@wrongstack/core 0.2.0 → 0.3.2

package/dist/extension/index.d.ts

@@ -0,0 +1,9 @@
+export { A as AfterIterationHook, p as AfterRunHook, q as AfterToolExecutionHook, s as AgentExtension, B as BeforeIterationHook, w as BeforeRunHook, x as BeforeToolExecutionHook, E as ExtensionRegistry, O as OnErrorHook, G as ProviderRunnerFn, z as ProviderRunnerWrapper } from '../index-hWNybrNZ.js';
+import '../context-IovtuTf8.js';
+import '../logger-BMQgxvdy.js';
+import '../system-prompt-Dk1qm8ey.js';
+import '../observability-BhnVLBLS.js';
+import '../events-BHIQs4o1.js';
+import '../secret-scrubber-CgG2tV2B.js';
+import '../config-D2qvAxVd.js';
+import '../models-registry-Y2xbog0E.js';

package/dist/extension/index.js

@@ -0,0 +1,241 @@
+// src/types/errors.ts
+var WrongStackError = class extends Error {
+  code;
+  subsystem;
+  severity;
+  recoverable;
+  context;
+  constructor(opts) {
+    super(opts.message, { cause: opts.cause });
+    this.name = "WrongStackError";
+    this.code = opts.code;
+    this.subsystem = opts.subsystem;
+    this.severity = opts.severity ?? "error";
+    this.recoverable = opts.recoverable ?? false;
+    this.context = opts.context;
+  }
+  /**
+   * Render a one-line user-facing description.
+   * Subclasses should override for domain-specific formatting.
+   */
+  describe() {
+    const ctx = this.context ? ` ${formatContext(this.context)}` : "";
+    return `${this.code}: ${this.message}${ctx}`;
+  }
+};
+function formatContext(ctx) {
+  const parts = Object.entries(ctx).filter(([, v]) => v !== void 0).slice(0, 3).map(([k, v]) => `${k}=${String(v)}`);
+  return parts.length > 0 ? `[${parts.join(" ")}]` : "";
+}
+
+// src/extension/registry.ts
+var ExtensionRegistry = class {
+  extensions = [];
+  promptContributors = [];
+  log;
+  setLogger(log) {
+    this.log = log;
+  }
+  /**
+   * Register a system prompt contributor. Returns an unregister function.
+   * Contributors are called on every system prompt build in registration
+   * order. Their output blocks are inserted after the core environment
+   * block, before the mode and plan blocks.
+   */
+  registerSystemPromptContributor(c) {
+    this.promptContributors.push(c);
+    return () => {
+      const idx = this.promptContributors.indexOf(c);
+      if (idx >= 0) this.promptContributors.splice(idx, 1);
+    };
+  }
+  /**
+   * Build all registered system prompt contributions.
+   * Failures are caught and logged — one bad contributor doesn't
+   * break the prompt assembly.
+   */
+  async buildSystemPromptContributions(ctx) {
+    const blocks = [];
+    for (const c of this.promptContributors) {
+      try {
+        const contributed = await c(ctx);
+        blocks.push(...contributed);
+      } catch (err) {
+        this.log?.error("SystemPromptContributor failed", err);
+      }
+    }
+    return blocks;
+  }
+  /**
+   * Returns the live array of contributors (readonly snapshot for
+   * passing to DefaultSystemPromptBuilder at build time).
+   */
+  listSystemPromptContributors() {
+    return this.promptContributors;
+  }
+  /**
+   * Register an extension. Duplicate names are rejected.
+   * Returns an unregister function.
+   */
+  register(ext) {
+    if (this.extensions.some((e) => e.name === ext.name)) {
+      throw new WrongStackError({
+        message: `Extension "${ext.name}" already registered`,
+        code: "REGISTRY_DUPLICATE",
+        subsystem: "container",
+        context: { extension: ext.name }
+      });
+    }
+    this.extensions.push(ext);
+    return () => this.unregister(ext.name);
+  }
+  /**
+   * Register an extension, silently replacing any previous registration
+   * with the same name. Use this when overriding a default extension.
+   */
+  registerOrReplace(ext) {
+    const idx = this.extensions.findIndex((e) => e.name === ext.name);
+    if (idx >= 0) this.extensions.splice(idx, 1);
+    return this.register(ext);
+  }
+  /**
+   * Unregister an extension by name. Returns true if found.
+   */
+  unregister(name) {
+    const idx = this.extensions.findIndex((e) => e.name === name);
+    if (idx === -1) return false;
+    this.extensions.splice(idx, 1);
+    return true;
+  }
+  /**
+   * List registered extension names in order.
+   */
+  list() {
+    return this.extensions.map((e) => e.name);
+  }
+  /**
+   * Check if an extension with the given name is registered.
+   */
+  has(name) {
+    return this.extensions.some((e) => e.name === name);
+  }
+  /**
+   * Remove all registered extensions and contributors.
+   */
+  clear() {
+    this.extensions.length = 0;
+    this.promptContributors.length = 0;
+  }
+  // ── Hook runners ─────────────────────────────────────────────────
+  async runBeforeRun(...args) {
+    const snapshot = [...this.extensions];
+    for (const ext of snapshot) {
+      if (!ext.beforeRun) continue;
+      try {
+        await ext.beforeRun(...args);
+      } catch (err) {
+        this.log?.error(`Extension "${ext.name}" beforeRun hook failed`, err);
+      }
+    }
+  }
+  async runAfterRun(...args) {
+    const snapshot = [...this.extensions];
+    for (const ext of snapshot) {
+      if (!ext.afterRun) continue;
+      try {
+        await ext.afterRun(...args);
+      } catch (err) {
+        this.log?.error(`Extension "${ext.name}" afterRun hook failed`, err);
+      }
+    }
+  }
+  async runBeforeIteration(...args) {
+    const snapshot = [...this.extensions];
+    for (const ext of snapshot) {
+      if (!ext.beforeIteration) continue;
+      try {
+        await ext.beforeIteration(...args);
+      } catch (err) {
+        this.log?.error(`Extension "${ext.name}" beforeIteration hook failed`, err);
+      }
+    }
+  }
+  async runAfterIteration(...args) {
+    const snapshot = [...this.extensions];
+    for (const ext of snapshot) {
+      if (!ext.afterIteration) continue;
+      try {
+        await ext.afterIteration(...args);
+      } catch (err) {
+        this.log?.error(`Extension "${ext.name}" afterIteration hook failed`, err);
+      }
+    }
+  }
+  /**
+   * Run onError hooks in order. The first hook that returns a non-void
+   * result wins; subsequent hooks are skipped.
+   */
+  async runOnError(...args) {
+    const snapshot = [...this.extensions];
+    for (const ext of snapshot) {
+      if (!ext.onError) continue;
+      try {
+        const result = await ext.onError(...args);
+        if (result) return result;
+      } catch (err) {
+        this.log?.error(`Extension "${ext.name}" onError hook failed`, err);
+      }
+    }
+  }
+  /**
+   * Build a composed provider runner. Extensions with `wrapProviderRunner`
+   * form a middleware-style chain: the innermost extension wraps the
+   * default runner, each subsequent wrapper wraps the previous.
+   */
+  wrapProviderRunner(inner) {
+    const wrappers = this.extensions.filter((e) => e.wrapProviderRunner).map((e) => ({ name: e.name, wrap: e.wrapProviderRunner }));
+    if (wrappers.length === 0) return inner;
+    let composed = inner;
+    for (let i = wrappers.length - 1; i >= 0; i--) {
+      const wrapper = wrappers[i];
+      const next = composed;
+      composed = async (ctx, req) => {
+        try {
+          return await wrapper.wrap(ctx, req, next);
+        } catch (err) {
+          this.log?.error(`Extension "${wrapper.name}" wrapProviderRunner failed`, err);
+          throw err;
+        }
+      };
+    }
+    return composed;
+  }
+  async runBeforeToolExecution(...args) {
+    let toolUses = args[1];
+    const snapshot = [...this.extensions];
+    for (const ext of snapshot) {
+      if (!ext.beforeToolExecution) continue;
+      try {
+        toolUses = await ext.beforeToolExecution(args[0], toolUses);
+      } catch (err) {
+        this.log?.error(`Extension "${ext.name}" beforeToolExecution hook failed`, err);
+      }
+    }
+    return toolUses;
+  }
+  async runAfterToolExecution(...args) {
+    const snapshot = [...this.extensions];
+    for (const ext of snapshot) {
+      if (!ext.afterToolExecution) continue;
+      try {
+        await ext.afterToolExecution(...args);
+      } catch (err) {
+        this.log?.error(`Extension "${ext.name}" afterToolExecution hook failed`, err);
+      }
+    }
+  }
+};
+
+export { ExtensionRegistry };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/extension/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/types/errors.ts","../../src/extension/registry.ts"],"names":[],"mappings":";AAiEO,IAAM,eAAA,GAAN,cAA8B,KAAA,CAAM;AAAA,EAChC,IAAA;AAAA,EACA,SAAA;AAAA,EACA,QAAA;AAAA,EACA,WAAA;AAAA,EACA,OAAA;AAAA,EAET,YAAY,IAAA,EAQT;AACD,IAAA,KAAA,CAAM,KAAK,OAAA,EAAS,EAAE,KAAA,EAAO,IAAA,CAAK,OAAO,CAAA;AACzC,IAAA,IAAA,CAAK,IAAA,GAAO,iBAAA;AACZ,IAAA,IAAA,CAAK,OAAO,IAAA,CAAK,IAAA;AACjB,IAAA,IAAA,CAAK,YAAY,IAAA,CAAK,SAAA;AACtB,IAAA,IAAA,CAAK,QAAA,GAAW,KAAK,QAAA,IAAY,OAAA;AACjC,IAAA,IAAA,CAAK,WAAA,GAAc,KAAK,WAAA,IAAe,KAAA;AACvC,IAAA,IAAA,CAAK,UAAU,IAAA,CAAK,OAAA;AAAA,EACtB;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,QAAA,GAAmB;AACjB,IAAA,MAAM,GAAA,GAAM,KAAK,OAAA,GAAU,CAAA,CAAA,EAAI,cAAc,IAAA,CAAK,OAAO,CAAC,CAAA,CAAA,GAAK,EAAA;AAC/D,IAAA,OAAO,GAAG,IAAA,CAAK,IAAI,KAAK,IAAA,CAAK,OAAO,GAAG,GAAG,CAAA,CAAA;AAAA,EAC5C;AACF,CAAA;AAEA,SAAS,cAAc,GAAA,EAAsC;AAC3D,EAAA,MAAM,KAAA,GAAQ,MAAA,CAAO,OAAA,CAAQ,GAAG,CAAA,CAC7B,MAAA,CAAO,CAAC,GAAG,CAAC,CAAA,KAAM,CAAA,KAAM,MAAS,CAAA,CACjC,KAAA,CAAM,CAAA,EAAG,CAAC,CAAA,CACV,GAAA,CAAI,CAAC,CAAC,CAAA,EAAG,CAAC,CAAA,KAAM,CAAA,EAAG,CAAC,CAAA,CAAA,EAAI,MAAA,CAAO,CAAC,CAAC,CAAA,CAAE,CAAA;AACtC,EAAA,OAAO,KAAA,CAAM,SAAS,CAAA,GAAI,CAAA,CAAA,EAAI,MAAM,IAAA,CAAK,GAAG,CAAC,CAAA,CAAA,CAAA,GAAM,EAAA;AACrD;;;ACjFO,IAAM,oBAAN,MAAwB;AAAA,EACZ,aAA+B,EAAC;AAAA,EAChC,qBAAgD,EAAC;AAAA,EAC1D,GAAA;AAAA,EAER,UAAU,GAAA,EAAmB;AAC3B,IAAA,IAAA,CAAK,GAAA,GAAM,GAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,gCAAgC,CAAA,EAAwC;AACtE,IAAA,IAAA,CAAK,kBAAA,CAAmB,KAAK,CAAC,CAAA;AAC9B,IAAA,OAAO,MAAM;AACX,MAAA,MAAM,GAAA,GAAM,IAAA,CAAK,kBAAA,CAAmB,OAAA,CAAQ,CAAC,CAAA;AAC7C,MAAA,IAAI,OAAO,CAAA,EAAG,IAAA,CAAK,kBAAA,CAAmB,MAAA,CAAO,KAAK,CAAC,CAAA;AAAA,IACrD,CAAA;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,+BACJ,GAAA,EACsB;AACtB,IAAA,MAAM,SAAsB,EAAC;AAC7B,IAAA,KAAA,MAAW,CAAA,IAAK,KAAK,kBAAA,EAAoB;AACvC,MAAA,IAAI;AACF,QAAA,MAAM,WAAA,GAAc,MAAM,CAAA,CAAE,GAAG,CAAA;AAC/B,QAAA,MAAA,CAAO,IAAA,CAAK,GAAG,WAAW,CAAA;AAAA,MAC5B,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,GAAA,EAAK,KAAA,CAAM,gCAAA,EAAkC,GAAG,CAAA;AAAA,MACvD;AAAA,IACF;AACA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,4BAAA,GAAmE;AACjE,IAAA,OAAO,IAAA,CAAK,kBAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,SAAS,GAAA,EAAiC;AACxC,IAAA,IAAI,IAAA,CAAK,WAAW,IAAA,CAAK,CAAC,MAAM,CAAA,CAAE,IAAA,KAAS,GAAA,CAAI,IAAI,CAAA,EAAG;AACpD,MAAA,MAAM,IAAI,eAAA,CAAgB;AAAA,QACxB,OAAA,EAAS,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,CAAA,oBAAA,CAAA;AAAA,QAC/B,IAAA,EAAM,oBAAA;AAAA,QACN,SAAA,EAAW,WAAA;AAAA,QACX,OAAA,EAAS,EAAE,SAAA,EAAW,GAAA,CAAI,IAAA;AAAK,OAChC,CAAA;AAAA,IACH;AACA,IAAA,IAAA,CAAK,UAAA,CAAW,KAAK,GAAG,CAAA;AACxB,IAAA,OAAO,MAAM,IAAA,CAAK,UAAA,CAAW,GAAA,CAAI,IAAI,CAAA;AAAA,EACvC;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,kBAAkB,GAAA,EAAiC;AACjD,IAAA,MAAM,GAAA,GAAM,KAAK,UAAA,CAAW,SAAA,CAAU,CAAC,CAAA,KAAM,CAAA,CAAE,IAAA,KAAS,GAAA,CAAI,IAAI,CAAA;AAChE,IAAA,IAAI,OAAO,CAAA,EAAG,IAAA,CAAK,UAAA,CAAW,MAAA,CAAO,KAAK,CAAC,CAAA;AAC3C,IAAA,OAAO,IAAA,CAAK,SAAS,GAAG,CAAA;AAAA,EAC1B;AAAA;AAAA;AAAA;AAAA,EAKA,WAAW,IAAA,EAAuB;AAChC,IAAA,MAAM,GAAA,GAAM,KAAK,UAAA,CAAW,SAAA,CAAU,CAAC,CAAA,KAAM,CAAA,CAAE,SAAS,IAAI,CAAA;AAC5D,IAAA,IAAI,GAAA,KAAQ,IAAI,OAAO,KAAA;AACvB,IAAA,IAAA,CAAK,UAAA,CAAW,MAAA,CAAO,GAAA,EAAK,CAAC,CAAA;AAC7B,IAAA,OAAO,IAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,IAAA,GAA0B;AACxB,IAAA,OAAO,KAAK,UAAA,CAAW,GAAA,CAAI,CAAC,CAAA,KAAM,EAAE,IAAI,CAAA;AAAA,EAC1C;AAAA;AAAA;AAAA;AAAA,EAKA,IAAI,IAAA,EAAuB;AACzB,IAAA,OAAO,KAAK,UAAA,CAAW,IAAA,CAAK,CAAC,CAAA,KAAM,CAAA,CAAE,SAAS,IAAI,CAAA;AAAA,EACpD;AAAA;AAAA;AAAA;AAAA,EAKA,KAAA,GAAc;AACZ,IAAA,IAAA,CAAK,WAAW,MAAA,GAAS,CAAA;AACzB,IAAA,IAAA,CAAK,mBAAmB,MAAA,GAAS,CAAA;AAAA,EACnC;AAAA;AAAA,EAIA,MAAM,gBAAgB,IAAA,EAAgD;AACpE,IAAA,MAAM,QAAA,GAAW,CAAC,GAAG,IAAA,CAAK,UAAU,CAAA;AACpC,IAAA,KAAA,MAAW,OAAO,QAAA,EAAU;AAC1B,MAAA,IAAI,CAAC,IAAI,SAAA,EAAW;AACpB,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,SAAA,CAAU,GAAG,IAAI,CAAA;AAAA,MAC7B,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,2BAA2B,GAAG,CAAA;AAAA,MACtE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,eAAe,IAAA,EAA+C;AAClE,IAAA,MAAM,QAAA,GAAW,CAAC,GAAG,IAAA,CAAK,UAAU,CAAA;AACpC,IAAA,KAAA,MAAW,OAAO,QAAA,EAAU;AAC1B,MAAA,IAAI,CAAC,IAAI,QAAA,EAAU;AACnB,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,QAAA,CAAS,GAAG,IAAI,CAAA;AAAA,MAC5B,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,0BAA0B,GAAG,CAAA;AAAA,MACrE;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,sBAAsB,IAAA,EAAsD;AAChF,IAAA,MAAM,QAAA,GAAW,CAAC,GAAG,IAAA,CAAK,UAAU,CAAA;AACpC,IAAA,KAAA,MAAW,OAAO,QAAA,EAAU;AAC1B,MAAA,IAAI,CAAC,IAAI,eAAA,EAAiB;AAC1B,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,eAAA,CAAgB,GAAG,IAAI,CAAA;AAAA,MACnC,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,iCAAiC,GAAG,CAAA;AAAA,MAC5E;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,qBAAqB,IAAA,EAAqD;AAC9E,IAAA,MAAM,QAAA,GAAW,CAAC,GAAG,IAAA,CAAK,UAAU,CAAA;AACpC,IAAA,KAAA,MAAW,OAAO,QAAA,EAAU;AAC1B,MAAA,IAAI,CAAC,IAAI,cAAA,EAAgB;AACzB,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,cAAA,CAAe,GAAG,IAAI,CAAA;AAAA,MAClC,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,gCAAgC,GAAG,CAAA;AAAA,MAC3E;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,cACD,IAAA,EAGH;AACA,IAAA,MAAM,QAAA,GAAW,CAAC,GAAG,IAAA,CAAK,UAAU,CAAA;AACpC,IAAA,KAAA,MAAW,OAAO,QAAA,EAAU;AAC1B,MAAA,IAAI,CAAC,IAAI,OAAA,EAAS;AAClB,MAAA,IAAI;AACF,QAAA,MAAM,MAAA,GAAS,MAAM,GAAA,CAAI,OAAA,CAAQ,GAAG,IAAI,CAAA;AACxC,QAAA,IAAI,QAAQ,OAAO,MAAA;AAAA,MACrB,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,yBAAyB,GAAG,CAAA;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,mBAAmB,KAAA,EAA2C;AAC5D,IAAA,MAAM,WAAW,IAAA,CAAK,UAAA,CACnB,OAAO,CAAC,CAAA,KAAM,EAAE,kBAAkB,CAAA,CAClC,IAAI,CAAC,CAAA,MAAO,EAAE,IAAA,EAAM,CAAA,CAAE,MAAM,IAAA,EAAM,CAAA,CAAE,oBAAoB,CAAE,CAAA;AAE7D,IAAA,IAAI,QAAA,CAAS,MAAA,KAAW,CAAA,EAAG,OAAO,KAAA;AAGlC,IAAA,IAAI,QAAA,GAA6B,KAAA;AACjC,IAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC7C,MAAA,MAAM,OAAA,GAAU,SAAS,CAAC,CAAA;AAC1B,MAAA,MAAM,IAAA,GAAO,QAAA;AACb,MAAA,QAAA,GAAW,OAAO,KAAK,GAAA,KAAQ;AAC7B,QAAA,IAAI;AACF,UAAA,OAAO,MAAM,OAAA,CAAQ,IAAA,CAAK,GAAA,EAAK,KAAK,IAAI,CAAA;AAAA,QAC1C,SAAS,GAAA,EAAK;AACZ,UAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,OAAA,CAAQ,IAAI,+BAA+B,GAAG,CAAA;AAC5E,UAAA,MAAM,GAAA;AAAA,QACR;AAAA,MACF,CAAA;AAAA,IACF;AACA,IAAA,OAAO,QAAA;AAAA,EACT;AAAA,EAEA,MAAM,0BACD,IAAA,EAC8C;AACjD,IAAA,IAAI,QAAA,GAAW,KAAK,CAAC,CAAA;AACrB,IAAA,MAAM,QAAA,GAAW,CAAC,GAAG,IAAA,CAAK,UAAU,CAAA;AACpC,IAAA,KAAA,MAAW,OAAO,QAAA,EAAU;AAC1B,MAAA,IAAI,CAAC,IAAI,mBAAA,EAAqB;AAC9B,MAAA,IAAI;AACF,QAAA,QAAA,GAAW,MAAM,GAAA,CAAI,mBAAA,CAAoB,IAAA,CAAK,CAAC,GAAG,QAAQ,CAAA;AAAA,MAC5D,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,qCAAqC,GAAG,CAAA;AAAA,MAChF;AAAA,IACF;AACA,IAAA,OAAO,QAAA;AAAA,EACT;AAAA,EAEA,MAAM,yBAAyB,IAAA,EAAyD;AACtF,IAAA,MAAM,QAAA,GAAW,CAAC,GAAG,IAAA,CAAK,UAAU,CAAA;AACpC,IAAA,KAAA,MAAW,OAAO,QAAA,EAAU;AAC1B,MAAA,IAAI,CAAC,IAAI,kBAAA,EAAoB;AAC7B,MAAA,IAAI;AACF,QAAA,MAAM,GAAA,CAAI,kBAAA,CAAmB,GAAG,IAAI,CAAA;AAAA,MACtC,SAAS,GAAA,EAAK;AACZ,QAAA,IAAA,CAAK,KAAK,KAAA,CAAM,CAAA,WAAA,EAAc,GAAA,CAAI,IAAI,oCAAoC,GAAG,CAAA;AAAA,MAC/E;AAAA,IACF;AAAA,EACF;AACF","file":"index.js","sourcesContent":["/**\n * WrongStack error hierarchy.\n *\n * Every error thrown by the framework is a `WrongStackError` with a\n * machine-readable `code`, a `subsystem` tag, and a `severity` level.\n * This lets consumers (CLI, TUI, plugins, tests) branch on structured\n * data instead of parsing error messages.\n */\n\n// ── Error codes ──────────────────────────────────────────────────────\n\nexport type ErrorCode =\n  // Provider\n  | 'PROVIDER_RATE_LIMITED'\n  | 'PROVIDER_AUTH_FAILED'\n  | 'PROVIDER_OVERLOADED'\n  | 'PROVIDER_INVALID_REQUEST'\n  | 'PROVIDER_SERVER_ERROR'\n  | 'PROVIDER_NETWORK_ERROR'\n  | 'PROVIDER_CONTEXT_OVERFLOW'\n  // Tool\n  | 'TOOL_NOT_FOUND'\n  | 'TOOL_PERMISSION_DENIED'\n  | 'TOOL_EXECUTION_FAILED'\n  | 'TOOL_TIMEOUT'\n  | 'TOOL_INPUT_INVALID'\n  // Config\n  | 'CONFIG_INVALID'\n  | 'CONFIG_NOT_FOUND'\n  | 'CONFIG_PARSE_FAILED'\n  | 'CONFIG_MIGRATION_NEEDED'\n  // Plugin\n  | 'PLUGIN_LOAD_FAILED'\n  | 'PLUGIN_API_MISMATCH'\n  | 'PLUGIN_MISSING_DEPENDENCY'\n  // Agent\n  | 'AGENT_ITERATION_LIMIT'\n  | 'AGENT_CONTEXT_OVERFLOW'\n  | 'AGENT_ABORTED'\n  | 'AGENT_RUN_FAILED'\n  // Session\n  | 'SESSION_NOT_FOUND'\n  | 'SESSION_CORRUPTED'\n  | 'SESSION_WRITE_FAILED'\n  // Container / Registry\n  | 'CONTAINER_TOKEN_ALREADY_BOUND'\n  | 'CONTAINER_TOKEN_NOT_BOUND'\n  | 'REGISTRY_DUPLICATE'\n  | 'REGISTRY_NOT_FOUND'\n  // General\n  | 'UNKNOWN';\n\nexport type ErrorSubsystem =\n  | 'provider'\n  | 'tool'\n  | 'config'\n  | 'plugin'\n  | 'agent'\n  | 'session'\n  | 'container'\n  | 'general';\nexport type ErrorSeverity = 'fatal' | 'error' | 'warning';\n\n// ── Base error class ─────────────────────────────────────────────────\n\nexport class WrongStackError extends Error {\n  readonly code: ErrorCode;\n  readonly subsystem: ErrorSubsystem;\n  readonly severity: ErrorSeverity;\n  readonly recoverable: boolean;\n  readonly context?: Record<string, unknown>;\n\n  constructor(opts: {\n    message: string;\n    code: ErrorCode;\n    subsystem: ErrorSubsystem;\n    severity?: ErrorSeverity;\n    recoverable?: boolean;\n    context?: Record<string, unknown>;\n    cause?: unknown;\n  }) {\n    super(opts.message, { cause: opts.cause });\n    this.name = 'WrongStackError';\n    this.code = opts.code;\n    this.subsystem = opts.subsystem;\n    this.severity = opts.severity ?? 'error';\n    this.recoverable = opts.recoverable ?? false;\n    this.context = opts.context;\n  }\n\n  /**\n   * Render a one-line user-facing description.\n   * Subclasses should override for domain-specific formatting.\n   */\n  describe(): string {\n    const ctx = this.context ? ` ${formatContext(this.context)}` : '';\n    return `${this.code}: ${this.message}${ctx}`;\n  }\n}\n\nfunction formatContext(ctx: Record<string, unknown>): string {\n  const parts = Object.entries(ctx)\n    .filter(([, v]) => v !== undefined)\n    .slice(0, 3)\n    .map(([k, v]) => `${k}=${String(v)}`);\n  return parts.length > 0 ? `[${parts.join(' ')}]` : '';\n}\n\n// ── Specific error classes ───────────────────────────────────────────\n\n/**\n * Tool execution errors — thrown by ToolExecutor and individual tools.\n */\nexport class ToolError extends WrongStackError {\n  readonly toolName: string;\n\n  constructor(opts: {\n    message: string;\n    code: Extract<\n      ErrorCode,\n      | 'TOOL_NOT_FOUND'\n      | 'TOOL_PERMISSION_DENIED'\n      | 'TOOL_EXECUTION_FAILED'\n      | 'TOOL_TIMEOUT'\n      | 'TOOL_INPUT_INVALID'\n    >;\n    toolName: string;\n    recoverable?: boolean;\n    context?: Record<string, unknown>;\n    cause?: unknown;\n  }) {\n    super({\n      message: opts.message,\n      code: opts.code,\n      subsystem: 'tool',\n      recoverable: opts.recoverable,\n      context: { tool: opts.toolName, ...opts.context },\n      cause: opts.cause,\n    });\n    this.name = 'ToolError';\n    this.toolName = opts.toolName;\n  }\n}\n\n/**\n * Config loading / validation errors.\n */\nexport class ConfigError extends WrongStackError {\n  constructor(opts: {\n    message: string;\n    code: Extract<\n      ErrorCode,\n      'CONFIG_INVALID' | 'CONFIG_NOT_FOUND' | 'CONFIG_PARSE_FAILED' | 'CONFIG_MIGRATION_NEEDED'\n    >;\n    context?: Record<string, unknown>;\n    cause?: unknown;\n  }) {\n    super({\n      message: opts.message,\n      code: opts.code,\n      subsystem: 'config',\n      severity: 'fatal',\n      recoverable: false,\n      context: opts.context,\n      cause: opts.cause,\n    });\n    this.name = 'ConfigError';\n  }\n}\n\n/**\n * Plugin loading / lifecycle errors.\n */\nexport class PluginError extends WrongStackError {\n  readonly pluginName: string;\n\n  constructor(opts: {\n    message: string;\n    code: Extract<\n      ErrorCode,\n      'PLUGIN_LOAD_FAILED' | 'PLUGIN_API_MISMATCH' | 'PLUGIN_MISSING_DEPENDENCY'\n    >;\n    pluginName: string;\n    context?: Record<string, unknown>;\n    cause?: unknown;\n  }) {\n    super({\n      message: opts.message,\n      code: opts.code,\n      subsystem: 'plugin',\n      severity: 'error',\n      recoverable: opts.code === 'PLUGIN_MISSING_DEPENDENCY',\n      context: { plugin: opts.pluginName, ...opts.context },\n      cause: opts.cause,\n    });\n    this.name = 'PluginError';\n    this.pluginName = opts.pluginName;\n  }\n}\n\n/**\n * Agent runtime errors — thrown by Agent.run when a non-WrongStackError\n * escapes the inner loop, so callers always see a structured error.\n */\nexport class AgentError extends WrongStackError {\n  constructor(opts: {\n    message: string;\n    code: Extract<\n      ErrorCode,\n      'AGENT_ITERATION_LIMIT' | 'AGENT_CONTEXT_OVERFLOW' | 'AGENT_ABORTED' | 'AGENT_RUN_FAILED'\n    >;\n    recoverable?: boolean;\n    context?: Record<string, unknown>;\n    cause?: unknown;\n  }) {\n    super({\n      message: opts.message,\n      code: opts.code,\n      subsystem: 'agent',\n      severity: opts.code === 'AGENT_ABORTED' ? 'warning' : 'error',\n      recoverable: opts.recoverable ?? opts.code === 'AGENT_ITERATION_LIMIT',\n      context: opts.context,\n      cause: opts.cause,\n    });\n    this.name = 'AgentError';\n  }\n}\n\n/**\n * Wrap an arbitrary thrown value into a `WrongStackError` so the caller\n * always gets a structured error. Pass-throughs WrongStackError instances\n * unchanged; raw `Error`s and primitives get an `AGENT_RUN_FAILED` wrapper\n * with the original preserved as `cause`.\n */\nexport function toWrongStackError(\n  err: unknown,\n  code: Extract<ErrorCode, 'AGENT_RUN_FAILED' | 'AGENT_ABORTED' | 'UNKNOWN'> = 'AGENT_RUN_FAILED',\n): WrongStackError {\n  if (err instanceof WrongStackError) return err;\n  const message = err instanceof Error ? err.message : String(err);\n  return new AgentError({\n    message,\n    code: code === 'UNKNOWN' ? 'AGENT_RUN_FAILED' : code,\n    cause: err,\n  });\n}\n\n/**\n * Session storage errors.\n */\nexport class SessionError extends WrongStackError {\n  readonly sessionId?: string;\n\n  constructor(opts: {\n    message: string;\n    code: Extract<ErrorCode, 'SESSION_NOT_FOUND' | 'SESSION_CORRUPTED' | 'SESSION_WRITE_FAILED'>;\n    sessionId?: string;\n    context?: Record<string, unknown>;\n    cause?: unknown;\n  }) {\n    super({\n      message: opts.message,\n      code: opts.code,\n      subsystem: 'session',\n      severity: opts.code === 'SESSION_WRITE_FAILED' ? 'error' : 'warning',\n      recoverable: opts.code !== 'SESSION_CORRUPTED',\n      context: { sessionId: opts.sessionId, ...opts.context },\n      cause: opts.cause,\n    });\n    this.name = 'SessionError';\n    this.sessionId = opts.sessionId;\n  }\n}\n\n// ── Type guards ──────────────────────────────────────────────────────\n\nexport function isWrongStackError(err: unknown): err is WrongStackError {\n  return err instanceof WrongStackError;\n}\n\nexport function isToolError(err: unknown): err is ToolError {\n  return err instanceof ToolError;\n}\n\nexport function isConfigError(err: unknown): err is ConfigError {\n  return err instanceof ConfigError;\n}\n\nexport function isPluginError(err: unknown): err is PluginError {\n  return err instanceof PluginError;\n}\n\nexport function isSessionError(err: unknown): err is SessionError {\n  return err instanceof SessionError;\n}\n\nexport function isAgentError(err: unknown): err is AgentError {\n  return err instanceof AgentError;\n}\n","/**\n * ExtensionRegistry — manages AgentExtension registrations.\n *\n * Extensions are called in registration order at each lifecycle phase.\n * Each extension hook failure is caught and logged independently so\n * one bad extension can't take down the agent.\n */\n\nimport type { TextBlock } from '../types/blocks.js';\nimport { WrongStackError } from '../types/errors.js';\nimport type { Logger } from '../types/logger.js';\nimport type { SystemPromptContributor } from '../types/system-prompt-contributor.js';\nimport type {\n  AfterIterationHook,\n  AfterRunHook,\n  AfterToolExecutionHook,\n  AgentExtension,\n  BeforeIterationHook,\n  BeforeRunHook,\n  BeforeToolExecutionHook,\n  OnErrorHook,\n  ProviderRunnerFn,\n  ProviderRunnerWrapper,\n} from './extension-points.js';\n\nexport class ExtensionRegistry {\n  private readonly extensions: AgentExtension[] = [];\n  private readonly promptContributors: SystemPromptContributor[] = [];\n  private log: Logger | undefined;\n\n  setLogger(log: Logger): void {\n    this.log = log;\n  }\n\n  /**\n   * Register a system prompt contributor. Returns an unregister function.\n   * Contributors are called on every system prompt build in registration\n   * order. Their output blocks are inserted after the core environment\n   * block, before the mode and plan blocks.\n   */\n  registerSystemPromptContributor(c: SystemPromptContributor): () => void {\n    this.promptContributors.push(c);\n    return () => {\n      const idx = this.promptContributors.indexOf(c);\n      if (idx >= 0) this.promptContributors.splice(idx, 1);\n    };\n  }\n\n  /**\n   * Build all registered system prompt contributions.\n   * Failures are caught and logged — one bad contributor doesn't\n   * break the prompt assembly.\n   */\n  async buildSystemPromptContributions(\n    ctx: Parameters<SystemPromptContributor>[0],\n  ): Promise<TextBlock[]> {\n    const blocks: TextBlock[] = [];\n    for (const c of this.promptContributors) {\n      try {\n        const contributed = await c(ctx);\n        blocks.push(...contributed);\n      } catch (err) {\n        this.log?.error('SystemPromptContributor failed', err);\n      }\n    }\n    return blocks;\n  }\n\n  /**\n   * Returns the live array of contributors (readonly snapshot for\n   * passing to DefaultSystemPromptBuilder at build time).\n   */\n  listSystemPromptContributors(): readonly SystemPromptContributor[] {\n    return this.promptContributors;\n  }\n\n  /**\n   * Register an extension. Duplicate names are rejected.\n   * Returns an unregister function.\n   */\n  register(ext: AgentExtension): () => void {\n    if (this.extensions.some((e) => e.name === ext.name)) {\n      throw new WrongStackError({\n        message: `Extension \"${ext.name}\" already registered`,\n        code: 'REGISTRY_DUPLICATE',\n        subsystem: 'container',\n        context: { extension: ext.name },\n      });\n    }\n    this.extensions.push(ext);\n    return () => this.unregister(ext.name);\n  }\n\n  /**\n   * Register an extension, silently replacing any previous registration\n   * with the same name. Use this when overriding a default extension.\n   */\n  registerOrReplace(ext: AgentExtension): () => void {\n    const idx = this.extensions.findIndex((e) => e.name === ext.name);\n    if (idx >= 0) this.extensions.splice(idx, 1);\n    return this.register(ext);\n  }\n\n  /**\n   * Unregister an extension by name. Returns true if found.\n   */\n  unregister(name: string): boolean {\n    const idx = this.extensions.findIndex((e) => e.name === name);\n    if (idx === -1) return false;\n    this.extensions.splice(idx, 1);\n    return true;\n  }\n\n  /**\n   * List registered extension names in order.\n   */\n  list(): readonly string[] {\n    return this.extensions.map((e) => e.name);\n  }\n\n  /**\n   * Check if an extension with the given name is registered.\n   */\n  has(name: string): boolean {\n    return this.extensions.some((e) => e.name === name);\n  }\n\n  /**\n   * Remove all registered extensions and contributors.\n   */\n  clear(): void {\n    this.extensions.length = 0;\n    this.promptContributors.length = 0;\n  }\n\n  // ── Hook runners ─────────────────────────────────────────────────\n\n  async runBeforeRun(...args: Parameters<BeforeRunHook>): Promise<void> {\n    const snapshot = [...this.extensions];\n    for (const ext of snapshot) {\n      if (!ext.beforeRun) continue;\n      try {\n        await ext.beforeRun(...args);\n      } catch (err) {\n        this.log?.error(`Extension \"${ext.name}\" beforeRun hook failed`, err);\n      }\n    }\n  }\n\n  async runAfterRun(...args: Parameters<AfterRunHook>): Promise<void> {\n    const snapshot = [...this.extensions];\n    for (const ext of snapshot) {\n      if (!ext.afterRun) continue;\n      try {\n        await ext.afterRun(...args);\n      } catch (err) {\n        this.log?.error(`Extension \"${ext.name}\" afterRun hook failed`, err);\n      }\n    }\n  }\n\n  async runBeforeIteration(...args: Parameters<BeforeIterationHook>): Promise<void> {\n    const snapshot = [...this.extensions];\n    for (const ext of snapshot) {\n      if (!ext.beforeIteration) continue;\n      try {\n        await ext.beforeIteration(...args);\n      } catch (err) {\n        this.log?.error(`Extension \"${ext.name}\" beforeIteration hook failed`, err);\n      }\n    }\n  }\n\n  async runAfterIteration(...args: Parameters<AfterIterationHook>): Promise<void> {\n    const snapshot = [...this.extensions];\n    for (const ext of snapshot) {\n      if (!ext.afterIteration) continue;\n      try {\n        await ext.afterIteration(...args);\n      } catch (err) {\n        this.log?.error(`Extension \"${ext.name}\" afterIteration hook failed`, err);\n      }\n    }\n  }\n\n  /**\n   * Run onError hooks in order. The first hook that returns a non-void\n   * result wins; subsequent hooks are skipped.\n   */\n  async runOnError(\n    ...args: Parameters<OnErrorHook>\n  ): Promise<\n    { action: 'retry'; model?: string } | { action: 'fail' } | { action: 'continue' } | void\n  > {\n    const snapshot = [...this.extensions];\n    for (const ext of snapshot) {\n      if (!ext.onError) continue;\n      try {\n        const result = await ext.onError(...args);\n        if (result) return result;\n      } catch (err) {\n        this.log?.error(`Extension \"${ext.name}\" onError hook failed`, err);\n      }\n    }\n  }\n\n  /**\n   * Build a composed provider runner. Extensions with `wrapProviderRunner`\n   * form a middleware-style chain: the innermost extension wraps the\n   * default runner, each subsequent wrapper wraps the previous.\n   */\n  wrapProviderRunner(inner: ProviderRunnerFn): ProviderRunnerFn {\n    const wrappers = this.extensions\n      .filter((e) => e.wrapProviderRunner)\n      .map((e) => ({ name: e.name, wrap: e.wrapProviderRunner! }));\n\n    if (wrappers.length === 0) return inner;\n\n    // Build chain from innermost to outermost\n    let composed: ProviderRunnerFn = inner;\n    for (let i = wrappers.length - 1; i >= 0; i--) {\n      const wrapper = wrappers[i]!;\n      const next = composed;\n      composed = async (ctx, req) => {\n        try {\n          return await wrapper.wrap(ctx, req, next);\n        } catch (err) {\n          this.log?.error(`Extension \"${wrapper.name}\" wrapProviderRunner failed`, err);\n          throw err;\n        }\n      };\n    }\n    return composed;\n  }\n\n  async runBeforeToolExecution(\n    ...args: Parameters<BeforeToolExecutionHook>\n  ): Promise<Parameters<BeforeToolExecutionHook>[1]> {\n    let toolUses = args[1];\n    const snapshot = [...this.extensions];\n    for (const ext of snapshot) {\n      if (!ext.beforeToolExecution) continue;\n      try {\n        toolUses = await ext.beforeToolExecution(args[0], toolUses);\n      } catch (err) {\n        this.log?.error(`Extension \"${ext.name}\" beforeToolExecution hook failed`, err);\n      }\n    }\n    return toolUses;\n  }\n\n  async runAfterToolExecution(...args: Parameters<AfterToolExecutionHook>): Promise<void> {\n    const snapshot = [...this.extensions];\n    for (const ext of snapshot) {\n      if (!ext.afterToolExecution) continue;\n      try {\n        await ext.afterToolExecution(...args);\n      } catch (err) {\n        this.log?.error(`Extension \"${ext.name}\" afterToolExecution hook failed`, err);\n      }\n    }\n  }\n}\n"]}