@imdigitalashish/zpi 0.1.1

package/dist/core/footer-data-provider.js

@@ -0,0 +1,134 @@
+import { existsSync, readFileSync, statSync, watch } from "fs";
+import { dirname, join, resolve } from "path";
+/**
+ * Find the git HEAD path by walking up from cwd.
+ * Handles both regular git repos (.git is a directory) and worktrees (.git is a file).
+ */
+function findGitHeadPath() {
+    let dir = process.cwd();
+    while (true) {
+        const gitPath = join(dir, ".git");
+        if (existsSync(gitPath)) {
+            try {
+                const stat = statSync(gitPath);
+                if (stat.isFile()) {
+                    const content = readFileSync(gitPath, "utf8").trim();
+                    if (content.startsWith("gitdir: ")) {
+                        const gitDir = content.slice(8);
+                        const headPath = resolve(dir, gitDir, "HEAD");
+                        if (existsSync(headPath))
+                            return headPath;
+                    }
+                }
+                else if (stat.isDirectory()) {
+                    const headPath = join(gitPath, "HEAD");
+                    if (existsSync(headPath))
+                        return headPath;
+                }
+            }
+            catch {
+                return null;
+            }
+        }
+        const parent = dirname(dir);
+        if (parent === dir)
+            return null;
+        dir = parent;
+    }
+}
+/**
+ * Provides git branch and extension statuses - data not otherwise accessible to extensions.
+ * Token stats, model info available via ctx.sessionManager and ctx.model.
+ */
+export class FooterDataProvider {
+    extensionStatuses = new Map();
+    cachedBranch = undefined;
+    gitWatcher = null;
+    branchChangeCallbacks = new Set();
+    availableProviderCount = 0;
+    constructor() {
+        this.setupGitWatcher();
+    }
+    /** Current git branch, null if not in repo, "detached" if detached HEAD */
+    getGitBranch() {
+        if (this.cachedBranch !== undefined)
+            return this.cachedBranch;
+        try {
+            const gitHeadPath = findGitHeadPath();
+            if (!gitHeadPath) {
+                this.cachedBranch = null;
+                return null;
+            }
+            const content = readFileSync(gitHeadPath, "utf8").trim();
+            this.cachedBranch = content.startsWith("ref: refs/heads/") ? content.slice(16) : "detached";
+        }
+        catch {
+            this.cachedBranch = null;
+        }
+        return this.cachedBranch;
+    }
+    /** Extension status texts set via ctx.ui.setStatus() */
+    getExtensionStatuses() {
+        return this.extensionStatuses;
+    }
+    /** Subscribe to git branch changes. Returns unsubscribe function. */
+    onBranchChange(callback) {
+        this.branchChangeCallbacks.add(callback);
+        return () => this.branchChangeCallbacks.delete(callback);
+    }
+    /** Internal: set extension status */
+    setExtensionStatus(key, text) {
+        if (text === undefined) {
+            this.extensionStatuses.delete(key);
+        }
+        else {
+            this.extensionStatuses.set(key, text);
+        }
+    }
+    /** Internal: clear extension statuses */
+    clearExtensionStatuses() {
+        this.extensionStatuses.clear();
+    }
+    /** Number of unique providers with available models (for footer display) */
+    getAvailableProviderCount() {
+        return this.availableProviderCount;
+    }
+    /** Internal: update available provider count */
+    setAvailableProviderCount(count) {
+        this.availableProviderCount = count;
+    }
+    /** Internal: cleanup */
+    dispose() {
+        if (this.gitWatcher) {
+            this.gitWatcher.close();
+            this.gitWatcher = null;
+        }
+        this.branchChangeCallbacks.clear();
+    }
+    setupGitWatcher() {
+        if (this.gitWatcher) {
+            this.gitWatcher.close();
+            this.gitWatcher = null;
+        }
+        const gitHeadPath = findGitHeadPath();
+        if (!gitHeadPath)
+            return;
+        // Watch the directory containing HEAD, not HEAD itself.
+        // Git uses atomic writes (write temp, rename over HEAD), which changes the inode.
+        // fs.watch on a file stops working after the inode changes.
+        const gitDir = dirname(gitHeadPath);
+        try {
+            this.gitWatcher = watch(gitDir, (_eventType, filename) => {
+                if (filename === "HEAD") {
+                    this.cachedBranch = undefined;
+                    for (const cb of this.branchChangeCallbacks)
+                        cb();
+                }
+            });
+        }
+        catch {
+            // Silently fail if we can't watch
+        }
+    }
+}
+//# sourceMappingURL=footer-data-provider.js.map

