<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="组织名称" prop="organization">
              <el-input v-model="queryParams.organization" placeholder="请输入组织名称" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="项目名称" prop="project">
              <el-input v-model="queryParams.project" placeholder="请输入项目名称" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="私钥" prop="privateKey">
              <el-input v-model="queryParams.privateKey" placeholder="请输入私钥" clearable @keyup.enter="handleQuery" />
            </el-form-item>
            <el-form-item label="公钥" prop="publicKey">
              <el-input v-model="queryParams.publicKey" 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:project:add']">新增</el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button type="success" plain icon="Edit" :disabled="single" @click="handleUpdate()" v-hasPermi="['system:project:edit']">修改</el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button type="danger" plain icon="Delete" :disabled="multiple" @click="handleDelete()" v-hasPermi="['system:project:remove']">删除</el-button>
          </el-col>
          <el-col :span="1.5">
            <el-button type="warning" plain icon="Download" @click="handleExport" v-hasPermi="['system:project: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="projectList" @selection-change="handleSelectionChange">
        <el-table-column type="selection" width="55" align="center" />
        <el-table-column label="组织名称" align="center" prop="organization" />
        <el-table-column label="项目名称" align="center" prop="project" />
        <el-table-column label="私钥" align="center" prop="privateKey" />
        <el-table-column label="公钥" align="center" prop="publicKey" />
        <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
          <template #default="scope">
            <div class="flex space-x-2">
              <el-tooltip content="修改" placement="top">
                <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)" v-hasPermi="['system:project:edit']"></el-button>
              </el-tooltip>
              <el-tooltip content="删除" placement="top">
                <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)" v-hasPermi="['system:project:remove']"></el-button>
              </el-tooltip>
              <div class="flex items-center">
                <button
                    @click="handleButtonClick('model', scope.row)"
                    class="text-green-500 hover:text-green-600 border border-transparent hover:border-green-600 hover:border-1 py-1 px-2 rounded-md text-xs"
                >
                  配置模型
                </button>

                <div class="mx-1 text-gray-400 h-full">|</div>
                <button
                    @click="handleButtonClick('knowledge', scope.row)"
                    class="text-green-500 hover:text-green-600 border border-transparent hover:border-green-600 hover:border-1 py-1 px-2 rounded-md text-xs"
                >
                  配置知识库
                </button>
                <div class="mx-1 text-gray-400 h-full">|</div>
                <button
                    @click="handleButtonClick('agent', scope.row)"
                    class="text-green-500 hover:text-green-600 border border-transparent hover:border-green-600 hover:border-1 py-1 px-2 rounded-md text-xs"
                >
                  配置智能体
                </button>
              </div>
            </div>
          </template>
        </el-table-column>
      </el-table>
      <el-dialog
          v-model="showModelDialog"
          title="配置模型"
          width="800px"
          append-to-body
      >
        <el-card shadow="never" v-loading="modelLoading">
          <el-checkbox-group v-model="selectedModelIds">
            <el-row :gutter="16">
              <el-col
                  :span="8"
                  v-for="model in modelList"
                  :key="model.id"
                  class="mb-4"
              >
                <el-checkbox
                    :label="model.id"
                    class="w-full flex items-center"
                >
                  <div class="ml-2">
                    <div class="font-medium">{{ model.modelName }}</div>
                    <div class="text-xs text-gray-500">
                      {{ model.company }} · {{ model.introduce }}
                    </div>
                  </div>
                </el-checkbox>
              </el-col>
            </el-row>
          </el-checkbox-group>

          <div v-if="modelList.length === 0" class="text-center py-8 text-gray-500">
            暂无可用模型
          </div>
        </el-card>

        <template #footer>
          <div class="dialog-footer">
            <el-button @click="showModelDialog = false">取消</el-button>
            <el-button type="primary" @click="submitModelConfig">确定</el-button>
          </div>
        </template>
      </el-dialog>

      <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="projectFormRef" :model="form" :rules="rules" label-width="80px">
        <el-form-item label="组织名称" prop="organization">
          <el-input v-model="form.organization" type="textarea" placeholder="请输入内容" />
        </el-form-item>
        <el-form-item label="项目名称" prop="project">
          <el-input v-model="form.project" type="textarea" placeholder="请输入内容" />
        </el-form-item>
        <el-form-item label="私钥" prop="privateKey">
          <el-input v-model="form.privateKey" type="textarea" placeholder="请输入内容" />
        </el-form-item>
        <el-form-item label="公钥" prop="publicKey">
          <el-input v-model="form.publicKey" 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>
    <!-- 智能体配置对话框 -->
    <el-dialog
        v-model="showAgentDialog"
        title="配置智能体"
        width="800px"
        append-to-body
    >
      <el-card shadow="never" v-loading="agentLoading">
        <el-checkbox-group v-model="selectedAgentIds">
          <el-row :gutter="16">
            <el-col
                :span="8"
                v-for="agent in agentList"
                :key="agent.id"
                class="mb-4"
            >
              <el-checkbox
                  :label="agent.id"
                  class="w-full flex items-center"
              >
                <div class="ml-2">
                  <div class="font-medium">{{ agent.name }}</div>
                  <div class="text-xs text-gray-500">
                    {{ agent.description }}
                  </div>
                </div>
              </el-checkbox>
            </el-col>
          </el-row>
        </el-checkbox-group>

        <div v-if="agentList.length === 0" class="text-center py-8 text-gray-500">
          暂无可用智能体
        </div>
      </el-card>

      <template #footer>
        <div class="dialog-footer">
          <el-button @click="showAgentDialog = false">取消</el-button>
          <el-button type="primary" @click="submitAgentConfig">确定</el-button>
        </div>
      </template>
    </el-dialog>

    <!-- 知识库配置对话框 -->
    <el-dialog
        v-model="showKnowledgeDialog"
        title="配置知识库"
        width="800px"
        append-to-body
    >
      <el-card shadow="never" v-loading="knowledgeLoading">
        <el-checkbox-group v-model="selectedKnowledgeIds">
          <el-row :gutter="16">
            <el-col
                :span="8"
                v-for="knowledge in knowledgeList"
                :key="knowledge.id"
                class="mb-4"
            >
              <el-checkbox
                  :label="knowledge.id"
                  class="w-full flex items-center"
              >
                <div class="ml-2">
                  <div class="font-medium">{{ knowledge.knowledgeName }}</div>
                  <div class="text-xs text-gray-500">
                    {{ knowledge.description }}
                  </div>
                </div>
              </el-checkbox>
            </el-col>
          </el-row>
        </el-checkbox-group>

        <div v-if="knowledgeList.length === 0" class="text-center py-8 text-gray-500">
          暂无可用知识库
        </div>
      </el-card>

      <template #footer>
        <div class="dialog-footer">
          <el-button @click="showKnowledgeDialog = false">取消</el-button>
          <el-button type="primary" @click="submitKnowledgeConfig">确定</el-button>
        </div>
      </template>
    </el-dialog>
  </div>
