import { writable, derived, readable } from 'svelte/store'; import localforage from 'localforage'; import type { ExtensionsDirectory } from 'dbgate-types'; import invalidateCommands from './commands/invalidateCommands'; import getElectron from './utility/getElectron'; import { getSettings, useConfig, useSettings } from './utility/metadataLoaders'; import _ from 'lodash'; import { safeJsonParse } from 'dbgate-tools'; import { apiCall } from './utility/api'; import { getOpenedTabsStorageName, isAdminPage } from './utility/pageDefs'; import { switchCurrentDatabase } from './utility/common'; export interface TabDefinition { title: string; closedTime?: number; icon: string; props: any; selected: boolean; busy: boolean; tabid: string; tabComponent: string; tabOrder?: number; multiTabIndex?: number; unsaved?: boolean; tabPreviewMode?: boolean; focused?: boolean; } export function writableWithStorage(defaultValue: T, storageName) { const init = localStorage.getItem(storageName); const res = writable(init ? safeJsonParse(init, defaultValue, true) : defaultValue); res.subscribe(value => { localStorage.setItem(storageName, JSON.stringify(value)); }); return res; } export function writableWithForage(defaultValue: T, storageName, safeConvertor?) { const res = writable(defaultValue); res.subscribe(value => { localforage.setItem(storageName, value); }); localforage.getItem(storageName).then(value => { if (value == null) { const migrated = localStorage.getItem(storageName); if (migrated) { localStorage.removeItem(storageName); const parsed = safeJsonParse(migrated, defaultValue, true); localforage.setItem(storageName, parsed); res.set(parsed as T); } } else { res.set(safeConvertor ? safeConvertor(value) : (value as T)); } }); return res; } export function writableSettingsValue(defaultValue: T, storageName) { const res = derived(useSettings(), $settings => { const obj = $settings || {}; // console.log('GET SETTINGS', $settings, storageName, obj[storageName]); return obj[storageName] ?? defaultValue; }); return { ...res, set: value => apiCall('config/update-settings', { [storageName]: value }), update: async func => { const settings = await getSettings(); const newValue = func(settings[storageName] ?? defaultValue); apiCall('config/update-settings', { [storageName]: newValue }); }, }; } function subscribeCssVariable(store, transform, cssVariable) { store.subscribe(value => document.documentElement.style.setProperty(cssVariable, transform(value))); } export const selectedWidget = writableWithStorage( isAdminPage() ? 'admin' : 'database', isAdminPage() ? 'selectedAdminWidget' : 'selectedWidget' ); export const lockedDatabaseMode = writableWithStorage(false, 'lockedDatabaseMode'); export const visibleWidgetSideBar = writableWithStorage(true, 'visibleWidgetSideBar'); export const visibleSelectedWidget = derived( [selectedWidget, visibleWidgetSideBar], ([$selectedWidget, $visibleWidgetSideBar]) => ($visibleWidgetSideBar ? $selectedWidget : null) ); export const emptyConnectionGroupNames = writableWithStorage([], 'emptyConnectionGroupNames'); export const collapsedConnectionGroupNames = writableWithStorage([], 'collapsedConnectionGroupNames'); export const openedConnections = writable([]); export const temporaryOpenedConnections = writable([]); export const openedSingleDatabaseConnections = writable([]); export const expandedConnections = writable([]); export const currentDatabase = writableWithStorage(null, 'currentDatabase'); export const openedTabs = writableWithForage([], getOpenedTabsStorageName(), x => [...(x || [])]); export const copyRowsFormat = writableWithStorage('textWithoutHeaders', 'copyRowsFormat'); export const extensions = writable(null); export const visibleCommandPalette = writable(null); export const commands = writable({}); export const currentTheme = getElectron() ? writableSettingsValue('theme-light', 'currentTheme') : writableWithStorage('theme-light', 'currentTheme'); export const currentEditorTheme = getElectron() ? writableSettingsValue(null, 'currentEditorTheme') : writableWithStorage(null, 'currentEditorTheme'); export const currentEditorKeybindigMode = getElectron() ? writableSettingsValue(null, 'currentEditorKeybindigMode') : writableWithStorage(null, 'currentEditorKeybindigMode'); export const currentEditorWrapEnabled = getElectron() ? writableSettingsValue(false, 'currentEditorWrapEnabled') : writableWithStorage(false, 'currentEditorWrapEnabled'); export const currentEditorFontSize = getElectron() ? writableSettingsValue(null, 'currentEditorFontSize') : writableWithStorage(null, 'currentEditorFontSize'); export const currentEditorFont = writableSettingsValue(null, 'editor.fontFamily'); export const allowedSendToAiService = writableSettingsValue(false, 'ai.allowSendModels'); export const activeTabId = derived([openedTabs], ([$openedTabs]) => $openedTabs.find(x => x.selected)?.tabid); export const activeTab = derived([openedTabs], ([$openedTabs]) => $openedTabs.find(x => x.selected)); export const recentDatabases = writableWithStorage([], 'recentDatabases'); export const pinnedDatabases = writableWithStorage([], 'pinnedDatabases'); export const pinnedTables = writableWithStorage([], 'pinnedTables'); export const commandsSettings = writable({}); export const allResultsInOneTabDefault = writableWithStorage(false, 'allResultsInOneTabDefault'); export const commandsCustomized = derived([commands, commandsSettings], ([$commands, $commandsSettings]) => _.mapValues($commands, (v, k) => ({ // @ts-ignore ...v, ...$commandsSettings[k], })) ); export const appUpdateStatus = writable(null); export const appUpdaterActive = writable(false); export const draggingTab = writable(null); export const draggingTabTarget = writable(null); export const draggingDbGroup = writable(null); export const draggingDbGroupTarget = writable(null); // export const visibleToolbar = writableWithStorage(true, 'visibleToolbar'); export const visibleToolbar = writable(false); export const leftPanelWidth = writableWithStorage(300, 'leftPanelWidth'); export const currentDropDownMenu = writable(null); export const openedModals = writable([]); export const draggedPinnedObject = writable(null); export const openedSnackbars = writable([]); export const nullStore = readable(null, () => {}); export const currentArchive = writableWithStorage('default', 'currentArchive'); export const currentApplication = writableWithStorage(null, 'currentApplication'); export const isFileDragActive = writable(false); export const selectedCellsCallback = writable(null); export const loadingPluginStore = writable({ loaded: false, loadingPackageName: null, }); export const activeDbKeysStore = writableWithStorage({}, 'activeDbKeysStore'); export const appliedCurrentSchema = writable(null); export const loadingSchemaLists = writable({}); // dict [`${conid}::${database}`]: true export const lastUsedDefaultActions = writableWithStorage({}, 'lastUsedDefaultActions'); export const selectedDatabaseObjectAppObject = writable(null); export const focusedConnectionOrDatabase = writable<{ conid: string; database?: string; connection: any }>(null); export const focusedTreeDbKey = writable<{ key: string; root: string; type: string; text: string }>(null); export const DEFAULT_OBJECT_SEARCH_SETTINGS = { pureName: true, schemaName: false, columnName: false, columnDataType: false, tableComment: true, columnComment: false, sqlObjectText: false, tableEngine: false, }; export const DEFAULT_CONNECTION_SEARCH_SETTINGS = { displayName: true, server: true, user: false, engine: false, database: true, }; export const databaseObjectAppObjectSearchSettings = writableWithStorage( DEFAULT_OBJECT_SEARCH_SETTINGS, 'databaseObjectAppObjectSearchSettings2' ); export const connectionAppObjectSearchSettings = writableWithStorage( DEFAULT_CONNECTION_SEARCH_SETTINGS, 'connectionAppObjectSearchSettings2' ); export const currentThemeDefinition = derived([currentTheme, extensions], ([$currentTheme, $extensions]) => $extensions.themes.find(x => x.themeClassName == $currentTheme) ); export const openedConnectionsWithTemporary = derived( [openedConnections, temporaryOpenedConnections, openedSingleDatabaseConnections], ([$openedConnections, $temporaryOpenedConnections, $openedSingleDatabaseConnections]) => _.uniq([ ...$openedConnections, ...$temporaryOpenedConnections.map(x => x.conid), ...$openedSingleDatabaseConnections, ]) ); let nativeMenuOnStartup = null; export const visibleTitleBar = derived(useSettings(), $settings => { const electron = getElectron(); if (!electron) return false; // console.log('visibleTitleBar:settings', $settings); if (!$settings) return false; if (nativeMenuOnStartup == null) { nativeMenuOnStartup = !!$settings['app.useNativeMenu']; } // console.log('nativeMenuOnStartup', nativeMenuOnStartup); return !$settings['app.fullscreen'] && !nativeMenuOnStartup; }); export const alignDataGridNumbersToRight = derived(useSettings(), $settings => { return !!$settings?.['dataGrid.alignNumbersRight']; }); export const visibleHamburgerMenuWidget = derived(useSettings(), $settings => { const electron = getElectron(); if (!electron) return true; if (!$settings) return false; return !!$settings['app.fullscreen']; }); subscribeCssVariable(visibleSelectedWidget, x => (x ? 1 : 0), '--dim-visible-left-panel'); // subscribeCssVariable(visibleToolbar, x => (x ? 1 : 0), '--dim-visible-toolbar'); subscribeCssVariable(leftPanelWidth, x => `${x}px`, '--dim-left-panel-width'); subscribeCssVariable(visibleTitleBar, x => (x ? 1 : 0), '--dim-visible-titlebar'); subscribeCssVariable(lockedDatabaseMode, x => (x ? 0 : 1), '--dim-visible-tabs-databases'); subscribeCssVariable(alignDataGridNumbersToRight, x => (x ? 'right' : 'left'), '--data-grid-numbers-align'); let activeTabIdValue = null; activeTabId.subscribe(value => { activeTabIdValue = value; invalidateCommands(); }); export const getActiveTabId = () => activeTabIdValue; let visibleCommandPaletteValue = null; visibleCommandPalette.subscribe(value => { visibleCommandPaletteValue = value; invalidateCommands(); }); export const getVisibleCommandPalette = () => visibleCommandPaletteValue; let visibleToolbarValue = null; visibleToolbar.subscribe(value => { visibleToolbarValue = value; invalidateCommands(); }); export const getVisibleToolbar = () => visibleToolbarValue; let openedTabsValue = null; openedTabs.subscribe(value => { openedTabsValue = value; invalidateCommands(); }); export const getOpenedTabs = () => openedTabsValue; let commandsValue = null; commands.subscribe(value => { commandsValue = value; const electron = getElectron(); if (electron) { electron.send('update-commands', JSON.stringify(value)); } }); export const getCommands = () => commandsValue; let activeTabValue = null; activeTab.subscribe(value => { activeTabValue = value; }); export const getActiveTab = () => activeTabValue; let currentConfigValue = null; export const getCurrentConfig = () => currentConfigValue; let recentDatabasesValue = null; recentDatabases.subscribe(value => { recentDatabasesValue = value; }); export const getRecentDatabases = () => _.compact(recentDatabasesValue); let pinnedDatabasesValue = null; pinnedDatabases.subscribe(value => { pinnedDatabasesValue = value; }); export const getPinnedDatabases = () => _.compact(pinnedDatabasesValue); let lockedDatabaseModeValue = null; lockedDatabaseMode.subscribe(value => { lockedDatabaseModeValue = value; }); export const getLockedDatabaseMode = () => lockedDatabaseModeValue; let currentDatabaseValue = null; currentDatabase.subscribe(value => { currentDatabaseValue = value; if (value?.connection?._id) { if (value?.connection?.singleDatabase) { openedSingleDatabaseConnections.update(x => _.uniq([...x, value?.connection?._id])); } else { openedConnections.update(x => _.uniq([...x, value?.connection?._id])); expandedConnections.update(x => _.uniq([...x, value?.connection?._id])); } } invalidateCommands(); }); export const getCurrentDatabase = () => currentDatabaseValue; let currentSettingsValue = null; export const getCurrentSettings = () => currentSettingsValue || {}; let extensionsValue: ExtensionsDirectory = null; extensions.subscribe(value => { extensionsValue = value; }); export const getExtensions = () => extensionsValue; let openedConnectionsValue = null; openedConnections.subscribe(value => { openedConnectionsValue = value; }); export const getOpenedConnections = () => openedConnectionsValue; export function subscribeApiDependendStores() { useSettings().subscribe(value => { currentSettingsValue = value; commandsSettings.set((value || {}).commands || {}); invalidateCommands(); }); useConfig().subscribe(value => { currentConfigValue = value; invalidateCommands(); if (value.singleDbConnection) { switchCurrentDatabase(value.singleDbConnection); } }); } let currentArchiveValue = null; currentArchive.subscribe(value => { currentArchiveValue = value; }); export const getCurrentArchive = () => currentArchiveValue; let appUpdaterActiveValue = false; appUpdaterActive.subscribe(value => { appUpdaterActiveValue = value; }); export const getAppUpdaterActive = () => appUpdaterActiveValue; let appliedCurrentSchemaValue = null; appliedCurrentSchema.subscribe(value => { appliedCurrentSchemaValue = value; }); export const getAppliedCurrentSchema = () => appliedCurrentSchemaValue; let selectedDatabaseObjectAppObjectValue = null; selectedDatabaseObjectAppObject.subscribe(value => { selectedDatabaseObjectAppObjectValue = value; }); export const getSelectedDatabaseObjectAppObject = () => selectedDatabaseObjectAppObjectValue; let openedModalsValue = []; openedModals.subscribe(value => { openedModalsValue = value; }); export const getOpenedModals = () => openedModalsValue; let focusedConnectionOrDatabaseValue = null; focusedConnectionOrDatabase.subscribe(value => { focusedConnectionOrDatabaseValue = value; }); export const getFocusedConnectionOrDatabase = () => focusedConnectionOrDatabaseValue; let openedSingleDatabaseConnectionsValue = []; openedSingleDatabaseConnections.subscribe(value => { openedSingleDatabaseConnectionsValue = value; }); export const getOpenedSingleDatabaseConnections = () => openedSingleDatabaseConnectionsValue; let lastUsedDefaultActionsValue = {}; lastUsedDefaultActions.subscribe(value => { lastUsedDefaultActionsValue = value; }); export const getLastUsedDefaultActions = () => lastUsedDefaultActionsValue; let databaseObjectAppObjectSearchSettingsValue: typeof DEFAULT_OBJECT_SEARCH_SETTINGS = { ...DEFAULT_OBJECT_SEARCH_SETTINGS, }; databaseObjectAppObjectSearchSettings.subscribe(value => { databaseObjectAppObjectSearchSettingsValue = value; }); export const getDatabaseObjectAppObjectSearchSettings = () => databaseObjectAppObjectSearchSettingsValue; let connectionAppObjectSearchSettingsValue: typeof DEFAULT_CONNECTION_SEARCH_SETTINGS = { ...DEFAULT_CONNECTION_SEARCH_SETTINGS, }; connectionAppObjectSearchSettings.subscribe(value => { connectionAppObjectSearchSettingsValue = value; }); export const getConnectionAppObjectSearchSettings = () => connectionAppObjectSearchSettingsValue; let focusedTreeDbKeyValue = null; focusedTreeDbKey.subscribe(value => { focusedTreeDbKeyValue = value; }); export const getFocusedTreeDbKey = () => focusedTreeDbKeyValue; window['__changeCurrentTheme'] = theme => currentTheme.set(theme);