@gjczone/pi-swarm 0.1.2

package/dist/state/recovery.js

@@ -0,0 +1,149 @@
+/**
+ * state/recovery — crash recovery and stale run detection.
+ *
+ * On session start, scans for incomplete runs.  If the parent Pi
+ * process is no longer alive, marks the run as abandoned.  Completed
+ * runs older than 7 days are cleaned up automatically.
+ */
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { resolveCrewRoot, listActiveRuns, readManifest, updateManifest, deleteRunState, appendEvent, } from "./persistence.js";
+// ---------------------------------------------------------------------------
+// Constants
+// ---------------------------------------------------------------------------
+/** Runs completed more than this many ms ago are eligible for cleanup. */
+const COMPLETED_RUN_RETENTION_MS = 7 * 24 * 60 * 60 * 1000; // 7 days
+/** Runs with status "running" older than this without a heartbeat are stale. */
+const STALE_RUN_THRESHOLD_MS = 30 * 60 * 1000; // 30 minutes
+/**
+ * Run recovery on session start.
+ *
+ * - Marks stale "running" runs as abandoned.
+ * - Cleans up expired completed runs.
+ * - Returns lists of resumable and abandoned runs.
+ */
+export function recoverRuns(cwd) {
+    const crewRoot = resolveCrewRoot(cwd);
+    const runsDir = path.join(crewRoot, "state", "runs");
+    if (!fs.existsSync(runsDir)) {
+        return { resumable: [], abandoned: [], cleanedUp: [] };
+    }
+    const runIds = listActiveRuns(crewRoot);
+    const now = Date.now();
+    const resumable = [];
+    const abandoned = [];
+    const cleanedUp = [];
+    for (const runId of runIds) {
+        const manifest = readManifest(crewRoot, runId);
+        if (!manifest) {
+            // Orphaned directory — clean up
+            try {
+                deleteRunState(crewRoot, runId);
+                cleanedUp.push(runId);
+            }
+            catch {
+                // Best effort
+            }
+            continue;
+        }
+        switch (manifest.status) {
+            case "running": {
+                // Check for staleness
+                const age = now - manifest.startedAt;
+                if (age > STALE_RUN_THRESHOLD_MS) {
+                    // Mark as abandoned
+                    const updated = {
+                        ...manifest,
+                        status: "abandoned",
+                        completedAt: now,
+                    };
+                    updateManifest(crewRoot, updated);
+                    appendEvent(crewRoot, runId, {
+                        type: "run.abandoned",
+                        reason: "stale",
+                        timestamp: new Date().toISOString(),
+                    });
+                    abandoned.push(updated);
+                }
+                else {
+                    // Still potentially resumable
+                    resumable.push(manifest);
+                }
+                break;
+            }
+            case "completed":
+            case "failed": {
+                // Clean up expired runs
+                const completedAt = manifest.completedAt ?? manifest.startedAt;
+                const age = now - completedAt;
+                if (age > COMPLETED_RUN_RETENTION_MS) {
+                    try {
+                        deleteRunState(crewRoot, runId);
+                        cleanedUp.push(runId);
+                    }
+                    catch {
+                        // Best effort
+                    }
+                }
+                break;
+            }
+            case "abandoned":
+                // Already abandoned — check if we should clean up
+                {
+                    const abandonedAt = manifest.completedAt ?? manifest.startedAt;
+                    if (now - abandonedAt > COMPLETED_RUN_RETENTION_MS) {
+                        try {
+                            deleteRunState(crewRoot, runId);
+                            cleanedUp.push(runId);
+                        }
+                        catch {
+                            // Best effort
+                        }
+                    }
+                }
+                break;
+        }
+    }
+    return { resumable, abandoned, cleanedUp };
+}
+/**
+ * Check whether a run has unfinished tasks.
+ * Used to decide whether to offer resume or start fresh.
+ */
+export function hasUnfinishedTasks(crewRoot, runId) {
+    const manifest = readManifest(crewRoot, runId);
+    if (!manifest)
+        return false;
+    if (manifest.status === "completed")
+        return false;
+    // Check if any agent status files indicate unfinished work
+    const agentsDir = path.join(crewRoot, "state", "runs", runId, "agents");
+    if (!fs.existsSync(agentsDir))
+        return manifest.status === "running";
+    // If there are agent IDs in the manifest, all are accounted for
+    // If fewer agents ran than expected, there is unfinished work
+    return true;
+}
+/**
+ * Get resumable agent IDs from a run.
+ * Returns a map of agentId → last known status.
+ */
+export function getResumableAgents(crewRoot, runId) {
+    const manifest = readManifest(crewRoot, runId);
+    const result = new Map();
+    if (!manifest)
+        return result;
+    for (const agentId of manifest.agentIds) {
+        const statusDir = path.join(crewRoot, "state", "runs", runId, "agents", agentId);
+        if (fs.existsSync(statusDir)) {
+            // Agent has state on disk — it can be resumed
+            result.set(agentId, "resumable");
+        }
+        else {
+            // Agent was never started
+            result.set(agentId, "not_started");
+        }
+    }
+    return result;
+}
+//# sourceMappingURL=recovery.js.map