package/dist/core/footer-data-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"footer-data-provider.js","sourceRoot":"","sources":["../../src/core/footer-data-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAkB,YAAY,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,IAAI,CAAC;AAC/E,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAE9C;;;GAGG;AACH,SAAS,eAAe,GAAkB;IACzC,IAAI,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IACxB,OAAO,IAAI,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC;gBACJ,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAC/B,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;oBACnB,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;oBACrD,IAAI,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;wBACpC,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;wBAChC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;wBAC9C,IAAI,UAAU,CAAC,QAAQ,CAAC;4BAAE,OAAO,QAAQ,CAAC;oBAC3C,CAAC;gBACF,CAAC;qBAAM,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;oBAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;oBACvC,IAAI,UAAU,CAAC,QAAQ,CAAC;wBAAE,OAAO,QAAQ,CAAC;gBAC3C,CAAC;YACF,CAAC;YAAC,MAAM,CAAC;gBACR,OAAO,IAAI,CAAC;YACb,CAAC;QACF,CAAC;QACD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,MAAM,KAAK,GAAG;YAAE,OAAO,IAAI,CAAC;QAChC,GAAG,GAAG,MAAM,CAAC;IACd,CAAC;AAAA,CACD;AAED;;;GAGG;AACH,MAAM,OAAO,kBAAkB;IACtB,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC9C,YAAY,GAA8B,SAAS,CAAC;IACpD,UAAU,GAAqB,IAAI,CAAC;IACpC,qBAAqB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC9C,sBAAsB,GAAG,CAAC,CAAC;IAEnC,cAAc;QACb,IAAI,CAAC,eAAe,EAAE,CAAC;IAAA,CACvB;IAED,2EAA2E;IAC3E,YAAY,GAAkB;QAC7B,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS;YAAE,OAAO,IAAI,CAAC,YAAY,CAAC;QAE9D,IAAI,CAAC;YACJ,MAAM,WAAW,GAAG,eAAe,EAAE,CAAC;YACtC,IAAI,CAAC,WAAW,EAAE,CAAC;gBAClB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;gBACzB,OAAO,IAAI,CAAC;YACb,CAAC;YACD,MAAM,OAAO,GAAG,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;YACzD,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QAC7F,CAAC;QAAC,MAAM,CAAC;YACR,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC1B,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC;IAAA,CACzB;IAED,wDAAwD;IACxD,oBAAoB,GAAgC;QACnD,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAAA,CAC9B;IAED,qEAAqE;IACrE,cAAc,CAAC,QAAoB,EAAc;QAChD,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACzC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAAA,CACzD;IAED,qCAAqC;IACrC,kBAAkB,CAAC,GAAW,EAAE,IAAwB,EAAQ;QAC/D,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACvC,CAAC;IAAA,CACD;IAED,yCAAyC;IACzC,sBAAsB,GAAS;QAC9B,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;IAAA,CAC/B;IAED,4EAA4E;IAC5E,yBAAyB,GAAW;QACnC,OAAO,IAAI,CAAC,sBAAsB,CAAC;IAAA,CACnC;IAED,gDAAgD;IAChD,yBAAyB,CAAC,KAAa,EAAQ;QAC9C,IAAI,CAAC,sBAAsB,GAAG,KAAK,CAAC;IAAA,CACpC;IAED,wBAAwB;IACxB,OAAO,GAAS;QACf,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACxB,CAAC;QACD,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;IAAA,CACnC;IAEO,eAAe,GAAS;QAC/B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACxB,CAAC;QAED,MAAM,WAAW,GAAG,eAAe,EAAE,CAAC;QACtC,IAAI,CAAC,WAAW;YAAE,OAAO;QAEzB,wDAAwD;QACxD,kFAAkF;QAClF,4DAA4D;QAC5D,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;QAEpC,IAAI,CAAC;YACJ,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,CAAC;gBACzD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;oBACzB,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;oBAC9B,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,qBAAqB;wBAAE,EAAE,EAAE,CAAC;gBACnD,CAAC;YAAA,CACD,CAAC,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACR,kCAAkC;QACnC,CAAC;IAAA,CACD;CACD","sourcesContent":["import { existsSync, type FSWatcher, readFileSync, statSync, watch } from \"fs\";\r\nimport { dirname, join, resolve } from \"path\";\r\n\r\n/**\r\n * Find the git HEAD path by walking up from cwd.\r\n * Handles both regular git repos (.git is a directory) and worktrees (.git is a file).\r\n */\r\nfunction findGitHeadPath(): string | null {\r\n\tlet dir = process.cwd();\r\n\twhile (true) {\r\n\t\tconst gitPath = join(dir, \".git\");\r\n\t\tif (existsSync(gitPath)) {\r\n\t\t\ttry {\r\n\t\t\t\tconst stat = statSync(gitPath);\r\n\t\t\t\tif (stat.isFile()) {\r\n\t\t\t\t\tconst content = readFileSync(gitPath, \"utf8\").trim();\r\n\t\t\t\t\tif (content.startsWith(\"gitdir: \")) {\r\n\t\t\t\t\t\tconst gitDir = content.slice(8);\r\n\t\t\t\t\t\tconst headPath = resolve(dir, gitDir, \"HEAD\");\r\n\t\t\t\t\t\tif (existsSync(headPath)) return headPath;\r\n\t\t\t\t\t}\r\n\t\t\t\t} else if (stat.isDirectory()) {\r\n\t\t\t\t\tconst headPath = join(gitPath, \"HEAD\");\r\n\t\t\t\t\tif (existsSync(headPath)) return headPath;\r\n\t\t\t\t}\r\n\t\t\t} catch {\r\n\t\t\t\treturn null;\r\n\t\t\t}\r\n\t\t}\r\n\t\tconst parent = dirname(dir);\r\n\t\tif (parent === dir) return null;\r\n\t\tdir = parent;\r\n\t}\r\n}\r\n\r\n/**\r\n * Provides git branch and extension statuses - data not otherwise accessible to extensions.\r\n * Token stats, model info available via ctx.sessionManager and ctx.model.\r\n */\r\nexport class FooterDataProvider {\r\n\tprivate extensionStatuses = new Map<string, string>();\r\n\tprivate cachedBranch: string | null | undefined = undefined;\r\n\tprivate gitWatcher: FSWatcher | null = null;\r\n\tprivate branchChangeCallbacks = new Set<() => void>();\r\n\tprivate availableProviderCount = 0;\r\n\r\n\tconstructor() {\r\n\t\tthis.setupGitWatcher();\r\n\t}\r\n\r\n\t/** Current git branch, null if not in repo, \"detached\" if detached HEAD */\r\n\tgetGitBranch(): string | null {\r\n\t\tif (this.cachedBranch !== undefined) return this.cachedBranch;\r\n\r\n\t\ttry {\r\n\t\t\tconst gitHeadPath = findGitHeadPath();\r\n\t\t\tif (!gitHeadPath) {\r\n\t\t\t\tthis.cachedBranch = null;\r\n\t\t\t\treturn null;\r\n\t\t\t}\r\n\t\t\tconst content = readFileSync(gitHeadPath, \"utf8\").trim();\r\n\t\t\tthis.cachedBranch = content.startsWith(\"ref: refs/heads/\") ? content.slice(16) : \"detached\";\r\n\t\t} catch {\r\n\t\t\tthis.cachedBranch = null;\r\n\t\t}\r\n\t\treturn this.cachedBranch;\r\n\t}\r\n\r\n\t/** Extension status texts set via ctx.ui.setStatus() */\r\n\tgetExtensionStatuses(): ReadonlyMap<string, string> {\r\n\t\treturn this.extensionStatuses;\r\n\t}\r\n\r\n\t/** Subscribe to git branch changes. Returns unsubscribe function. */\r\n\tonBranchChange(callback: () => void): () => void {\r\n\t\tthis.branchChangeCallbacks.add(callback);\r\n\t\treturn () => this.branchChangeCallbacks.delete(callback);\r\n\t}\r\n\r\n\t/** Internal: set extension status */\r\n\tsetExtensionStatus(key: string, text: string | undefined): void {\r\n\t\tif (text === undefined) {\r\n\t\t\tthis.extensionStatuses.delete(key);\r\n\t\t} else {\r\n\t\t\tthis.extensionStatuses.set(key, text);\r\n\t\t}\r\n\t}\r\n\r\n\t/** Internal: clear extension statuses */\r\n\tclearExtensionStatuses(): void {\r\n\t\tthis.extensionStatuses.clear();\r\n\t}\r\n\r\n\t/** Number of unique providers with available models (for footer display) */\r\n\tgetAvailableProviderCount(): number {\r\n\t\treturn this.availableProviderCount;\r\n\t}\r\n\r\n\t/** Internal: update available provider count */\r\n\tsetAvailableProviderCount(count: number): void {\r\n\t\tthis.availableProviderCount = count;\r\n\t}\r\n\r\n\t/** Internal: cleanup */\r\n\tdispose(): void {\r\n\t\tif (this.gitWatcher) {\r\n\t\t\tthis.gitWatcher.close();\r\n\t\t\tthis.gitWatcher = null;\r\n\t\t}\r\n\t\tthis.branchChangeCallbacks.clear();\r\n\t}\r\n\r\n\tprivate setupGitWatcher(): void {\r\n\t\tif (this.gitWatcher) {\r\n\t\t\tthis.gitWatcher.close();\r\n\t\t\tthis.gitWatcher = null;\r\n\t\t}\r\n\r\n\t\tconst gitHeadPath = findGitHeadPath();\r\n\t\tif (!gitHeadPath) return;\r\n\r\n\t\t// Watch the directory containing HEAD, not HEAD itself.\r\n\t\t// Git uses atomic writes (write temp, rename over HEAD), which changes the inode.\r\n\t\t// fs.watch on a file stops working after the inode changes.\r\n\t\tconst gitDir = dirname(gitHeadPath);\r\n\r\n\t\ttry {\r\n\t\t\tthis.gitWatcher = watch(gitDir, (_eventType, filename) => {\r\n\t\t\t\tif (filename === \"HEAD\") {\r\n\t\t\t\t\tthis.cachedBranch = undefined;\r\n\t\t\t\t\tfor (const cb of this.branchChangeCallbacks) cb();\r\n\t\t\t\t}\r\n\t\t\t});\r\n\t\t} catch {\r\n\t\t\t// Silently fail if we can't watch\r\n\t\t}\r\n\t}\r\n}\r\n\r\n/** Read-only view for extensions - excludes setExtensionStatus, setAvailableProviderCount and dispose */\r\nexport type ReadonlyFooterDataProvider = Pick<\r\n\tFooterDataProvider,\r\n\t\"getGitBranch\" | \"getExtensionStatuses\" | \"getAvailableProviderCount\" | \"onBranchChange\"\r\n>;\r\n"]}

package/dist/core/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Core modules shared between all run modes.
+ */
+export { AgentSession, type AgentSessionConfig, type AgentSessionEvent, type AgentSessionEventListener, type ModelCycleResult, type PromptOptions, type SessionStats, } from "./agent-session.js";
+export { type BashExecutorOptions, type BashResult, executeBash, executeBashWithOperations } from "./bash-executor.js";
+export type { CompactionResult } from "./compaction/index.js";
+export { createEventBus, type EventBus, type EventBusController } from "./event-bus.js";
+export { type AgentEndEvent, type AgentStartEvent, type AgentToolResult, type AgentToolUpdateCallback, type BeforeAgentStartEvent, type ContextEvent, discoverAndLoadExtensions, type ExecOptions, type ExecResult, type Extension, type ExtensionAPI, type ExtensionCommandContext, type ExtensionContext, type ExtensionError, type ExtensionEvent, type ExtensionFactory, type ExtensionFlag, type ExtensionHandler, ExtensionRunner, type ExtensionShortcut, type ExtensionUIContext, type LoadExtensionsResult, type MessageRenderer, type RegisteredCommand, type SessionBeforeCompactEvent, type SessionBeforeForkEvent, type SessionBeforeSwitchEvent, type SessionBeforeTreeEvent, type SessionCompactEvent, type SessionForkEvent, type SessionShutdownEvent, type SessionStartEvent, type SessionSwitchEvent, type SessionTreeEvent, type ToolCallEvent, type ToolDefinition, type ToolRenderResultOptions, type ToolResultEvent, type TurnEndEvent, type TurnStartEvent, wrapToolsWithExtensions, } from "./extensions/index.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/core/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACN,YAAY,EACZ,KAAK,kBAAkB,EACvB,KAAK,iBAAiB,EACtB,KAAK,yBAAyB,EAC9B,KAAK,gBAAgB,EACrB,KAAK,aAAa,EAClB,KAAK,YAAY,GACjB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,KAAK,mBAAmB,EAAE,KAAK,UAAU,EAAE,WAAW,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AACvH,YAAY,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,KAAK,QAAQ,EAAE,KAAK,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAGxF,OAAO,EACN,KAAK,aAAa,EAClB,KAAK,eAAe,EACpB,KAAK,eAAe,EACpB,KAAK,uBAAuB,EAC5B,KAAK,qBAAqB,EAC1B,KAAK,YAAY,EACjB,yBAAyB,EACzB,KAAK,WAAW,EAChB,KAAK,UAAU,EACf,KAAK,SAAS,EACd,KAAK,YAAY,EACjB,KAAK,uBAAuB,EAC5B,KAAK,gBAAgB,EACrB,KAAK,cAAc,EACnB,KAAK,cAAc,EACnB,KAAK,gBAAgB,EACrB,KAAK,aAAa,EAClB,KAAK,gBAAgB,EACrB,eAAe,EACf,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EACvB,KAAK,oBAAoB,EACzB,KAAK,eAAe,EACpB,KAAK,iBAAiB,EACtB,KAAK,yBAAyB,EAC9B,KAAK,sBAAsB,EAC3B,KAAK,wBAAwB,EAC7B,KAAK,sBAAsB,EAC3B,KAAK,mBAAmB,EACxB,KAAK,gBAAgB,EACrB,KAAK,oBAAoB,EACzB,KAAK,iBAAiB,EACtB,KAAK,kBAAkB,EACvB,KAAK,gBAAgB,EACrB,KAAK,aAAa,EAClB,KAAK,cAAc,EACnB,KAAK,uBAAuB,EAC5B,KAAK,eAAe,EACpB,KAAK,YAAY,EACjB,KAAK,cAAc,EACnB,uBAAuB,GACvB,MAAM,uBAAuB,CAAC","sourcesContent":["/**\r\n * Core modules shared between all run modes.\r\n */\r\n\r\nexport {\r\n\tAgentSession,\r\n\ttype AgentSessionConfig,\r\n\ttype AgentSessionEvent,\r\n\ttype AgentSessionEventListener,\r\n\ttype ModelCycleResult,\r\n\ttype PromptOptions,\r\n\ttype SessionStats,\r\n} from \"./agent-session.js\";\r\nexport { type BashExecutorOptions, type BashResult, executeBash, executeBashWithOperations } from \"./bash-executor.js\";\r\nexport type { CompactionResult } from \"./compaction/index.js\";\r\nexport { createEventBus, type EventBus, type EventBusController } from \"./event-bus.js\";\r\n\r\n// Extensions system\r\nexport {\r\n\ttype AgentEndEvent,\r\n\ttype AgentStartEvent,\r\n\ttype AgentToolResult,\r\n\ttype AgentToolUpdateCallback,\r\n\ttype BeforeAgentStartEvent,\r\n\ttype ContextEvent,\r\n\tdiscoverAndLoadExtensions,\r\n\ttype ExecOptions,\r\n\ttype ExecResult,\r\n\ttype Extension,\r\n\ttype ExtensionAPI,\r\n\ttype ExtensionCommandContext,\r\n\ttype ExtensionContext,\r\n\ttype ExtensionError,\r\n\ttype ExtensionEvent,\r\n\ttype ExtensionFactory,\r\n\ttype ExtensionFlag,\r\n\ttype ExtensionHandler,\r\n\tExtensionRunner,\r\n\ttype ExtensionShortcut,\r\n\ttype ExtensionUIContext,\r\n\ttype LoadExtensionsResult,\r\n\ttype MessageRenderer,\r\n\ttype RegisteredCommand,\r\n\ttype SessionBeforeCompactEvent,\r\n\ttype SessionBeforeForkEvent,\r\n\ttype SessionBeforeSwitchEvent,\r\n\ttype SessionBeforeTreeEvent,\r\n\ttype SessionCompactEvent,\r\n\ttype SessionForkEvent,\r\n\ttype SessionShutdownEvent,\r\n\ttype SessionStartEvent,\r\n\ttype SessionSwitchEvent,\r\n\ttype SessionTreeEvent,\r\n\ttype ToolCallEvent,\r\n\ttype ToolDefinition,\r\n\ttype ToolRenderResultOptions,\r\n\ttype ToolResultEvent,\r\n\ttype TurnEndEvent,\r\n\ttype TurnStartEvent,\r\n\twrapToolsWithExtensions,\r\n} from \"./extensions/index.js\";\r\n"]}

