@sema-agent/server 1.185.0

package/dist/plugins/pg-rate-limiter.d.ts

@@ -0,0 +1,49 @@
+import type { Pool, PoolClient } from "pg";
+import type { RateDecision, RateGate } from "../observability/rate-limit.js";
+import { type CounterDegradeHook } from "./write-behind-counter.js";
+export interface CounterTable {
+    table: string;
+    keyCol: string;
+    valCol: string;
+}
+export declare class PgWriteBehindCounter {
+    private readonly pool;
+    private readonly t;
+    private readonly windowMs;
+    private readonly now;
+    private readonly onDegraded?;
+    private readonly queryTimeoutMs;
+    private pendingDelta;
+    private fleetTotal;
+    private timer?;
+    private flushing;
+    private flushStartedAt;
+    private failStreak;
+    constructor(pool: Pool, t: CounterTable, windowMs: number, now?: () => number, onDegraded?: CounterDegradeHook | undefined, queryTimeoutMs?: number);
+    private bump;
+    bucket(): number;
+    private rk;
+    used(key: string): number;
+    addLocal(key: string, n: number): void;
+    retryAfterSec(): number;
+    flush(): Promise<void>;
+    private flushOnce;
+    refresh(): Promise<void>;
+    reap(): Promise<number>;
+    startRefresh(intervalMs?: number): this;
+    stop(): void;
+}
+export declare class PgRateLimiter implements RateGate {
+    private readonly limit;
+    private readonly counter;
+    constructor(pool: Pool, limit: number, windowMs?: number, now?: () => number, onDegraded?: CounterDegradeHook, queryTimeoutMs?: number);
+    check(key: string): RateDecision;
+    sweep(): void;
+    flush(): Promise<void>;
+    refresh(): Promise<void>;
+    startRefresh(intervalMs?: number): this;
+    stop(): void;
+}
+export declare const PG_RATE_LIMITER_SCHEMA: string[];
+export declare function ensureSchema(pool: Pool | PoolClient): Promise<void>;
+//# sourceMappingURL=pg-rate-limiter.d.ts.map

package/dist/plugins/pg-rate-limiter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-rate-limiter.d.ts","sourceRoot":"","sources":["../../src/plugins/pg-rate-limiter.ts"],"names":[],"mappings":"AAoBA,OAAO,KAAK,EAAE,IAAI,EAAE,UAAU,EAAe,MAAM,IAAI,CAAC;AACxD,OAAO,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAC7E,OAAO,EAAoC,KAAK,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAUtG,MAAM,WAAW,YAAY;IAE3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;CAChB;AAID,qBAAa,oBAAoB;IAU7B,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,CAAC;IAClB,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,GAAG;IACpB,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;IAG5B,OAAO,CAAC,QAAQ,CAAC,cAAc;IAhBjC,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,UAAU,CAA6B;IAC/C,OAAO,CAAC,KAAK,CAAC,CAAiC;IAC/C,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,cAAc,CAAK;IAE3B,OAAO,CAAC,UAAU,CAAsD;gBAGrD,IAAI,EAAE,IAAI,EACV,CAAC,EAAE,YAAY,EACf,QAAQ,EAAE,MAAM,EAChB,GAAG,GAAE,MAAM,MAAyB,EACpC,UAAU,CAAC,EAAE,kBAAkB,YAAA,EAG/B,cAAc,GAAE,MAAyC;IAI5E,OAAO,CAAC,IAAI;IAaZ,MAAM,IAAI,MAAM;IAGhB,OAAO,CAAC,EAAE;IAKV,IAAI,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM;IAIzB,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,GAAG,IAAI;IAMtC,aAAa,IAAI,MAAM;IAMjB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;YAoBd,SAAS;IAqCjB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAmBxB,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC;IAU7B,YAAY,CAAC,UAAU,SAAQ,GAAG,IAAI;IAKtC,IAAI,IAAI,IAAI;CAIb;AAED,qBAAa,aAAc,YAAW,QAAQ;IAGpB,OAAO,CAAC,QAAQ,CAAC,KAAK;IAF9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAuB;gBAEnC,IAAI,EAAE,IAAI,EAAmB,KAAK,EAAE,MAAM,EAAE,QAAQ,SAAS,EAAE,GAAG,CAAC,EAAE,MAAM,MAAM,EAAE,UAAU,CAAC,EAAE,kBAAkB,EAAE,cAAc,CAAC,EAAE,MAAM;IAIvJ,KAAK,CAAC,GAAG,EAAE,MAAM,GAAG,YAAY;IAUhC,KAAK,IAAI,IAAI;IAKb,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAGtB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAIxB,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI;IAIvC,IAAI,IAAI,IAAI;CAGb;AAOD,eAAO,MAAM,sBAAsB,EAAE,MAAM,EAS1C,CAAC;AAGF,wBAAsB,YAAY,CAAC,IAAI,EAAE,IAAI,GAAG,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAEzE"}

