@lobehub/lobehub 2.0.0-next.142 → 2.0.0-next.143

package/apps/desktop/src/preload/streamer.test.ts

@@ -0,0 +1,365 @@
+import type { ProxyTRPCRequestParams } from '@lobechat/electron-client-ipc';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
+
+// Mock electron module
+const mockIpcRendererOn = vi.fn();
+const mockIpcRendererOnce = vi.fn();
+const mockIpcRendererSend = vi.fn();
+const mockIpcRendererRemoveAllListeners = vi.fn();
+
+vi.mock('electron', () => ({
+  ipcRenderer: {
+    on: mockIpcRendererOn,
+    once: mockIpcRendererOnce,
+    removeAllListeners: mockIpcRendererRemoveAllListeners,
+    send: mockIpcRendererSend,
+  },
+}));
+
+// Mock uuid
+vi.mock('uuid', () => ({
+  v4: vi.fn(() => 'test-request-id-123'),
+}));
+
+const { onStreamInvoke } = await import('./streamer');
+
+describe('onStreamInvoke', () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it('should set up stream listeners and send start event', () => {
+    const params: ProxyTRPCRequestParams = {
+      input: { query: 'test' },
+      path: 'test.endpoint',
+      type: 'query',
+    };
+
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Verify listeners are registered
+    expect(mockIpcRendererOn).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+      expect.any(Function),
+    );
+    expect(mockIpcRendererOnce).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+      expect.any(Function),
+    );
+    expect(mockIpcRendererOnce).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+      expect.any(Function),
+    );
+    expect(mockIpcRendererOnce).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+      expect.any(Function),
+    );
+
+    // Verify start event is sent
+    expect(mockIpcRendererSend).toHaveBeenCalledWith('stream:start', {
+      ...params,
+      requestId: 'test-request-id-123',
+    });
+  });
+
+  it('should invoke onData callback when data is received', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the data listener callback
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    // Simulate data event
+    const testData = Buffer.from('test data');
+    dataListener?.(null, testData);
+
+    expect(callbacks.onData).toHaveBeenCalledWith(new Uint8Array(testData));
+  });
+
+  it('should invoke onResponse callback when response is received', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the response listener callback
+    const responseListener = mockIpcRendererOnce.mock.calls.find((call) =>
+      call[0].includes('stream:response'),
+    )?.[1];
+
+    // Simulate response event
+    const testResponse = {
+      headers: { 'content-type': 'application/json' },
+      status: 200,
+      statusText: 'OK',
+    };
+    responseListener?.(null, testResponse);
+
+    expect(callbacks.onResponse).toHaveBeenCalledWith(testResponse);
+  });
+
+  it('should invoke onEnd callback and cleanup when stream ends', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the end listener callback
+    const endListener = mockIpcRendererOnce.mock.calls.find((call) =>
+      call[0].includes('stream:end'),
+    )?.[1];
+
+    // Simulate end event
+    endListener?.(null);
+
+    expect(callbacks.onEnd).toHaveBeenCalled();
+
+    // Verify cleanup
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+    );
+  });
+
+  it('should invoke onError callback and cleanup when error occurs', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    // Get the error listener callback
+    const errorListener = mockIpcRendererOnce.mock.calls.find((call) =>
+      call[0].includes('stream:error'),
+    )?.[1];
+
+    // Simulate error event
+    const testError = new Error('Stream processing failed');
+    errorListener?.(null, testError);
+
+    expect(callbacks.onError).toHaveBeenCalledWith(testError);
+
+    // Verify cleanup
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+    );
+  });
+
+  it('should return cleanup function that removes all listeners', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    const cleanup = onStreamInvoke(params, callbacks);
+
+    // Call cleanup function
+    cleanup();
+
+    // Verify all listeners are removed
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:data:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:end:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:error:test-request-id-123',
+    );
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledWith(
+      'stream:response:test-request-id-123',
+    );
+  });
+
+  it('should handle multiple data chunks', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    // Simulate multiple data chunks
+    const chunk1 = Buffer.from('chunk1');
+    const chunk2 = Buffer.from('chunk2');
+    const chunk3 = Buffer.from('chunk3');
+
+    dataListener?.(null, chunk1);
+    dataListener?.(null, chunk2);
+    dataListener?.(null, chunk3);
+
+    expect(callbacks.onData).toHaveBeenCalledTimes(3);
+    expect(callbacks.onData).toHaveBeenNthCalledWith(1, new Uint8Array(chunk1));
+    expect(callbacks.onData).toHaveBeenNthCalledWith(2, new Uint8Array(chunk2));
+    expect(callbacks.onData).toHaveBeenNthCalledWith(3, new Uint8Array(chunk3));
+  });
+
+  it('should handle complex request parameters', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {
+        filters: { active: true },
+        query: 'complex query',
+        sort: { field: 'date', order: 'desc' },
+      },
+      path: 'complex.nested.endpoint',
+      type: 'mutation',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    expect(mockIpcRendererSend).toHaveBeenCalledWith('stream:start', {
+      ...params,
+      requestId: 'test-request-id-123',
+    });
+  });
+
+  it('should not invoke callbacks after cleanup', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    const cleanup = onStreamInvoke(params, callbacks);
+
+    // Cleanup immediately
+    cleanup();
+
+    // Try to trigger callbacks after cleanup (this simulates late events)
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    // Since listeners are removed, this shouldn't do anything
+    // The actual behavior depends on electron's implementation
+    // But we can verify cleanup was called
+    expect(mockIpcRendererRemoveAllListeners).toHaveBeenCalledTimes(4);
+  });
+
+  it('should handle empty buffer data', () => {
+    const callbacks = {
+      onData: vi.fn(),
+      onEnd: vi.fn(),
+      onError: vi.fn(),
+      onResponse: vi.fn(),
+    };
+
+    const params: ProxyTRPCRequestParams = {
+      input: {},
+      path: 'test',
+      type: 'query',
+    };
+
+    onStreamInvoke(params, callbacks);
+
+    const dataListener = mockIpcRendererOn.mock.calls.find((call) =>
+      call[0].includes('stream:data'),
+    )?.[1];
+
+    const emptyBuffer = Buffer.from('');
+    dataListener?.(null, emptyBuffer);
+
+    expect(callbacks.onData).toHaveBeenCalledWith(new Uint8Array(emptyBuffer));
+  });
+});

