@lobehub/lobehub 2.0.0-next.13 → 2.0.0-next.14

package/src/app/[variants]/(main)/chat/features/Migration/const.ts

@@ -1,15 +0,0 @@
-export const MIGRATE_KEY = 'migrated';
-export enum UpgradeStatus {
-  START,
-  UPGRADING,
-  UPGRADED,
-  UPGRADE_FAILED,
-}
-
-export const V1DB_NAME = 'LobeHub';
-export const V1DB_TABLE_NAME = 'LOBE_CHAT';
-
-export interface MigrationError {
-  message: string;
-  stack: string;
-}

package/src/app/[variants]/(main)/chat/features/Migration/index.tsx

@@ -1,50 +0,0 @@
-'use client';
-
-import { Spin } from 'antd';
-import dynamic from 'next/dynamic';
-import { memo, useEffect, useState } from 'react';
-
-import { isServerMode } from '@/const/version';
-import { useGlobalStore } from '@/store/global';
-import { systemStatusSelectors } from '@/store/global/selectors';
-
-import { V2DBReader } from './DBReader';
-
-const Modal = dynamic(() => import('./Modal'), { loading: () => <Spin fullscreen />, ssr: false });
-
-const Migration = memo(() => {
-  const [dbState, setDbState] = useState(null);
-  const [open, setOpen] = useState(false);
-
-  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
-
-  const checkMigration = async () => {
-    const isMigrated = localStorage.getItem('V2DB_IS_MIGRATED');
-    // if db have migrated already, don't show modal
-    if (isMigrated || isServerMode) return;
-
-    const dbReader = new V2DBReader([
-      'messages',
-      'files',
-      'plugins',
-      'sessionGroups',
-      'sessions',
-      'topics',
-      'users',
-    ]);
-    const data = await dbReader.readAllData();
-    console.log('migration data:', data);
-    const state = await dbReader.convertToImportData(data);
-    console.log('import state', state);
-    setDbState(state as any);
-    setOpen(true);
-  };
-
-  useEffect(() => {
-    if (isPgliteInited) checkMigration();
-  }, [isPgliteInited]);
-
-  return open && <Modal open={open} setOpen={setOpen} state={dbState} />;
-});
-
-export default Migration;

package/src/app/[variants]/loading/Client/Content.tsx

