import request from '@/utils/request';
import type { ApiResult } from '@/api';
import { MODULES_API_URL } from '@/config/setting';

/**
 * 获取知识库文档列表
 */
export async function getKnowledgeBaseDocuments(kbId: string, pageSize: number, pageNumber: number) {
  const res = await request.get<ApiResult<unknown>>(
      MODULES_API_URL + '/ai/knowledgeBase/documents',
      {
        params: { kbId, pageSize, pageNumber }
      }
  );
  if (res.data.code === 0) {
    return res.data.data;
  }
  return Promise.reject(new Error(res.data.message));
}

/**
 * 删除知识库文档
 */
export async function deleteKnowledgeBaseDocument(kbId: string, fileIds: string) {
  const res = await request.delete<ApiResult<unknown>>(
      MODULES_API_URL + '/ai/knowledgeBase/document',
      {
        params: { kbId, fileIds }
      }
  );
  if (res.data.code === 0) {
    return res.data.data;
  }
  return Promise.reject(new Error(res.data.message));
}

/**
 * 上传知识库文档
 */
export async function uploadKnowledgeBaseDocument(kbId: string, files: File[]) {
  const formData = new FormData();
  files.forEach(file => {
    formData.append('files', file);
  });
  const res = await request.post<ApiResult<unknown>>(
      MODULES_API_URL + '/ai/knowledgeBase/upload',
      formData,
      {
        params: { kbId },
        headers: {
          'Content-Type': 'multipart/form-data'
        }
      }
  );
  if (res.data.code === 0) {
    return res.data.data;
  }
  return Promise.reject(new Error(res.data.message));
}