package/dist/state/recovery.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recovery.js","sourceRoot":"","sources":["../../src/state/recovery.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EACL,eAAe,EACf,cAAc,EACd,YAAY,EACZ,cAAc,EACd,cAAc,EAEd,WAAW,GACZ,MAAM,kBAAkB,CAAC;AAE1B,8EAA8E;AAC9E,YAAY;AACZ,8EAA8E;AAE9E,0EAA0E;AAC1E,MAAM,0BAA0B,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,SAAS;AAErE,gFAAgF;AAChF,MAAM,sBAAsB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAe5D;;;;;;GAMG;AACH,MAAM,UAAU,WAAW,CAAC,GAAW;IACrC,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;IACtC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAErD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IACzD,CAAC;IAED,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,CAAC,CAAC;IACxC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,MAAM,SAAS,GAAkB,EAAE,CAAC;IACpC,MAAM,SAAS,GAAkB,EAAE,CAAC;IACpC,MAAM,SAAS,GAAa,EAAE,CAAC;IAE/B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC/C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,gCAAgC;YAChC,IAAI,CAAC;gBACH,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;gBAChC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;YAAC,MAAM,CAAC;gBACP,cAAc;YAChB,CAAC;YACD,SAAS;QACX,CAAC;QAED,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC;YACxB,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,sBAAsB;gBACtB,MAAM,GAAG,GAAG,GAAG,GAAG,QAAQ,CAAC,SAAS,CAAC;gBACrC,IAAI,GAAG,GAAG,sBAAsB,EAAE,CAAC;oBACjC,oBAAoB;oBACpB,MAAM,OAAO,GAAgB;wBAC3B,GAAG,QAAQ;wBACX,MAAM,EAAE,WAAW;wBACnB,WAAW,EAAE,GAAG;qBACjB,CAAC;oBACF,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBAClC,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE;wBAC3B,IAAI,EAAE,eAAe;wBACrB,MAAM,EAAE,OAAO;wBACf,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBACpC,CAAC,CAAC;oBACH,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC1B,CAAC;qBAAM,CAAC;oBACN,8BAA8B;oBAC9B,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC3B,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,WAAW,CAAC;YACjB,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,wBAAwB;gBACxB,MAAM,WAAW,GAAG,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAAC,SAAS,CAAC;gBAC/D,MAAM,GAAG,GAAG,GAAG,GAAG,WAAW,CAAC;gBAC9B,IAAI,GAAG,GAAG,0BAA0B,EAAE,CAAC;oBACrC,IAAI,CAAC;wBACH,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;wBAChC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACxB,CAAC;oBAAC,MAAM,CAAC;wBACP,cAAc;oBAChB,CAAC;gBACH,CAAC;gBACD,MAAM;YACR,CAAC;YACD,KAAK,WAAW;gBACd,kDAAkD;gBAClD,CAAC;oBACC,MAAM,WAAW,GAAG,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAAC,SAAS,CAAC;oBAC/D,IAAI,GAAG,GAAG,WAAW,GAAG,0BAA0B,EAAE,CAAC;wBACnD,IAAI,CAAC;4BACH,cAAc,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;4BAChC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBACxB,CAAC;wBAAC,MAAM,CAAC;4BACP,cAAc;wBAChB,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,MAAM;QACV,CAAC;IACH,CAAC;IAED,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;AAC7C,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAChC,QAAgB,EAChB,KAAa;IAEb,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC/C,IAAI,CAAC,QAAQ;QAAE,OAAO,KAAK,CAAC;IAC5B,IAAI,QAAQ,CAAC,MAAM,KAAK,WAAW;QAAE,OAAO,KAAK,CAAC;IAElD,2DAA2D;IAC3D,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CACzB,QAAQ,EACR,OAAO,EACP,MAAM,EACN,KAAK,EACL,QAAQ,CACT,CAAC;IACF,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,QAAQ,CAAC,MAAM,KAAK,SAAS,CAAC;IAEpE,gEAAgE;IAChE,8DAA8D;IAC9D,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAChC,QAAgB,EAChB,KAAa;IAEb,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC/C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEzC,IAAI,CAAC,QAAQ;QAAE,OAAO,MAAM,CAAC;IAE7B,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;QACxC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CACzB,QAAQ,EACR,OAAO,EACP,MAAM,EACN,KAAK,EACL,QAAQ,EACR,OAAO,CACR,CAAC;QACF,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7B,8CAA8C;YAC9C,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,0BAA0B;YAC1B,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/swarm/command.d.ts

