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

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

@@ -1,290 +0,0 @@
-import {
-  ImportMessage,
-  ImportSession,
-  ImportSessionGroup,
-  ImportTopic,
-  ImporterEntryData,
-} from '@/types/importer';
-
-interface V2DB_File {
-  /**
-   * create Time
-   */
-  createdAt: number;
-  /**
-   * file data array buffer
-   */
-  data: ArrayBuffer;
-  /**
-   * file type
-   * @example 'image/png'
-   */
-  fileType: string;
-  id: string;
-  metadata: any;
-  /**
-   * file name
-   * @example 'test.png'
-   */
-  name: string;
-  /**
-   * the mode database save the file
-   * local mean save the raw file into data
-   * url mean upload the file to a cdn and then save the url
-   */
-  saveMode: 'local' | 'url';
-  /**
-   * file size
-   */
-  size: number;
-  /**
-   * file url if saveMode is url
-   */
-  url: string;
-}
-
-interface V2DB_MESSAGE {
-  content: string;
-  createdAt: number;
-  error?: any;
-  favorite: 0 | 1;
-  files?: string[];
-  fromModel?: string;
-  fromProvider?: string;
-  id: string;
-  observationId?: string;
-  // foreign keys
-  parentId?: string;
-  plugin?: any;
-  pluginError?: any;
-  pluginState?: any;
-
-  quotaId?: string;
-  role: string;
-  sessionId?: string;
-  tool_call_id?: string;
-  tools?: object[];
-  topicId?: string;
-
-  traceId?: string;
-  translate?: object | false;
-  tts?: any;
-  updatedAt: number;
-}
-
-interface DB_Plugin {
-  createdAt: number;
-  id: string;
-  identifier: string;
-  manifest?: object;
-  settings?: object;
-  type: 'plugin' | 'customPlugin';
-  updatedAt: number;
-}
-
-interface DB_Session {
-  config: object;
-  createdAt: number;
-  group?: string;
-  // 原 Agent 类型
-  id: string;
-  meta: object;
-  pinned?: number;
-  type?: 'agent' | 'group';
-  updatedAt: number;
-}
-
-interface DB_SessionGroup {
-  createdAt: number;
-  id: string;
-  name: string;
-  sort?: number;
-  updatedAt: number;
-}
-
-interface DB_Topic {
-  createdAt: number;
-  favorite?: number;
-  id: string;
-  sessionId?: string;
-  title: string;
-  updatedAt: number;
-}
-
-interface DB_User {
-  avatar?: string;
-  createdAt: number;
-  id: string;
-  settings: object;
-  updatedAt: number;
-  uuid: string;
-}
-
-interface MigrationData {
-  files: V2DB_File[];
-  messages: V2DB_MESSAGE[];
-  plugins: DB_Plugin[];
-  sessionGroups: DB_SessionGroup[];
-  sessions: DB_Session[];
-  topics: DB_Topic[];
-  users: DB_User[];
-}
-
-const LOBE_CHAT_LOCAL_DB_NAME = 'LOBE_CHAT_DB';
-
-const V2DB_LASET_SCHEMA_VERSION = 7;
-export class V2DBReader {
-  private dbName: string = LOBE_CHAT_LOCAL_DB_NAME;
-  private storeNames: string[];
-
-  constructor(storeNames: string[]) {
-    this.storeNames = storeNames;
-  }
-
-  /**
-   * 读取所有数据
-   */
-  async readAllData(): Promise<MigrationData> {
-    try {
-      // 打开数据库连接
-      const db = await this.openDB();
-
-      // 并行读取所有表的数据
-      const results = await Promise.all(
-        this.storeNames.map((storeName) => this.readStore(db, storeName)),
-      );
-
-      // 构建返回结果
-      const migrationData = this.storeNames.reduce((acc, storeName, index) => {
-        // @ts-expect-error
-        acc[storeName] = results[index];
-        return acc;
-      }, {} as MigrationData);
-
-      // 关闭数据库连接
-      db.close();
-
-      return migrationData;
-    } catch (error) {
-      console.error('读取数据库失败:', error);
-      throw error;
-    }
-  }
-
-  async convertToImportData(data: MigrationData): Promise<ImporterEntryData> {
-    // 转换 messages
-    const messages = data.messages.map(
-      (msg): ImportMessage => ({
-        // 使用原有的 id
-        content: msg.content,
-        createdAt: msg.createdAt,
-        // 处理 error
-        error: msg.error || msg.pluginError,
-
-        // 处理额外信息
-        extra: {
-          fromModel: msg.fromModel,
-          fromProvider: msg.fromProvider,
-          translate: msg.translate as any,
-          tts: msg.tts,
-        },
-
-        files: msg.files,
-        id: msg.id,
-
-        // 复制原有字段
-        observationId: msg.observationId,
-        parentId: msg.parentId,
-        plugin: msg.plugin,
-        pluginState: msg.pluginState,
-        quotaId: msg.quotaId,
-        role: msg.role as any,
-        sessionId: msg.sessionId,
-        tool_call_id: msg.tool_call_id,
-        tools: msg.tools as any,
-
-        topicId: msg.topicId,
-
-        traceId: msg.traceId,
-
-        updatedAt: msg.updatedAt,
-      }),
-    );
-
-    // 转换 sessionGroups
-    const sessionGroups = data.sessionGroups.map(
-      (group): ImportSessionGroup => ({
-        createdAt: group.createdAt,
-        id: group.id,
-        // 使用原有的 id
-        name: group.name,
-        sort: group.sort || null,
-        updatedAt: group.updatedAt,
-      }),
-    );
-
-    // 转换 sessions
-    const sessions = data.sessions.map(
-      (session): ImportSession => ({
-        // 使用原有的 id
-        config: session.config as any,
-        createdAt: new Date(session.createdAt).toString(),
-        group: session.group,
-        id: session.id,
-        meta: session.meta as any,
-        pinned: session.pinned ? true : undefined,
-        type: session.type || 'agent',
-        updatedAt: new Date(session.updatedAt).toString(),
-      }),
-    );
-
-    const topics = data.topics.map(
-      (topic): ImportTopic => ({
-        ...topic,
-        favorite: topic.favorite ? true : undefined,
-      }),
-    );
-
-    return {
-      messages,
-      sessionGroups,
-      sessions,
-      topics,
-      version: V2DB_LASET_SCHEMA_VERSION,
-    };
-  }
-
-  /**
-   * 打开数据库
-   */
-  private openDB(): Promise<IDBDatabase> {
-    return new Promise((resolve, reject) => {
-      const request = indexedDB.open(this.dbName);
-
-      // eslint-disable-next-line unicorn/prefer-add-event-listener
-      request.onerror = () => {
-        reject(request.error);
-      };
-      request.onsuccess = () => resolve(request.result);
-    });
-  }
-
-  /**
-   * 读取单个存储对象的所有数据
-   */
-  private readStore(db: IDBDatabase, storeName: string): Promise<any[]> {
-    return new Promise((resolve, reject) => {
-      try {
-        const transaction = db.transaction(storeName, 'readonly');
-        const store = transaction.objectStore(storeName);
-        const request = store.getAll();
-
-        // eslint-disable-next-line unicorn/prefer-add-event-listener
-        request.onerror = () => reject(request.error);
-        request.onsuccess = () => resolve(request.result);
-      } catch (error) {
-        reject(error);
-      }
-    });
-  }
-}

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

