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

package/src/features/InitClientDB/features/DatabaseRepair/Repair.tsx

@@ -1,218 +0,0 @@
-'use client';
-
-import { Alert, Button, CodeEditor, Icon, Text } from '@lobehub/ui';
-import { Card, List } from 'antd';
-import { createStyles } from 'antd-style';
-import dayjs from 'dayjs';
-import isEqual from 'fast-deep-equal';
-import { AlertCircle, CheckCircle, Play } from 'lucide-react';
-import { memo, useState } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import { clientDB, updateMigrationRecord } from '@/database/client/db';
-import { useGlobalStore } from '@/store/global';
-import { clientDBSelectors } from '@/store/global/selectors';
-
-// 使用 antd-style 创建样式
-const useStyles = createStyles(({ css, token }) => ({
-  actionBar: css`
-    display: flex;
-    justify-content: flex-end;
-    margin-block-start: ${token.marginSM}px;
-  `,
-  card: css`
-    margin-block-end: ${token.marginMD}px;
-    border-radius: ${token.borderRadiusLG}px;
-  `,
-  codeBlock: css`
-    overflow-x: auto;
-
-    margin-block: ${token.marginSM}px;
-    margin-inline: 0;
-    padding: ${token.paddingSM}px;
-    border-radius: ${token.borderRadiusSM}px;
-
-    font-family: ${token.fontFamilyCode};
-    font-size: 12px;
-
-    background: ${token.colorBgTextHover};
-  `,
-  container: css`
-    overflow-y: auto;
-    padding: 0;
-  `,
-  header: css`
-    display: flex;
-    align-items: center;
-    margin-block-end: ${token.marginMD}px;
-  `,
-  migrationItem: css`
-    display: flex;
-    align-items: center;
-    justify-content: space-between;
-
-    padding-block: ${token.paddingSM}px;
-    padding-inline: 0;
-    border-block-end: none;
-  `,
-  spotlight: css`
-    margin-block-end: ${token.marginLG}px;
-  `,
-
-  templateCard: css`
-    margin-block-end: ${token.marginSM}px;
-  `,
-}));
-
-interface QueryResult {
-  message: string;
-  success: boolean;
-}
-const Repair = memo(() => {
-  const { t } = useTranslation('common');
-
-  const { styles } = useStyles();
-  const [sqlQuery, setSqlQuery] = useState('');
-  const [queryResult, setQueryResult] = useState<QueryResult | null>(null);
-  const [showSQLHash, setShowSQLHash] = useState('');
-
-  const errorMigrations = useGlobalStore(clientDBSelectors.errorMigrations, isEqual);
-
-  const handleExecuteSQL = async () => {
-    try {
-      // 移除 BEGIN 和 COMMIT
-      let processedSQL = sqlQuery.replaceAll(/BEGIN;|COMMIT;/g, '');
-
-      // 分割 SQL 语句（使用 statement-breakpoint 作为分隔符）
-      const migrationQueries = processedSQL.split('--> statement-breakpoint');
-
-      // 手动管理事务
-      await clientDB.execute('BEGIN;');
-
-      try {
-        for (const query of migrationQueries) {
-          if (query.trim()) {
-            await clientDB.execute(query);
-          }
-        }
-        await clientDB.execute('COMMIT;');
-
-        setQueryResult({
-          message: `SQL executed successfully.`,
-          success: true,
-        });
-      } catch (error) {
-        await clientDB.execute('ROLLBACK;');
-
-        throw error;
-      }
-    } catch (error) {
-      setQueryResult({ message: (error as Error).message, success: false });
-    }
-  };
-
-  return (
-    <div>
-      <Card
-        className={styles.card}
-        extra={<Text type="secondary">{t('clientDB.solve.repair.desc')}</Text>}
-        title={t('clientDB.solve.repair.title')}
-        variant={'borderless'}
-      >
-        <List
-          dataSource={errorMigrations}
-          renderItem={(migration) => (
-            <>
-              <List.Item
-                actions={[
-                  <Button
-                    icon={<Icon icon={Play} />}
-                    key="retry"
-                    onClick={() => {
-                      setShowSQLHash(!!showSQLHash ? '' : migration.hash);
-                      setSqlQuery(migration.sql.join('--> statement-breakpoint\n'));
-                    }}
-                  >
-                    {t('clientDB.solve.repair.runSQL')}
-                  </Button>,
-                ]}
-                className={styles.migrationItem}
-                key={migration.hash}
-              >
-                <List.Item.Meta
-                  description={
-                    <Flexbox>
-                      {t('clientDB.solve.diagnosis.createdAt')}
-                      {dayjs(migration.folderMillis).format('YYYY-MM-DD hh:mm:ss')} ·
-                    </Flexbox>
-                  }
-                  title={migration.sql[0]}
-                />
-              </List.Item>
-              {showSQLHash === migration.hash && (
-                <Card
-                  className={styles.card}
-                  extra={<Text type="secondary">{t('clientDB.solve.repair.sql.desc')}</Text>}
-                  title={t('clientDB.solve.repair.sql.title')}
-                >
-                  <Flexbox gap={16}>
-                    <CodeEditor
-                      language={'sql'}
-                      onValueChange={(e) => setSqlQuery(e)}
-                      placeholder={t('clientDB.solve.repair.sql.placeholder')}
-                      style={{ fontFamily: 'monospace', height: 300 }}
-                      value={sqlQuery}
-                    />
-                    <Flexbox horizontal style={{ justifyContent: 'space-between' }}>
-                      <Button onClick={() => setSqlQuery('')}>
-                        {t('clientDB.solve.repair.sql.clear')}
-                      </Button>
-                      <Button disabled={!sqlQuery.trim()} onClick={handleExecuteSQL} type="primary">
-                        {t('clientDB.solve.repair.sql.run')}
-                      </Button>
-                    </Flexbox>
-
-                    {queryResult && (
-                      <Alert
-                        action={
-                          queryResult.success && (
-                            <Button
-                              key="complete"
-                              onClick={() => updateMigrationRecord(migration.hash)}
-                              size={'small'}
-                              type="primary"
-                            >
-                              {t('clientDB.solve.repair.sql.markFinished')}
-                            </Button>
-                          )
-                        }
-                        description={
-                          <pre style={{ marginTop: '8px', whiteSpace: 'pre-wrap' }}>
-                            {queryResult.message}
-                          </pre>
-                        }
-                        icon={
-                          queryResult.success ? (
-                            <AlertCircle size={16} />
-                          ) : (
-                            <CheckCircle size={16} />
-                          )
-                        }
-                        message={t('clientDB.solve.repair.sql.result')}
-                        showIcon
-                        type={queryResult.success ? 'success' : 'error'}
-                      />
-                    )}
-                  </Flexbox>
-                </Card>
-              )}
-            </>
-          )}
-        />
-      </Card>
-    </div>
-  );
-});
-
-export default Repair;

