@astrofoundry/pi-astro 0.2.1

package/extensions/astro-agents/spawn.ts

@@ -0,0 +1,162 @@
+import { spawn } from "node:child_process";
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import type { AgentConfig } from "./discovery.ts";
+
+interface AssistantMessageContent {
+    type: string;
+    text?: string;
+}
+
+interface AssistantMessage {
+    role?: string;
+    content?: AssistantMessageContent[];
+}
+
+interface AgentEndEvent {
+    type: "agent_end";
+    messages?: AssistantMessage[];
+}
+
+interface UnknownEvent {
+    type?: string;
+    [key: string]: unknown;
+}
+
+export interface SpawnResult {
+    text: string;
+    exitCode: number;
+    stderr: string;
+}
+
+function getPiInvocation(args: string[]): { command: string; args: string[] } {
+    const currentScript = process.argv[1];
+    const isBunVirtualScript = currentScript?.startsWith("/$bunfs/root/");
+    if (currentScript && !isBunVirtualScript && fs.existsSync(currentScript)) {
+        return { command: process.execPath, args: [currentScript, ...args] };
+    }
+
+    const execName = path.basename(process.execPath).toLowerCase();
+    const isGenericRuntime = /^(node|bun)(\.exe)?$/.test(execName);
+    if (!isGenericRuntime) {
+        return { command: process.execPath, args };
+    }
+
+    return { command: "pi", args };
+}
+
+function extractTextFromMessages(messages: AssistantMessage[]): string {
+    for (let i = messages.length - 1; i >= 0; i--) {
+        const msg = messages[i];
+        if (msg.role !== "assistant" || !msg.content) continue;
+        const text = msg.content
+            .filter((part) => part.type === "text" && typeof part.text === "string")
+            .map((part) => part.text)
+            .join("\n")
+            .trim();
+        if (text.length > 0) return text;
+    }
+    return "";
+}
+
+function writeSystemPromptTempFile(agentName: string, prompt: string): { dir: string; filePath: string } {
+    const dir = fs.mkdtempSync(path.join(os.tmpdir(), `astro-agent-${agentName}-`));
+    const filePath = path.join(dir, "system-prompt.txt");
+    fs.writeFileSync(filePath, prompt, "utf-8");
+    return { dir, filePath };
+}
+
+export interface RunAgentOptions {
+    agent: AgentConfig;
+    task: string;
+    cwd: string;
+    signal?: AbortSignal | undefined;
+}
+
+export async function runAgent(options: RunAgentOptions): Promise<SpawnResult> {
+    const { agent, task, cwd, signal } = options;
+
+    const args = ["--mode", "json", "-p", "--no-session"];
+    let tmpDir: string | null = null;
+
+    if (agent.systemPrompt.trim()) {
+        const tmp = writeSystemPromptTempFile(agent.name, agent.systemPrompt);
+        tmpDir = tmp.dir;
+        args.push("--append-system-prompt", tmp.filePath);
+    }
+
+    args.push(`Task: ${task}`);
+
+    try {
+        return await new Promise<SpawnResult>((resolve) => {
+            const invocation = getPiInvocation(args);
+            const proc = spawn(invocation.command, invocation.args, {
+                cwd,
+                shell: false,
+                stdio: ["ignore", "pipe", "pipe"],
+            });
+
+            let stdoutBuffer = "";
+            let stderrBuffer = "";
+            const collectedMessages: AssistantMessage[] = [];
+
+            const handleLine = (line: string): void => {
+                if (!line.trim()) return;
+                let event: UnknownEvent;
+                try {
+                    event = JSON.parse(line) as UnknownEvent;
+                } catch {
+                    return;
+                }
+                if (event.type === "agent_end") {
+                    const end = event as AgentEndEvent;
+                    if (end.messages) collectedMessages.push(...end.messages);
+                }
+            };
+
+            proc.stdout.on("data", (chunk: Buffer) => {
+                stdoutBuffer += chunk.toString("utf-8");
+                let newlineIndex = stdoutBuffer.indexOf("\n");
+                while (newlineIndex !== -1) {
+                    const line = stdoutBuffer.slice(0, newlineIndex);
+                    stdoutBuffer = stdoutBuffer.slice(newlineIndex + 1);
+                    handleLine(line);
+                    newlineIndex = stdoutBuffer.indexOf("\n");
+                }
+            });
+
+            proc.stderr.on("data", (chunk: Buffer) => {
+                stderrBuffer += chunk.toString("utf-8");
+            });
+
+            const abortHandler = (): void => {
+                proc.kill("SIGTERM");
+            };
+            signal?.addEventListener("abort", abortHandler, { once: true });
+
+            proc.on("close", (code) => {
+                signal?.removeEventListener("abort", abortHandler);
+                if (stdoutBuffer.trim()) handleLine(stdoutBuffer);
+                resolve({
+                    text: extractTextFromMessages(collectedMessages),
+                    exitCode: code ?? 0,
+                    stderr: stderrBuffer,
+                });
+            });
+
+            proc.on("error", (err) => {
+                signal?.removeEventListener("abort", abortHandler);
+                resolve({
+                    text: "",
+                    exitCode: 1,
+                    stderr: stderrBuffer + "\n" + err.message,
+                });
+            });
+        });
+    } finally {
+        if (tmpDir) {
+            fs.rmSync(tmpDir, { recursive: true, force: true });
+        }
+    }
+}

