@hydra-acp/budgeter 0.1.4 → 0.1.5

package/dist/index.js

@@ -1,65 +1,5 @@
 #!/usr/bin/env node
-import { readFileSync } from "node:fs";
-import { dirname, resolve } from "node:path";
-import { fileURLToPath } from "node:url";
-import { loadConfig } from "./config.js";
-import { BudgeterBridge } from "./bridge.js";
-import { stateFilePath } from "./paths.js";
-import { DEFAULT_RULE } from "./rule.js";
-import { deleteStateFile } from "./tracker.js";
-import { logger, setDebug } from "./util/log.js";
-const log = logger("main");
-function readVersion() {
-    try {
-        const here = dirname(fileURLToPath(import.meta.url));
-        const pkg = JSON.parse(readFileSync(resolve(here, "../package.json"), "utf8"));
-        return pkg.version ?? "unknown";
-    }
-    catch {
-        return "unknown";
-    }
-}
-function runReset() {
-    deleteStateFile(stateFilePath());
-    process.stdout.write("hydra-acp-budgeter accumulated cost reset\n");
-}
-async function runTransformer() {
-    const config = loadConfig();
-    setDebug(config.debug);
-    const statePath = stateFilePath();
-    const bridge = new BudgeterBridge({
-        daemonWsUrl: config.hydraWsUrl,
-        token: config.hydraToken,
-        softLimit: config.softLimit,
-        hardLimit: config.hardLimit,
-        currency: config.currency,
-        rule: DEFAULT_RULE,
-        statePath,
-    });
-    bridge.start();
-    const shutdown = (sig) => {
-        log.info(`${sig} received — shutting down`);
-        bridge.stop();
-        setTimeout(() => process.exit(0), 200).unref();
-    };
-    process.on("SIGINT", () => shutdown("SIGINT"));
-    process.on("SIGTERM", () => shutdown("SIGTERM"));
-    log.info(`hydra-acp-budgeter up; daemon=${config.hydraDaemonUrl} soft=${config.softLimit} hard=${config.hardLimit} ${config.currency} state=${statePath}`);
-}
-async function main() {
-    const argv = process.argv.slice(2);
-    if (argv.includes("--version") || argv.includes("-v")) {
-        process.stdout.write(`hydra-acp-budgeter ${readVersion()}\n`);
-        return;
-    }
-    if (argv[0] === "reset") {
-        runReset();
-        return;
-    }
-    await runTransformer();
-}
-main().catch((err) => {
-    process.stderr.write(`hydra-acp-budgeter: ${err.message}\n`);
-    process.exit(1);
-});
-//# sourceMappingURL=index.js.map
+import{readFileSync as a}from"node:fs";import{dirname as c,resolve as d}from"node:path";import{fileURLToPath as m}from"node:url";import{loadConfig as u}from"./config.js";import{BudgeterBridge as p}from"./bridge.js";import{stateFilePath as n}from"./paths.js";import{DEFAULT_RULE as f}from"./rule.js";import{deleteStateFile as g}from"./tracker.js";import{logger as h,setDebug as l}from"./util/log.js";const s=h("main");function y(){try{const r=c(m(import.meta.url));return JSON.parse(a(d(r,"../package.json"),"utf8")).version??"unknown"}catch{return"unknown"}}function v(){g(n()),process.stdout.write(`hydra-acp-budgeter accumulated cost reset
+`)}async function w(){const r=u();l(r.debug);const e=n(),t=new p({daemonWsUrl:r.hydraWsUrl,token:r.hydraToken,softLimit:r.softLimit,hardLimit:r.hardLimit,currency:r.currency,rule:f,statePath:e});t.start();const o=i=>{s.info(`${i} received \u2014 shutting down`),t.stop(),setTimeout(()=>process.exit(0),200).unref()};process.on("SIGINT",()=>o("SIGINT")),process.on("SIGTERM",()=>o("SIGTERM")),s.info(`hydra-acp-budgeter up; daemon=${r.hydraDaemonUrl} soft=${r.softLimit} hard=${r.hardLimit} ${r.currency} state=${e}`)}async function L(){const r=process.argv.slice(2);if(r.includes("--version")||r.includes("-v")){process.stdout.write(`hydra-acp-budgeter ${y()}
+`);return}if(r[0]==="reset"){v();return}await w()}L().catch(r=>{process.stderr.write(`hydra-acp-budgeter: ${r.message}
+`),process.exit(1)});

package/dist/paths.js