package/apps/desktop/vitest.config.mts

@@ -5,6 +5,7 @@ export default defineConfig({
   test: {
     alias: {
       '@': resolve(__dirname, './src/main'),
+      '~common': resolve(__dirname, './src/common'),
     },
     coverage: {
       all: false,

package/changelog/v1.json

@@ -1,4 +1,13 @@
 [
+  {
+    "children": {
+      "features": [
+        "Support market cloud endpoint mcp."
+      ]
+    },
+    "date": "2025-12-02",
+    "version": "2.0.0-next.143"
+  },
   {
     "children": {
       "fixes": [

package/locales/ar/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "إلغاء",
+    "confirm": "تأكيد التفويض",
+    "description": {
+      "and": "و",
+      "prefix": "بالنقر على تأكيد التفويض، فإنك توافق على",
+      "privacy": "سياسة الخصوصية",
+      "terms": "شروط الخدمة"
+    },
+    "title": "تأكيد التفويض"
+  },
   "callback": {
     "buttons": {
       "close": "إغلاق النافذة"
@@ -33,8 +44,10 @@
     "stateMissing": "لم يتم العثور على حالة التفويض، يرجى المحاولة مرة أخرى."
   },
   "messages": {
+    "authorized": "تم تفويض خدمة LobeHub بنجاح",
     "loading": "جارٍ بدء عملية التفويض...",
     "success": {
+      "cloudMcpInstall": "تم التفويض بنجاح! يمكنك الآن تثبيت إضافة Cloud MCP.",
       "submit": "تم التفويض بنجاح! يمكنك الآن نشر المساعد.",
       "upload": "تم التفويض بنجاح! يمكنك الآن نشر إصدار جديد."
     }

package/locales/bg-BG/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Отказ",
+    "confirm": "Разреши използването",
+    "description": {
+      "and": "и",
+      "prefix": "С натискане на „Разреши използването“ се съгласявате с",
+      "privacy": "Политиката за поверителност",
+      "terms": "Условията за ползване"
+    },
+    "title": "Потвърждаване на оторизацията"
+  },
   "callback": {
     "buttons": {
       "close": "Затвори прозореца"
@@ -33,8 +44,10 @@
     "stateMissing": "Състоянието на упълномощаване не бе намерено, моля опитайте отново."
   },
   "messages": {
+    "authorized": "Успешно упълномощаване на услугата LobeHub",
     "loading": "Стартиране на процеса по упълномощаване...",
     "success": {
+      "cloudMcpInstall": "Успешно разрешение! Вече можете да инсталирате приставката Cloud MCP.",
       "submit": "Упълномощаването е успешно! Вече можете да публикувате помощник.",
       "upload": "Упълномощаването е успешно! Вече можете да публикувате нова версия."
     }

package/locales/de-DE/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Abbrechen",
+    "confirm": "Autorisieren",
+    "description": {
+      "and": "und",
+      "prefix": "Mit dem Klick auf „Autorisieren“ stimmen Sie den",
+      "privacy": "Datenschutzbestimmungen",
+      "terms": "Nutzungsbedingungen"
+    },
+    "title": "Autorisierung bestätigen"
+  },
   "callback": {
     "buttons": {
       "close": "Fenster schließen"
@@ -33,8 +44,10 @@
     "stateMissing": "Autorisierungsstatus nicht gefunden, bitte erneut versuchen."
   },
   "messages": {
+    "authorized": "LobeHub-Dienst erfolgreich autorisiert",
     "loading": "Autorisierungsvorgang wird gestartet...",
     "success": {
+      "cloudMcpInstall": "Autorisierung erfolgreich! Sie können jetzt das Cloud MCP-Plugin installieren.",
       "submit": "Autorisierung erfolgreich! Du kannst jetzt einen Assistenten veröffentlichen.",
       "upload": "Autorisierung erfolgreich! Du kannst jetzt eine neue Version veröffentlichen."
     }

package/locales/en-US/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Cancel",
+    "confirm": "Authorize",
+    "description": {
+      "and": "and",
+      "prefix": "By clicking Authorize, you agree to",
+      "privacy": "Privacy Policy",
+      "terms": "Terms of Service"
+    },
+    "title": "Confirm Authorization"
+  },
   "callback": {
     "buttons": {
       "close": "Close Window"
@@ -33,8 +44,10 @@
     "stateMissing": "Authorization state not found. Please try again."
   },
   "messages": {
+    "authorized": "LobeHub service authorized successfully",
     "loading": "Starting authorization process...",
     "success": {
+      "cloudMcpInstall": "Authorization successful! You can now install the Cloud MCP plugin.",
       "submit": "Authorization successful! You can now publish your assistant.",
       "upload": "Authorization successful! You can now publish a new version."
     }

package/locales/es-ES/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Cancelar",
+    "confirm": "Autorizar uso",
+    "description": {
+      "and": "y",
+      "prefix": "Al hacer clic en autorizar uso, se considera que aceptas",
+      "privacy": "la Política de Privacidad",
+      "terms": "los Términos de Servicio"
+    },
+    "title": "Confirmar autorización"
+  },
   "callback": {
     "buttons": {
       "close": "Cerrar ventana"
@@ -33,8 +44,10 @@
     "stateMissing": "No se encontró el estado de autenticación, por favor inténtalo de nuevo."
   },
   "messages": {
+    "authorized": "Autorización del servicio LobeHub realizada con éxito",
     "loading": "Iniciando el proceso de autenticación...",
     "success": {
+      "cloudMcpInstall": "¡Autorización exitosa! Ahora puedes instalar el complemento Cloud MCP.",
       "submit": "¡Autenticación exitosa! Ahora puedes publicar asistentes.",
       "upload": "¡Autenticación exitosa! Ahora puedes publicar una nueva versión."
     }

package/locales/fa-IR/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "لغو",
+    "confirm": "تأیید مجوز استفاده",
+    "description": {
+      "and": "و",
+      "prefix": "با کلیک بر روی تأیید مجوز استفاده، شما با",
+      "privacy": "سیاست حفظ حریم خصوصی",
+      "terms": "شرایط خدمات"
+    },
+    "title": "تأیید مجوز"
+  },
   "callback": {
     "buttons": {
       "close": "بستن پنجره"
@@ -33,8 +44,10 @@
     "stateMissing": "وضعیت احراز هویت یافت نشد، لطفاً دوباره تلاش کنید."
   },
   "messages": {
+    "authorized": "دسترسی به سرویس LobeHub با موفقیت تأیید شد",
     "loading": "در حال آغاز فرآیند احراز هویت...",
     "success": {
+      "cloudMcpInstall": "مجوز با موفقیت صادر شد! اکنون می‌توانید افزونه Cloud MCP را نصب کنید.",
       "submit": "احراز هویت با موفقیت انجام شد! اکنون می‌توانید دستیار را منتشر کنید.",
       "upload": "احراز هویت با موفقیت انجام شد! اکنون می‌توانید نسخه جدیدی منتشر کنید."
     }

package/locales/fr-FR/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Annuler",
+    "confirm": "Autoriser l'utilisation",
+    "description": {
+      "and": "et",
+      "prefix": "En cliquant sur « Autoriser l'utilisation », vous acceptez",
+      "privacy": "la Politique de confidentialité",
+      "terms": "les Conditions d'utilisation"
+    },
+    "title": "Confirmer l'autorisation"
+  },
   "callback": {
     "buttons": {
       "close": "Fermer la fenêtre"
@@ -33,8 +44,10 @@
     "stateMissing": "État d'autorisation introuvable, veuillez réessayer."
   },
   "messages": {
+    "authorized": "Autorisation du service LobeHub réussie",
     "loading": "Lancement du processus d'autorisation...",
     "success": {
+      "cloudMcpInstall": "Autorisation réussie ! Vous pouvez maintenant installer le plugin Cloud MCP.",
       "submit": "Autorisation réussie ! Vous pouvez maintenant publier un assistant.",
       "upload": "Autorisation réussie ! Vous pouvez maintenant publier une nouvelle version."
     }

package/locales/it-IT/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Annulla",
+    "confirm": "Autorizza l'utilizzo",
+    "description": {
+      "and": "e",
+      "prefix": "Facendo clic su 'Autorizza l'utilizzo' si accetta",
+      "privacy": "l'Informativa sulla privacy",
+      "terms": "i Termini di servizio"
+    },
+    "title": "Conferma autorizzazione"
+  },
   "callback": {
     "buttons": {
       "close": "Chiudi finestra"
@@ -33,8 +44,10 @@
     "stateMissing": "Stato di autorizzazione non trovato, riprova."
   },
   "messages": {
+    "authorized": "Autorizzazione al servizio LobeHub avvenuta con successo",
     "loading": "Avvio del processo di autorizzazione in corso...",
     "success": {
+      "cloudMcpInstall": "Autorizzazione avvenuta con successo! Ora puoi installare il plugin Cloud MCP.",
       "submit": "Autorizzazione riuscita! Ora puoi pubblicare l'assistente.",
       "upload": "Autorizzazione riuscita! Ora puoi pubblicare una nuova versione."
     }

package/locales/ja-JP/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "キャンセル",
+    "confirm": "使用を許可する",
+    "description": {
+      "and": "および",
+      "prefix": "「使用を許可する」をクリックすると、以下に同意したものとみなされます：",
+      "privacy": "プライバシーポリシー",
+      "terms": "利用規約"
+    },
+    "title": "認証の確認"
+  },
   "callback": {
     "buttons": {
       "close": "ウィンドウを閉じる"
@@ -33,8 +44,10 @@
     "stateMissing": "認証状態が見つかりません。もう一度お試しください。"
   },
   "messages": {
+    "authorized": "LobeHub サービスの認証に成功しました",
     "loading": "認証プロセスを開始しています...",
     "success": {
+      "cloudMcpInstall": "許可が成功しました！これで Cloud MCP プラグインをインストールできます。",
       "submit": "認証に成功しました！アシスタントを公開できます。",
       "upload": "認証に成功しました！新しいバージョンを公開できます。"
     }

package/locales/ko-KR/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "취소",
+    "confirm": "사용 권한 부여",
+    "description": {
+      "and": "및",
+      "prefix": "사용 권한 부여를 클릭하면 다음에 동의한 것으로 간주됩니다",
+      "privacy": "개인정보 처리방침",
+      "terms": "서비스 이용약관"
+    },
+    "title": "권한 확인"
+  },
   "callback": {
     "buttons": {
       "close": "창 닫기"
@@ -33,8 +44,10 @@
     "stateMissing": "인증 상태를 찾을 수 없습니다. 다시 시도해 주세요."
   },
   "messages": {
+    "authorized": "LobeHub 서비스가 성공적으로 인증되었습니다",
     "loading": "인증 절차를 시작하는 중입니다...",
     "success": {
+      "cloudMcpInstall": "권한 부여 완료! 이제 Cloud MCP 플러그인을 설치할 수 있습니다.",
       "submit": "인증 성공! 이제 어시스턴트를 게시할 수 있습니다.",
       "upload": "인증 성공! 이제 새 버전을 게시할 수 있습니다."
     }

package/locales/nl-NL/marketAuth.json

@@ -1,4 +1,15 @@
 {
+  "authorize": {
+    "cancel": "Annuleren",
+    "confirm": "Toestemming geven",
+    "description": {
+      "and": "en",
+      "prefix": "Door op 'Toestemming geven' te klikken, ga je akkoord met onze",
+      "privacy": "Privacybeleid",
+      "terms": "Servicevoorwaarden"
+    },
+    "title": "Autorisatie bevestigen"
+  },
   "callback": {
     "buttons": {
       "close": "Venster sluiten"
@@ -33,8 +44,10 @@
     "stateMissing": "Autorisatiestatus niet gevonden, probeer het opnieuw."
   },
   "messages": {
+    "authorized": "LobeHub-service succesvol geautoriseerd",
     "loading": "Autorisatieproces wordt gestart...",
     "success": {
+      "cloudMcpInstall": "Toestemming succesvol! Je kunt nu de Cloud MCP-plugin installeren.",
       "submit": "Autorisatie geslaagd! Je kunt nu een assistent publiceren.",
       "upload": "Autorisatie geslaagd! Je kunt nu een nieuwe versie publiceren."
     }