package/package.json

@@ -0,0 +1,72 @@
+{
+  "name": "@astrofoundry/pi-astro",
+  "version": "0.2.1",
+  "description": "Personal pi customizations (extensions, skills, prompts, themes) for the pi coding agent.",
+  "keywords": [
+    "pi-package"
+  ],
+  "license": "ISC",
+  "type": "module",
+  "publishConfig": {
+    "access": "public"
+  },
+  "pi": {
+    "extensions": [
+      "./extensions"
+    ],
+    "skills": [
+      "./skills"
+    ],
+    "prompts": [
+      "./prompts"
+    ],
+    "themes": [
+      "./themes"
+    ]
+  },
+  "files": [
+    "extensions",
+    "skills",
+    "prompts",
+    "themes",
+    "README.md"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/astrofoundry/pi-astro.git"
+  },
+  "bugs": {
+    "url": "https://github.com/astrofoundry/pi-astro/issues"
+  },
+  "homepage": "https://github.com/astrofoundry/pi-astro#readme",
+  "peerDependencies": {
+    "@mariozechner/pi-agent-core": "*",
+    "@mariozechner/pi-ai": "*",
+    "@mariozechner/pi-coding-agent": "*",
+    "@mariozechner/pi-tui": "*",
+    "typebox": "*"
+  },
+  "devDependencies": {
+    "@eslint/js": "^10.0.1",
+    "@mariozechner/pi-agent-core": "^0.70.0",
+    "@mariozechner/pi-ai": "^0.70.0",
+    "@mariozechner/pi-coding-agent": "^0.70.0",
+    "@mariozechner/pi-tui": "^0.70.0",
+    "@types/node": "^25.6.0",
+    "eslint": "^10.2.1",
+    "globals": "^17.5.0",
+    "typebox": "^1.1.33",
+    "typescript": "^6.0.3",
+    "typescript-eslint": "^8.59.0",
+    "vitest": "^4.1.5"
+  },
+  "scripts": {
+    "test": "vitest run --passWithNoTests",
+    "lint": "eslint .",
+    "typecheck": "tsc --noEmit",
+    "check": "tsc --noEmit && eslint . && vitest run --passWithNoTests",
+    "release:patch": "v=$(pnpm version patch --no-git-tag-version) && git add -A && git commit -m \"Release $v\" && git tag $v && git push && git push origin $v",
+    "release:minor": "v=$(pnpm version minor --no-git-tag-version) && git add -A && git commit -m \"Release $v\" && git tag $v && git push && git push origin $v",
+    "release:major": "v=$(pnpm version major --no-git-tag-version) && git add -A && git commit -m \"Release $v\" && git tag $v && git push && git push origin $v"
+  }
+}