import {
  BatchMigrationCreateComponentFieldInput,
  BatchMigrationCreateComponentInput,
  BatchMigrationCreateComponentUnionFieldInput,
  BatchMigrationCreateEnumerableFieldInput,
  BatchMigrationCreateEnumerationInput,
  BatchMigrationCreateLocaleInput,
  BatchMigrationCreateModelInput,
  BatchMigrationCreateRelationalFieldInput,
  BatchMigrationCreateSimpleFieldInput,
  BatchMigrationCreateStageInput,
  BatchMigrationCreateUnionFieldInput,
  Client,
} from '@hygraph/management-sdk'
import { Schema } from './types'

interface MigrationParams {
  name?: string
  endpoint: string
  authToken: string
}

export class UpsertClient extends Client {
  constructor(
    params: MigrationParams,
    private schema: Schema,
  ) {
    super(params)
  }

  upsertModel(data: BatchMigrationCreateModelInput) {
    const exists = this.schema.models.some((m) => m.apiId === data.apiId)
    return exists ? this.createModel(data) : this.updateModel(data)
  }

  upsertComponent(data: BatchMigrationCreateComponentInput) {
    const exists = this.schema.models.some((m) => m.apiId === data.apiId)
    return exists ? this.createComponent(data) : this.updateComponent(data)
  }

  upsertSimpleField(data: BatchMigrationCreateSimpleFieldInput) {
    const model = this.schema.models.find((m) => m.apiId === data.parentApiId)
    const exists = model?.fields.some((f) => f.apiId === data.apiId)
    return exists
      ? this.createSimpleField(data)
      : this.updateSimpleField({ ...data, embeddableModels: undefined })
  }

  // upsertRemoteField(data: BatchMigrationCreateRemoteFieldInput) {
  //   const model = this.schema.models.find((m) => m.apiId === data.parentApiId)
  //   const exists = model?.fields.some((f) => f.apiId === data.apiId)
  //   return exists ? this.createRemoteField(data) : this.updateRemoteField(data)
  // }

  upsertRelationalField(data: BatchMigrationCreateRelationalFieldInput) {
    const model = this.schema.models.find((m) => m.apiId === data.parentApiId)
    const exists = model?.fields.some((f) => f.apiId === data.apiId)
    return exists ? this.createRelationalField(data) : this.updateRelationalField(data)
  }

  upsertUnionField(data: BatchMigrationCreateUnionFieldInput) {
    const model = this.schema.models.find((m) => m.apiId === data.parentApiId)
    const exists = model?.fields.some((f) => f.apiId === data.apiId)
    return exists ? this.createUnionField(data) : this.updateUnionField(data)
  }

  upsertComponentField(data: BatchMigrationCreateComponentFieldInput) {
    const model = this.schema.models.find((m) => m.apiId === data.parentApiId)
    const exists = model?.fields.some((f) => f.apiId === data.apiId)
    return exists ? this.createComponentField(data) : this.updateComponentField(data)
  }

  upsertComponentUnionField(data: BatchMigrationCreateComponentUnionFieldInput) {
    const model = this.schema.models.find((m) => m.apiId === data.parentApiId)
    const exists = model?.fields.some((f) => f.apiId === data.apiId)
    return exists ? this.createComponentUnionField(data) : this.updateComponentUnionField(data)
  }

  upsertEnumeration(data: BatchMigrationCreateEnumerationInput) {
    const exists = this.schema.enumerations.some((e) => e.apiId === data.apiId)
    return exists ? this.createEnumeration(data) : this.updateEnumeration(data)
  }

  upsertEnumerableField(data: BatchMigrationCreateEnumerableFieldInput) {
    const model = this.schema.models.find((m) => m.apiId === data.parentApiId)
    const exists = model?.fields.some((f) => f.apiId === data.apiId)
    return exists ? this.createEnumerableField(data) : this.updateEnumerableField(data)
  }

  upsertStage(data: BatchMigrationCreateStageInput) {
    const exists = this.schema.stages.some((m) => m.apiId === data.apiId)
    return exists ? this.createStage(data) : this.updateStage(data)
  }

  upsertLocale(data: BatchMigrationCreateLocaleInput) {
    const exists = this.schema.locales.some((m) => m.apiId === data.apiId)
    return exists ? this.createLocale(data) : this.updateLocale(data)
  }
}