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

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;

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();
-  });
-});