ai-project-manage-cli 3.0.9 → 3.0.11

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,12 @@ 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";
+import { appendFileSync } from "fs";
+import { resolve as resolve2 } from "path";
 function runConnect(opts) {
   void (async () => {
     const cfg = await ensureApmConfig();
@@ -240,6 +248,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 +267,142 @@ 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"
+        });
+        let IN_PROGRESS = false;
+        const events = [];
+        let data2 = {};
+        const run = await agent.send(payload.prompt);
+        for await (const event of run.stream()) {
+          if (!IN_PROGRESS) {
+            await api.cliRequirements.updateTaskStatus({
+              taskId: payload.taskId,
+              requirementId: payload.requirementId,
+              status: "IN_PROGRESS"
+            });
+            IN_PROGRESS = true;
+          }
+          data2 = {};
+          if (event.type === "assistant") {
+            const output = event.message.content[0].text;
+            console.log(`[Output]`, output);
+            if (data2.type && data2.type !== "output") {
+              events.push(data2);
+              data2 = {};
+            } else {
+              data2.type = "output";
+              if (!data2.content) {
+                data2.content = "";
+              }
+              data2.content += output;
+            }
+          }
+          if (event.type === "thinking") {
+            const thinking = event.text;
+            console.log(`[Thinking]`, thinking);
+            if (data2.type && data2.type !== "thinking") {
+              events.push(data2);
+              data2 = {};
+            } else {
+              data2.type = "thinking";
+              if (!data2.content) {
+                data2.content = "";
+              }
+              data2.content += thinking;
+            }
+          }
+          if (event.type === "tool_call") {
+            if (event.status === "completed" || event.status === "error") {
+              console.log(`[ToolCall(${event.call_id}) Result]`);
+              console.log(JSON.stringify(event.args));
+              console.log(JSON.stringify(event.result, null, 2));
+              console.log("---------------\u8C03\u7528\u5B8C\u6210-----------------");
+              if (data2.type && data2.type !== "tool_call") {
+                events.push(data2);
+                data2 = {};
+              } else {
+                data2.type = "tool_call";
+                data2.call_id = event.call_id;
+                data2.args = event.args;
+                data2.result = event.result;
+                data2.status = event.status;
+              }
+            }
+          }
+          if (event.type === "task") {
+            console.log(`[Task:${event.status}]`);
+            console.log("--------------------------------");
+            console.log(event.text || "\u65E0");
+            console.log("--------------------------------");
+            if (data2.type && data2.type !== "task") {
+              events.push(data2);
+              data2 = {};
+            } else {
+              data2.type = "task";
+              data2.status = event.status;
+              data2.text = event.text;
+            }
+          }
+          if (event.type === "request") {
+            console.log(JSON.stringify(event, null, 2));
+            if (data2.type && data2.type !== "request") {
+              events.push(data2);
+              data2 = {};
+            } else {
+              data2 = event;
+            }
+          }
+        }
+        if (data2.type) {
+          events.push(data2);
+          data2 = {};
+        }
+        await api.cliRequirements.updateTaskStatus({
+          taskId: payload.taskId,
+          requirementId: payload.requirementId,
+          status: "PENDING_ACCEPTANCE"
+        });
+        await api.cliRequirements.updateDevStatus({
+          requirementId: payload.requirementId,
+          status: "IDLE"
+        });
+        console.log("[Done]");
+        const sessionsDir = resolve2(
+          payload.cwd,
+          `.apm/workitems/${payload.requirementId}/sessions`
+        );
+        ensureDirExists(sessionsDir);
+        const sessionFile = resolve2(sessionsDir, `${run.agentId}.md`);
+        for (const event of events) {
+          appendFileSync(sessionFile, JSON.stringify(event, null, 2) + "\n=====================================================\n");
+        }
+      } catch {
+        console.error("[apm] \u65E0\u6CD5\u89E3\u6790 WebSocket \u6D88\u606F:", text);
+      }
     });
     ws.on("error", (err) => {
       console.error("[apm] WebSocket \u9519\u8BEF:", err.message);
@@ -334,13 +476,13 @@ async function runLogin(opts) {
 
 // src/commands/branch.ts
 import { execFile } from "child_process";
-import { resolve as resolve2 } from "path";
+import { resolve as resolve3 } from "path";
 import { promisify } from "util";
 var execFileAsync = promisify(execFile);
 async function fetchBaselineBranchFromApi(requirementId, cwd) {
   const cfg = await ensureLoggedConfig();
   const api = createApmApiClient(cfg);
-  const workdirPath = resolve2(cwd);
+  const workdirPath = resolve3(cwd);
   const { baselineBranch } = await api.cliRequirements.branchBaseline({
     requirementId,
     workdirPath
@@ -725,11 +867,11 @@ import path5 from "node:path";
 
 // src/commands/deploy/lib/apm-config.ts
 import { existsSync as existsSync3, readFileSync as readFileSync6 } from "node:fs";
-import { resolve as resolve3 } from "node:path";
+import { resolve as resolve4 } from "node:path";
 function loadApmConfig(options) {
-  const p = resolve3(
+  const p = resolve4(
     process.cwd(),
-    options?.configPath ?? resolve3(WORKSPACE_APM_DIR, "apm.config.json")
+    options?.configPath ?? resolve4(WORKSPACE_APM_DIR, "apm.config.json")
   );
   if (!existsSync3(p)) {
     console.error(`\u672A\u627E\u5230\u914D\u7F6E\u6587\u4EF6\uFF1A${p}`);
@@ -973,17 +1115,17 @@ var DockerodeClient = class {
     await this.client.getImage(image).remove({ force: true });
   }
   async pullImage(image, auth) {
-    const stream = await new Promise((resolve4, reject) => {
+    const stream = await new Promise((resolve5, reject) => {
       const pullOptions = auth ? { authconfig: auth } : void 0;
       this.client.pull(image, pullOptions, (err, output) => {
         if (err || !output) {
           reject(err ?? new Error("docker pull \u8FD4\u56DE\u7A7A\u8F93\u51FA"));
           return;
         }
-        resolve4(output);
+        resolve5(output);
       });
     });
-    await new Promise((resolve4, reject) => {
+    await new Promise((resolve5, reject) => {
       this.client.modem.followProgress(
         stream,
         (err) => {
@@ -991,7 +1133,7 @@ var DockerodeClient = class {
             reject(err);
             return;
           }
-          resolve4();
+          resolve5();
         },
         () => void 0
       );
@@ -1136,7 +1278,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 {
@@ -1162,7 +1304,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"
@@ -1180,7 +1322,7 @@ var CommandRunner = class {
   static execWithOutput(command, cwd) {
     try {
       Logger.info(`\u6267\u884C\u547D\u4EE4: ${command}`);
-      execSync(command, {
+      execSync2(command, {
         cwd,
         stdio: "inherit"
       });
@@ -1507,14 +1649,14 @@ var MinioClient = class {
   async deleteObjectsByPrefix(bucket, prefix) {
     const objectsStream = this.inner.listObjectsV2(bucket, prefix, true);
     const keys = [];
-    await new Promise((resolve4, reject) => {
+    await new Promise((resolve5, reject) => {
       objectsStream.on("data", (obj) => {
         if (obj.name) {
           keys.push(obj.name);
         }
       });
       objectsStream.on("error", reject);
-      objectsStream.on("end", resolve4);
+      objectsStream.on("end", resolve5);
     });
     const chunkSize = 500;
     for (let i = 0; i < keys.length; i += chunkSize) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-project-manage-cli",
-  "version": "3.0.9",
+  "version": "3.0.11",
   "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"
   }
 }