@lobehub/chat 1.76.1 → 1.77.1

package/src/components/IndexCard/index.tsx

@@ -0,0 +1,143 @@
+import { ActionIcon } from '@lobehub/ui';
+import { createStyles } from 'antd-style';
+import { ChevronDown, ChevronRight } from 'lucide-react';
+import { ReactNode, memo } from 'react';
+import { Center, Flexbox, FlexboxProps } from 'react-layout-kit';
+
+const useStyles = createStyles(({ css, token, responsive }) => ({
+  card: css`
+    position: relative;
+
+    overflow: hidden;
+
+    border: 1px solid ${token.colorBorderSecondary};
+    border-radius: ${token.borderRadiusLG}px;
+
+    background: ${token.colorBgContainer};
+  `,
+  desc: css`
+    font-size: 14px;
+    line-height: 1.4;
+    color: ${token.colorTextDescription};
+    ${responsive.mobile} {
+      font-size: 12px;
+    }
+  `,
+  expend: css`
+    position: absolute;
+    inset-block-end: -12px;
+    inset-inline-start: 50%;
+    transform: translateX(-50%);
+
+    border: 1px solid ${token.colorBorderSecondary};
+    border-radius: 50%;
+
+    background: ${token.colorBgContainer};
+  `,
+  header: css`
+    border-block-end: 1px solid ${token.colorBorderSecondary};
+    background: ${token.colorFillQuaternary};
+  `,
+  more: css`
+    border: 1px solid ${token.colorBorderSecondary};
+  `,
+  title: css`
+    font-size: 16px;
+    font-weight: bold;
+    line-height: 1.4;
+    ${responsive.mobile} {
+      font-size: 14px;
+    }
+  `,
+}));
+
+interface IndexCardProps extends Omit<FlexboxProps, 'title'> {
+  desc?: ReactNode;
+  expand?: boolean;
+  extra?: ReactNode;
+  icon?: ReactNode;
+  moreTooltip?: string;
+  onExpand?: () => void;
+  onMoreClick?: () => void;
+  title?: ReactNode;
+}
+
+const IndexCard = memo<IndexCardProps>(
+  ({
+    expand = true,
+    onExpand,
+    icon,
+    className,
+    onMoreClick,
+    title,
+    extra,
+    moreTooltip,
+    desc,
+    children,
+    ...rest
+  }) => {
+    const { styles } = useStyles();
+    return (
+      <Flexbox
+        style={{
+          marginBottom: !expand ? 12 : undefined,
+          position: 'relative',
+        }}
+      >
+        <Flexbox
+          className={styles.card}
+          style={{
+            paddingBottom: !expand ? 12 : undefined,
+          }}
+        >
+          {title && (
+            <Flexbox
+              align={'center'}
+              className={styles.header}
+              gap={16}
+              horizontal
+              justify={'space-between'}
+              padding={16}
+            >
+              <Flexbox align={'center'} gap={12} horizontal>
+                {icon}
+                <Flexbox>
+                  <div className={styles.title}>{title}</div>
+                  {desc && <div className={styles.desc}>{desc}</div>}
+                </Flexbox>
+              </Flexbox>
+              <Flexbox align={'center'} gap={8} horizontal>
+                {extra}
+                {onMoreClick && (
+                  <ActionIcon
+                    className={styles.more}
+                    icon={ChevronRight}
+                    onClick={onMoreClick}
+                    size={{ blockSize: 32, borderRadius: '50%', fontSize: 16 }}
+                    title={moreTooltip}
+                  />
+                )}
+              </Flexbox>
+            </Flexbox>
+          )}
+          <Flexbox className={className} gap={16} padding={16} width={'100%'} {...rest}>
+            {children}
+          </Flexbox>
+        </Flexbox>
+        {!expand && (
+          <Center className={styles.expend} height={24} width={24}>
+            <ActionIcon
+              icon={ChevronDown}
+              onClick={onExpand}
+              size={{ blockSize: 24, borderRadius: '50%', fontSize: 16 }}
+            />
+          </Center>
+        )}
+      </Flexbox>
+    );
+  },
+);
+
+IndexCard.displayName = 'IndexCard';
+
+export default IndexCard;

