@urateam/core 0.1.3 → 0.1.8

package/dist/__tests__/pm-audit-retention-step.test.js

@@ -0,0 +1,126 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import { createPmScheduler } from "../pm/scheduler.js";
+import { createDb } from "../db/client.js";
+import { auditEvents } from "../db/schema.js";
+import { installTestProLicense, restoreLicense } from "./helpers/license.js";
+function stubActions() {
+    return {
+        evaluateBudget: vi.fn().mockResolvedValue({
+            scopes: [
+                {
+                    scope: { kind: "global" },
+                    scopeLabel: "global",
+                    limit: 100,
+                    used: 0,
+                    percent: 0,
+                    tier: "ok",
+                },
+            ],
+            worstTier: "ok",
+            promoteBlocked: false,
+            activeCount: 0,
+        }),
+        recoverRetriableRuns: vi.fn().mockResolvedValue({ recovered: [], exhausted: [] }),
+        recoverStuckInProgressIssues: vi.fn().mockResolvedValue([]),
+        triageNewIssues: vi.fn().mockResolvedValue([]),
+        resolveApprovals: vi.fn().mockResolvedValue({ resolved: 0, stillPending: 0 }),
+        promoteReadyIssues: vi.fn().mockResolvedValue([]),
+        deprioritizeStaleIssues: vi.fn().mockResolvedValue([]),
+        cancelAbandonedIssues: vi.fn().mockResolvedValue([]),
+        postDigest: vi.fn().mockResolvedValue(undefined),
+        getActiveFileMaps: vi.fn().mockResolvedValue(new Map()),
+        predictConflict: vi.fn().mockResolvedValue({ overlapRisk: "none", likelyFiles: [], reasoning: "" }),
+    };
+}
+function baseConfig(extra = {}) {
+    return {
+        enabled: true,
+        cronIntervalMs: 1800000,
+        triageBatchSize: 3,
+        maxInFlight: 3,
+        dailyTokenBudget: 100,
+        slackChannelId: "C123",
+        teamIds: ["team-1"],
+        ...extra,
+    };
+}
+describe("pm tick audit retention sweep", () => {
+    beforeEach(async () => {
+        await installTestProLicense("enterprise");
+    });
+    afterEach(async () => {
+        await restoreLicense();
+    });
+    it("deletes audit_events older than retentionDays", async () => {
+        const db = await createDb({ connectionString: ":memory:" });
+        const oldTs = new Date(Date.now() - 400 * 86400000);
+        const freshTs = new Date();
+        await db.insert(auditEvents).values([
+            {
+                id: "old",
+                timestamp: oldTs,
+                eventType: "pm.issue_promoted",
+                actor: "pm-agent",
+                actorType: "pm-agent",
+                payload: "{}",
+            },
+            {
+                id: "fresh",
+                timestamp: freshTs,
+                eventType: "pm.issue_promoted",
+                actor: "pm-agent",
+                actorType: "pm-agent",
+                payload: "{}",
+            },
+        ]);
+        const scheduler = createPmScheduler({
+            config: baseConfig({ auditLog: { retentionDays: 365 } }),
+            db: db,
+            linearApiKey: "",
+            slackBotToken: "",
+            actions: stubActions(),
+        });
+        await scheduler.tick();
+        const rows = await db.select().from(auditEvents);
+        const ids = rows.map((r) => r.id);
+        expect(ids).not.toContain("old");
+        expect(ids).toContain("fresh");
+    });
+    it("uses default retention (365d) when config.auditLog is unset", async () => {
+        const db = await createDb({ connectionString: ":memory:" });
+        await db.insert(auditEvents).values([
+            {
+                id: "ancient",
+                timestamp: new Date(Date.now() - 400 * 86400000),
+                eventType: "pm.issue_promoted",
+                actor: "pm-agent",
+                actorType: "pm-agent",
+                payload: "{}",
+            },
+        ]);
+        const scheduler = createPmScheduler({
+            config: baseConfig(),
+            db: db,
+            linearApiKey: "",
+            slackBotToken: "",
+            actions: stubActions(),
+        });
+        await scheduler.tick();
+        const rows = await db.select().from(auditEvents);
+        expect(rows.find((r) => r.id === "ancient")).toBeUndefined();
+    });
+    it("tick does not throw when retention sweep fails", async () => {
+        const db = await createDb({ connectionString: ":memory:" });
+        // Break the audit_events table so pruneAuditLog throws.
+        await db.run?.("DROP TABLE audit_events");
+        const scheduler = createPmScheduler({
+            config: baseConfig({ auditLog: { retentionDays: 30 } }),
+            db: db,
+            linearApiKey: "",
+            slackBotToken: "",
+            actions: stubActions(),
+        });
+        await expect(scheduler.tick()).resolves.toBeUndefined();
+    });
+});
+//# sourceMappingURL=pm-audit-retention-step.test.js.map