package/dist/plugins/pg-rate-limiter.js

@@ -0,0 +1,193 @@
+import { DEFAULT_COUNTER_QUERY_TIMEOUT_MS } from "./write-behind-counter.js";
+const SEP = "\u0000";
+export class PgWriteBehindCounter {
+    pool;
+    t;
+    windowMs;
+    now;
+    onDegraded;
+    queryTimeoutMs;
+    pendingDelta = new Map();
+    fleetTotal = new Map();
+    timer;
+    flushing = false;
+    flushStartedAt = 0;
+    failStreak = new Map();
+    constructor(pool, t, windowMs, now = () => Date.now(), onDegraded, queryTimeoutMs = DEFAULT_COUNTER_QUERY_TIMEOUT_MS) {
+        this.pool = pool;
+        this.t = t;
+        this.windowMs = windowMs;
+        this.now = now;
+        this.onDegraded = onDegraded;
+        this.queryTimeoutMs = queryTimeoutMs;
+    }
+    bump(kind, failed, error) {
+        if (!this.onDegraded)
+            return;
+        const prev = this.failStreak.get(kind) ?? 0;
+        if (!failed && prev === 0)
+            return;
+        const next = failed ? prev + 1 : 0;
+        this.failStreak.set(kind, next);
+        try {
+            this.onDegraded({ table: this.t.table, kind, streak: next, prevStreak: prev, ...(failed && error !== undefined ? { error: error instanceof Error ? error.message : String(error) } : {}) });
+        }
+        catch {
+        }
+    }
+    bucket() {
+        return Math.floor(this.now() / this.windowMs);
+    }
+    rk(key) {
+        return `${key}${SEP}${this.bucket()}`;
+    }
+    used(key) {
+        const rk = this.rk(key);
+        return (this.fleetTotal.get(rk) ?? 0) + (this.pendingDelta.get(rk) ?? 0);
+    }
+    addLocal(key, n) {
+        if (n <= 0)
+            return;
+        const rk = this.rk(key);
+        this.pendingDelta.set(rk, (this.pendingDelta.get(rk) ?? 0) + n);
+    }
+    retryAfterSec() {
+        return Math.ceil(((this.bucket() + 1) * this.windowMs - this.now()) / 1000);
+    }
+    async flush() {
+        if (this.flushing) {
+            if (this.now() - this.flushStartedAt > 60_000)
+                this.bump("stalled", true, "flush round stalled >60s (hung query?)");
+            return;
+        }
+        this.flushing = true;
+        this.flushStartedAt = this.now();
+        try {
+            await this.flushOnce();
+        }
+        finally {
+            this.flushing = false;
+            this.bump("stalled", false);
+        }
+    }
+    async flushOnce() {
+        const { table, keyCol, valCol } = this.t;
+        let attempted = false;
+        let failure;
+        let hadFailure = false;
+        for (const [rk, delta] of [...this.pendingDelta]) {
+            if (delta <= 0)
+                continue;
+            const sep = rk.lastIndexOf(SEP);
+            const key = rk.slice(0, sep);
+            const windowBucket = Number(rk.slice(sep + 1));
+            attempted = true;
+            try {
+                const q = {
+                    text: `INSERT INTO ${table} (${keyCol}, window_bucket, ${valCol}, updated_at) VALUES ($1,$2,$3,$4) ` +
+                        `ON CONFLICT (${keyCol}, window_bucket) DO UPDATE SET ${valCol} = ${table}.${valCol} + EXCLUDED.${valCol}, updated_at = EXCLUDED.updated_at`,
+                    values: [key, windowBucket, delta, new Date()],
+                    query_timeout: this.queryTimeoutMs,
+                };
+                await this.pool.query(q);
+                const remaining = (this.pendingDelta.get(rk) ?? 0) - delta;
+                if (remaining > 0)
+                    this.pendingDelta.set(rk, remaining);
+                else
+                    this.pendingDelta.delete(rk);
+            }
+            catch (err) {
+                hadFailure = true;
+                failure = err;
+            }
+        }
+        if (attempted)
+            this.bump("flush", hadFailure, failure);
+        await this.refresh();
+        await this.reap();
+    }
+    async refresh() {
+        const { table, keyCol, valCol } = this.t;
+        const b = this.bucket();
+        try {
+            const q = { text: `SELECT ${keyCol}, ${valCol} FROM ${table} WHERE window_bucket = $1`, values: [b], query_timeout: this.queryTimeoutMs };
+            const res = await this.pool.query(q);
+            const next = new Map();
+            for (const row of res.rows) {
+                next.set(`${String(row[keyCol])}${SEP}${b}`, Number(row[valCol]));
+            }
+            this.fleetTotal = next;
+            this.bump("refresh", false);
+        }
+        catch (err) {
+            this.bump("refresh", true, err);
+        }
+    }
+    async reap() {
+        try {
+            const q = { text: `DELETE FROM ${this.t.table} WHERE window_bucket < $1`, values: [this.bucket()], query_timeout: this.queryTimeoutMs };
+            const res = await this.pool.query(q);
+            return res.rowCount ?? 0;
+        }
+        catch {
+            return 0;
+        }
+    }
+    startRefresh(intervalMs = 5_000) {
+        this.timer ??= setInterval(() => void this.flush(), intervalMs);
+        this.timer.unref?.();
+        return this;
+    }
+    stop() {
+        if (this.timer)
+            clearInterval(this.timer);
+        this.timer = undefined;
+    }
+}
+export class PgRateLimiter {
+    limit;
+    counter;
+    constructor(pool, limit, windowMs = 60_000, now, onDegraded, queryTimeoutMs) {
+        this.limit = limit;
+        this.counter = new PgWriteBehindCounter(pool, { table: "rate_limit", keyCol: "limit_key", valCol: "n" }, windowMs, now, onDegraded, queryTimeoutMs);
+    }
+    check(key) {
+        if (this.limit <= 0)
+            return { allowed: true, retryAfterSec: 0, remaining: Number.POSITIVE_INFINITY };
+        const usedBefore = this.counter.used(key);
+        this.counter.addLocal(key, 1);
+        return usedBefore < this.limit
+            ? { allowed: true, retryAfterSec: 0, remaining: this.limit - usedBefore - 1 }
+            : { allowed: false, retryAfterSec: Math.max(1, this.counter.retryAfterSec()), remaining: 0 };
+    }
+    sweep() {
+    }
+    flush() {
+        return this.counter.flush();
+    }
+    refresh() {
+        return this.counter.refresh();
+    }
+    startRefresh(intervalMs) {
+        this.counter.startRefresh(intervalMs);
+        return this;
+    }
+    stop() {
+        this.counter.stop();
+    }
+}
+export const PG_RATE_LIMITER_SCHEMA = [
+    `CREATE TABLE IF NOT EXISTS rate_limit (
+     limit_key     VARCHAR(190)   NOT NULL,
+     window_bucket BIGINT         NOT NULL,
+     n             INTEGER        NOT NULL DEFAULT 0,
+     updated_at    TIMESTAMPTZ(3) NOT NULL,
+     PRIMARY KEY (limit_key, window_bucket)
+   )`,
+    `CREATE INDEX IF NOT EXISTS idx_rate_limit_window ON rate_limit (window_bucket)`,
+];
+export async function ensureSchema(pool) {
+    for (const stmt of PG_RATE_LIMITER_SCHEMA)
+        await pool.query(stmt);
+}
+//# sourceMappingURL=pg-rate-limiter.js.map

