replace eventsource with reconnecting-eventsource to optimize the code.

(cherry picked from commit 505f86d)

Author: aheizi

package-lock.json

@@ -333,7 +333,6 @@
 		"delay": "^6.0.0",
 		"diff": "^5.2.0",
 		"diff-match-patch": "^1.0.5",
-		"eventsource": "^3.0.5",
 		"fast-deep-equal": "^3.1.3",
 		"fastest-levenshtein": "^1.0.16",
 		"get-folder-size": "^5.0.0",
@@ -351,6 +350,7 @@
 		"pretty-bytes": "^6.1.1",
 		"puppeteer-chromium-resolver": "^23.0.0",
 		"puppeteer-core": "^23.4.0",
+		"reconnecting-eventsource": "^1.6.4",
 		"serialize-error": "^11.0.3",
 		"simple-git": "^3.27.0",
 		"sound-play": "^1.1.0",

package.json

@@ -1,7 +1,7 @@
 import { Client } from "@modelcontextprotocol/sdk/client/index.js"
 import { StdioClientTransport, StdioServerParameters } from "@modelcontextprotocol/sdk/client/stdio.js"
 import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js"
-import { EventSource } from "eventsource"
+import ReconnectingEventSource from "reconnecting-eventsource"
 import {
 	CallToolResultSchema,
 	ListResourcesResultSchema,
@@ -243,8 +243,16 @@ export class McpHub {
 						headers: config.headers,
 					},
 				}
-				global.EventSource = EventSource
-				transport = new SSEClientTransport(new URL(config.url), sseOptions)
+				// Configure ReconnectingEventSource options
+				const reconnectingEventSourceOptions = {
+					max_retry_time: 5000, // Maximum retry time in milliseconds
+					withCredentials: config.headers?.["Authorization"] ? true : false, // Enable credentials if Authorization header exists
+				}
+				global.EventSource = ReconnectingEventSource
+				transport = new SSEClientTransport(new URL(config.url), {
+					...sseOptions,
+					eventSourceInit: reconnectingEventSourceOptions,
+				})
 
 				// Set up SSE specific error handling
 				transport.onerror = async (error) => {
@@ -255,11 +263,6 @@ export class McpHub {
 						this.appendErrorMessage(connection, error.message)
 					}
 					await this.notifyWebviewOfServerChanges()
-
-					// Attempt reconnection for SSE
-					if (!this.isDisposed) {
-						await this.handleSSEReconnection(name, config)
-					}
 				}
 			}
 
@@ -295,26 +298,6 @@ export class McpHub {
 		}
 	}
 
-	private async handleSSEReconnection(name: string, config: z.infer<typeof ServerConfigSchema>): Promise<void> {
-		const maxRetries = 3
-		let retryCount = 0
-
-		while (retryCount < maxRetries) {
-			try {
-				await delay(Math.pow(2, retryCount) * 1000) // Exponential backoff
-				await this.connectToServer(name, config)
-				return
-			} catch (error) {
-				retryCount++
-				console.error(`Retry ${retryCount} failed for ${name}:`, error)
-			}
-		}
-
-		vscode.window.showErrorMessage(
-			`Failed to reconnect to ${name} after ${maxRetries} attempts. Please check your connection.`,
-		)
-	}
-
 	private appendErrorMessage(connection: McpConnection, error: string) {
 		const newError = connection.server.error ? `${connection.server.error}\n${error}` : error
 		connection.server.error = newError //.slice(0, 800)