package/dist/core/index.js

@@ -0,0 +1,9 @@
+/**
+ * Core modules shared between all run modes.
+ */
+export { AgentSession, } from "./agent-session.js";
+export { executeBash, executeBashWithOperations } from "./bash-executor.js";
+export { createEventBus } from "./event-bus.js";
+// Extensions system
+export { discoverAndLoadExtensions, ExtensionRunner, wrapToolsWithExtensions, } from "./extensions/index.js";
+//# sourceMappingURL=index.js.map

package/dist/core/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/core/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACN,YAAY,GAOZ,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAA6C,WAAW,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAEvH,OAAO,EAAE,cAAc,EAA0C,MAAM,gBAAgB,CAAC;AAExF,oBAAoB;AACpB,OAAO,EAON,yBAAyB,EAYzB,eAAe,EAsBf,uBAAuB,GACvB,MAAM,uBAAuB,CAAC","sourcesContent":["/**\r\n * Core modules shared between all run modes.\r\n */\r\n\r\nexport {\r\n\tAgentSession,\r\n\ttype AgentSessionConfig,\r\n\ttype AgentSessionEvent,\r\n\ttype AgentSessionEventListener,\r\n\ttype ModelCycleResult,\r\n\ttype PromptOptions,\r\n\ttype SessionStats,\r\n} from \"./agent-session.js\";\r\nexport { type BashExecutorOptions, type BashResult, executeBash, executeBashWithOperations } from \"./bash-executor.js\";\r\nexport type { CompactionResult } from \"./compaction/index.js\";\r\nexport { createEventBus, type EventBus, type EventBusController } from \"./event-bus.js\";\r\n\r\n// Extensions system\r\nexport {\r\n\ttype AgentEndEvent,\r\n\ttype AgentStartEvent,\r\n\ttype AgentToolResult,\r\n\ttype AgentToolUpdateCallback,\r\n\ttype BeforeAgentStartEvent,\r\n\ttype ContextEvent,\r\n\tdiscoverAndLoadExtensions,\r\n\ttype ExecOptions,\r\n\ttype ExecResult,\r\n\ttype Extension,\r\n\ttype ExtensionAPI,\r\n\ttype ExtensionCommandContext,\r\n\ttype ExtensionContext,\r\n\ttype ExtensionError,\r\n\ttype ExtensionEvent,\r\n\ttype ExtensionFactory,\r\n\ttype ExtensionFlag,\r\n\ttype ExtensionHandler,\r\n\tExtensionRunner,\r\n\ttype ExtensionShortcut,\r\n\ttype ExtensionUIContext,\r\n\ttype LoadExtensionsResult,\r\n\ttype MessageRenderer,\r\n\ttype RegisteredCommand,\r\n\ttype SessionBeforeCompactEvent,\r\n\ttype SessionBeforeForkEvent,\r\n\ttype SessionBeforeSwitchEvent,\r\n\ttype SessionBeforeTreeEvent,\r\n\ttype SessionCompactEvent,\r\n\ttype SessionForkEvent,\r\n\ttype SessionShutdownEvent,\r\n\ttype SessionStartEvent,\r\n\ttype SessionSwitchEvent,\r\n\ttype SessionTreeEvent,\r\n\ttype ToolCallEvent,\r\n\ttype ToolDefinition,\r\n\ttype ToolRenderResultOptions,\r\n\ttype ToolResultEvent,\r\n\ttype TurnEndEvent,\r\n\ttype TurnStartEvent,\r\n\twrapToolsWithExtensions,\r\n} from \"./extensions/index.js\";\r\n"]}

package/dist/core/keybindings.d.ts

@@ -0,0 +1,55 @@
+import { type EditorAction, type KeyId } from "@mariozechner/pi-tui";
+/**
+ * Application-level actions (coding agent specific).
+ */
+export type AppAction = "interrupt" | "clear" | "exit" | "suspend" | "cycleThinkingLevel" | "cycleModelForward" | "cycleModelBackward" | "selectModel" | "expandTools" | "toggleThinking" | "toggleSessionNamedFilter" | "externalEditor" | "followUp" | "dequeue" | "pasteImage" | "newSession" | "tree" | "fork" | "resume";
+/**
+ * All configurable actions.
+ */
+export type KeyAction = AppAction | EditorAction;
+/**
+ * Full keybindings configuration (app + editor actions).
+ */
+export type KeybindingsConfig = {
+    [K in KeyAction]?: KeyId | KeyId[];
+};
+/**
+ * Default application keybindings.
+ */
+export declare const DEFAULT_APP_KEYBINDINGS: Record<AppAction, KeyId | KeyId[]>;
+/**
+ * All default keybindings (app + editor).
+ */
+export declare const DEFAULT_KEYBINDINGS: Required<KeybindingsConfig>;
+/**
+ * Manages all keybindings (app + editor).
+ */
+export declare class KeybindingsManager {
+    private config;
+    private appActionToKeys;
+    private constructor();
+    /**
+     * Create from config file and set up editor keybindings.
+     */
+    static create(agentDir?: string): KeybindingsManager;
+    /**
+     * Create in-memory.
+     */
+    static inMemory(config?: KeybindingsConfig): KeybindingsManager;
+    private static loadFromFile;
+    private buildMaps;
+    /**
+     * Check if input matches an app action.
+     */
+    matches(data: string, action: AppAction): boolean;
+    /**
+     * Get keys bound to an app action.
+     */
+    getKeys(action: AppAction): KeyId[];
+    /**
+     * Get the full effective config.
+     */
+    getEffectiveConfig(): Required<KeybindingsConfig>;
+}
+export type { EditorAction, KeyId };
+//# sourceMappingURL=keybindings.d.ts.map

