@solongate/proxy 0.30.1 → 0.31.0

package/dist/index.js

@@ -534,13 +534,16 @@ function isAlreadyProtected(server) {
   }
   return false;
 }
-function wrapServer(serverName, server, policy) {
+function wrapServer(serverName, server, policy, agentName) {
   const env = { ...server.env ?? {} };
   env.SOLONGATE_API_KEY = "${SOLONGATE_API_KEY}";
   const proxyArgs = ["-y", "@solongate/proxy@latest"];
   if (policy) {
     proxyArgs.push("--policy", policy);
   }
+  if (agentName) {
+    proxyArgs.push("--agent-name", agentName);
+  }
   proxyArgs.push("--verbose", "--", server.command, ...server.args ?? []);
   return {
     command: "npx",
@@ -814,8 +817,22 @@ function installCursorConfig(clientDir, guardCmd, auditCmd, _stopCmd, wrappedMcp
     changed = true;
   }
   if (wrappedMcpConfig) {
+    const cursorMcpConfig = { mcpServers: {} };
+    for (const [name, server] of Object.entries(wrappedMcpConfig.mcpServers)) {
+      const args = [...server.args ?? []];
+      const agentIdx = args.indexOf("--agent-name");
+      if (agentIdx >= 0 && agentIdx + 1 < args.length) {
+        args[agentIdx + 1] = "cursor";
+      } else {
+        const sepIdx = args.indexOf("--");
+        if (sepIdx >= 0) {
+          args.splice(sepIdx, 0, "--agent-name", "cursor");
+        }
+      }
+      cursorMcpConfig.mcpServers[name] = { ...server, args };
+    }
     const mcpPath = join(clientDir, "mcp.json");
-    const mcpStr = JSON.stringify(wrappedMcpConfig, null, 2) + "\n";
+    const mcpStr = JSON.stringify(cursorMcpConfig, null, 2) + "\n";
     const existingMcp = existsSync4(mcpPath) ? readFileSync4(mcpPath, "utf-8") : "";
     if (mcpStr !== existingMcp) {
       writeFileSync2(mcpPath, mcpStr);
@@ -846,7 +863,19 @@ function installGeminiConfig(clientDir, guardCmd, auditCmd, _stopCmd, wrappedMcp
   }
   const merged = { ...existing };
   if (wrappedMcpConfig) {
-    merged.mcpServers = wrappedMcpConfig.mcpServers;
+    const geminiMcp = {};
+    for (const [name, server] of Object.entries(wrappedMcpConfig.mcpServers)) {
+      const args = [...server.args ?? []];
+      const agentIdx = args.indexOf("--agent-name");
+      if (agentIdx >= 0 && agentIdx + 1 < args.length) {
+        args[agentIdx + 1] = "gemini-cli";
+      } else {
+        const sepIdx = args.indexOf("--");
+        if (sepIdx >= 0) args.splice(sepIdx, 0, "--agent-name", "gemini-cli");
+      }
+      geminiMcp[name] = { ...server, args };
+    }
+    merged.mcpServers = geminiMcp;
   }
   merged.hooks = {
     BeforeTool: [
@@ -6220,11 +6249,13 @@ var SolonGateProxy = class {
       if (this.server) {
         const clientVersion = this.server.getClientVersion();
         log2(`MCP clientInfo raw: ${JSON.stringify(clientVersion)}`);
-        if (clientVersion?.name) {
+        if (clientVersion?.name && !this.config.agentName) {
           const normalized = this.normalizeAgentName(clientVersion.name);
           this.agentId = normalized.id;
           this.agentName = normalized.name;
           log2(`Agent identified from MCP clientInfo: ${this.agentName} (raw: ${clientVersion.name})`);
+        } else if (this.config.agentName) {
+          log2(`Agent identity from --agent-name flag: ${this.agentName} (clientInfo: ${clientVersion?.name || "none"})`);
         }
       }
     };

package/dist/init.js

@@ -118,13 +118,16 @@ function isAlreadyProtected(server) {
   }
   return false;
 }
-function wrapServer(serverName, server, policy) {
+function wrapServer(serverName, server, policy, agentName) {
   const env = { ...server.env ?? {} };
   env.SOLONGATE_API_KEY = "${SOLONGATE_API_KEY}";
   const proxyArgs = ["-y", "@solongate/proxy@latest"];
   if (policy) {
     proxyArgs.push("--policy", policy);
   }
+  if (agentName) {
+    proxyArgs.push("--agent-name", agentName);
+  }
   proxyArgs.push("--verbose", "--", server.command, ...server.args ?? []);
   return {
     command: "npx",
@@ -400,8 +403,22 @@ function installCursorConfig(clientDir, guardCmd, auditCmd, _stopCmd, wrappedMcp
     changed = true;
   }
   if (wrappedMcpConfig) {
+    const cursorMcpConfig = { mcpServers: {} };
+    for (const [name, server] of Object.entries(wrappedMcpConfig.mcpServers)) {
+      const args = [...server.args ?? []];
+      const agentIdx = args.indexOf("--agent-name");
+      if (agentIdx >= 0 && agentIdx + 1 < args.length) {
+        args[agentIdx + 1] = "cursor";
+      } else {
+        const sepIdx = args.indexOf("--");
+        if (sepIdx >= 0) {
+          args.splice(sepIdx, 0, "--agent-name", "cursor");
+        }
+      }
+      cursorMcpConfig.mcpServers[name] = { ...server, args };
+    }
     const mcpPath = join(clientDir, "mcp.json");
-    const mcpStr = JSON.stringify(wrappedMcpConfig, null, 2) + "\n";
+    const mcpStr = JSON.stringify(cursorMcpConfig, null, 2) + "\n";
     const existingMcp = existsSync(mcpPath) ? readFileSync(mcpPath, "utf-8") : "";
     if (mcpStr !== existingMcp) {
       writeFileSync(mcpPath, mcpStr);
@@ -432,7 +449,19 @@ function installGeminiConfig(clientDir, guardCmd, auditCmd, _stopCmd, wrappedMcp
   }
   const merged = { ...existing };
   if (wrappedMcpConfig) {
-    merged.mcpServers = wrappedMcpConfig.mcpServers;
+    const geminiMcp = {};
+    for (const [name, server] of Object.entries(wrappedMcpConfig.mcpServers)) {
+      const args = [...server.args ?? []];
+      const agentIdx = args.indexOf("--agent-name");
+      if (agentIdx >= 0 && agentIdx + 1 < args.length) {
+        args[agentIdx + 1] = "gemini-cli";
+      } else {
+        const sepIdx = args.indexOf("--");
+        if (sepIdx >= 0) args.splice(sepIdx, 0, "--agent-name", "gemini-cli");
+      }
+      geminiMcp[name] = { ...server, args };
+    }
+    merged.mcpServers = geminiMcp;
   }
   merged.hooks = {
     BeforeTool: [

package/dist/lib.js

@@ -4460,11 +4460,13 @@ var SolonGateProxy = class {
       if (this.server) {
         const clientVersion = this.server.getClientVersion();
         log2(`MCP clientInfo raw: ${JSON.stringify(clientVersion)}`);
-        if (clientVersion?.name) {
+        if (clientVersion?.name && !this.config.agentName) {
           const normalized = this.normalizeAgentName(clientVersion.name);
           this.agentId = normalized.id;
           this.agentName = normalized.name;
           log2(`Agent identified from MCP clientInfo: ${this.agentName} (raw: ${clientVersion.name})`);
+        } else if (this.config.agentName) {
+          log2(`Agent identity from --agent-name flag: ${this.agentName} (clientInfo: ${clientVersion?.name || "none"})`);
         }
       }
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@solongate/proxy",
-  "version": "0.30.1",
+  "version": "0.31.0",
   "description": "AI tool security proxy — protect any AI tool server with customizable policies, path/command constraints, rate limiting, and audit logging. Zero code changes required.",
   "type": "module",
   "bin": {