DocsGPT/frontend/src/api/services/userService.ts

import { getSessionToken } from '../../utils/providerUtils';
import apiClient from '../client';
import endpoints from '../endpoints';

const userService = {
  getConfig: (): Promise<any> => apiClient.get(endpoints.USER.CONFIG, null),
  getNewToken: (): Promise<any> =>
    apiClient.get(endpoints.USER.NEW_TOKEN, null),
  getDocs: (token: string | null): Promise<any> =>
    apiClient.get(`${endpoints.USER.DOCS}`, token),
  getDocsWithPagination: (query: string, token: string | null): Promise<any> =>
    apiClient.get(`${endpoints.USER.DOCS_PAGINATED}?${query}`, token),
  getAPIKeys: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.API_KEYS, token),
  createAPIKey: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.CREATE_API_KEY, data, token),
  deleteAPIKey: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.DELETE_API_KEY, data, token),
  getAgent: (id: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.AGENT(id), token),
  getAgents: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.AGENTS, token),
  createAgent: (data: any, token: string | null): Promise<any> =>
    apiClient.postFormData(endpoints.USER.CREATE_AGENT, data, token),
  updateAgent: (
    agent_id: string,
    data: any,
    token: string | null,
  ): Promise<any> =>
    apiClient.putFormData(endpoints.USER.UPDATE_AGENT(agent_id), data, token),
  deleteAgent: (id: string, token: string | null): Promise<any> =>
    apiClient.delete(endpoints.USER.DELETE_AGENT(id), token),
  getPinnedAgents: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.PINNED_AGENTS, token),
  togglePinAgent: (id: string, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.TOGGLE_PIN_AGENT(id), {}, token),
  getSharedAgent: (id: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.SHARED_AGENT(id), token),
  getSharedAgents: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.SHARED_AGENTS, token),
  shareAgent: (data: any, token: string | null): Promise<any> =>
    apiClient.put(endpoints.USER.SHARE_AGENT, data, token),
  removeSharedAgent: (id: string, token: string | null): Promise<any> =>
    apiClient.delete(endpoints.USER.REMOVE_SHARED_AGENT(id), token),
  getTemplateAgents: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.TEMPLATE_AGENTS, token),
  adoptAgent: (id: string, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.ADOPT_AGENT(id), {}, token),
  getAgentWebhook: (id: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.AGENT_WEBHOOK(id), token),
  getPrompts: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.PROMPTS, token),
  createPrompt: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.CREATE_PROMPT, data, token),
  deletePrompt: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.DELETE_PROMPT, data, token),
  updatePrompt: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.UPDATE_PROMPT, data, token),
  getSinglePrompt: (id: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.SINGLE_PROMPT(id), token),
  deletePath: (docPath: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.DELETE_PATH(docPath), token),
  getTaskStatus: (task_id: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.TASK_STATUS(task_id), token),
  getMessageAnalytics: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.MESSAGE_ANALYTICS, data, token),
  getTokenAnalytics: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.TOKEN_ANALYTICS, data, token),
  getFeedbackAnalytics: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.FEEDBACK_ANALYTICS, data, token),
  getLogs: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.LOGS, data, token),
  manageSync: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.MANAGE_SYNC, data, token),
  getAvailableTools: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.GET_AVAILABLE_TOOLS, token),
  getUserTools: (token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.GET_USER_TOOLS, token),
  createTool: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.CREATE_TOOL, data, token),
  updateToolStatus: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.UPDATE_TOOL_STATUS, data, token),
  updateTool: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.UPDATE_TOOL, data, token),
  deleteTool: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.DELETE_TOOL, data, token),
  getDocumentChunks: (
    docId: string,
    page: number,
    perPage: number,
    token: string | null,
    path?: string,
    search?: string,
  ): Promise<any> =>
    apiClient.get(
      endpoints.USER.GET_CHUNKS(docId, page, perPage, path, search),
      token,
    ),
  addChunk: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.ADD_CHUNK, data, token),
  deleteChunk: (
    docId: string,
    chunkId: string,
    token: string | null,
  ): Promise<any> =>
    apiClient.delete(endpoints.USER.DELETE_CHUNK(docId, chunkId), token),
  updateChunk: (data: any, token: string | null): Promise<any> =>
    apiClient.put(endpoints.USER.UPDATE_CHUNK, data, token),
  getDirectoryStructure: (docId: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.DIRECTORY_STRUCTURE(docId), token),
  manageSourceFiles: (data: FormData, token: string | null): Promise<any> =>
    apiClient.postFormData(endpoints.USER.MANAGE_SOURCE_FILES, data, token),
  testMCPConnection: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.MCP_TEST_CONNECTION, data, token),
  saveMCPServer: (data: any, token: string | null): Promise<any> =>
    apiClient.post(endpoints.USER.MCP_SAVE_SERVER, data, token),
  getMCPOAuthStatus: (task_id: string, token: string | null): Promise<any> =>
    apiClient.get(endpoints.USER.MCP_OAUTH_STATUS(task_id), token),
  syncConnector: (
    docId: string,
    provider: string,
    token: string | null,
  ): Promise<any> => {
    const sessionToken = getSessionToken(provider);
    return apiClient.post(
      endpoints.USER.SYNC_CONNECTOR,
      {
        source_id: docId,
        session_token: sessionToken,
        provider: provider,
      },
      token,
    );
  },
};

export default userService;