package/dist/core/keybindings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"keybindings.d.ts","sourceRoot":"","sources":["../../src/core/keybindings.ts"],"names":[],"mappings":"AAAA,OAAO,EAEN,KAAK,YAAY,EAGjB,KAAK,KAAK,EAGV,MAAM,sBAAsB,CAAC;AAK9B;;GAEG;AACH,MAAM,MAAM,SAAS,GAClB,WAAW,GACX,OAAO,GACP,MAAM,GACN,SAAS,GACT,oBAAoB,GACpB,mBAAmB,GACnB,oBAAoB,GACpB,aAAa,GACb,aAAa,GACb,gBAAgB,GAChB,0BAA0B,GAC1B,gBAAgB,GAChB,UAAU,GACV,SAAS,GACT,YAAY,GACZ,YAAY,GACZ,MAAM,GACN,MAAM,GACN,QAAQ,CAAC;AAEZ;;GAEG;AACH,MAAM,MAAM,SAAS,GAAG,SAAS,GAAG,YAAY,CAAC;AAEjD;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG;KAC9B,CAAC,IAAI,SAAS,CAAC,CAAC,EAAE,KAAK,GAAG,KAAK,EAAE;CAClC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,uBAAuB,EAAE,MAAM,CAAC,SAAS,EAAE,KAAK,GAAG,KAAK,EAAE,CAoBtE,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,mBAAmB,EAAE,QAAQ,CAAC,iBAAiB,CAG3D,CAAC;AA6BF;;GAEG;AACH,qBAAa,kBAAkB;IAC9B,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,eAAe,CAA0B;IAEjD,OAAO,eAIN;IAED;;OAEG;IACH,MAAM,CAAC,MAAM,CAAC,QAAQ,GAAE,MAAsB,GAAG,kBAAkB,CAgBlE;IAED;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAE,iBAAsB,GAAG,kBAAkB,CAElE;IAED,OAAO,CAAC,MAAM,CAAC,YAAY;IAS3B,OAAO,CAAC,SAAS;IAiBjB;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,GAAG,OAAO,CAOhD;IAED;;OAEG;IACH,OAAO,CAAC,MAAM,EAAE,SAAS,GAAG,KAAK,EAAE,CAElC;IAED;;OAEG;IACH,kBAAkB,IAAI,QAAQ,CAAC,iBAAiB,CAAC,CAQhD;CACD;AAGD,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC","sourcesContent":["import {\r\n\tDEFAULT_EDITOR_KEYBINDINGS,\r\n\ttype EditorAction,\r\n\ttype EditorKeybindingsConfig,\r\n\tEditorKeybindingsManager,\r\n\ttype KeyId,\r\n\tmatchesKey,\r\n\tsetEditorKeybindings,\r\n} from \"@mariozechner/pi-tui\";\r\nimport { existsSync, readFileSync } from \"fs\";\r\nimport { join } from \"path\";\r\nimport { getAgentDir } from \"../config.js\";\r\n\r\n/**\r\n * Application-level actions (coding agent specific).\r\n */\r\nexport type AppAction =\r\n\t| \"interrupt\"\r\n\t| \"clear\"\r\n\t| \"exit\"\r\n\t| \"suspend\"\r\n\t| \"cycleThinkingLevel\"\r\n\t| \"cycleModelForward\"\r\n\t| \"cycleModelBackward\"\r\n\t| \"selectModel\"\r\n\t| \"expandTools\"\r\n\t| \"toggleThinking\"\r\n\t| \"toggleSessionNamedFilter\"\r\n\t| \"externalEditor\"\r\n\t| \"followUp\"\r\n\t| \"dequeue\"\r\n\t| \"pasteImage\"\r\n\t| \"newSession\"\r\n\t| \"tree\"\r\n\t| \"fork\"\r\n\t| \"resume\";\r\n\r\n/**\r\n * All configurable actions.\r\n */\r\nexport type KeyAction = AppAction | EditorAction;\r\n\r\n/**\r\n * Full keybindings configuration (app + editor actions).\r\n */\r\nexport type KeybindingsConfig = {\r\n\t[K in KeyAction]?: KeyId | KeyId[];\r\n};\r\n\r\n/**\r\n * Default application keybindings.\r\n */\r\nexport const DEFAULT_APP_KEYBINDINGS: Record<AppAction, KeyId | KeyId[]> = {\r\n\tinterrupt: \"escape\",\r\n\tclear: \"ctrl+c\",\r\n\texit: \"ctrl+d\",\r\n\tsuspend: \"ctrl+z\",\r\n\tcycleThinkingLevel: \"shift+tab\",\r\n\tcycleModelForward: \"ctrl+p\",\r\n\tcycleModelBackward: \"shift+ctrl+p\",\r\n\tselectModel: \"ctrl+l\",\r\n\texpandTools: \"ctrl+o\",\r\n\ttoggleThinking: \"ctrl+t\",\r\n\ttoggleSessionNamedFilter: \"ctrl+n\",\r\n\texternalEditor: \"ctrl+g\",\r\n\tfollowUp: \"alt+enter\",\r\n\tdequeue: \"alt+up\",\r\n\tpasteImage: \"ctrl+v\",\r\n\tnewSession: [],\r\n\ttree: [],\r\n\tfork: [],\r\n\tresume: [],\r\n};\r\n\r\n/**\r\n * All default keybindings (app + editor).\r\n */\r\nexport const DEFAULT_KEYBINDINGS: Required<KeybindingsConfig> = {\r\n\t...DEFAULT_EDITOR_KEYBINDINGS,\r\n\t...DEFAULT_APP_KEYBINDINGS,\r\n};\r\n\r\n// App actions list for type checking\r\nconst APP_ACTIONS: AppAction[] = [\r\n\t\"interrupt\",\r\n\t\"clear\",\r\n\t\"exit\",\r\n\t\"suspend\",\r\n\t\"cycleThinkingLevel\",\r\n\t\"cycleModelForward\",\r\n\t\"cycleModelBackward\",\r\n\t\"selectModel\",\r\n\t\"expandTools\",\r\n\t\"toggleThinking\",\r\n\t\"toggleSessionNamedFilter\",\r\n\t\"externalEditor\",\r\n\t\"followUp\",\r\n\t\"dequeue\",\r\n\t\"pasteImage\",\r\n\t\"newSession\",\r\n\t\"tree\",\r\n\t\"fork\",\r\n\t\"resume\",\r\n];\r\n\r\nfunction isAppAction(action: string): action is AppAction {\r\n\treturn APP_ACTIONS.includes(action as AppAction);\r\n}\r\n\r\n/**\r\n * Manages all keybindings (app + editor).\r\n */\r\nexport class KeybindingsManager {\r\n\tprivate config: KeybindingsConfig;\r\n\tprivate appActionToKeys: Map<AppAction, KeyId[]>;\r\n\r\n\tprivate constructor(config: KeybindingsConfig) {\r\n\t\tthis.config = config;\r\n\t\tthis.appActionToKeys = new Map();\r\n\t\tthis.buildMaps();\r\n\t}\r\n\r\n\t/**\r\n\t * Create from config file and set up editor keybindings.\r\n\t */\r\n\tstatic create(agentDir: string = getAgentDir()): KeybindingsManager {\r\n\t\tconst configPath = join(agentDir, \"keybindings.json\");\r\n\t\tconst config = KeybindingsManager.loadFromFile(configPath);\r\n\t\tconst manager = new KeybindingsManager(config);\r\n\r\n\t\t// Set up editor keybindings globally\r\n\t\t// Include both editor actions and expandTools (shared between app and editor)\r\n\t\tconst editorConfig: EditorKeybindingsConfig = {};\r\n\t\tfor (const [action, keys] of Object.entries(config)) {\r\n\t\t\tif (!isAppAction(action) || action === \"expandTools\") {\r\n\t\t\t\teditorConfig[action as EditorAction] = keys;\r\n\t\t\t}\r\n\t\t}\r\n\t\tsetEditorKeybindings(new EditorKeybindingsManager(editorConfig));\r\n\r\n\t\treturn manager;\r\n\t}\r\n\r\n\t/**\r\n\t * Create in-memory.\r\n\t */\r\n\tstatic inMemory(config: KeybindingsConfig = {}): KeybindingsManager {\r\n\t\treturn new KeybindingsManager(config);\r\n\t}\r\n\r\n\tprivate static loadFromFile(path: string): KeybindingsConfig {\r\n\t\tif (!existsSync(path)) return {};\r\n\t\ttry {\r\n\t\t\treturn JSON.parse(readFileSync(path, \"utf-8\"));\r\n\t\t} catch {\r\n\t\t\treturn {};\r\n\t\t}\r\n\t}\r\n\r\n\tprivate buildMaps(): void {\r\n\t\tthis.appActionToKeys.clear();\r\n\r\n\t\t// Set defaults for app actions\r\n\t\tfor (const [action, keys] of Object.entries(DEFAULT_APP_KEYBINDINGS)) {\r\n\t\t\tconst keyArray = Array.isArray(keys) ? keys : [keys];\r\n\t\t\tthis.appActionToKeys.set(action as AppAction, [...keyArray]);\r\n\t\t}\r\n\r\n\t\t// Override with user config (app actions only)\r\n\t\tfor (const [action, keys] of Object.entries(this.config)) {\r\n\t\t\tif (keys === undefined || !isAppAction(action)) continue;\r\n\t\t\tconst keyArray = Array.isArray(keys) ? keys : [keys];\r\n\t\t\tthis.appActionToKeys.set(action, keyArray);\r\n\t\t}\r\n\t}\r\n\r\n\t/**\r\n\t * Check if input matches an app action.\r\n\t */\r\n\tmatches(data: string, action: AppAction): boolean {\r\n\t\tconst keys = this.appActionToKeys.get(action);\r\n\t\tif (!keys) return false;\r\n\t\tfor (const key of keys) {\r\n\t\t\tif (matchesKey(data, key)) return true;\r\n\t\t}\r\n\t\treturn false;\r\n\t}\r\n\r\n\t/**\r\n\t * Get keys bound to an app action.\r\n\t */\r\n\tgetKeys(action: AppAction): KeyId[] {\r\n\t\treturn this.appActionToKeys.get(action) ?? [];\r\n\t}\r\n\r\n\t/**\r\n\t * Get the full effective config.\r\n\t */\r\n\tgetEffectiveConfig(): Required<KeybindingsConfig> {\r\n\t\tconst result = { ...DEFAULT_KEYBINDINGS };\r\n\t\tfor (const [action, keys] of Object.entries(this.config)) {\r\n\t\t\tif (keys !== undefined) {\r\n\t\t\t\t(result as KeybindingsConfig)[action as KeyAction] = keys;\r\n\t\t\t}\r\n\t\t}\r\n\t\treturn result;\r\n\t}\r\n}\r\n\r\n// Re-export for convenience\r\nexport type { EditorAction, KeyId };\r\n"]}