package/src/features/InitClientDB/features/DatabaseRepair/index.tsx

@@ -1,91 +0,0 @@
-'use client';
-
-import { Button, Segmented } from '@lobehub/ui';
-import { createStyles } from 'antd-style';
-import { AlertCircle, ArrowLeft, PenToolIcon as Tool } from 'lucide-react';
-import { memo, useState } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import Backup from './Backup';
-import Diagnosis from './Diagnosis';
-import Repair from './Repair';
-
-const useStyles = createStyles(({ css, token }) => ({
-  backButton: css`
-    margin-inline-end: ${token.marginSM}px;
-    padding-block: 4px;
-    padding-inline: 8px;
-  `,
-}));
-
-enum DatabaseRepairTab {
-  Backup = 'backup',
-  Diagnosis = 'diagnosis',
-  Repair = 'repair',
-}
-
-interface DatabaseRepairProps {
-  setShowRepair: (value: boolean) => void;
-}
-
-const DatabaseRepair = memo<DatabaseRepairProps>(({ setShowRepair }) => {
-  const { t } = useTranslation('common');
-  const { styles } = useStyles();
-  const [activeTab, setActiveTab] = useState<DatabaseRepairTab>(DatabaseRepairTab.Diagnosis);
-
-  return (
-    <Flexbox gap={12}>
-      <Flexbox horizontal justify={'space-between'}>
-        <Button
-          className={styles.backButton}
-          icon={<ArrowLeft size={16} />}
-          onClick={() => setShowRepair(false)}
-          type="text"
-        >
-          {t('back')}
-        </Button>
-        <Segmented
-          onChange={(v) => setActiveTab(v as DatabaseRepairTab)}
-          options={[
-            {
-              label: (
-                <Flexbox align={'center'} horizontal>
-                  <AlertCircle size={16} style={{ marginRight: 8 }} />
-                  {t('clientDB.solve.tabs.diagnosis')}
-                </Flexbox>
-              ),
-              value: 'diagnosis',
-            },
-            {
-              label: (
-                <Flexbox align={'center'} horizontal>
-                  <Tool size={16} style={{ marginRight: 8 }} />
-                  {t('clientDB.solve.tabs.repair')}
-                </Flexbox>
-              ),
-              value: 'repair',
-            },
-            // {
-            //   label: (
-            //     <Flexbox align={'center'} horizontal>
-            //       <Shield size={16} style={{ marginRight: 8 }} />
-            //       {t('clientDB.solve.tabs.backup')}
-            //     </Flexbox>
-            //   ),
-            //   value: 'backup',
-            // },
-          ]}
-          value={activeTab}
-        />
-        <Flexbox width={36} />
-      </Flexbox>
-
-      {activeTab === DatabaseRepairTab.Diagnosis && <Diagnosis />}
-      {activeTab === DatabaseRepairTab.Repair && <Repair />}
-      {activeTab === DatabaseRepairTab.Backup && <Backup />}
-    </Flexbox>
-  );
-});
-
-export default DatabaseRepair;