</template>

<script setup name="Project" lang="ts">
import { listProject, getProject, delProject, addProject, updateProject, listModel, listAgent, listKnowledgeInfo } from '@/api/system/project';
import { ProjectVO, ProjectQuery, ProjectForm, ModelVO, AgentVO, KnowledgeInfoVO } from '@/api/system/project/types';

const { proxy } = getCurrentInstance() as ComponentInternalInstance;

const projectList = ref<ProjectVO[]>([]);
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 projectFormRef = ref<ElFormInstance>();

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

const initFormData: ProjectForm = {
  id: undefined,
  deptId: undefined,
  organization: undefined,
  project: undefined,
  privateKey: undefined,
  publicKey: undefined,
  modelIds: [],
  ragIds: [],
  agentIds: [],
}

const data = reactive({
  form: {...initFormData},
  queryParams: {
    pageNum: 1,
    pageSize: 10,
    deptId: undefined,
    organization: undefined,
    project: undefined,
    privateKey: undefined,
    publicKey: undefined,
    modelIds: undefined,
    ragIds: undefined,
    agentIds: undefined,
    params: {}
  },
  rules: {
    organization: [
      { required: true, message: "组织名称不能为空", trigger: "blur" }
    ],
    project: [
      { required: true, message: "项目名称不能为空", trigger: "blur" }
    ],
  }
});

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