package/dist/plugins/pg-rate-limiter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-rate-limiter.js","sourceRoot":"","sources":["../../src/plugins/pg-rate-limiter.ts"],"names":[],"mappings":"AAsBA,OAAO,EAAE,gCAAgC,EAA2B,MAAM,2BAA2B,CAAC;AAQtG,MAAM,GAAG,GAAG,QAAQ,CAAC;AAWrB,MAAM,OAAO,oBAAoB;IAUZ;IACA;IACA;IACA;IACA;IAGA;IAhBX,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,UAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;IACvC,KAAK,CAAkC;IACvC,QAAQ,GAAG,KAAK,CAAC;IACjB,cAAc,GAAG,CAAC,CAAC;IAEnB,UAAU,GAAG,IAAI,GAAG,EAA2C,CAAC;IAExE,YACmB,IAAU,EACV,CAAe,EACf,QAAgB,EAChB,MAAoB,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,EACpC,UAA+B,EAG/B,iBAAyB,gCAAgC;QAPzD,SAAI,GAAJ,IAAI,CAAM;QACV,MAAC,GAAD,CAAC,CAAc;QACf,aAAQ,GAAR,QAAQ,CAAQ;QAChB,QAAG,GAAH,GAAG,CAAiC;QACpC,eAAU,GAAV,UAAU,CAAqB;QAG/B,mBAAc,GAAd,cAAc,CAA2C;IACzE,CAAC;IAGI,IAAI,CAAC,IAAqC,EAAE,MAAe,EAAE,KAAe;QAClF,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,OAAO;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,IAAI,IAAI,KAAK,CAAC;YAAE,OAAO;QAClC,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAChC,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,CAAC,MAAM,IAAI,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9L,CAAC;QAAC,MAAM,CAAC;QAET,CAAC;IACH,CAAC;IAED,MAAM;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IACO,EAAE,CAAC,GAAW;QACpB,OAAO,GAAG,GAAG,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC;IACxC,CAAC;IAGD,IAAI,CAAC,GAAW;QACd,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;QACxB,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3E,CAAC;IACD,QAAQ,CAAC,GAAW,EAAE,CAAS;QAC7B,IAAI,CAAC,IAAI,CAAC;YAAE,OAAO;QACnB,MAAM,EAAE,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAClE,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;IAC9E,CAAC;IAID,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAMlB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,GAAG,MAAM;gBAAE,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,wCAAwC,CAAC,CAAC;YACpH,OAAO;QACT,CAAC;QACD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;QACzB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC;YACtB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;QACzC,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,OAAgB,CAAC;QACrB,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACjD,IAAI,KAAK,IAAI,CAAC;gBAAE,SAAS;YACzB,MAAM,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YAChC,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAC7B,MAAM,YAAY,GAAG,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;YAC/C,SAAS,GAAG,IAAI,CAAC;YACjB,IAAI,CAAC;gBACH,MAAM,CAAC,GAAqB;oBAC1B,IAAI,EACF,eAAe,KAAK,KAAK,MAAM,oBAAoB,MAAM,qCAAqC;wBAC9F,gBAAgB,MAAM,kCAAkC,MAAM,MAAM,KAAK,IAAI,MAAM,eAAe,MAAM,oCAAoC;oBAC9I,MAAM,EAAE,CAAC,GAAG,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,IAAI,EAAE,CAAC;oBAC9C,aAAa,EAAE,IAAI,CAAC,cAAc;iBACnC,CAAC;gBACF,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAEzB,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC;gBAC3D,IAAI,SAAS,GAAG,CAAC;oBAAE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;;oBACnD,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBAGb,UAAU,GAAG,IAAI,CAAC;gBAClB,OAAO,GAAG,GAAG,CAAC;YAChB,CAAC;QACH,CAAC;QACD,IAAI,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACvD,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;IACpB,CAAC;IAGD,KAAK,CAAC,OAAO;QACX,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACxB,IAAI,CAAC;YACH,MAAM,CAAC,GAAqB,EAAE,IAAI,EAAE,UAAU,MAAM,KAAK,MAAM,SAAS,KAAK,2BAA2B,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;YAC5J,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAA0B,CAAC,CAAC,CAAC;YAC9D,MAAM,IAAI,GAAG,IAAI,GAAG,EAAkB,CAAC;YACvC,KAAK,MAAM,GAAG,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC3B,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACpE,CAAC;YACD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YAEb,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,IAAI;QACR,IAAI,CAAC;YACH,MAAM,CAAC,GAAqB,EAAE,IAAI,EAAE,eAAe,IAAI,CAAC,CAAC,CAAC,KAAK,2BAA2B,EAAE,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC;YAC1J,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACrC,OAAO,GAAG,CAAC,QAAQ,IAAI,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAED,YAAY,CAAC,UAAU,GAAG,KAAK;QAC7B,IAAI,CAAC,KAAK,KAAK,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,EAAE,EAAE,UAAU,CAAC,CAAC;QAChE,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI;QACF,IAAI,IAAI,CAAC,KAAK;YAAE,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;IACzB,CAAC;CACF;AAED,MAAM,OAAO,aAAa;IAGiB;IAFxB,OAAO,CAAuB;IAE/C,YAAY,IAAU,EAAmB,KAAa,EAAE,QAAQ,GAAG,MAAM,EAAE,GAAkB,EAAE,UAA+B,EAAE,cAAuB;QAA9G,UAAK,GAAL,KAAK,CAAQ;QACpD,IAAI,CAAC,OAAO,GAAG,IAAI,oBAAoB,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;IACtJ,CAAC;IAED,KAAK,CAAC,GAAW;QACf,IAAI,IAAI,CAAC,KAAK,IAAI,CAAC;YAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,iBAAiB,EAAE,CAAC;QACrG,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC9B,OAAO,UAAU,GAAG,IAAI,CAAC,KAAK;YAC5B,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,KAAK,GAAG,UAAU,GAAG,CAAC,EAAE;YAC7E,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IACjG,CAAC;IAGD,KAAK;IAEL,CAAC;IAGD,KAAK;QACH,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;IACD,OAAO;QACL,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IAChC,CAAC;IAED,YAAY,CAAC,UAAmB;QAC9B,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QACtC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI;QACF,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IACtB,CAAC;CACF;AAOD,MAAM,CAAC,MAAM,sBAAsB,GAAa;IAC9C;;;;;;KAMG;IACH,gFAAgF;CACjF,CAAC;AAGF,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,IAAuB;IACxD,KAAK,MAAM,IAAI,IAAI,sBAAsB;QAAE,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AACpE,CAAC"}

