import request, { fileUploadRequest, fileDownloadRequest } from '@/utils/request'

export const createDataset = (data: any) => {
    return request({
      url: `/brichat-service/datasetManage/create`,
      method: 'post',
      data
    })
  }
  export const updateDataset = (data: any) => {
    return request({
      url: `/brichat-service/datasetManage/update`,
      method: 'post',
      data
    })
  }


export const deleteDataset = (id: string) => {
    return request({
      url: `/brichat-service/datasetManage/${id}`,
      method: 'delete',
    })
  }

  export const getUsersAndDatasets = () => {
    return request({
      url: '/brichat-service/datasetManage/usersLinkDataset',
      method: 'get',
    })
  }
  
  export const getCurrUserDatasets = (userId: number, pageNo: number = 1, pageSize: number = 9) => {
    return request({
      url: '/brichat-service/datasetManage/currUserDatasets',
      method: 'get',
      params: {
        userId,
        pageNo,
        pageSize
      }
    })
  }
  
  
  export const getAllDatasets = () => {
    return request({
      url: '/brichat-service/datasetManage/allDataset',
      method: 'get',
    })
  }
  
  interface UserBindDatasetDto {
    userId: number;
    datasetIds: string[];
  }
  
  export const bindDatasets = (data: UserBindDatasetDto) => {
    return request({
      url: '/brichat-service/datasetManage/bindDatasets',
      method: 'post',
      data
    })
  }
  
  interface UnbindDatasetParams {
    userId: number;
    datasetId: string;
  }
  
  export const unbindDataset = (data: UnbindDatasetParams) => {
    return request({
      url: '/brichat-service/datasetManage/unbindDataset',
      method: 'post',
      params: data
    })
  }
  

  export const getDatasetDocPage = (data: any) => {
    return request({
      url: '/brichat-service/datasetManage/document/page',
      method: 'post',
      data  
    })
  }

  export const deleteDocument = (params: any) => {
    return request({
      url: '/brichat-service/datasetManage/document/delete',
      method: 'delete',
      params
    })
  }

  interface RenameParams {
  documentId: string
  newName: string
}
  export const renameDocument = (data: RenameParams) => {
    return request({
      url: '/brichat-service/datasetManage/document/rename',
      method: 'post',
      data
    })
  }

  export const uploadDocument = (data: FormData) => {
    return fileUploadRequest({
      url: '/brichat-service/datasetManage/document/upload',
      method: 'post',
      data
    })
  }

  /**
   * 下载文档文件
   * @param documentId 文档ID
   * @param datasetId 数据集ID
   */
  export const downloadDocument = (documentId: string, datasetId: string) => {
    return fileDownloadRequest({
      url: '/brichat-service/datasetManage/document/download',
      method: 'get',
      params: {
        documentId,
        datasetId
      }
    })
  }

  export const previewDocumentUrl = (documentId: string, datasetId: string) => {
    return request({
      url: '/brichat-service/datasetManage/document/previewUrl',
      method: 'get',
      params: {
        documentId,
        datasetId
      }
    })
  }

  /**
   * 获取深度解析任务列表
   */
  export const getDeepAnalysisList = () => {
    return request({
      url: '/brichat-service/datasetManage/document/deepAnalysisList',
      method: 'get'
    })
  }

export function retrieval(data: { query: string; searchMethod: string }) {
  return request({
    url: `/brichat-service/knowledge-base/retrieval`,
    method: 'post',
    data
  })
}