package/src/features/InitClientDB/index.tsx

@@ -1,37 +0,0 @@
-'use client';
-
-import { Spin } from 'antd';
-import dynamic from 'next/dynamic';
-import { memo } from 'react';
-
-import { useGlobalStore } from '@/store/global';
-import { systemStatusSelectors } from '@/store/global/selectors';
-
-const Modal = dynamic(() => import('./EnableModal'), {
-  loading: () => <Spin fullscreen />,
-  ssr: false,
-});
-
-const InitIndicator = dynamic(() => import('./InitIndicator'), {
-  ssr: false,
-});
-
-interface InitClientDBProps {
-  bottom?: number;
-}
-
-const InitClientDB = memo<InitClientDBProps>(({ bottom }) => {
-  const isPgliteNotEnabled = useGlobalStore(systemStatusSelectors.isPgliteNotEnabled);
-  const isPgliteNotInited = useGlobalStore(systemStatusSelectors.isPgliteNotInited);
-
-  return (
-    <>
-      {/* 当用户没有设置启用 pglite 时，强弹窗引导用户来开启弹窗 */}
-      {isPgliteNotEnabled && <Modal open={isPgliteNotEnabled} />}
-      {/* 当用户已经启用 pglite 但没有初始化时，展示初始化指示器 */}
-      {isPgliteNotInited && <InitIndicator bottom={bottom} show={isPgliteNotInited} />}
-    </>
-  );
-});
-
-export default InitClientDB;

package/src/libs/trpc/client/edge.ts

@@ -1,26 +0,0 @@
-import { createTRPCClient, httpBatchLink } from '@trpc/client';
-import superjson from 'superjson';
-
-import { isDesktop } from '@/const/version';
-import type { EdgeRouter } from '@/server/routers/edge';
-import { fetchWithDesktopRemoteRPC } from '@/utils/electron/desktopRemoteRPCFetch';
-
-export const edgeClient = createTRPCClient<EdgeRouter>({
-  links: [
-    httpBatchLink({
-      fetch: isDesktop
-        ? // eslint-disable-next-line no-undef
-          (input, init) => fetchWithDesktopRemoteRPC(input as string, init as RequestInit)
-        : undefined,
-      headers: async () => {
-        // dynamic import to avoid circular dependency
-        const { createHeaderWithAuth } = await import('@/services/_auth');
-
-        return createHeaderWithAuth();
-      },
-      maxURLLength: 2083,
-      transformer: superjson,
-      url: '/trpc/edge',
-    }),
-  ],
-});

package/src/libs/trpc/edge/context.ts

