Merge pull request #92 from levante-hub/fix/runtime-implementation

fix: runtime installation improvements and auto-install for MCP servers

Author: olivermontes

package.json

@@ -110,6 +110,7 @@
     "update-electron-app": "^3.1.1",
     "use-stick-to-bottom": "^1.1.1",
     "vaul": "^1.1.2",
+    "winston": "^3.18.3",
     "zod": "^4.1.12",
     "zustand": "^5.0.8"
   },

pnpm-lock.yaml

@@ -43,7 +43,16 @@ export function registerConfigurationHandlers(
           results[serverId] = { success: true };
           logger.mcp.info("Successfully reconnected MCP server", { serverId });
         } catch (error: any) {
-          results[serverId] = { success: false, error: error.message };
+          // Handle runtime errors gracefully during batch refresh
+          if (error.message === 'RUNTIME_NOT_FOUND') {
+            results[serverId] = {
+              success: false,
+              error: "Runtime not available. Please check your connection and try again."
+            };
+          } else {
+            results[serverId] = { success: false, error: error.message };
+          }
+
           logger.mcp.error("Failed to reconnect MCP server", {
             serverId,
             error: error.message,

src/main/ipc/mcpHandlers/configuration.ts

@@ -21,11 +21,11 @@ let mcpService: IMCPService;
 const configManager = new MCPConfigurationManager();
 const logger = getLogger();
 
-export function registerMCPHandlers() {
+export async function registerMCPHandlers() {
   try {
     // Create MCP service based on user preferences
     const uiPreferences = preferencesService.getAll();
-    mcpService = MCPServiceFactory.createFromUIPreferences(uiPreferences);
+    mcpService = await MCPServiceFactory.createFromUIPreferences(uiPreferences);
 
     logger.mcp.info("MCP service created via factory", {
       sdk: uiPreferences.mcp?.sdk || "mcp-use",

src/main/ipc/mcpHandlers/index.ts

@@ -4,16 +4,6 @@ import { getLogger } from "../../services/logging";
 const logger = getLogger();
 
 export function registerRegistryHandlers(mcpService: any, configManager: any) {
-  // Diagnose system for MCP compatibility
-  ipcMain.handle("levante/mcp/diagnose-system", async () => {
-    try {
-      const diagnosis = await mcpService.diagnoseSystem();
-      return { success: true, data: diagnosis };
-    } catch (error: any) {
-      return { success: false, error: error.message };
-    }
-  });
-
   // Get MCP registry information
   ipcMain.handle("levante/mcp/get-registry", async () => {
     try {

src/main/ipc/mcpHandlers/registry.ts

@@ -109,7 +109,7 @@ export async function initializeServices(): Promise<void> {
  * Should be called after service initialization
  * @param getMainWindow - Function to get current main window reference
  */
-export function registerIPCHandlers(getMainWindow: () => BrowserWindow | null): void {
+export async function registerIPCHandlers(getMainWindow: () => BrowserWindow | null): Promise<void> {
   // Service-specific handlers
   setupDatabaseHandlers();
   setupPreferencesHandlers();
@@ -119,7 +119,7 @@ export function registerIPCHandlers(getMainWindow: () => BrowserWindow | null):
   setupLoggerHandlers();
   setupWizardHandlers();
   setupProfileHandlers();
-  registerMCPHandlers();
+  await registerMCPHandlers();
   registerDebugHandlers();
   registerAnalyticsHandlers();
 

src/main/lifecycle/initialization.ts

@@ -46,28 +46,39 @@ let mainWindow: BrowserWindow | null = null;
 
 // App ready event
 app.whenReady().then(async () => {
-  // Initialize all services
-  await initializeServices();
+  try {
+    // Initialize all services
+    await initializeServices();
 
-  // Register all IPC handlers
-  registerIPCHandlers(() => mainWindow);
+    // Register all IPC handlers
+    await registerIPCHandlers(() => mainWindow);
 
-  // Create main window
-  mainWindow = createMainWindow();
+    // Create main window
+    mainWindow = createMainWindow();
 
-  // Track app open (fire and forget, don't block UI)
-  analyticsService.trackAppOpen().catch(() => { });
+    // Track app open (fire and forget, don't block UI)
+    analyticsService.trackAppOpen().catch(() => { });
 
-  // Create application menu
-  createApplicationMenu(mainWindow);
+    // Create application menu
+    createApplicationMenu(mainWindow);
 
-  // Register main window with services
-  deepLinkService.setMainWindow(mainWindow);
-  oauthCallbackServer.setMainWindow(mainWindow);
+    // Register main window with services
+    deepLinkService.setMainWindow(mainWindow);
+    oauthCallbackServer.setMainWindow(mainWindow);
 
-  // Register app event handlers
-  registerAppEvents(() => mainWindow);
+    // Register app event handlers
+    registerAppEvents(() => mainWindow);
 
-  // Setup deep link handling (Windows/Linux)
-  setupDeepLinkHandling();
+    // Setup deep link handling (Windows/Linux)
+    setupDeepLinkHandling();
+  } catch (error) {
+    console.error('Fatal error during app initialization:', error);
+    // Show error dialog and quit
+    const { dialog } = await import('electron');
+    dialog.showErrorBox(
+      'Initialization Error',
+      `Failed to start the application:\n\n${error instanceof Error ? error.message : String(error)}`
+    );
+    app.quit();
+  }
 });

src/main/main.ts

@@ -17,6 +17,12 @@ import type { MCPRegistry } from "./types.js";
  * must implement, allowing them to be used interchangeably via the factory pattern.
  */
 export interface IMCPService {
+  /**
+   * Initialize the MCP service. Must be called before using other methods.
+   * This allows for async initialization like configuring loggers.
+   */
+  initialize(): Promise<void>;
+
   /**
    * Connect to an MCP server with the given configuration.
    * @param config - Server configuration including transport details
@@ -98,16 +104,6 @@ export interface IMCPService {
     alternative?: string;
   }>;
 
-  /**
-   * Run system diagnostics for MCP compatibility.
-   * @returns Diagnostic results with issues and recommendations
-   */
-  diagnoseSystem(): Promise<{
-    success: boolean;
-    issues: string[];
-    recommendations: string[];
-  }>;
-
   // ==========================================
   // MCP Resources methods
   // ==========================================

src/main/services/mcp/IMCPService.ts

@@ -7,7 +7,6 @@ import type {
 } from "../../types/mcp.js";
 import { getLogger } from "../logging";
 import { createTransport, handleConnectionError } from "./transports.js";
-import { diagnoseSystem } from "./diagnostics.js";
 import { loadMCPRegistry } from "./registry.js";
 import type { MCPRegistry } from "./types";
 import { RuntimeManager } from "../runtime/runtimeManager";
@@ -367,13 +366,4 @@ export class MCPService {
       };
     }
   }
-
-  // Diagnose system for MCP compatibility
-  async diagnoseSystem(): Promise<{
-    success: boolean;
-    issues: string[];
-    recommendations: string[];
-  }> {
-    return await diagnoseSystem();
-  }
 }