memshell-party-cli 0.2.1 → 0.2.2

package/README.md

@@ -246,7 +246,7 @@ Example Claude Code / Claude Desktop config:
 ```
 
 Exposed tools: `list_servers`, `list_config`, `list_packers`, `list_command_configs`,
-`generate_memshell`, `generate_probe`, `parse_classname`, `server_version`.
+`generate_memshell`, `generate_probe`, `parse_classname`, `server_version`, `connect_test`.
 
 ## AI skill (Claude Code)
 

package/README.zh-CN.md

@@ -243,7 +243,7 @@ Claude Code / Claude Desktop 配置示例：
 ```
 
 暴露的工具：`list_servers`、`list_config`、`list_packers`、`list_command_configs`、
-`generate_memshell`、`generate_probe`、`parse_classname`、`server_version`。
+`generate_memshell`、`generate_probe`、`parse_classname`、`server_version`、`connect_test`。
 
 ## AI 技能（Claude Code）
 

package/dist/cli.js

@@ -1380,6 +1380,17 @@ var probeInput = {
   reqParamName: z.string().optional(),
   commandTemplate: z.string().optional()
 };
+var connectInput = {
+  url: z.string().describe("URL of the deployed shell"),
+  tool: z.enum(["godzilla", "behinder", "suo5"]).describe("shell tool to test"),
+  pass: z.string().optional().describe("password (godzilla default: pass; behinder default: rebeyond)"),
+  key: z.string().optional().describe("godzilla key (default: key)"),
+  headerName: z.string().optional().describe("gate header name from generate_memshell shellToolConfig (usually User-Agent)"),
+  headerValue: z.string().optional().describe("gate header value from generate_memshell shellToolConfig \u2014 required for MemShellParty shells"),
+  suo5Mode: z.enum(["auto", "v2", "v1"]).optional().describe("suo5 protocol variant (default: auto)"),
+  insecure: z.boolean().optional().describe("skip TLS certificate verification"),
+  timeoutMs: z.number().optional().describe("request timeout in milliseconds (default 30000)")
+};
 function createMcpServer(client) {
   const server = new McpServer({ name: "memshell-party", version: CLI_VERSION });
   server.registerTool(
@@ -1482,6 +1493,39 @@ function createMcpServer(client) {
       }
     }
   );
+  server.registerTool(
+    "connect_test",
+    {
+      title: "Test shell connection",
+      description: "Test whether a deployed Godzilla / Behinder / suo5 shell is alive and the credentials work, by performing the tool's real protocol handshake. Pass the gate header (headerName/headerValue) from generate_memshell output for MemShellParty shells.",
+      inputSchema: connectInput
+    },
+    async (args) => {
+      try {
+        const common = {
+          headerName: args.headerName,
+          headerValue: args.headerValue,
+          timeoutMs: args.timeoutMs,
+          insecure: args.insecure
+        };
+        let result;
+        switch (args.tool) {
+          case "godzilla":
+            result = await testGodzilla(args.url, args.pass ?? "pass", args.key ?? "key", common);
+            break;
+          case "behinder":
+            result = await testBehinder(args.url, args.pass ?? "rebeyond", common);
+            break;
+          case "suo5":
+            result = await testSuo5(args.url, { ...common, mode: args.suo5Mode });
+            break;
+        }
+        return jsonContent(result);
+      } catch (err) {
+        return errorContent(err);
+      }
+    }
+  );
   return server;
 }
 async function startMcpStdio(client) {