@tomsun28/pizza 0.1.0 → 0.1.1

package/dist/modes/rpc/rpc-client.js

@@ -0,0 +1,405 @@
+/**
+ * RPC Client for programmatic access to the coding agent.
+ *
+ * Spawns the agent in RPC mode and provides a typed API for all operations.
+ */
+import { spawn } from "node:child_process";
+import { attachJsonlLineReader, serializeJsonLine } from "./jsonl.js";
+// ============================================================================
+// RPC Client
+// ============================================================================
+export class RpcClient {
+    constructor(options = {}) {
+        this.options = options;
+        this.process = null;
+        this.stopReadingStdout = null;
+        this.eventListeners = [];
+        this.pendingRequests = new Map();
+        this.requestId = 0;
+        this.stderr = "";
+    }
+    /**
+     * Start the RPC agent process.
+     */
+    async start() {
+        if (this.process) {
+            throw new Error("Client already started");
+        }
+        const cliPath = this.options.cliPath ?? "dist/cli.js";
+        const args = ["--mode", "rpc"];
+        if (this.options.provider) {
+            args.push("--provider", this.options.provider);
+        }
+        if (this.options.model) {
+            args.push("--model", this.options.model);
+        }
+        if (this.options.args) {
+            args.push(...this.options.args);
+        }
+        this.process = spawn("node", [cliPath, ...args], {
+            cwd: this.options.cwd,
+            env: { ...process.env, ...this.options.env },
+            stdio: ["pipe", "pipe", "pipe"],
+        });
+        // Collect stderr for debugging
+        this.process.stderr?.on("data", (data) => {
+            this.stderr += data.toString();
+            process.stderr.write(data);
+        });
+        // Set up strict JSONL reader for stdout.
+        this.stopReadingStdout = attachJsonlLineReader(this.process.stdout, (line) => {
+            this.handleLine(line);
+        });
+        // Wait a moment for process to initialize
+        await new Promise((resolve) => setTimeout(resolve, 100));
+        if (this.process.exitCode !== null) {
+            throw new Error(`Agent process exited immediately with code ${this.process.exitCode}. Stderr: ${this.stderr}`);
+        }
+    }
+    /**
+     * Stop the RPC agent process.
+     */
+    async stop() {
+        if (!this.process)
+            return;
+        this.stopReadingStdout?.();
+        this.stopReadingStdout = null;
+        this.process.kill("SIGTERM");
+        // Wait for process to exit
+        await new Promise((resolve) => {
+            const timeout = setTimeout(() => {
+                this.process?.kill("SIGKILL");
+                resolve();
+            }, 1000);
+            this.process?.on("exit", () => {
+                clearTimeout(timeout);
+                resolve();
+            });
+        });
+        this.process = null;
+        this.pendingRequests.clear();
+    }
+    /**
+     * Subscribe to agent events.
+     */
+    onEvent(listener) {
+        this.eventListeners.push(listener);
+        return () => {
+            const index = this.eventListeners.indexOf(listener);
+            if (index !== -1) {
+                this.eventListeners.splice(index, 1);
+            }
+        };
+    }
+    /**
+     * Get collected stderr output (useful for debugging).
+     */
+    getStderr() {
+        return this.stderr;
+    }
+    // =========================================================================
+    // Command Methods
+    // =========================================================================
+    /**
+     * Send a prompt to the agent.
+     * Returns immediately after sending; use onEvent() to receive streaming events.
+     * Use waitForIdle() to wait for completion.
+     */
+    async prompt(message, images) {
+        await this.send({ type: "prompt", message, images });
+    }
+    /**
+     * Queue a steering message to interrupt the agent mid-run.
+     */
+    async steer(message, images) {
+        await this.send({ type: "steer", message, images });
+    }
+    /**
+     * Queue a follow-up message to be processed after the agent finishes.
+     */
+    async followUp(message, images) {
+        await this.send({ type: "follow_up", message, images });
+    }
+    /**
+     * Abort current operation.
+     */
+    async abort() {
+        await this.send({ type: "abort" });
+    }
+    /**
+     * Rewind to a branch point, or no-op to continue the eternal conversation.
+     * @param targetEventId - Optional event id to fork from
+     */
+    async rewind(targetEventId) {
+        const response = await this.send({ type: "rewind", targetEventId });
+        return this.getData(response);
+    }
+    /**
+     * Get current session state.
+     */
+    async getState() {
+        const response = await this.send({ type: "get_state" });
+        return this.getData(response);
+    }
+    /**
+     * Set model by provider and ID.
+     */
+    async setModel(provider, modelId) {
+        const response = await this.send({ type: "set_model", provider, modelId });
+        return this.getData(response);
+    }
+    /**
+     * Cycle to next model.
+     */
+    async cycleModel() {
+        const response = await this.send({ type: "cycle_model" });
+        return this.getData(response);
+    }
+    /**
+     * Get list of available models.
+     */
+    async getAvailableModels() {
+        const response = await this.send({ type: "get_available_models" });
+        return this.getData(response).models;
+    }
+    /**
+     * Set thinking level.
+     */
+    async setThinkingLevel(level) {
+        await this.send({ type: "set_thinking_level", level });
+    }
+    /**
+     * Cycle thinking level.
+     */
+    async cycleThinkingLevel() {
+        const response = await this.send({ type: "cycle_thinking_level" });
+        return this.getData(response);
+    }
+    /**
+     * Set steering mode.
+     */
+    async setSteeringMode(mode) {
+        await this.send({ type: "set_steering_mode", mode });
+    }
+    /**
+     * Set follow-up mode.
+     */
+    async setFollowUpMode(mode) {
+        await this.send({ type: "set_follow_up_mode", mode });
+    }
+    /**
+     * Compact session context.
+     */
+    async compact(customInstructions) {
+        const response = await this.send({ type: "compact", customInstructions });
+        return this.getData(response);
+    }
+    /**
+     * Set auto-compaction enabled/disabled.
+     */
+    async setAutoCompaction(enabled) {
+        await this.send({ type: "set_auto_compaction", enabled });
+    }
+    /**
+     * Set auto-retry enabled/disabled.
+     */
+    async setAutoRetry(enabled) {
+        await this.send({ type: "set_auto_retry", enabled });
+    }
+    /**
+     * Abort in-progress retry.
+     */
+    async abortRetry() {
+        await this.send({ type: "abort_retry" });
+    }
+    /**
+     * Execute a bash command.
+     */
+    async bash(command) {
+        const response = await this.send({ type: "bash", command });
+        return this.getData(response);
+    }
+    /**
+     * Abort running bash command.
+     */
+    async abortBash() {
+        await this.send({ type: "abort_bash" });
+    }
+    /**
+     * Get session statistics.
+     */
+    async getSessionStats() {
+        const response = await this.send({ type: "get_session_stats" });
+        return this.getData(response);
+    }
+    /**
+     * Export session to HTML.
+     */
+    async exportHtml(outputPath) {
+        const response = await this.send({ type: "export_html", outputPath });
+        return this.getData(response);
+    }
+    /**
+     * Switch to a different session file.
+     * @returns Object with `cancelled: true` if an extension cancelled the switch
+     */
+    async switchSession(sessionPath) {
+        const response = await this.send({ type: "switch_session", sessionPath });
+        return this.getData(response);
+    }
+    /**
+     * Fork from a specific message.
+     * @returns Object with `text` (the message text) and `cancelled` (if extension cancelled)
+     */
+    async fork(entryId) {
+        const response = await this.send({ type: "fork", entryId });
+        return this.getData(response);
+    }
+    /**
+     * Clone the current active branch into a new session.
+     * @returns Object with `cancelled: true` if an extension cancelled the clone
+     */
+    async clone() {
+        const response = await this.send({ type: "clone" });
+        return this.getData(response);
+    }
+    /**
+     * Get messages available for forking.
+     */
+    async getForkMessages() {
+        const response = await this.send({ type: "get_fork_messages" });
+        return this.getData(response).messages;
+    }
+    /**
+     * Get text of last assistant message.
+     */
+    async getLastAssistantText() {
+        const response = await this.send({ type: "get_last_assistant_text" });
+        return this.getData(response).text;
+    }
+    /**
+     * Get all messages in the session.
+     */
+    async getMessages() {
+        const response = await this.send({ type: "get_messages" });
+        return this.getData(response).messages;
+    }
+    /**
+     * Get available commands (extension commands, prompt templates, skills).
+     */
+    async getCommands() {
+        const response = await this.send({ type: "get_commands" });
+        return this.getData(response).commands;
+    }
+    // =========================================================================
+    // Helpers
+    // =========================================================================
+    /**
+     * Wait for agent to become idle (no streaming).
+     * Resolves when AGENT_TURN_COMPLETED is received.
+     */
+    waitForIdle(timeout = 60000) {
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                unsubscribe();
+                reject(new Error(`Timeout waiting for agent to become idle. Stderr: ${this.stderr}`));
+            }, timeout);
+            const unsubscribe = this.onEvent((event) => {
+                if (isIdleEvent(event)) {
+                    clearTimeout(timer);
+                    unsubscribe();
+                    resolve();
+                }
+            });
+        });
+    }
+    /**
+     * Collect events until agent becomes idle.
+     */
+    collectEvents(timeout = 60000) {
+        return new Promise((resolve, reject) => {
+            const events = [];
+            const timer = setTimeout(() => {
+                unsubscribe();
+                reject(new Error(`Timeout collecting events. Stderr: ${this.stderr}`));
+            }, timeout);
+            const unsubscribe = this.onEvent((event) => {
+                events.push(event);
+                if (isIdleEvent(event)) {
+                    clearTimeout(timer);
+                    unsubscribe();
+                    resolve(events);
+                }
+            });
+        });
+    }
+    /**
+     * Send prompt and wait for completion, returning all events.
+     */
+    async promptAndWait(message, images, timeout = 60000) {
+        const eventsPromise = this.collectEvents(timeout);
+        await this.prompt(message, images);
+        return eventsPromise;
+    }
+    // =========================================================================
+    // Internal
+    // =========================================================================
+    handleLine(line) {
+        try {
+            const data = JSON.parse(line);
+            // Check if it's a response to a pending request
+            if (data.type === "response" && data.id && this.pendingRequests.has(data.id)) {
+                const pending = this.pendingRequests.get(data.id);
+                this.pendingRequests.delete(data.id);
+                pending.resolve(data);
+                return;
+            }
+            // Otherwise it's an event
+            for (const listener of this.eventListeners) {
+                listener(data);
+            }
+        }
+        catch {
+            // Ignore non-JSON lines
+        }
+    }
+    async send(command) {
+        if (!this.process?.stdin) {
+            throw new Error("Client not started");
+        }
+        const id = `req_${++this.requestId}`;
+        const fullCommand = { ...command, id };
+        return new Promise((resolve, reject) => {
+            this.pendingRequests.set(id, { resolve, reject });
+            const timeout = setTimeout(() => {
+                this.pendingRequests.delete(id);
+                reject(new Error(`Timeout waiting for response to ${command.type}. Stderr: ${this.stderr}`));
+            }, 30000);
+            this.pendingRequests.set(id, {
+                resolve: (response) => {
+                    clearTimeout(timeout);
+                    resolve(response);
+                },
+                reject: (error) => {
+                    clearTimeout(timeout);
+                    reject(error);
+                },
+            });
+            this.process.stdin.write(serializeJsonLine(fullCommand));
+        });
+    }
+    getData(response) {
+        if (!response.success) {
+            const errorResponse = response;
+            throw new Error(errorResponse.error);
+        }
+        // Type assertion: we trust response.data matches T based on the command sent.
+        // This is safe because each public method specifies the correct T for its command.
+        const successResponse = response;
+        return successResponse.data;
+    }
+}
+function isIdleEvent(event) {
+    const type = event.type;
+    return type === "AGENT_TURN_COMPLETED" || type === "agent_end";
+}

package/dist/modes/rpc/rpc-mode.d.ts

@@ -0,0 +1,19 @@
+/**
+ * RPC mode: Headless operation with JSON stdin/stdout protocol.
+ *
+ * Used for embedding the agent in other applications.
+ * Receives commands as JSON on stdin, outputs events and responses as JSON on stdout.
+ *
+ * Protocol:
+ * - Commands: JSON objects with `type` field, optional `id` for correlation
+ * - Responses: JSON objects with `type: "response"`, `command`, `success`, and optional `data`/`error`
+ * - Events: TypedEvent objects streamed as they occur
+ * - Extension UI: Extension UI requests are emitted, client responds with extension_ui_response
+ */
+import type { SessionFacade } from "../../core/session-facade.js";
+export type { RpcCommand, RpcExtensionUIRequest, RpcExtensionUIResponse, RpcResponse, RpcSessionState, } from "./rpc-types.js";
+/**
+ * Run RPC mode against the event-sourced facade.
+ * Events are emitted as raw EventStore TypedEvent JSON lines.
+ */
+export declare function runRpcModeWithFacade(facade: SessionFacade): Promise<never>;