package/src/components/ProgressItem/index.tsx

@@ -0,0 +1,75 @@
+import { Progress } from 'antd';
+import { createStyles, useResponsive } from 'antd-style';
+import { CSSProperties, memo } from 'react';
+import { Flexbox } from 'react-layout-kit';
+
+const useStyles = createStyles(({ css, token }) => ({
+  desc: css`
+    height: 20px;
+    font-size: 12px;
+    line-height: 20px;
+    color: ${token.colorTextTertiary};
+  `,
+  title: css`
+    font-size: 15px;
+    font-weight: bold;
+    color: ${token.colorTextSecondary};
+  `,
+}));
+
+interface ProgressItemProps {
+  className?: string;
+  desc?: string;
+  legend?: string;
+  padding?: number;
+  percent: number;
+  style?: CSSProperties;
+  title: string;
+  usage: {
+    total: string | number;
+    used: string | number;
+  };
+}
+
+const ProgressItem = memo<ProgressItemProps>(
+  ({ legend, title, desc, usage, percent, style, className }) => {
+    const { mobile } = useResponsive();
+    const { styles, theme } = useStyles();
+
+    return (
+      <Flexbox className={className} paddingInline={16} style={style} width={'100%'}>
+        <Flexbox align={'center'} horizontal justify={'space-between'} width={'100%'}>
+          <Flexbox align={'center'} gap={8} horizontal>
+            {legend && (
+              <Flexbox
+                height={8}
+                style={{
+                  background: theme.geekblue,
+                  borderRadius: '50%',
+                  flex: 'none',
+                }}
+                width={8}
+              />
+            )}
+            <Flexbox align={'baseline'} gap={mobile ? 0 : 8} horizontal={!mobile}>
+              <div className={styles.title}>{title}</div>
+              {desc && <div className={styles.desc}>{desc}</div>}
+            </Flexbox>
+          </Flexbox>
+          <div>
+            <span style={{ fontWeight: 'bold' }}>{usage.used}</span>
+            {['', '/', usage.total].join(' ')}
+          </div>
+        </Flexbox>
+        <Progress
+          percent={percent}
+          showInfo={false}
+          size={'small'}
+          strokeColor={theme.colorPrimary}
+        />
+      </Flexbox>
+    );
+  },
+);
+
+export default ProgressItem;

package/src/database/models/__tests__/session.test.ts

