<template>
  <Dialog :title="dialogTitle" v-model="dialogVisible">
    <el-form
      ref="formRef"
      :model="formData"
      :rules="formRules"
      label-width="100px"
      v-loading="formLoading"
    >
      <el-form-item label="向量模型编号" prop="embeddingModelId">
        <el-select
          v-model="formData.embeddingModelId"
          placeholder="请选择向量模型"
          clearable
          @change="handleModelChange"
        >
          <el-option
            v-for="model in models"
            :key="model.id"
            :label="model.name"
            :value="model.id"
          />
        </el-select>
      </el-form-item>
      <el-form-item label="业务名称" prop="name">
        <el-input v-model="formData.name" placeholder="请输入业务名称" />
      </el-form-item>
      <el-form-item label="数据源" prop="datasourceId">
        <el-select v-model="formData.datasourceId" placeholder="请选择数据源" clearable>
          <el-option
            v-for="item in datasources"
            :key="item.id"
            :label="item.name"
            :value="item.id"
          />
        </el-select>
      </el-form-item>
      <el-form-item label="表名" prop="tableNames">
        <el-select v-model="formData.tableNames" placeholder="请选择表名" clearable multiple>
          <el-option v-for="item in tableNames" :key="item" :label="item" :value="item" />
        </el-select>
      </el-form-item>
    </el-form>
    <template #footer>
      <el-button @click="submitForm" type="primary" :disabled="formLoading">确 定</el-button>
      <el-button @click="dialogVisible = false">取 消</el-button>
    </template>
  </Dialog>
</template>
<script setup lang="ts">
import { Nl2sqlApi, Nl2sqlVO } from '@/api/ai/nl2sql'
import { AiModelTypeEnum } from '@/views/ai/utils/constants'
import { ModelApi, ModelVO } from '@/api/ai/model/model'
import * as DataSourceConfigApi from '@/api/infra/dataSourceConfig'

/** nl2sql配置 表单 */
defineOptions({ name: 'ChatNl2sqlForm' })

const { t } = useI18n() // 国际化
const message = useMessage() // 消息弹窗

const dialogVisible = ref(false) // 弹窗的是否展示
const dialogTitle = ref('') // 弹窗的标题
const formLoading = ref(false) // 表单的加载中：1）修改时的数据加载；2）提交的按钮禁用
const formType = ref('') // 表单的类型：create - 新增；update - 修改
const models = ref([] as ModelVO[]) // 聊天模型列表
const datasources = ref<DataSourceConfigApi.DataSourceConfigVO[]>([]) // 数据源列表
const tableNames = ref<string[]>([]) // 数据源列表
const formData = ref({
  id: undefined,
  embeddingModelId: undefined,
  embeddingModel: '',
  name: undefined,
  datasourceId: undefined,
  tableNames: []
})
const formRules = reactive({
  embeddingModelId: [{ required: true, message: '向量模型编号不能为空', trigger: 'blur' }],
  name: [{ required: true, message: '业务名称不能为空', trigger: 'blur' }],
  datasourceId: [{ required: true, message: '数据源不能为空', trigger: 'blur' }],
  tableNames: [{ required: true, message: '表名不能为空', trigger: 'blur' }]
})
const formRef = ref() // 表单 Ref

/** 打开弹窗 */
const open = async (type: string, id?: number) => {
  dialogVisible.value = true
  dialogTitle.value = t('action.' + type)
  formType.value = type
  resetForm()
  formLoading.value = false
  // 修改时，设置数据
  if (id) {
    formLoading.value = true
    try {
      formData.value = await Nl2sqlApi.getChatNl2sql(id)
    } finally {
      formLoading.value = false
    }
  }
  models.value = await ModelApi.getModelSimpleList(AiModelTypeEnum.EMBEDDING)
  datasources.value = await DataSourceConfigApi.getDataSourceConfigList()
}
const handleModelChange = (selectedId: number | undefined) => {
  if (!selectedId) {
    formData.value.embeddingModel = ''
    return
  }
  const selectedModel = models.value.find((model) => model.id === selectedId)
  formData.value.embeddingModel = selectedModel ? selectedModel.name : ''
}
defineExpose({ open }) // 提供 open 方法，用于打开弹窗
/** 加载表名列表 */
const loadTableNames = async () => {
  if (formData.value.datasourceId === undefined || formData.value.datasourceId === null) {
    tableNames.value = []
    return
  }

  try {
    formLoading.value = true
    // 这里调用API获取表名列表，假设后端有提供这样的接口
    tableNames.value = await DataSourceConfigApi.getTableNames(formData.value.datasourceId)
  } catch (error) {
    message.error('获取表名列表失败: ' + error.message)
    tableNames.value = []
  } finally {
    formLoading.value = false
  }
}
// 监听数据源变化，自动加载表名
watch(
  () => formData.value.datasourceId,
  async (newVal) => {
    if (newVal != null && newVal !== '') {
      await loadTableNames()
    } else {
      formData.value.tableNames = []
      tableNames.value = []
    }
  }
)
/** 提交表单 */
const emit = defineEmits(['success']) // 定义 success 事件，用于操作成功后的回调
const submitForm = async () => {
  // 校验表单
  await formRef.value.validate()
  // 提交请求
  formLoading.value = true
  try {
    const data = formData.value as unknown as Nl2sqlVO
    if (formType.value === 'create') {
      await Nl2sqlApi.createChatNl2sql(data)
      message.success(t('common.createSuccess'))
    } else {
      await Nl2sqlApi.updateChatNl2sql(data)
      message.success(t('common.updateSuccess'))
    }
    dialogVisible.value = false
    // 发送操作成功的事件
    emit('success')
  } finally {
    formLoading.value = false
  }
}

/** 重置表单 */
const resetForm = () => {
  formData.value = {
    id: undefined,
    embeddingModelId: undefined,
    embeddingModel: undefined,
    name: undefined,
    datasourceId: undefined,
    tableNames: []
  }
  formRef.value?.resetFields()
}
</script>
