@lobehub/chat 1.74.2 → 1.74.3

package/src/locales/default/common.ts

@@ -41,9 +41,12 @@ export default {
       title: '初始化 PGlite 数据库',
     },
     error: {
-      desc: '非常抱歉，Pglite 数据库初始化过程发生异常。请点击按钮重试。如多次重试后仍重复出错，请 <1>提交问题</1> ，我们将会第一时间帮你排查',
+      desc: '非常抱歉，Pglite 数据库初始化过程发生异常，请点击右侧按钮重试。如果你具有 SQL 运维经验，可以尝试点击下方自助解决按钮自行解决。如多次重试后仍重复出错，请点击左下方反馈问题，我们将会第一时间帮你排查',
       detail: '错误原因：[{{type}}] {{message}}，明细如下：',
+      detailTitle: '错误原因',
+      report: '反馈问题',
       retry: '重试',
+      selfSolve: '自助解决',
       title: '数据库初始化失败',
     },
     initing: {
@@ -83,6 +86,55 @@ export default {
       desc: '立即想用',
       title: 'PGlite 数据库已就绪',
     },
+
+    solve: {
+      backup: {
+        backup: '备份',
+        backupSuccess: '备份成功',
+        desc: '导出当前数据库中的关键数据',
+        export: '导出全部数据',
+        exportDesc: '导出的数据将以JSON格式保存，可用于后续恢复或分析。',
+        reset: {
+          alert: '警告',
+          alertDesc: '以下操作可能会导致数据丢失。请确保您已备份重要数据后再继续。',
+          button: '完全重置数据库（删除所有数据）',
+          confirm: {
+            desc: '此操作将会删除所有数据并且不可撤销，确认继续吗？',
+            title: '确认重置数据库',
+          },
+          desc: '在迁移无法恢复的情况下重置数据库',
+          title: '数据库重置',
+        },
+        restore: '恢复',
+        restoreSuccess: '恢复成功',
+        title: '数据备份',
+      },
+      diagnosis: {
+        createdAt: '创建时间',
+        migratedAt: '迁移完成时间',
+        sql: '迁移SQL',
+        title: '迁移状态',
+      },
+      repair: {
+        desc: '手动管理迁移状态',
+        runSQL: '自定义执行',
+        sql: {
+          clear: '清空',
+          desc: '执行自定义SQL语句修复数据库问题',
+          markFinished: '标记为已完成',
+          placeholder: '输入SQL语句...',
+          result: '执行结果',
+          run: '执行',
+          title: 'SQL执行器',
+        },
+        title: '迁移控制',
+      },
+      tabs: {
+        backup: '备份恢复',
+        diagnosis: '诊断',
+        repair: '修复',
+      },
+    },
   },
   close: '关闭',
   contact: '联系我们',

package/src/store/global/actions/clientDb.ts

@@ -31,8 +31,14 @@ export const clientDBSlice: StateCreator<
 
     const { initializeDB } = await import('@/database/client/db');
     await initializeDB({
-      onError: (error) => {
-        set({ initClientDBError: error });
+      onError: ({ error, migrationsSQL, migrationTableItems }) => {
+        set({
+          initClientDBError: error,
+          initClientDBMigrations: {
+            sqls: migrationsSQL,
+            tableRecords: migrationTableItems,
+          },
+        });
       },
       onProgress: (data) => {
         set({ initClientDBProcess: data });
@@ -43,8 +49,18 @@ export const clientDBSlice: StateCreator<
       },
     });
   },