@@ -1,12 +1 @@
-import { homedir } from "node:os";
-import { resolve } from "node:path";
-// hydra-acp injects HYDRA_ACP_HOME when it spawns transformers; this file
-// is invoked both in that role and as a one-shot subcommand from the
-// user's shell, so fall back to ~/.hydra-acp when the env isn't set.
-function hydraHome() {
-    return process.env.HYDRA_ACP_HOME ?? resolve(homedir(), ".hydra-acp");
-}
-export function stateFilePath() {
-    return resolve(hydraHome(), "budgeter-cost.json");
-}
-//# sourceMappingURL=paths.js.map
+import{homedir as t}from"node:os";import{resolve as r}from"node:path";function e(){return process.env.HYDRA_ACP_HOME??r(t(),".hydra-acp")}function i(){return r(e(),"budgeter-cost.json")}export{i as stateFilePath};

package/dist/router.js

@@ -1,161 +1 @@
-// Translates incoming hydra-acp/transformer/message and hydra-acp/transformer/session_event
-// envelopes into BudgetEvents, runs them through the rule, and dispatches
-// the resulting verdicts via the Enforcer. Unlike the notifier's
-// EventRouter (one instance per session), this is a singleton — the
-// transformer process holds one connection for every session, so the
-// router keeps a sessionId → SessionMeta map.
-export class EventRouter {
-    rule;
-    tracker;
-    enforcer;
-    log;
-    metas = new Map();
-    constructor(rule, tracker, enforcer, log) {
-        this.rule = rule;
-        this.tracker = tracker;
-        this.enforcer = enforcer;
-        this.log = log;
-    }
-    setMeta(sessionId, meta) {
-        this.metas.set(sessionId, meta);
-    }
-    forgetSession(sessionId) {
-        this.metas.delete(sessionId);
-    }
-    // Fired by the bridge when an agent→client session/update notification
-    // is being routed. usage_update is the only kind we care about for cost
-    // tracking; other kinds may still flow through the rule so a custom
-    // rule can observe (but the default rule ignores them).
-    async onResponseUpdate(sessionId, update) {
-        const kind = typeof update.sessionUpdate === "string" ? update.sessionUpdate : "";
-        // session_info_update may carry an updated title; mirror the notifier
-        // and refresh meta so subsequent warning messages reflect the latest.
-        if (kind === "session_info_update") {
-            this.applySessionInfoUpdate(sessionId, update);
-        }
-        if (kind !== "usage_update") {
-            return;
-        }
-        this.tracker.applyUsageUpdate(sessionId, update);
-        const transition = this.tracker.consumeStateTransition();
-        if (transition !== undefined) {
-            await this.fire({
-                sessionId,
-                kind: "threshold_cross",
-                raw: { to: transition, from: this.tracker.state },
-                meta: this.metaFor(sessionId),
-                budget: this.tracker.snapshotFor(sessionId),
-            });
-        }
-        // Also run a plain usage_update event so a custom rule can act on
-        // every tick (e.g. emit a status notification every $1 spent). The
-        // default rule ignores this kind.
-        await this.fire({
-            sessionId,
-            kind: "usage_update",
-            raw: update,
-            meta: this.metaFor(sessionId),
-            budget: this.tracker.snapshotFor(sessionId),
-        });
-    }
-    // Fired by the bridge for a request:session/prompt intercept. Returns
-    // the stop payload to send back to the daemon, or undefined to let the
-    // prompt continue. The rule decides — the default rule rejects when
-    // we're in the "hard" state.
-    async onPromptRequest(sessionId, envelope) {
-        const verdict = await this.runRule({
-            sessionId,
-            kind: "prompt_request",
-            raw: { envelope },
-            meta: this.metaFor(sessionId),
-            budget: this.tracker.snapshotFor(sessionId),
-        });
-        if (!verdict) {
-            return undefined;
-        }
-        return this.enforcer.dispatch(sessionId, verdict);
-    }
-    // Fired by the bridge for lifecycle:session.opened and session.closed.
-    async onLifecycle(event, sessionId) {
-        if (event === "session.opened") {
-            await this.fire({
-                sessionId,
-                kind: "session_opened",
-                raw: {},
-                meta: this.metaFor(sessionId),
-                budget: this.tracker.snapshotFor(sessionId),
-            });
-            return;
-        }
-        if (event === "session.closed") {
-            // Note: we deliberately do NOT drop the session's cost from the
-            // tracker — total spend is sticky across session.closed so the
-            // budget reflects every dollar this transformer has seen. Reset
-            // via `hydra-acp-budgeter reset` (deletes the state file) when
-            // you want to zero it.
-            await this.fire({
-                sessionId,
-                kind: "session_closed",
-                raw: {},
-                meta: this.metaFor(sessionId),
-                budget: this.tracker.snapshotFor(sessionId),
-            });
-            this.forgetSession(sessionId);
-        }
-    }
-    async fire(ev) {
-        const verdict = await this.runRule(ev);
-        if (!verdict) {
-            return;
-        }
-        // For non-prompt events, only the warn side can fire — reject without
-        // a prompt to attach it to is a no-op. Log and skip so a misconfigured
-        // rule doesn't fail silently.
-        if (verdict.reject && ev.kind !== "prompt_request") {
-            this.log.warn(`rule returned reject on kind=${ev.kind} for ${ev.sessionId.slice(0, 12)} — only valid on prompt_request; ignored`);
-        }
-        if (verdict.warn) {
-            await this.enforcer.warn(ev.sessionId, verdict.warn);
-        }
-    }
-    async runRule(ev) {
-        try {
-            return await this.rule(ev);
-        }
-        catch (err) {
-            this.log.warn(`rule threw on kind=${ev.kind} sessionId=${ev.sessionId.slice(0, 12)}: ${err.message}; skipping`);
-            return null;
-        }
-    }
-    metaFor(sessionId) {
-        return this.metas.get(sessionId) ?? {};
-    }
-    applySessionInfoUpdate(sessionId, update) {
-        const cur = { ...this.metaFor(sessionId) };
-        let changed = false;
-        if (typeof update.title === "string" && cur.title !== update.title) {
-            cur.title = update.title;
-            changed = true;
-        }
-        const agentId = readHydraAgentId(update._meta);
-        if (agentId !== undefined && cur.agentId !== agentId) {
-            cur.agentId = agentId;
-            changed = true;
-        }
-        if (changed) {
-            this.metas.set(sessionId, cur);
-        }
-    }
-}
-function readHydraAgentId(meta) {
-    if (!meta || typeof meta !== "object" || Array.isArray(meta)) {
-        return undefined;
-    }
-    const ns = meta["hydra-acp"];
-    if (!ns || typeof ns !== "object" || Array.isArray(ns)) {
-        return undefined;
-    }
-    const v = ns.agentId;
-    return typeof v === "string" ? v : undefined;
-}
-//# sourceMappingURL=router.js.map
+class a{constructor(t,e,r,n){this.rule=t;this.tracker=e;this.enforcer=r;this.log=n}rule;tracker;enforcer;log;metas=new Map;setMeta(t,e){this.metas.set(t,e)}forgetSession(t){this.metas.delete(t)}async onResponseUpdate(t,e){const r=typeof e.sessionUpdate=="string"?e.sessionUpdate:"";if(r==="session_info_update"&&this.applySessionInfoUpdate(t,e),r!=="usage_update")return;this.tracker.applyUsageUpdate(t,e);const n=this.tracker.consumeStateTransition();n!==void 0&&await this.fire({sessionId:t,kind:"threshold_cross",raw:{to:n,from:this.tracker.state},meta:this.metaFor(t),budget:this.tracker.snapshotFor(t)}),await this.fire({sessionId:t,kind:"usage_update",raw:e,meta:this.metaFor(t),budget:this.tracker.snapshotFor(t)})}async onPromptRequest(t,e){const r=await this.runRule({sessionId:t,kind:"prompt_request",raw:{envelope:e},meta:this.metaFor(t),budget:this.tracker.snapshotFor(t)});if(r)return this.enforcer.dispatch(t,r)}async onLifecycle(t,e){if(t==="session.opened"){await this.fire({sessionId:e,kind:"session_opened",raw:{},meta:this.metaFor(e),budget:this.tracker.snapshotFor(e)});return}t==="session.closed"&&(await this.fire({sessionId:e,kind:"session_closed",raw:{},meta:this.metaFor(e),budget:this.tracker.snapshotFor(e)}),this.forgetSession(e))}async fire(t){const e=await this.runRule(t);e&&(e.reject&&t.kind!=="prompt_request"&&this.log.warn(`rule returned reject on kind=${t.kind} for ${t.sessionId.slice(0,12)} \u2014 only valid on prompt_request; ignored`),e.warn&&await this.enforcer.warn(t.sessionId,e.warn))}async runRule(t){try{return await this.rule(t)}catch(e){return this.log.warn(`rule threw on kind=${t.kind} sessionId=${t.sessionId.slice(0,12)}: ${e.message}; skipping`),null}}metaFor(t){return this.metas.get(t)??{}}applySessionInfoUpdate(t,e){const r={...this.metaFor(t)};let n=!1;typeof e.title=="string"&&r.title!==e.title&&(r.title=e.title,n=!0);const s=o(e._meta);s!==void 0&&r.agentId!==s&&(r.agentId=s,n=!0),n&&this.metas.set(t,r)}}function o(i){if(!i||typeof i!="object"||Array.isArray(i))return;const t=i["hydra-acp"];if(!t||typeof t!="object"||Array.isArray(t))return;const e=t.agentId;return typeof e=="string"?e:void 0}export{a as EventRouter};

