@reconcrap/boss-recommend-mcp 1.2.2 → 1.2.4

package/src/index.js

@@ -4,7 +4,11 @@ import { spawn } from "node:child_process";
 import { createRequire } from "node:module";
 import process from "node:process";
 import { fileURLToPath } from "node:url";
-import { runRecommendPipeline } from "./pipeline.js";
+import {
+  getFeaturedCalibrationResolution,
+  runRecommendCalibration
+} from "./adapters.js";
+import { runRecommendPipeline } from "./pipeline.js";
 import {
   RUN_MODE_ASYNC,
   RUN_STAGE_PREFLIGHT,
@@ -28,11 +32,13 @@ import {
 const require = createRequire(import.meta.url);
 const { version: SERVER_VERSION } = require("../package.json");
 
-const TOOL_START_RUN = "start_recommend_pipeline_run";
-const TOOL_GET_RUN = "get_recommend_pipeline_run";
-const TOOL_CANCEL_RUN = "cancel_recommend_pipeline_run";
-const TOOL_PAUSE_RUN = "pause_recommend_pipeline_run";
-const TOOL_RESUME_RUN = "resume_recommend_pipeline_run";
+const TOOL_START_RUN = "start_recommend_pipeline_run";
+const TOOL_GET_RUN = "get_recommend_pipeline_run";
+const TOOL_CANCEL_RUN = "cancel_recommend_pipeline_run";
+const TOOL_PAUSE_RUN = "pause_recommend_pipeline_run";
+const TOOL_RESUME_RUN = "resume_recommend_pipeline_run";
+const TOOL_RUN_FEATURED_CALIBRATION = "run_featured_calibration";
+const TOOL_GET_FEATURED_CALIBRATION_STATUS = "get_featured_calibration_status";
 
 const SERVER_NAME = "boss-recommend-mcp";
 const FRAMING_UNKNOWN = "unknown";
@@ -141,7 +147,7 @@ function createJsonRpcError(id, code, message) {
   };
 }
 
-function createRunInputSchema() {
+function createRunInputSchema() {
   return {
     type: "object",
     properties: {
@@ -290,10 +296,33 @@ function createRunInputSchema() {
     required: ["instruction"],
     additionalProperties: false
   };
-}
-
-function createToolsSchema() {
-  return [
+}
+
+function createRunFeaturedCalibrationInputSchema() {
+  return {
+    type: "object",
+    properties: {
+      port: {
+        type: "integer",
+        minimum: 1,
+        description: "可选，Boss Chrome 远程调试端口（默认读取配置或 9222）"
+      },
+      timeout_ms: {
+        type: "integer",
+        minimum: 1000,
+        description: "可选，等待收藏点击的超时时间（毫秒）"
+      },
+      output: {
+        type: "string",
+        description: "可选，校准文件输出路径（默认 favorite-calibration.json）"
+      }
+    },
+    additionalProperties: false
+  };
+}
+
+function createToolsSchema() {
+  return [
     {
       name: TOOL_START_RUN,
       description: "异步启动 Boss 推荐页流水线（含同步门禁预检）；只有在前置确认与页面就绪通过后才返回 run_id。",
@@ -335,20 +364,34 @@ function createToolsSchema() {
         additionalProperties: false
       }
     },
-    {
-      name: TOOL_RESUME_RUN,
-      description: "继续指定 run_id 的 paused 流水线；沿用原 CSV 与 checkpoint 续跑。",
-      inputSchema: {
+    {
+      name: TOOL_RESUME_RUN,
+      description: "继续指定 run_id 的 paused 流水线；沿用原 CSV 与 checkpoint 续跑。",
+      inputSchema: {
         type: "object",
         properties: {
           run_id: { type: "string" }
         },
-        required: ["run_id"],
-        additionalProperties: false
-      }
-    }
-  ];
-}
+        required: ["run_id"],
+        additionalProperties: false
+      }
+    },
+    {
+      name: TOOL_RUN_FEATURED_CALIBRATION,
+      description: "手动执行精选页收藏按钮校准。执行前请先在 Boss 推荐页切换到精选 tab 并打开任意候选人详情页。",
+      inputSchema: createRunFeaturedCalibrationInputSchema()
+    },
+    {
+      name: TOOL_GET_FEATURED_CALIBRATION_STATUS,
+      description: "查询精选页收藏校准文件与校准脚本可用性。",
+      inputSchema: {
+        type: "object",
+        properties: {},
+        additionalProperties: false
+      }
+    }
+  ];
+}
 
 function createToolResultResponse(id, payload, isError = false) {
   return {
@@ -367,7 +410,7 @@ function createToolResultResponse(id, payload, isError = false) {
   };
 }
 
-function validateRunArgs(args) {
+function validateRunArgs(args) {
   if (!args || typeof args !== "object") {
     return "arguments must be an object";
   }
@@ -375,9 +418,37 @@ function validateRunArgs(args) {
     return "instruction is required and must be a string";
   }
   return null;
-}
-
-function getLastOutputLine(text) {
+}
+
+function validateRunFeaturedCalibrationArgs(args) {
+  if (!args || typeof args !== "object" || Array.isArray(args)) {
+    return "arguments must be an object";
+  }
+
+  if (Object.prototype.hasOwnProperty.call(args, "port")) {
+    const port = Number.parseInt(String(args.port), 10);
+    if (!Number.isFinite(port) || port <= 0) {
+      return "port must be a positive integer";
+    }
+  }
+
+  if (Object.prototype.hasOwnProperty.call(args, "timeout_ms")) {
+    const timeoutMs = Number.parseInt(String(args.timeout_ms), 10);
+    if (!Number.isFinite(timeoutMs) || timeoutMs < 1000) {
+      return "timeout_ms must be an integer >= 1000";
+    }
+  }
+
+  if (Object.prototype.hasOwnProperty.call(args, "output")) {
+    if (typeof args.output !== "string" || !normalizeText(args.output)) {
+      return "output must be a non-empty string when provided";
+    }
+  }
+
+  return null;
+}
+
+function getLastOutputLine(text) {
   const lines = String(text || "")
     .split(/\r?\n/)
     .map((line) => normalizeText(line))
@@ -1219,8 +1290,57 @@ function handleResumeRunTool(args) {
     message: "已恢复 Recommend 流水线（detached）。默认不自动轮询；如需进度请按需调用 get_recommend_pipeline_run。"
   };
 }
-
-async function handleRequest(message, workspaceRoot) {
+
+function handleGetFeaturedCalibrationStatusTool(workspaceRoot) {
+  const resolution = getFeaturedCalibrationResolution(workspaceRoot);
+  return {
+    status: "CALIBRATION_STATUS",
+    ready: resolution.calibration_usable === true,
+    calibration_path: resolution.calibration_path,
+    calibration_exists: resolution.calibration_exists,
+    calibration_usable: resolution.calibration_usable,
+    calibration_script_path: resolution.calibration_script_path,
+    message: resolution.calibration_usable
+      ? "精选页收藏校准文件可用。"
+      : "精选页收藏校准文件不存在或无效。"
+  };
+}
+
+async function handleRunFeaturedCalibrationTool({ workspaceRoot, args }) {
+  const result = await runRecommendCalibration(workspaceRoot, {
+    port: args.port,
+    timeoutMs: args.timeout_ms,
+    output: args.output
+  });
+
+  if (!result?.ok) {
+    return {
+      status: "FAILED",
+      error: {
+        code: result?.error?.code || "CALIBRATION_REQUIRED",
+        message: result?.error?.message || "精选页收藏校准失败，请在推荐页精选 tab 打开候选人详情后点击收藏按钮再重试。",
+        retryable: true
+      },
+      calibration_path: result?.calibration_path || null,
+      calibration_script_path: result?.calibration_script_path || null,
+      debug_port: result?.debug_port || null,
+      diagnostics: {
+        stdout_last_line: getLastOutputLine(result?.stdout),
+        stderr_last_line: getLastOutputLine(result?.stderr)
+      }
+    };
+  }
+
+  return {
+    status: "CALIBRATED",
+    message: "精选页收藏按钮校准完成，可重新执行 start_recommend_pipeline_run。",
+    calibration_path: result.calibration_path,
+    calibration_script_path: result.calibration_script_path,
+    debug_port: result.debug_port
+  };
+}
+
+async function handleRequest(message, workspaceRoot) {
   if (!message || message.jsonrpc !== "2.0") {
     return createJsonRpcError(null, -32600, "Invalid JSON-RPC request");
   }
@@ -1262,34 +1382,45 @@ async function handleRequest(message, workspaceRoot) {
     const toolName = params?.name;
     const args = params?.arguments || {};
 
-    if (toolName === TOOL_START_RUN) {
-      const inputError = validateRunArgs(args);
-      if (inputError) {
-        return createJsonRpcError(id, -32602, inputError);
-      }
-    }
-
-    if ([TOOL_GET_RUN, TOOL_CANCEL_RUN, TOOL_PAUSE_RUN, TOOL_RESUME_RUN].includes(toolName)) {
-      if (!args || typeof args.run_id !== "string" || !normalizeText(args.run_id)) {
-        return createJsonRpcError(id, -32602, "run_id is required and must be a string");
-      }
+    if (toolName === TOOL_START_RUN) {
+      const inputError = validateRunArgs(args);
+      if (inputError) {
+        return createJsonRpcError(id, -32602, inputError);
+      }
+    }
+
+    if (toolName === TOOL_RUN_FEATURED_CALIBRATION) {
+      const inputError = validateRunFeaturedCalibrationArgs(args);
+      if (inputError) {
+        return createJsonRpcError(id, -32602, inputError);
+      }
+    }
+
+    if ([TOOL_GET_RUN, TOOL_CANCEL_RUN, TOOL_PAUSE_RUN, TOOL_RESUME_RUN].includes(toolName)) {
+      if (!args || typeof args.run_id !== "string" || !normalizeText(args.run_id)) {
+        return createJsonRpcError(id, -32602, "run_id is required and must be a string");
+      }
     }
 
     try {
       let payload;
-      if (toolName === TOOL_START_RUN) {
-        payload = await handleStartRunTool({ workspaceRoot, args });
-      } else if (toolName === TOOL_GET_RUN) {
-        payload = handleGetRunTool(args);
-      } else if (toolName === TOOL_CANCEL_RUN) {
-        payload = handleCancelRunTool(args);
-      } else if (toolName === TOOL_PAUSE_RUN) {
-        payload = handlePauseRunTool(args);
-      } else if (toolName === TOOL_RESUME_RUN) {
-        payload = handleResumeRunTool(args);
-      } else {
-        return createJsonRpcError(id, -32602, `Unknown tool: ${toolName || ""}`);
-      }
+      if (toolName === TOOL_START_RUN) {
+        payload = await handleStartRunTool({ workspaceRoot, args });
+      } else if (toolName === TOOL_GET_RUN) {
+        payload = handleGetRunTool(args);
+      } else if (toolName === TOOL_CANCEL_RUN) {
+        payload = handleCancelRunTool(args);
+      } else if (toolName === TOOL_PAUSE_RUN) {
+        payload = handlePauseRunTool(args);
+      } else if (toolName === TOOL_RESUME_RUN) {
+        payload = handleResumeRunTool(args);
+      } else if (toolName === TOOL_GET_FEATURED_CALIBRATION_STATUS) {
+        payload = handleGetFeaturedCalibrationStatusTool(workspaceRoot);
+      } else if (toolName === TOOL_RUN_FEATURED_CALIBRATION) {
+        payload = await handleRunFeaturedCalibrationTool({ workspaceRoot, args });
+      } else {
+        return createJsonRpcError(id, -32602, `Unknown tool: ${toolName || ""}`);
+      }
       const isError = payload?.status === "FAILED";
       return createToolResultResponse(id, payload, isError);
     } catch (error) {