@@ -0,0 +1,36 @@
+/**
+ * swarm/command — /swarm slash command handler.
+ *
+ * Supports:
+ *   /swarm on       — enable swarm mode (manual trigger)
+ *   /swarm off      — disable swarm mode
+ *   /swarm          — toggle swarm mode
+ *   /swarm <task>   — enable swarm mode + send task (one-shot)
+ *
+ * Ported from MoonshotAI/kimi-code's swarm command.
+ */
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import type { SwarmModeTrigger } from "./mode.js";
+export interface SwarmCommandHost {
+    /** The pi extension API. */
+    readonly pi: ExtensionAPI;
+    /** Current swarm mode state. */
+    swarmActive: boolean;
+    /** Set swarm mode state. */
+    setSwarmActive(active: boolean, trigger: SwarmModeTrigger): void;
+    /** Log / status helpers. */
+    /** Send a normal user input prompt. */
+    sendNormalUserInput(prompt: string): void;
+    /** Show a status message to the user. */
+    showStatus(message: string): void;
+    /** Show an error message to the user. */
+    showError(message: string): void;
+    /** Check if a model is configured. */
+    hasModel(): boolean;
+    /** Get the current permission mode. */
+    getPermissionMode(): string;
+    /** Set the permission mode. */
+    setPermissionMode(mode: string): Promise<void>;
+}
+export declare function registerSwarmCommand(pi: ExtensionAPI, host: SwarmCommandHost): void;
+//# sourceMappingURL=command.d.ts.map

package/dist/swarm/command.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"command.d.ts","sourceRoot":"","sources":["../../src/swarm/command.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,YAAY,EAA2B,MAAM,iCAAiC,CAAC;AAC7F,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAMlD,MAAM,WAAW,gBAAgB;IAC/B,4BAA4B;IAC5B,QAAQ,CAAC,EAAE,EAAE,YAAY,CAAC;IAC1B,gCAAgC;IAChC,WAAW,EAAE,OAAO,CAAC;IACrB,4BAA4B;IAC5B,cAAc,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,GAAG,IAAI,CAAC;IACjE,4BAA4B;IAC5B,uCAAuC;IACvC,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1C,yCAAyC;IACzC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,yCAAyC;IACzC,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,sCAAsC;IACtC,QAAQ,IAAI,OAAO,CAAC;IACpB,uCAAuC;IACvC,iBAAiB,IAAI,MAAM,CAAC;IAC5B,+BAA+B;IAC/B,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAChD;AAED,wBAAgB,oBAAoB,CAClC,EAAE,EAAE,YAAY,EAChB,IAAI,EAAE,gBAAgB,GACrB,IAAI,CAmCN"}

package/dist/swarm/command.js