package/dist/rule.js

@@ -1,86 +1 @@
-const CURRENCY_SYMBOLS = {
-    USD: "$",
-    EUR: "€",
-    GBP: "£",
-    JPY: "¥",
-    CNY: "¥",
-    CAD: "C$",
-    AUD: "A$",
-    NZD: "NZ$",
-    CHF: "CHF ",
-    HKD: "HK$",
-    SGD: "S$",
-    INR: "₹",
-    KRW: "₩",
-    BRL: "R$",
-    MXN: "MX$",
-};
-function fmtMoney(amount, currency) {
-    const fixed = amount.toFixed(2);
-    const sym = CURRENCY_SYMBOLS[currency.toUpperCase()];
-    if (sym) {
-        return `${sym}${fixed}`;
-    }
-    return `${currency} ${fixed}`;
-}
-// The rule the budgeter runs on every event. Strategy:
-//   - threshold_cross to "soft": warn (one-shot, the tracker only fires
-//     crosses once per transition)
-//   - threshold_cross to "hard": warn (also one-shot)
-//   - prompt_request in "hard" state: reject with stopReason "refusal"
-//     and an explanation in the message
-//   - session_opened in "hard" state: warn the new session so the user
-//     understands why their next prompt will bounce
-//   - usage_update and session_closed: do nothing
-export const DEFAULT_RULE = (ev) => {
-    const { budget } = ev;
-    const totalStr = fmtMoney(budget.total, budget.currency);
-    const softStr = fmtMoney(budget.soft, budget.currency);
-    const hardStr = fmtMoney(budget.hard, budget.currency);
-    if (ev.kind === "threshold_cross") {
-        const to = (ev.raw.to ?? "ok");
-        if (to === "hard") {
-            return {
-                warn: {
-                    title: `🛑 Budget hard limit hit`,
-                    body: `Spent ${totalStr} ≥ ${hardStr} hard limit. Further prompts will be rejected until the budget is reset.`,
-                },
-            };
-        }
-        return null;
-    }
-    // Warn on every turn that reports a cost while over the soft limit.
-    if (ev.kind === "usage_update" && budget.state !== "ok" && typeof ev.raw.cost?.amount === "number") {
-        const label = budget.state === "hard" ? "🛑 Over hard limit" : "💰 Over soft limit";
-        return {
-            warn: {
-                title: `${label} · ${totalStr} spent`,
-                body: budget.state === "hard"
-                    ? `Hard limit ${hardStr} reached. Prompts will be rejected until budget is reset.`
-                    : `Soft limit ${softStr} reached (hard: ${hardStr}).`,
-            },
-        };
-    }
-    if (ev.kind === "prompt_request" && budget.state === "hard") {
-        return {
-            warn: {
-                title: `🛑 Prompt blocked — budget exceeded`,
-                body: `Spent ${totalStr} ≥ ${hardStr} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`,
-            },
-            reject: {
-                message: `Budget exceeded: spent ${totalStr} ≥ ${hardStr} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`,
-                stopReason: "refusal",
-            },
-        };
-    }
-    if (ev.kind === "session_opened" && budget.state === "hard") {
-        return {
-            warn: {
-                title: `🛑 Session opened over budget`,
-                body: `Total spend ${totalStr} ≥ ${hardStr}. Prompts on this session will be rejected.`,
-            },
-        };
-    }
-    return null;
-};
-//# sourceMappingURL=rule.js.map
+const d={USD:"$",EUR:"\u20AC",GBP:"\xA3",JPY:"\xA5",CNY:"\xA5",CAD:"C$",AUD:"A$",NZD:"NZ$",CHF:"CHF ",HKD:"HK$",SGD:"S$",INR:"\u20B9",KRW:"\u20A9",BRL:"R$",MXN:"MX$"};function o(e,t){const r=e.toFixed(2),s=d[t.toUpperCase()];return s?`${s}${r}`:`${t} ${r}`}const u=e=>{const{budget:t}=e,r=o(t.total,t.currency),s=o(t.soft,t.currency),n=o(t.hard,t.currency);return e.kind==="threshold_cross"?(e.raw.to??"ok")==="hard"?{warn:{title:"\u{1F6D1} Budget hard limit hit",body:`Spent ${r} \u2265 ${n} hard limit. Further prompts will be rejected until the budget is reset.`}}:null:e.kind==="usage_update"&&t.state!=="ok"&&typeof e.raw.cost?.amount=="number"?{warn:{title:`${t.state==="hard"?"\u{1F6D1} Over hard limit":"\u{1F4B0} Over soft limit"} \xB7 ${r} spent`,body:t.state==="hard"?`Hard limit ${n} reached. Prompts will be rejected until budget is reset.`:`Soft limit ${s} reached (hard: ${n}).`}}:e.kind==="prompt_request"&&t.state==="hard"?{warn:{title:"\u{1F6D1} Prompt blocked \u2014 budget exceeded",body:`Spent ${r} \u2265 ${n} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`},reject:{message:`Budget exceeded: spent ${r} \u2265 ${n} hard limit. Reset the budget or raise HYDRA_ACP_BUDGETER_HARD to continue.`,stopReason:"refusal"}}:e.kind==="session_opened"&&t.state==="hard"?{warn:{title:"\u{1F6D1} Session opened over budget",body:`Total spend ${r} \u2265 ${n}. Prompts on this session will be rejected.`}}:null};export{u as DEFAULT_RULE};

