@tea-agent/loop-agent 0.40.0-next.9 → 0.41.0-beta.1

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

@@ -0,0 +1,179 @@
+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) {
+    const base = spec.startsWith("npm:") ? spec.slice(4) : spec;
+    // 剥掉尾部 @<version>（如 pi-lens@3.8.72 → pi-lens），但保留 scoped 包名开头的 @
+    // （如 @scope/pkg → @scope/pkg，@scope/pkg@1.0.0 → @scope/pkg）。
+    const at = base.lastIndexOf("@");
+    if (at > 0)
+        return base.slice(0, at);
+    return base;
+}
+function listInstalledPackageNames() {
+    const settings = readSettings();
+    return (settings.packages ?? []).map(packageNameFromSpec);
+}
+function isPackageInstalledOnDisk(pkg) {
+    return existsSync(path.join(nodeModulesRoot(), ...pkg.split("/")));
+}
+/** 读取已安装包的实际版本（node_modules 里 package.json 的 version）。 */
+function readInstalledVersion(pkg) {
+    try {
+        const manifest = JSON.parse(readFileSync(path.join(nodeModulesRoot(), ...pkg.split("/"), "package.json"), "utf8"));
+        return typeof manifest.version === "string" ? manifest.version : undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+/** 目录 + 每个插件的已装状态（settings.packages 声明 + node_modules 存在）+ 已装版本。 */
+export function listPiPlugins() {
+    const installed = new Set(listInstalledPackageNames());
+    const plugins = catalogData.plugins.map((p) => {
+        const isInstalled = installed.has(p.name) && isPackageInstalledOnDisk(p.name);
+        return {
+            ...p,
+            installed: isInstalled,
+            installedVersion: isInstalled ? readInstalledVersion(p.name) : undefined,
+        };
+    });
+    return { summary: catalogData.summary, plugins };
+}
+let cachedRegistry;
+/** 读取 npm 配置的 registry（`npm config get registry`），失败回退官方源。 */
+function getNpmRegistry() {
+    if (cachedRegistry)
+        return Promise.resolve(cachedRegistry);
+    return new Promise((resolve) => {
+        const fallback = () => {
+            cachedRegistry = "https://registry.npmjs.org/";
+            resolve(cachedRegistry);
+        };
+        try {
+            const child = spawn("npm", ["config", "get", "registry"], {
+                stdio: ["ignore", "pipe", "pipe"],
+                cwd: getPiAgentDir(),
+                shell: process.platform === "win32",
+            });
+            let out = "";
+            child.stdout.on("data", (d) => (out += d.toString("utf8")));
+            child.on("close", () => {
+                const reg = out.trim();
+                cachedRegistry = reg || "https://registry.npmjs.org/";
+                resolve(cachedRegistry);
+            });
+            child.on("error", fallback);
+        }
+        catch {
+            fallback();
+        }
+    });
+}
+/** 从 npm 配置的仓库拉取某包的全部已发布版本 + latest。 */
+export async function listPiPluginVersions(name) {
+    const registry = (await getNpmRegistry()).replace(/\/+$/, "");
+    const url = `${registry}/${encodeURIComponent(name).replace(/%2F/g, "/")}`;
+    try {
+        const res = await fetch(url, {
+            headers: { accept: "application/json", "user-agent": "loop-agent/1.0" },
+        });
+        if (!res.ok)
+            return { ok: false, error: `npm registry HTTP ${res.status}` };
+        const body = (await res.json());
+        const versions = body.versions && typeof body.versions === "object"
+            ? Object.keys(body.versions).sort((a, b) => b.localeCompare(a, undefined, { numeric: true }))
+            : [];
+        const latest = body["dist-tags"]?.latest;
+        return {
+            ok: true,
+            versions,
+            latest: typeof latest === "string" ? latest : undefined,
+        };
+    }
+    catch (error) {
+        return {
+            ok: false,
+            error: error instanceof Error ? error.message : String(error),
+        };
+    }
+}
+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, version) {
+    const entry = findCatalogPlugin(name);
+    if (!entry) {
+        return { ok: false, error: `包 ${name} 不在插件目录中` };
+    }
+    if (!entry.valid) {
+        return {
+            ok: false,
+            error: `包 ${name} 未声明 pi.extensions，无法作为 Pi 扩展使用`,
+        };
+    }
+    const spec = version && version.trim() ? `npm:${name}@${version.trim()}` : `npm:${name}`;
+    const result = await runPi(["install", spec]);
+    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 };
+}