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

export class ConfigItemList {
  constructor(public configItems: ConfigItem[]) {}

  static valueOf(items: ConfigItem[]): ConfigItemList {
    return new ConfigItemList(items)
  }

  toArray(): Object[] {
    return this.configItems.map(item => item.toObject())
  }
}

interface LabelMappingForUI {
  from: string,
  to: string
}

export interface ParametersForUI {
  name: string,
  value: string | object[],
  type?: string,
  items?: string[]
}

export interface Fields {
  modelName: string,
  modelAttrs: ParametersForUI[],
  template: string,
  labelMapping: LabelMappingForUI[],
  taskType: string
}

export class ConfigItem {
  constructor(
    public id: number,
    public modelName: string,
    public modelAttrs: object,
    public template: string,
    public labelMapping: object,
    public taskType: string,
  ) {}

  static valueOf(
    { id, model_name, model_attrs, template, label_mapping, task_type }:
    { id: number, model_name: string, model_attrs: object, template: string, label_mapping: object, task_type: string }
  ): ConfigItem {
    return new ConfigItem(id, model_name, model_attrs, template, label_mapping, task_type)
  }

  static parseFromUI(
    { modelName, modelAttrs, template, labelMapping, taskType }: Fields): ConfigItem {
    const mapping = labelMapping.reduce((a, x) => ({...a, [x.from]: x.to}), {})
    const attributes: {[key: string]: any} = modelAttrs.reduce((a, x) => ({...a, [x.name]: x.value}), {})
    for (const [key, value] of Object.entries(attributes)) {
      if (Array.isArray(value)) {
        attributes[key] = value.reduce((a, x) => ({...a, [x.key]: x.value}), {})
      }
    }
    return new ConfigItem(99999, modelName, attributes, template, mapping, taskType)
  }

  toObject(): object {
    return {
      id: this.id,
      modelName: this.modelName,
      modelAttrs: this.modelAttrs,
      template: this.template,
      labelMapping: this.labelMapping,
      taskType: this.taskType
    }
  }

  toAPI(): object {
    return {
      id: this.id,
      model_name: this.modelName,
      model_attrs: this.modelAttrs,
      template: this.template,
      label_mapping: this.labelMapping,
      task_type: this.taskType
    }
  }
}