package/dist/tracker.js

@@ -1,264 +1 @@
-import { mkdirSync, readFileSync, renameSync, unlinkSync, writeFileSync } from "node:fs";
-import { dirname } from "node:path";
-import { logger } from "./util/log.js";
-const log = logger("tracker");
-export class CostTracker {
-    opts;
-    sessions = new Map();
-    currentState = "ok";
-    // Last JSON we wrote to disk. The watcher uses this to ignore its own
-    // writes — if the file's content matches, we did it.
-    lastWrittenJson = "";
-    constructor(opts) {
-        this.opts = opts;
-        if (opts.statePath) {
-            this.loadFromDisk();
-        }
-    }
-    applyUsageUpdate(sessionId, update) {
-        log.debug(`usage_update session=${sessionId.slice(0, 12)} raw=${JSON.stringify(update).slice(0, 200)}`);
-        const cost = readCost(update);
-        if (cost === undefined) {
-            log.debug(`readCost returned undefined for session=${sessionId.slice(0, 12)}`);
-            return this.snapshotFor(sessionId);
-        }
-        const prior = this.sessions.get(sessionId) ?? { cost: 0, baseline: 0, currency: undefined };
-        const next = {
-            cost: Math.max(prior.cost, cost.amount),
-            baseline: prior.baseline,
-            currency: cost.currency ?? prior.currency,
-        };
-        this.sessions.set(sessionId, next);
-        this.persist();
-        return this.snapshotFor(sessionId);
-    }
-    // Public snapshot accessor for the synthetic events (session_opened,
-    // prompt_request) that don't carry a cost in their own envelope.
-    snapshotFor(sessionId) {
-        const per = this.sessions.get(sessionId);
-        const total = this.totalCost();
-        return {
-            total,
-            perSession: per ? Math.max(0, per.cost - per.baseline) : 0,
-            currency: per?.currency ?? this.opts.currency,
-            soft: this.opts.softLimit,
-            hard: this.opts.hardLimit,
-            state: deriveState(total, this.opts.softLimit, this.opts.hardLimit),
-        };
-    }
-    // Returns the new state if this call transitioned to a higher tier
-    // (ok→soft, ok→hard, soft→hard). Returns undefined if the state didn't
-    // change or if it dropped (e.g. a reset zeroed everything).
-    consumeStateTransition() {
-        const next = deriveState(this.totalCost(), this.opts.softLimit, this.opts.hardLimit);
-        if (rank(next) > rank(this.currentState)) {
-            this.currentState = next;
-            return next;
-        }
-        if (next !== this.currentState) {
-            this.currentState = next;
-        }
-        return undefined;
-    }
-    get state() {
-        return this.currentState;
-    }
-    // Baseline all sessions from their current cost so effective spend
-    // reads zero without discarding the agent's running total.
-    reset() {
-        this.baselineFromCurrent();
-        this.persist();
-    }
-    baselineFromCurrent() {
-        for (const [id, s] of this.sessions) {
-            this.sessions.set(id, { ...s, baseline: s.cost });
-        }
-        this.currentState = "ok";
-    }
-    // Re-read the state file and replace in-memory state if it changed
-    // from what we last wrote. Called from the fs.watch handler in bridge.
-    // Returns true when state was adopted (caller can react if it cares).
-    adoptFromDisk() {
-        if (!this.opts.statePath) {
-            return false;
-        }
-        let raw;
-        try {
-            raw = readFileSync(this.opts.statePath, "utf8");
-        }
-        catch (err) {
-            const e = err;
-            if (e.code === "ENOENT") {
-                // File deleted = reset. Baseline all sessions from current cost
-                // so effective spend reads zero without discarding the agent's
-                // running total (it will keep reporting the same numbers).
-                if (this.sessions.size === 0) {
-                    return false;
-                }
-                log.info("state file removed externally — baselining from current totals");
-                this.baselineFromCurrent();
-                this.lastWrittenJson = "";
-                return true;
-            }
-            log.warn(`read state failed: ${e.message}`);
-            return false;
-        }
-        if (raw === this.lastWrittenJson) {
-            return false;
-        }
-        // Don't advance currentState — leave it at the pre-load value so
-        // consumeStateTransition detects any upward crossing on the next tick.
-        const adopted = this.applyPersisted(raw, false);
-        if (adopted) {
-            this.lastWrittenJson = raw;
-            log.info(`adopted state from disk (total=${this.totalCost().toFixed(2)})`);
-        }
-        return adopted;
-    }
-    loadFromDisk() {
-        if (!this.opts.statePath) {
-            return;
-        }
-        let raw;
-        try {
-            raw = readFileSync(this.opts.statePath, "utf8");
-        }
-        catch (err) {
-            const e = err;
-            if (e.code !== "ENOENT") {
-                log.warn(`read state failed: ${e.message}; starting fresh`);
-            }
-            return;
-        }
-        if (this.applyPersisted(raw)) {
-            this.lastWrittenJson = raw;
-            log.info(`loaded state from ${this.opts.statePath} (total=${this.totalCost().toFixed(2)})`);
-        }
-    }
-    applyPersisted(raw, updateState = true) {
-        let parsed;
-        try {
-            parsed = JSON.parse(raw);
-        }
-        catch (err) {
-            log.warn(`state file malformed: ${err.message}; ignoring`);
-            return false;
-        }
-        if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
-            log.warn("state file is not an object; ignoring");
-            return false;
-        }
-        const obj = parsed;
-        const sessions = obj.sessions ?? {};
-        if (typeof sessions !== "object" || Array.isArray(sessions)) {
-            log.warn("state.sessions is not an object; ignoring");
-            return false;
-        }
-        this.sessions.clear();
-        for (const [id, val] of Object.entries(sessions)) {
-            if (val && typeof val === "object" && typeof val.cost === "number") {
-                const v = val;
-                this.sessions.set(id, {
-                    cost: v.cost,
-                    baseline: typeof v.baseline === "number" ? v.baseline : 0,
-                    currency: typeof v.currency === "string" ? v.currency : undefined,
-                });
-            }
-        }
-        if (updateState) {
-            this.currentState = deriveState(this.totalCost(), this.opts.softLimit, this.opts.hardLimit);
-        }
-        return true;
-    }
-    persist() {
-        if (!this.opts.statePath) {
-            return;
-        }
-        const payload = {
-            version: 1,
-            sessions: Object.fromEntries(this.sessions),
-        };
-        const json = JSON.stringify(payload, null, 2);
-        if (json === this.lastWrittenJson) {
-            return;
-        }
-        try {
-            mkdirSync(dirname(this.opts.statePath), { recursive: true });
-            const tmp = `${this.opts.statePath}.tmp`;
-            writeFileSync(tmp, json, { encoding: "utf8", mode: 0o600 });
-            renameSync(tmp, this.opts.statePath);
-            this.lastWrittenJson = json;
-        }
-        catch (err) {
-            log.warn(`persist failed: ${err.message}`);
-        }
-    }
-    totalCost() {
-        let sum = 0;
-        for (const s of this.sessions.values()) {
-            sum += Math.max(0, s.cost - s.baseline);
-        }
-        return sum;
-    }
-}
-// Delete the persisted state file. Used by the `reset` subcommand when
-// no live process is running (or as the message the live process picks
-// up via fs.watch).
-export function deleteStateFile(statePath) {
-    try {
-        unlinkSync(statePath);
-        return true;
-    }
-    catch (err) {
-        const e = err;
-        if (e.code === "ENOENT") {
-            return false;
-        }
-        throw err;
-    }
-}
-function deriveState(total, soft, hard) {
-    if (total >= hard) {
-        return "hard";
-    }
-    if (total >= soft) {
-        return "soft";
-    }
-    return "ok";
-}
-function rank(state) {
-    if (state === "hard")
-        return 2;
-    if (state === "soft")
-        return 1;
-    return 0;
-}
-function readCost(update) {
-    // hydra injects cumulative cost via _meta.hydra-acp.cumulativeCost on
-    // usage_update for sessions that have lived across multiple agents.
-    // Prefer that when present so resurrects don't undercount.
-    const cumulative = readCumulativeFromMeta(update._meta);
-    const cost = (update.cost ?? undefined);
-    const amount = cumulative !== undefined
-        ? cumulative
-        : typeof cost?.amount === "number"
-            ? cost.amount
-            : undefined;
-    if (amount === undefined) {
-        return undefined;
-    }
-    const currency = typeof cost?.currency === "string" ? cost.currency : undefined;
-    return { amount, currency };
-}
-function readCumulativeFromMeta(meta) {
-    if (!meta || typeof meta !== "object" || Array.isArray(meta)) {
-        return undefined;
-    }
-    const ns = meta["hydra-acp"];
-    if (!ns || typeof ns !== "object" || Array.isArray(ns)) {
-        return undefined;
-    }
-    const v = ns.cumulativeCost;
-    return typeof v === "number" ? v : undefined;
-}
-//# sourceMappingURL=tracker.js.map
+import{mkdirSync as l,readFileSync as d,renameSync as p,unlinkSync as m,writeFileSync as y}from"node:fs";import{dirname as g}from"node:path";import{logger as b}from"./util/log.js";const n=b("tracker");class x{constructor(t){this.opts=t;t.statePath&&this.loadFromDisk()}opts;sessions=new Map;currentState="ok";lastWrittenJson="";applyUsageUpdate(t,e){n.debug(`usage_update session=${t.slice(0,12)} raw=${JSON.stringify(e).slice(0,200)}`);const s=S(e);if(s===void 0)return n.debug(`readCost returned undefined for session=${t.slice(0,12)}`),this.snapshotFor(t);const i=this.sessions.get(t)??{cost:0,baseline:0,currency:void 0},o={cost:Math.max(i.cost,s.amount),baseline:i.baseline,currency:s.currency??i.currency};return this.sessions.set(t,o),this.persist(),this.snapshotFor(t)}snapshotFor(t){const e=this.sessions.get(t),s=this.totalCost();return{total:s,perSession:e?Math.max(0,e.cost-e.baseline):0,currency:e?.currency??this.opts.currency,soft:this.opts.softLimit,hard:this.opts.hardLimit,state:f(s,this.opts.softLimit,this.opts.hardLimit)}}consumeStateTransition(){const t=f(this.totalCost(),this.opts.softLimit,this.opts.hardLimit);if(h(t)>h(this.currentState))return this.currentState=t,t;t!==this.currentState&&(this.currentState=t)}get state(){return this.currentState}reset(){this.baselineFromCurrent(),this.persist()}baselineFromCurrent(){for(const[t,e]of this.sessions)this.sessions.set(t,{...e,baseline:e.cost});this.currentState="ok"}adoptFromDisk(){if(!this.opts.statePath)return!1;let t;try{t=d(this.opts.statePath,"utf8")}catch(s){const i=s;return i.code==="ENOENT"?this.sessions.size===0?!1:(n.info("state file removed externally \u2014 baselining from current totals"),this.baselineFromCurrent(),this.lastWrittenJson="",!0):(n.warn(`read state failed: ${i.message}`),!1)}if(t===this.lastWrittenJson)return!1;const e=this.applyPersisted(t,!1);return e&&(this.lastWrittenJson=t,n.info(`adopted state from disk (total=${this.totalCost().toFixed(2)})`)),e}loadFromDisk(){if(!this.opts.statePath)return;let t;try{t=d(this.opts.statePath,"utf8")}catch(e){const s=e;s.code!=="ENOENT"&&n.warn(`read state failed: ${s.message}; starting fresh`);return}this.applyPersisted(t)&&(this.lastWrittenJson=t,n.info(`loaded state from ${this.opts.statePath} (total=${this.totalCost().toFixed(2)})`))}applyPersisted(t,e=!0){let s;try{s=JSON.parse(t)}catch(c){return n.warn(`state file malformed: ${c.message}; ignoring`),!1}if(!s||typeof s!="object"||Array.isArray(s))return n.warn("state file is not an object; ignoring"),!1;const o=s.sessions??{};if(typeof o!="object"||Array.isArray(o))return n.warn("state.sessions is not an object; ignoring"),!1;this.sessions.clear();for(const[c,u]of Object.entries(o))if(u&&typeof u=="object"&&typeof u.cost=="number"){const a=u;this.sessions.set(c,{cost:a.cost,baseline:typeof a.baseline=="number"?a.baseline:0,currency:typeof a.currency=="string"?a.currency:void 0})}return e&&(this.currentState=f(this.totalCost(),this.opts.softLimit,this.opts.hardLimit)),!0}persist(){if(!this.opts.statePath)return;const t={version:1,sessions:Object.fromEntries(this.sessions)},e=JSON.stringify(t,null,2);if(e!==this.lastWrittenJson)try{l(g(this.opts.statePath),{recursive:!0});const s=`${this.opts.statePath}.tmp`;y(s,e,{encoding:"utf8",mode:384}),p(s,this.opts.statePath),this.lastWrittenJson=e}catch(s){n.warn(`persist failed: ${s.message}`)}}totalCost(){let t=0;for(const e of this.sessions.values())t+=Math.max(0,e.cost-e.baseline);return t}}function E(r){try{return m(r),!0}catch(t){if(t.code==="ENOENT")return!1;throw t}}function f(r,t,e){return r>=e?"hard":r>=t?"soft":"ok"}function h(r){return r==="hard"?2:r==="soft"?1:0}function S(r){const t=k(r._meta),e=r.cost??void 0,s=t!==void 0?t:typeof e?.amount=="number"?e.amount:void 0;if(s===void 0)return;const i=typeof e?.currency=="string"?e.currency:void 0;return{amount:s,currency:i}}function k(r){if(!r||typeof r!="object"||Array.isArray(r))return;const t=r["hydra-acp"];if(!t||typeof t!="object"||Array.isArray(t))return;const e=t.cumulativeCost;return typeof e=="number"?e:void 0}export{x as CostTracker,E as deleteStateFile};