@@ -1,71 +0,0 @@
-import { ClientSecretPayload } from '@lobechat/types';
-import { User } from 'next-auth';
-import { NextRequest } from 'next/server';
-
-import { LOBE_CHAT_AUTH_HEADER, enableClerk, enableNextAuth } from '@/const/auth';
-import { ClerkAuth, IClerkAuth } from '@/libs/clerk-auth';
-
-export interface AuthContext {
-  authorizationHeader?: string | null;
-  clerkAuth?: IClerkAuth;
-  jwtPayload?: ClientSecretPayload | null;
-  nextAuth?: User;
-  userId?: string | null;
-}
-
-/**
- * Inner function for `createContext` where we create the context.
- * This is useful for testing when we don't want to mock Next.js' request/response
- */
-export const createContextInner = async (params?: {
-  authorizationHeader?: string | null;
-  clerkAuth?: IClerkAuth;
-  nextAuth?: User;
-  userId?: string | null;
-}): Promise<AuthContext> => ({
-  authorizationHeader: params?.authorizationHeader,
-  clerkAuth: params?.clerkAuth,
-  nextAuth: params?.nextAuth,
-  userId: params?.userId,
-});
-
-export type EdgeContext = Awaited<ReturnType<typeof createContextInner>>;
-
-/**
- * Creates context for an incoming request
- * @link https://trpc.io/docs/v11/context
- */
-export const createEdgeContext = async (request: NextRequest): Promise<EdgeContext> => {
-  // for API-response caching see https://trpc.io/docs/v11/caching
-
-  const authorization = request.headers.get(LOBE_CHAT_AUTH_HEADER);
-
-  let userId;
-  let auth;
-
-  if (enableClerk) {
-    const clerkAuth = new ClerkAuth();
-    const result = clerkAuth.getAuthFromRequest(request);
-    auth = result.clerkAuth;
-    userId = result.userId;
-
-    return createContextInner({ authorizationHeader: authorization, clerkAuth: auth, userId });
-  }
-
-  if (enableNextAuth) {
-    try {
-      const { default: NextAuth } = await import('@/libs/next-auth');
-
-      const session = await NextAuth.auth();
-      if (session && session?.user?.id) {
-        auth = session.user;
-        userId = session.user.id;
-      }
-      return createContextInner({ authorizationHeader: authorization, nextAuth: auth, userId });
-    } catch (e) {
-      console.error('next auth err', e);
-    }
-  }
-
-  return createContextInner({ authorizationHeader: authorization, userId });
-};

package/src/libs/trpc/edge/index.ts

@@ -1,45 +0,0 @@
-/**
- * This is your entry point to setup the root configuration for tRPC on the server.
- * - `initTRPC` should only be used once per app.
- * - We export only the functionality that we use so we can enforce which base procedures should be used
- *
- * Learn how to create protected base procedures and other things below:
- * @link https://trpc.io/docs/v11/router
- * @link https://trpc.io/docs/v11/procedures
- */
-import { DESKTOP_USER_ID } from '@/const/desktop';
-import { isDesktop } from '@/const/version';
-
-import { edgeTrpc } from './init';
-import { jwtPayloadChecker } from './middleware/jwtPayload';
-
-/**
- * Create a router
- * @link https://trpc.io/docs/v11/router
- */
-export const router = edgeTrpc.router;
-
-/**
- * Create an unprotected procedure
- * @link https://trpc.io/docs/v11/procedures
- **/
-export const publicProcedure = edgeTrpc.procedure.use(({ next, ctx }) => {
-  return next({
-    ctx: { userId: isDesktop ? DESKTOP_USER_ID : ctx.userId },
-  });
-});
-
-// procedure that asserts that the user add the password
-export const passwordProcedure = edgeTrpc.procedure.use(jwtPayloadChecker);
-
-/**
- * Merge multiple routers together
- * @link https://trpc.io/docs/v11/merging-routers
- */
-export const mergeRouters = edgeTrpc.mergeRouters;
-
-/**
- * Create a server-side caller
- * @link https://trpc.io/docs/v11/server/server-side-calls
- */
-export const createCallerFactory = edgeTrpc.createCallerFactory;

package/src/libs/trpc/edge/init.ts