// 对话框状态
const showAgentDialog = ref(false);
const agentLoading = ref(false);
const agentList = ref<AgentVO[]>([]);
const selectedAgentIds = ref<(string | number)[]>([]);

const showKnowledgeDialog = ref(false);
const knowledgeLoading = ref(false);
const knowledgeList = ref<KnowledgeInfoVO[]>([]);
const selectedKnowledgeIds = ref<(string | number)[]>([]);

const showModelDialog = ref(false);
const modelLoading = ref(false);
const modelList = ref<ModelVO[]>([]);
const selectedModelIds = ref<(string | number)[]>([]);
const currentProject = ref<ProjectVO | null>(null);

/** 查询项目列表 */
const getList = async () => {
  loading.value = true;
  try {
    const res = await listProject(queryParams.value);
    projectList.value = res.rows.map(item => ({
      ...item,
      modelIds: item.modelIds || [],
      ragIds: item.ragIds || [],
      agentIds: item.agentIds || []
    }));
    total.value = res.total;
  } catch (error) {
    console.error('获取项目列表失败:', error);
    proxy?.$modal.msgError("获取项目列表失败");
  } finally {
    loading.value = false;
  }
}

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

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

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

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

/** 多选框选中数据 */
const handleSelectionChange = (selection: ProjectVO[]) => {
  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?: ProjectVO) => {
  reset();
  const _id = row?.id || ids.value[0];
  try {
    const res = await getProject(_id);
    Object.assign(form.value, res.data);
    dialog.visible = true;
    dialog.title = "修改项目";
  } catch (error) {
    console.error('获取项目详情失败:', error);
    proxy?.$modal.msgError("获取项目详情失败");
  }
}

/** 提交按钮 */
const submitForm = () => {
  projectFormRef.value?.validate(async (valid: boolean) => {
    if (valid) {
      buttonLoading.value = true;
      try {
        if (form.value.id) {
          await updateProject(form.value);
        } else {
          await addProject(form.value);
        }
        proxy?.$modal.msgSuccess("操作成功");
        dialog.visible = false;
        await getList();
      } catch (error) {
        console.error('操作失败:', error);
        proxy?.$modal.msgError("操作失败");
      } finally {
        buttonLoading.value = false;
      }
    }
  });
}

/** 删除按钮操作 */
const handleDelete = async (row?: ProjectVO) => {
  const _ids = row?.id || ids.value;
  try {
    await proxy?.$modal.confirm('是否确认删除项目编号为"' + _ids + '"的数据项？');
    await delProject(_ids);
    proxy?.$modal.msgSuccess("删除成功");
    await getList();
  } catch (error) {
    console.error('删除失败:', error);
  }
}

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

const fetchModelList = async () => {
  try {
    modelLoading.value = true;
    const res = await listModel({
      pageNum: 1,
      pageSize: 1000,
      status: 0
    });
    modelList.value = res.rows;
  } catch (error) {
    console.error('获取模型列表失败:', error);
    proxy?.$modal.msgError("获取模型列表失败");
  } finally {
    modelLoading.value = false;
  }
};

const fetchAgentList = async () => {
  try {
    agentLoading.value = true;
    const res = await listAgent({
      pageNum: 1,
      pageSize: 1000,
    });
    agentList.value = res.rows;
  } catch (error) {
    console.error('获取智能体列表失败:', error);
    proxy?.$modal.msgError("获取智能体列表失败");
  } finally {
    agentLoading.value = false;
  }
};

const fetchKnowledgeList = async () => {
  try {
    knowledgeLoading.value = true;
    const res = await listKnowledgeInfo({
      pageNum: 1,
      pageSize: 1000,
      share: 1
    });
    knowledgeList.value = res.rows;
  } catch (error) {
    console.error('获取知识库列表失败:', error);
    proxy?.$modal.msgError("获取知识库列表失败");
  } finally {
    knowledgeLoading.value = false;
  }
};

