ai-project-manage-cli 3.0.8 → 3.0.10

package/dist/index.js

@@ -114,6 +114,10 @@ var requestConfig = {
     updateDevStatus: defineEndpoint({
       method: "POST",
       path: "/cli/requirements/update-dev-status"
+    }),
+    updateTaskStatus: defineEndpoint({
+      method: "POST",
+      path: "/cli/requirements/update-task-status"
     })
   },
   requirementArtifact: {
@@ -223,8 +227,10 @@ async function runComment(requirementId, file, model) {
 }
 
 // src/commands/connect.ts
+import { execSync } from "child_process";
 import { randomUUID } from "crypto";
 import WebSocket from "ws";
+import { Agent } from "@cursor/sdk";
 function runConnect(opts) {
   void (async () => {
     const cfg = await ensureApmConfig();
@@ -240,6 +246,7 @@ function runConnect(opts) {
       console.error(`[apm] \u914D\u7F6E\u6587\u4EF6\u8DEF\u5F84: ${APM_CONFIG_PATH}`);
       process.exit(1);
     }
+    const api = createApmApiClient(cfg);
     const url = buildAgentWsUrl(cfg.baseUrl, cfg.token);
     console.error(`[apm] \u8FDE\u63A5 ${url.replace(cfg.token, "<token>")} \u2026`);
     const ws = new WebSocket(url);
@@ -258,9 +265,88 @@ function runConnect(opts) {
       const interval = setInterval(sendHeartbeat, 15e3);
       ws.on("close", () => clearInterval(interval));
     });
-    ws.on("message", (data) => {
+    ws.on("message", async (data) => {
       const text = typeof data === "string" ? data : data.toString();
-      console.log(text);
+      try {
+        const msg = JSON.parse(text);
+        if (msg.type !== "JOB") {
+          return;
+        }
+        const payload = msg.payload;
+        try {
+          execSync(`apm pull ${payload.requirementId}`, {
+            cwd: payload.cwd,
+            encoding: "utf8"
+          });
+        } catch (pullErr) {
+          console.error("[apm] apm pull \u5931\u8D25:", pullErr);
+          throw pullErr;
+        }
+        const agent = await Agent.create({
+          apiKey: payload.apiKey,
+          model: { id: payload.model ?? "default" },
+          local: { cwd: payload.cwd }
+        });
+        await api.cliRequirements.updateDevStatus({
+          requirementId: payload.requirementId,
+          status: "WORKING"
+        });
+        const run = await agent.send(payload.prompt);
+        for await (const event of run.stream()) {
+          if (event.type === "system") {
+            console.log("[Ready]", JSON.stringify(event));
+            await api.cliRequirements.updateTaskStatus({
+              taskId: payload.taskId,
+              requirementId: payload.requirementId,
+              status: "IN_PROGRESS"
+            });
+          }
+          if (event.type === "user") {
+            console.log(
+              `[Input]`,
+              JSON.stringify(event.message.content[0].text)
+            );
+          }
+          if (event.type === "assistant") {
+            console.log(`[Output]`, JSON.stringify(event.message.content));
+          }
+          if (event.type === "thinking") {
+            console.log(`[Thinking]`, JSON.stringify(event.text));
+          }
+          if (event.type === "tool_call") {
+            if (event.result) {
+              console.log(
+                `[ToolCall(${event.call_id}) Result]`,
+                JSON.stringify(event.result)
+              );
+            }
+            console.log(
+              `[ToolCall(${event.call_id}) Args]`,
+              JSON.stringify(event.args)
+            );
+          }
+          if (event.type === "task") {
+            console.log(`[Task:${event.status}]`);
+            console.log("--------------------------------");
+            console.log(event.text || "\u65E0");
+            console.log("--------------------------------");
+          }
+          if (event.type === "request") {
+            console.log(JSON.stringify(event, null, 2));
+          }
+        }
+        await api.cliRequirements.updateTaskStatus({
+          taskId: payload.taskId,
+          requirementId: payload.requirementId,
+          status: "PENDING_ACCEPTANCE"
+        });
+        await api.cliRequirements.updateDevStatus({
+          requirementId: payload.requirementId,
+          status: "IDLE"
+        });
+      } catch {
+        console.error("[apm] \u65E0\u6CD5\u89E3\u6790 WebSocket \u6D88\u606F:", text);
+      }
     });
     ws.on("error", (err) => {
       console.error("[apm] WebSocket \u9519\u8BEF:", err.message);
@@ -516,6 +602,26 @@ function tasksForStatusYaml(tasks) {
     executor: t.executorType
   }));
 }
+function escapeForCdata(text) {
+  return text.replace(/\]\]>/g, "]]]]><![CDATA[>");
+}
+function defectsToXml(defects) {
+  const sorted = [...defects].sort(
+    (a, b) => new Date(a.createdAt ?? 0).getTime() - new Date(b.createdAt ?? 0).getTime()
+  );
+  const lines = ["<defects>"];
+  for (const d of sorted) {
+    lines.push(`  <defect id="${xmlEscape(d.id)}">`);
+    lines.push(`    <status>${xmlEscape(d.status)}</status>`);
+    lines.push(`    <current><![CDATA[${escapeForCdata(d.currentState ?? "")}]]></current>`);
+    lines.push(
+      `    <expected><![CDATA[${escapeForCdata(d.expectedEffect ?? "")}]]></expected>`
+    );
+    lines.push(`  </defect>`);
+  }
+  lines.push("</defects>", "");
+  return lines.join("\n");
+}
 async function runPull(requirementId) {
   const cfg = await ensureLoggedConfig();
   const api = createApmApiClient(cfg);
@@ -560,7 +666,7 @@ async function runPull(requirementId) {
     ""
   ].join("\n");
   writeFileSync3(join4(WORKITEMS_DIR, "reviews.xml"), reviewsXml, "utf8");
-  const defectsXml = unknownArrayToXml("defects", "defect", data.defects ?? []);
+  const defectsXml = defectsToXml(data.defects ?? []);
   writeFileSync3(join4(WORKITEMS_DIR, "defect.xml"), defectsXml, "utf8");
   const testCasesXml = unknownArrayToXml(
     "testcases",
@@ -1116,7 +1222,7 @@ function assertDeployImageTag(tag) {
 import { platform } from "node:os";
 
 // src/commands/deploy/lib/backend-deploy/command-runner.ts
-import { execSync } from "child_process";
+import { execSync as execSync2 } from "child_process";
 
 // src/commands/deploy/lib/backend-deploy/logger.ts
 var Logger = class {
@@ -1142,7 +1248,7 @@ var CommandRunner = class {
   static exec(command, cwd) {
     try {
       Logger.info(`\u6267\u884C\u547D\u4EE4: ${command}`);
-      const result = execSync(command, {
+      const result = execSync2(command, {
         cwd,
         encoding: "utf8",
         stdio: "pipe"
@@ -1160,7 +1266,7 @@ var CommandRunner = class {
   static execWithOutput(command, cwd) {
     try {
       Logger.info(`\u6267\u884C\u547D\u4EE4: ${command}`);
-      execSync(command, {
+      execSync2(command, {
         cwd,
         stdio: "inherit"
       });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-project-manage-cli",
-  "version": "3.0.8",
+  "version": "3.0.10",
   "description": "命令行工具：后续用于调用平台后端 API 完成运维与自动化操作",
   "type": "module",
   "private": false,
@@ -19,7 +19,9 @@
     "test": "vitest run",
     "test:watch": "vitest",
     "test:init": "vitest run tests/cli-init.test.ts",
-    "test:pull": "vitest run tests/cli-pull.test.ts"
+    "test:pull": "vitest run tests/cli-pull.test.ts",
+    "test:cursor": "vitest run tests/cursor.test.ts"
+
   },
   "devDependencies": {
     "@types/node": "^22.0.0",
@@ -38,6 +40,7 @@
     "commander": "~14.0.3",
     "yaml": "~2.8.4",
     "minio": "~8.0.7",
-    "dockerode": "~5.0.0"
+    "dockerode": "~5.0.0",
+    "@cursor/sdk": "~1.0.12"
   }
 }