@lobehub/chat/src/features/DevPanel/PostgresViewer/usePgTable.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import useSWR from 'swr';

import { tableViewerService } from '@/services/tableViewer';
import { useGlobalStore } from '@/store/global';
import { systemStatusSelectors } from '@/store/global/selectors';

const FETCH_TABLES = 'fetch-tables';
const FETCH_TABLE_COLUMN_KEY = (tableName: string) => ['fetch-table-columns', tableName];
export const FETCH_TABLE_DATA_KEY = (tableName: string) => ['fetch-table-data', tableName];

export const useFetchTables = () => {
  const isDBInited = useGlobalStore(systemStatusSelectors.isDBInited);

  return useSWR(isDBInited ? FETCH_TABLES : null, () => tableViewerService.getAllTables());
};

export const useTableColumns = (tableName?: string) => {
  const isDBInited = useGlobalStore(systemStatusSelectors.isDBInited);

  return useSWR(isDBInited && tableName ? FETCH_TABLE_COLUMN_KEY(tableName) : null, ([, table]) =>
    tableViewerService.getTableDetails(table),
  );
};

export const usePgTable = (tableName?: string) => {
  const isDBInited = useGlobalStore(systemStatusSelectors.isDBInited);

  return useSWR(isDBInited && tableName ? FETCH_TABLE_DATA_KEY(tableName) : null, ([, table]) =>
    tableViewerService.getTableData(table),
  );
};