import { TrackedArray, TrackedObject } from "@ember-compat/tracked-built-ins";
import RestModel from "discourse/models/rest";

const CREATE_ATTRIBUTES = [
  "id",
  "name",
  "description",
  "parameters",
  "script",
  "summary",
  "enabled",
];

export default class AiTool extends RestModel {
  createProperties() {
    return this.getProperties(CREATE_ATTRIBUTES);
  }

  updateProperties() {
    return this.getProperties(CREATE_ATTRIBUTES);
  }

  workingCopy() {
    const attrs = this.getProperties(CREATE_ATTRIBUTES);

    attrs.parameters = new TrackedArray(
      attrs.parameters?.map((p) => {
        const parameter = new TrackedObject(p);

        //Backwards-compatibility code.
        // TODO(roman): Remove aug 2024. Leave only else clause.
        if (parameter.enum_values) {
          parameter.enum = new TrackedArray(parameter.enum_values);
          delete parameter.enum_values;
        } else {
          parameter.enum = new TrackedArray(parameter.enum);
        }

        return parameter;
      })
    );

    return this.store.createRecord("ai-tool", attrs);
  }
}