fix stdin issue with API key

Author: kapravel

.gitignore

@@ -1,2 +1,4 @@
 node_modules/
-*.log
+*.log
+.vscode/
+build/index.js.map

build/index.js

@@ -40,9 +40,8 @@ const logger = pino({
 
 const SOCKET_API_URL = "https://api.socket.dev/v0/purl?alerts=false&compact=false&fixable=false&licenseattrib=false&licensedetails=false";
 
-let SOCKET_API_KEY = process.env.SOCKET_API_KEY || "";
-if (!SOCKET_API_KEY) {
-    logger.error("SOCKET_API_KEY environment variable is not set");
+// Function to get API key interactively (only for HTTP mode)
+async function getApiKeyInteractively(): Promise<string> {
     const rl = readline.createInterface({
         input: process.stdin,
         output: process.stderr
@@ -60,15 +59,11 @@ if (!SOCKET_API_KEY) {
         process.exit(1);
     }
 
-    SOCKET_API_KEY = apiKey;
+    return apiKey;
 }
 
-const SOCKET_HEADERS = {
-  "user-agent": `socket-mcp/${VERSION}`,  
-  "accept": "application/x-ndjson",
-  "content-type": "application/json",
-  "authorization": `Bearer ${SOCKET_API_KEY}`
-};
+// Initialize API key
+let SOCKET_API_KEY = process.env.SOCKET_API_KEY || "";
 
 // Transport management
 const transports: Record<string, StreamableHTTPServerTransport> = {};
@@ -164,10 +159,8 @@ server.tool(
                                 .map(([key, value]) => `${key}: ${value}`)
                                 .join(', ');
 
-                            const packageName = jsonData.name || 'unknown';
                             results.push(`${purl}: ${scoreEntries}`);
                         } else {
-                            const packageName = jsonData.name || 'unknown';
                             results.push(`${purl}: No score found`);
                         }
                     }
@@ -180,7 +173,6 @@ server.tool(
                             .map(([key, value]) => `${key}: ${value}`)
                             .join(', ');
 
-                        const packageName = jsonData.package?.name || 'unknown';
                         results.push(`${purl}: ${scoreEntries}`);
                     }
                 }
@@ -221,6 +213,28 @@ server.tool(
 const useHttp = process.env.MCP_HTTP_MODE === 'true' || process.argv.includes('--http');
 const port = parseInt(process.env.MCP_PORT || '3000', 10);
 
+// Validate API key - in stdio mode, we can't prompt interactively
+if (!SOCKET_API_KEY) {
+    if (useHttp) {
+        // In HTTP mode, we can prompt for the API key
+        logger.error("SOCKET_API_KEY environment variable is not set");
+        SOCKET_API_KEY = await getApiKeyInteractively();
+    } else {
+        // In stdio mode, we must have the API key as an environment variable
+        logger.error("SOCKET_API_KEY environment variable is required in stdio mode");
+        logger.error("Please set the SOCKET_API_KEY environment variable and try again");
+        process.exit(1);
+    }
+}
+
+// Now that we have the API key, set up the headers
+const SOCKET_HEADERS = {
+  "user-agent": `socket-mcp/${VERSION}`,  
+  "accept": "application/x-ndjson",
+  "content-type": "application/json",
+  "authorization": `Bearer ${SOCKET_API_KEY}`
+};
+
 if (useHttp) {
   // HTTP mode with Server-Sent Events
   logger.info(`Starting HTTP server on port ${port}`);

src/index.ts

@@ -8,6 +8,7 @@
     "strict": true,
     "esModuleInterop": true,
     "skipLibCheck": true,
+    "sourceMap": true,
     "forceConsistentCasingInFileNames": true
   },
   "include": ["src/**/*"],