@lobehub/lobehub 2.0.0-next.2 → 2.0.0-next.20

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

@@ -1,162 +0,0 @@
-import { LobeTool } from '@lobechat/types';
-import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
-import { beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { PluginModel } from '@/database/_deprecated/models/plugin';
-import { DB_Plugin } from '@/database/_deprecated/schemas/plugin';
-import { LobeToolCustomPlugin } from '@/types/tool/plugin';
-
-import { ClientService } from './_deprecated';
-import { InstallPluginParams } from './type';
-
-const pluginService = new ClientService();
-
-// Mocking modules and functions
-
-vi.mock('@/database/_deprecated/models/plugin', () => ({
-  PluginModel: {
-    getList: vi.fn(),
-    create: vi.fn(),
-    delete: vi.fn(),
-    update: vi.fn(),
-    clear: vi.fn(),
-  },
-}));
-
-beforeEach(() => {
-  vi.resetAllMocks();
-});
-
-describe('PluginService', () => {
-  describe('installPlugin', () => {
-    it('should install a plugin', async () => {
-      // Arrange
-      const fakePlugin = {
-        identifier: 'test-plugin',
-        manifest: { name: 'TestPlugin', version: '1.0.0' } as unknown as LobeChatPluginManifest,
-        type: 'plugin',
-      } as InstallPluginParams;
-      vi.mocked(PluginModel.create).mockResolvedValue(fakePlugin);
-
-      // Act
-      const installedPlugin = await pluginService.installPlugin(fakePlugin);
-
-      // Assert
-      expect(PluginModel.create).toHaveBeenCalledWith(fakePlugin);
-      expect(installedPlugin).toEqual(fakePlugin);
-    });
-  });
-
-  describe('getInstalledPlugins', () => {
-    it('should return a list of installed plugins', async () => {
-      // Arrange
-      const fakePlugins = [{ identifier: 'test-plugin', type: 'plugin' }] as LobeTool[];
-      vi.mocked(PluginModel.getList).mockResolvedValue(fakePlugins as DB_Plugin[]);
-
-      // Act
-      const installedPlugins = await pluginService.getInstalledPlugins();
-
-      // Assert
-      expect(PluginModel.getList).toHaveBeenCalled();
-      expect(installedPlugins).toEqual(fakePlugins);
-    });
-  });
-
-  describe('uninstallPlugin', () => {
-    it('should uninstall a plugin', async () => {
-      // Arrange
-      const identifier = 'test-plugin';
-      vi.mocked(PluginModel.delete).mockResolvedValue();
-
-      // Act
-      const result = await pluginService.uninstallPlugin(identifier);
-
-      // Assert
-      expect(PluginModel.delete).toHaveBeenCalledWith(identifier);
-      expect(result).toBe(undefined);
-    });
-  });
-
-  describe('createCustomPlugin', () => {
-    it('should create a custom plugin', async () => {
-      // Arrange
-      const customPlugin = {
-        identifier: 'custom-plugin',
-        manifest: {},
-        type: 'customPlugin',
-      } as LobeToolCustomPlugin;
-      vi.mocked(PluginModel.create).mockResolvedValue(customPlugin);
-
-      // Act
-      const result = await pluginService.createCustomPlugin(customPlugin);
-
-      // Assert
-      expect(PluginModel.create).toHaveBeenCalledWith({
-        ...customPlugin,
-        type: 'customPlugin',
-      });
-      expect(result).toEqual(customPlugin);
-    });
-  });
-
-  describe('updatePlugin', () => {
-    it('should update a plugin', async () => {
-      // Arrange
-      const id = 'plugin-id';
-      const value = { settings: { ab: '1' } } as unknown as LobeToolCustomPlugin;
-      vi.mocked(PluginModel.update).mockResolvedValue(1);
-
-      // Act
-      const result = await pluginService.updatePlugin(id, value);
-
-      // Assert
-      expect(PluginModel.update).toHaveBeenCalledWith(id, value);
-      expect(result).toEqual(undefined);
-    });
-  });
-
-  describe('updatePluginManifest', () => {
-    it('should update a plugin manifest', async () => {
-      // Arrange
-      const id = 'plugin-id';
-      const manifest = { name: 'NewPluginManifest' } as unknown as LobeChatPluginManifest;
-      vi.mocked(PluginModel.update).mockResolvedValue(1);
-
-      // Act
-      const result = await pluginService.updatePluginManifest(id, manifest);
-
-      // Assert
-      expect(PluginModel.update).toHaveBeenCalledWith(id, { manifest });
-      expect(result).toEqual(undefined);
-    });
-  });
-
-  describe('removeAllPlugins', () => {
-    it('should remove all plugins', async () => {
-      // Arrange
-      vi.mocked(PluginModel.clear).mockResolvedValue(undefined);
-
-      // Act
-      const result = await pluginService.removeAllPlugins();
-
-      // Assert
-      expect(PluginModel.clear).toHaveBeenCalled();
-      expect(result).toBe(undefined);
-    });
-  });
-
-  describe('updatePluginSettings', () => {
-    it('should update plugin settings', async () => {
-      // Arrange
-      const id = 'plugin-id';
-      const settings = { color: 'blue' };
-
-      // Act
-      const result = await pluginService.updatePluginSettings(id, settings);
-
-      // Assert
-      expect(PluginModel.update).toHaveBeenCalledWith(id, { settings });
-      expect(result).toEqual(undefined);
-    });
-  });
-});

package/src/services/plugin/_deprecated.ts

@@ -1,42 +0,0 @@
-import { LobeTool } from '@lobechat/types';
-import { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
-
-import { PluginModel } from '@/database/_deprecated/models/plugin';
-import { LobeToolCustomPlugin } from '@/types/tool/plugin';
-
-import { IPluginService, InstallPluginParams } from './type';
-
-export class ClientService implements IPluginService {
-  installPlugin = async (plugin: InstallPluginParams) => {
-    return PluginModel.create(plugin);
-  };
-
-  getInstalledPlugins = () => {
-    return PluginModel.getList() as Promise<LobeTool[]>;
-  };
-
-  uninstallPlugin(identifier: string) {
-    return PluginModel.delete(identifier);
-  }
-
-  async createCustomPlugin(customPlugin: LobeToolCustomPlugin) {
-    return PluginModel.create({ ...customPlugin, type: 'customPlugin' });
-  }
-
-  async updatePlugin(id: string, value: Partial<LobeToolCustomPlugin>) {
-    await PluginModel.update(id, value);
-    return;
-  }
-  async updatePluginManifest(id: string, manifest: LobeChatPluginManifest) {
-    await PluginModel.update(id, { manifest });
-  }
-
-  async removeAllPlugins() {
-    return PluginModel.clear();
-  }
-
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async updatePluginSettings(id: string, settings: any, _?: AbortSignal) {
-    await PluginModel.update(id, { settings });
-  }
-}

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

@@ -1,177 +0,0 @@
-import { LobeTool } from '@lobechat/types';
-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 { userInstalledPlugins, users } from '@/database/schemas';
-import { LobeToolCustomPlugin } from '@/types/tool/plugin';
-
-import { ClientService } from './client';
-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.userInstalledPlugins.findFirst({
-        where: eq(userInstalledPlugins.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(userInstalledPlugins)
-        .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(userInstalledPlugins).values([{ identifier, type: 'plugin', userId }]);
-
-      // Act
-      await pluginService.uninstallPlugin(identifier);
-
-      // Assert
-      const result = await clientDB.query.userInstalledPlugins.findFirst({
-        where: eq(userInstalledPlugins.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.userInstalledPlugins.findFirst({
-        where: eq(userInstalledPlugins.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(userInstalledPlugins).values([{ identifier, type: 'plugin', userId }]);
-
-      // Act
-      await pluginService.updatePlugin(identifier, value);
-
-      // Assert
-      const result = await clientDB.query.userInstalledPlugins.findFirst({
-        where: eq(userInstalledPlugins.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(userInstalledPlugins).values([{ identifier, type: 'plugin', userId }]);
-
-      // Act
-      await pluginService.updatePluginManifest(identifier, manifest);
-
-      // Assert
-      const result = await clientDB.query.userInstalledPlugins.findFirst({
-        where: eq(userInstalledPlugins.identifier, identifier),
-      });
-      expect(result).toMatchObject({ manifest });
-    });
-  });
-
-  describe('removeAllPlugins', () => {
-    it('should remove all plugins', async () => {
-      // Arrange
-      await clientDB.insert(userInstalledPlugins).values([
-        { identifier: '123', type: 'plugin', userId },
-        { identifier: '234', type: 'plugin', userId },
-      ]);
-
-      // Act
-      await pluginService.removeAllPlugins();
-
-      // Assert
-      const result = await clientDB.query.userInstalledPlugins.findMany({
-        where: eq(userInstalledPlugins.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(userInstalledPlugins)
-        .values([{ identifier: id, type: 'plugin', userId }]);
-
-      // Act
-      await pluginService.updatePluginSettings(id, settings);
-
-      // Assert
-      const result = await clientDB.query.userInstalledPlugins.findFirst({
-        where: eq(userInstalledPlugins.identifier, id),
-      });
-
-      expect(result).toMatchObject({ settings });
-    });
-  });
-});

package/src/services/plugin/client.ts

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