package/dist/core/keybindings.js

@@ -0,0 +1,153 @@
+import { DEFAULT_EDITOR_KEYBINDINGS, EditorKeybindingsManager, matchesKey, setEditorKeybindings, } from "@mariozechner/pi-tui";
+import { existsSync, readFileSync } from "fs";
+import { join } from "path";
+import { getAgentDir } from "../config.js";
+/**
+ * Default application keybindings.
+ */
+export const DEFAULT_APP_KEYBINDINGS = {
+    interrupt: "escape",
+    clear: "ctrl+c",
+    exit: "ctrl+d",
+    suspend: "ctrl+z",
+    cycleThinkingLevel: "shift+tab",
+    cycleModelForward: "ctrl+p",
+    cycleModelBackward: "shift+ctrl+p",
+    selectModel: "ctrl+l",
+    expandTools: "ctrl+o",
+    toggleThinking: "ctrl+t",
+    toggleSessionNamedFilter: "ctrl+n",
+    externalEditor: "ctrl+g",
+    followUp: "alt+enter",
+    dequeue: "alt+up",
+    pasteImage: "ctrl+v",
+    newSession: [],
+    tree: [],
+    fork: [],
+    resume: [],
+};
+/**
+ * All default keybindings (app + editor).
+ */
+export const DEFAULT_KEYBINDINGS = {
+    ...DEFAULT_EDITOR_KEYBINDINGS,
+    ...DEFAULT_APP_KEYBINDINGS,
+};
+// App actions list for type checking
+const APP_ACTIONS = [
+    "interrupt",
+    "clear",
+    "exit",
+    "suspend",
+    "cycleThinkingLevel",
+    "cycleModelForward",
+    "cycleModelBackward",
+    "selectModel",
+    "expandTools",
+    "toggleThinking",
+    "toggleSessionNamedFilter",
+    "externalEditor",
+    "followUp",
+    "dequeue",
+    "pasteImage",
+    "newSession",
+    "tree",
+    "fork",
+    "resume",
+];
+function isAppAction(action) {
+    return APP_ACTIONS.includes(action);
+}
+/**
+ * Manages all keybindings (app + editor).
+ */
+export class KeybindingsManager {
+    config;
+    appActionToKeys;
+    constructor(config) {
+        this.config = config;
+        this.appActionToKeys = new Map();
+        this.buildMaps();
+    }
+    /**
+     * Create from config file and set up editor keybindings.
+     */
+    static create(agentDir = getAgentDir()) {
+        const configPath = join(agentDir, "keybindings.json");
+        const config = KeybindingsManager.loadFromFile(configPath);
+        const manager = new KeybindingsManager(config);
+        // Set up editor keybindings globally
+        // Include both editor actions and expandTools (shared between app and editor)
+        const editorConfig = {};
+        for (const [action, keys] of Object.entries(config)) {
+            if (!isAppAction(action) || action === "expandTools") {
+                editorConfig[action] = keys;
+            }
+        }
+        setEditorKeybindings(new EditorKeybindingsManager(editorConfig));
+        return manager;
+    }
+    /**
+     * Create in-memory.
+     */
+    static inMemory(config = {}) {
+        return new KeybindingsManager(config);
+    }
+    static loadFromFile(path) {
+        if (!existsSync(path))
+            return {};
+        try {
+            return JSON.parse(readFileSync(path, "utf-8"));
+        }
+        catch {
+            return {};
+        }
+    }
+    buildMaps() {
+        this.appActionToKeys.clear();
+        // Set defaults for app actions
+        for (const [action, keys] of Object.entries(DEFAULT_APP_KEYBINDINGS)) {
+            const keyArray = Array.isArray(keys) ? keys : [keys];
+            this.appActionToKeys.set(action, [...keyArray]);
+        }
+        // Override with user config (app actions only)
+        for (const [action, keys] of Object.entries(this.config)) {
+            if (keys === undefined || !isAppAction(action))
+                continue;
+            const keyArray = Array.isArray(keys) ? keys : [keys];
+            this.appActionToKeys.set(action, keyArray);
+        }
+    }
+    /**
+     * Check if input matches an app action.
+     */
+    matches(data, action) {
+        const keys = this.appActionToKeys.get(action);
+        if (!keys)
+            return false;
+        for (const key of keys) {
+            if (matchesKey(data, key))
+                return true;
+        }
+        return false;
+    }
+    /**
+     * Get keys bound to an app action.
+     */
+    getKeys(action) {
+        return this.appActionToKeys.get(action) ?? [];
+    }
+    /**
+     * Get the full effective config.
+     */
+    getEffectiveConfig() {
+        const result = { ...DEFAULT_KEYBINDINGS };
+        for (const [action, keys] of Object.entries(this.config)) {
+            if (keys !== undefined) {
+                result[action] = keys;
+            }
+        }
+        return result;
+    }
+}
+//# sourceMappingURL=keybindings.js.map

