joystream/cli/src/commands/content-directory/addClassSchema.ts

import ContentDirectoryCommandBase from '../../base/ContentDirectoryCommandBase'
import AddClassSchemaSchema from '@joystream/cd-schemas/schemas/extrinsics/AddClassSchema.schema.json'
import { AddClassSchema } from '@joystream/cd-schemas/types/extrinsics/AddClassSchema'
import { InputParser } from '@joystream/cd-schemas'
import { JsonSchemaPrompter, JsonSchemaCustomPrompts } from '../../helpers/JsonSchemaPrompt'
import { JSONSchema } from '@apidevtools/json-schema-ref-parser'
import { IOFlags, getInputJson, saveOutputJson } from '../../helpers/InputOutput'
import { Class } from '@joystream/types/content-directory'

export default class AddClassSchemaCommand extends ContentDirectoryCommandBase {
  static description = 'Add a new schema to a class inside content directory. Requires lead access.'

  static flags = {
    ...IOFlags,
  }

  async run() {
    const account = await this.getRequiredSelectedAccount()
    await this.requireLead()
    await this.requestAccountDecoding(account)

    const { input, output } = this.parse(AddClassSchemaCommand).flags

    let inputJson = await getInputJson<AddClassSchema>(input)
    if (!inputJson) {
      let selectedClass: Class | undefined
      const customPrompts: JsonSchemaCustomPrompts = [
        [
          'className',
          async () => {
            selectedClass = await this.promptForClass('Select a class to add schema to')
            return selectedClass.name.toString()
          },
        ],
        [
          'existingProperties',
          async () => {
            const choices = selectedClass!.properties.map((p, i) => ({ name: `${i}: ${p.name.toString()}`, value: i }))
            if (!choices.length) {
              return []
            }
            return await this.simplePrompt({
              type: 'checkbox',
              message: 'Choose existing properties to keep',
              choices,
            })
          },
        ],
        [
          /^newProperties\[\d+\]\.property_type\.(Single|Vector\.vec_type)\.Reference/,
          async () => this.promptForClassReference(),
        ],
        [/^newProperties\[\d+\]\.property_type\.(Single|Vector\.vec_type)\.Text/, { message: 'Provide TextMaxLength' }],
        [
          /^newProperties\[\d+\]\.property_type\.(Single|Vector\.vec_type)\.Hash/,
          { message: 'Provide HashedTextMaxLength' },
        ],
      ]

      const prompter = new JsonSchemaPrompter<AddClassSchema>(
        AddClassSchemaSchema as JSONSchema,
        undefined,
        customPrompts
      )

      inputJson = await prompter.promptAll()
    }

    this.jsonPrettyPrint(JSON.stringify(inputJson))
    const confirmed = await this.simplePrompt({ type: 'confirm', message: 'Do you confirm the provided input?' })

    if (confirmed) {
      saveOutputJson(output, `${inputJson.className}Schema.json`, inputJson)
      const inputParser = new InputParser(this.getOriginalApi())
      this.log('Sending the extrinsic...')
      await this.sendAndFollowTx(account, await inputParser.parseAddClassSchemaExtrinsic(inputJson))
    }
  }
}