Add HTTP headers support and batch update for routing configuration (#151)

src/controllers/serverController.ts

@@ -88,6 +88,24 @@ export const createServer = async (req: Request, res: Response): Promise<void> =
       return;
     }
 
+    // Validate headers if provided
+    if (config.headers && typeof config.headers !== 'object') {
+      res.status(400).json({
+        success: false,
+        message: 'Headers must be an object',
+      });
+      return;
+    }
+
+    // Validate that headers are only used with sse and streamable-http types
+    if (config.headers && config.type === 'stdio') {
+      res.status(400).json({
+        success: false,
+        message: 'Headers are not supported for stdio server type',
+      });
+      return;
+    }
+
     const result = await addServer(name, config);
     if (result.success) {
       notifyToolChanged();
@@ -187,6 +205,24 @@ export const updateServer = async (req: Request, res: Response): Promise<void> =
       return;
     }
 
+    // Validate headers if provided
+    if (config.headers && typeof config.headers !== 'object') {
+      res.status(400).json({
+        success: false,
+        message: 'Headers must be an object',
+      });
+      return;
+    }
+
+    // Validate that headers are only used with sse and streamable-http types
+    if (config.headers && config.type === 'stdio') {
+      res.status(400).json({
+        success: false,
+        message: 'Headers are not supported for stdio server type',
+      });
+      return;
+    }
+
     const result = await updateMcpServer(name, config);
     if (result.success) {
       notifyToolChanged();

src/services/mcpService.ts

@@ -89,11 +89,27 @@ export const initializeClientsFromSettings = (isInit: boolean): ServerInfo[] =>
 
     let transport;
     if (conf.type === 'streamable-http') {
-      transport = new StreamableHTTPClientTransport(new URL(conf.url || ''));
+      const options: any = {};
+      if (conf.headers && Object.keys(conf.headers).length > 0) {
+        options.requestInit = {
+          headers: conf.headers,
+        };
+      }
+      transport = new StreamableHTTPClientTransport(new URL(conf.url || ''), options);
     } else if (conf.url) {
       // Default to SSE only when 'conf.type' is not specified and 'conf.url' is available
-      transport = new SSEClientTransport(new URL(conf.url));
-    } else if (conf.command && conf.args) {      // If type is stdio or if command and args are provided without type
+      const options: any = {};
+      if (conf.headers && Object.keys(conf.headers).length > 0) {
+        options.eventSourceInit = {
+          headers: conf.headers,
+        };
+        options.requestInit = {
+          headers: conf.headers,
+        };
+      }
+      transport = new SSEClientTransport(new URL(conf.url), options);
+    } else if (conf.command && conf.args) {
+      // If type is stdio or if command and args are provided without type
       const env: Record<string, string> = {
         ...(process.env as Record<string, string>), // Inherit all environment variables from parent process
         ...replaceEnvVars(conf.env || {}), // Override with configured env vars

src/types/index.ts

@@ -108,6 +108,7 @@ export interface ServerConfig {
   command?: string; // Command to execute for stdio-based servers
   args?: string[]; // Arguments for the command
   env?: Record<string, string>; // Environment variables
+  headers?: Record<string, string>; // HTTP headers for SSE/streamable-http servers
   enabled?: boolean; // Flag to enable/disable the server
 }