import useFetch from './useFetch'; import axios from './axios'; import _ from 'lodash'; import { cacheGet, cacheSet, getCachedPromise } from './cache'; import stableStringify from 'json-stable-stringify'; const databaseInfoLoader = ({ conid, database }) => ({ url: 'database-connections/structure', params: { conid, database }, reloadTrigger: `database-structure-changed-${conid}-${database}`, transform: db => { const allForeignKeys = _.flatten(db.tables.map(x => x.foreignKeys)); return { ...db, tables: db.tables.map(table => ({ ...table, dependencies: allForeignKeys.filter( x => x.refSchemaName == table.schemaName && x.refTableName == table.pureName ), })), }; }, }); // const tableInfoLoader = ({ conid, database, schemaName, pureName }) => ({ // url: 'metadata/table-info', // params: { conid, database, schemaName, pureName }, // reloadTrigger: `database-structure-changed-${conid}-${database}`, // }); // const sqlObjectInfoLoader = ({ objectTypeField, conid, database, schemaName, pureName }) => ({ // url: 'metadata/sql-object-info', // params: { objectTypeField, conid, database, schemaName, pureName }, // reloadTrigger: `database-structure-changed-${conid}-${database}`, // }); const connectionInfoLoader = ({ conid }) => ({ url: 'connections/get', params: { conid }, reloadTrigger: 'connection-list-changed', }); const configLoader = () => ({ url: 'config/get', params: {}, reloadTrigger: 'config-changed', }); const platformInfoLoader = () => ({ url: 'config/platform-info', params: {}, reloadTrigger: 'platform-info-changed', }); const favoritesLoader = () => ({ url: 'files/favorites', params: {}, reloadTrigger: 'files-changed-favorites', }); // const sqlObjectListLoader = ({ conid, database }) => ({ // url: 'metadata/list-objects', // params: { conid, database }, // reloadTrigger: `database-structure-changed-${conid}-${database}`, // }); const databaseStatusLoader = ({ conid, database }) => ({ url: 'database-connections/status', params: { conid, database }, reloadTrigger: `database-status-changed-${conid}-${database}`, }); const databaseListLoader = ({ conid }) => ({ url: 'server-connections/list-databases', params: { conid }, reloadTrigger: `database-list-changed-${conid}`, }); const archiveFoldersLoader = () => ({ url: 'archive/folders', params: {}, reloadTrigger: `archive-folders-changed`, }); const archiveFilesLoader = ({ folder }) => ({ url: 'archive/files', params: { folder }, reloadTrigger: `archive-files-changed-${folder}`, }); const serverStatusLoader = () => ({ url: 'server-connections/server-status', params: {}, reloadTrigger: `server-status-changed`, }); const connectionListLoader = () => ({ url: 'connections/list', params: {}, reloadTrigger: `connection-list-changed`, }); const installedPluginsLoader = () => ({ url: 'plugins/installed', params: {}, reloadTrigger: `installed-plugins-changed`, }); const filesLoader = ({ folder }) => ({ url: 'files/list', params: { folder }, reloadTrigger: `files-changed-${folder}`, }); const allFilesLoader = () => ({ url: 'files/list-all', params: {}, reloadTrigger: `all-files-changed`, }); async function getCore(loader, args) { const { url, params, reloadTrigger, transform } = loader(args); const key = stableStringify({ url, ...params }); async function doLoad() { const resp = await axios.request({ method: 'get', url, params, }); return (transform || (x => x))(resp.data); } const fromCache = cacheGet(key); if (fromCache) return fromCache; const res = await getCachedPromise(key, doLoad); cacheSet(key, res, reloadTrigger); return res; } function useCore(loader, args) { const { url, params, reloadTrigger, transform } = loader(args); const cacheKey = stableStringify({ url, ...params }); const res = useFetch({ url, params, reloadTrigger, cacheKey, transform, }); return res; } /** @returns {Promise} */ export function getDatabaseInfo(args) { return getCore(databaseInfoLoader, args); } /** @returns {import('dbgate-types').DatabaseInfo} */ export function useDatabaseInfo(args) { return useCore(databaseInfoLoader, args); } export async function getDbCore(args, objectTypeField = undefined) { const db = await getDatabaseInfo(args); if (!db) return null; return db[objectTypeField || args.objectTypeField].find( x => x.pureName == args.pureName && x.schemaName == args.schemaName ); } export function useDbCore(args, objectTypeField = undefined) { const db = useDatabaseInfo(args); if (!db) return null; return db[objectTypeField || args.objectTypeField].find( x => x.pureName == args.pureName && x.schemaName == args.schemaName ); } /** @returns {Promise} */ export function getTableInfo(args) { return getDbCore(args, 'tables'); } /** @returns {import('dbgate-types').TableInfo} */ export function useTableInfo(args) { return useDbCore(args, 'tables'); } /** @returns {Promise} */ export function getViewInfo(args) { return getDbCore(args, 'views'); } /** @returns {import('dbgate-types').ViewInfo} */ export function useViewInfo(args) { return useDbCore(args, 'views'); } export function getSqlObjectInfo(args) { return getDbCore(args); } export function useSqlObjectInfo(args) { return useDbCore(args); } /** @returns {Promise} */ export function getConnectionInfo(args) { return getCore(connectionInfoLoader, args); } /** @returns {import('dbgate-types').StoredConnection} */ export function useConnectionInfo(args) { return useCore(connectionInfoLoader, args); } // export function getSqlObjectList(args) { // return getCore(sqlObjectListLoader, args); // } // export function useSqlObjectList(args) { // return useCore(sqlObjectListLoader, args); // } export function getDatabaseStatus(args) { return getCore(databaseStatusLoader, args); } export function useDatabaseStatus(args) { return useCore(databaseStatusLoader, args); } export function getDatabaseList(args) { return getCore(databaseListLoader, args); } export function useDatabaseList(args) { return useCore(databaseListLoader, args); } export function getServerStatus() { return getCore(serverStatusLoader, {}); } export function useServerStatus() { return useCore(serverStatusLoader, {}); } export function getConnectionList() { return getCore(connectionListLoader, {}); } export function useConnectionList() { return useCore(connectionListLoader, {}); } export function getConfig() { return getCore(configLoader, {}) || {}; } export function useConfig() { return useCore(configLoader, {}) || {}; } export function getPlatformInfo() { return getCore(platformInfoLoader, {}) || {}; } export function usePlatformInfo() { return useCore(platformInfoLoader, {}) || {}; } export function getArchiveFiles(args) { return getCore(archiveFilesLoader, args); } export function useArchiveFiles(args) { return useCore(archiveFilesLoader, args); } export function getArchiveFolders(args) { return getCore(archiveFoldersLoader, args); } export function useArchiveFolders(args) { return useCore(archiveFoldersLoader, args); } export function getInstalledPlugins(args) { return getCore(installedPluginsLoader, args) || []; } export function useInstalledPlugins(args) { return useCore(installedPluginsLoader, args) || []; } export function getFiles(args) { return getCore(filesLoader, args); } export function useFiles(args) { return useCore(filesLoader, args); } export function getAllFiles(args) { return getCore(allFilesLoader, args); } export function useAllFiles(args) { return useCore(allFilesLoader, args); } export function getFavorites(args) { return getCore(favoritesLoader, args); } export function useFavorites(args) { return useCore(favoritesLoader, args); }