joystream/storage-node-v2/src/commands/leader/update-dynamic-bag-policy.ts

import ApiCommandBase from '../../command-base/ApiCommandBase'
import { updateNumberOfStorageBucketsInDynamicBagCreationPolicy } from '../../services/runtime/extrinsics'
import { flags } from '@oclif/command'
import logger from '../../services/logger'
import { parseDynamicBagType } from '../../services/helpers/bagTypes'
import { DynamicBagTypeKey } from '@joystream/types/storage'

/**
 * CLI command:
 * Updates dynamic bag creation policy - storage bucket number for new dynamic
 * bag.
 *
 * @remarks
 * Storage working group leader command. Requires storage WG leader priviliges.
 * Shell command: "leader:update-dynamic-bag-policy"
 */
export default class LeaderUpdateDynamicBagPolicy extends ApiCommandBase {
  static description = 'Update number of storage buckets used in the dynamic bag creation policy.'

  static flags = {
    number: flags.integer({
      char: 'n',
      required: true,
      description: 'New storage buckets number',
    }),
    bagType: flags.enum<DynamicBagTypeKey>({
      char: 't',
      description: 'Dynamic bag type (Channel, Member).',
      options: ['Channel', 'Member'],
      required: true,
    }),
    ...ApiCommandBase.flags,
  }

  async run(): Promise<void> {
    const { flags } = this.parse(LeaderUpdateDynamicBagPolicy)

    logger.info('Update dynamic bag creation policy....')
    if (flags.dev) {
      await this.ensureDevelopmentChain()
    }

    const account = this.getAccount(flags)
    const newNumber = flags.number

    const api = await this.getApi()
    const dynamicBagType = parseDynamicBagType(flags.bagType)
    const success = await updateNumberOfStorageBucketsInDynamicBagCreationPolicy(
      api,
      account,
      dynamicBagType,
      newNumber
    )

    this.exitAfterRuntimeCall(success)
  }
}