<template>
  <div class="p-2">
    <transition :enter-active-class="proxy?.animate.searchAnimate.enter" :leave-active-class="proxy?.animate.searchAnimate.leave">
      <div v-show="showSearch" class="mb-[10px]">
        <el-card shadow="hover">
          <el-form ref="queryFormRef" :model="queryParams" :inline="true">
            <el-form-item label="知识库ID" prop="knowledgeCode">
              <el-input v-model="queryParams.knowledgeCode" placeholder="请输入知识库ID" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="知识库名称" prop="knowledgeName">
              <el-input v-model="queryParams.knowledgeName" placeholder="请输入知识库名称" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="描述" prop="description">
              <el-input v-model="queryParams.description" placeholder="请输入描述" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="用户ID" prop="userId">
              <el-input v-model="queryParams.userId" placeholder="请输入用户ID" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="是否公开知识库" prop="share">
              <el-input v-model="queryParams.share" placeholder="请输入是否公开知识库" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="知识分隔符" prop="knowledgeSeparator">
              <el-input v-model="queryParams.knowledgeSeparator" placeholder="请输入知识分隔符" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="提问分隔符" prop="questionSeparator">
              <el-input v-model="queryParams.questionSeparator" placeholder="请输入提问分隔符" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="重叠字符数" prop="overlapChar">
              <el-input v-model="queryParams.overlapChar" placeholder="请输入重叠字符数" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="知识库中检索的条数" prop="retrieveLimit">
              <el-input v-model="queryParams.retrieveLimit" placeholder="请输入知识库中检索的条数" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="文本块大小" prop="textBlockSize">
              <el-input v-model="queryParams.textBlockSize" placeholder="请输入文本块大小" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="向量库" prop="vectorModelName">
              <el-input v-model="queryParams.vectorModelName" placeholder="请输入向量库" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="向量模型" prop="embeddingModelName">
              <el-input v-model="queryParams.embeddingModelName" placeholder="请输入向量模型" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="系统提示词" prop="systemPrompt">
              <el-input v-model="queryParams.systemPrompt" placeholder="请输入系统提示词" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item>
              <el-button type="primary" icon="Search" @click="handleQuery">搜索</el-button>
              <el-button icon="Refresh" @click="resetQuery">重置</el-button>
            </el-form-item>
          </el-form>
        </el-card>
      </div>
    </transition>

    <el-card shadow="never">
      <template #header>
        <el-row :gutter="10" class="mb8">
          <el-col :span="1.5">
            <el-button type="primary" plain icon="Plus" @click="handleAdd" v-hasPermi="['system:knowledgeInfo:add']">新增</el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button type="success" plain icon="Edit" :disabled="single" @click="handleUpdate()" v-hasPermi="['system:knowledgeInfo:edit']">修改</el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button type="danger" plain icon="Delete" :disabled="multiple" @click="handleDelete()" v-hasPermi="['system:knowledgeInfo:remove']">删除</el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button type="warning" plain icon="Download" @click="handleExport" v-hasPermi="['system:knowledgeInfo:export']">导出</el-button>
          </el-col>
          <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
        </el-row>
      </template>

      <el-table v-loading="loading" border :data="knowledgeInfoList" @selection-change="handleSelectionChange">
        <el-table-column type="selection" width="55" align="center" />
        <el-table-column label="" align="center" prop="id" v-if="true" />
        <el-table-column label="知识库ID" align="center" prop="knowledgeCode" />
        <el-table-column label="知识库名称" align="center" prop="knowledgeName" />
        <el-table-column label="描述" align="center" prop="description" />
        <el-table-column label="用户ID" align="center" prop="userId" />
        <el-table-column label="是否公开知识库" align="center" prop="share" />
        <el-table-column label="知识分隔符" align="center" prop="knowledgeSeparator" />
        <el-table-column label="提问分隔符" align="center" prop="questionSeparator" />
        <el-table-column label="重叠字符数" align="center" prop="overlapChar" />
        <el-table-column label="知识库中检索的条数" align="center" prop="retrieveLimit" />
        <el-table-column label="文本块大小" align="center" prop="textBlockSize" />
        <el-table-column label="向量库" align="center" prop="vectorModelName" />
        <el-table-column label="向量模型" align="center" prop="embeddingModelName" />
        <el-table-column label="系统提示词" align="center" prop="systemPrompt" />
        <el-table-column label="备注" align="center" prop="remark" />
        <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
          <template #default="scope">
            <el-tooltip content="修改" placement="top">
              <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)" v-hasPermi="['system:knowledgeInfo:edit']"></el-button>
            </el-tooltip>
            <el-tooltip content="删除" placement="top">
              <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)" v-hasPermi="['system:knowledgeInfo:remove']"></el-button>
            </el-tooltip>
          </template>
        </el-table-column>
      </el-table>

      <pagination v-show="total > 0" :total="total" v-model:page="queryParams.pageNum" v-model:limit="queryParams.pageSize" @pagination="getList" />
    </el-card>
    <!-- 添加或修改知识库对话框 -->
    <el-dialog :title="dialog.title" v-model="dialog.visible" width="500px" append-to-body>
      <el-form ref="knowledgeInfoFormRef" :model="form" :rules="rules" label-width="80px">
        <el-form-item label="知识库ID" prop="knowledgeCode">
          <el-input v-model="form.knowledgeCode" placeholder="请输入知识库ID" />
        </el-form-item>
        <el-form-item label="知识库名称" prop="knowledgeName">
          <el-input v-model="form.knowledgeName" placeholder="请输入知识库名称" />
        </el-form-item>
        <el-form-item label="描述" prop="description">
          <el-input v-model="form.description" placeholder="请输入描述" />
        </el-form-item>
        <el-form-item label="用户ID" prop="userId">
          <el-input v-model="form.userId" placeholder="请输入用户ID" />
        </el-form-item>
        <el-form-item label="是否公开知识库" prop="share">
          <el-input v-model="form.share" placeholder="请输入是否公开知识库" />
        </el-form-item>
        <el-form-item label="知识分隔符" prop="knowledgeSeparator">
          <el-input v-model="form.knowledgeSeparator" placeholder="请输入知识分隔符" />
        </el-form-item>
        <el-form-item label="提问分隔符" prop="questionSeparator">
          <el-input v-model="form.questionSeparator" placeholder="请输入提问分隔符" />
        </el-form-item>
        <el-form-item label="重叠字符数" prop="overlapChar">
          <el-input v-model="form.overlapChar" placeholder="请输入重叠字符数" />
        </el-form-item>
        <el-form-item label="知识库中检索的条数" prop="retrieveLimit">
          <el-input v-model="form.retrieveLimit" placeholder="请输入知识库中检索的条数" />
        </el-form-item>
        <el-form-item label="文本块大小" prop="textBlockSize">
          <el-input v-model="form.textBlockSize" placeholder="请输入文本块大小" />
        </el-form-item>
        <el-form-item label="向量库" prop="vectorModelName">
          <el-input v-model="form.vectorModelName" placeholder="请输入向量库" />
        </el-form-item>
        <el-form-item label="向量模型" prop="embeddingModelName">
          <el-input v-model="form.embeddingModelName" placeholder="请输入向量模型" />
        </el-form-item>
        <el-form-item label="系统提示词" prop="systemPrompt">
          <el-input v-model="form.systemPrompt" placeholder="请输入系统提示词" />
        </el-form-item>
        <el-form-item label="备注" prop="remark">
            <el-input v-model="form.remark" type="textarea" placeholder="请输入内容" />
        </el-form-item>
      </el-form>
      <template #footer>
        <div class="dialog-footer">
          <el-button :loading="buttonLoading" type="primary" @click="submitForm">确 定</el-button>
          <el-button @click="cancel">取 消</el-button>
        </div>
      </template>
    </el-dialog>
  </div>
