import axiosInstance from './axiosInstance'; import _ from 'lodash'; import { cacheGet, cacheSet, getCachedPromise } from './cache'; import stableStringify from 'json-stable-stringify'; import { cacheClean } from './cache'; import socket from './socket'; import getAsArray from './getAsArray'; import { DatabaseInfo } from 'dbgate-types'; import { derived } from 'svelte/store'; import { extendDatabaseInfo } from 'dbgate-tools'; import { setLocalStorage } from '../utility/storageCache'; const databaseInfoLoader = ({ conid, database }) => ({ url: 'database-connections/structure', params: { conid, database }, reloadTrigger: `database-structure-changed-${conid}-${database}`, transform: extendDatabaseInfo, }); // 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 settingsLoader = () => ({ url: 'config/get-settings', params: {}, reloadTrigger: 'settings-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}`, onLoaded: value => { if (value?.length > 0) setLocalStorage(`database_list_${conid}`, value); }, }); const serverVersionLoader = ({ conid }) => ({ url: 'server-connections/version', params: { conid }, reloadTrigger: `server-version-changed-${conid}`, }); const databaseServerVersionLoader = ({ conid, database }) => ({ url: 'database-connections/server-version', params: { conid, database }, reloadTrigger: `database-server-version-changed-${conid}-${database}`, }); 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`, }); const authTypesLoader = ({ engine }) => ({ url: 'plugins/auth-types', params: { engine }, reloadTrigger: `installed-plugins-changed`, }); async function getCore(loader, args) { const { url, params, reloadTrigger, transform, onLoaded } = loader(args); const key = stableStringify({ url, ...params }); async function doLoad() { const resp = await axiosInstance.request({ method: 'get', url, params, }); const res = (transform || (x => x))(resp.data); if (onLoaded) onLoaded(res); return res; } 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, onLoaded } = loader(args); const cacheKey = stableStringify({ url, ...params }); return { subscribe: onChange => { async function handleReload() { async function doLoad() { const resp = await axiosInstance.request({ method: 'get', params, url, }); const res = (transform || (x => x))(resp.data); if (onLoaded) onLoaded(res); return res; } if (cacheKey) { const fromCache = cacheGet(cacheKey); if (fromCache) { onChange(fromCache); } else { try { const res = await getCachedPromise(cacheKey, doLoad); cacheSet(cacheKey, res, reloadTrigger); onChange(res); } catch (err) { console.error('Error when using cached promise', err); cacheClean(cacheKey); const res = await doLoad(); cacheSet(cacheKey, res, reloadTrigger); onChange(res); } } } else { const res = await doLoad(); onChange(res); } } if (reloadTrigger && !socket) { console.error('Socket not available, reloadTrigger not planned'); } handleReload(); if (reloadTrigger && socket) { for (const item of getAsArray(reloadTrigger)) { socket.on(item, handleReload); } return () => { for (const item of getAsArray(reloadTrigger)) { socket.off(item, handleReload); } }; } }, }; // const useTrack = track => ({ // subscribe: onChange => { // onChange('TRACK ' + track); // if (track) { // const handle = setInterval(() => onChange('TRACK ' + track + ';' + new Date()), 1000); // // console.log("ON", track); // const oldTrack = track; // return () => { // clearInterval(handle); // // console.log("OFF", oldTrack); // }; // } // }, // }); // 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 dbStore = useDatabaseInfo(args); if (!dbStore) return null; return derived(dbStore, db => { if (!db) return null; if (_.isArray(objectTypeField)) { for (const field of objectTypeField) { const res = db[field || args.objectTypeField].find( x => x.pureName == args.pureName && x.schemaName == args.schemaName ); if (res) return res; } } else { 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', 'matviews']); } /** @returns {import('dbgate-types').CollectionInfo} */ export function useCollectionInfo(args) { return useDbCore(args, 'collections'); } 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 getServerVersion(args) { return getCore(serverVersionLoader, args); } export function useServerVersion(args) { return useCore(serverVersionLoader, args); } export function getDatabaseServerVersion(args) { return getCore(databaseServerVersionLoader, args); } export function useDatabaseServerVersion(args) { return useCore(databaseServerVersionLoader, 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 getSettings() { return getCore(settingsLoader, {}); } export function useSettings() { return useCore(settingsLoader, {}); } 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); } export function getAuthTypes(args) { return getCore(authTypesLoader, args); } export function useAuthTypes(args) { return useCore(authTypesLoader, args); }