-  markPgliteEnabled: () => {
+  markPgliteEnabled: async () => {
     get().updateSystemStatus({ isEnablePglite: true });
+
+    if (navigator.storage && !!navigator.storage.persist) {
+      // 1. Check if persistent permission has been obtained
+      const isPersisted = await navigator.storage.persisted();
+
+      // 2. If the persistent permission has not been obtained, request permission
+      if (!isPersisted) {
+        await navigator.storage.persist();
+      }
+    }
   },
   useInitClientDB: (params) =>
     useOnlyFetchOnceSWR('initClientDB', () => get().initializeClientDB(params)),

package/src/store/global/initialState.ts

@@ -1,7 +1,7 @@
 import type { ThemeMode } from 'antd-style';
 import { AppRouterInstance } from 'next/dist/shared/lib/app-router-context.shared-runtime';
 
-import { DatabaseLoadingState } from '@/types/clientDB';
+import { DatabaseLoadingState, MigrationSQL, MigrationTableItem } from '@/types/clientDB';
 import { LocaleMode } from '@/types/locale';
 import { SessionDefaultGroup } from '@/types/session';
 import { AsyncLocalStorage } from '@/utils/localStorage';
@@ -73,6 +73,11 @@ export interface SystemStatus {
 export interface GlobalState {
   hasNewVersion?: boolean;
   initClientDBError?: Error;
+  initClientDBMigrations?: {
+    sqls: MigrationSQL[];
+    tableRecords: MigrationTableItem[];
+  };
+
   initClientDBProcess?: { costTime?: number; phase: 'wasm' | 'dependencies'; progress: number };
   /**
    * 客户端数据库初始化状态

package/src/store/global/selectors/clientDB.ts

@@ -0,0 +1,43 @@
+import { GlobalState } from '@/store/global/initialState';
+
+const initClientDBMigrationSqls = (s: GlobalState) => {
+  return s.initClientDBMigrations?.sqls || [];
+};
+
+const displayMigrationStatus = (s: GlobalState) => {
+  const sql = s.initClientDBMigrations?.sqls || [];
+  const tableRecords = s.initClientDBMigrations?.tableRecords || [];
+
+  return (
+    sql
+
+      .map((item, index) => {
+        const recordInTable = tableRecords.find((record) => record.hash === item.hash);
+        return {
+          createdAt: new Date(item.folderMillis),
+          desc: item.sql[0],
+          folderMillis: item.folderMillis,
+          id: item.hash,
+          index: index + 1,
+          migratedAt: recordInTable ? new Date(recordInTable.created_at) : undefined,
+          sql: item.sql,
+          status: !!recordInTable ? 'success' : 'error',
+        };
+      })
+      // 时间倒序
+      .sort((a, b) => b.folderMillis - a.folderMillis)
+  );
+};
+
+const errorMigrations = (s: GlobalState) => {
+  const sql = s.initClientDBMigrations?.sqls || [];
+  const tableRecords = s.initClientDBMigrations?.tableRecords || [];
+
+  return sql.filter((item) => !tableRecords.some((record) => record.hash === item.hash));
+};
+
+export const clientDBSelectors = {
+  displayMigrationStatus,
+  errorMigrations,
+  initClientDBMigrationSqls,
+};

package/src/store/global/selectors/index.ts

@@ -1,2 +1,3 @@
+export * from './clientDB';
 export * from './general';
 export * from './systemStatus';

package/src/store/user/slices/settings/selectors/general.test.ts

@@ -5,6 +5,20 @@ import { merge } from '@/utils/merge';
 import { userGeneralSettingsSelectors } from './general';
 
 describe('settingsSelectors', () => {
+  describe('generalConfig', () => {
+    it('should return general settings', () => {
+      const s: UserState = merge(initialState, {
+        settings: {
+          general: { fontSize: 12 },
+        },
+      });
+
+      const result = userGeneralSettingsSelectors.config(s as UserStore);
+
+      expect(result).toEqual({ fontSize: 12 });
+    });
+  });
+
   describe('fontSize', () => {
     it('should return the fontSize', () => {
       const s: UserState = merge(initialState, {
@@ -18,4 +32,80 @@ describe('settingsSelectors', () => {
       expect(result).toBe(12);
     });
   });
+
+  describe('neutralColor', () => {
+    it('should return undefined if general settings not exists', () => {
+      const s: UserState = merge(initialState, {
+        settings: {
+          general: undefined,
+        },
+      });
+
+      const result = userGeneralSettingsSelectors.neutralColor(s as UserStore);
+
+      expect(result).toBeUndefined();
+    });
+
+    it('should return undefined if neutralColor not set', () => {
+      const s: UserState = merge(initialState, {
+        settings: {
+          general: {},
+        },
+      });
+
+      const result = userGeneralSettingsSelectors.neutralColor(s as UserStore);
+
+      expect(result).toBeUndefined();
+    });
+
+    it('should return the neutralColor', () => {
+      const s: UserState = merge(initialState, {
+        settings: {
+          general: { neutralColor: '#000000' },
+        },
+      });
+
+      const result = userGeneralSettingsSelectors.neutralColor(s as UserStore);
+
+      expect(result).toBe('#000000');
+    });
+  });
+
+  describe('primaryColor', () => {
+    it('should return undefined if general settings not exists', () => {
+      const s: UserState = merge(initialState, {
+        settings: {
+          general: undefined,
+        },
+      });
+
+      const result = userGeneralSettingsSelectors.primaryColor(s as UserStore);
+
+      expect(result).toBeUndefined();
+    });
+
+    it('should return undefined if primaryColor not set', () => {
+      const s: UserState = merge(initialState, {
+        settings: {
+          general: {},
+        },
+      });
+
+      const result = userGeneralSettingsSelectors.primaryColor(s as UserStore);
+
+      expect(result).toBeUndefined();
+    });
+
+    it('should return the primaryColor', () => {
+      const s: UserState = merge(initialState, {
+        settings: {
+          general: { primaryColor: '#ffffff' },
+        },
+      });
+
+      const result = userGeneralSettingsSelectors.primaryColor(s as UserStore);
+
+      expect(result).toBe('#ffffff');
+    });
+  });
 });

package/src/types/clientDB.ts

@@ -27,3 +27,16 @@ export interface ClientDBLoadingProgress {
 }
 
 export type OnStageChange = (state: DatabaseLoadingState) => void;
+
+export interface MigrationSQL {
+  bps: boolean;
+  folderMillis: number;
+  hash: string;
+  sql: string[];
+}
+
+export interface MigrationTableItem {
+  created_at: number;
+  hash: string;
+  id: number;
+}