@@ -1,26 +0,0 @@
-/**
- * This is your entry point to setup the root configuration for tRPC on the server.
- * - `initTRPC` should only be used once per app.
- * - We export only the functionality that we use so we can enforce which base procedures should be used
- *
- * Learn how to create protected base procedures and other things below:
- * @link https://trpc.io/docs/v11/router
- * @link https://trpc.io/docs/v11/procedures
- */
-import { initTRPC } from '@trpc/server';
-import superjson from 'superjson';
-
-import type { EdgeContext } from './context';
-
-export const edgeTrpc = initTRPC.context<EdgeContext>().create({
-  /**
-   * @link https://trpc.io/docs/v11/error-formatting
-   */
-  errorFormatter({ shape }) {
-    return shape;
-  },
-  /**
-   * @link https://trpc.io/docs/v11/data-transformers
-   */
-  transformer: superjson,
-});

package/src/libs/trpc/edge/middleware/jwtPayload.test.ts

@@ -1,75 +0,0 @@
-// @vitest-environment node
-import * as utils from '@lobechat/utils/server';
-import { TRPCError } from '@trpc/server';
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { createCallerFactory } from '@/libs/trpc/edge';
-import { AuthContext, createContextInner } from '@/libs/trpc/edge/context';
-import { edgeTrpc as trpc } from '@/libs/trpc/edge/init';
-
-import { jwtPayloadChecker } from './jwtPayload';
-
-const appRouter = trpc.router({
-  protectedQuery: trpc.procedure.use(jwtPayloadChecker).query(async ({ ctx }) => {
-    return ctx.jwtPayload;
-  }),
-});
-
-const createCaller = createCallerFactory(appRouter);
-let ctx: AuthContext;
-let router: ReturnType<typeof createCaller>;
-
-vi.mock('@/libs/next-auth/edge', () => {
-  return {
-    auth: vi.fn().mockResolvedValue(undefined),
-  };
-});
-
-beforeEach(() => {
-  vi.resetAllMocks();
-});
-
-describe('passwordChecker middleware', () => {
-  it('should throw UNAUTHORIZED error if authorizationHeader is not present in context', async () => {
-    ctx = await createContextInner();
-    router = createCaller(ctx);
-
-    await expect(router.protectedQuery()).rejects.toThrow(new TRPCError({ code: 'UNAUTHORIZED' }));
-  });
-
-  it('should call next with jwtPayload in context if access code is correct', async () => {
-    const jwtPayload = { accessCode: '123' };
-
-    vi.spyOn(utils, 'getXorPayload').mockResolvedValue(jwtPayload);
-
-    ctx = await createContextInner({ authorizationHeader: 'Bearer token' });
-    router = createCaller(ctx);
-
-    const data = await router.protectedQuery();
-
-    expect(data).toEqual(jwtPayload);
-  });
-
-  it('should call next with jwtPayload in context if no access codes are set', async () => {
-    const jwtPayload = {};
-    vi.spyOn(utils, 'getXorPayload').mockResolvedValue(jwtPayload);
-
-    ctx = await createContextInner({ authorizationHeader: 'Bearer token' });
-    router = createCaller(ctx);
-
-    const data = await router.protectedQuery();
-
-    expect(data).toEqual(jwtPayload);
-  });
-  it('should call next with jwtPayload in context if  access codes is undefined', async () => {
-    const jwtPayload = {};
-    vi.spyOn(utils, 'getXorPayload').mockResolvedValue(jwtPayload);
-
-    ctx = await createContextInner({ authorizationHeader: 'Bearer token' });
-    router = createCaller(ctx);
-
-    const data = await router.protectedQuery();
-
-    expect(data).toEqual(jwtPayload);
-  });
-});

package/src/libs/trpc/edge/middleware/jwtPayload.ts

@@ -1,14 +0,0 @@
-import { getXorPayload } from '@lobechat/utils/server';
-import { TRPCError } from '@trpc/server';
-
-import { edgeTrpc } from '../init';
-
-export const jwtPayloadChecker = edgeTrpc.middleware(async (opts) => {
-  const { ctx } = opts;
-
-  if (!ctx.authorizationHeader) throw new TRPCError({ code: 'UNAUTHORIZED' });
-
-  const jwtPayload = getXorPayload(ctx.authorizationHeader);
-
-  return opts.next({ ctx: { jwtPayload } });
-});

package/src/migrations/FromV0ToV1.ts

@@ -1,10 +0,0 @@
-import type { Migration, MigrationData } from './VersionController';
-
-export class MigrationV0ToV1 implements Migration {
-  // from this version to start migration
-  version = 0;
-
-  migrate(data: MigrationData): MigrationData {
-    return data;
-  }
-}