package/dist/__tests__/pm-audit-retention-step.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pm-audit-retention-step.test.js","sourceRoot":"","sources":["../../src/__tests__/pm-audit-retention-step.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAE7E,SAAS,WAAW;IAClB,OAAO;QACL,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACxC,MAAM,EAAE;gBACN;oBACE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE;oBAClC,UAAU,EAAE,QAAQ;oBACpB,KAAK,EAAE,GAAG;oBACV,IAAI,EAAE,CAAC;oBACP,OAAO,EAAE,CAAC;oBACV,IAAI,EAAE,IAAa;iBACpB;aACF;YACD,SAAS,EAAE,IAAa;YACxB,cAAc,EAAE,KAAK;YACrB,WAAW,EAAE,CAAC;SACf,CAAC;QACF,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;QACjF,4BAA4B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC3D,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC9C,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;QAC7E,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACjD,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACtD,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QACpD,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAChD,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,GAAG,EAAE,CAAC;QACvD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;KACpG,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,QAAiC,EAAE;IACrD,OAAO;QACL,OAAO,EAAE,IAAI;QACb,cAAc,EAAE,OAAO;QACvB,eAAe,EAAE,CAAC;QAClB,WAAW,EAAE,CAAC;QACd,gBAAgB,EAAE,GAAG;QACrB,cAAc,EAAE,MAAM;QACtB,OAAO,EAAE,CAAC,QAAQ,CAAC;QACnB,GAAG,KAAK;KACT,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,qBAAqB,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,cAAc,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,EAAE,GAAG,MAAM,QAAQ,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,CAAC,CAAC;QAE5D,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,QAAQ,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;QAC3B,MAAO,EAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC;YAC3C;gBACE,EAAE,EAAE,KAAK;gBACT,SAAS,EAAE,KAAK;gBAChB,SAAS,EAAE,mBAAmB;gBAC9B,KAAK,EAAE,UAAU;gBACjB,SAAS,EAAE,UAAU;gBACrB,OAAO,EAAE,IAAI;aACd;YACD;gBACE,EAAE,EAAE,OAAO;gBACX,SAAS,EAAE,OAAO;gBAClB,SAAS,EAAE,mBAAmB;gBAC9B,KAAK,EAAE,UAAU;gBACjB,SAAS,EAAE,UAAU;gBACrB,OAAO,EAAE,IAAI;aACd;SACF,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,iBAAiB,CAAC;YAClC,MAAM,EAAE,UAAU,CAAC,EAAE,QAAQ,EAAE,EAAE,aAAa,EAAE,GAAG,EAAE,EAAE,CAAC;YACxD,EAAE,EAAE,EAAS;YACb,YAAY,EAAE,EAAE;YAChB,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,WAAW,EAAS;SAC9B,CAAC,CAAC;QAEH,MAAM,SAAS,CAAC,IAAI,EAAE,CAAC;QAEvB,MAAM,IAAI,GAAG,MAAO,EAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,EAAE,GAAG,MAAM,QAAQ,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,CAAC,CAAC;QAE5D,MAAO,EAAU,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC;YAC3C;gBACE,EAAE,EAAE,SAAS;gBACb,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,QAAQ,CAAC;gBAChD,SAAS,EAAE,mBAAmB;gBAC9B,KAAK,EAAE,UAAU;gBACjB,SAAS,EAAE,UAAU;gBACrB,OAAO,EAAE,IAAI;aACd;SACF,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,iBAAiB,CAAC;YAClC,MAAM,EAAE,UAAU,EAAE;YACpB,EAAE,EAAE,EAAS;YACb,YAAY,EAAE,EAAE;YAChB,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,WAAW,EAAS;SAC9B,CAAC,CAAC;QAEH,MAAM,SAAS,CAAC,IAAI,EAAE,CAAC;QAEvB,MAAM,IAAI,GAAG,MAAO,EAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,EAAE,GAAG,MAAM,QAAQ,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,CAAC,CAAC;QAE5D,wDAAwD;QACxD,MAAO,EAAU,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,CAAC;QAEnD,MAAM,SAAS,GAAG,iBAAiB,CAAC;YAClC,MAAM,EAAE,UAAU,CAAC,EAAE,QAAQ,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,EAAE,CAAC;YACvD,EAAE,EAAE,EAAS;YACb,YAAY,EAAE,EAAE;YAChB,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,WAAW,EAAS;SAC9B,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/pm-budget-alerts.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=pm-budget-alerts.test.d.ts.map