@@ -0,0 +1,113 @@
+/**
+ * swarm/command — /swarm slash command handler.
+ *
+ * Supports:
+ *   /swarm on       — enable swarm mode (manual trigger)
+ *   /swarm off      — disable swarm mode
+ *   /swarm          — toggle swarm mode
+ *   /swarm <task>   — enable swarm mode + send task (one-shot)
+ *
+ * Ported from MoonshotAI/kimi-code's swarm command.
+ */
+export function registerSwarmCommand(pi, host) {
+    pi.registerCommand("swarm", {
+        description: "Turn swarm mode on/off, or run a one-shot swarm task. " +
+            "Usage: /swarm on|off, or /swarm <task>",
+        async handler(args, ctx) {
+            const prompt = args.trim();
+            const mode = swarmModeSubcommand(prompt);
+            // Subcommand: on / off
+            if (mode !== undefined) {
+                await applySwarmMode(host, mode, `/swarm ${prompt}`, ctx);
+                return;
+            }
+            // No args: toggle
+            if (prompt.length === 0) {
+                await applySwarmMode(host, !host.swarmActive, "/swarm", ctx);
+                return;
+            }
+            // Task mode: enable swarm + send prompt
+            if (!host.hasModel()) {
+                host.showError("No model configured. Please set a model first.");
+                return;
+            }
+            await startSwarmTask(host, prompt, ctx);
+        },
+    });
+}
+// ---------------------------------------------------------------------------
+// Command logic
+// ---------------------------------------------------------------------------
+async function applySwarmMode(host, enabled, commandText, ctx) {
+    if (enabled && host.swarmActive) {
+        host.showStatus("Swarm mode is already on.");
+        return;
+    }
+    if (!enabled && !host.swarmActive) {
+        host.showStatus("Swarm mode is already off.");
+        return;
+    }
+    if (enabled) {
+        // If in manual permission mode, ask for confirmation
+        if (host.getPermissionMode() === "manual") {
+            const confirmed = await ctx.ui?.confirm("Swarm Mode", "Swarm mode works best with auto or yolo permission. " +
+                "Switch to auto permission mode and enable swarm?");
+            if (!confirmed) {
+                host.showStatus("Swarm mode not enabled.");
+                return;
+            }
+            await host.setPermissionMode("auto");
+        }
+        host.setSwarmActive(true, "manual");
+        host.showStatus("Swarm mode enabled. AgentSwarm tool is now auto-approved.");
+        // Insert swarm mode marker (via pi.sendMessage for TUI rendering)
+        host.pi.sendMessage?.({
+            customType: "swarm:marker",
+            content: "active",
+            display: true,
+        });
+    }
+    else {
+        host.setSwarmActive(false, "manual");
+        host.showStatus("Swarm mode disabled.");
+        host.pi.sendMessage?.({
+            customType: "swarm:marker",
+            content: "inactive",
+            display: true,
+        });
+    }
+}
+async function startSwarmTask(host, prompt, ctx) {
+    // Enable swarm mode if not already active
+    if (!host.swarmActive) {
+        if (host.getPermissionMode() === "manual") {
+            const confirmed = await ctx.ui?.confirm("Swarm Task", "Starting a swarm task. Switch to auto permission mode?");
+            if (!confirmed) {
+                host.showStatus("Swarm task cancelled.");
+                return;
+            }
+            await host.setPermissionMode("auto");
+        }
+        host.setSwarmActive(true, "task");
+    }
+    // TUI marker
+    host.pi.sendMessage?.({
+        customType: "swarm:marker",
+        content: "active",
+        display: true,
+    });
+    // Send the prompt
+    host.sendNormalUserInput(prompt);
+}
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+function swarmModeSubcommand(input) {
+    const command = input.toLowerCase();
+    if (command === "on")
+        return true;
+    if (command === "off")
+        return false;
+    return undefined;
+}
+//# sourceMappingURL=command.js.map

