import { get, post } from './request';

export interface RagStore {
  id?: string;
  storeCode: string;
  storeName: string;
  description?: string | null;
  status?: string | null;
  remark?: string | null;
  createTime?: string | null;
  updateTime?: string | null;
}

export interface RagStoreOverview {
  totalStores: number;
  totalDocuments: number;
  totalChunks?: number | null;
  retrievableStores: number;
}

export interface RagStoreDocumentOverview {
  storeId: string;
  storeName?: string | null;
  documentCount: number;
  enabledDocumentCount: number;
  parsedDocumentCount: number;
  indexedDocumentCount: number;
  lastUploadTime?: string | null;
}

export interface RagStoreQueryRequest {
  storeCode?: string;
  storeName?: string;
  status?: string;
}

export type RagStoreSaveRequest = RagStore;

export function listRagStores() {
  return post<RagStore[]>('/rag/store/list');
}

export function queryRagStores(query?: RagStoreQueryRequest) {
  return post<RagStore[], RagStoreQueryRequest | undefined>('/rag/store/query', query);
}

export function getRagStoreById(id: string) {
  return get<RagStore>('/rag/store/detail', {
    params: { id },
  });
}

export function getRagStoreOverview() {
  return get<RagStoreOverview>('/rag/store/overview');
}

export function getRagStoreDocumentOverview(storeId: string) {
  return get<RagStoreDocumentOverview>('/rag/store/documentOverview', {
    params: { storeId },
  });
}

export function saveRagStore(data: RagStoreSaveRequest) {
  return post<boolean, RagStoreSaveRequest>('/rag/store/save', data);
}

export function deleteRagStore(id: string) {
  return post<boolean>('/rag/store/delete', undefined, {
    params: { id },
  });
}