@lobehub/chat 1.36.46 → 1.37.1

package/src/services/message/type.ts

@@ -1,4 +1,4 @@
-import { DB_Message } from '@/database/_deprecated/schemas/message';
+import { MessageItem } from '@/database/schemas';
 import {
   ChatMessage,
   ChatMessageError,
@@ -11,7 +11,7 @@ import {
 
 export interface IMessageService {
   createMessage(data: CreateMessageParams): Promise<string>;
-  batchCreateMessages(messages: ChatMessage[]): Promise<any>;
+  batchCreateMessages(messages: MessageItem[]): Promise<any>;
 
   getMessages(sessionId: string, topicId?: string): Promise<ChatMessage[]>;
   getAllMessages(): Promise<ChatMessage[]>;
@@ -20,11 +20,10 @@ export interface IMessageService {
   countTodayMessages(): Promise<number>;
 
   updateMessageError(id: string, error: ChatMessageError): Promise<any>;
-  updateMessage(id: string, message: Partial<DB_Message>): Promise<any>;
+  updateMessage(id: string, message: Partial<MessageItem>): Promise<any>;
   updateMessageTTS(id: string, tts: Partial<ChatTTS> | false): Promise<any>;
   updateMessageTranslate(id: string, translate: Partial<ChatTranslate> | false): Promise<any>;
   updateMessagePluginState(id: string, value: Record<string, any>): Promise<any>;
-  bindMessagesToTopic(topicId: string, messageIds: string[]): Promise<any>;
 
   removeMessage(id: string): Promise<any>;
   removeMessages(ids: string[]): Promise<any>;

package/src/services/plugin/{client.test.ts → _deprecated.test.ts}

@@ -6,7 +6,7 @@ import { DB_Plugin } from '@/database/_deprecated/schemas/plugin';
 import { LobeTool } from '@/types/tool';
 import { LobeToolCustomPlugin } from '@/types/tool/plugin';
 
-import { ClientService } from './client';
+import { ClientService } from './_deprecated';
 import { InstallPluginParams } from './type';
 
 const pluginService = new ClientService();

package/src/services/plugin/index.ts

@@ -1,5 +1,9 @@
-import { ClientService } from './client';
+import { ClientService as DeprecatedService } from './_deprecated';
+import { ClientService } from './pglite';
 import { ServerService } from './server';
 
+const clientService =
+  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
+
 export const pluginService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : new ClientService();
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;

package/src/services/plugin/pglite.test.ts

@@ -0,0 +1,175 @@
+import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
+import { eq } from 'drizzle-orm';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { clientDB, initializeDB } from '@/database/client/db';
+import { installedPlugins, users } from '@/database/schemas';
+import { LobeTool } from '@/types/tool';
+import { LobeToolCustomPlugin } from '@/types/tool/plugin';
+
+import { ClientService } from './pglite';
+import { InstallPluginParams } from './type';
+
+// Mocking modules and functions
+
+const userId = 'message-db';
+const pluginService = new ClientService(userId);
+
+// Mock data
+beforeEach(async () => {
+  await initializeDB();
+
+  // 在每个测试用例之前，重置表数据
+  await clientDB.transaction(async (trx) => {
+    await trx.delete(users);
+    await trx.insert(users).values([{ id: userId }, { id: '456' }]);
+  });
+});
+
+describe('PluginService', () => {
+  describe('installPlugin', () => {
+    it('should install a plugin', async () => {
+      // Arrange
+      const fakePlugin = {
+        identifier: 'test-plugin-d',
+        manifest: { name: 'TestPlugin', version: '1.0.0' } as unknown as LobeChatPluginManifest,
+        type: 'plugin',
+      } as InstallPluginParams;
+
+      // Act
+      await pluginService.installPlugin(fakePlugin);
+
+      // Assert
+      const result = await clientDB.query.installedPlugins.findFirst({
+        where: eq(installedPlugins.identifier, fakePlugin.identifier),
+      });
+      expect(result).toMatchObject(fakePlugin);
+    });
+  });
+
+  describe('getInstalledPlugins', () => {
+    it('should return a list of installed plugins', async () => {
+      // Arrange
+      const fakePlugins = [{ identifier: 'test-plugin', type: 'plugin' }] as LobeTool[];
+      await clientDB
+        .insert(installedPlugins)
+        .values([{ identifier: 'test-plugin', type: 'plugin', userId }]);
+      // Act
+      const data = await pluginService.getInstalledPlugins();
+
+      // Assert
+      expect(data).toMatchObject(fakePlugins);
+    });
+  });
+
+  describe('uninstallPlugin', () => {
+    it('should uninstall a plugin', async () => {
+      // Arrange
+      const identifier = 'test-plugin';
+      await clientDB.insert(installedPlugins).values([{ identifier, type: 'plugin', userId }]);
+
+      // Act
+      await pluginService.uninstallPlugin(identifier);
+
+      // Assert
+      const result = await clientDB.query.installedPlugins.findFirst({
+        where: eq(installedPlugins.identifier, identifier),
+      });
+      expect(result).toBe(undefined);
+    });
+  });
+
+  describe('createCustomPlugin', () => {
+    it('should create a custom plugin', async () => {
+      // Arrange
+      const customPlugin = {
+        identifier: 'custom-plugin-x',
+        manifest: {},
+        type: 'customPlugin',
+      } as LobeToolCustomPlugin;
+
+      // Act
+      await pluginService.createCustomPlugin(customPlugin);
+
+      // Assert
+      const result = await clientDB.query.installedPlugins.findFirst({
+        where: eq(installedPlugins.identifier, customPlugin.identifier),
+      });
+      expect(result).toMatchObject(customPlugin);
+    });
+  });
+
+  describe('updatePlugin', () => {
+    it('should update a plugin', async () => {
+      // Arrange
+      const identifier = 'plugin-id';
+      const value = { customParams: { ab: '1' } } as unknown as LobeToolCustomPlugin;
+      await clientDB.insert(installedPlugins).values([{ identifier, type: 'plugin', userId }]);
+
+      // Act
+      await pluginService.updatePlugin(identifier, value);
+
+      // Assert
+      const result = await clientDB.query.installedPlugins.findFirst({
+        where: eq(installedPlugins.identifier, identifier),
+      });
+      expect(result).toMatchObject(value);
+    });
+  });
+
+  describe('updatePluginManifest', () => {
+    it('should update a plugin manifest', async () => {
+      // Arrange
+      const identifier = 'plugin-id';
+      const manifest = { name: 'NewPluginManifest' } as unknown as LobeChatPluginManifest;
+      await clientDB.insert(installedPlugins).values([{ identifier, type: 'plugin', userId }]);
+
+      // Act
+      await pluginService.updatePluginManifest(identifier, manifest);
+
+      // Assert
+      const result = await clientDB.query.installedPlugins.findFirst({
+        where: eq(installedPlugins.identifier, identifier),
+      });
+      expect(result).toMatchObject({ manifest });
+    });
+  });
+
+  describe('removeAllPlugins', () => {
+    it('should remove all plugins', async () => {
+      // Arrange
+      await clientDB.insert(installedPlugins).values([
+        { identifier: '123', type: 'plugin', userId },
+        { identifier: '234', type: 'plugin', userId },
+      ]);
+
+      // Act
+      await pluginService.removeAllPlugins();
+
+      // Assert
+      const result = await clientDB.query.installedPlugins.findMany({
+        where: eq(installedPlugins.userId, userId),
+      });
+      expect(result.length).toEqual(0);
+    });
+  });
+
+  describe('updatePluginSettings', () => {
+    it('should update plugin settings', async () => {
+      // Arrange
+      const id = 'plugin-id';
+      const settings = { color: 'blue' };
+      await clientDB.insert(installedPlugins).values([{ identifier: id, type: 'plugin', userId }]);
+
+      // Act
+      await pluginService.updatePluginSettings(id, settings);
+
+      // Assert
+      const result = await clientDB.query.installedPlugins.findFirst({
+        where: eq(installedPlugins.identifier, id),
+      });
+
+      expect(result).toMatchObject({ settings });
+    });
+  });
+});

package/src/services/plugin/pglite.ts

@@ -0,0 +1,51 @@
+import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
+
+import { clientDB } from '@/database/client/db';
+import { PluginModel } from '@/database/server/models/plugin';
+import { BaseClientService } from '@/services/baseClientService';
+import { LobeTool } from '@/types/tool';
+import { LobeToolCustomPlugin } from '@/types/tool/plugin';
+
+import { IPluginService, InstallPluginParams } from './type';
+
+export class ClientService extends BaseClientService implements IPluginService {
+  private get pluginModel(): PluginModel {
+    return new PluginModel(clientDB as any, this.userId);
+  }
+
+  installPlugin = async (plugin: InstallPluginParams) => {
+    await this.pluginModel.create(plugin);
+    return;
+  };
+
+  getInstalledPlugins = () => {
+    return this.pluginModel.query() as Promise<LobeTool[]>;
+  };
+
+  async uninstallPlugin(identifier: string) {
+    await this.pluginModel.delete(identifier);
+    return;
+  }
+
+  async createCustomPlugin(customPlugin: LobeToolCustomPlugin) {
+    await this.pluginModel.create({ ...customPlugin, type: 'customPlugin' });
+    return;
+  }
+
+  async updatePlugin(id: string, value: LobeToolCustomPlugin) {
+    await this.pluginModel.update(id, value);
+    return;
+  }
+  async updatePluginManifest(id: string, manifest: LobeChatPluginManifest) {
+    await this.pluginModel.update(id, { manifest });
+  }
+
+  async removeAllPlugins() {
+    await this.pluginModel.deleteAll();
+  }
+
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  async updatePluginSettings(id: string, settings: any, _?: AbortSignal) {
+    await this.pluginModel.update(id, { settings });
+  }
+}

package/src/services/session/{client.test.ts → _deprecated.test.ts}

@@ -5,7 +5,7 @@ import { SessionGroupModel } from '@/database/_deprecated/models/sessionGroup';
 import { LobeAgentConfig } from '@/types/agent';
 import { LobeAgentSession, LobeSessionType, SessionGroups } from '@/types/session';
 
-import { ClientService } from './client';
+import { ClientService } from './_deprecated';
 
 const sessionService = new ClientService();
 

package/src/services/session/{client.ts → _deprecated.ts}

@@ -166,7 +166,7 @@ export class ClientService implements ISessionService {
   }
 
   async updateSessionGroup(id: string, data: Partial<SessionGroupItem>) {
-    return SessionGroupModel.update(id, data);
+    return SessionGroupModel.update(id, data as any);
   }
 
   async updateSessionGroupOrder(sortMap: { id: string; sort: number }[]) {

package/src/services/session/index.ts

@@ -1,5 +1,9 @@
-import { ClientService } from './client';
+import { ClientService as DeprecatedService } from './_deprecated';
+import { ClientService } from './pglite';
 import { ServerService } from './server';
 
+const clientService =
+  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
+
 export const sessionService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : new ClientService();
+  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;