@hydra-acp/budgeter 0.1.2

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Sam Magnuson
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,144 @@
+# hydra-acp-budgeter
+
+Cost-budget transformer extension for [hydra-acp](https://github.com/smagnuso/hydra-acp). Watches every session's `usage_update` events, warns attached clients when total spend crosses a soft limit, and rejects further prompts once it crosses a hard limit — with a human-readable reason returned to the client.
+
+Runs as a daemon-managed *transformer* (not a client extension): it connects once, declares its intercepts via `transformer/initialize`, and sits inside the daemon's message pipeline for every live session.
+
+## Install
+
+From npm (recommended once published):
+
+```sh
+npm install -g @hydra-acp/budgeter
+```
+
+This drops a `hydra-acp-budgeter` binary on your PATH.
+
+Or from source:
+
+```sh
+git clone git@github.com:smagnuso/hydra-acp-budgeter.git ~/dev/hydra-acp-budgeter
+cd ~/dev/hydra-acp-budgeter
+npm install
+npm run build
+```
+
+Register the transformer with hydra. If installed via npm:
+
+```sh
+hydra-acp transformers add hydra-acp-budgeter --command hydra-acp-budgeter
+```
+
+Or pointed at a local build:
+
+```sh
+hydra-acp transformers add hydra-acp-budgeter \
+  --command node \
+  --args ~/dev/hydra-acp-budgeter/dist/index.js
+```
+
+That registers the transformer but does **not** wire it into sessions yet. You also need to add it to `defaultTransformers` in `~/.hydra-acp/config.json` — there's no CLI command for this, edit the file directly:
+
+```json
+{
+  "transformers": {
+    "hydra-acp-budgeter": {
+      "command": ["node"],
+      "args": ["/home/you/dev/hydra-acp-budgeter/dist/index.js"]
+    }
+  },
+  "defaultTransformers": ["hydra-acp-budgeter"]
+}
+```
+
+Without `defaultTransformers`, the transformer process runs but no sessions route through it.
+
+On `hydra-acp daemon start`, hydra spawns hydra-acp-budgeter with these env
+vars set: `HYDRA_ACP_DAEMON_URL`, `HYDRA_ACP_TOKEN`, `HYDRA_ACP_WS_URL`,
+`HYDRA_ACP_HOME`, `HYDRA_ACP_TRANSFORMER_NAME`. Stdout/stderr land in
+`~/.hydra-acp/transformers/hydra-acp-budgeter.log`. Lifecycle is managed with
+`hydra-acp transformers start|stop|restart hydra-acp-budgeter` and
+`hydra-acp transformers log hydra-acp-budgeter -f` to tail.
+
+## Behavior
+
+Tracks each session's running cost from `usage_update` events (the `cost.amount` the agent reports, or `_meta.hydra-acp.cumulativeCost` when present), sums across sessions, and acts at two thresholds:
+
+**1. Soft limit crossed** (`total ≥ HYDRA_ACP_BUDGETER_SOFT`)
+- Emits a single warning `session/update` (`agent_message_chunk`) to every attached client on the session that triggered the cross.
+- Body: `Spent $X.XX of $S.SS soft (hard: $H.HH). Heads up — prompts will be rejected at the hard limit.`
+- Fires **once per upward transition**, not on every tick.
+
+**2. Hard limit crossed** (`total ≥ HYDRA_ACP_BUDGETER_HARD`)
+- Same one-shot warning, this time on the session that pushed us over.
+- Body: `Spent $X.XX ≥ $H.HH hard limit. Further prompts will be rejected until the budget is reset.`
+
+**3. Prompts while over hard limit**
+- Any `session/prompt` from any session is intercepted at `request:session/prompt` and replaced with a stop response:
+  ```json
+  {
+    "stopReason": "refusal",
+    "_meta": {
+      "hydra-acp": {
+        "budgeter": {
+          "message": "Budget exceeded: spent $X.XX ≥ $H.HH hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue."
+        }
+      }
+    }
+  }
+  ```
+- The agent never sees the prompt. The client's pending `session/prompt` resolves with the stop payload; well-behaved renderers (TUI, Zed, agent-shell) will surface the `stopReason: refusal` and the `_meta.hydra-acp.budgeter.message` body.
+
+**4. New session opens while over hard limit**
+- The budgeter fires the same warning style on `session.opened` so the user knows their next prompt will bounce, even before they send it.
+
+### State and reset
+
+The per-session cost map is persisted to `~/.hydra-acp/budgeter-cost.json`, atomically rewritten on every `usage_update`. The running budgeter reads it on startup and `fs.watch`es it for external mutations — so daemon restarts preserve the running total, and a reset from elsewhere is picked up live without restarting.
+
+Spend is sticky across `session.closed`: a closed session's cost stays in the total until you reset.
+
+To zero the budget:
+
+```sh
+hydra-acp-budgeter reset
+```
+
+That deletes the state file. If the transformer is running, its watcher adopts the deletion and the in-memory total drops to zero on the next tick (≤50ms). If it isn't running, the file is just gone and the next start begins at zero.
+
+## Configuration
+
+Create `~/.hydra-acp/budgeter.conf` (override path via `HYDRA_ACP_BUDGETER_CONF`):
+
+```
+# ~/.hydra-acp/budgeter.conf
+SOFT=5
+HARD=10
+CURRENCY=USD
+DEBUG=false
+```
+
+The file is optional — all keys have defaults and the transformer works without it. Environment variables always win over file values, so you can temporarily override a limit without editing the file.
+
+| Key / env var | Default | Purpose |
+|---|---|---|
+| `SOFT` / `HYDRA_ACP_BUDGETER_SOFT` | `5` | Soft limit (warning threshold) |
+| `HARD` / `HYDRA_ACP_BUDGETER_HARD` | `10` | Hard limit (rejection threshold). Must be ≥ soft. |
+| `CURRENCY` / `HYDRA_ACP_BUDGETER_CURRENCY` | `USD` | ISO-3 currency code for formatted messages |
+| `DEBUG` | `false` | Verbose logging |
+| `HYDRA_ACP_DAEMON_URL` | `http://127.0.0.1:8765` | Daemon HTTP endpoint (injected by hydra) |
+| `HYDRA_ACP_TOKEN` | *(required)* | Daemon auth token (injected by hydra) |
+| `HYDRA_ACP_WS_URL` | derived | Override WS endpoint |
+
+## How it works
+
+- Connects via WebSocket to `/acp`, calls `initialize`, then `transformer/initialize` declaring the intercepts:
+  - `response:session/update`   — observe `usage_update` to track cost
+  - `request:session/prompt`    — reject when over hard limit
+  - `lifecycle:session.opened`  — warn brand-new sessions that are already over budget
+  - `lifecycle:session.closed`  — fires session_closed rule event (cost stays sticky)
+- For every `transformer/message` the daemon dispatches, the budgeter responds with `{ action: "continue" }` (observe-only on response side, allow on request side when under budget) or `{ action: "stop", payload: { stopReason: "refusal", _meta: ... } }` (when over hard limit).
+- Warnings are emitted via `hydra-acp/emit_message` with `route: "chain"` and `method: "session/update"` so they flow back through the daemon's broadcast machinery and reach every attached client.
+- All cost state is in-memory; restart the transformer to reset.
+
+For a working example of the transformer protocol the budgeter speaks, see [`hydra-acp/cli/examples/transformer-observe.mjs`](https://github.com/smagnuso/hydra-acp/blob/main/cli/examples/transformer-observe.mjs).

package/dist/acp/protocol.js

@@ -0,0 +1,13 @@
+// ACP wire protocol version this transformer speaks. Single source of
+// truth for the initialize handshake; never a literal at the callsite.
+export const ACP_PROTOCOL_VERSION = 1;
+export function isRequest(m) {
+    return "method" in m && "id" in m;
+}
+export function isNotification(m) {
+    return "method" in m && !("id" in m);
+}
+export function isResponse(m) {
+    return !("method" in m) && "id" in m;
+}
+//# sourceMappingURL=protocol.js.map

package/dist/acp/protocol.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol.js","sourceRoot":"","sources":["../../src/acp/protocol.ts"],"names":[],"mappings":"AAAA,sEAAsE;AACtE,uEAAuE;AACvE,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC;AA6BtC,MAAM,UAAU,SAAS,CAAC,CAAiB;IACzC,OAAO,QAAQ,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,CAAiB;IAC9C,OAAO,QAAQ,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;AACvC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,CAAiB;IAC1C,OAAO,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,IAAI,IAAI,IAAI,CAAC,CAAC;AACvC,CAAC"}

package/dist/acp/transformer.js

@@ -0,0 +1,188 @@
+import { EventEmitter } from "node:events";
+import { readFileSync } from "node:fs";
+import { WebSocket } from "ws";
+import { logger } from "../util/log.js";
+import { ACP_PROTOCOL_VERSION, isNotification, isRequest, isResponse, } from "./protocol.js";
+const pkg = JSON.parse(readFileSync(new URL("../../package.json", import.meta.url), "utf8"));
+const log = logger("acp");
+// One WebSocket to the daemon for the entire transformer process. Unlike
+// the notifier's AcpAttach (one WS per session), the daemon multiplexes
+// every session's transformer/message + transformer/session_event over
+// this single connection, keyed by sessionId in the params.
+export class TransformerClient extends EventEmitter {
+    opts;
+    ws;
+    nextId = 1;
+    pending = new Map();
+    connected = false;
+    constructor(opts) {
+        super();
+        this.opts = opts;
+    }
+    get isConnected() {
+        return this.connected;
+    }
+    start() {
+        log.debug(`connecting ${this.opts.daemonWsUrl}`);
+        const subprotocols = ["acp.v1", `hydra-acp-token.${this.opts.token}`];
+        let ws;
+        try {
+            ws = new WebSocket(this.opts.daemonWsUrl, subprotocols);
+        }
+        catch (err) {
+            this.emit("error", err);
+            return;
+        }
+        this.ws = ws;
+        ws.on("open", () => {
+            this.connected = true;
+            log.info(`ws open`);
+            void this.handshake()
+                .then(() => {
+                this.emit("open");
+            })
+                .catch((err) => {
+                this.emit("error", err);
+                try {
+                    this.ws?.close();
+                }
+                catch {
+                    void 0;
+                }
+            });
+        });
+        ws.on("message", (data, isBinary) => {
+            if (isBinary) {
+                return;
+            }
+            const text = data.toString("utf8");
+            try {
+                const parsed = JSON.parse(text);
+                this.onMessage(parsed);
+            }
+            catch (err) {
+                log.warn(`parse error: ${err.message}; raw=${text.slice(0, 200)}`);
+            }
+        });
+        ws.on("error", (err) => {
+            log.warn(`ws error: ${err.message}`);
+            this.emit("error", err);
+        });
+        ws.on("close", (code, reason) => {
+            const hadError = code >= 4000 || code === 1006 || code === 1011;
+            const reasonText = reason.toString("utf8");
+            this.connected = false;
+            log.info(`ws closed code=${code}${reasonText ? ` reason=${reasonText}` : ""}`);
+            for (const [, p] of this.pending) {
+                p.reject(new Error("ws closed"));
+            }
+            this.pending.clear();
+            this.emit("close", { hadError });
+        });
+    }
+    stop() {
+        if (this.ws && this.ws.readyState !== WebSocket.CLOSED) {
+            try {
+                this.ws.close();
+            }
+            catch {
+                void 0;
+            }
+        }
+    }
+    async request(method, params) {
+        const id = this.nextId++;
+        const msg = {
+            jsonrpc: "2.0",
+            id,
+            method,
+            ...(params !== undefined ? { params } : {}),
+        };
+        this.write(msg);
+        return new Promise((resolve, reject) => {
+            this.pending.set(id, {
+                resolve: (resp) => {
+                    if (resp.error) {
+                        reject(new Error(`${resp.error.code}: ${resp.error.message}`));
+                    }
+                    else {
+                        resolve(resp.result);
+                    }
+                },
+                reject,
+            });
+        });
+    }
+    notify(method, params) {
+        const msg = {
+            jsonrpc: "2.0",
+            method,
+            ...(params !== undefined ? { params } : {}),
+        };
+        this.write(msg);
+    }
+    reply(id, result) {
+        const msg = { jsonrpc: "2.0", id, result };
+        this.write(msg);
+    }
+    replyError(id, code, message) {
+        const msg = {
+            jsonrpc: "2.0",
+            id,
+            error: { code, message },
+        };
+        this.write(msg);
+    }
+    async handshake() {
+        try {
+            await this.request("initialize", {
+                protocolVersion: ACP_PROTOCOL_VERSION,
+                clientCapabilities: {
+                    fs: { readTextFile: false, writeTextFile: false },
+                    terminal: false,
+                },
+                clientInfo: { name: "hydra-acp-budgeter", version: pkg.version },
+            });
+        }
+        catch (err) {
+            log.warn(`initialize failed: ${err.message}`);
+        }
+        try {
+            await this.request("transformer/initialize", {
+                intercepts: this.opts.intercepts,
+            });
+            log.info(`transformer/initialize ok intercepts=${this.opts.intercepts.join(",")}`);
+        }
+        catch (err) {
+            log.warn(`transformer/initialize failed: ${err.message}`);
+            throw err;
+        }
+    }
+    write(msg) {
+        if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
+            log.warn(`drop write to closed ws: ${JSON.stringify(msg)}`);
+            return;
+        }
+        this.ws.send(JSON.stringify(msg));
+    }
+    onMessage(m) {
+        if (isResponse(m)) {
+            const p = this.pending.get(m.id);
+            if (p) {
+                this.pending.delete(m.id);
+                p.resolve(m);
+            }
+            else {
+                log.debug(`unmatched response id=${String(m.id)}`);
+            }
+            this.emit("response", m);
+        }
+        else if (isRequest(m)) {
+            this.emit("request", m);
+        }
+        else if (isNotification(m)) {
+            this.emit("notification", m);
+        }
+    }
+}
+//# sourceMappingURL=transformer.js.map

package/dist/acp/transformer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transformer.js","sourceRoot":"","sources":["../../src/acp/transformer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EACL,oBAAoB,EAMpB,cAAc,EACd,SAAS,EACT,UAAU,GACX,MAAM,eAAe,CAAC;AAEvB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CACpB,YAAY,CAAC,IAAI,GAAG,CAAC,oBAAoB,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAC9C,CAAC;AAEzB,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;AAwB1B,yEAAyE;AACzE,wEAAwE;AACxE,uEAAuE;AACvE,4DAA4D;AAC5D,MAAM,OAAO,iBAAkB,SAAQ,YAAqC;IAM7C;IALrB,EAAE,CAAwB;IAC1B,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;IAC/C,SAAS,GAAG,KAAK,CAAC;IAE1B,YAA6B,IAA8B;QACzD,KAAK,EAAE,CAAC;QADmB,SAAI,GAAJ,IAAI,CAA0B;IAE3D,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,KAAK;QACH,GAAG,CAAC,KAAK,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACjD,MAAM,YAAY,GAAG,CAAC,QAAQ,EAAE,mBAAmB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACtE,IAAI,EAAa,CAAC;QAClB,IAAI,CAAC;YACH,EAAE,GAAG,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QAC1D,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAY,CAAC,CAAC;YACjC,OAAO;QACT,CAAC;QACD,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QAEb,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACpB,KAAK,IAAI,CAAC,SAAS,EAAE;iBAClB,IAAI,CAAC,GAAG,EAAE;gBACT,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpB,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;gBACtB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAY,CAAC,CAAC;gBACjC,IAAI,CAAC;oBACH,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC;gBACnB,CAAC;gBAAC,MAAM,CAAC;oBACP,KAAK,CAAC,CAAC;gBACT,CAAC;YACH,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE;YAClC,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO;YACT,CAAC;YACD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACnC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAmB,CAAC;gBAClD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACzB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,GAAG,CAAC,IAAI,CACN,gBAAiB,GAAa,CAAC,OAAO,SAAS,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CACpE,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACrB,GAAG,CAAC,IAAI,CAAC,aAAa,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACrC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAC9B,MAAM,QAAQ,GAAG,IAAI,IAAI,IAAI,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,IAAI,CAAC;YAChE,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,GAAG,CAAC,IAAI,CACN,kBAAkB,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,WAAW,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACrE,CAAC;YACF,KAAK,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjC,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YACnC,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,MAAM,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAClB,CAAC;YAAC,MAAM,CAAC;gBACP,KAAK,CAAC,CAAC;YACT,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,OAAO,CAAc,MAAc,EAAE,MAAgB;QACzD,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,GAAG,GAAmB;YAC1B,OAAO,EAAE,KAAK;YACd,EAAE;YACF,MAAM;YACN,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5C,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChB,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE;gBACnB,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;oBAChB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;wBACf,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;oBACjE,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,IAAI,CAAC,MAAW,CAAC,CAAC;oBAC5B,CAAC;gBACH,CAAC;gBACD,MAAM;aACP,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,MAAc,EAAE,MAAgB;QACrC,MAAM,GAAG,GAAwB;YAC/B,OAAO,EAAE,KAAK;YACd,MAAM;YACN,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5C,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,EAAa,EAAE,MAAe;QAClC,MAAM,GAAG,GAAoB,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC;QAC5D,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,CAAC;IAED,UAAU,CAAC,EAAa,EAAE,IAAY,EAAE,OAAe;QACrD,MAAM,GAAG,GAAoB;YAC3B,OAAO,EAAE,KAAK;YACd,EAAE;YACF,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;SACzB,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE;gBAC/B,eAAe,EAAE,oBAAoB;gBACrC,kBAAkB,EAAE;oBAClB,EAAE,EAAE,EAAE,YAAY,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE;oBACjD,QAAQ,EAAE,KAAK;iBAChB;gBACD,UAAU,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE;aACjE,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,IAAI,CAAC,sBAAuB,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,wBAAwB,EAAE;gBAC3C,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;aACjC,CAAC,CAAC;YACH,GAAG,CAAC,IAAI,CACN,wCAAwC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CACzE,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,IAAI,CAAC,kCAAmC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YACrE,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,GAAmB;QAC/B,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACtD,GAAG,CAAC,IAAI,CAAC,4BAA4B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC5D,OAAO;QACT,CAAC;QACD,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC;IAEO,SAAS,CAAC,CAAiB;QACjC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;YAClB,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACjC,IAAI,CAAC,EAAE,CAAC;gBACN,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBAC1B,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACf,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,KAAK,CAAC,yBAAyB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YACrD,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QAC3B,CAAC;aAAM,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QAC1B,CAAC;aAAM,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;CACF"}

package/dist/bridge.js

@@ -0,0 +1,196 @@
+import { watch } from "node:fs";
+import { basename, dirname } from "node:path";
+import { TransformerClient } from "./acp/transformer.js";
+import { Enforcer } from "./enforce.js";
+import { EventRouter } from "./router.js";
+import { CostTracker } from "./tracker.js";
+import { logger } from "./util/log.js";
+const log = logger("bridge");
+// The set of intercepts the budgeter declares to the daemon. Kept in one
+// place so the rule, the router, and the README stay in agreement.
+//
+//   response:session/update    — observe usage_update for cost tracking
+//   request:session/prompt     — reject when over hard limit
+//   lifecycle:session.opened   — warn on new sessions while over budget
+//   lifecycle:session.closed   — drop per-session cost state
+const BUDGETER_INTERCEPTS = [
+    "response:session/update",
+    "request:session/prompt",
+    "lifecycle:session.opened",
+    "lifecycle:session.closed",
+];
+// One bridge per budgeter process. Owns the WS connection, the cost
+// tracker, the enforcer, and the router. The mirror of NotifierBridge,
+// except scoped to the whole process rather than to a single session.
+export class BudgeterBridge {
+    opts;
+    client;
+    tracker;
+    enforcer;
+    router;
+    watcher;
+    watchTimer;
+    stopped = false;
+    constructor(opts) {
+        this.opts = opts;
+        this.client = new TransformerClient({
+            daemonWsUrl: opts.daemonWsUrl,
+            token: opts.token,
+            intercepts: BUDGETER_INTERCEPTS,
+        });
+        this.tracker = new CostTracker({
+            softLimit: opts.softLimit,
+            hardLimit: opts.hardLimit,
+            currency: opts.currency,
+            statePath: opts.statePath,
+        });
+        this.enforcer = new Enforcer(this.client, log);
+        this.router = new EventRouter(opts.rule, this.tracker, this.enforcer, log);
+    }
+    start() {
+        this.client.on("request", (r) => this.onRequest(r));
+        this.client.on("notification", (n) => this.onNotification(n));
+        this.client.on("error", (err) => {
+            log.warn(`client error: ${err.message}`);
+        });
+        this.client.start();
+        this.startWatcher();
+    }
+    stop() {
+        if (this.stopped) {
+            return;
+        }
+        this.stopped = true;
+        if (this.watcher) {
+            this.watcher.close();
+            this.watcher = undefined;
+        }
+        if (this.watchTimer) {
+            clearTimeout(this.watchTimer);
+            this.watchTimer = undefined;
+        }
+        this.client.stop();
+    }
+    // Watch the state file's parent directory so we still see events when
+    // the file is created later (e.g. first usage_update of a fresh run)
+    // or deleted entirely (the reset subcommand). fs.watch on a missing
+    // file throws on some platforms, but the parent dir is created by
+    // CostTracker.persist before any write happens, and the daemon writes
+    // the .pid file there even sooner — so the dir reliably exists by
+    // the time start() runs. We still try/catch in case it doesn't.
+    startWatcher() {
+        if (!this.opts.statePath) {
+            return;
+        }
+        const dir = dirname(this.opts.statePath);
+        const file = basename(this.opts.statePath);
+        try {
+            this.watcher = watch(dir, (eventType, filename) => {
+                if (filename && filename !== file) {
+                    return;
+                }
+                // fs.watch can fire 1–N times per logical change (especially
+                // when our own atomic-rename hits it). Debounce briefly so we
+                // do at most one re-read per burst.
+                if (this.watchTimer) {
+                    return;
+                }
+                this.watchTimer = setTimeout(() => {
+                    this.watchTimer = undefined;
+                    try {
+                        this.tracker.adoptFromDisk();
+                    }
+                    catch (err) {
+                        log.warn(`adoptFromDisk failed: ${err.message}`);
+                    }
+                }, 50);
+            });
+            log.debug(`watching ${this.opts.statePath}`);
+        }
+        catch (err) {
+            log.warn(`fs.watch failed for ${dir}: ${err.message}`);
+        }
+    }
+    onRequest(r) {
+        if (r.method !== "transformer/message") {
+            // The daemon only sends transformer/message requests to us. Anything
+            // else is an error on the daemon side or a future protocol kind we
+            // don't yet understand. Continue rather than guessing.
+            this.client.reply(r.id, { action: "continue" });
+            return;
+        }
+        const params = (r.params ?? {});
+        const sessionId = typeof params.sessionId === "string" ? params.sessionId : "";
+        const phase = params.phase;
+        const method = params.method;
+        if (!sessionId || !phase || !method) {
+            this.client.reply(r.id, { action: "continue" });
+            return;
+        }
+        if (phase === "response" && method === "session/update") {
+            // Response side is observe-only — we always continue and run the
+            // tracker update asynchronously. The daemon proceeds with the
+            // original envelope; any warning we want to emit goes out via a
+            // separate emit_message call.
+            this.client.reply(r.id, { action: "continue" });
+            const envelope = params.envelope;
+            const update = envelope && typeof envelope === "object" && !Array.isArray(envelope)
+                ? envelope.update
+                : undefined;
+            if (update) {
+                void this.router
+                    .onResponseUpdate(sessionId, update)
+                    .catch((err) => log.warn(`response update error: ${err.message}`));
+            }
+            return;
+        }
+        if (phase === "request" && method === "session/prompt") {
+            // Request side: ask the rule. If it returns a reject payload, stop;
+            // otherwise let the prompt continue. The reply waits on the rule —
+            // the daemon's forwardRequest is awaiting our response, so a slow
+            // rule pauses the prompt, which is acceptable for the rare reject
+            // path.
+            void this.router
+                .onPromptRequest(sessionId, params.envelope)
+                .then((rejectPayload) => {
+                if (rejectPayload) {
+                    this.client.reply(r.id, { action: "stop", payload: rejectPayload });
+                }
+                else {
+                    this.client.reply(r.id, { action: "continue" });
+                }
+            })
+                .catch((err) => {
+                log.warn(`prompt request error: ${err.message}`);
+                this.client.reply(r.id, { action: "continue" });
+            });
+            return;
+        }
+        // Unknown phase/method — declare nothing and let the daemon proceed.
+        this.client.reply(r.id, { action: "continue" });
+    }
+    onNotification(n) {
+        if (n.method !== "transformer/session_event") {
+            return;
+        }
+        const params = (n.params ?? {});
+        const event = typeof params.event === "string" ? params.event : "";
+        const sessionId = typeof params.sessionId === "string" ? params.sessionId : "";
+        if (!event || !sessionId) {
+            return;
+        }
+        if (event === "session.opened") {
+            void this.router
+                .onLifecycle("session.opened", sessionId)
+                .catch((err) => log.warn(`session.opened error: ${err.message}`));
+            return;
+        }
+        if (event === "session.closed") {
+            void this.router
+                .onLifecycle("session.closed", sessionId)
+                .catch((err) => log.warn(`session.closed error: ${err.message}`));
+            return;
+        }
+    }
+}
+//# sourceMappingURL=bridge.js.map

package/dist/bridge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bridge.js","sourceRoot":"","sources":["../src/bridge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAkB,MAAM,SAAS,CAAC;AAChD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAOzD,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAEvC,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;AAgB7B,yEAAyE;AACzE,mEAAmE;AACnE,EAAE;AACF,wEAAwE;AACxE,6DAA6D;AAC7D,wEAAwE;AACxE,6DAA6D;AAC7D,MAAM,mBAAmB,GAAG;IAC1B,yBAAyB;IACzB,wBAAwB;IACxB,0BAA0B;IAC1B,0BAA0B;CAC3B,CAAC;AAEF,oEAAoE;AACpE,uEAAuE;AACvE,sEAAsE;AACtE,MAAM,OAAO,cAAc;IASI;IARZ,MAAM,CAAoB;IAC1B,OAAO,CAAc;IACrB,QAAQ,CAAW;IACnB,MAAM,CAAc;IAC7B,OAAO,CAAwB;IAC/B,UAAU,CAA6B;IACvC,OAAO,GAAG,KAAK,CAAC;IAExB,YAA6B,IAAmB;QAAnB,SAAI,GAAJ,IAAI,CAAe;QAC9C,IAAI,CAAC,MAAM,GAAG,IAAI,iBAAiB,CAAC;YAClC,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU,EAAE,mBAAmB;SAChC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,GAAG,IAAI,WAAW,CAAC;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAC3B,IAAI,CAAC,IAAI,EACT,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,QAAQ,EACb,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YAC9B,GAAG,CAAC,IAAI,CAAC,iBAAiB,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACpB,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAED,IAAI;QACF,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;QAC3B,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC9B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC9B,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IACrB,CAAC;IAED,sEAAsE;IACtE,qEAAqE;IACrE,oEAAoE;IACpE,kEAAkE;IAClE,sEAAsE;IACtE,kEAAkE;IAClE,gEAAgE;IACxD,YAAY;QAClB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QACD,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,CAAC;YACH,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,EAAE,QAAQ,EAAE,EAAE;gBAChD,IAAI,QAAQ,IAAI,QAAQ,KAAK,IAAI,EAAE,CAAC;oBAClC,OAAO;gBACT,CAAC;gBACD,6DAA6D;gBAC7D,8DAA8D;gBAC9D,oCAAoC;gBACpC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACpB,OAAO;gBACT,CAAC;gBACD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,GAAG,EAAE;oBAChC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;oBAC5B,IAAI,CAAC;wBACH,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;oBAC/B,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,GAAG,CAAC,IAAI,CAAC,yBAA0B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC9D,CAAC;gBACH,CAAC,EAAE,EAAE,CAAC,CAAC;YACT,CAAC,CAAC,CAAC;YACH,GAAG,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,IAAI,CAAC,uBAAuB,GAAG,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,CAAiB;QACjC,IAAI,CAAC,CAAC,MAAM,KAAK,qBAAqB,EAAE,CAAC;YACvC,qEAAqE;YACrE,mEAAmE;YACnE,uDAAuD;YACvD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAChD,OAAO;QACT,CAAC;QACD,MAAM,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM,IAAI,EAAE,CAAsC,CAAC;QACrE,MAAM,SAAS,GAAG,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC3B,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,SAAS,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAChD,OAAO;QACT,CAAC;QAED,IAAI,KAAK,KAAK,UAAU,IAAI,MAAM,KAAK,gBAAgB,EAAE,CAAC;YACxD,iEAAiE;YACjE,8DAA8D;YAC9D,gEAAgE;YAChE,8BAA8B;YAC9B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;YACjC,MAAM,MAAM,GACV,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;gBAClE,CAAC,CAAG,QAAiC,CAAC,MAEtB;gBAChB,CAAC,CAAC,SAAS,CAAC;YAChB,IAAI,MAAM,EAAE,CAAC;gBACX,KAAK,IAAI,CAAC,MAAM;qBACb,gBAAgB,CAAC,SAAS,EAAE,MAAM,CAAC;qBACnC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,0BAA2B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAClF,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,KAAK,KAAK,SAAS,IAAI,MAAM,KAAK,gBAAgB,EAAE,CAAC;YACvD,oEAAoE;YACpE,mEAAmE;YACnE,kEAAkE;YAClE,kEAAkE;YAClE,QAAQ;YACR,KAAK,IAAI,CAAC,MAAM;iBACb,eAAe,CAAC,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC;iBAC3C,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE;gBACtB,IAAI,aAAa,EAAE,CAAC;oBAClB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;gBACtE,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACb,GAAG,CAAC,IAAI,CAAC,yBAA0B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC5D,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAClD,CAAC,CAAC,CAAC;YACL,OAAO;QACT,CAAC;QAED,qEAAqE;QACrE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;IAClD,CAAC;IAEO,cAAc,CAAC,CAAsB;QAC3C,IAAI,CAAC,CAAC,MAAM,KAAK,2BAA2B,EAAE,CAAC;YAC7C,OAAO;QACT,CAAC;QACD,MAAM,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM,IAAI,EAAE,CAAqC,CAAC;QACpE,MAAM,KAAK,GAAG,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,MAAM,SAAS,GAAG,OAAO,MAAM,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/E,IAAI,CAAC,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QACD,IAAI,KAAK,KAAK,gBAAgB,EAAE,CAAC;YAC/B,KAAK,IAAI,CAAC,MAAM;iBACb,WAAW,CAAC,gBAAgB,EAAE,SAAS,CAAC;iBACxC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,yBAA0B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC/E,OAAO;QACT,CAAC;QACD,IAAI,KAAK,KAAK,gBAAgB,EAAE,CAAC;YAC/B,KAAK,IAAI,CAAC,MAAM;iBACb,WAAW,CAAC,gBAAgB,EAAE,SAAS,CAAC;iBACxC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,yBAA0B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC/E,OAAO;QACT,CAAC;IACH,CAAC;CACF"}