@tea-agent/loop-agent 0.40.0-next.12 → 0.40.0-next.14

package/dist/worker/console/pi-plugins.js

@@ -0,0 +1,100 @@
+import { existsSync, mkdirSync, readFileSync } from "node:fs";
+import { spawn } from "node:child_process";
+import os from "node:os";
+import path from "node:path";
+import catalogData from "./pi-plugins.catalog.json" with { type: "json" };
+export function getPiAgentDir() {
+    const envDir = process.env.PI_CODING_AGENT_DIR;
+    return envDir ? envDir : path.join(os.homedir(), ".pi", "agent");
+}
+function npmRoot() {
+    return path.join(getPiAgentDir(), "npm");
+}
+function nodeModulesRoot() {
+    return path.join(npmRoot(), "node_modules");
+}
+function settingsPath() {
+    return path.join(getPiAgentDir(), "settings.json");
+}
+function readSettings() {
+    try {
+        if (!existsSync(settingsPath()))
+            return {};
+        return JSON.parse(readFileSync(settingsPath(), "utf8"));
+    }
+    catch {
+        return {};
+    }
+}
+function packageNameFromSpec(spec) {
+    return spec.startsWith("npm:") ? spec.slice(4) : spec;
+}
+function listInstalledPackageNames() {
+    const settings = readSettings();
+    return (settings.packages ?? []).map(packageNameFromSpec);
+}
+function isPackageInstalledOnDisk(pkg) {
+    return existsSync(path.join(nodeModulesRoot(), ...pkg.split("/")));
+}
+/** 目录 + 每个插件的已装状态（settings.packages 声明 + node_modules 存在）。 */
+export function listPiPlugins() {
+    const installed = new Set(listInstalledPackageNames());
+    const plugins = catalogData.plugins.map((p) => ({
+        ...p,
+        installed: installed.has(p.name) && isPackageInstalledOnDisk(p.name),
+    }));
+    return { summary: catalogData.summary, plugins };
+}
+function runPi(args) {
+    return new Promise((resolve) => {
+        mkdirSync(getPiAgentDir(), { recursive: true });
+        // Windows 上 pi 是 pi.cmd（无 .exe），.cmd/.bat 只能 shell:true 解析。
+        const child = spawn("pi", args, {
+            stdio: ["ignore", "pipe", "pipe"],
+            cwd: getPiAgentDir(),
+            shell: process.platform === "win32",
+        });
+        let stdout = "";
+        let stderr = "";
+        child.stdout.on("data", (d) => (stdout += d.toString("utf8")));
+        child.stderr.on("data", (d) => (stderr += d.toString("utf8")));
+        child.on("close", (code) => resolve({ code: code ?? -1, stdout, stderr }));
+        child.on("error", (err) => resolve({ code: -1, stdout, stderr: String(err?.message ?? err) }));
+    });
+}
+/** 校验安装目标是否在目录中，且该包确实声明了 pi.extensions（可安装为 Pi 扩展）。 */
+export function findCatalogPlugin(name) {
+    return catalogData.plugins.find((p) => p.name === name);
+}
+/** 全局（用户级 ~/.pi/agent）安装：调用 `pi install npm:<name>`。 */
+export async function installPiPlugin(name) {
+    const entry = findCatalogPlugin(name);
+    if (!entry) {
+        return { ok: false, error: `包 ${name} 不在插件目录中` };
+    }
+    if (!entry.valid) {
+        return {
+            ok: false,
+            error: `包 ${name} 未声明 pi.extensions，无法作为 Pi 扩展使用`,
+        };
+    }
+    const result = await runPi(["install", `npm:${name}`]);
+    if (result.code !== 0) {
+        return {
+            ok: false,
+            error: `pi install 失败 (exit ${result.code}): ${(result.stderr || result.stdout).slice(0, 400)}`,
+        };
+    }
+    return { ok: true };
+}
+/** 全局卸载：调用 `pi remove npm:<name>`。 */
+export async function uninstallPiPlugin(name) {
+    const result = await runPi(["remove", `npm:${name}`]);
+    if (result.code !== 0) {
+        return {
+            ok: false,
+            error: `pi remove 失败 (exit ${result.code}): ${(result.stderr || result.stdout).slice(0, 400)}`,
+        };
+    }
+    return { ok: true };
+}