package/dist/plugins/pg-resume-anchor-store.d.ts

@@ -0,0 +1,14 @@
+import type { Pool } from "pg";
+export declare class PgResumeAnchorStore {
+    private readonly pool;
+    constructor(pool: Pool);
+    put(sessionId: string, eventId: string, entryId: string, owner: string | null): Promise<void>;
+    resolve(sessionId: string, eventId: string, owner: string | null): Promise<string | undefined>;
+    listBySession(sessionId: string): Promise<Array<{
+        eventId: string;
+        entryId: string;
+        owner: string | null;
+    }>>;
+    deleteBySession(sessionId: string): Promise<number>;
+}
+//# sourceMappingURL=pg-resume-anchor-store.d.ts.map

package/dist/plugins/pg-resume-anchor-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-resume-anchor-store.d.ts","sourceRoot":"","sources":["../../src/plugins/pg-resume-anchor-store.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAE/B,qBAAa,mBAAmB;IAClB,OAAO,CAAC,QAAQ,CAAC,IAAI;gBAAJ,IAAI,EAAE,IAAI;IAEjC,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ7F,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAU9F,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,CAAC,CAAC;IAU5G,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAO1D"}

package/dist/plugins/pg-resume-anchor-store.js

@@ -0,0 +1,23 @@
+export class PgResumeAnchorStore {
+    pool;
+    constructor(pool) {
+        this.pool = pool;
+    }
+    async put(sessionId, eventId, entryId, owner) {
+        await this.pool.query("INSERT INTO resume_anchor (session_id, event_id, entry_id, owner, created_at) VALUES ($1,$2,$3,$4,$5) " +
+            "ON CONFLICT (session_id, event_id) DO UPDATE SET entry_id = EXCLUDED.entry_id, owner = EXCLUDED.owner", [sessionId, eventId, entryId, owner, new Date()]);
+    }
+    async resolve(sessionId, eventId, owner) {
+        const r = await this.pool.query("SELECT entry_id FROM resume_anchor WHERE session_id = $1 AND event_id = $2 AND owner IS NOT DISTINCT FROM $3", [sessionId, eventId, owner]);
+        return r.rows[0] ? String(r.rows[0].entry_id) : undefined;
+    }
+    async listBySession(sessionId) {
+        const r = await this.pool.query("SELECT event_id, entry_id, owner FROM resume_anchor WHERE session_id = $1", [sessionId]);
+        return r.rows.map((row) => ({ eventId: String(row.event_id), entryId: String(row.entry_id), owner: row.owner == null ? null : String(row.owner) }));
+    }
+    async deleteBySession(sessionId) {
+        const r = await this.pool.query("DELETE FROM resume_anchor WHERE session_id = $1", [sessionId]);
+        return r.rowCount ?? 0;
+    }
+}
+//# sourceMappingURL=pg-resume-anchor-store.js.map