package/dist/swarm/command.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"command.js","sourceRoot":"","sources":["../../src/swarm/command.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AA+BH,MAAM,UAAU,oBAAoB,CAClC,EAAgB,EAChB,IAAsB;IAEtB,EAAE,CAAC,eAAe,CAAC,OAAO,EAAE;QAC1B,WAAW,EACT,wDAAwD;YACxD,wCAAwC;QAC1C,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,GAA4B;YACtD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAC3B,MAAM,IAAI,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAEzC,uBAAuB;YACvB,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,MAAM,cAAc,CAAC,IAAI,EAAE,IAAI,EAAE,UAAU,MAAM,EAAE,EAAE,GAAG,CAAC,CAAC;gBAC1D,OAAO;YACT,CAAC;YAED,kBAAkB;YAClB,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,MAAM,cAAc,CAClB,IAAI,EACJ,CAAC,IAAI,CAAC,WAAW,EACjB,QAAQ,EACR,GAAG,CACJ,CAAC;gBACF,OAAO;YACT,CAAC;YAED,wCAAwC;YACxC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,CAAC;gBACrB,IAAI,CAAC,SAAS,CAAC,gDAAgD,CAAC,CAAC;gBACjE,OAAO;YACT,CAAC;YAED,MAAM,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1C,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED,8EAA8E;AAC9E,gBAAgB;AAChB,8EAA8E;AAE9E,KAAK,UAAU,cAAc,CAC3B,IAAsB,EACtB,OAAgB,EAChB,WAAmB,EACnB,GAA4B;IAE5B,IAAI,OAAO,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QAChC,IAAI,CAAC,UAAU,CAAC,2BAA2B,CAAC,CAAC;QAC7C,OAAO;IACT,CAAC;IACD,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QAClC,IAAI,CAAC,UAAU,CAAC,4BAA4B,CAAC,CAAC;QAC9C,OAAO;IACT,CAAC;IAED,IAAI,OAAO,EAAE,CAAC;QACZ,qDAAqD;QACrD,IAAI,IAAI,CAAC,iBAAiB,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC1C,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,EAAE,EAAE,OAAO,CACrC,YAAY,EACZ,sDAAsD;gBACpD,kDAAkD,CACrD,CAAC;YACF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC;gBAC3C,OAAO;YACT,CAAC;YACD,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACpC,IAAI,CAAC,UAAU,CAAC,2DAA2D,CAAC,CAAC;QAE7E,kEAAkE;QAClE,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;YACpB,UAAU,EAAE,cAAc;YAC1B,OAAO,EAAE,QAAQ;YACjB,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACrC,IAAI,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC;QAExC,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;YACpB,UAAU,EAAE,cAAc;YAC1B,OAAO,EAAE,UAAU;YACnB,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,IAAsB,EACtB,MAAc,EACd,GAA4B;IAE5B,0CAA0C;IAC1C,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACtB,IAAI,IAAI,CAAC,iBAAiB,EAAE,KAAK,QAAQ,EAAE,CAAC;YAC1C,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,EAAE,EAAE,OAAO,CACrC,YAAY,EACZ,wDAAwD,CACzD,CAAC;YACF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,IAAI,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC;gBACzC,OAAO;YACT,CAAC;YACD,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACvC,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACpC,CAAC;IAED,aAAa;IACb,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;QACpB,UAAU,EAAE,cAAc;QAC1B,OAAO,EAAE,QAAQ;QACjB,OAAO,EAAE,IAAI;KACd,CAAC,CAAC;IAEH,kBAAkB;IAClB,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;AACnC,CAAC;AAED,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E,SAAS,mBAAmB,CAC1B,KAAa;IAEb,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IACpC,IAAI,OAAO,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAClC,IAAI,OAAO,KAAK,KAAK;QAAE,OAAO,KAAK,CAAC;IACpC,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/swarm/mode.d.ts

@@ -0,0 +1,58 @@
+/**
+ * swarm/mode — SwarmMode state machine.
+ *
+ * Tracks whether swarm mode is active and manages the lifecycle
+ * of entering/exiting swarm mode with system reminders.
+ *
+ * Ported from MoonshotAI/kimi-code's SwarmMode.
+ */
+/** How swarm mode was triggered. */
+export type SwarmModeTrigger = "manual" | "task" | "tool";
+/**
+ * Minimal interface for the agent object that SwarmMode needs.
+ * The real agent is provided by the extension context.
+ */
+export interface SwarmModeAgent {
+    readonly context: {
+        /**
+         * Append a system reminder to the agent context.
+         * The reminder is injected into the model's context window.
+         */
+        appendSystemReminder(text: string, origin: {
+            kind: string;
+            variant: string;
+        }): void;
+        /**
+         * Pop a matched message from the context.
+         * Returns true if a message was removed.
+         */
+        popMatchedMessage(predicate: (origin: unknown) => boolean): boolean;
+    };
+    /** Emit a status update event. */
+    emitStatusUpdated(): void;
+    /** Log a record to the agent's event log. */
+    records: {
+        logRecord(entry: {
+            type: string;
+            trigger?: string;
+        }): void;
+    };
+}
+export declare class SwarmMode {
+    private readonly agent;
+    private active;
+    constructor(agent: SwarmModeAgent);
+    /** Enter swarm mode with the given trigger. No-op if already active. */
+    enter(trigger: SwarmModeTrigger): void;
+    /** Restore a previously saved swarm mode trigger (e.g., on session resume). */
+    restoreEnter(trigger: SwarmModeTrigger): void;
+    /** Exit swarm mode. No-op if not active. */
+    exit(): void;
+    /** Whether swarm mode is currently active. */
+    get isActive(): boolean;
+    /** Whether swarm mode should auto-exit after the current operation. */
+    get shouldAutoExit(): boolean;
+    /** The current trigger, or null if not active. */
+    get currentTrigger(): SwarmModeTrigger | null;
+}
+//# sourceMappingURL=mode.d.ts.map

package/dist/swarm/mode.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mode.d.ts","sourceRoot":"","sources":["../../src/swarm/mode.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,oCAAoC;AACpC,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,CAAC;AAE1D;;;GAGG;AACH,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,OAAO,EAAE;QAChB;;;WAGG;QACH,oBAAoB,CAClB,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,GACxC,IAAI,CAAC;QACR;;;WAGG;QACH,iBAAiB,CACf,SAAS,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,OAAO,GACtC,OAAO,CAAC;KACZ,CAAC;IACF,kCAAkC;IAClC,iBAAiB,IAAI,IAAI,CAAC;IAC1B,6CAA6C;IAC7C,OAAO,EAAE;QACP,SAAS,CAAC,KAAK,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,OAAO,CAAC,EAAE,MAAM,CAAA;SAAE,GAAG,IAAI,CAAC;KAC5D,CAAC;CACH;AA0BD,qBAAa,SAAS;IAGR,OAAO,CAAC,QAAQ,CAAC,KAAK;IAFlC,OAAO,CAAC,MAAM,CAAiC;gBAElB,KAAK,EAAE,cAAc;IAElD,wEAAwE;IACxE,KAAK,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;IAmBtC,+EAA+E;IAC/E,YAAY,CAAC,OAAO,EAAE,gBAAgB,GAAG,IAAI;IAI7C,4CAA4C;IAC5C,IAAI,IAAI,IAAI;IAgCZ,8CAA8C;IAC9C,IAAI,QAAQ,IAAI,OAAO,CAEtB;IAED,uEAAuE;IACvE,IAAI,cAAc,IAAI,OAAO,CAE5B;IAED,kDAAkD;IAClD,IAAI,cAAc,IAAI,gBAAgB,GAAG,IAAI,CAE5C;CACF"}

package/dist/swarm/mode.js

@@ -0,0 +1,87 @@
+/**
+ * swarm/mode — SwarmMode state machine.
+ *
+ * Tracks whether swarm mode is active and manages the lifecycle
+ * of entering/exiting swarm mode with system reminders.
+ *
+ * Ported from MoonshotAI/kimi-code's SwarmMode.
+ */
+// ---------------------------------------------------------------------------
+// System reminder templates (from kimi-code)
+// ---------------------------------------------------------------------------
+const SWARM_MODE_ENTER_REMINDER = `
+Swarm mode is now active. While swarm mode is active:
+- The AgentSwarm tool is auto-approved (no permission required).
+- You may call AgentSwarm freely to parallelise work.
+- When you finish the swarm work, call AgentSwarm with
+  resume_agent_ids to continue unfinished subagents.
+- Prefer splitting large tasks into many small, independent items
+  so the swarm stays efficient.
+`.trim();
+const SWARM_MODE_EXIT_REMINDER = `
+Swarm mode has been deactivated.
+- The AgentSwarm tool now requires permission approval again.
+- Continue working normally.
+`.trim();
+// ---------------------------------------------------------------------------
+// SwarmMode
+// ---------------------------------------------------------------------------
+export class SwarmMode {
+    agent;
+    active = null;
+    constructor(agent) {
+        this.agent = agent;
+    }
+    /** Enter swarm mode with the given trigger. No-op if already active. */
+    enter(trigger) {
+        if (this.active !== null)
+            return;
+        this.agent.records.logRecord({
+            type: "swarm_mode.enter",
+            trigger,
+        });
+        this.active = trigger;
+        if (trigger !== "tool") {
+            this.agent.context.appendSystemReminder(SWARM_MODE_ENTER_REMINDER, { kind: "injection", variant: "swarm_mode" });
+        }
+        this.agent.emitStatusUpdated();
+    }
+    /** Restore a previously saved swarm mode trigger (e.g., on session resume). */
+    restoreEnter(trigger) {
+        this.active = trigger;
+    }
+    /** Exit swarm mode. No-op if not active. */
+    exit() {
+        if (this.active === null)
+            return;
+        this.agent.records.logRecord({ type: "swarm_mode.exit" });
+        const trigger = this.active;
+        this.active = null;
+        this.agent.emitStatusUpdated();
+        if (trigger === "tool")
+            return;
+        // Try to pop the enter reminder first
+        if (this.agent.context.popMatchedMessage((origin) => {
+            const o = origin;
+            return (o?.kind === "injection" &&
+                o?.variant === "swarm_mode");
+        })) {
+            return;
+        }
+        // If we couldn't pop it, inject an exit reminder
+        this.agent.context.appendSystemReminder(SWARM_MODE_EXIT_REMINDER, { kind: "injection", variant: "swarm_mode_exit" });
+    }
+    /** Whether swarm mode is currently active. */
+    get isActive() {
+        return this.active !== null;
+    }
+    /** Whether swarm mode should auto-exit after the current operation. */
+    get shouldAutoExit() {
+        return this.active === "task" || this.active === "tool";
+    }
+    /** The current trigger, or null if not active. */
+    get currentTrigger() {
+        return this.active;
+    }
+}
+//# sourceMappingURL=mode.js.map

package/dist/swarm/mode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mode.js","sourceRoot":"","sources":["../../src/swarm/mode.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAmCH,8EAA8E;AAC9E,6CAA6C;AAC7C,8EAA8E;AAE9E,MAAM,yBAAyB,GAAG;;;;;;;;CAQjC,CAAC,IAAI,EAAE,CAAC;AAET,MAAM,wBAAwB,GAAG;;;;CAIhC,CAAC,IAAI,EAAE,CAAC;AAET,8EAA8E;AAC9E,YAAY;AACZ,8EAA8E;AAE9E,MAAM,OAAO,SAAS;IAGS;IAFrB,MAAM,GAA4B,IAAI,CAAC;IAE/C,YAA6B,KAAqB;QAArB,UAAK,GAAL,KAAK,CAAgB;IAAG,CAAC;IAEtD,wEAAwE;IACxE,KAAK,CAAC,OAAyB;QAC7B,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI;YAAE,OAAO;QAEjC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC;YAC3B,IAAI,EAAE,kBAAkB;YACxB,OAAO;SACR,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QAEtB,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;YACvB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,oBAAoB,CACrC,yBAAyB,EACzB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,EAAE,CAC7C,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;IACjC,CAAC;IAED,+EAA+E;IAC/E,YAAY,CAAC,OAAyB;QACpC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;IACxB,CAAC;IAED,4CAA4C;IAC5C,IAAI;QACF,IAAI,IAAI,CAAC,MAAM,KAAK,IAAI;YAAE,OAAO;QAEjC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,OAAO,KAAK,MAAM;YAAE,OAAO;QAE/B,sCAAsC;QACtC,IACE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAClC,CAAC,MAAe,EAAE,EAAE;YAClB,MAAM,CAAC,GAAG,MAAoD,CAAC;YAC/D,OAAO,CACL,CAAC,EAAE,IAAI,KAAK,WAAW;gBACvB,CAAC,EAAE,OAAO,KAAK,YAAY,CAC5B,CAAC;QACJ,CAAC,CACF,EACD,CAAC;YACD,OAAO;QACT,CAAC;QAED,iDAAiD;QACjD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,oBAAoB,CACrC,wBAAwB,EACxB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAClD,CAAC;IACJ,CAAC;IAED,8CAA8C;IAC9C,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC;IAC9B,CAAC;IAED,uEAAuE;IACvE,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,MAAM,KAAK,MAAM,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,CAAC;IAC1D,CAAC;IAED,kDAAkD;IAClD,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;CACF"}

package/dist/swarm/tool.d.ts

@@ -0,0 +1,11 @@
+/**
+ * swarm/tool — AgentSwarm tool registration.
+ *
+ * Registers the `AgentSwarm` tool that the LLM can call to launch
+ * multiple subagents from a shared prompt template.
+ *
+ * Ported from MoonshotAI/kimi-code's AgentSwarmTool.
+ */
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+export declare function registerAgentSwarmTool(pi: ExtensionAPI): void;
+//# sourceMappingURL=tool.d.ts.map

package/dist/swarm/tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../src/swarm/tool.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AA0CpE,wBAAgB,sBAAsB,CACpC,EAAE,EAAE,YAAY,GACf,IAAI,CAoIN"}