@lobehub/lobehub 2.0.0-next.163 → 2.0.0-next.165

package/apps/desktop/src/main/controllers/__tests__/SystemCtr.test.ts

@@ -2,9 +2,38 @@ import { ThemeMode } from '@lobechat/electron-client-ipc';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import type { App } from '@/core/App';
+import type { IpcContext } from '@/utils/ipc';
+import { IpcHandler } from '@/utils/ipc/base';
 
 import SystemController from '../SystemCtr';
 
+const { ipcHandlers, ipcMainHandleMock } = vi.hoisted(() => {
+  const handlers = new Map<string, (event: any, ...args: any[]) => any>();
+  const handle = vi.fn((channel: string, handler: any) => {
+    handlers.set(channel, handler);
+  });
+  return { ipcHandlers: handlers, ipcMainHandleMock: handle };
+});
+
+const invokeIpc = async <T = any>(
+  channel: string,
+  payload?: any,
+  context?: Partial<IpcContext>,
+): Promise<T> => {
+  const handler = ipcHandlers.get(channel);
+  if (!handler) throw new Error(`IPC handler for ${channel} not found`);
+
+  const fakeEvent = {
+    sender: context?.sender ?? ({ id: 'test' } as any),
+  };
+
+  if (payload === undefined) {
+    return handler(fakeEvent);
+  }
+
+  return handler(fakeEvent, payload);
+};
+
 // Mock logger
 vi.mock('@/utils/logger', () => ({
   createLogger: () => ({
@@ -21,6 +50,9 @@ vi.mock('electron', () => ({
     getLocale: vi.fn(() => 'en-US'),
     getPath: vi.fn((name: string) => `/mock/path/${name}`),
   },
+  ipcMain: {
+    handle: ipcMainHandleMock,
+  },
   nativeTheme: {
     on: vi.fn(),
     shouldUseDarkColors: false,
@@ -38,19 +70,6 @@ vi.mock('electron-is', () => ({
   macOS: vi.fn(() => true),
 }));
 
-// Mock node:fs
-vi.mock('node:fs', () => ({
-  readFileSync: vi.fn(),
-  writeFileSync: vi.fn(),
-}));
-
-// Mock @/const/dir
-vi.mock('@/const/dir', () => ({
-  DB_SCHEMA_HASH_FILENAME: 'db-schema-hash.txt',
-  LOCAL_DATABASE_DIR: 'database',
-  userDataDir: '/mock/user/data',
-}));
-
 // Mock browserManager
 const mockBrowserManager = {
   broadcastToAllWindows: vi.fn(),
@@ -80,12 +99,15 @@ describe('SystemController', () => {
 
   beforeEach(() => {
     vi.clearAllMocks();
+    ipcHandlers.clear();
+    ipcMainHandleMock.mockClear();
+    (IpcHandler.getInstance() as any).registeredChannels?.clear();
     controller = new SystemController(mockApp);
   });
 
   describe('getAppState', () => {
     it('should return app state with system info', async () => {
-      const result = await controller.getAppState();
+      const result = await invokeIpc('system.getAppState');
 
       expect(result).toMatchObject({
         arch: expect.any(String),
@@ -108,7 +130,7 @@ describe('SystemController', () => {
       const { nativeTheme } = await import('electron');
       Object.defineProperty(nativeTheme, 'shouldUseDarkColors', { value: true });
 
-      const result = await controller.getAppState();
+      const result = await invokeIpc('system.getAppState');
 
       expect(result.systemAppearance).toBe('dark');
 
@@ -121,7 +143,7 @@ describe('SystemController', () => {
     it('should check accessibility on macOS', async () => {
       const { systemPreferences } = await import('electron');
 
-      controller.checkAccessibilityForMacOS();
+      await invokeIpc('system.checkAccessibilityForMacOS');
 
       expect(systemPreferences.isTrustedAccessibilityClient).toHaveBeenCalledWith(true);
     });
@@ -130,7 +152,7 @@ describe('SystemController', () => {
       const { macOS } = await import('electron-is');
       vi.mocked(macOS).mockReturnValue(false);
 
-      const result = controller.checkAccessibilityForMacOS();
+      const result = await invokeIpc('system.checkAccessibilityForMacOS');
 
       expect(result).toBeUndefined();
 
@@ -143,7 +165,7 @@ describe('SystemController', () => {
     it('should open external link', async () => {
       const { shell } = await import('electron');
 
-      await controller.openExternalLink('https://example.com');
+      await invokeIpc('system.openExternalLink', 'https://example.com');
 
       expect(shell.openExternal).toHaveBeenCalledWith('https://example.com');
     });
@@ -151,7 +173,7 @@ describe('SystemController', () => {
 
   describe('updateLocale', () => {
     it('should update locale and broadcast change', async () => {
-      const result = await controller.updateLocale('zh-CN');
+      const result = await invokeIpc('system.updateLocale', 'zh-CN');
 
       expect(mockStoreManager.set).toHaveBeenCalledWith('locale', 'zh-CN');
       expect(mockI18n.changeLanguage).toHaveBeenCalledWith('zh-CN');
@@ -162,7 +184,7 @@ describe('SystemController', () => {
     });
 
     it('should use system locale when set to auto', async () => {
-      await controller.updateLocale('auto');
+      await invokeIpc('system.updateLocale', 'auto');
 
       expect(mockI18n.changeLanguage).toHaveBeenCalledWith('en-US');
     });
@@ -172,7 +194,7 @@ describe('SystemController', () => {
     it('should update theme mode and broadcast change', async () => {
       const themeMode: ThemeMode = 'dark';
 
-      await controller.updateThemeModeHandler(themeMode);
+      await invokeIpc('system.updateThemeModeHandler', themeMode);
 
       expect(mockStoreManager.set).toHaveBeenCalledWith('themeMode', 'dark');
       expect(mockBrowserManager.broadcastToAllWindows).toHaveBeenCalledWith('themeChanged', {
@@ -182,58 +204,6 @@ describe('SystemController', () => {
     });
   });
 
-  describe('getDatabasePath', () => {
-    it('should return database path', async () => {
-      const result = await controller.getDatabasePath();
-
-      expect(result).toBe('/mock/storage/database');
-    });
-  });
-
-  describe('getDatabaseSchemaHash', () => {
-    it('should return schema hash when file exists', async () => {
-      const { readFileSync } = await import('node:fs');
-      vi.mocked(readFileSync).mockReturnValue('abc123');
-
-      const result = await controller.getDatabaseSchemaHash();
-
-      expect(result).toBe('abc123');
-    });
-
-    it('should return undefined when file does not exist', async () => {
-      const { readFileSync } = await import('node:fs');
-      vi.mocked(readFileSync).mockImplementation(() => {
-        throw new Error('File not found');
-      });
-
-      const result = await controller.getDatabaseSchemaHash();
-
-      expect(result).toBeUndefined();
-    });
-  });
-
-  describe('getUserDataPath', () => {
-    it('should return user data path', async () => {
-      const result = await controller.getUserDataPath();
-
-      expect(result).toBe('/mock/user/data');
-    });
-  });
-
-  describe('setDatabaseSchemaHash', () => {
-    it('should write schema hash to file', async () => {
-      const { writeFileSync } = await import('node:fs');
-
-      await controller.setDatabaseSchemaHash('newhash123');
-
-      expect(writeFileSync).toHaveBeenCalledWith(
-        '/mock/storage/db-schema-hash.txt',
-        'newhash123',
-        'utf8',
-      );
-    });
-  });
-
   describe('afterAppReady', () => {
     it('should initialize system theme listener', async () => {
       const { nativeTheme } = await import('electron');

package/apps/desktop/src/main/controllers/__tests__/SystemServerCtr.test.ts

@@ -0,0 +1,75 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+import type { App } from '@/core/App';
+
+import SystemServerCtr from '../SystemServerCtr';
+
+vi.mock('@/utils/logger', () => ({
+  createLogger: () => ({
+    debug: vi.fn(),
+    error: vi.fn(),
+    info: vi.fn(),
+    warn: vi.fn(),
+  }),
+}));
+
+vi.mock('node:fs', () => ({
+  readFileSync: vi.fn(),
+  writeFileSync: vi.fn(),
+}));
+
+vi.mock('@/const/dir', () => ({
+  DB_SCHEMA_HASH_FILENAME: 'db-schema-hash.txt',
+  LOCAL_DATABASE_DIR: 'database',
+  userDataDir: '/mock/user/data',
+}));
+
+const mockApp = {
+  appStoragePath: '/mock/storage',
+} as unknown as App;
+
+describe('SystemServerCtr', () => {
+  let controller: SystemServerCtr;
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+    controller = new SystemServerCtr(mockApp);
+  });
+
+  it('returns database path', async () => {
+    await expect(controller.getDatabasePath()).resolves.toBe('/mock/storage/database');
+  });
+
+  it('reads schema hash when file exists', async () => {
+    const { readFileSync } = await import('node:fs');
+    vi.mocked(readFileSync).mockReturnValue('hash123');
+
+    await expect(controller.getDatabaseSchemaHash()).resolves.toBe('hash123');
+    expect(readFileSync).toHaveBeenCalledWith('/mock/storage/db-schema-hash.txt', 'utf8');
+  });
+
+  it('returns undefined when schema hash file missing', async () => {
+    const { readFileSync } = await import('node:fs');
+    vi.mocked(readFileSync).mockImplementation(() => {
+      throw new Error('missing');
+    });
+
+    await expect(controller.getDatabaseSchemaHash()).resolves.toBeUndefined();
+  });
+
+  it('returns user data path', async () => {
+    await expect(controller.getUserDataPath()).resolves.toBe('/mock/user/data');
+  });
+
+  it('writes schema hash to disk', async () => {
+    const { writeFileSync } = await import('node:fs');
+
+    await controller.setDatabaseSchemaHash('newhash');
+
+    expect(writeFileSync).toHaveBeenCalledWith(
+      '/mock/storage/db-schema-hash.txt',
+      'newhash',
+      'utf8',
+    );
+  });
+});

package/apps/desktop/src/main/controllers/__tests__/TrayMenuCtr.test.ts

@@ -1,12 +1,24 @@
-import { afterAll, beforeEach, describe, expect, it, vi } from 'vitest';
-import { 
+import {
   ShowTrayNotificationParams,
   UpdateTrayIconParams,
-  UpdateTrayTooltipParams
+  UpdateTrayTooltipParams,
 } from '@lobechat/electron-client-ipc';
+import { afterAll, beforeEach, describe, expect, it, vi } from 'vitest';
 
 import type { App } from '@/core/App';
 
+import TrayMenuCtr from '../TrayMenuCtr';
+
+const { ipcMainHandleMock } = vi.hoisted(() => ({
+  ipcMainHandleMock: vi.fn(),
+}));
+
+vi.mock('electron', () => ({
+  ipcMain: {
+    handle: ipcMainHandleMock,
+  },
+}));
+
 // 模拟 logger
 vi.mock('@/utils/logger', () => ({
   createLogger: () => ({
@@ -15,8 +27,6 @@ vi.mock('@/utils/logger', () => ({
   }),
 }));
 
-import TrayMenuCtr from '../TrayMenuCtr';
-
 // 保存原始平台，确保测试结束后能恢复
 const originalPlatform = process.platform;
 
@@ -45,6 +55,7 @@ describe('TrayMenuCtr', () => {
 
   beforeEach(() => {
     vi.clearAllMocks();
+    ipcMainHandleMock.mockClear();
     // 为每个测试重置 mockedTray
     mockGetMainTray.mockReset();
     trayMenuCtr = new TrayMenuCtr(mockApp);
@@ -69,7 +80,7 @@ describe('TrayMenuCtr', () => {
     it('should display balloon notification on Windows platform', async () => {
       // 模拟 Windows 平台
       Object.defineProperty(process, 'platform', { value: 'win32' });
-      
+
       const mockedTray = {
         displayBalloon: mockDisplayBalloon,
       };
@@ -125,9 +136,9 @@ describe('TrayMenuCtr', () => {
 
       expect(mockGetMainTray).toHaveBeenCalled();
       expect(mockDisplayBalloon).not.toHaveBeenCalled();
-      expect(result).toEqual({ 
+      expect(result).toEqual({
         error: 'Tray notifications are only supported on Windows platform',
-        success: false
+        success: false,
       });
     });
   });
@@ -136,7 +147,7 @@ describe('TrayMenuCtr', () => {
     it('should update tray icon on Windows platform', async () => {
       // 模拟 Windows 平台
       Object.defineProperty(process, 'platform', { value: 'win32' });
-      
+
       const mockedTray = {
         updateIcon: mockUpdateIcon,
       };
@@ -156,7 +167,7 @@ describe('TrayMenuCtr', () => {
     it('should handle errors when updating icon', async () => {
       // 模拟 Windows 平台
       Object.defineProperty(process, 'platform', { value: 'win32' });
-      
+
       const error = new Error('Failed to update icon');
       const mockedTray = {
         updateIcon: vi.fn().mockImplementation(() => {
@@ -198,7 +209,7 @@ describe('TrayMenuCtr', () => {
     it('should update tray tooltip on Windows platform', async () => {
       // 模拟 Windows 平台
       Object.defineProperty(process, 'platform', { value: 'win32' });
-      
+
       const mockedTray = {
         updateTooltip: mockUpdateTooltip,
       };
@@ -234,7 +245,7 @@ describe('TrayMenuCtr', () => {
     it('should return error when tooltip is not provided', async () => {
       // 模拟 Windows 平台
       Object.defineProperty(process, 'platform', { value: 'win32' });
-      
+
       const mockedTray = {
         updateTooltip: mockUpdateTooltip,
       };
@@ -253,4 +264,4 @@ describe('TrayMenuCtr', () => {
       });
     });
   });
-}); 
+});

package/apps/desktop/src/main/controllers/__tests__/UpdaterCtr.test.ts

@@ -2,6 +2,8 @@ import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import type { App } from '@/core/App';
 
+import UpdaterCtr from '../UpdaterCtr';
+
 // 模拟 logger
 vi.mock('@/utils/logger', () => ({
   createLogger: () => ({
@@ -9,7 +11,15 @@ vi.mock('@/utils/logger', () => ({
   }),
 }));
 
-import UpdaterCtr from '../UpdaterCtr';
+const { ipcMainHandleMock } = vi.hoisted(() => ({
+  ipcMainHandleMock: vi.fn(),
+}));
+
+vi.mock('electron', () => ({
+  ipcMain: {
+    handle: ipcMainHandleMock,
+  },
+}));
 
 // 模拟 App 及其依赖项
 const mockCheckForUpdates = vi.fn();
@@ -31,6 +41,7 @@ describe('UpdaterCtr', () => {
 
   beforeEach(() => {
     vi.clearAllMocks();
+    ipcMainHandleMock.mockClear();
     updaterCtr = new UpdaterCtr(mockApp);
   });
 
@@ -79,4 +90,4 @@ describe('UpdaterCtr', () => {
       await expect(updaterCtr.downloadUpdate()).rejects.toThrow(error);
     });
   });
-}); 
+});

package/apps/desktop/src/main/controllers/__tests__/UploadFileCtr.test.ts

@@ -1,9 +1,33 @@
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 import type { App } from '@/core/App';
+import { IpcHandler } from '@/utils/ipc/base';
 
 import UploadFileCtr from '../UploadFileCtr';
 
+const { ipcHandlers, ipcMainHandleMock } = vi.hoisted(() => {
+  const handlers = new Map<string, (event: any, ...args: any[]) => any>();
+  const handle = vi.fn((channel: string, handler: any) => {
+    handlers.set(channel, handler);
+  });
+  return { ipcHandlers: handlers, ipcMainHandleMock: handle };
+});
+
+const invokeIpc = async <T = any>(channel: string, payload?: any): Promise<T> => {
+  const handler = ipcHandlers.get(channel);
+  if (!handler) throw new Error(`IPC handler for ${channel} not found`);
+
+  const fakeEvent = { sender: { id: 'test' } as any };
+  if (payload === undefined) return handler(fakeEvent);
+  return handler(fakeEvent, payload);
+};
+
+vi.mock('electron', () => ({
+  ipcMain: {
+    handle: ipcMainHandleMock,
+  },
+}));
+
 // Mock FileService module to prevent electron dependency issues
 vi.mock('@/services/fileSrv', () => ({
   default: class MockFileService {},
@@ -12,9 +36,6 @@ vi.mock('@/services/fileSrv', () => ({
 // Mock FileService instance methods
 const mockFileService = {
   uploadFile: vi.fn(),
-  getFilePath: vi.fn(),
-  getFileHTTPURL: vi.fn(),
-  deleteFiles: vi.fn(),
 };
 
 const mockApp = {
@@ -26,6 +47,9 @@ describe('UploadFileCtr', () => {
 
   beforeEach(() => {
     vi.clearAllMocks();
+    ipcHandlers.clear();
+    ipcMainHandleMock.mockClear();
+    (IpcHandler.getInstance() as any).registeredChannels?.clear();
     controller = new UploadFileCtr(mockApp);
   });
 
@@ -41,7 +65,7 @@ describe('UploadFileCtr', () => {
       const expectedResult = { id: 'file-id-123', url: '/files/file-id-123' };
       mockFileService.uploadFile.mockResolvedValue(expectedResult);
 
-      const result = await controller.uploadFile(params);
+      const result = await invokeIpc('upload.uploadFile', params);
 
       expect(result).toEqual(expectedResult);
       expect(mockFileService.uploadFile).toHaveBeenCalledWith(params);
@@ -58,110 +82,7 @@ describe('UploadFileCtr', () => {
       const error = new Error('Upload failed');
       mockFileService.uploadFile.mockRejectedValue(error);
 
-      await expect(controller.uploadFile(params)).rejects.toThrow('Upload failed');
-    });
-  });
-
-  describe('getFileUrlById', () => {
-    it('should get file path by id successfully', async () => {
-      const fileId = 'file-id-123';
-      const expectedPath = '/files/abc123.txt';
-      mockFileService.getFilePath.mockResolvedValue(expectedPath);
-
-      const result = await controller.getFileUrlById(fileId);
-
-      expect(result).toBe(expectedPath);
-      expect(mockFileService.getFilePath).toHaveBeenCalledWith(fileId);
-    });
-
-    it('should handle get file path error', async () => {
-      const fileId = 'non-existent-id';
-      const error = new Error('File not found');
-      mockFileService.getFilePath.mockRejectedValue(error);
-
-      await expect(controller.getFileUrlById(fileId)).rejects.toThrow('File not found');
-    });
-  });
-
-  describe('getFileHTTPURL', () => {
-    it('should get file HTTP URL successfully', async () => {
-      const filePath = '/files/abc123.txt';
-      const expectedUrl = 'http://localhost:3000/files/abc123.txt';
-      mockFileService.getFileHTTPURL.mockResolvedValue(expectedUrl);
-
-      const result = await controller.getFileHTTPURL(filePath);
-
-      expect(result).toBe(expectedUrl);
-      expect(mockFileService.getFileHTTPURL).toHaveBeenCalledWith(filePath);
-    });
-
-    it('should handle get HTTP URL error', async () => {
-      const filePath = '/files/abc123.txt';
-      const error = new Error('Failed to generate URL');
-      mockFileService.getFileHTTPURL.mockRejectedValue(error);
-
-      await expect(controller.getFileHTTPURL(filePath)).rejects.toThrow('Failed to generate URL');
-    });
-  });
-
-  describe('deleteFiles', () => {
-    it('should delete files successfully', async () => {
-      const paths = ['/files/file1.txt', '/files/file2.txt'];
-      mockFileService.deleteFiles.mockResolvedValue(undefined);
-
-      await controller.deleteFiles(paths);
-
-      expect(mockFileService.deleteFiles).toHaveBeenCalledWith(paths);
-    });
-
-    it('should handle delete files error', async () => {
-      const paths = ['/files/file1.txt'];
-      const error = new Error('Delete failed');
-      mockFileService.deleteFiles.mockRejectedValue(error);
-
-      await expect(controller.deleteFiles(paths)).rejects.toThrow('Delete failed');
-    });
-
-    it('should handle empty paths array', async () => {
-      const paths: string[] = [];
-      mockFileService.deleteFiles.mockResolvedValue(undefined);
-
-      await controller.deleteFiles(paths);
-
-      expect(mockFileService.deleteFiles).toHaveBeenCalledWith([]);
-    });
-  });
-
-  describe('createFile', () => {
-    it('should create file successfully', async () => {
-      const params = {
-        hash: 'xyz789',
-        path: '/test/newfile.txt',
-        content: 'bmV3IGZpbGUgY29udGVudA==',
-        filename: 'newfile.txt',
-        type: 'text/plain',
-      };
-      const expectedResult = { id: 'new-file-id', url: '/files/new-file-id' };
-      mockFileService.uploadFile.mockResolvedValue(expectedResult);
-
-      const result = await controller.createFile(params);
-
-      expect(result).toEqual(expectedResult);
-      expect(mockFileService.uploadFile).toHaveBeenCalledWith(params);
-    });
-
-    it('should handle create file error', async () => {
-      const params = {
-        hash: 'xyz789',
-        path: '/test/newfile.txt',
-        content: 'bmV3IGZpbGUgY29udGVudA==',
-        filename: 'newfile.txt',
-        type: 'text/plain',
-      };
-      const error = new Error('Create failed');
-      mockFileService.uploadFile.mockRejectedValue(error);
-
-      await expect(controller.createFile(params)).rejects.toThrow('Create failed');
+      await expect(invokeIpc('upload.uploadFile', params)).rejects.toThrow('Upload failed');
     });
   });
 });

package/apps/desktop/src/main/controllers/__tests__/UploadFileServerCtr.test.ts

@@ -0,0 +1,55 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+import type { App } from '@/core/App';
+
+import UploadFileServerCtr from '../UploadFileServerCtr';
+
+vi.mock('@/services/fileSrv', () => ({
+  default: class MockFileService {},
+}));
+
+const mockFileService = {
+  getFileHTTPURL: vi.fn(),
+  getFilePath: vi.fn(),
+  deleteFiles: vi.fn(),
+  uploadFile: vi.fn(),
+};
+
+const mockApp = {
+  getService: vi.fn(() => mockFileService),
+} as unknown as App;
+
+describe('UploadFileServerCtr', () => {
+  let controller: UploadFileServerCtr;
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+    controller = new UploadFileServerCtr(mockApp);
+  });
+
+  it('gets file path by id', async () => {
+    mockFileService.getFilePath.mockResolvedValue('path');
+    await expect(controller.getFileUrlById('id')).resolves.toBe('path');
+    expect(mockFileService.getFilePath).toHaveBeenCalledWith('id');
+  });
+
+  it('gets HTTP URL', async () => {
+    mockFileService.getFileHTTPURL.mockResolvedValue('url');
+    await expect(controller.getFileHTTPURL('/path')).resolves.toBe('url');
+    expect(mockFileService.getFileHTTPURL).toHaveBeenCalledWith('/path');
+  });
+
+  it('deletes files', async () => {
+    mockFileService.deleteFiles.mockResolvedValue(undefined);
+    await controller.deleteFiles(['a']);
+    expect(mockFileService.deleteFiles).toHaveBeenCalledWith(['a']);
+  });
+
+  it('creates files via upload service', async () => {
+    const params = { filename: 'file' } as any;
+    mockFileService.uploadFile.mockResolvedValue({ success: true });
+
+    await expect(controller.createFile(params)).resolves.toEqual({ success: true });
+    expect(mockFileService.uploadFile).toHaveBeenCalledWith(params);
+  });
+});

package/apps/desktop/src/main/controllers/_template.ts

@@ -1,7 +1,7 @@
-import { ControllerModule, ipcClientEvent } from './index';
+import { ControllerModule, IpcMethod } from './index';
 
 export default class DevtoolsCtr extends ControllerModule {
-  @ipcClientEvent('openDevtools')
+  @IpcMethod()
   async openDevtools() {
     const devtoolsBrowser = this.app.browserManager.retrieveByIdentifier('devtools');
     devtoolsBrowser.show();