@@ -1,35 +0,0 @@
-import { Button } from '@lobehub/ui';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-
-import { BRANDING_NAME } from '@/const/branding';
-
-const ExportConfigButton = memo<{ primary?: boolean; state: any }>(({ state, primary }) => {
-  const { t } = useTranslation('migration');
-
-  const exportData = () => {
-    const config = { exportType: 'sessions', state, version: 1 };
-
-    const url = URL.createObjectURL(
-      new Blob([JSON.stringify(config)], { type: 'application/json' }),
-    );
-
-    const a = document.createElement('a');
-    a.href = url;
-    a.download = `${BRANDING_NAME}-backup-v1.json`;
-
-    document.body.append(a);
-    a.click();
-
-    URL.revokeObjectURL(url);
-    a.remove();
-  };
-
-  return (
-    <Button onClick={exportData} size={'large'} type={primary ? 'primary' : undefined}>
-      {t('dbV1.action.downloadBackup')}
-    </Button>
-  );
-});
-
-export default ExportConfigButton;

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

@@ -1,120 +0,0 @@
-import { Alert, Button, Highlighter, Icon } from '@lobehub/ui';
-import { Popconfirm, Result } from 'antd';
-import { useTheme } from 'antd-style';
-import { createStore, del, get, set } from 'idb-keyval';
-import { ShieldAlert } from 'lucide-react';
-import Link from 'next/link';
-import { lighten } from 'polished';
-import { memo } from 'react';
-import { Trans, useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-import Balancer from 'react-wrap-balancer';
-
-import { GITHUB_ISSUES } from '@/const/url';
-import { githubService } from '@/services/github';
-
-import ExportConfigButton from './ExportConfigButton';
-import UpgradeButton from './UpgradeButton';
-import { MigrationError, UpgradeStatus, V1DB_NAME, V1DB_TABLE_NAME } from './const';
-
-const clearLocal = async () => {
-  const store = createStore(V1DB_NAME, V1DB_TABLE_NAME);
-  // delete the state key and back it up to state_backup for reproduce
-  const state = await get('state', store);
-  await del('state');
-  await set('state_backup', state, store);
-  location.reload();
-};
-
-interface FailedProps {
-  error?: MigrationError;
-  setError: (error: MigrationError) => void;
-  setUpgradeStatus: (status: UpgradeStatus) => void;
-  state: any;
-  upgradeStatus: UpgradeStatus;
-}
-const Failed = memo<FailedProps>(({ error, state, setUpgradeStatus, setError, upgradeStatus }) => {
-  const { t } = useTranslation('migration');
-  const theme = useTheme();
-
-  return (
-    <Result
-      extra={
-        <Flexbox gap={24}>
-          {!!error && (
-            <Alert
-              extra={
-                <Highlighter actionIconSize={'small'} language={'json'}>
-                  {JSON.stringify(error)}
-                </Highlighter>
-              }
-              message={error.message}
-              style={{ flex: 1 }}
-              type={'error'}
-            />
-          )}
-          <Flexbox
-            gap={16}
-            horizontal
-            justify={'center'}
-            style={{
-              alignSelf: 'center',
-              flexWrap: 'wrap',
-            }}
-          >
-            <ExportConfigButton primary state={state} />
-            <Popconfirm
-              arrow={false}
-              okButtonProps={{
-                danger: true,
-                type: 'primary',
-              }}
-              onConfirm={clearLocal}
-              styles={{
-                body: { background: lighten(0.03, theme.colorBgElevated) },
-                root: { width: 340 },
-              }}
-              title={t('dbV1.clear.confirm')}
-            >
-              <Button size={'large'}>{t('dbV1.action.clearDB')}</Button>
-            </Popconfirm>
-            <UpgradeButton
-              primary={false}
-              setError={setError}
-              setUpgradeStatus={setUpgradeStatus}
-              state={state}
-              upgradeStatus={upgradeStatus}
-            >
-              {t('dbV1.action.reUpgrade')}
-            </UpgradeButton>
-          </Flexbox>
-        </Flexbox>
-      }
-      icon={<Icon icon={ShieldAlert} />}
-      status={'error'}
-      style={{ paddingBlock: 24, width: 450 }}
-      subTitle={
-        <Balancer>
-          <Trans i18nKey="dbV1.upgrade.error.subTitle" ns={'migration'}>
-            非常抱歉，数据库升级过程发生异常。请重试升级，或
-            <Link
-              aria-label={'issue'}
-              href={GITHUB_ISSUES}
-              onClick={(e) => {
-                e.preventDefault();
-                githubService.submitDBV1UpgradeError(1, error!);
-              }}
-              target="_blank"
-            >
-              提交问题
-            </Link>
-            我们将会第一时间帮你排查问题。
-          </Trans>
-        </Balancer>
-      }
-      title={t('dbV1.upgrade.error.title')}
-    />
-  );
-});
-
-export default Failed;

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

@@ -1,81 +0,0 @@
-import { Button, Icon } from '@lobehub/ui';
-import { Result } from 'antd';
-import { CheckCircle, CpuIcon } from 'lucide-react';
-import { memo, useState } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Center } from 'react-layout-kit';
-
-import DataStyleModal from '@/components/DataStyleModal';
-import { BRANDING_NAME } from '@/const/branding';
-
-import Failed from './Failed';
-import MigrationStart from './Start';
-import { MigrationError, UpgradeStatus } from './const';
-
-interface MigrationModalProps {
-  open: boolean;
-  setOpen: (open: boolean) => void;
-  state: any;
-}
-
-const MigrationModal = memo<MigrationModalProps>(({ setOpen, open, state: dbState }) => {
-  const { t } = useTranslation('migration');
-  const [upgradeStatus, setUpgradeStatus] = useState<UpgradeStatus>(UpgradeStatus.START);
-
-  const [error, setError] = useState<MigrationError>();
-
-  const close = () => {
-    setOpen(false);
-  };
-
-  const renderContent = () => {
-    switch (upgradeStatus) {
-      case UpgradeStatus.START:
-      case UpgradeStatus.UPGRADING: {
-        return (
-          <MigrationStart
-            setError={setError}
-            setUpgradeStatus={setUpgradeStatus}
-            state={dbState}
-            upgradeStatus={upgradeStatus}
-          />
-        );
-      }
-      case UpgradeStatus.UPGRADED: {
-        return (
-          <Result
-            extra={
-              <Button onClick={close} size={'large'} type={'primary'}>
-                {t('dbV1.action.start')}
-              </Button>
-            }
-            icon={<Icon icon={CheckCircle} />}
-            status={'success'}
-            style={{ paddingBlock: 24 }}
-            subTitle={t('dbV1.upgrade.success.subTitle', { appName: BRANDING_NAME })}
-            title={t('dbV1.upgrade.success.title')}
-          />
-        );
-      }
-      case UpgradeStatus.UPGRADE_FAILED: {
-        return (
-          <Failed
-            error={error}
-            setError={setError}
-            setUpgradeStatus={setUpgradeStatus}
-            state={dbState}
-            upgradeStatus={upgradeStatus}
-          />
-        );
-      }
-    }
-  };
-
-  return (
-    <DataStyleModal icon={CpuIcon} open={open} title={t('dbV1.title', { appName: BRANDING_NAME })}>
-      <Center gap={48}>{renderContent()}</Center>
-    </DataStyleModal>
-  );
-});
-
-export default MigrationModal;

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