@@ -489,6 +489,7 @@ describe('SessionModel', () => {
     it('should delete a session and its associated topics and messages', async () => {
       // Create a session
       const sessionId = '1';
+      await serverDB.insert(users).values([{ id: '456' }]);
       await serverDB.insert(sessions).values({ id: sessionId, userId });
 
       // Create some topics and messages associated with the session
@@ -500,6 +501,11 @@ describe('SessionModel', () => {
         { id: '1', sessionId, userId, role: 'user' },
         { id: '2', sessionId, userId, role: 'assistant' },
       ]);
+      await serverDB.insert(agents).values([
+        { id: 'a1', userId },
+        { id: 'a2', userId: '456' },
+      ]);
+      await serverDB.insert(agentsToSessions).values([{ agentId: 'a1', userId, sessionId: '1' }]);
 
       // Delete the session
       await sessionModel.delete(sessionId);
@@ -514,6 +520,7 @@ describe('SessionModel', () => {
       expect(
         await serverDB.select().from(messages).where(eq(messages.sessionId, sessionId)),
       ).toHaveLength(0);
+      expect(await serverDB.select().from(agents).where(eq(agents.userId, userId))).toHaveLength(0);
     });
 
     it('should not delete sessions belonging to other users', async () => {
@@ -576,6 +583,8 @@ describe('SessionModel', () => {
       // Create some sessions
       const sessionIds = ['1', '2', '3'];
       await serverDB.insert(sessions).values(sessionIds.map((id) => ({ id, userId })));
+      await serverDB.insert(agents).values([{ id: '1', userId }]);
+      await serverDB.insert(agentsToSessions).values([{ sessionId: '1', agentId: '1', userId }]);
 
       // Create some topics and messages associated with the sessions
       await serverDB.insert(topics).values([
@@ -602,6 +611,7 @@ describe('SessionModel', () => {
       expect(
         await serverDB.select().from(messages).where(inArray(messages.sessionId, sessionIds)),
       ).toHaveLength(0);
+      expect(await serverDB.select().from(agents)).toHaveLength(0);
     });
 
     it('should not delete sessions belonging to other users', async () => {
@@ -710,6 +720,14 @@ describe('SessionModel', () => {
           { id: 'm1', sessionId: '1', userId, role: 'user' },
           { id: 'm2', sessionId: '2', userId, role: 'assistant' },
         ]);
+        await trx.insert(agents).values([
+          { id: 'a1', userId },
+          { id: 'a2', userId },
+        ]);
+        await trx.insert(agentsToSessions).values([
+          { agentId: 'a1', sessionId: '1', userId },
+          { agentId: 'a2', sessionId: '2', userId },
+        ]);
       });
 
       await sessionModel.deleteAll();
@@ -723,6 +741,9 @@ describe('SessionModel', () => {
         .from(messages)
         .where(eq(messages.userId, userId));
       expect(remainingMessages).toHaveLength(0);
+
+      const agentsTopics = await serverDB.select().from(agents).where(eq(agents.userId, userId));
+      expect(agentsTopics).toHaveLength(0);
     });
   });
 

package/src/database/repositories/dataExporter/index.test.ts

@@ -0,0 +1,330 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { clientDB, initializeDB } from '@/database/client/db';
+import {
+  agents,
+  agentsKnowledgeBases,
+  agentsToSessions,
+  files,
+  filesToSessions,
+  globalFiles,
+  knowledgeBaseFiles,
+  knowledgeBases,
+  messages,
+  sessionGroups,
+  sessions,
+  topics,
+  userSettings,
+  users,
+} from '@/database/schemas';
+import { LobeChatDatabase } from '@/database/type';
+
+import { DATA_EXPORT_CONFIG, DataExporterRepos } from './index';
+
+let db = clientDB as LobeChatDatabase;
+
+// 设置测试数据
+describe('DataExporterRepos', () => {
+  // 测试数据 ID
+  const testIds = {
+    userId: 'test-user-id',
+    fileId: 'test-file-id',
+    fileHash: 'test-file-hash',
+    sessionId: 'test-session-id',
+    agentId: 'test-agent-id',
+    topicId: 'test-topic-id',
+    messageId: 'test-message-id',
+    knowledgeBaseId: 'test-kb-id',
+  };
+
+  // 设置测试环境
+  let userId: string = testIds.userId;
+
+  const setupTestData = async () => {
+    await db.transaction(async (trx) => {
+      // 用户数据
+      await trx.insert(users).values({
+        id: testIds.userId,
+        username: 'testuser',
+        email: 'test@example.com',
+      });
+
+      // 用户设置
+      await trx.insert(userSettings).values({
+        id: testIds.userId,
+        general: { theme: 'light' },
+      });
+
+      // 全局文件
+      await trx.insert(globalFiles).values({
+        hashId: testIds.fileHash,
+        fileType: 'text/plain',
+        size: 1024,
+        url: 'https://example.com/test-file.txt',
+        creator: testIds.userId,
+      });
+
+      // 文件数据
+      await trx.insert(files).values({
+        id: testIds.fileId,
+        userId: testIds.userId,
+        fileType: 'text/plain',
+        fileHash: testIds.fileHash,
+        name: 'test-file.txt',
+        size: 1024,
+        url: 'https://example.com/test-file.txt',
+      });
+
+      // 会话组
+      await trx.insert(sessionGroups).values({
+        name: 'Test Group',
+        userId: testIds.userId,
+      });
+
+      // 会话
+      await trx.insert(sessions).values({
+        id: testIds.sessionId,
+        slug: 'test-session',
+        title: 'Test Session',
+        userId: testIds.userId,
+      });
+
+      // 主题
+      await trx.insert(topics).values({
+        id: testIds.topicId,
+        title: 'Test Topic',
+        sessionId: testIds.sessionId,
+        userId: testIds.userId,
+      });
+
+      // 消息
+      await trx.insert(messages).values({
+        id: testIds.messageId,
+        role: 'user',
+        content: 'Hello, world!',
+        userId: testIds.userId,
+        sessionId: testIds.sessionId,
+        topicId: testIds.topicId,
+      });
+
+      // 代理
+      await trx.insert(agents).values({
+        id: testIds.agentId,
+        title: 'Test Agent',
+        userId: testIds.userId,
+      });
+
+      // 代理到会话的关联
+      await trx.insert(agentsToSessions).values({
+        agentId: testIds.agentId,
+        sessionId: testIds.sessionId,
+        userId: testIds.userId,
+      });
+
+      // 文件到会话的关联
+      await trx.insert(filesToSessions).values({
+        fileId: testIds.fileId,
+        sessionId: testIds.sessionId,
+        userId: testIds.userId,
+      });
+
+      // 知识库
+      await trx.insert(knowledgeBases).values({
+        id: testIds.knowledgeBaseId,
+        name: 'Test Knowledge Base',
+        userId: testIds.userId,
+      });
+
+      // 知识库文件
+      await trx.insert(knowledgeBaseFiles).values({
+        knowledgeBaseId: testIds.knowledgeBaseId,
+        fileId: testIds.fileId,
+        userId: testIds.userId,
+      });
+
+      // 代理知识库
+      await trx.insert(agentsKnowledgeBases).values({
+        agentId: testIds.agentId,
+        knowledgeBaseId: testIds.knowledgeBaseId,
+        userId: testIds.userId,
+      });
+    });
+  };
+
+  beforeEach(async () => {
+    // 创建内存数据库
+    await initializeDB();
+
+    // 插入测试数据
+    await setupTestData();
+  });
+
+  afterEach(async () => {
+    await db.delete(users);
+    await db.delete(globalFiles);
+
+    vi.restoreAllMocks();
+  });
+
+  describe('export', () => {
+    it('should export all user data correctly', async () => {
+      // 创建导出器实例
+      const dataExporter = new DataExporterRepos(db, userId);
+
+      // 执行导出
+      const result = await dataExporter.export();
+
+      // 验证基础表导出结果
+      // expect(result).toHaveProperty('users');
+      // expect(result.users).toHaveLength(1);
+      // expect(result.users[0]).toHaveProperty('id', testIds.userId);
+      // expect(result.users[0]).not.toHaveProperty('userId'); // userId 字段应该被移除
+
+      expect(result).toHaveProperty('userSettings');
+      expect(result.userSettings).toHaveLength(1);
+      expect(result.userSettings[0]).toHaveProperty('id', testIds.userId);
+
+      // expect(result).toHaveProperty('files');
+      // expect(result.files).toHaveLength(1);
+      // expect(result.files[0]).toHaveProperty('id', testIds.fileId);
+      // expect(result.files[0]).toHaveProperty('fileHash', testIds.fileHash);
+      // expect(result.files[0]).not.toHaveProperty('userId');
+
+      expect(result).toHaveProperty('sessions');
+      expect(result.sessions).toHaveLength(1);
+      expect(result.sessions[0]).toHaveProperty('id', testIds.sessionId);
+
+      expect(result).toHaveProperty('topics');
+      expect(result.topics).toHaveLength(1);
+      expect(result.topics[0]).toHaveProperty('id', testIds.topicId);
+
+      expect(result).toHaveProperty('messages');
+      expect(result.messages).toHaveLength(1);
+      expect(result.messages[0]).toHaveProperty('id', testIds.messageId);
+
+      expect(result).toHaveProperty('agents');
+      expect(result.agents).toHaveLength(1);
+      expect(result.agents[0]).toHaveProperty('id', testIds.agentId);
+
+      // expect(result).toHaveProperty('knowledgeBases');
+      // expect(result.knowledgeBases).toHaveLength(1);
+      // expect(result.knowledgeBases[0]).toHaveProperty('id', testIds.knowledgeBaseId);
+
+      // 验证关联表导出结果
+      // expect(result).toHaveProperty('globalFiles');
+      // expect(result.globalFiles).toHaveLength(1);
+      // expect(result.globalFiles[0]).toHaveProperty('hashId', testIds.fileHash);
+
+      expect(result).toHaveProperty('agentsToSessions');
+      expect(result.agentsToSessions).toHaveLength(1);
+      expect(result.agentsToSessions[0]).toHaveProperty('agentId', testIds.agentId);
+      expect(result.agentsToSessions[0]).toHaveProperty('sessionId', testIds.sessionId);
+
+      // expect(result).toHaveProperty('filesToSessions');
+      // expect(result.filesToSessions).toHaveLength(1);
+      // expect(result.filesToSessions[0]).toHaveProperty('fileId', testIds.fileId);
+      // expect(result.filesToSessions[0]).toHaveProperty('sessionId', testIds.sessionId);
+
+      // expect(result).toHaveProperty('knowledgeBaseFiles');
+      // expect(result.knowledgeBaseFiles).toHaveLength(1);
+      // expect(result.knowledgeBaseFiles[0]).toHaveProperty(
+      //   'knowledgeBaseId',
+      //   testIds.knowledgeBaseId,
+      // );
+      // expect(result.knowledgeBaseFiles[0]).toHaveProperty('fileId', testIds.fileId);
+    });
+
+    it('should handle empty database gracefully', async () => {
+      // 清空数据库
+
+      await db.delete(users);
+      await db.delete(globalFiles);
+
+      // 创建导出器实例
+      const dataExporter = new DataExporterRepos(db, userId);
+
+      // 执行导出
+      const result = await dataExporter.export();
+
+      // 验证所有表都返回空数组
+      DATA_EXPORT_CONFIG.baseTables.forEach(({ table }) => {
+        expect(result).toHaveProperty(table);
+        expect(result[table]).toEqual([]);
+      });
+
+      DATA_EXPORT_CONFIG.relationTables.forEach(({ table }) => {
+        expect(result).toHaveProperty(table);
+        expect(result[table]).toEqual([]);
+      });
+    });
+
+    it('should handle database query errors', async () => {
+      // 模拟查询错误
+      // @ts-ignore
+      vi.spyOn(db.query.users, 'findMany').mockRejectedValueOnce(new Error('Database error'));
+
+      // 创建导出器实例
+      const dataExporter = new DataExporterRepos(db, userId);
+
+      // 执行导出
+      const result = await dataExporter.export();
+
+      // 验证其他表仍然被导出
+      expect(result).toHaveProperty('sessions');
+      expect(result.sessions).toHaveLength(1);
+    });
+
+    it.skip('should skip relation tables when source tables have no data', async () => {
+      // 删除文件数据，这将导致 globalFiles 表被跳过
+      await db.delete(files);
+
+      // 创建导出器实例
+      const dataExporter = new DataExporterRepos(db, userId);
+
+      // 执行导出
+      const result = await dataExporter.export();
+
+      // 验证文件表为空
+      // expect(result).toHaveProperty('files');
+      // expect(result.files).toEqual([]);
+
+      // 验证关联表也为空
+      // expect(result).toHaveProperty('globalFiles');
+      // expect(result.globalFiles).toEqual([]);
+    });
+
+    it('should export data for a different user', async () => {
+      // 创建另一个用户
+      const anotherUserId = 'another-user-id';
+      await db.transaction(async (trx) => {
+        await trx.insert(users).values({
+          id: anotherUserId,
+          username: 'anotheruser',
+          email: 'another@example.com',
+        });
+        await trx.insert(sessions).values({
+          id: 'another-session-id',
+          slug: 'another-session',
+          title: 'Another Session',
+          userId: anotherUserId,
+        });
+      });
+
+      // 创建导出器实例，使用另一个用户 ID
+      const dataExporter = new DataExporterRepos(db, anotherUserId);
+
+      // 执行导出
+      const result = await dataExporter.export();
+
+      // 验证只导出了另一个用户的数据
+      // expect(result).toHaveProperty('users');
+      // expect(result.users).toHaveLength(1);
+      // expect(result.users[0]).toHaveProperty('id', anotherUserId);
+
+      expect(result).toHaveProperty('sessions');
+      expect(result.sessions).toHaveLength(1);
+      expect(result.sessions[0]).not.toHaveProperty('userId', anotherUserId);
+      expect(result.sessions[0]).toHaveProperty('id', 'another-session-id');
+    });
+  });
+});