@lobehub/chat 1.71.4 → 1.72.0

package/src/hooks/useImgToClipboard.ts

@@ -0,0 +1,29 @@
+import { App } from 'antd';
+import { t } from 'i18next';
+import { useState } from 'react';
+
+import { ImageType, getImageUrl } from './useScreenshot';
+
+export const useImgToClipboard = ({ id = '#preview', width }: { id?: string; width?: number }) => {
+  const [loading, setLoading] = useState(false);
+  const { message } = App.useApp();
+
+  const handleCopy = async () => {
+    setLoading(true);
+    try {
+      const dataUrl = await getImageUrl({ id, imageType: ImageType.PNG, width });
+      const blob = await fetch(dataUrl).then((res) => res.blob());
+      navigator.clipboard.write([new ClipboardItem({ [blob.type]: blob })]);
+      setLoading(false);
+      message.success(t('copySuccess', { defaultValue: 'Copy Success', ns: 'common' }));
+    } catch (error) {
+      console.error('Failed to copy image', error);
+      setLoading(false);
+    }
+  };
+
+  return {
+    loading,
+    onCopy: handleCopy,
+  };
+};

package/src/hooks/useScreenshot.ts

@@ -31,6 +31,58 @@ export const imageTypeOptions: SegmentedProps['options'] = [
   },
 ];
 
