// packages/backend/src/repositories/settings.repository.ts import { getDbInstance, runDb, getDb as getDbRow, allDb } from '../database/connection'; import { SidebarConfig } from '../types/settings.types'; // <-- Correct import path import * as sqlite3 from 'sqlite3'; // Import sqlite3 for Database type hint // Define keys for specific settings const SIDEBAR_CONFIG_KEY = 'sidebarConfig'; export interface Setting { key: string; value: string; } // Define the expected row structure from the database if different from Setting // In this case, it seems Setting matches the SELECT columns. type DbSettingRow = Setting; export const settingsRepository = { async getAllSettings(): Promise { try { const db = await getDbInstance(); const rows = await allDb(db, 'SELECT key, value FROM settings'); return rows; } catch (err: any) { console.error('[Repository] 获取所有设置时出错:', err.message); throw new Error('获取设置失败'); } }, async getSetting(key: string): Promise { console.log(`[Repository] Attempting to get setting with key: ${key}`); try { const db = await getDbInstance(); // Use the correct type for the expected row structure const row = await getDbRow<{ value: string }>(db, 'SELECT value FROM settings WHERE key = ?', [key]); const value = row ? row.value : null; console.log(`[Repository] Found value for key ${key}:`, value); return value; } catch (err: any) { console.error(`[Repository] 获取设置项 ${key} 时出错:`, err.message); throw new Error(`获取设置项 ${key} 失败`); } }, async setSetting(key: string, value: string): Promise { const now = Math.floor(Date.now() / 1000); // Use seconds const sql = `INSERT INTO settings (key, value, created_at, updated_at) VALUES (?, ?, ?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value, updated_at = excluded.updated_at`; const params = [key, value, now, now]; console.log(`[Repository] Attempting to set setting. Key: ${key}, Value: ${value}`); console.log(`[Repository] Executing SQL: ${sql} with params: ${JSON.stringify(params)}`); try { const db = await getDbInstance(); const result = await runDb(db, sql, params); console.log(`[Repository] Successfully set setting for key: ${key}. Rows affected: ${result.changes}`); } catch (err: any) { console.error(`[Repository] 设置设置项 ${key} 时出错:`, err.message); throw new Error(`设置设置项 ${key} 失败`); } }, async deleteSetting(key: string): Promise { // Return boolean indicating success console.log(`[Repository] Attempting to delete setting with key: ${key}`); const sql = 'DELETE FROM settings WHERE key = ?'; try { const db = await getDbInstance(); const result = await runDb(db, sql, [key]); console.log(`[Repository] Successfully deleted setting for key: ${key}. Rows affected: ${result.changes}`); return result.changes > 0; // Return true if a row was deleted } catch (err: any) { console.error(`[Repository] 删除设置项 ${key} 时出错:`, err.message); throw new Error(`删除设置项 ${key} 失败`); } }, async setMultipleSettings(settings: Record): Promise { console.log('[Repository] setMultipleSettings called with:', JSON.stringify(settings)); // Use Promise.all with the async setSetting method // Note: 'this' inside map refers to the settingsRepository object correctly here const promises = Object.entries(settings).map(([key, value]) => this.setSetting(key, value) ); try { await Promise.all(promises); console.log('[Repository] setMultipleSettings finished successfully.'); } catch (error) { console.error('[Repository] setMultipleSettings failed:', error); // Re-throw the error or handle it as needed throw new Error('批量设置失败'); } }, }; // --- Specific Setting Getters/Setters --- /** * 获取侧栏配置 * @returns Promise - Returns the parsed config or default */ export const getSidebarConfig = async (): Promise => { const defaultValue: SidebarConfig = { left: [], right: [] }; try { const jsonString = await settingsRepository.getSetting(SIDEBAR_CONFIG_KEY); if (jsonString) { try { const config = JSON.parse(jsonString); // Basic validation if (config && Array.isArray(config.left) && Array.isArray(config.right)) { // TODO: Add deeper validation if needed (e.g., check if items are valid PaneName) return config as SidebarConfig; } console.warn(`[SettingsRepo] Invalid sidebarConfig format found in DB: ${jsonString}. Returning default.`); } catch (parseError) { console.error(`[SettingsRepo] Failed to parse sidebarConfig JSON from DB: ${jsonString}`, parseError); } } } catch (error) { console.error(`[SettingsRepo] Error fetching sidebar config setting (key: ${SIDEBAR_CONFIG_KEY}):`, error); } // Return default if not found, invalid, or error occurred return defaultValue; }; /** * 设置侧栏配置 * @param config - The sidebar configuration object */ export const setSidebarConfig = async (config: SidebarConfig): Promise => { try { // Basic validation before stringifying if (!config || typeof config !== 'object' || !Array.isArray(config.left) || !Array.isArray(config.right)) { throw new Error('Invalid sidebar config object provided.'); } // TODO: Add deeper validation if needed (e.g., check PaneName validity) const jsonString = JSON.stringify(config); await settingsRepository.setSetting(SIDEBAR_CONFIG_KEY, jsonString); } catch (error) { console.error(`[SettingsRepo] Error setting sidebar config (key: ${SIDEBAR_CONFIG_KEY}):`, error); throw new Error('Failed to save sidebar configuration.'); } }; // --- Initialization --- /** * Ensures default settings exist in the settings table. * This function should be called during database initialization. * @param db - The active database instance */ export const ensureDefaultSettingsExist = async (db: sqlite3.Database): Promise => { const defaultSettings: Record = { language: 'en', // Default language ipWhitelistEnabled: 'false', ipWhitelist: '', maxLoginAttempts: '5', loginBanDuration: '300', // 5 minutes in seconds focusSwitcherSequence: JSON.stringify(["quickCommandsSearch", "commandHistorySearch", "fileManagerSearch", "commandInput", "terminalSearch"]), // Default focus sequence navBarVisible: 'true', // Default nav bar visibility layoutTree: 'null', // Default layout tree (null initially) autoCopyOnSelect: 'false', // Default auto copy setting showPopupFileEditor: 'false', // Default popup editor setting shareFileEditorTabs: 'true', // Default editor tab sharing dockerStatusIntervalSeconds: '5', // Default Docker refresh interval dockerDefaultExpand: 'false', // Default Docker expand state statusMonitorIntervalSeconds: '3', // Default Status Monitor interval [SIDEBAR_CONFIG_KEY]: JSON.stringify({ left: [], right: [] }), // Default sidebar config // Add other default settings here }; const nowSeconds = Math.floor(Date.now() / 1000); const sqlInsertOrIgnore = `INSERT OR IGNORE INTO settings (key, value, created_at, updated_at) VALUES (?, ?, ?, ?)`; console.log('[SettingsRepo] Ensuring default settings exist...'); try { for (const [key, value] of Object.entries(defaultSettings)) { await runDb(db, sqlInsertOrIgnore, [key, value, nowSeconds, nowSeconds]); } console.log('[SettingsRepo] Default settings check complete.'); } catch (err: any) { console.error(`[SettingsRepo] Error ensuring default settings:`, err.message); throw new Error(`Failed to ensure default settings: ${err.message}`); } };