package/dist/plugins/pg-resume-anchor-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-resume-anchor-store.js","sourceRoot":"","sources":["../../src/plugins/pg-resume-anchor-store.ts"],"names":[],"mappings":"AAOA,MAAM,OAAO,mBAAmB;IACD;IAA7B,YAA6B,IAAU;QAAV,SAAI,GAAJ,IAAI,CAAM;IAAG,CAAC;IAE3C,KAAK,CAAC,GAAG,CAAC,SAAiB,EAAE,OAAe,EAAE,OAAe,EAAE,KAAoB;QACjF,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACnB,wGAAwG;YACtG,uGAAuG,EACzG,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,IAAI,EAAE,CAAC,CACjD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB,EAAE,OAAe,EAAE,KAAoB;QACpE,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAC7B,8GAA8G,EAC9G,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,CAC5B,CAAC;QACF,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5D,CAAC;IAID,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAC7B,2EAA2E,EAC3E,CAAC,SAAS,CAAC,CACZ,CAAC;QACF,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;IACtJ,CAAC;IAID,KAAK,CAAC,eAAe,CAAC,SAAiB;QACrC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAC7B,iDAAiD,EACjD,CAAC,SAAS,CAAC,CACZ,CAAC;QACF,OAAO,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC;IACzB,CAAC;CACF"}