+export const getImageUrl = async ({
+  imageType,
+  id = '#preview',
+  width,
+}: {
+  id?: string;
+  imageType: ImageType;
+  width?: number;
+}) => {
+  let screenshotFn: any;
+  switch (imageType) {
+    case ImageType.JPG: {
+      screenshotFn = domToJpeg;
+      break;
+    }
+    case ImageType.PNG: {
+      screenshotFn = domToPng;
+      break;
+    }
+    case ImageType.SVG: {
+      screenshotFn = domToSvg;
+      break;
+    }
+    case ImageType.WEBP: {
+      screenshotFn = domToWebp;
+      break;
+    }
+  }
+
+  const dom: HTMLDivElement = document.querySelector(id) as HTMLDivElement;
+  let copy: HTMLDivElement = dom;
+
+  if (width) {
+    copy = dom.cloneNode(true) as HTMLDivElement;
+    copy.style.width = `${width}px`;
+    document.body.append(copy);
+  }
+
+  const dataUrl = await screenshotFn(width ? copy : dom, {
+    features: {
+      // 不启用移除控制符，否则会导致 safari emoji 报错
+      removeControlCharacter: false,
+    },
+    scale: 2,
+    width,
+  });
+
+  if (width && copy) copy?.remove();
+
+  return dataUrl;
+};
+
 export const useScreenshot = ({
   imageType,
   title = 'share',
@@ -47,46 +99,7 @@ export const useScreenshot = ({
   const handleDownload = useCallback(async () => {
     setLoading(true);
     try {
-      let screenshotFn: any;
-      switch (imageType) {
-        case ImageType.JPG: {
-          screenshotFn = domToJpeg;
-          break;
-        }
-        case ImageType.PNG: {
-          screenshotFn = domToPng;
-          break;
-        }
-        case ImageType.SVG: {
-          screenshotFn = domToSvg;
-          break;
-        }
-        case ImageType.WEBP: {
-          screenshotFn = domToWebp;
-          break;
-        }
-      }
-
-      const dom: HTMLDivElement = document.querySelector(id) as HTMLDivElement;
-      let copy: HTMLDivElement = dom;
-
-      if (width) {
-        copy = dom.cloneNode(true) as HTMLDivElement;
-        copy.style.width = `${width}px`;
-        document.body.append(copy);
-      }
-
-      const dataUrl = await screenshotFn(width ? copy : dom, {
-        features: {
-          // 不启用移除控制符，否则会导致 safari emoji 报错
-          removeControlCharacter: false,
-        },
-        scale: 2,
-        width,
-      });
-
-      if (width && copy) copy?.remove();
-
+      const dataUrl = await getImageUrl({ id, imageType, width });
       const link = document.createElement('a');
       link.download = `${BRANDING_NAME}_${title}_${dayjs().format('YYYY-MM-DD')}.${imageType}`;
       link.href = dataUrl;

package/src/services/file/client.test.ts

@@ -92,7 +92,7 @@ describe('FileService', () => {
         hashId: '123tttt',
       };
 
-      await clientDB.insert(globalFiles).values(file);
+      await clientDB.insert(globalFiles).values({ ...file, creator: userId });
 
       await clientDB.insert(files).values({
         id: fileId,
@@ -174,6 +174,7 @@ describe('FileService', () => {
       await clientDB.insert(globalFiles).values({
         ...mockFile,
         hashId: hash,
+        creator: userId,
       });
       await clientDB.insert(files).values({
         id: '1',

package/src/services/message/client.test.ts

@@ -284,7 +284,7 @@ describe('MessageClientService', () => {
     it('should update the plugin state of a message', async () => {
       // Setup
       await clientDB.insert(messages).values({ id: mockMessageId, role: 'user', userId });
-      await clientDB.insert(messagePlugins).values({ id: mockMessageId });
+      await clientDB.insert(messagePlugins).values({ id: mockMessageId, userId });
       const key = 'stateKey';
       const value = 'stateValue';
       const newPluginState = { [key]: value };
@@ -304,7 +304,7 @@ describe('MessageClientService', () => {
     it('should update the plugin arguments object of a message', async () => {
       // Setup
       await clientDB.insert(messages).values({ id: mockMessageId, role: 'user', userId });
-      await clientDB.insert(messagePlugins).values({ id: mockMessageId });
+      await clientDB.insert(messagePlugins).values({ id: mockMessageId, userId });
       const value = 'stateValue';
 
       // Execute
@@ -319,7 +319,7 @@ describe('MessageClientService', () => {
     it('should update the plugin arguments string of a message', async () => {
       // Setup
       await clientDB.insert(messages).values({ id: mockMessageId, role: 'user', userId });
-      await clientDB.insert(messagePlugins).values({ id: mockMessageId });
+      await clientDB.insert(messagePlugins).values({ id: mockMessageId, userId });
       const value = 'stateValue';
       // Execute
       await messageService.updateMessagePluginArguments(

package/src/services/session/client.test.ts

@@ -32,7 +32,9 @@ beforeEach(async () => {
     await trx.insert(users).values([{ id: userId }, { id: '456' }]);
     await trx.insert(sessions).values([{ id: mockSessionId, userId }]);
     await trx.insert(agents).values([{ id: mockAgentId, userId }]);
-    await trx.insert(agentsToSessions).values([{ agentId: mockAgentId, sessionId: mockSessionId }]);
+    await trx
+      .insert(agentsToSessions)
+      .values([{ agentId: mockAgentId, sessionId: mockSessionId, userId }]);
     await trx.insert(sessionGroups).values([
       { id: 'group-1', name: 'group-A', sort: 2, userId },
       { id: 'group-2', name: 'group-B', sort: 1, userId },
@@ -176,7 +178,7 @@ describe('SessionService', () => {
       await clientDB.insert(agents).values({ userId, id: 'agent-1', title: 'Session Name' });
       await clientDB
         .insert(agentsToSessions)
-        .values({ agentId: 'agent-1', sessionId: mockSessionId });
+        .values({ agentId: 'agent-1', sessionId: mockSessionId, userId });
 
       // Execute
       const keyword = 'Name';
@@ -201,7 +203,7 @@ describe('SessionService', () => {
       await clientDB.insert(agents).values({ userId, id: 'agent-1' });
       await clientDB
         .insert(agentsToSessions)
-        .values({ agentId: 'agent-1', sessionId: 'duplicated-session-id' });
+        .values({ agentId: 'agent-1', sessionId: 'duplicated-session-id', userId });
 
       // Execute
       const duplicatedSessionId = await sessionService.cloneSession(mockSessionId, newTitle);

package/src/types/message/base.ts

@@ -117,3 +117,10 @@ export interface UpdateMessageParams {
   toolCalls?: MessageToolCall[];
   tools?: ChatToolPayload[] | null;
 }
+
+export interface NewMessageQueryParams {
+  embeddingsId: string;
+  messageId: string;
+  rewriteQuery: string;
+  userQuery: string;
+}

package/vitest.server.config.ts

@@ -15,7 +15,7 @@ export default defineConfig({
       reportsDirectory: './coverage/server',
     },
     environment: 'node',
-    include: ['src/database/server/**/**/*.test.ts'],
+    include: ['src/database/models/**/**/*.test.ts', 'src/database/server/**/**/*.test.ts'],
     poolOptions: {
       threads: { singleThread: true },
     },

package/src/database/server/models/user.test.ts

@@ -1,58 +0,0 @@
-// @vitest-environment node
-import { TRPCError } from '@trpc/server';
-import { describe, expect, it, vi } from 'vitest';
-
-import { UserModel, UserNotFoundError } from '@/database/server/models/user';
-
-describe('UserNotFoundError', () => {
-  it('should extend TRPCError with correct code and message', () => {
-    const error = new UserNotFoundError();
-
-    expect(error).toBeInstanceOf(TRPCError);
-    expect(error.code).toBe('UNAUTHORIZED');
-    expect(error.message).toBe('user not found');
-  });
-});
-
-describe('UserModel', () => {
-  const mockDb = {
-    query: {
-      users: {
-        findFirst: vi.fn(),
-      },
-    },
-  };
-
-  const mockUserId = 'test-user-id';
-  const userModel = new UserModel(mockDb as any, mockUserId);
-
-  describe('getUserRegistrationDuration', () => {
-    it('should return default values when user not found', async () => {
-      mockDb.query.users.findFirst.mockResolvedValue(null);
-
-      const result = await userModel.getUserRegistrationDuration();
-
-      expect(result).toEqual({
-        createdAt: expect.any(String),
-        duration: 1,
-        updatedAt: expect.any(String),
-      });
-    });
-
-    it('should calculate duration correctly for existing user', async () => {
-      const createdAt = new Date('2024-01-01');
-      mockDb.query.users.findFirst.mockResolvedValue({
-        createdAt,
-      });
-
-      const result = await userModel.getUserRegistrationDuration();
-
-      expect(result).toEqual({
-        createdAt: '2024-01-01',
-        duration: expect.any(Number),
-        updatedAt: expect.any(String),
-      });
-      expect(result.duration).toBeGreaterThan(0);
-    });
-  });
-});