package/dist/core/keybindings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"keybindings.js","sourceRoot":"","sources":["../../src/core/keybindings.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,0BAA0B,EAG1B,wBAAwB,EAExB,UAAU,EACV,oBAAoB,GACpB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAsC3C;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAuC;IAC1E,SAAS,EAAE,QAAQ;IACnB,KAAK,EAAE,QAAQ;IACf,IAAI,EAAE,QAAQ;IACd,OAAO,EAAE,QAAQ;IACjB,kBAAkB,EAAE,WAAW;IAC/B,iBAAiB,EAAE,QAAQ;IAC3B,kBAAkB,EAAE,cAAc;IAClC,WAAW,EAAE,QAAQ;IACrB,WAAW,EAAE,QAAQ;IACrB,cAAc,EAAE,QAAQ;IACxB,wBAAwB,EAAE,QAAQ;IAClC,cAAc,EAAE,QAAQ;IACxB,QAAQ,EAAE,WAAW;IACrB,OAAO,EAAE,QAAQ;IACjB,UAAU,EAAE,QAAQ;IACpB,UAAU,EAAE,EAAE;IACd,IAAI,EAAE,EAAE;IACR,IAAI,EAAE,EAAE;IACR,MAAM,EAAE,EAAE;CACV,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAgC;IAC/D,GAAG,0BAA0B;IAC7B,GAAG,uBAAuB;CAC1B,CAAC;AAEF,qCAAqC;AACrC,MAAM,WAAW,GAAgB;IAChC,WAAW;IACX,OAAO;IACP,MAAM;IACN,SAAS;IACT,oBAAoB;IACpB,mBAAmB;IACnB,oBAAoB;IACpB,aAAa;IACb,aAAa;IACb,gBAAgB;IAChB,0BAA0B;IAC1B,gBAAgB;IAChB,UAAU;IACV,SAAS;IACT,YAAY;IACZ,YAAY;IACZ,MAAM;IACN,MAAM;IACN,QAAQ;CACR,CAAC;AAEF,SAAS,WAAW,CAAC,MAAc,EAAuB;IACzD,OAAO,WAAW,CAAC,QAAQ,CAAC,MAAmB,CAAC,CAAC;AAAA,CACjD;AAED;;GAEG;AACH,MAAM,OAAO,kBAAkB;IACtB,MAAM,CAAoB;IAC1B,eAAe,CAA0B;IAEjD,YAAoB,MAAyB,EAAE;QAC9C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,IAAI,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,SAAS,EAAE,CAAC;IAAA,CACjB;IAED;;OAEG;IACH,MAAM,CAAC,MAAM,CAAC,QAAQ,GAAW,WAAW,EAAE,EAAsB;QACnE,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,kBAAkB,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAC3D,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAE/C,qCAAqC;QACrC,8EAA8E;QAC9E,MAAM,YAAY,GAA4B,EAAE,CAAC;QACjD,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,MAAM,KAAK,aAAa,EAAE,CAAC;gBACtD,YAAY,CAAC,MAAsB,CAAC,GAAG,IAAI,CAAC;YAC7C,CAAC;QACF,CAAC;QACD,oBAAoB,CAAC,IAAI,wBAAwB,CAAC,YAAY,CAAC,CAAC,CAAC;QAEjE,OAAO,OAAO,CAAC;IAAA,CACf;IAED;;OAEG;IACH,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAsB,EAAE,EAAsB;QACnE,OAAO,IAAI,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAAA,CACtC;IAEO,MAAM,CAAC,YAAY,CAAC,IAAY,EAAqB;QAC5D,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;YAAE,OAAO,EAAE,CAAC;QACjC,IAAI,CAAC;YACJ,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,EAAE,CAAC;QACX,CAAC;IAAA,CACD;IAEO,SAAS,GAAS;QACzB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAE7B,+BAA+B;QAC/B,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,EAAE,CAAC;YACtE,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAmB,EAAE,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC;QAC9D,CAAC;QAED,+CAA+C;QAC/C,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1D,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBAAE,SAAS;YACzD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC5C,CAAC;IAAA,CACD;IAED;;OAEG;IACH,OAAO,CAAC,IAAY,EAAE,MAAiB,EAAW;QACjD,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI;YAAE,OAAO,KAAK,CAAC;QACxB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACxB,IAAI,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC;QACxC,CAAC;QACD,OAAO,KAAK,CAAC;IAAA,CACb;IAED;;OAEG;IACH,OAAO,CAAC,MAAiB,EAAW;QACnC,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IAAA,CAC9C;IAED;;OAEG;IACH,kBAAkB,GAAgC;QACjD,MAAM,MAAM,GAAG,EAAE,GAAG,mBAAmB,EAAE,CAAC;QAC1C,KAAK,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1D,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,MAA4B,CAAC,MAAmB,CAAC,GAAG,IAAI,CAAC;YAC3D,CAAC;QACF,CAAC;QACD,OAAO,MAAM,CAAC;IAAA,CACd;CACD","sourcesContent":["import {\r\n\tDEFAULT_EDITOR_KEYBINDINGS,\r\n\ttype EditorAction,\r\n\ttype EditorKeybindingsConfig,\r\n\tEditorKeybindingsManager,\r\n\ttype KeyId,\r\n\tmatchesKey,\r\n\tsetEditorKeybindings,\r\n} from \"@mariozechner/pi-tui\";\r\nimport { existsSync, readFileSync } from \"fs\";\r\nimport { join } from \"path\";\r\nimport { getAgentDir } from \"../config.js\";\r\n\r\n/**\r\n * Application-level actions (coding agent specific).\r\n */\r\nexport type AppAction =\r\n\t| \"interrupt\"\r\n\t| \"clear\"\r\n\t| \"exit\"\r\n\t| \"suspend\"\r\n\t| \"cycleThinkingLevel\"\r\n\t| \"cycleModelForward\"\r\n\t| \"cycleModelBackward\"\r\n\t| \"selectModel\"\r\n\t| \"expandTools\"\r\n\t| \"toggleThinking\"\r\n\t| \"toggleSessionNamedFilter\"\r\n\t| \"externalEditor\"\r\n\t| \"followUp\"\r\n\t| \"dequeue\"\r\n\t| \"pasteImage\"\r\n\t| \"newSession\"\r\n\t| \"tree\"\r\n\t| \"fork\"\r\n\t| \"resume\";\r\n\r\n/**\r\n * All configurable actions.\r\n */\r\nexport type KeyAction = AppAction | EditorAction;\r\n\r\n/**\r\n * Full keybindings configuration (app + editor actions).\r\n */\r\nexport type KeybindingsConfig = {\r\n\t[K in KeyAction]?: KeyId | KeyId[];\r\n};\r\n\r\n/**\r\n * Default application keybindings.\r\n */\r\nexport const DEFAULT_APP_KEYBINDINGS: Record<AppAction, KeyId | KeyId[]> = {\r\n\tinterrupt: \"escape\",\r\n\tclear: \"ctrl+c\",\r\n\texit: \"ctrl+d\",\r\n\tsuspend: \"ctrl+z\",\r\n\tcycleThinkingLevel: \"shift+tab\",\r\n\tcycleModelForward: \"ctrl+p\",\r\n\tcycleModelBackward: \"shift+ctrl+p\",\r\n\tselectModel: \"ctrl+l\",\r\n\texpandTools: \"ctrl+o\",\r\n\ttoggleThinking: \"ctrl+t\",\r\n\ttoggleSessionNamedFilter: \"ctrl+n\",\r\n\texternalEditor: \"ctrl+g\",\r\n\tfollowUp: \"alt+enter\",\r\n\tdequeue: \"alt+up\",\r\n\tpasteImage: \"ctrl+v\",\r\n\tnewSession: [],\r\n\ttree: [],\r\n\tfork: [],\r\n\tresume: [],\r\n};\r\n\r\n/**\r\n * All default keybindings (app + editor).\r\n */\r\nexport const DEFAULT_KEYBINDINGS: Required<KeybindingsConfig> = {\r\n\t...DEFAULT_EDITOR_KEYBINDINGS,\r\n\t...DEFAULT_APP_KEYBINDINGS,\r\n};\r\n\r\n// App actions list for type checking\r\nconst APP_ACTIONS: AppAction[] = [\r\n\t\"interrupt\",\r\n\t\"clear\",\r\n\t\"exit\",\r\n\t\"suspend\",\r\n\t\"cycleThinkingLevel\",\r\n\t\"cycleModelForward\",\r\n\t\"cycleModelBackward\",\r\n\t\"selectModel\",\r\n\t\"expandTools\",\r\n\t\"toggleThinking\",\r\n\t\"toggleSessionNamedFilter\",\r\n\t\"externalEditor\",\r\n\t\"followUp\",\r\n\t\"dequeue\",\r\n\t\"pasteImage\",\r\n\t\"newSession\",\r\n\t\"tree\",\r\n\t\"fork\",\r\n\t\"resume\",\r\n];\r\n\r\nfunction isAppAction(action: string): action is AppAction {\r\n\treturn APP_ACTIONS.includes(action as AppAction);\r\n}\r\n\r\n/**\r\n * Manages all keybindings (app + editor).\r\n */\r\nexport class KeybindingsManager {\r\n\tprivate config: KeybindingsConfig;\r\n\tprivate appActionToKeys: Map<AppAction, KeyId[]>;\r\n\r\n\tprivate constructor(config: KeybindingsConfig) {\r\n\t\tthis.config = config;\r\n\t\tthis.appActionToKeys = new Map();\r\n\t\tthis.buildMaps();\r\n\t}\r\n\r\n\t/**\r\n\t * Create from config file and set up editor keybindings.\r\n\t */\r\n\tstatic create(agentDir: string = getAgentDir()): KeybindingsManager {\r\n\t\tconst configPath = join(agentDir, \"keybindings.json\");\r\n\t\tconst config = KeybindingsManager.loadFromFile(configPath);\r\n\t\tconst manager = new KeybindingsManager(config);\r\n\r\n\t\t// Set up editor keybindings globally\r\n\t\t// Include both editor actions and expandTools (shared between app and editor)\r\n\t\tconst editorConfig: EditorKeybindingsConfig = {};\r\n\t\tfor (const [action, keys] of Object.entries(config)) {\r\n\t\t\tif (!isAppAction(action) || action === \"expandTools\") {\r\n\t\t\t\teditorConfig[action as EditorAction] = keys;\r\n\t\t\t}\r\n\t\t}\r\n\t\tsetEditorKeybindings(new EditorKeybindingsManager(editorConfig));\r\n\r\n\t\treturn manager;\r\n\t}\r\n\r\n\t/**\r\n\t * Create in-memory.\r\n\t */\r\n\tstatic inMemory(config: KeybindingsConfig = {}): KeybindingsManager {\r\n\t\treturn new KeybindingsManager(config);\r\n\t}\r\n\r\n\tprivate static loadFromFile(path: string): KeybindingsConfig {\r\n\t\tif (!existsSync(path)) return {};\r\n\t\ttry {\r\n\t\t\treturn JSON.parse(readFileSync(path, \"utf-8\"));\r\n\t\t} catch {\r\n\t\t\treturn {};\r\n\t\t}\r\n\t}\r\n\r\n\tprivate buildMaps(): void {\r\n\t\tthis.appActionToKeys.clear();\r\n\r\n\t\t// Set defaults for app actions\r\n\t\tfor (const [action, keys] of Object.entries(DEFAULT_APP_KEYBINDINGS)) {\r\n\t\t\tconst keyArray = Array.isArray(keys) ? keys : [keys];\r\n\t\t\tthis.appActionToKeys.set(action as AppAction, [...keyArray]);\r\n\t\t}\r\n\r\n\t\t// Override with user config (app actions only)\r\n\t\tfor (const [action, keys] of Object.entries(this.config)) {\r\n\t\t\tif (keys === undefined || !isAppAction(action)) continue;\r\n\t\t\tconst keyArray = Array.isArray(keys) ? keys : [keys];\r\n\t\t\tthis.appActionToKeys.set(action, keyArray);\r\n\t\t}\r\n\t}\r\n\r\n\t/**\r\n\t * Check if input matches an app action.\r\n\t */\r\n\tmatches(data: string, action: AppAction): boolean {\r\n\t\tconst keys = this.appActionToKeys.get(action);\r\n\t\tif (!keys) return false;\r\n\t\tfor (const key of keys) {\r\n\t\t\tif (matchesKey(data, key)) return true;\r\n\t\t}\r\n\t\treturn false;\r\n\t}\r\n\r\n\t/**\r\n\t * Get keys bound to an app action.\r\n\t */\r\n\tgetKeys(action: AppAction): KeyId[] {\r\n\t\treturn this.appActionToKeys.get(action) ?? [];\r\n\t}\r\n\r\n\t/**\r\n\t * Get the full effective config.\r\n\t */\r\n\tgetEffectiveConfig(): Required<KeybindingsConfig> {\r\n\t\tconst result = { ...DEFAULT_KEYBINDINGS };\r\n\t\tfor (const [action, keys] of Object.entries(this.config)) {\r\n\t\t\tif (keys !== undefined) {\r\n\t\t\t\t(result as KeybindingsConfig)[action as KeyAction] = keys;\r\n\t\t\t}\r\n\t\t}\r\n\t\treturn result;\r\n\t}\r\n}\r\n\r\n// Re-export for convenience\r\nexport type { EditorAction, KeyId };\r\n"]}

package/dist/core/memory.d.ts

