doccano/frontend/domain/models/autoLabeling/template.ts

import { Fields, ParametersForUI } from '~/domain/models/autoLabeling/config'

export interface Schema {
  title: string,
  type: string,
  properties: object
}

export interface ConfigResponse {
  name: string,
  schema: Schema,
  template: string
}

export class ConfigTemplateItem {
  constructor(
    private schema: Schema,
    public template: string
  ) {}

  static valueOf(
    { schema, template }:
    { schema: Schema, template: string }
  ): ConfigTemplateItem {
    return new ConfigTemplateItem(schema, template)
  }

  get modelName(): string {
    return this.schema.title
  }

  get fields(): ParametersForUI[] {
    const response: ParametersForUI[] = []
    for (const [key, value] of Object.entries(this.schema.properties)) {
      if ('type' in value && value.type === 'string') {
        response.push({name: key, type: 'textField', value: ''})
      } else if ('anyOf' in value) {
        response.push(
          {
            name: key,
            type: 'selectField',
            value: '',
            items: value.anyOf.map(
              (item: {'const': string, 'type': string}) => item.const
            )
          }
        )
      } else if ('type' in value && value.type === 'object') {
        response.push(
          {
            name: key,
            type: 'objectField',
            value: []
          }
        )
      }
    }
    return response
  }

  toObject(): Fields {
    return {
      modelName: this.modelName,
      template: this.template,
      modelAttrs: this.fields,
      labelMapping: [],
      taskType: ''
    }
  }
}