</template>

<script setup name="KnowledgeInfo" lang="ts">
import { listKnowledgeInfo, getKnowledgeInfo, delKnowledgeInfo, addKnowledgeInfo, updateKnowledgeInfo } from '@/api/system/knowledgeInfo';
import { KnowledgeInfoVO, KnowledgeInfoQuery, KnowledgeInfoForm } from '@/api/system/knowledgeInfo/types';

const { proxy } = getCurrentInstance() as ComponentInternalInstance;

const knowledgeInfoList = ref<KnowledgeInfoVO[]>([]);
const buttonLoading = ref(false);
const loading = ref(true);
const showSearch = ref(true);
const ids = ref<Array<string | number>>([]);
const single = ref(true);
const multiple = ref(true);
const total = ref(0);

const queryFormRef = ref<ElFormInstance>();
const knowledgeInfoFormRef = ref<ElFormInstance>();

const dialog = reactive<DialogOption>({
  visible: false,
  title: ''
});

const initFormData: KnowledgeInfoForm = {
  id: undefined,
  knowledgeCode: undefined,
  knowledgeName: undefined,
  description: undefined,
  userId: undefined,
  share: undefined,
  knowledgeSeparator: undefined,
  questionSeparator: undefined,
  overlapChar: undefined,
  retrieveLimit: undefined,
  textBlockSize: undefined,
  vectorModelName: undefined,
  embeddingModelName: undefined,
  systemPrompt: undefined,
  remark: undefined
}
const data = reactive<PageData<KnowledgeInfoForm, KnowledgeInfoQuery>>({
  form: {...initFormData},
  queryParams: {
    pageNum: 1,
    pageSize: 10,
    knowledgeCode: undefined,
    knowledgeName: undefined,
    description: undefined,
    userId: undefined,
    share: undefined,
    knowledgeSeparator: undefined,
    questionSeparator: undefined,
    overlapChar: undefined,
    retrieveLimit: undefined,
    textBlockSize: undefined,
    vectorModelName: undefined,
    embeddingModelName: undefined,
    systemPrompt: undefined,
    params: {
    }
  },
  rules: {
  }
});