@@ -1,48 +0,0 @@
-import dynamic from 'next/dynamic';
-import React, { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-
-import FullscreenLoading from '@/components/Loading/FullscreenLoading';
-import { useGlobalStore } from '@/store/global';
-import { systemStatusSelectors } from '@/store/global/selectors';
-import { DatabaseLoadingState } from '@/types/clientDB';
-
-import { CLIENT_LOADING_STAGES } from '../stage';
-
-const InitError = dynamic(() => import('./Error'), { ssr: false });
-
-interface InitProps {
-  setActiveStage: (value: string) => void;
-}
-
-const Init = memo<InitProps>(({ setActiveStage }) => {
-  const useInitClientDB = useGlobalStore((s) => s.useInitClientDB);
-
-  useInitClientDB({ onStateChange: setActiveStage });
-
-  return null;
-});
-
-interface ContentProps {
-  loadingStage: string;
-  setActiveStage: (value: string) => void;
-}
-
-const Content = memo<ContentProps>(({ loadingStage, setActiveStage }) => {
-  const { t } = useTranslation('common');
-  const isPgliteNotInited = useGlobalStore(systemStatusSelectors.isPgliteNotInited);
-  const isError = useGlobalStore((s) => s.initClientDBStage === DatabaseLoadingState.Error);
-
-  return (
-    <>
-      {isPgliteNotInited && <Init setActiveStage={setActiveStage} />}
-      <FullscreenLoading
-        activeStage={CLIENT_LOADING_STAGES.indexOf(loadingStage)}
-        contentRender={isError && <InitError />}
-        stages={CLIENT_LOADING_STAGES.map((key) => t(`appLoading.${key}` as any))}
-      />
-    </>
-  );
-});
-
-export default Content;

package/src/app/[variants]/loading/Client/Error.tsx

@@ -1,27 +0,0 @@
-import { Button } from '@lobehub/ui';
-import React from 'react';
-import { useTranslation } from 'react-i18next';
-import { Center } from 'react-layout-kit';
-
-import ErrorResult from '@/features/InitClientDB/ErrorResult';
-
-const InitError = () => {
-  const { t } = useTranslation('common');
-
-  return (
-    <ErrorResult>
-      {({ setOpen }) => (
-        <Center gap={8}>
-          {t('appLoading.failed')}
-          <div>
-            <Button onClick={() => setOpen(true)} type={'primary'}>
-              {t('appLoading.showDetail')}
-            </Button>
-          </div>
-        </Center>
-      )}
-    </ErrorResult>
-  );
-};
-
-export default InitError;

package/src/app/[variants]/loading/Client/Redirect.tsx

@@ -1,47 +0,0 @@
-'use client';
-
-import { useRouter } from 'next/navigation';
-import { memo, useEffect } from 'react';
-
-import { useGlobalStore } from '@/store/global';
-import { systemStatusSelectors } from '@/store/global/selectors';
-import { useUserStore } from '@/store/user';
-
-import { AppLoadingStage } from '../stage';
-
-interface RedirectProps {
-  setActiveStage: (value: AppLoadingStage) => void;
-}
-
-const Redirect = memo<RedirectProps>(({ setActiveStage }) => {
-  const router = useRouter();
-  const isUserStateInit = useUserStore((s) => s.isUserStateInit);
-
-  const isPgliteNotEnabled = useGlobalStore(systemStatusSelectors.isPgliteNotEnabled);
-
-  const navToChat = () => {
-    setActiveStage(AppLoadingStage.GoToChat);
-    router.replace('/chat');
-  };
-
-  useEffect(() => {
-    // if pglite is not enabled, redirect to chat
-    if (isPgliteNotEnabled) {
-      navToChat();
-      return;
-    }
-
-    // if user state not init, wait for loading
-    if (!isUserStateInit) {
-      setActiveStage(AppLoadingStage.InitUser);
-      return;
-    }
-
-    // finally check the conversation status
-    navToChat();
-  }, [isUserStateInit, isPgliteNotEnabled]);
-
-  return null;
-});
-
-export default Redirect;

package/src/app/[variants]/loading/Client/index.tsx

@@ -1,22 +0,0 @@
-'use client';
-
-import { useState } from 'react';
-
-import { AppLoadingStage } from '../stage';
-import Content from './Content';
-import Redirect from './Redirect';
-
-const ClientMode = () => {
-  const [activeStage, setActiveStage] = useState<string>(AppLoadingStage.Initializing);
-
-  return (
-    <>
-      <Content loadingStage={activeStage} setActiveStage={setActiveStage} />
-      <Redirect setActiveStage={setActiveStage} />
-    </>
-  );
-};
-
-ClientMode.displayName = 'ClientMode';
-
-export default ClientMode;

package/src/components/InnerLink.tsx

@@ -1,20 +0,0 @@
-'use client';
-
-import Link, { LinkProps } from 'next/link';
-import { AnchorHTMLAttributes, ReactNode } from 'react';
-
-import { useServerConfigStore } from '@/store/serverConfig';
-
-interface InnerLinkProps
-  extends Omit<AnchorHTMLAttributes<HTMLAnchorElement>, keyof LinkProps>,
-    LinkProps {
-  children?: ReactNode | undefined;
-}
-
-const InnerLink = ({ href, ...props }: InnerLinkProps) => {
-  const variants = useServerConfigStore((s) => s.segmentVariants);
-
-  return <Link {...props} as={href} href={`/${variants}${href}`} />;
-};
-
-export default InnerLink;

package/src/database/_deprecated/core/__tests__/db-upgrade.test.ts

@@ -1,42 +0,0 @@
-import Dexie from 'dexie';
-import { afterEach, beforeEach, describe, expect, it } from 'vitest';
-
-import { BrowserDB } from '../db';
-import { dbSchemaV3 } from '../schemas';
-import { LOBE_CHAT_LOCAL_DB_NAME } from '../types/db';
-
-beforeEach(async () => {
-  // 确保在测试开始前删除数据库
-  await Dexie.delete(LOBE_CHAT_LOCAL_DB_NAME);
-});
-
-afterEach(async () => {
-  // 确保在测试结束后删除数据库
-  await Dexie.delete(LOBE_CHAT_LOCAL_DB_NAME);
-});
-
-describe('LocalDB migration', () => {
-  it('should upgrade from version 3 to version 4 correctly', async () => {
-    // 使用版本3的schema创建数据库实例并填充测试数据
-    const dbV3 = new Dexie(LOBE_CHAT_LOCAL_DB_NAME);
-    dbV3.version(3).stores(dbSchemaV3);
-    await dbV3.open();
-    await dbV3.table('sessions').bulkAdd([
-      { id: 's1', group: 'pinned' },
-      { id: 's2', group: 'default' },
-    ]);
-    dbV3.close();
-
-    // 创建新的数据库实例，包含版本4的迁移逻辑
-    const dbV4 = new BrowserDB();
-    await dbV4.open();
-
-    // 验证迁移后的数据
-    const updatedSession1 = await dbV4.sessions.get('s1');
-    const updatedSession2 = await dbV4.sessions.get('s2');
-    expect(updatedSession1).toEqual({ id: 's1', pinned: 1, group: 'default' });
-    expect(updatedSession2).toEqual({ id: 's2', pinned: 0, group: 'default' });
-
-    dbV4.close();
-  });
-});

package/src/database/_deprecated/core/__tests__/db.test.ts

@@ -1,79 +0,0 @@
-import { afterEach, beforeEach, describe, expect, it } from 'vitest';
-
-import { DB_File } from '../../schemas/files';
-import { BrowserDB } from '../db';
-import { DBModel } from '../types/db';
-
-describe('LocalDB', () => {
-  let db: BrowserDB;
-
-  beforeEach(() => {
-    db = new BrowserDB();
-  });
-
-  afterEach(async () => {
-    await db.delete();
-    db.close();
-  });
-
-  it('should be instantiated with the correct schema', () => {
-    const filesTable = db.files;
-
-    expect(filesTable).toBeDefined();
-  });
-
-  it('should allow adding a file', async () => {
-    const file: DBModel<DB_File> = {
-      id: 'file1',
-      name: 'testfile.txt',
-      data: new ArrayBuffer(3),
-      saveMode: 'local',
-      fileType: 'plain/text',
-      size: 3,
-      createdAt: Date.now(),
-      updatedAt: Date.now(),
-    };
-
-    await db.files.add(file);
-
-    expect(await db.files.get(file.id)).toEqual(file);
-  });
-
-  it('should allow updating a file', async () => {
-    const file: DBModel<DB_File> = {
-      id: 'file1',
-      name: 'testfile.txt',
-      data: new ArrayBuffer(3),
-      saveMode: 'local',
-      fileType: 'plain/text',
-      size: 3,
-      createdAt: Date.now(),
-      updatedAt: Date.now(),
-    };
-    await db.files.add(file);
-
-    // Act
-    await db.files.update(file.id, { name: 'update.txt' });
-
-    // Assert
-    expect(await db.files.get(file.id)).toHaveProperty('name', 'update.txt');
-  });
-
-  it('should allow deleting a file', async () => {
-    const file: DBModel<DB_File> = {
-      id: 'file1',
-      name: 'testfile.txt',
-      data: new ArrayBuffer(3),
-      saveMode: 'local',
-      fileType: 'plain/text',
-      size: 3,
-      updatedAt: Date.now(),
-      createdAt: Date.now(),
-    };
-    await db.files.add(file);
-
-    await db.files.delete(file.id);
-
-    expect(await db.files.get(file.id)).toBeUndefined();
-  });
-});

package/src/database/_deprecated/core/__tests__/model.test.ts

@@ -1,55 +0,0 @@
-import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
-import { z } from 'zod';
-
-import { BaseModel } from '../model';
-
-// Define a mock schema for testing
-const mockSchema = z.object({
-  name: z.string(),
-  content: z.string(),
-});
-
-// Define a mock table name
-const mockTableName = 'files';
-
-describe('BaseModel', () => {
-  let baseModel: BaseModel<typeof mockTableName>;
-
-  beforeEach(() => {
-    baseModel = new BaseModel(mockTableName, mockSchema);
-    // Mock the console.error to test error logging
-    vi.spyOn(console, 'error').mockImplementation(() => {});
-  });
-
-  afterEach(() => {
-    vi.restoreAllMocks();
-    // console.error = originalConsoleError;
-  });
-
-  it('should have a table property', () => {
-    expect(baseModel.table).toBeDefined();
-  });
-
-  describe('add method', () => {
-    it('should add a valid record to the database', async () => {
-      const validData = {
-        name: 'testfile.txt',
-        content: 'Hello, World!',
-      };
-
-      const result = await baseModel['_addWithSync'](validData);
-
-      expect(result).toHaveProperty('id');
-      expect(console.error).not.toHaveBeenCalled();
-    });
-
-    it('should throw an error and log to console when adding an invalid record', async () => {
-      const invalidData = {
-        name: 123, // Invalid type, should be a string
-        content: 'Hello, World!',
-      };
-
-      await expect(baseModel['_addWithSync'](invalidData)).rejects.toThrow(TypeError);
-    });
-  });
-});

package/src/database/_deprecated/core/db.ts

@@ -1,246 +0,0 @@
-import Dexie, { Transaction } from 'dexie';
-
-import { MigrationLLMSettings } from '@/migrations/FromV3ToV4';
-import { MigrationAgentChatConfig } from '@/migrations/FromV5ToV6';
-import { MigrationKeyValueSettings } from '@/migrations/FromV6ToV7';
-import { uuid } from '@/utils/uuid';
-
-import { DB_File } from '../schemas/files';
-import { DB_Message } from '../schemas/message';
-import { DB_Plugin } from '../schemas/plugin';
-import { DB_Session } from '../schemas/session';
-import { DB_SessionGroup } from '../schemas/sessionGroup';
-import { DB_Topic } from '../schemas/topic';
-import { DB_User } from '../schemas/user';
-import { migrateSettingsToUser } from './migrations/migrateSettingsToUser';
-import {
-  dbSchemaV1,
-  dbSchemaV2,
-  dbSchemaV3,
-  dbSchemaV4,
-  dbSchemaV5,
-  dbSchemaV6,
-  dbSchemaV7,
-  dbSchemaV9,
-} from './schemas';
-import { DBModel, LOBE_CHAT_LOCAL_DB_NAME } from './types/db';
-
-export interface LobeDBSchemaMap {
-  files: DB_File;
-  messages: DB_Message;
-  plugins: DB_Plugin;
-  sessionGroups: DB_SessionGroup;
-  sessions: DB_Session;
-  topics: DB_Topic;
-  users: DB_User;
-}
-
-// Define a local DB
-export class BrowserDB extends Dexie {
-  public files: BrowserDBTable<'files'>;
-  public sessions: BrowserDBTable<'sessions'>;
-  public messages: BrowserDBTable<'messages'>;
-  public topics: BrowserDBTable<'topics'>;
-  public plugins: BrowserDBTable<'plugins'>;
-  public sessionGroups: BrowserDBTable<'sessionGroups'>;
-  public users: BrowserDBTable<'users'>;
-
-  constructor() {
-    super(LOBE_CHAT_LOCAL_DB_NAME);
-    this.version(1).stores(dbSchemaV1);
-    this.version(2).stores(dbSchemaV2);
-    this.version(3).stores(dbSchemaV3);
-    this.version(4)
-      .stores(dbSchemaV4)
-      .upgrade((trans) => this.upgradeToV4(trans));
-
-    this.version(5)
-      .stores(dbSchemaV5)
-      .upgrade((trans) => this.upgradeToV5(trans));
-
-    this.version(6)
-      .stores(dbSchemaV6)
-      .upgrade((trans) => this.upgradeToV6(trans));
-
-    this.version(7)
-      .stores(dbSchemaV7)
-      .upgrade((trans) => this.upgradeToV7(trans));
-
-    this.version(8)
-      .stores(dbSchemaV7)
-      .upgrade((trans) => this.upgradeToV8(trans));
-
-    this.version(9)
-      .stores(dbSchemaV9)
-      .upgrade((trans) => this.upgradeToV9(trans));
-
-    this.version(10)
-      .stores(dbSchemaV9)
-      .upgrade((trans) => this.upgradeToV10(trans));
-
-    this.version(11)
-      .stores(dbSchemaV9)
-      .upgrade((trans) => this.upgradeToV11(trans));
-
-    this.files = this.table('files');
-    this.sessions = this.table('sessions');
-    this.messages = this.table('messages');
-    this.topics = this.table('topics');
-    this.plugins = this.table('plugins');
-    this.sessionGroups = this.table('sessionGroups');
-    this.users = this.table('users');
-  }
-
-  /**
-   * 2024.01.22
-   *
-   * DB V3 to V4
-   * from `group = pinned` to `pinned:true`
-   */
-  upgradeToV4 = async (trans: Transaction) => {
-    const sessions = trans.table('sessions');
-    await sessions.toCollection().modify((session) => {
-      // translate boolean to number
-      session.pinned = session.group === 'pinned' ? 1 : 0;
-      session.group = 'default';
-    });
-  };
-
-  /**
-   * 2024.01.29
-   * settings from localStorage to indexedDB
-   */
-  upgradeToV5 = async (trans: Transaction) => {
-    const users = trans.table('users');
-
-    // if no user, create one
-    if ((await users.count()) === 0) {
-      const data = localStorage.getItem('LOBE_SETTINGS');
-
-      if (data) {
-        let json;
-
-        try {
-          json = JSON.parse(data);
-        } catch {
-          /* empty */
-        }
-
-        if (!json?.state?.settings) return;
-
-        const settings = json.state.settings;
-
-        const user = migrateSettingsToUser(settings);
-        await users.add(user);
-      }
-    }
-  };
-
-  /**
-   * 2024.02.27
-   * add uuid to user
-   */
-  upgradeToV6 = async (trans: Transaction) => {
-    const users = trans.table('users');
-
-    await users.toCollection().modify((user: DB_User) => {
-      if (!user.uuid) user.uuid = uuid();
-    });
-  };
-
-  /**
-   * 2024.03.14
-   * add `id` in plugins
-   */
-  upgradeToV7 = async (trans: Transaction) => {
-    const plugins = trans.table('plugins');
-
-    await plugins.toCollection().modify((plugin: DB_Plugin) => {
-      plugin.id = plugin.identifier;
-    });
-  };
-
-  upgradeToV8 = async (trans: Transaction) => {
-    const users = trans.table('users');
-    await users.toCollection().modify((user: DB_User) => {
-      if (user.settings) {
-        user.settings = MigrationLLMSettings.migrateSettings(user.settings as any);
-      }
-    });
-  };
-
-  /**
-   * 2024.05.11
-   *
-   * message role=function to role=tool
-   */
-  upgradeToV9 = async (trans: Transaction) => {
-    const messages = trans.table('messages');
-    await messages.toCollection().modify(async (message: DBModel<DB_Message>) => {
-      if ((message.role as string) === 'function') {
-        const origin = Object.assign({}, message);
-
-        const toolCallId = `tool_call_${message.id}`;
-        const assistantMessageId = `tool_calls_${message.id}`;
-
-        message.role = 'tool';
-        message.tool_call_id = toolCallId;
-        message.parentId = assistantMessageId;
-
-        await messages.add({
-          ...origin,
-          content: '',
-          createdAt: message.createdAt - 10,
-          error: undefined,
-          id: assistantMessageId,
-          role: 'assistant',
-          tools: [{ ...message.plugin!, id: toolCallId }],
-          updatedAt: message.updatedAt - 10,
-        } as DBModel<DB_Message>);
-      }
-    });
-  };
-
-  /**
-   * 2024.05.25
-   * migrate some agent config to chatConfig
-   */
-  upgradeToV10 = async (trans: Transaction) => {
-    const sessions = trans.table('sessions');
-    await sessions.toCollection().modify(async (session: DBModel<DB_Session>) => {
-      if (session.config)
-        session.config = MigrationAgentChatConfig.migrateChatConfig(session.config as any);
-    });
-  };
-
-  /**
-   * 2024.05.27
-   * migrate apiKey in languageModel to keyVaults
-   */
-  upgradeToV11 = async (trans: Transaction) => {
-    const users = trans.table('users');
-
-    await users.toCollection().modify((user: DB_User) => {
-      if (user.settings) {
-        user.settings = MigrationKeyValueSettings.migrateSettings(user.settings as any);
-      }
-    });
-  };
-}
-
-export const browserDB = new BrowserDB();
-
-// ================================================ //
-// ================================================ //
-// ================================================ //
-// ================================================ //
-// ================================================ //
-
-// types helper
-export type BrowserDBSchema = {
-  [t in keyof LobeDBSchemaMap]: {
-    model: LobeDBSchemaMap[t];
-    table: Dexie.Table<DBModel<LobeDBSchemaMap[t]>, string>;
-  };
-};
-type BrowserDBTable<T extends keyof LobeDBSchemaMap> = BrowserDBSchema[T]['table'];

package/src/database/_deprecated/core/index.ts

@@ -1,2 +0,0 @@
-export { browserDB } from './db';
-export * from './model';