@@ -0,0 +1,64 @@
+/**
+ * Persistent memory system — procedural, episodic, and semantic memories
+ * stored per-project in .pi/memory/ as JSON files.
+ */
+export interface ProceduralMemory {
+    id: string;
+    name: string;
+    trigger: string;
+    steps: string[];
+    tags: string[];
+    created: string;
+    updated: string;
+    sourceSession: string;
+}
+export interface EpisodicMemory {
+    id: string;
+    summary: string;
+    details: string[];
+    reflection?: {
+        mistakes: string[];
+        lessons: string[];
+    };
+    tags: string[];
+    date: string;
+    sourceSession: string;
+}
+export interface SemanticMemory {
+    id: string;
+    category: "preference" | "architecture" | "convention" | "fact";
+    text: string;
+    tags: string[];
+    created: string;
+    sourceSession: string;
+}
+export interface MemoryStore<T> {
+    memories: T[];
+}
+export type MemoryType = "procedural" | "episodic" | "semantic";
+export declare function getMemoryDir(cwd: string): string;
+export declare function loadStore<T>(cwd: string, filename: string): MemoryStore<T>;
+export declare function saveStore<T>(cwd: string, filename: string, store: MemoryStore<T>): void;
+export declare function generateId(prefix: string, store: MemoryStore<{
+    id: string;
+}>): string;
+export declare function nowISO(): string;
+export declare function getSessionId(): string;
+export declare function getStoreFile(type: MemoryType): string;
+export declare function searchMemories<T extends {
+    tags: string[];
+}>(memories: T[], query: string, getText: (m: T) => string): T[];
+export declare function compactEpisodicMemories(memories: EpisodicMemory[]): EpisodicMemory[];
+export declare function formatProceduralForDisplay(m: ProceduralMemory): string;
+export declare function formatEpisodicForDisplay(m: EpisodicMemory): string;
+export declare function formatSemanticForDisplay(m: SemanticMemory): string;
+export declare function buildMemoryPromptSection(cwd: string): string;
+export declare function formatTimestamp(ts: number): string;
+export interface MemoryCounts {
+    semantic: number;
+    procedural: number;
+    episodic: number;
+    total: number;
+}
+export declare function getMemoryCounts(cwd: string): MemoryCounts;
+//# sourceMappingURL=memory.d.ts.map