package/dist/worker/console/routes.js

@@ -8,6 +8,7 @@ import { buildOperatorSurfaceHealthV1 } from "./operator-surface-health.js";
 import { ROUTES } from "../observe/routes.js";
 import { evaluateMutationGate, injectConfirmationTokenScript, isMutationMethod, } from "./security.js";
 import { listDagRunsForRecovery } from "./resolve-dag-run-for-task.js";
+import { installPiPlugin, listPiPlugins, uninstallPiPlugin, } from "./pi-plugins.js";
 import { pickNativeWorkspaceDirectory } from "./native-directory-picker.js";
 export function sendJson(res, status, body, extraHeaders) {
     const payload = JSON.stringify(body);
@@ -462,6 +463,94 @@ async function handleNightAuxTickerPost(req, res, ctx) {
     const status = ticker.setEnabled(body.enabled);
     sendJson(res, 200, { ok: true, ticker: status });
 }
+async function handlePiPluginInstall(req, res) {
+    let body;
+    try {
+        body = (await readJsonBody(req));
+    }
+    catch (error) {
+        sendJson(res, 400, {
+            ok: false,
+            error: {
+                code: "INVALID_INPUT",
+                message: error instanceof Error ? error.message : String(error),
+            },
+        });
+        return;
+    }
+    const name = typeof body.name === "string" ? body.name.trim() : "";
+    if (!name) {
+        sendJson(res, 400, {
+            ok: false,
+            error: { code: "INVALID_INPUT", message: "missing package name" },
+        });
+        return;
+    }
+    try {
+        const result = await installPiPlugin(name);
+        if (!result.ok) {
+            sendJson(res, 400, {
+                ok: false,
+                error: { code: "INSTALL_FAILED", message: result.error },
+            });
+            return;
+        }
+        sendJson(res, 200, { ok: true });
+    }
+    catch (error) {
+        sendJson(res, 500, {
+            ok: false,
+            error: {
+                code: "INTERNAL_ERROR",
+                message: error instanceof Error ? error.message : String(error),
+            },
+        });
+    }
+}
+async function handlePiPluginUninstall(req, res) {
+    let body;
+    try {
+        body = (await readJsonBody(req));
+    }
+    catch (error) {
+        sendJson(res, 400, {
+            ok: false,
+            error: {
+                code: "INVALID_INPUT",
+                message: error instanceof Error ? error.message : String(error),
+            },
+        });
+        return;
+    }
+    const name = typeof body.name === "string" ? body.name.trim() : "";
+    if (!name) {
+        sendJson(res, 400, {
+            ok: false,
+            error: { code: "INVALID_INPUT", message: "missing package name" },
+        });
+        return;
+    }
+    try {
+        const result = await uninstallPiPlugin(name);
+        if (!result.ok) {
+            sendJson(res, 400, {
+                ok: false,
+                error: { code: "UNINSTALL_FAILED", message: result.error },
+            });
+            return;
+        }
+        sendJson(res, 200, { ok: true });
+    }
+    catch (error) {
+        sendJson(res, 500, {
+            ok: false,
+            error: {
+                code: "INTERNAL_ERROR",
+                message: error instanceof Error ? error.message : String(error),
+            },
+        });
+    }
+}
 export async function handleConsoleOperatorRequest(req, res, ctx) {
     const method = (req.method ?? "GET").toUpperCase();
     const url = new URL(req.url ?? "/", "http://localhost");
@@ -495,6 +584,10 @@ export async function handleConsoleOperatorRequest(req, res, ctx) {
         await handleNightAuxTickerGet(req, res, ctx);
         return true;
     }
+    if (method === "GET" && pathname === "/api/operator/v1/pi-plugins") {
+        sendJson(res, 200, { ok: true, data: listPiPlugins() });
+        return true;
+    }
     const opMatch = pathname.match(/^\/api\/operator\/v1\/operations\/([^/]+)(?:\/(events))?$/);
     if (opMatch) {
         const operationId = decodeURIComponent(opMatch[1]);
@@ -526,6 +619,14 @@ export async function handleConsoleOperatorRequest(req, res, ctx) {
             await handleNightAuxTickerPost(req, res, ctx);
             return true;
         }
+        if (method === "POST" && pathname === "/api/operator/v1/pi-plugins/install") {
+            await handlePiPluginInstall(req, res);
+            return true;
+        }
+        if (method === "POST" && pathname === "/api/operator/v1/pi-plugins/uninstall") {
+            await handlePiPluginUninstall(req, res);
+            return true;
+        }
         // Allow other operator POSTs only under known routes; unknown → 404
         sendJson(res, 404, {
             ok: false,
@@ -587,9 +688,11 @@ function pendingReadinessSummary(inspectReady) {
  *   POST   /api/operator/v1/workspaces/pick        — host native folder chooser
  *   POST   /api/operator/v1/workspaces             — body { "name": "<label>",
  *                                                     "directoryPath": "<dir>" }
+ *   PATCH  /api/operator/v1/workspaces/:workspaceId — body { "beforeWorkspaceId":
+ *                                                     "<id>" | null }
  *   DELETE /api/operator/v1/workspaces/:workspaceId — soft remove (restore on re-add)
  *
- * POST/DELETE are mutations and pass the Console mutation gate. GET carries no
+ * POST/PATCH/DELETE are mutations and pass the Console mutation gate. GET carries no
  * workspace selection parameter.
  *
  * AC-006 (2026-08-23): GET is NON-BLOCKING — registry + process-level
@@ -598,8 +701,53 @@ function pendingReadinessSummary(inspectReady) {
  * requests. A hanging probe can never delay the 200 list.
  */
 export async function handleWorkspaceRoutes(req, res, registry, deps, url, pathname, method, readinessDeps) {
+    if (method === "GET" &&
+        pathname === "/api/operator/v1/workspaces/events") {
+        res.writeHead(200, {
+            "Content-Type": "text/event-stream; charset=utf-8",
+            "Cache-Control": "no-cache, no-transform",
+            Connection: "keep-alive",
+            "X-Accel-Buffering": "no",
+        });
+        const sendRevision = (event, revision) => {
+            if (res.destroyed || res.writableEnded)
+                return;
+            res.write(`id: ${revision}\nevent: ${event}\ndata: ${JSON.stringify({ schemaVersion: 1, revision })}\n\n`);
+        };
+        let baselineSent = false;
+        let pendingRevision = 0;
+        const unsubscribe = registry.subscribe((revision) => {
+            if (!baselineSent)
+                pendingRevision = Math.max(pendingRevision, revision);
+            else
+                sendRevision("workspace_registry_changed", revision);
+        });
+        const initial = await registry.snapshotVisible();
+        sendRevision("workspace_registry_ready", initial.revision);
+        baselineSent = true;
+        if (pendingRevision > initial.revision) {
+            sendRevision("workspace_registry_changed", pendingRevision);
+        }
+        const heartbeat = setInterval(() => {
+            if (!res.destroyed && !res.writableEnded)
+                res.write(": keepalive\n\n");
+        }, 15_000);
+        heartbeat.unref();
+        let closed = false;
+        const close = () => {
+            if (closed)
+                return;
+            closed = true;
+            clearInterval(heartbeat);
+            unsubscribe();
+        };
+        req.once("aborted", close);
+        res.once("close", close);
+        return;
+    }
     if (method === "GET" && pathname === "/api/operator/v1/workspaces") {
-        const records = await registry.listVisible();
+        const snapshot = await registry.snapshotVisible();
+        const records = snapshot.workspaces;
         const inspectReady = readinessDeps.inspectReady;
         const readinessCache = readinessDeps.readinessCache ?? defaultWorkspaceReadinessCache;
         const workspaces = records.map((record) => {
@@ -611,6 +759,7 @@ export async function handleWorkspaceRoutes(req, res, registry, deps, url, pathn
                 repoFingerprint: record.repoFingerprint,
                 createdAt: record.createdAt,
                 lastAccessedAt: record.lastAccessedAt,
+                sessionIds: record.sessionIds,
                 readiness: cached
                     ? {
                         state: cached.state,
@@ -622,7 +771,8 @@ export async function handleWorkspaceRoutes(req, res, registry, deps, url, pathn
         // Respond FIRST, then refresh readiness in the background (AC-006).
         sendJson(res, 200, {
             ok: true,
-            schemaVersion: 1,
+            schemaVersion: 2,
+            revision: snapshot.revision,
             workspaces,
         });
         void Promise.all(records.map(async (record) => {
@@ -789,8 +939,73 @@ export async function handleWorkspaceRoutes(req, res, registry, deps, url, pathn
         });
         return;
     }
-    const removeMatch = pathname.match(/^\/api\/operator\/v1\/workspaces\/([^/]+)$/);
-    if (method === "DELETE" && removeMatch) {
+    const workspaceMatch = pathname.match(/^\/api\/operator\/v1\/workspaces\/([^/]+)$/);
+    if (method === "PATCH" && workspaceMatch) {
+        const failure = evaluateMutationGate(req, {
+            cookieName: deps.bootToken.cookieName,
+            bootToken: deps.bootToken.value,
+            confirmationToken: deps.bootToken.confirmationToken,
+            consoleOrigin: deps.getConsoleOrigin(),
+            allowNonLoopbackAccess: deps.allowNonLoopbackAccess,
+        });
+        if (failure) {
+            sendJson(res, failure.status, {
+                ok: false,
+                error: { code: failure.code, message: failure.message },
+            });
+            return;
+        }
+        let body;
+        try {
+            body = (await readJsonBody(req));
+        }
+        catch (error) {
+            sendJson(res, 400, {
+                ok: false,
+                error: {
+                    code: "INVALID_INPUT",
+                    message: error instanceof Error ? error.message : String(error),
+                },
+            });
+            return;
+        }
+        if (body.beforeWorkspaceId !== null &&
+            typeof body.beforeWorkspaceId !== "string") {
+            sendJson(res, 400, {
+                ok: false,
+                error: {
+                    code: "INVALID_INPUT",
+                    message: "beforeWorkspaceId must be a workspace id or null",
+                },
+            });
+            return;
+        }
+        const workspaceId = decodeURIComponent(workspaceMatch[1]);
+        const beforeWorkspaceId = typeof body.beforeWorkspaceId === "string"
+            ? body.beforeWorkspaceId
+            : null;
+        const reordered = await registry.insertBefore(workspaceId, beforeWorkspaceId);
+        if (!reordered.ok) {
+            sendJson(res, 404, {
+                ok: false,
+                error: {
+                    code: reordered.code,
+                    message: reordered.code === "ANCHOR_NOT_FOUND"
+                        ? `unknown workspace anchor: ${beforeWorkspaceId}`
+                        : `unknown workspace: ${workspaceId}`,
+                },
+            });
+            return;
+        }
+        sendJson(res, 200, {
+            ok: true,
+            workspaceId,
+            beforeWorkspaceId,
+            changed: reordered.changed,
+        });
+        return;
+    }
+    if (method === "DELETE" && workspaceMatch) {
         const failure = evaluateMutationGate(req, {
             cookieName: deps.bootToken.cookieName,
             bootToken: deps.bootToken.value,
@@ -805,7 +1020,7 @@ export async function handleWorkspaceRoutes(req, res, registry, deps, url, pathn
             });
             return;
         }
-        const workspaceId = decodeURIComponent(removeMatch[1]);
+        const workspaceId = decodeURIComponent(workspaceMatch[1]);
         const removed = await registry.remove(workspaceId);
         if (!removed) {
             sendJson(res, 404, {