package/dist/plugins/pg-run-store.d.ts

@@ -0,0 +1,76 @@
+import type { TaskResult, TaskStatus } from "@sema-agent/core";
+import { type UsageRow } from "../usage-analytics.js";
+import type { Pool } from "pg";
+import type { RunRecord, RunEvent, SessionSummary } from "./tidb-run-store.js";
+export declare class PgRunStore {
+    private readonly pool;
+    constructor(pool: Pool);
+    private tx;
+    createRun(taskId: string, sessionId: string, owner: string | null, instanceId: string, meta?: {
+        jobId?: string | null;
+        source?: string | null;
+        objectivePreview?: string | null;
+    }): Promise<{
+        ok: true;
+    } | {
+        ok: false;
+        activeTaskId: string;
+    }>;
+    requestCancel(taskId: string, owner: string | null): Promise<boolean>;
+    isCancelRequested(taskId: string, owner: string | null): Promise<boolean>;
+    requestPreempt(taskId: string, owner: string | null): Promise<boolean>;
+    isPreemptRequested(taskId: string, owner: string | null): Promise<boolean>;
+    heartbeat(taskId: string, owner: string | null): Promise<void>;
+    appendEvent(taskId: string, seq: number, type: string, data: unknown): Promise<void>;
+    maxSeq(taskId: string): Promise<number>;
+    setTerminal(taskId: string, status: TaskStatus, result: TaskResult | null, error: string | null): Promise<void>;
+    setSuspended(taskId: string): Promise<void>;
+    setNeedsReview(taskId: string): Promise<void>;
+    markResuming(taskId: string): Promise<boolean>;
+    getActiveTaskId(sessionId: string): Promise<string | undefined>;
+    private rowToRun;
+    getRun(taskId: string): Promise<RunRecord | undefined>;
+    listRuns(opts: {
+        status?: string;
+        jobId?: string;
+        source?: string;
+        owner?: string;
+        cursor?: {
+            createdAt: string;
+            taskId: string;
+        };
+        limit: number;
+    }): Promise<RunRecord[]>;
+    listSessions(opts: {
+        owner?: string;
+        cursor?: {
+            lastActivityAt: string;
+            sessionId: string;
+        };
+        limit: number;
+        q?: string;
+    }): Promise<SessionSummary[]>;
+    deleteBySession(sessionId: string, owner: string | null): Promise<{
+        removed: number;
+    } | {
+        active: string;
+    }>;
+    sourceSummary(sinceMs: number): Promise<Array<{
+        source: string | null;
+        status: string;
+        count: number;
+    }>>;
+    usageScan(fromMs: number, toMs: number, opts?: {
+        owner?: string;
+        limit?: number;
+    }): Promise<{
+        rows: UsageRow[];
+        truncated: boolean;
+    }>;
+    retainedFrom(taskId: string): Promise<number>;
+    getEvents(taskId: string, afterSeq: number): Promise<RunEvent[]>;
+    reapStale(olderThanMs: number): Promise<number>;
+    reapSuspended(olderThanMs: number): Promise<number>;
+    failSuspendedWithExpiredCheckpoint(): Promise<number>;
+}
+//# sourceMappingURL=pg-run-store.d.ts.map

