import { defHttp } from '/@/utils/http/axios';
import { useMessage } from '/@/hooks/web/useMessage';

const { createConfirm } = useMessage();

enum Api {
  list = '/embeddings/embeddings/list',
  save = '/embeddings/embeddings/add',
  edit = '/embeddings/embeddings/edit',
  deleteOne = '/embeddings/embeddings/delete',
  deleteBatch = '/embeddings/embeddings/deleteBatch',
  importExcel = '/embeddings/embeddings/importExcel',
  exportXls = '/embeddings/embeddings/exportXls',
}

/**
 * 导出api
 * @param params
 */
export const getExportUrl = Api.exportXls;

/**
 * 导入api
 */
export const getImportUrl = Api.importExcel;

/**
 * 列表接口
 * @param params
 */
export const list = async (params) => {
  try {
    // 设置足够大的size获取所有数据
    const res = await defHttp.get({
      url: Api.list,
      params: { ...params, size: 1000 }
    });

    if (res?.records && Array.isArray(res.records)) {
      // 将metadata中的字段提取到顶层
      res.records = res.records.map(item => ({
        ...item,
        ...item.metadata, // 将metadata中的字段展开
        docName: item.metadata?.docName || '',
        knowledgeId: item.metadata?.knowledgeId || '',
        docId: item.metadata?.docId || '',
        index: item.metadata?.index || '',
      }));
    }
    return res;
  } catch (error) {
    console.error("Error fetching data:", error);
    return {
      records: [],
      total: 0,
      size: 10,
      current: 1,
      pages: 0
    };
  }
}

/**
 * 删除单个
 */
export const deleteOne = (params, handleSuccess) => {
  return defHttp.delete({ url: Api.deleteOne, params }, { joinParamsToUrl: true }).then(() => {
    handleSuccess();
  });
};

/**
 * 批量删除
 * @param params
 */
export const batchDelete = (params, handleSuccess) => {
  createConfirm({
    iconType: 'warning',
    title: '确认删除',
    content: '是否删除选中数据',
    okText: '确认',
    cancelText: '取消',
    onOk: () => {
      return defHttp.delete({ url: Api.deleteBatch, data: params }, { joinParamsToUrl: true }).then(() => {
        handleSuccess();
      });
    },
  });
};

/**
 * 保存或者更新
 * @param params
 */
export const saveOrUpdate = (params, isUpdate) => {
  const url = isUpdate ? Api.edit : Api.save;
  return defHttp.post({ url: url, params });
};