package/dist/util/log.js

@@ -1,46 +1,2 @@
-let debugEnabled = false;
-export function setDebug(on) {
-    debugEnabled = on;
-}
-function emit(level, scope, args) {
-    const ts = new Date().toISOString();
-    const stream = level === "error" || level === "warn" ? process.stderr : process.stdout;
-    stream.write(`[${ts}] ${level} [${scope}] ${formatArgs(args)}\n`);
-}
-function formatArgs(args) {
-    return args
-        .map((a) => {
-        if (typeof a === "string") {
-            return a;
-        }
-        if (a instanceof Error) {
-            return a.stack ?? a.message;
-        }
-        try {
-            return JSON.stringify(a);
-        }
-        catch {
-            return String(a);
-        }
-    })
-        .join(" ");
-}
-export function logger(scope) {
-    return {
-        debug(...args) {
-            if (debugEnabled) {
-                emit("debug", scope, args);
-            }
-        },
-        info(...args) {
-            emit("info", scope, args);
-        },
-        warn(...args) {
-            emit("warn", scope, args);
-        },
-        error(...args) {
-            emit("error", scope, args);
-        },
-    };
-}
-//# sourceMappingURL=log.js.map
+let t=!1;function s(r){t=r}function o(r,n,e){const i=new Date().toISOString();(r==="error"||r==="warn"?process.stderr:process.stdout).write(`[${i}] ${r} [${n}] ${u(e)}
+`)}function u(r){return r.map(n=>{if(typeof n=="string")return n;if(n instanceof Error)return n.stack??n.message;try{return JSON.stringify(n)}catch{return String(n)}}).join(" ")}function d(r){return{debug(...n){t&&o("debug",r,n)},info(...n){o("info",r,n)},warn(...n){o("warn",r,n)},error(...n){o("error",r,n)}}}export{d as logger,s as setDebug};