package/dist/__tests__/pm-budget-alerts.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pm-budget-alerts.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/pm-budget-alerts.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/pm-budget-alerts.test.js

@@ -0,0 +1,128 @@
+import { describe, it, expect, beforeEach, vi } from "vitest";
+import Database from "better-sqlite3";
+import { drizzle } from "drizzle-orm/better-sqlite3";
+import * as schema from "../db/schema.js";
+import { maybeFireAlerts } from "../pm/budget-alerts.js";
+import { getCreateTablesDDL } from "../db/client.js";
+import { _setSchemaDriver } from "../db/schema.js";
+function makeDb() {
+    _setSchemaDriver("sqlite");
+    const sqlite = new Database(":memory:");
+    sqlite.exec(getCreateTablesDDL("sqlite"));
+    return drizzle(sqlite, { schema });
+}
+function scopeAt(kind, id, percent) {
+    return {
+        scope: kind === "global"
+            ? { kind: "global" }
+            : kind === "team"
+                ? { kind: "team", teamId: id }
+                : { kind: "repo", repoUrl: id },
+        scopeLabel: kind === "global" ? "global" : `${kind} ${id}`,
+        limit: 1_000_000,
+        used: Math.floor(1_000_000 * (percent / 100)),
+        percent,
+        tier: percent >= 100
+            ? "blocked-100"
+            : percent >= 80
+                ? "warn-80"
+                : percent >= 50
+                    ? "warn-50"
+                    : "ok",
+    };
+}
+function evaluationWith(scopes) {
+    const tierRank = { ok: 0, "warn-50": 1, "warn-80": 2, "blocked-100": 3 };
+    let worst = "ok";
+    for (const s of scopes) {
+        if (tierRank[s.tier] > tierRank[worst])
+            worst = s.tier;
+    }
+    return {
+        scopes,
+        worstTier: worst,
+        promoteBlocked: worst === "blocked-100",
+        activeCount: 0,
+    };
+}
+describe("maybeFireAlerts", () => {
+    let db;
+    let postSlack;
+    beforeEach(() => {
+        db = makeDb();
+        postSlack = vi.fn().mockResolvedValue(undefined);
+    });
+    it("skips scopes at tier ok", async () => {
+        const evaluation = evaluationWith([scopeAt("global", "", 10)]);
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        expect(postSlack).not.toHaveBeenCalled();
+    });
+    it("fires a message at 50%", async () => {
+        const evaluation = evaluationWith([scopeAt("global", "", 55)]);
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(1);
+        const [channel, blocks] = postSlack.mock.calls[0];
+        expect(channel).toBe("C_TEST");
+        const json = JSON.stringify(blocks);
+        expect(json).toContain("global");
+        expect(json).toContain("55");
+    });
+    it("fires 50 and 80 when a scope is at 80%", async () => {
+        const evaluation = evaluationWith([scopeAt("team", "team-a", 82)]);
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(2);
+    });
+    it("fires 50, 80, and 100 when a scope is blocked", async () => {
+        const evaluation = evaluationWith([scopeAt("repo", "r", 105)]);
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(3);
+        const joined = postSlack.mock.calls.map((c) => JSON.stringify(c[1])).join("\n");
+        expect(joined).toContain("blocked");
+    });
+    it("dedup: same threshold same day fires exactly once", async () => {
+        const evaluation = evaluationWith([scopeAt("global", "", 60)]);
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(1);
+    });
+    it("dedup: different scopes same threshold fire separately", async () => {
+        const evaluation = evaluationWith([
+            scopeAt("team", "team-a", 60),
+            scopeAt("team", "team-b", 60),
+        ]);
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(2);
+    });
+    it("dedup: threshold escalation from 50 to 80 fires only the new one on second call", async () => {
+        const first = evaluationWith([scopeAt("global", "", 55)]);
+        await maybeFireAlerts(first, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(1);
+        const second = evaluationWith([scopeAt("global", "", 85)]);
+        await maybeFireAlerts(second, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(2);
+    });
+    it("compensating delete: a failed Slack post leaves no dedup row, so next call retries", async () => {
+        const evaluation = evaluationWith([scopeAt("global", "", 55)]);
+        // First call: postSlack throws, row should be rolled back
+        const failingPost = vi
+            .fn()
+            .mockRejectedValueOnce(new Error("slack is down"));
+        await maybeFireAlerts(evaluation, db, failingPost, "C_TEST");
+        expect(failingPost).toHaveBeenCalledTimes(1);
+        // Second call with a working post: should re-fire because the row was rolled back
+        await maybeFireAlerts(evaluation, db, postSlack, "C_TEST");
+        expect(postSlack).toHaveBeenCalledTimes(1);
+    });
+    it("DB insert failure is swallowed and does NOT call Slack", async () => {
+        const evaluation = evaluationWith([scopeAt("global", "", 55)]);
+        // Construct a db whose .insert() throws
+        const brokenDb = {
+            insert: () => {
+                throw new Error("db is down");
+            },
+        };
+        await maybeFireAlerts(evaluation, brokenDb, postSlack, "C_TEST");
+        expect(postSlack).not.toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=pm-budget-alerts.test.js.map

package/dist/__tests__/pm-budget-alerts.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pm-budget-alerts.test.js","sourceRoot":"","sources":["../../src/__tests__/pm-budget-alerts.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,KAAK,MAAM,MAAM,iBAAiB,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,SAAS,MAAM;IACb,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC3B,MAAM,MAAM,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC;IACxC,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC1C,OAAO,OAAO,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AACrC,CAAC;AAED,SAAS,OAAO,CAAC,IAAgC,EAAE,EAAU,EAAE,OAAe;IAC5E,OAAO;QACL,KAAK,EACH,IAAI,KAAK,QAAQ;YACf,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE;YAC7B,CAAC,CAAC,IAAI,KAAK,MAAM;gBACf,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,MAAM,EAAE,EAAE,EAAE;gBACvC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,EAAE,EAAE;QAC9C,UAAU,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,EAAE,EAAE;QAC1D,KAAK,EAAE,SAAS;QAChB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,OAAO,GAAG,GAAG,CAAC,CAAC;QAC7C,OAAO;QACP,IAAI,EACF,OAAO,IAAI,GAAG;YACZ,CAAC,CAAE,aAAuB;YAC1B,CAAC,CAAC,OAAO,IAAI,EAAE;gBACb,CAAC,CAAE,SAAmB;gBACtB,CAAC,CAAC,OAAO,IAAI,EAAE;oBACb,CAAC,CAAE,SAAmB;oBACtB,CAAC,CAAE,IAAc;KAC1B,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,MAAoC;IAC1D,MAAM,QAAQ,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAW,CAAC;IAClF,IAAI,KAAK,GAA0B,IAAI,CAAC;IACxC,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;QACvB,IAAI,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC;YAAE,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC;IACzD,CAAC;IACD,OAAO;QACL,MAAM;QACN,SAAS,EAAE,KAAK;QAChB,cAAc,EAAE,KAAK,KAAK,aAAa;QACvC,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,EAA6B,CAAC;IAClC,IAAI,SAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,GAAG,MAAM,EAAE,CAAC;QACd,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChF,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC/D,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,UAAU,GAAG,cAAc,CAAC;YAChC,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC;YAC7B,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC;SAC9B,CAAC,CAAC;QACH,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,eAAe,CAAC,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACtD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE3C,MAAM,MAAM,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3D,MAAM,eAAe,CAAC,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACvD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QAClG,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAE/D,0DAA0D;QAC1D,MAAM,WAAW,GAAG,EAAE;aACnB,EAAE,EAAoB;aACtB,qBAAqB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;QACrD,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;QAC7D,MAAM,CAAC,WAAW,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE7C,kFAAkF;QAClF,MAAM,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,UAAU,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC;QAE/D,wCAAwC;QACxC,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE,GAAG,EAAE;gBACX,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;YAChC,CAAC;SACsB,CAAC;QAE1B,MAAM,eAAe,CAAC,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACjE,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/pm-budget-refused-event.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=pm-budget-refused-event.test.d.ts.map

package/dist/__tests__/pm-budget-refused-event.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pm-budget-refused-event.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/pm-budget-refused-event.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/pm-budget-refused-event.test.js

@@ -0,0 +1,141 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import { createPmScheduler } from "../pm/scheduler.js";
+import { createDb } from "../db/client.js";
+import { auditEvents } from "../db/schema.js";
+import { installTestProLicense, restoreLicense } from "./helpers/license.js";
+function mixedBlockedEvaluation() {
+    return {
+        scopes: [
+            {
+                scope: { kind: "global" },
+                scopeLabel: "global",
+                limit: 100,
+                used: 100,
+                percent: 100,
+                tier: "blocked-100",
+            },
+            {
+                scope: { kind: "team", teamId: "T1" },
+                scopeLabel: "team T1",
+                limit: 50,
+                used: 60,
+                percent: 120,
+                tier: "blocked-100",
+            },
+            {
+                scope: { kind: "repo", repoUrl: "https://github.com/x/y" },
+                scopeLabel: "repo x",
+                limit: 100,
+                used: 10,
+                percent: 10,
+                tier: "ok",
+            },
+        ],
+        worstTier: "blocked-100",
+        promoteBlocked: true,
+        blockReason: "global at 100%",
+        activeCount: 0,
+    };
+}
+describe("budget.run_refused audit event", () => {
+    beforeEach(async () => {
+        await installTestProLicense("enterprise");
+    });
+    afterEach(async () => {
+        await restoreLicense();
+    });
+    it("writes one event per blocked scope from evaluateBudget", async () => {
+        const db = await createDb({ connectionString: ":memory:" });
+        const mockActions = {
+            evaluateBudget: vi.fn().mockResolvedValue(mixedBlockedEvaluation()),
+            recoverRetriableRuns: vi.fn().mockResolvedValue({ recovered: [], exhausted: [] }),
+            recoverStuckInProgressIssues: vi.fn().mockResolvedValue([]),
+            triageNewIssues: vi.fn().mockResolvedValue([]),
+            resolveApprovals: vi.fn().mockResolvedValue({ resolved: 0, stillPending: 0 }),
+            promoteReadyIssues: vi.fn().mockResolvedValue([]),
+            deprioritizeStaleIssues: vi.fn().mockResolvedValue([]),
+            cancelAbandonedIssues: vi.fn().mockResolvedValue([]),
+            postDigest: vi.fn().mockResolvedValue(undefined),
+            getActiveFileMaps: vi.fn().mockResolvedValue(new Map()),
+            predictConflict: vi.fn().mockResolvedValue({ overlapRisk: "none", likelyFiles: [], reasoning: "" }),
+        };
+        const scheduler = createPmScheduler({
+            config: {
+                enabled: true,
+                cronIntervalMs: 1800000,
+                triageBatchSize: 3,
+                maxInFlight: 3,
+                dailyTokenBudget: 100,
+                slackChannelId: "C123",
+                teamIds: ["team-1"],
+            },
+            db: db,
+            linearApiKey: "",
+            slackBotToken: "",
+            actions: mockActions,
+        });
+        await scheduler.tick();
+        const rows = await db.select().from(auditEvents);
+        const refused = rows.filter((r) => r.eventType === "budget.run_refused");
+        expect(refused).toHaveLength(2);
+        const scopes = refused.map((r) => r.scope).sort();
+        expect(scopes).toEqual(["global", "team:T1"]);
+        for (const row of refused) {
+            const payload = JSON.parse(row.payload);
+            expect(payload).toHaveProperty("scopeType");
+            expect(payload).toHaveProperty("tokensUsed");
+            expect(payload).toHaveProperty("limit");
+            expect(payload).toHaveProperty("utilization");
+        }
+    });
+    it("does not write audit events when no scopes are blocked", async () => {
+        const db = await createDb({ connectionString: ":memory:" });
+        const mockActions = {
+            evaluateBudget: vi.fn().mockResolvedValue({
+                scopes: [
+                    {
+                        scope: { kind: "global" },
+                        scopeLabel: "global",
+                        limit: 100,
+                        used: 10,
+                        percent: 10,
+                        tier: "ok",
+                    },
+                ],
+                worstTier: "ok",
+                promoteBlocked: false,
+                activeCount: 0,
+            }),
+            recoverRetriableRuns: vi.fn().mockResolvedValue({ recovered: [], exhausted: [] }),
+            recoverStuckInProgressIssues: vi.fn().mockResolvedValue([]),
+            triageNewIssues: vi.fn().mockResolvedValue([]),
+            resolveApprovals: vi.fn().mockResolvedValue({ resolved: 0, stillPending: 0 }),
+            promoteReadyIssues: vi.fn().mockResolvedValue([]),
+            deprioritizeStaleIssues: vi.fn().mockResolvedValue([]),
+            cancelAbandonedIssues: vi.fn().mockResolvedValue([]),
+            postDigest: vi.fn().mockResolvedValue(undefined),
+            getActiveFileMaps: vi.fn().mockResolvedValue(new Map()),
+            predictConflict: vi.fn().mockResolvedValue({ overlapRisk: "none", likelyFiles: [], reasoning: "" }),
+        };
+        const scheduler = createPmScheduler({
+            config: {
+                enabled: true,
+                cronIntervalMs: 1800000,
+                triageBatchSize: 3,
+                maxInFlight: 3,
+                dailyTokenBudget: 100,
+                slackChannelId: "C123",
+                teamIds: ["team-1"],
+            },
+            db: db,
+            linearApiKey: "",
+            slackBotToken: "",
+            actions: mockActions,
+        });
+        await scheduler.tick();
+        const rows = await db.select().from(auditEvents);
+        const refused = rows.filter((r) => r.eventType === "budget.run_refused");
+        expect(refused).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=pm-budget-refused-event.test.js.map

package/dist/__tests__/pm-budget-refused-event.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pm-budget-refused-event.test.js","sourceRoot":"","sources":["../../src/__tests__/pm-budget-refused-event.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAG7E,SAAS,sBAAsB;IAC7B,OAAO;QACL,MAAM,EAAE;YACN;gBACE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE;gBAClC,UAAU,EAAE,QAAQ;gBACpB,KAAK,EAAE,GAAG;gBACV,IAAI,EAAE,GAAG;gBACT,OAAO,EAAE,GAAG;gBACZ,IAAI,EAAE,aAAsB;aAC7B;YACD;gBACE,KAAK,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,MAAM,EAAE,IAAI,EAAE;gBAC9C,UAAU,EAAE,SAAS;gBACrB,KAAK,EAAE,EAAE;gBACT,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,GAAG;gBACZ,IAAI,EAAE,aAAsB;aAC7B;YACD;gBACE,KAAK,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,wBAAwB,EAAE;gBACnE,UAAU,EAAE,QAAQ;gBACpB,KAAK,EAAE,GAAG;gBACV,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,EAAE;gBACX,IAAI,EAAE,IAAa;aACpB;SACF;QACD,SAAS,EAAE,aAAa;QACxB,cAAc,EAAE,IAAI;QACpB,WAAW,EAAE,gBAAgB;QAC7B,WAAW,EAAE,CAAC;KACf,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC9C,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,qBAAqB,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,cAAc,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,GAAG,MAAM,QAAQ,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,CAAC,CAAC;QAE5D,MAAM,WAAW,GAAG;YAClB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,sBAAsB,EAAE,CAAC;YACnE,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;YACjF,4BAA4B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC3D,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC9C,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;YAC7E,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACjD,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACtD,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACpD,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAChD,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,GAAG,EAAE,CAAC;YACvD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;SACpG,CAAC;QAEF,MAAM,SAAS,GAAG,iBAAiB,CAAC;YAClC,MAAM,EAAE;gBACN,OAAO,EAAE,IAAI;gBACb,cAAc,EAAE,OAAO;gBACvB,eAAe,EAAE,CAAC;gBAClB,WAAW,EAAE,CAAC;gBACd,gBAAgB,EAAE,GAAG;gBACrB,cAAc,EAAE,MAAM;gBACtB,OAAO,EAAE,CAAC,QAAQ,CAAC;aACb;YACR,EAAE,EAAE,EAAS;YACb,YAAY,EAAE,EAAE;YAChB,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,WAAkB;SAC5B,CAAC,CAAC;QAEH,MAAM,SAAS,CAAC,IAAI,EAAE,CAAC;QAEvB,MAAM,IAAI,GAAG,MAAO,EAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,oBAAoB,CAAC,CAAC;QAC9E,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;QAE9C,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAChD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,GAAG,MAAM,QAAQ,CAAC,EAAE,gBAAgB,EAAE,UAAU,EAAE,CAAC,CAAC;QAE5D,MAAM,WAAW,GAAG;YAClB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACxC,MAAM,EAAE;oBACN;wBACE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE;wBAClC,UAAU,EAAE,QAAQ;wBACpB,KAAK,EAAE,GAAG;wBACV,IAAI,EAAE,EAAE;wBACR,OAAO,EAAE,EAAE;wBACX,IAAI,EAAE,IAAa;qBACpB;iBACF;gBACD,SAAS,EAAE,IAAI;gBACf,cAAc,EAAE,KAAK;gBACrB,WAAW,EAAE,CAAC;aACK,CAAC;YACtB,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;YACjF,4BAA4B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC3D,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC9C,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;YAC7E,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACjD,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACtD,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACpD,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAChD,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,GAAG,EAAE,CAAC;YACvD,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;SACpG,CAAC;QAEF,MAAM,SAAS,GAAG,iBAAiB,CAAC;YAClC,MAAM,EAAE;gBACN,OAAO,EAAE,IAAI;gBACb,cAAc,EAAE,OAAO;gBACvB,eAAe,EAAE,CAAC;gBAClB,WAAW,EAAE,CAAC;gBACd,gBAAgB,EAAE,GAAG;gBACrB,cAAc,EAAE,MAAM;gBACtB,OAAO,EAAE,CAAC,QAAQ,CAAC;aACb;YACR,EAAE,EAAE,EAAS;YACb,YAAY,EAAE,EAAE;YAChB,aAAa,EAAE,EAAE;YACjB,OAAO,EAAE,WAAkB;SAC5B,CAAC,CAAC;QAEH,MAAM,SAAS,CAAC,IAAI,EAAE,CAAC;QAEvB,MAAM,IAAI,GAAG,MAAO,EAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,oBAAoB,CAAC,CAAC;QAC9E,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/pm-budget.test.js

@@ -1,65 +1,161 @@
 import { describe, it, expect } from "vitest";
-import { checkBudgetGuards } from "../pm/budget.js";
-/**
- * Mock DB that simulates a Drizzle query builder chain.
- * budget.ts now uses Drizzle select().from().where() instead of raw SQL.
- */
+import { evaluateBudget } from "../pm/budget.js";
 function mockDb(rows) {
-    const mapped = rows.length > 0 ? {
-        totalIn: rows[0].totalIn ?? 0,
-        totalOut: rows[0].totalOut ?? 0,
-        activeCount: rows[0].activeCount ?? 0,
-    } : { totalIn: 0, totalOut: 0, activeCount: 0 };
     const chain = {
         select: () => chain,
         from: () => chain,
-        where: () => Promise.resolve([mapped]),
+        where: () => chain,
+        groupBy: () => Promise.resolve(rows),
     };
     return chain;
 }
-describe("checkBudgetGuards", () => {
-    it("returns promoteBlocked=false when under limits", async () => {
-        const db = mockDb([{ totalIn: 100000, totalOut: 50000, activeCount: 1 }]);
-        const result = await checkBudgetGuards({
-            db,
-            maxInFlight: 3,
-            dailyTokenBudget: 5000000,
-        });
+function baseConfig(overrides = {}) {
+    return {
+        enabled: true,
+        dailyTokenBudget: 5_000_000,
+        slackChannelId: "C_TEST",
+        teamIds: ["team-a"],
+        maxInFlight: 3,
+        cronIntervalMs: 1_800_000,
+        triageBatchSize: 3,
+        stuckIssueRecovery: true,
+        stuckIssueTargetState: "Backlog",
+        stuckIssueMaxPerTick: 5,
+        ...overrides,
+    };
+}
+describe("evaluateBudget", () => {
+    it("returns ok for empty spend with default config", async () => {
+        const db = mockDb([]);
+        const result = await evaluateBudget({ db, config: baseConfig() });
+        expect(result.worstTier).toBe("ok");
         expect(result.promoteBlocked).toBe(false);
+        expect(result.scopes).toHaveLength(1);
+        expect(result.scopes[0].scope.kind).toBe("global");
+        expect(result.scopes[0].used).toBe(0);
+        expect(result.scopes[0].percent).toBe(0);
+        expect(result.activeCount).toBe(0);
+    });
+    it("computes global scope from rows", async () => {
+        const db = mockDb([
+            { linearTeamId: "team-a", repoUrl: "github.com/org/repo", totalTokens: 2_500_000, activeCount: 1 },
+        ]);
+        const result = await evaluateBudget({ db, config: baseConfig() });
+        const global = result.scopes.find((s) => s.scope.kind === "global");
+        expect(global.used).toBe(2_500_000);
+        expect(global.percent).toBe(50);
+        expect(global.tier).toBe("warn-50");
         expect(result.activeCount).toBe(1);
-        expect(result.tokenSpendPercent).toBeLessThan(80);
     });
-    it("blocks when activeCount >= maxInFlight", async () => {
-        const db = mockDb([{ totalIn: 100000, totalOut: 50000, activeCount: 3 }]);
-        const result = await checkBudgetGuards({
+    it("tier transitions: 0/50/80/100", async () => {
+        const cases = [
+            [0, "ok"],
+            [49, "ok"],
+            [50, "warn-50"],
+            [79, "warn-50"],
+            [80, "warn-80"],
+            [99, "warn-80"],
+            [100, "blocked-100"],
+            [150, "blocked-100"],
+        ];
+        for (const [percent, expected] of cases) {
+            const used = (5_000_000 * percent) / 100;
+            const db = mockDb([
+                { linearTeamId: "team-a", repoUrl: "r", totalTokens: used, activeCount: 0 },
+            ]);
+            const result = await evaluateBudget({ db, config: baseConfig() });
+            const global = result.scopes.find((s) => s.scope.kind === "global");
+            expect({ percent, tier: global.tier }).toEqual({ percent, tier: expected });
+        }
+    });
+    it("per-team scope uses perTeam override", async () => {
+        const db = mockDb([
+            { linearTeamId: "team-a", repoUrl: "r", totalTokens: 1_600_000, activeCount: 0 },
+        ]);
+        const result = await evaluateBudget({
             db,
-            maxInFlight: 3,
-            dailyTokenBudget: 5000000,
+            config: baseConfig({
+                budgets: { perTeam: { "team-a": 2_000_000 } },
+            }),
         });
-        expect(result.promoteBlocked).toBe(true);
-        expect(result.reason).toContain("maxInFlight");
+        const teamScope = result.scopes.find((s) => s.scope.kind === "team" && s.scope.teamId === "team-a");
+        expect(teamScope).toBeDefined();
+        expect(teamScope.limit).toBe(2_000_000);
+        expect(teamScope.percent).toBe(80);
+        expect(teamScope.tier).toBe("warn-80");
+    });
+    it("per-team scope falls back to budgets.default when team not in perTeam", async () => {
+        const db = mockDb([
+            { linearTeamId: "team-z", repoUrl: "r", totalTokens: 500_000, activeCount: 0 },
+        ]);
+        const result = await evaluateBudget({
+            db,
+            config: baseConfig({
+                budgets: {
+                    default: 1_000_000,
+                    perTeam: { "team-a": 500_000 },
+                },
+            }),
+        });
+        const teamScope = result.scopes.find((s) => s.scope.kind === "team" && s.scope.teamId === "team-z");
+        expect(teamScope.limit).toBe(1_000_000);
+        expect(teamScope.percent).toBe(50);
+    });
+    it("per-repo scope uses perRepo override", async () => {
+        const db = mockDb([
+            { linearTeamId: "team-a", repoUrl: "github.com/org/secret", totalTokens: 900_000, activeCount: 0 },
+        ]);
+        const result = await evaluateBudget({
+            db,
+            config: baseConfig({
+                budgets: { perRepo: { "github.com/org/secret": 1_000_000 } },
+            }),
+        });
+        const repoScope = result.scopes.find((s) => s.scope.kind === "repo" && s.scope.repoUrl === "github.com/org/secret");
+        expect(repoScope.limit).toBe(1_000_000);
+        expect(repoScope.percent).toBe(90);
+        expect(repoScope.tier).toBe("warn-80");
     });
-    it("blocks when token spend >= 80%", async () => {
-        const db = mockDb([{ totalIn: 3000000, totalOut: 1100000, activeCount: 1 }]);
-        const result = await checkBudgetGuards({
+    it("both per-team and per-repo can apply — worstTier is the max", async () => {
+        const db = mockDb([
+            { linearTeamId: "team-a", repoUrl: "github.com/org/secret", totalTokens: 5_200_000, activeCount: 1 },
+        ]);
+        const result = await evaluateBudget({
             db,
-            maxInFlight: 3,
-            dailyTokenBudget: 5000000,
+            config: baseConfig({
+                dailyTokenBudget: 20_000_000, // keep global well below 100% so only repo blocks
+                budgets: {
+                    perTeam: { "team-a": 10_000_000 }, // 52% — warn-50
+                    perRepo: { "github.com/org/secret": 5_000_000 }, // 104% — blocked-100
+                },
+            }),
         });
+        expect(result.worstTier).toBe("blocked-100");
         expect(result.promoteBlocked).toBe(true);
-        expect(result.reason).toContain("token budget");
-        expect(result.tokenSpendPercent).toBeGreaterThanOrEqual(80);
+        expect(result.blockReason).toContain("github.com/org/secret");
     });
-    it("handles empty DB rows gracefully", async () => {
-        const db = mockDb([]);
-        const result = await checkBudgetGuards({
+    it("legacy rows with NULL linearTeamId contribute to global only", async () => {
+        const db = mockDb([
+            { linearTeamId: null, repoUrl: "r1", totalTokens: 1_000_000, activeCount: 0 },
+            { linearTeamId: "team-a", repoUrl: "r1", totalTokens: 500_000, activeCount: 0 },
+        ]);
+        const result = await evaluateBudget({
             db,
-            maxInFlight: 3,
-            dailyTokenBudget: 5000000,
+            config: baseConfig({ budgets: { perTeam: { "team-a": 2_000_000 } } }),
         });
-        expect(result.promoteBlocked).toBe(false);
-        expect(result.activeCount).toBe(0);
-        expect(result.dailyTokensUsed).toBe(0);
+        const global = result.scopes.find((s) => s.scope.kind === "global");
+        expect(global.used).toBe(1_500_000);
+        const teamScope = result.scopes.find((s) => s.scope.kind === "team" && s.scope.teamId === "team-a");
+        expect(teamScope.used).toBe(500_000);
+    });
+    it("blocks when any scope is at 100%", async () => {
+        const db = mockDb([
+            { linearTeamId: "team-a", repoUrl: "r", totalTokens: 5_000_000, activeCount: 0 },
+        ]);
+        const result = await evaluateBudget({ db, config: baseConfig() });
+        expect(result.promoteBlocked).toBe(true);
+        expect(result.blockReason).toBeDefined();
+        expect(result.blockReason).toContain("global");
     });
 });
 //# sourceMappingURL=pm-budget.test.js.map