feat: create the v059x node-override.yaml via solo

src/commands/node/handlers.ts

@@ -122,7 +122,7 @@ export class NodeCommandHandlers implements CommandHandlers {
       this.tasks.checkNodePodsAreRunning(),
       this.tasks.populateServiceMap(),
       this.tasks.fetchPlatformSoftware('allNodeAliases'),
-      this.tasks.setupNetworkNodes('allNodeAliases'),
+      this.tasks.setupNetworkNodes('allNodeAliases', false),
       this.tasks.startNodes('allNodeAliases'),
       this.tasks.enablePortForwarding(),
       this.tasks.checkAllNodesAreActive('allNodeAliases'),
@@ -173,7 +173,7 @@ export class NodeCommandHandlers implements CommandHandlers {
       this.tasks.fetchPlatformSoftware('allNodeAliases'),
       this.tasks.downloadLastState(),
       this.tasks.uploadStateToNewNode(),
-      this.tasks.setupNetworkNodes('allNodeAliases'),
+      this.tasks.setupNetworkNodes('allNodeAliases', false),
       this.tasks.startNodes('allNodeAliases'),
       this.tasks.enablePortForwarding(),
       this.tasks.checkAllNodesAreActive('allNodeAliases'),
@@ -218,7 +218,7 @@ export class NodeCommandHandlers implements CommandHandlers {
       this.tasks.killNodesAndUpdateConfigMap(),
       this.tasks.checkNodePodsAreRunning(),
       this.tasks.fetchPlatformSoftware('allNodeAliases'),
-      this.tasks.setupNetworkNodes('allNodeAliases'),
+      this.tasks.setupNetworkNodes('allNodeAliases', false),
       this.tasks.startNodes('allNodeAliases'),
       this.tasks.enablePortForwarding(),
       this.tasks.checkAllNodesAreActive('allNodeAliases'),
@@ -717,7 +717,7 @@ export class NodeCommandHandlers implements CommandHandlers {
         this.tasks.identifyNetworkPods(),
         this.tasks.dumpNetworkNodesSaveState(),
         this.tasks.fetchPlatformSoftware('nodeAliases'),
-        this.tasks.setupNetworkNodes('nodeAliases'),
+        this.tasks.setupNetworkNodes('nodeAliases', true),
         this.tasks.startNodes('nodeAliases'),
         this.tasks.checkAllNodesAreActive('nodeAliases'),
         this.tasks.checkNodeProxiesAreActive(),
@@ -826,8 +826,7 @@ export class NodeCommandHandlers implements CommandHandlers {
         }),
         this.tasks.identifyNetworkPods(),
         this.tasks.fetchPlatformSoftware('nodeAliases'),
-        // TODO: change to isGenesis: true once we are ready to use genesis-network.json: this.tasks.setupNetworkNodes('nodeAliases', true),
-        this.tasks.setupNetworkNodes('nodeAliases', false),
+        this.tasks.setupNetworkNodes('nodeAliases', true),
         this.changeAllNodeStates(ConsensusNodeStates.SETUP),
       ],
       {

src/commands/node/tasks.ts

@@ -61,11 +61,13 @@
 import {type BaseCommand} from '../base.js';
 import {type NodeAddConfigClass} from './node_add_config.js';
 import {GenesisNetworkDataConstructor} from '../../core/genesis_network_models/genesis_network_data_constructor.js';
+import {NodeOverridesModel} from '../../core/node_overrides_model.js';
 import {type NamespaceName} from '../../core/kube/namespace_name.js';
 import {PodRef} from '../../core/kube/pod_ref.js';
 import {ContainerRef} from '../../core/kube/container_ref.js';
 import {NetworkNodes} from '../../core/network_nodes.js';
 import {container} from 'tsyringe-neo';
+import * as helpers from '../../core/helpers.js';
 
 export class NodeCommandTasks {
   private readonly accountManager: AccountManager;
@@ -903,8 +905,9 @@
     });
   }
 
-  setupNetworkNodes(nodeAliasesProperty: string, isGenesis: boolean = false) {
+  setupNetworkNodes(nodeAliasesProperty: string, isGenesis: boolean) {
     return new Task('Setup network nodes', async (ctx: any, task: ListrTaskWrapper<any, any, any>) => {
+      ctx.config.nodeAliases = helpers.parseNodeAliases(ctx.config.nodeAliasesUnparsed);
       if (isGenesis) {
         await this.generateGenesisNetworkJson(
           ctx.config.namespace,
@@ -914,6 +917,8 @@
         );
       }
 
+      await this.generateNodeOverridesJson(ctx.config.namespace, ctx.config.nodeAliases, ctx.config.stagingDir);
+
       const subTasks = [];
       for (const nodeAlias of ctx.config[nodeAliasesProperty]) {
         const podRef = ctx.config.podRefs[nodeAlias];
@@ -931,6 +936,15 @@
     });
   }
 
+  private async generateNodeOverridesJson(namespace: NamespaceName, nodeAliases: NodeAliases, stagingDir: string) {
+    const networkNodeServiceMap = await this.accountManager.getNodeServiceMap(namespace);
+
+    const nodeOverridesModel = new NodeOverridesModel(nodeAliases, networkNodeServiceMap);
+
+    const nodeOverridesJson = path.join(stagingDir, constants.NODE_OVERRIDE_FILE);
+    fs.writeFileSync(nodeOverridesJson, nodeOverridesModel.toYAML());
+  }
+
   /**
    * Generate genesis network json file
    * @private
@@ -951,6 +965,7 @@
     const genesisNetworkData = await GenesisNetworkDataConstructor.initialize(
       nodeAliases,
       this.keyManager,
+      this.accountManager,
       keysDir,
       networkNodeServiceMap,
       adminPublicKeys,

src/core/constants.ts

@@ -204,6 +204,7 @@ export const GRPC_PORT = +process.env.GRPC_PORT || 50_211;
 export const NETWORK_DESTROY_WAIT_TIMEOUT = +process.env.NETWORK_DESTROY_WAIT_TIMEOUT || 120;
 
 export const DEFAULT_LOCAL_CONFIG_FILE = 'local-config.yaml';
+export const NODE_OVERRIDE_FILE = 'node-overrides.yaml';
 export const IGNORED_NODE_ACCOUNT_ID = '0.0.0';
 
 export const UPLOADER_SECRET_NAME = 'uploader-mirror-secrets';

src/core/genesis_network_models/genesis_network_data_constructor.ts

@@ -14,56 +14,78 @@
 import {type NetworkNodeServices} from '../network_node_services.js';
 import {SoloError} from '../errors.js';
 import {Flags as flags} from '../../commands/flags.js';
+import {type AccountManager} from '../account_manager.js';
 
 /**
  * Used to construct the nodes data and convert them to JSON
  */
 export class GenesisNetworkDataConstructor implements ToJSON {
   public readonly nodes: Record<NodeAlias, GenesisNetworkNodeDataWrapper> = {};
   public readonly rosters: Record<NodeAlias, GenesisNetworkRosterEntryDataWrapper> = {};
-
+  private readonly initializationPromise: Promise<void>;
   private constructor(
     private readonly nodeAliases: NodeAliases,
     private readonly keyManager: KeyManager,
+    private readonly accountManager: AccountManager,
     private readonly keysDir: string,
     private readonly networkNodeServiceMap: Map<string, NetworkNodeServices>,
     adminPublicKeyMap: Map<NodeAlias, string>,
   ) {
-    nodeAliases.forEach(nodeAlias => {
-      const genesisPrivateKey = PrivateKey.fromStringED25519(constants.GENESIS_KEY);
-      const adminPubKey = PublicKey.fromStringED25519(adminPublicKeyMap[nodeAlias])
-        ? adminPublicKeyMap[nodeAlias]
-        : genesisPrivateKey.publicKey;
-
-      const nodeDataWrapper = new GenesisNetworkNodeDataWrapper(
-        +networkNodeServiceMap.get(nodeAlias).nodeId,
-        adminPubKey,
-        nodeAlias,
-      );
-      this.nodes[nodeAlias] = nodeDataWrapper;
-      nodeDataWrapper.accountId = AccountId.fromString(networkNodeServiceMap.get(nodeAlias).accountId);
-
-      const rosterDataWrapper = new GenesisNetworkRosterEntryDataWrapper(+networkNodeServiceMap.get(nodeAlias).nodeId);
-      this.rosters[nodeAlias] = rosterDataWrapper;
-      rosterDataWrapper.weight = this.nodes[nodeAlias].weight = constants.HEDERA_NODE_DEFAULT_STAKE_AMOUNT;
-
-      const externalPort = +constants.HEDERA_NODE_EXTERNAL_GOSSIP_PORT;
-      const namespace = networkNodeServiceMap.get(nodeAlias).namespace;
-      const externalIP = Templates.renderFullyQualifiedNetworkSvcName(namespace, nodeAlias);
-      // Add gossip endpoints
-      nodeDataWrapper.addGossipEndpoint(externalIP, externalPort);
-      rosterDataWrapper.addGossipEndpoint(externalIP, externalPort);
-
-      const haProxyFqdn = Templates.renderFullyQualifiedHaProxyName(nodeAlias, namespace);
-
-      // Add service endpoints
-      nodeDataWrapper.addServiceEndpoint(haProxyFqdn, constants.GRPC_PORT);
-    });
+    this.initializationPromise = (async () => {
+      nodeAliases.forEach(nodeAlias => {
+        let adminPubKey: PublicKey;
+        const accountId = AccountId.fromString(networkNodeServiceMap.get(nodeAlias).accountId);
+        const namespace = networkNodeServiceMap.get(nodeAlias).namespace;
+
+        if (adminPublicKeyMap.has(nodeAlias)) {
+          try {
+            if (PublicKey.fromStringED25519(adminPublicKeyMap[nodeAlias])) {
+              adminPubKey = adminPublicKeyMap[nodeAlias];
+            }
+          } catch {
+            // Ignore error
+          }
+        }
+
+        // not found existing one, generate a new key, and save to k8 secret
+        if (!adminPubKey) {
+          const newKey = PrivateKey.generate();
+          adminPubKey = newKey.publicKey;
+          this.accountManager.updateAccountKeys(namespace, accountId, newKey, true);
+        }
+
+        const nodeDataWrapper = new GenesisNetworkNodeDataWrapper(
+          +networkNodeServiceMap.get(nodeAlias).nodeId,
+          adminPubKey,
+          nodeAlias,
+        );
+        this.nodes[nodeAlias] = nodeDataWrapper;
+        nodeDataWrapper.accountId = accountId;
+
+        const rosterDataWrapper = new GenesisNetworkRosterEntryDataWrapper(
+          +networkNodeServiceMap.get(nodeAlias).nodeId,
+        );
+        this.rosters[nodeAlias] = rosterDataWrapper;
+        rosterDataWrapper.weight = this.nodes[nodeAlias].weight = constants.HEDERA_NODE_DEFAULT_STAKE_AMOUNT;
+
+        const externalPort = +constants.HEDERA_NODE_EXTERNAL_GOSSIP_PORT;
+        const externalIP = Templates.renderFullyQualifiedNetworkSvcName(namespace, nodeAlias);
+        // Add gossip endpoints
+        nodeDataWrapper.addGossipEndpoint(externalIP, externalPort);
+        rosterDataWrapper.addGossipEndpoint(externalIP, externalPort);
+
+        const haProxyFqdn = Templates.renderFullyQualifiedHaProxyName(nodeAlias, namespace);
+
+        // Add service endpoints
+        nodeDataWrapper.addServiceEndpoint(haProxyFqdn, constants.GRPC_PORT);
+      });
+    })();
   }
 
   public static async initialize(
     nodeAliases: NodeAliases,
     keyManager: KeyManager,
+    accountManager: AccountManager,
     keysDir: string,
     networkNodeServiceMap: Map<string, NetworkNodeServices>,
     adminPublicKeys: string[],
@@ -86,6 +108,7 @@
     const instance = new GenesisNetworkDataConstructor(
       nodeAliases,
       keyManager,
+      accountManager,
       keysDir,
       networkNodeServiceMap,
       adminPublicKeyMap,
@@ -100,6 +123,7 @@
    * Loads the gossipCaCertificate and grpcCertificateHash
    */
   private async load() {
+    await this.initializationPromise;
     await Promise.all(
       this.nodeAliases.map(async nodeAlias => {
         const signingCertFile = Templates.renderGossipPemPublicKeyFile(nodeAlias);

src/core/node_overrides_model.ts

@@ -0,0 +1,53 @@
+/**
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as yaml from 'yaml';
+import * as constants from './constants.js';
+import {Templates} from './templates.js';
+import {type NodeAliases} from '../types/aliases.js';
+import {type NetworkNodeServices} from './network_node_services.js';
+import {type GossipEndpoint} from '../types/index.js';
+
+export class NodeOverridesModel {
+  private readonly interfaceBindings: GossipEndpoint[] = [];
+  private readonly endpointOverrides: GossipEndpoint[] = [];
+
+  public constructor(nodeAliases: NodeAliases, networkNodeServiceMap: Map<string, NetworkNodeServices>) {
+    nodeAliases.forEach(nodeAlias => {
+      const nodeId = +networkNodeServiceMap.get(nodeAlias).nodeId;
+
+      const localClusterPort = +constants.HEDERA_NODE_EXTERNAL_GOSSIP_PORT;
+      const localClusterHostName = Templates.renderNetworkHeadlessSvcName(nodeAlias);
+
+      this.interfaceBindings.push({nodeId, hostname: localClusterHostName, port: localClusterPort});
+      // TODO future, add endpointOverrides for addresses external to cluster in multi-cluster support situation
+      //  this.endpointOverrides.push({nodeId, hostname: externalHostname, port: externalPort});
+    });
+  }
+
+  /**
+   * Converts the model to YAML as expected to be consumed inside node
+   * @returns the raw YAML as string
+   *
+   * @example
+   * gossip:
+   *   interfaceBindings:
+   *     - { "nodeId": 0, "hostname": "10.10.10.1", "port": 1234 }
+   *     - { "nodeId": 3, "hostname": "2001:db8:3333:4444:5555:6666:7777:8888", "port": 1237 }
+   *   endpointOverrides:
+   *     - { "nodeId": 5, "hostname": "10.10.10.11", "port": 1238 }
+   */
+  public toYAML(): string {
+    const gossipData: {interfaceBindings?: string[]; endpointOverrides?: string[]} = {};
+
+    if (this.interfaceBindings.length) {
+      gossipData.interfaceBindings = this.interfaceBindings.map(d => JSON.stringify(d));
+    }
+
+    if (this.endpointOverrides.length) {
+      gossipData.endpointOverrides = this.endpointOverrides.map(d => JSON.stringify(d));
+    }
+
+    return yaml.stringify({gossip: gossipData}).replaceAll(/'/g, '');
+  }
+}

src/core/platform_installer.ts

@@ -305,6 +305,9 @@
       const genesisNetworkJson = [path.join(stagingDir, 'genesis-network.json')];
       await this.copyFiles(podRef, genesisNetworkJson, `${constants.HEDERA_HAPI_PATH}/data/config`);
     }
+
+    const nodeOverridesYaml = [path.join(stagingDir, constants.NODE_OVERRIDE_FILE)];
+    await this.copyFiles(podRef, nodeOverridesYaml, `${constants.HEDERA_HAPI_PATH}/data/config`);
   }
 
   /**

src/core/templates.ts

@@ -26,7 +26,7 @@ export class Templates {
     return svcName.split('-').slice(1, -1).join('-') as NodeAlias;
   }
 
-  private static renderNetworkHeadlessSvcName(nodeAlias: NodeAlias): string {
+  public static renderNetworkHeadlessSvcName(nodeAlias: NodeAlias): string {
     return `network-${nodeAlias}`;
   }
 

src/types/index.ts

@@ -115,3 +115,14 @@ export interface GenesisNetworkRosterStructure {
   gossipEndpoint: ServiceEndpoint[];
   gossipCaCertificate: string;
 }
+
+export interface GossipEndpoint {
+  nodeId: number;
+  hostname: string;
+  port: number;
+}
+
+export interface NodeOverridesStructure {
+  interfaceBindings: GossipEndpoint[];
+  endpointOverrides: GossipEndpoint[];
+}