const openModelConfig = async (row: ProjectVO) => {
  currentProject.value = row;
  selectedModelIds.value = row.modelIds ? [...row.modelIds] : [];
  await fetchModelList();
  showModelDialog.value = true;
};

const openAgentConfig = async (row: ProjectVO) => {
  console.log(currentProject.value);
  currentProject.value = row;
  selectedAgentIds.value = row.agentIds ? [...row.agentIds] : [];
  await fetchAgentList();
  showAgentDialog.value = true;
};

const openKnowledgeConfig = async (row: ProjectVO) => {
  currentProject.value = row;
  selectedKnowledgeIds.value = row.ragIds ? [...row.ragIds] : [];
  await fetchKnowledgeList();
  showKnowledgeDialog.value = true;
};

const submitModelConfig = async () => {
  if (!currentProject.value) return;

  try {
    modelLoading.value = true;
    const modelIdsArray = Array.isArray(selectedModelIds.value)
        ? selectedModelIds.value
        : [selectedModelIds.value];

    // 先获取当前项目的完整数据
    const currentProjectData = await getProject(currentProject.value.id);

    // 只更新 modelIds，保留其他字段
    const payload = {
      ...currentProjectData.data, // 保留原有数据
      modelIds: modelIdsArray,
      model_ids: modelIdsArray // 如果需要兼容旧字段
    };

    await updateProject(payload);
    proxy?.$modal.msgSuccess("模型配置成功");
    showModelDialog.value = false;
    await getList();
  } catch (error) {
    console.error('模型配置失败:', error);
    proxy?.$modal.msgError("模型配置失败");
  } finally {
    modelLoading.value = false;
  }
};

const submitAgentConfig = async () => {
  if (!currentProject.value) return;

  try {
    agentLoading.value = true;
    const agentIdsArray = Array.isArray(selectedAgentIds.value)
        ? selectedAgentIds.value
        : [selectedAgentIds.value];

    // 先获取当前项目的完整数据
    const currentProjectData = await getProject(currentProject.value.id);

    // 只更新 agentIds，保留其他字段
    const payload = {
      ...currentProjectData.data, // 保留原有数据
      agentIds: agentIdsArray,
      agent_ids: agentIdsArray // 如果需要兼容旧字段
    };

    await updateProject(payload);
    proxy?.$modal.msgSuccess("智能体配置成功");
    showAgentDialog.value = false;
    await getList();
  } catch (error) {
    console.error('智能体配置失败:', error);
    proxy?.$modal.msgError("智能体配置失败");
  } finally {
    agentLoading.value = false;
  }
};

const submitKnowledgeConfig = async () => {
  if (!currentProject.value) return;

  try {
    knowledgeLoading.value = true;
    const knowledgeIdsArray = Array.isArray(selectedKnowledgeIds.value)
        ? selectedKnowledgeIds.value
        : [selectedKnowledgeIds.value];

    // 先获取当前项目的完整数据
    const currentProjectData = await getProject(currentProject.value.id);

    // 只更新 ragIds，保留其他字段
    const payload = {
      ...currentProjectData.data, // 保留原有数据
      ragIds: knowledgeIdsArray,
      rag_ids: knowledgeIdsArray // 如果需要兼容旧字段
    };

    await updateProject(payload);
    proxy?.$modal.msgSuccess("知识库配置成功");
    showKnowledgeDialog.value = false;
    await getList();
  } catch (error) {
    console.error('知识库配置失败:', error);
    proxy?.$modal.msgError("知识库配置失败");
  } finally {
    knowledgeLoading.value = false;
  }
};




const handleButtonClick = (type: 'model' | 'knowledge' | 'agent', row: ProjectVO) => {
  switch (type) {
    case 'model':
      openModelConfig(row);
      break;
    case 'knowledge':
      openKnowledgeConfig(row);
      break;
    case 'agent':
      openAgentConfig(row);
      break;
  }
};

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