Add '--mirror-node-version' flag to override default mirror node version

Signed-off-by: instamenta <[email protected]>

src/commands/flags.mjs

@@ -755,6 +755,16 @@ export const adminKey = {
  }
 }
 
+/** @type {CommandFlag} **/
+export const mirrorNodeVersion = {
+ constName: 'mirrorNodeVersion',
+ name: 'mirror-node-version',
+ definition: {
+ describe: 'Mirror node chart version',
+ defaultValue: '',
+ type: 'string'
+ }
+}
 /** @type {CommandFlag[]} **/
 export const allFlags = [
  accountId,
@@ -815,7 +825,8 @@ export const allFlags = [
  tlsPrivateKey,
  tlsPublicKey,
  updateAccountKeys,
- valuesFile
+ valuesFile,
+ mirrorNodeVersion
 ]
 
 /**

src/commands/mirror_node.mjs

@@ -60,7 +60,8 @@ export class MirrorNodeCommand extends BaseCommand {
  flags.profileFile,
  flags.profileName,
  flags.tlsClusterIssuerType,
- flags.valuesFile
+ flags.valuesFile,
+ flags.mirrorNodeVersion
  ]
  }
 
@@ -121,6 +122,10 @@ export class MirrorNodeCommand extends BaseCommand {
  config.hederaExplorerTlsLoadBalancerIp, config.hederaExplorerTlsHostName)
  }
 
+ if (config.mirrorNodeVersion) {
+ valuesArg += ` --set global.image.tag=${config.mirrorNodeVersion}`
+ }
+
  valuesArg += ` --set hedera-mirror-node.enabled=true --set hedera-explorer.enabled=${config.deployHederaExplorer}`
  return valuesArg
  }
@@ -147,7 +152,8 @@ export class MirrorNodeCommand extends BaseCommand {
  flags.hederaExplorerTlsHostName,
  flags.hederaExplorerTlsLoadBalancerIp,
  flags.tlsClusterIssuerType,
- flags.valuesFile
+ flags.valuesFile,
+ flags.mirrorNodeVersion
  ])
 
  await prompts.execute(task, self.configManager, MirrorNodeCommand.DEPLOY_FLAGS_LIST)
@@ -169,6 +175,7 @@ export class MirrorNodeCommand extends BaseCommand {
  * -- extra args --
  * @property {string} chartPath
  * @property {string} valuesArg
+ * @property {string} mirrorNodeVersion
  * -- methods --
  * @property {getUnusedConfigs} getUnusedConfigs
  */

src/commands/prompts.mjs

@@ -421,6 +421,14 @@ export async function promptPersistentVolumeClaims (task, input) {
  flags.persistentVolumeClaims.name)
 }
 
+export async function promptMirrorNodeVersion (task, input) {
+ return await promptToggle(task, input,
+ flags.mirrorNodeVersion.definition.defaultValue,
+ 'Would you like to choose mirror node version? ',
+ null,
+ flags.mirrorNodeVersion.name)
+}
+
 /**
  * @returns {Map<string, Function>}
  */
@@ -464,6 +472,7 @@ export function getPromptMap () {
  .set(flags.gossipEndpoints.name, promptGossipEndpoints)
  .set(flags.grpcEndpoints.name, promptGrpcEndpoints)
  .set(flags.endpointType.name, promptEndpointType)
+ .set(flags.mirrorNodeVersion.name, promptMirrorNodeVersion)
 }
 
 // build the prompt registry

src/core/chart_manager.mjs

@@ -39,28 +39,36 @@ export class ChartManager {
  *
  * @param {Map<string, string>} repoURLs - a map of name and chart repository URLs
  * @param {boolean} force - whether or not to update the repo
- * @returns {Promise<string[]>}
+ * @returns {Promise<string[]>} - returns the urls
  */
  async setup (repoURLs = constants.DEFAULT_CHART_REPO, force = true) {
  try {
- let forceUpdateArg = ''
- if (force) {
- forceUpdateArg = '--force-update'
- }
+ const forceUpdateArg = force ? '--force-update' : ''
 
- const urls = []
+ /** @type {Array<Promise<string>>} */
+ const promises = []
  for (const [name, url] of repoURLs.entries()) {
- this.logger.debug(`Adding repo ${name} -> ${url}`, { repoName: name, repoURL: url })
- await this.helm.repo('add', name, url, forceUpdateArg)
- urls.push(url)
+ promises.push(this.addRepo(name, url, forceUpdateArg))
  }
 
- return urls
+ return await Promise.all(promises) // urls
  } catch (e) {
  throw new FullstackTestingError(`failed to setup chart repositories: ${e.message}`, e)
  }
  }
 
+ /**
+ * @param {string} name
+ * @param {string} url
+ * @param {string} forceUpdateArg
+ * @returns {Promise<string>}
+ */
+ async addRepo (name, url, forceUpdateArg) {
+ this.logger.debug(`Adding repo ${name} -> ${url}`, { repoName: name, repoURL: url })
+ await this.helm.repo('add', name, url, forceUpdateArg)
+ return url
+ }
+
  /**
  * List available clusters
  * @param {string} namespaceName