package/dist/core/memory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../src/core/memory.ts"],"names":[],"mappings":"AAAA;;;GAGG;AASH,MAAM,WAAW,gBAAgB;IAChC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,UAAU,CAAC,EAAE;QACZ,QAAQ,EAAE,MAAM,EAAE,CAAC;QACnB,OAAO,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC;IACF,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,cAAc;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,YAAY,GAAG,cAAc,GAAG,YAAY,GAAG,MAAM,CAAC;IAChE,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,WAAW,CAAC,CAAC;IAC7B,QAAQ,EAAE,CAAC,EAAE,CAAC;CACd;AAED,MAAM,MAAM,UAAU,GAAG,YAAY,GAAG,UAAU,GAAG,UAAU,CAAC;AAMhE,wBAAgB,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAEhD;AAUD,wBAAgB,SAAS,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,WAAW,CAAC,CAAC,CAAC,CAW1E;AAED,wBAAgB,SAAS,CAAC,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,IAAI,CAIvF;AAED,wBAAgB,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,CAAC;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,CAAC,GAAG,MAAM,CAOrF;AAED,wBAAgB,MAAM,IAAI,MAAM,CAE/B;AAED,wBAAgB,YAAY,IAAI,MAAM,CAErC;AAQD,wBAAgB,YAAY,CAAC,IAAI,EAAE,UAAU,GAAG,MAAM,CAErD;AAMD,wBAAgB,cAAc,CAAC,CAAC,SAAS;IAAE,IAAI,EAAE,MAAM,EAAE,CAAA;CAAE,EAC1D,QAAQ,EAAE,CAAC,EAAE,EACb,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,MAAM,GACvB,CAAC,EAAE,CASL;AAMD,wBAAgB,uBAAuB,CAAC,QAAQ,EAAE,cAAc,EAAE,GAAG,cAAc,EAAE,CAgDpF;AAMD,wBAAgB,0BAA0B,CAAC,CAAC,EAAE,gBAAgB,GAAG,MAAM,CAGtE;AAED,wBAAgB,wBAAwB,CAAC,CAAC,EAAE,cAAc,GAAG,MAAM,CAclE;AAED,wBAAgB,wBAAwB,CAAC,CAAC,EAAE,cAAc,GAAG,MAAM,CAElE;AAMD,wBAAgB,wBAAwB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CA8E5D;AAeD,wBAAgB,eAAe,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,CAWlD;AAMD,MAAM,WAAW,YAAY;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;CACd;AAED,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,YAAY,CAKzD","sourcesContent":["/**\r\n * Persistent memory system — procedural, episodic, and semantic memories\r\n * stored per-project in .pi/memory/ as JSON files.\r\n */\r\n\r\nimport { existsSync, mkdirSync, readFileSync, writeFileSync } from \"node:fs\";\r\nimport { join } from \"node:path\";\r\n\r\n// ---------------------------------------------------------------------------\r\n// Types\r\n// ---------------------------------------------------------------------------\r\n\r\nexport interface ProceduralMemory {\r\n\tid: string;\r\n\tname: string;\r\n\ttrigger: string;\r\n\tsteps: string[];\r\n\ttags: string[];\r\n\tcreated: string;\r\n\tupdated: string;\r\n\tsourceSession: string;\r\n}\r\n\r\nexport interface EpisodicMemory {\r\n\tid: string;\r\n\tsummary: string;\r\n\tdetails: string[];\r\n\treflection?: {\r\n\t\tmistakes: string[];\r\n\t\tlessons: string[];\r\n\t};\r\n\ttags: string[];\r\n\tdate: string;\r\n\tsourceSession: string;\r\n}\r\n\r\nexport interface SemanticMemory {\r\n\tid: string;\r\n\tcategory: \"preference\" | \"architecture\" | \"convention\" | \"fact\";\r\n\ttext: string;\r\n\ttags: string[];\r\n\tcreated: string;\r\n\tsourceSession: string;\r\n}\r\n\r\nexport interface MemoryStore<T> {\r\n\tmemories: T[];\r\n}\r\n\r\nexport type MemoryType = \"procedural\" | \"episodic\" | \"semantic\";\r\n\r\n// ---------------------------------------------------------------------------\r\n// Storage\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function getMemoryDir(cwd: string): string {\r\n\treturn join(cwd, \".pi\", \"memory\");\r\n}\r\n\r\nfunction ensureMemoryDir(cwd: string): string {\r\n\tconst dir = getMemoryDir(cwd);\r\n\tif (!existsSync(dir)) {\r\n\t\tmkdirSync(dir, { recursive: true });\r\n\t}\r\n\treturn dir;\r\n}\r\n\r\nexport function loadStore<T>(cwd: string, filename: string): MemoryStore<T> {\r\n\tconst filepath = join(getMemoryDir(cwd), filename);\r\n\tif (!existsSync(filepath)) {\r\n\t\treturn { memories: [] };\r\n\t}\r\n\ttry {\r\n\t\tconst raw = readFileSync(filepath, \"utf-8\");\r\n\t\treturn JSON.parse(raw) as MemoryStore<T>;\r\n\t} catch {\r\n\t\treturn { memories: [] };\r\n\t}\r\n}\r\n\r\nexport function saveStore<T>(cwd: string, filename: string, store: MemoryStore<T>): void {\r\n\tconst dir = ensureMemoryDir(cwd);\r\n\tconst filepath = join(dir, filename);\r\n\twriteFileSync(filepath, JSON.stringify(store, null, 2), \"utf-8\");\r\n}\r\n\r\nexport function generateId(prefix: string, store: MemoryStore<{ id: string }>): string {\r\n\tlet max = 0;\r\n\tfor (const m of store.memories) {\r\n\t\tconst num = parseInt(m.id.replace(`${prefix}_`, \"\"), 10);\r\n\t\tif (!Number.isNaN(num) && num > max) max = num;\r\n\t}\r\n\treturn `${prefix}_${String(max + 1).padStart(3, \"0\")}`;\r\n}\r\n\r\nexport function nowISO(): string {\r\n\treturn new Date().toISOString();\r\n}\r\n\r\nexport function getSessionId(): string {\r\n\treturn `session_${Date.now()}`;\r\n}\r\n\r\nconst STORE_FILES: Record<MemoryType, string> = {\r\n\tprocedural: \"procedural.json\",\r\n\tepisodic: \"episodic.json\",\r\n\tsemantic: \"semantic.json\",\r\n};\r\n\r\nexport function getStoreFile(type: MemoryType): string {\r\n\treturn STORE_FILES[type];\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Search\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function searchMemories<T extends { tags: string[] }>(\r\n\tmemories: T[],\r\n\tquery: string,\r\n\tgetText: (m: T) => string,\r\n): T[] {\r\n\tconst lower = query.toLowerCase();\r\n\tconst queryWords = lower.split(/\\s+/).filter(Boolean);\r\n\treturn memories.filter((m) => {\r\n\t\tconst text = getText(m).toLowerCase();\r\n\t\tconst tagText = m.tags.join(\" \").toLowerCase();\r\n\t\tconst combined = `${text} ${tagText}`;\r\n\t\treturn queryWords.every((w) => combined.includes(w));\r\n\t});\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Compaction\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function compactEpisodicMemories(memories: EpisodicMemory[]): EpisodicMemory[] {\r\n\tif (memories.length <= 1) return memories;\r\n\r\n\tconst tagGroups = new Map<string, EpisodicMemory[]>();\r\n\tfor (const m of memories) {\r\n\t\tlet placed = false;\r\n\t\tfor (const [key, group] of tagGroups) {\r\n\t\t\tconst groupTags = new Set(key.split(\",\"));\r\n\t\t\tif (m.tags.some((t) => groupTags.has(t))) {\r\n\t\t\t\tgroup.push(m);\r\n\t\t\t\tconst mergedTags = new Set([...groupTags, ...m.tags]);\r\n\t\t\t\ttagGroups.delete(key);\r\n\t\t\t\ttagGroups.set([...mergedTags].join(\",\"), group);\r\n\t\t\t\tplaced = true;\r\n\t\t\t\tbreak;\r\n\t\t\t}\r\n\t\t}\r\n\t\tif (!placed) {\r\n\t\t\ttagGroups.set(m.tags.join(\",\") || m.id, [m]);\r\n\t\t}\r\n\t}\r\n\r\n\tconst compacted: EpisodicMemory[] = [];\r\n\tfor (const [, group] of tagGroups) {\r\n\t\tif (group.length <= 1) {\r\n\t\t\tcompacted.push(...group);\r\n\t\t\tcontinue;\r\n\t\t}\r\n\t\tconst allDetails = group.flatMap((m) => m.details);\r\n\t\tconst allTags = [...new Set(group.flatMap((m) => m.tags))];\r\n\t\tconst allMistakes = group.flatMap((m) => m.reflection?.mistakes ?? []);\r\n\t\tconst allLessons = group.flatMap((m) => m.reflection?.lessons ?? []);\r\n\t\tconst summaries = group.map((m) => m.summary);\r\n\r\n\t\tcompacted.push({\r\n\t\t\tid: group[0].id,\r\n\t\t\tsummary: `Consolidated: ${summaries.join(\"; \")}`,\r\n\t\t\tdetails: [...new Set(allDetails)],\r\n\t\t\treflection:\r\n\t\t\t\tallMistakes.length > 0 || allLessons.length > 0\r\n\t\t\t\t\t? { mistakes: [...new Set(allMistakes)], lessons: [...new Set(allLessons)] }\r\n\t\t\t\t\t: undefined,\r\n\t\t\ttags: allTags,\r\n\t\t\tdate: group[group.length - 1].date,\r\n\t\t\tsourceSession: group[group.length - 1].sourceSession,\r\n\t\t});\r\n\t}\r\n\treturn compacted;\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Display formatters\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function formatProceduralForDisplay(m: ProceduralMemory): string {\r\n\tconst steps = m.steps.map((s, i) => `  ${i + 1}. ${s}`).join(\"\\n\");\r\n\treturn `[${m.id}] ${m.name}\\n  Trigger: ${m.trigger}\\n  Tags: ${m.tags.join(\", \")}\\n  Steps:\\n${steps}\\n  Updated: ${m.updated}`;\r\n}\r\n\r\nexport function formatEpisodicForDisplay(m: EpisodicMemory): string {\r\n\tlet text = `[${m.id}] ${m.summary}\\n  Date: ${m.date}\\n  Tags: ${m.tags.join(\", \")}`;\r\n\tif (m.details.length > 0) {\r\n\t\ttext += `\\n  Details:\\n${m.details.map((d) => `    - ${d}`).join(\"\\n\")}`;\r\n\t}\r\n\tif (m.reflection) {\r\n\t\tif (m.reflection.mistakes.length > 0) {\r\n\t\t\ttext += `\\n  Mistakes:\\n${m.reflection.mistakes.map((x) => `    - ${x}`).join(\"\\n\")}`;\r\n\t\t}\r\n\t\tif (m.reflection.lessons.length > 0) {\r\n\t\t\ttext += `\\n  Lessons:\\n${m.reflection.lessons.map((x) => `    - ${x}`).join(\"\\n\")}`;\r\n\t\t}\r\n\t}\r\n\treturn text;\r\n}\r\n\r\nexport function formatSemanticForDisplay(m: SemanticMemory): string {\r\n\treturn `[${m.id}] (${m.category}) ${m.text}\\n  Tags: ${m.tags.join(\", \")}\\n  Created: ${m.created}`;\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// System prompt section\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function buildMemoryPromptSection(cwd: string): string {\r\n\tconst semantic = loadStore<SemanticMemory>(cwd, \"semantic.json\");\r\n\tconst procedural = loadStore<ProceduralMemory>(cwd, \"procedural.json\");\r\n\tconst episodic = loadStore<EpisodicMemory>(cwd, \"episodic.json\");\r\n\r\n\tif (semantic.memories.length === 0 && procedural.memories.length === 0 && episodic.memories.length === 0) {\r\n\t\treturn `\\n\\n<memory_system>\r\nYou have a persistent memory system that stores knowledge across sessions.\r\nCurrently no memories are stored. Use the memory_write tool to save:\r\n1. Procedural memories when the user guides you through a multi-step workflow.\r\n2. Semantic memories when the user states a preference, rule, correction, or fact.\r\n3. Episodic memories when a significant task completes.\r\n</memory_system>`;\r\n\t}\r\n\r\n\tconst parts: string[] = [];\r\n\r\n\tparts.push(`\\n\\n<memory_system>\r\nYou have a persistent memory system that stores knowledge across sessions.\r\nYou MUST use these memories to improve your responses. Do not ask the user\r\nto repeat things they have already taught you.\r\n\r\nAUTOMATIC MEMORY CAPTURE RULES:\r\n1. When the user guides you through a multi-step workflow (correcting steps,\r\n   adding steps, reordering), save it as a PROCEDURAL memory using the\r\n   memory_write tool. Extract the general workflow, not the specific instance.\r\n2. When the user states a preference or rule (\"always do X\", \"never do Y\",\r\n   \"I prefer X\"), save it as a SEMANTIC memory immediately.\r\n3. When the user corrects you or you make a mistake, save the lesson as a\r\n   SEMANTIC memory with category \"convention\".\r\n4. When a significant task completes, save a summary as an EPISODIC memory.\r\n5. If you detect a conflict with an existing memory, ask the user: \"You\r\n   previously told me [old]. You are now saying [new]. Should I update this?\"\r\n   Only update after confirmation.\r\n\r\nTIMESTAMP AWARENESS:\r\nEach user message includes a timestamp. Use these to detect time gaps.\r\nIf the user returns after a significant gap (>10 minutes), acknowledge it\r\nnaturally if relevant. Do not force it if the gap is not relevant.`);\r\n\r\n\tif (semantic.memories.length > 0) {\r\n\t\tparts.push(\"\\n<semantic_memories>\");\r\n\t\tfor (const m of semantic.memories) {\r\n\t\t\tparts.push(`  <memory id=\"${m.id}\" category=\"${m.category}\">${escapeXml(m.text)}</memory>`);\r\n\t\t}\r\n\t\tparts.push(\"</semantic_memories>\");\r\n\t}\r\n\r\n\tif (procedural.memories.length > 0) {\r\n\t\tparts.push(\"\\n<procedural_memories>\");\r\n\t\tfor (const m of procedural.memories) {\r\n\t\t\tconst steps = m.steps.map((s, i) => `${i + 1}. ${s}`).join(\"; \");\r\n\t\t\tparts.push(\r\n\t\t\t\t`  <procedure id=\"${m.id}\" name=\"${escapeXml(m.name)}\" trigger=\"${escapeXml(m.trigger)}\">${escapeXml(steps)}</procedure>`,\r\n\t\t\t);\r\n\t\t}\r\n\t\tparts.push(\"</procedural_memories>\");\r\n\t}\r\n\r\n\tif (episodic.memories.length > 0) {\r\n\t\tconst recent = episodic.memories.slice(-10);\r\n\t\tparts.push(\"\\n<recent_episodic_memories>\");\r\n\t\tfor (const m of recent) {\r\n\t\t\tconst details = m.details.join(\"; \");\r\n\t\t\tlet text = details;\r\n\t\t\tif (m.reflection?.lessons.length) {\r\n\t\t\t\ttext += ` | Lessons: ${m.reflection.lessons.join(\"; \")}`;\r\n\t\t\t}\r\n\t\t\tparts.push(\r\n\t\t\t\t`  <episode id=\"${m.id}\" date=\"${m.date}\" summary=\"${escapeXml(m.summary)}\">${escapeXml(text)}</episode>`,\r\n\t\t\t);\r\n\t\t}\r\n\t\tparts.push(\"</recent_episodic_memories>\");\r\n\t}\r\n\r\n\tparts.push(\"\\n</memory_system>\");\r\n\r\n\treturn parts.join(\"\\n\");\r\n}\r\n\r\nfunction escapeXml(str: string): string {\r\n\treturn str\r\n\t\t.replace(/&/g, \"&amp;\")\r\n\t\t.replace(/</g, \"&lt;\")\r\n\t\t.replace(/>/g, \"&gt;\")\r\n\t\t.replace(/\"/g, \"&quot;\")\r\n\t\t.replace(/'/g, \"&apos;\");\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Timestamp formatting\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function formatTimestamp(ts: number): string {\r\n\tconst d = new Date(ts);\r\n\treturn d.toLocaleString(\"en-US\", {\r\n\t\tyear: \"numeric\",\r\n\t\tmonth: \"short\",\r\n\t\tday: \"numeric\",\r\n\t\thour: \"2-digit\",\r\n\t\tminute: \"2-digit\",\r\n\t\tsecond: \"2-digit\",\r\n\t\thour12: false,\r\n\t});\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Memory counts (for status display)\r\n// ---------------------------------------------------------------------------\r\n\r\nexport interface MemoryCounts {\r\n\tsemantic: number;\r\n\tprocedural: number;\r\n\tepisodic: number;\r\n\ttotal: number;\r\n}\r\n\r\nexport function getMemoryCounts(cwd: string): MemoryCounts {\r\n\tconst semantic = loadStore<SemanticMemory>(cwd, \"semantic.json\").memories.length;\r\n\tconst procedural = loadStore<ProceduralMemory>(cwd, \"procedural.json\").memories.length;\r\n\tconst episodic = loadStore<EpisodicMemory>(cwd, \"episodic.json\").memories.length;\r\n\treturn { semantic, procedural, episodic, total: semantic + procedural + episodic };\r\n}\r\n"]}