package/dist/plugins/pg-run-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-run-store.d.ts","sourceRoot":"","sources":["../../src/plugins/pg-run-store.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC/D,OAAO,EAAuC,KAAK,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAE3F,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,IAAI,CAAC;AAC3C,OAAO,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAkB/E,qBAAa,UAAU;IACT,OAAO,CAAC,QAAQ,CAAC,IAAI;gBAAJ,IAAI,EAAE,IAAI;YAKzB,EAAE;IAoBV,SAAS,CACb,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,KAAK,EAAE,MAAM,GAAG,IAAI,EACpB,UAAU,EAAE,MAAM,EAClB,IAAI,GAAE;QAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;KAAO,GAC7F,OAAO,CAAC;QAAE,EAAE,EAAE,IAAI,CAAA;KAAE,GAAG;QAAE,EAAE,EAAE,KAAK,CAAC;QAAC,YAAY,EAAE,MAAM,CAAA;KAAE,CAAC;IAoCxD,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAMrE,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAMzE,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAMtE,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAM1E,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAI9D,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAUpF,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOvC,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,GAAG,IAAI,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAgB/G,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK3C,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK7C,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAQ9C,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAKrE,OAAO,CAAC,QAAQ;IAkBV,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;IAStD,QAAQ,CAAC,IAAI,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE;YAAE,SAAS,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IA+BzK,YAAY,CAAC,IAAI,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE;YAAE,cAAc,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,CAAC,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IA2CpJ,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG;QAAE,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IAiB3G,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IASxG,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAAC,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC;IAsBrI,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAM7C,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAUhE,SAAS,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAe/C,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAkBnD,kCAAkC,IAAI,OAAO,CAAC,MAAM,CAAC;CAe5D"}