@@ -1,108 +0,0 @@
-import { Icon } from '@lobehub/ui';
-import { createStyles } from 'antd-style';
-import { Database, SearchCheck, Zap } from 'lucide-react';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Center, Flexbox } from 'react-layout-kit';
-import Balancer from 'react-wrap-balancer';
-
-import { BRANDING_NAME } from '@/const/branding';
-
-import ExportConfigButton from './ExportConfigButton';
-import UpgradeButton, { UpgradeButtonProps } from './UpgradeButton';
-
-const useStyles = createStyles(({ css, token, isDarkMode, responsive }) => ({
-  desc: css`
-    width: 280px;
-    color: ${token.colorTextSecondary};
-
-    ${responsive.mobile} {
-      line-height: ${token.lineHeight};
-    }
-  `,
-  hint: css`
-    font-size: ${token.fontSizeSM}px;
-    color: ${token.colorTextTertiary};
-    text-align: center;
-  `,
-  icon: css`
-    color: ${isDarkMode ? token.blue : token.geekblue};
-  `,
-  iconCtn: css`
-    width: 72px;
-    height: 72px;
-    border-radius: 50%;
-    background: ${isDarkMode ? token.blue1 : token.geekblue1};
-  `,
-  intro: css`
-    ${responsive.mobile} {
-      width: 350px;
-      margin-block-start: 24px;
-      line-height: ${token.lineHeight};
-    }
-  `,
-
-  title: css`
-    font-size: ${token.fontSizeLG}px;
-    font-weight: bold;
-  `,
-}));
-
-const MigrationStart = memo<UpgradeButtonProps>((props) => {
-  const { t } = useTranslation('migration');
-  const { styles } = useStyles();
-
-  const features = [
-    {
-      avatar: Database,
-      desc: t('dbV1.features.capability.desc'),
-      title: t('dbV1.features.capability.title'),
-    },
-    {
-      avatar: Zap,
-      desc: t('dbV1.features.performance.desc'),
-      title: t('dbV1.features.performance.title'),
-    },
-    {
-      avatar: SearchCheck,
-      desc: t('dbV1.features.use.desc'),
-      title: t('dbV1.features.use.title'),
-    },
-  ];
-
-  return (
-    <>
-      <Flexbox>
-        <Flexbox className={styles.intro} style={{ textAlign: 'center' }} width={460}>
-          {t('dbV1.description', { appName: BRANDING_NAME })}
-        </Flexbox>
-      </Flexbox>
-      <Flexbox gap={32}>
-        {features.map((item) => {
-          return (
-            <Flexbox align={'flex-start'} gap={24} horizontal key={item.title}>
-              <Center className={styles.iconCtn}>
-                <Icon className={styles.icon} icon={item.avatar} size={36} />
-              </Center>
-              <Flexbox gap={8}>
-                <p className={styles.title}>{item.title}</p>
-                <p className={styles.desc}>{item.desc}</p>
-              </Flexbox>
-            </Flexbox>
-          );
-        })}
-      </Flexbox>
-      <Flexbox align={'center'} gap={16} style={{ paddingBottom: 16 }}>
-        <Flexbox gap={16} horizontal justify={'center'} style={{ flexWrap: 'wrap' }}>
-          <ExportConfigButton state={props.state} />
-          <UpgradeButton {...props} />
-        </Flexbox>
-        <Flexbox align={'center'} className={styles.hint}>
-          <Balancer>{t('dbV1.upgradeTip', { appName: BRANDING_NAME })}</Balancer>
-        </Flexbox>
-      </Flexbox>
-    </>
-  );
-});
-
-export default MigrationStart;

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