const { queryParams, form, rules } = toRefs(data);

/** 查询知识库列表 */
const getList = async () => {
  loading.value = true;
  const res = await listKnowledgeInfo(queryParams.value);
  knowledgeInfoList.value = res.rows;
  total.value = res.total;
  loading.value = false;
}

/** 取消按钮 */
const cancel = () => {
  reset();
  dialog.visible = false;
}

/** 表单重置 */
const reset = () => {
  form.value = {...initFormData};
  knowledgeInfoFormRef.value?.resetFields();
}

/** 搜索按钮操作 */
const handleQuery = () => {
  queryParams.value.pageNum = 1;
  getList();
}

/** 重置按钮操作 */
const resetQuery = () => {
  queryFormRef.value?.resetFields();
  handleQuery();
}

/** 多选框选中数据 */
const handleSelectionChange = (selection: KnowledgeInfoVO[]) => {
  ids.value = selection.map(item => item.id);
  single.value = selection.length != 1;
  multiple.value = !selection.length;
}

/** 新增按钮操作 */
const handleAdd = () => {
  reset();
  dialog.visible = true;
  dialog.title = "添加知识库";
}

/** 修改按钮操作 */
const handleUpdate = async (row?: KnowledgeInfoVO) => {
  reset();
  const _id = row?.id || ids.value[0]
  const res = await getKnowledgeInfo(_id);
  Object.assign(form.value, res.data);
  dialog.visible = true;
  dialog.title = "修改知识库";
}

/** 提交按钮 */
const submitForm = () => {
  knowledgeInfoFormRef.value?.validate(async (valid: boolean) => {
    if (valid) {
      buttonLoading.value = true;
      if (form.value.id) {
        await updateKnowledgeInfo(form.value).finally(() =>  buttonLoading.value = false);
      } else {
        await addKnowledgeInfo(form.value).finally(() =>  buttonLoading.value = false);
      }
      proxy?.$modal.msgSuccess("操作成功");
      dialog.visible = false;
      await getList();
    }
  });
}

/** 删除按钮操作 */
const handleDelete = async (row?: KnowledgeInfoVO) => {
  const _ids = row?.id || ids.value;
  await proxy?.$modal.confirm('是否确认删除知识库编号为"' + _ids + '"的数据项？').finally(() => loading.value = false);
  await delKnowledgeInfo(_ids);
  proxy?.$modal.msgSuccess("删除成功");
  await getList();
}

/** 导出按钮操作 */
const handleExport = () => {
  proxy?.download('system/knowledgeInfo/export', {
    ...queryParams.value
  }, `knowledgeInfo_${new Date().getTime()}.xlsx`)
}

onMounted(() => {
  getList();
});
</script>