@@ -1,71 +0,0 @@
-import { Button } from '@lobehub/ui';
-import { ReactNode, memo } from 'react';
-import { useTranslation } from 'react-i18next';
-
-import { ClientService } from '@/services/import/_deprecated';
-import { useChatStore } from '@/store/chat';
-import { useSessionStore } from '@/store/session';
-
-import { MigrationError, UpgradeStatus } from './const';
-
-export interface UpgradeButtonProps {
-  children?: ReactNode;
-  primary?: boolean;
-  setError: (error: MigrationError) => void;
-  setUpgradeStatus: (status: UpgradeStatus) => void;
-  state: any;
-  upgradeStatus: UpgradeStatus;
-}
-
-const UpgradeButton = memo<UpgradeButtonProps>(
-  ({ setUpgradeStatus, upgradeStatus, state, setError, primary = true, children }) => {
-    const { t } = useTranslation('migration');
-
-    const refreshSession = useSessionStore((s) => s.refreshSessions);
-    const [refreshMessages, refreshTopic] = useChatStore((s) => [
-      s.refreshMessages,
-      s.refreshTopic,
-    ]);
-
-    const upgrade = async () => {
-      try {
-        setUpgradeStatus(UpgradeStatus.UPGRADING);
-
-        const configService = new ClientService();
-        await configService.importConfigState({
-          exportType: 'sessions',
-          state: state,
-          version: 7,
-        });
-
-        await refreshSession();
-        await refreshMessages();
-        await refreshTopic();
-
-        localStorage.setItem('V2DB_IS_MIGRATED', '1');
-
-        setUpgradeStatus(UpgradeStatus.UPGRADED);
-
-        return { success: true };
-      } catch (error) {
-        setUpgradeStatus(UpgradeStatus.UPGRADE_FAILED);
-        const err = error as { message: string; stack: string };
-
-        setError({ message: err.message, stack: err.stack });
-      }
-    };
-
-    return (
-      <Button
-        color={primary ? undefined : 'default'}
-        loading={upgradeStatus === UpgradeStatus.UPGRADING}
-        onClick={upgrade}
-        size={'large'}
-      >
-        {children ?? t('dbV1.action.upgrade')}
-      </Button>
-    );
-  },
-);
-
-export default UpgradeButton;