@sema-agent/server 1.185.0

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

@@ -0,0 +1,107 @@
+import type { TaskResult, TaskStatus } from "@sema-agent/core";
+import { type UsageRow } from "../usage-analytics.js";
+import type { Pool } from "mysql2/promise";
+export interface RunRecord {
+    taskId: string;
+    sessionId: string;
+    owner: string | null;
+    status: "running" | TaskStatus;
+    result: TaskResult | null;
+    error: string | null;
+    errorCode: string | null;
+    jobId: string | null;
+    source: string | null;
+    instanceId: string | null;
+    objectivePreview: string | null;
+    createdAt: string;
+    updatedAt: string;
+}
+export interface SessionSummary {
+    sessionId: string;
+    owner: string | null;
+    lastActivityAt: string;
+    firstActivityAt: string;
+    runCount: number;
+    objectivePreview: string | null;
+    lastStatus: string;
+    lastRunId: string | null;
+    title: string | null;
+}
+export interface RunEvent {
+    seq: number;
+    type: string;
+    data: unknown;
+    ts: string;
+}
+export declare class TiDBRunStore {
+    private readonly pool;
+    constructor(pool: Pool);
+    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>;
+    private tx;
+    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=tidb-run-store.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-run-store.d.ts","sourceRoot":"","sources":["../../src/plugins/tidb-run-store.ts"],"names":[],"mappings":"AASA,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,EAAkD,MAAM,gBAAgB,CAAC;AAE3F,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,MAAM,EAAE,SAAS,GAAG,UAAU,CAAC;IAC/B,MAAM,EAAE,UAAU,GAAG,IAAI,CAAC;IAC1B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAErB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IAGzB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAGrB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAItB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAG1B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAKD,MAAM,WAAW,cAAc;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAErB,cAAc,EAAE,MAAM,CAAC;IACvB,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;IAEjB,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,UAAU,EAAE,MAAM,CAAC;IAGnB,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IAGzB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtB;AAED,MAAM,WAAW,QAAQ;IACvB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,OAAO,CAAC;IACd,EAAE,EAAE,MAAM,CAAC;CACZ;AAkBD,qBAAa,YAAY;IACX,OAAO,CAAC,QAAQ,CAAC,IAAI;gBAAJ,IAAI,EAAE,IAAI;IAUjC,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;IAgDxD,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAUrE,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAczE,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAUtE,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC;IAa1E,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAU9D,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAapF,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAO/B,EAAE;IAoBV,WAAW,CACf,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,UAAU,EAClB,MAAM,EAAE,UAAU,GAAG,IAAI,EACzB,KAAK,EAAE,MAAM,GAAG,IAAI,GACnB,OAAO,CAAC,IAAI,CAAC;IAmBV,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB3C,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAY7C,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAkB9C,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;IAWtD,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;IAwCzK,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;IA6DpJ,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;IAgC3G,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;IAWxG,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;IAwBrI,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAO7C,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAmBhE,SAAS,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAmC/C,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IA6BnD,kCAAkC,IAAI,OAAO,CAAC,MAAM,CAAC;CAoB5D"}

package/dist/plugins/tidb-run-store.js

@@ -0,0 +1,321 @@
+import { projectUsageStats, USAGE_SCAN_LIMIT } from "../usage-analytics.js";
+import { escapeLike } from "./tidb-tool-result-store.js";
+function parseJson(v) {
+    if (v == null)
+        return null;
+    if (typeof v !== "string")
+        return v;
+    try {
+        return JSON.parse(v);
+    }
+    catch {
+        return null;
+    }
+}
+function isDupKey(err) {
+    return Boolean(err) && err.code === "ER_DUP_ENTRY";
+}
+function iso(v) {
+    return v instanceof Date ? v.toISOString() : String(v);
+}
+export class TiDBRunStore {
+    pool;
+    constructor(pool) {
+        this.pool = pool;
+    }
+    async createRun(taskId, sessionId, owner, instanceId, meta = {}) {
+        const conn = await this.pool.getConnection();
+        try {
+            await conn.beginTransaction();
+            const now = new Date();
+            try {
+                await conn.query("INSERT INTO task_active (session_id, task_id, since) VALUES (?,?,?)", [
+                    sessionId,
+                    taskId,
+                    now,
+                ]);
+            }
+            catch (err) {
+                if (isDupKey(err)) {
+                    await conn.rollback();
+                    const [active] = await conn.query("SELECT task_id FROM task_active WHERE session_id = ?", [sessionId]);
+                    return { ok: false, activeTaskId: String(active[0]?.task_id ?? "") };
+                }
+                throw err;
+            }
+            await conn.query("INSERT INTO task_run (task_id, session_id, owner, status, instance_id, job_id, source, objective_preview, created_at, updated_at) " +
+                "VALUES (?,?,?,'running',?,?,?,?,?,?)", [taskId, sessionId, owner, instanceId, meta.jobId ?? null, meta.source ?? null, meta.objectivePreview ?? null, now, now]);
+            await conn.commit();
+            return { ok: true };
+        }
+        catch (err) {
+            try {
+                await conn.rollback();
+            }
+            catch {
+            }
+            throw err;
+        }
+        finally {
+            conn.release();
+        }
+    }
+    async requestCancel(taskId, owner) {
+        const [r] = await this.pool.query("UPDATE task_run SET cancel_requested = 1 WHERE task_id = ? AND owner <=> ? AND status = 'running'", [taskId, owner]);
+        return r.affectedRows > 0;
+    }
+    async isCancelRequested(taskId, owner) {
+        const [rows] = await this.pool.query("SELECT cancel_requested FROM task_run WHERE task_id = ? AND owner <=> ?", [taskId, owner]);
+        return Boolean(rows[0]?.cancel_requested);
+    }
+    async requestPreempt(taskId, owner) {
+        const [r] = await this.pool.query("UPDATE task_run SET preempt_requested = 1 WHERE task_id = ? AND owner <=> ? AND status = 'running'", [taskId, owner]);
+        return r.affectedRows > 0;
+    }
+    async isPreemptRequested(taskId, owner) {
+        const [rows] = await this.pool.query("SELECT preempt_requested FROM task_run WHERE task_id = ? AND owner <=> ?", [taskId, owner]);
+        return Boolean(rows[0]?.preempt_requested);
+    }
+    async heartbeat(taskId, owner) {
+        await this.pool.query("UPDATE task_run SET updated_at = ? WHERE task_id = ? AND owner <=> ? AND status = 'running'", [
+            new Date(),
+            taskId,
+            owner,
+        ]);
+    }
+    async appendEvent(taskId, seq, type, data) {
+        await this.pool.query("INSERT INTO task_event (task_id, seq, type, data, ts) VALUES (?,?,?,?,?)", [
+            taskId,
+            seq,
+            type,
+            data == null ? null : JSON.stringify(data),
+            new Date(),
+        ]);
+    }
+    async maxSeq(taskId) {
+        const [rows] = await this.pool.query("SELECT MAX(seq) AS m FROM task_event WHERE task_id = ?", [taskId]);
+        const m = rows[0]?.m;
+        return m == null ? 0 : Number(m);
+    }
+    async tx(fn) {
+        const conn = await this.pool.getConnection();
+        try {
+            await conn.beginTransaction();
+            await fn(conn);
+            await conn.commit();
+        }
+        catch (err) {
+            try {
+                await conn.rollback();
+            }
+            catch {
+            }
+            throw err;
+        }
+        finally {
+            conn.release();
+        }
+    }
+    async setTerminal(taskId, status, result, error) {
+        const errorCode = result?.errorCode ?? null;
+        await this.tx(async (conn) => {
+            await conn.query("UPDATE task_run SET status = ?, result = ?, error = ?, error_code = ?, updated_at = ? WHERE task_id = ?", [status, result == null ? null : JSON.stringify(result), error, errorCode, new Date(), taskId]);
+            await conn.query("DELETE FROM task_active WHERE task_id = ?", [taskId]);
+        });
+    }
+    async setSuspended(taskId) {
+        await this.pool.query("UPDATE task_run SET status = 'suspended', updated_at = ? WHERE task_id = ? AND status IN ('running','suspended')", [new Date(), taskId]);
+    }
+    async setNeedsReview(taskId) {
+        await this.pool.query("UPDATE task_run SET status = 'needs_review', updated_at = ? WHERE task_id = ? AND status IN ('running','needs_review')", [new Date(), taskId]);
+    }
+    async markResuming(taskId) {
+        const [res] = await this.pool.query("UPDATE task_run SET status = 'running', cancel_requested = 0, preempt_requested = 0, updated_at = ? WHERE task_id = ? AND status IN ('suspended','needs_review')", [new Date(), taskId]);
+        return res.affectedRows === 1;
+    }
+    async getActiveTaskId(sessionId) {
+        const [rows] = await this.pool.query("SELECT task_id FROM task_active WHERE session_id = ?", [sessionId]);
+        return rows[0] ? String(rows[0].task_id) : undefined;
+    }
+    rowToRun(r) {
+        return {
+            taskId: String(r.task_id),
+            sessionId: String(r.session_id),
+            owner: r.owner ?? null,
+            status: r.status,
+            result: parseJson(r.result),
+            error: r.error ?? null,
+            errorCode: r.error_code ?? null,
+            jobId: r.job_id ?? null,
+            source: r.source ?? null,
+            instanceId: r.instance_id ?? null,
+            objectivePreview: r.objective_preview ?? null,
+            createdAt: iso(r.created_at),
+            updatedAt: iso(r.updated_at),
+        };
+    }
+    async getRun(taskId) {
+        const [rows] = await this.pool.query("SELECT task_id, session_id, owner, status, result, error, error_code, job_id, source, instance_id, objective_preview, created_at, updated_at FROM task_run WHERE task_id = ?", [taskId]);
+        return rows[0] ? this.rowToRun(rows[0]) : undefined;
+    }
+    async listRuns(opts) {
+        if (opts.cursor && Number.isNaN(Date.parse(opts.cursor.createdAt)))
+            return [];
+        const where = [];
+        const params = [];
+        if (opts.status) {
+            where.push("status = ?");
+            params.push(opts.status);
+        }
+        if (opts.jobId) {
+            where.push("job_id = ?");
+            params.push(opts.jobId);
+        }
+        if (opts.source) {
+            where.push("source = ?");
+            params.push(opts.source);
+        }
+        if (opts.owner) {
+            where.push("owner = ?");
+            params.push(opts.owner);
+        }
+        if (opts.cursor) {
+            where.push("(created_at < ? OR (created_at = ? AND task_id < ?))");
+            params.push(new Date(opts.cursor.createdAt), new Date(opts.cursor.createdAt), opts.cursor.taskId);
+        }
+        const sql = "SELECT task_id, session_id, owner, status, result, error, error_code, job_id, source, objective_preview, created_at, updated_at FROM task_run" +
+            (where.length ? " WHERE " + where.join(" AND ") : "") +
+            " ORDER BY created_at DESC, task_id DESC LIMIT ?";
+        params.push(opts.limit);
+        const [rows] = await this.pool.query(sql, params);
+        return rows.map((r) => this.rowToRun(r));
+    }
+    async listSessions(opts) {
+        const params = [];
+        let innerWhere = "";
+        if (opts.owner) {
+            innerWhere = " WHERE owner = ?";
+            params.push(opts.owner);
+        }
+        const outer = ["rn = 1"];
+        if (opts.q) {
+            outer.push("EXISTS (SELECT 1 FROM task_run tr WHERE tr.session_id = t.session_id AND LOWER(tr.objective_preview) LIKE LOWER(?) ESCAPE '\\\\')");
+            params.push(`%${escapeLike(opts.q)}%`);
+        }
+        if (opts.cursor) {
+            outer.push("(last_activity < ? OR (last_activity = ? AND session_id < ?))");
+            params.push(new Date(opts.cursor.lastActivityAt), new Date(opts.cursor.lastActivityAt), opts.cursor.sessionId);
+        }
+        const sql = "SELECT session_id, owner, task_id, last_activity, first_activity, run_count, objective_preview, status FROM (" +
+            "SELECT session_id, owner, task_id, objective_preview, status," +
+            " ROW_NUMBER() OVER (PARTITION BY session_id ORDER BY created_at DESC, task_id DESC) AS rn," +
+            " MAX(created_at) OVER (PARTITION BY session_id) AS last_activity," +
+            " MIN(created_at) OVER (PARTITION BY session_id) AS first_activity," +
+            " COUNT(*) OVER (PARTITION BY session_id) AS run_count" +
+            " FROM task_run" + innerWhere + ") t WHERE " + outer.join(" AND ") +
+            " ORDER BY last_activity DESC, session_id DESC LIMIT ?";
+        params.push(opts.limit);
+        const [rows] = await this.pool.query(sql, params);
+        return rows.map((r) => ({
+            sessionId: String(r.session_id),
+            owner: r.owner ?? null,
+            lastActivityAt: iso(r.last_activity),
+            firstActivityAt: iso(r.first_activity),
+            runCount: Number(r.run_count),
+            objectivePreview: r.objective_preview ?? null,
+            lastStatus: String(r.status),
+            lastRunId: r.task_id ?? null,
+            title: null,
+        }));
+    }
+    async deleteBySession(sessionId, owner) {
+        let result = { removed: 0 };
+        await this.tx(async (conn) => {
+            const [active] = await conn.query("SELECT task_id FROM task_active WHERE session_id = ? FOR UPDATE", [sessionId]);
+            if (active[0]) {
+                result = { active: String(active[0].task_id) };
+                return;
+            }
+            await conn.query("DELETE te FROM task_event te JOIN task_run tr ON te.task_id = tr.task_id WHERE tr.session_id = ? AND tr.owner <=> ?", [sessionId, owner]);
+            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.session_id = ? AND tr.owner <=> ?", [sessionId, owner]);
+            const [res] = await conn.query("DELETE FROM task_run WHERE session_id = ? AND owner <=> ?", [sessionId, owner]);
+            result = { removed: res.affectedRows };
+        });
+        return result;
+    }
+    async sourceSummary(sinceMs) {
+        const [rows] = await this.pool.query("SELECT source, status, COUNT(*) AS c FROM task_run WHERE created_at >= ? GROUP BY source, status", [new Date(sinceMs)]);
+        return rows.map((r) => ({ source: r.source ?? null, status: String(r.status), count: Number(r.c) }));
+    }
+    async usageScan(fromMs, toMs, opts) {
+        const limit = Math.min(opts?.limit ?? USAGE_SCAN_LIMIT, USAGE_SCAN_LIMIT);
+        const where = ["created_at >= ?", "created_at < ?"];
+        const params = [new Date(fromMs), new Date(toMs)];
+        if (opts?.owner !== undefined) {
+            where.push("owner = ?");
+            params.push(opts.owner);
+        }
+        const [rows] = await this.pool.query(`SELECT owner, created_at, status, result FROM task_run WHERE ${where.join(" AND ")} ORDER BY created_at LIMIT ${limit + 1}`, params);
+        const truncated = rows.length > limit;
+        const page = truncated ? rows.slice(0, limit) : rows;
+        return {
+            truncated,
+            rows: page.map((r) => ({
+                owner: r.owner ?? null,
+                createdAtMs: new Date(r.created_at).getTime(),
+                status: String(r.status),
+                ...(r.result != null ? { stats: projectUsageStats(r.result) } : {}),
+            })),
+        };
+    }
+    async retainedFrom(taskId) {
+        const [rows] = await this.pool.query("SELECT MIN(seq) AS m FROM task_event WHERE task_id = ?", [taskId]);
+        const m = rows[0]?.m;
+        return m == null ? 0 : Number(m);
+    }
+    async getEvents(taskId, afterSeq) {
+        const [rows] = await this.pool.query("SELECT seq, type, data, ts FROM task_event WHERE task_id = ? AND seq > ? ORDER BY seq ASC", [taskId, afterSeq]);
+        return rows.map((r) => ({
+            seq: Number(r.seq),
+            type: String(r.type),
+            data: parseJson(r.data),
+            ts: iso(r.ts),
+        }));
+    }
+    async reapStale(olderThanMs) {
+        const cutoff = new Date(Date.now() - olderThanMs);
+        let reaped = 0;
+        await this.tx(async (conn) => {
+            const [res] = await conn.query("UPDATE task_run SET status = 'failed', error = 'run stalled (instance lost?)', updated_at = ? " +
+                "WHERE status = 'running' AND updated_at < ?", [new Date(), cutoff]);
+            reaped = res.affectedRows;
+            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.status NOT IN ('running','suspended','needs_review')");
+        });
+        return reaped;
+    }
+    async reapSuspended(olderThanMs) {
+        const cutoff = new Date(Date.now() - olderThanMs);
+        let reaped = 0;
+        await this.tx(async (conn) => {
+            const [res] = await conn.query("UPDATE task_run tr SET tr.status = 'failed', tr.error = 'approval expired before decision', tr.error_code = 'approval.expired', tr.updated_at = ? " +
+                "WHERE tr.status = 'suspended' AND tr.updated_at < ? " +
+                "AND NOT EXISTS (SELECT 1 FROM checkpoint c WHERE c.session_id = tr.session_id AND c.status = 'pending')", [new Date(), cutoff]);
+            reaped = res.affectedRows;
+            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.status NOT IN ('running','suspended','needs_review')");
+        });
+        return reaped;
+    }
+    async failSuspendedWithExpiredCheckpoint() {
+        let reaped = 0;
+        await this.tx(async (conn) => {
+            const [res] = await conn.query("UPDATE task_run tr SET tr.status = 'failed', tr.error = 'approval expired before decision', tr.error_code = 'approval.expired', tr.updated_at = ? " +
+                "WHERE tr.status IN ('suspended','needs_review') " +
+                "AND EXISTS (SELECT 1 FROM checkpoint ce WHERE ce.session_id = tr.session_id AND ce.status = 'expired') " +
+                "AND NOT EXISTS (SELECT 1 FROM checkpoint cp WHERE cp.session_id = tr.session_id AND cp.status = 'pending')", [new Date()]);
+            reaped = res.affectedRows;
+            await conn.query("DELETE ta FROM task_active ta JOIN task_run tr ON ta.task_id = tr.task_id WHERE tr.status NOT IN ('running','suspended','needs_review')");
+        });
+        return reaped;
+    }
+}
+//# sourceMappingURL=tidb-run-store.js.map

package/dist/plugins/tidb-run-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-run-store.js","sourceRoot":"","sources":["../../src/plugins/tidb-run-store.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAiB,MAAM,uBAAuB,CAAC;AAC3F,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AAyDzD,SAAS,SAAS,CAAI,CAAU;IAC9B,IAAI,CAAC,IAAI,IAAI;QAAE,OAAO,IAAI,CAAC;IAC3B,IAAI,OAAO,CAAC,KAAK,QAAQ;QAAE,OAAO,CAAM,CAAC;IACzC,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAM,CAAC;IAC5B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AACD,SAAS,QAAQ,CAAC,GAAY;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,IAAK,GAAyB,CAAC,IAAI,KAAK,cAAc,CAAC;AAC5E,CAAC;AACD,SAAS,GAAG,CAAC,CAAU;IACrB,OAAO,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,OAAO,YAAY;IACM;IAA7B,YAA6B,IAAU;QAAV,SAAI,GAAJ,IAAI,CAAM;IAAG,CAAC;IAU3C,KAAK,CAAC,SAAS,CACb,MAAc,EACd,SAAiB,EACjB,KAAoB,EACpB,UAAkB,EAClB,OAA4F,EAAE;QAE9F,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,CAAC,qEAAqE,EAAE;oBACtF,SAAS;oBACT,MAAM;oBACN,GAAG;iBACJ,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;oBAClB,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACtB,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAC/B,sDAAsD,EACtD,CAAC,SAAS,CAAC,CACZ,CAAC;oBACF,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC;gBACvE,CAAC;gBACD,MAAM,GAAG,CAAC;YACZ,CAAC;YACD,MAAM,IAAI,CAAC,KAAK,CACd,oIAAoI;gBAClI,sCAAsC,EACxC,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,IAAI,EAAE,IAAI,CAAC,gBAAgB,IAAI,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,CACzH,CAAC;YACF,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;YACpB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACxB,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IASD,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,KAAoB;QACtD,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAC/B,mGAAmG,EACnG,CAAC,MAAM,EAAE,KAAK,CAAC,CAChB,CAAC;QACF,OAAO,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC;IAC5B,CAAC;IAID,KAAK,CAAC,iBAAiB,CAAC,MAAc,EAAE,KAAoB;QAC1D,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,yEAAyE,EACzE,CAAC,MAAM,EAAE,KAAK,CAAC,CAChB,CAAC;QACF,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC;IAC5C,CAAC;IAQD,KAAK,CAAC,cAAc,CAAC,MAAc,EAAE,KAAoB;QACvD,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAC/B,oGAAoG,EACpG,CAAC,MAAM,EAAE,KAAK,CAAC,CAChB,CAAC;QACF,OAAO,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC;IAC5B,CAAC;IAID,KAAK,CAAC,kBAAkB,CAAC,MAAc,EAAE,KAAoB;QAC3D,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,0EAA0E,EAC1E,CAAC,MAAM,EAAE,KAAK,CAAC,CAChB,CAAC;QACF,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;IAC7C,CAAC;IAOD,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,KAAoB;QAClD,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,6FAA6F,EAAE;YACnH,IAAI,IAAI,EAAE;YACV,MAAM;YACN,KAAK;SACN,CAAC,CAAC;IACL,CAAC;IAID,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,GAAW,EAAE,IAAY,EAAE,IAAa;QACxE,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,0EAA0E,EAAE;YAChG,MAAM;YACN,GAAG;YACH,IAAI;YACJ,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;YAC1C,IAAI,IAAI,EAAE;SACX,CAAC,CAAC;IACL,CAAC;IAKD,KAAK,CAAC,MAAM,CAAC,MAAc;QACzB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wDAAwD,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QACzG,MAAM,CAAC,GAAI,IAAoC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACtD,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;IAGO,KAAK,CAAC,EAAE,CAAC,EAA2C;QAC1D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC9B,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC;YACf,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACxB,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IAID,KAAK,CAAC,WAAW,CACf,MAAc,EACd,MAAkB,EAClB,MAAyB,EACzB,KAAoB;QAIpB,MAAM,SAAS,GAAI,MAAwC,EAAE,SAAS,IAAI,IAAI,CAAC;QAC/E,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,IAAI,CAAC,KAAK,CACd,yGAAyG,EACzG,CAAC,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,MAAM,CAAC,CAC/F,CAAC;YACF,MAAM,IAAI,CAAC,KAAK,CAAC,2CAA2C,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC;IAQD,KAAK,CAAC,YAAY,CAAC,MAAc;QAM/B,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,kHAAkH,EAAE,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IAClK,CAAC;IAcD,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wHAAwH,EAAE,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;IACxK,CAAC;IAUD,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAUjC,kKAAkK,EAClK,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,CAAC,CACrB,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC;IAChC,CAAC;IAGD,KAAK,CAAC,eAAe,CAAC,SAAiB;QACrC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,sDAAsD,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC3H,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACvD,CAAC;IAEO,QAAQ,CAAC,CAAgB;QAC/B,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC;YAC/B,KAAK,EAAG,CAAC,CAAC,KAAuB,IAAI,IAAI;YACzC,MAAM,EAAE,CAAC,CAAC,MAA6B;YACvC,MAAM,EAAE,SAAS,CAAa,CAAC,CAAC,MAAM,CAAC;YACvC,KAAK,EAAG,CAAC,CAAC,KAAuB,IAAI,IAAI;YACzC,SAAS,EAAG,CAAC,CAAC,UAA4B,IAAI,IAAI;YAClD,KAAK,EAAG,CAAC,CAAC,MAAwB,IAAI,IAAI;YAC1C,MAAM,EAAG,CAAC,CAAC,MAAwB,IAAI,IAAI;YAC3C,UAAU,EAAG,CAAC,CAAC,WAA6B,IAAI,IAAI;YACpD,gBAAgB,EAAG,CAAC,CAAC,iBAAmC,IAAI,IAAI;YAChE,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC;YAC5B,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC;SAC7B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc;QACzB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,8KAA8K,EAC9K,CAAC,MAAM,CAAC,CACT,CAAC;QACF,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACtD,CAAC;IAKD,KAAK,CAAC,QAAQ,CAAC,IAAyI;QAItJ,IAAI,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAAE,OAAO,EAAE,CAAC;QAC9E,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,sDAAsD,CAAC,CAAC;YACnE,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACpG,CAAC;QACD,MAAM,GAAG,GACP,+IAA+I;YAC/I,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACrD,iDAAiD,CAAC;QACpD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,GAAG,EAAE,MAAM,CAAC,CAAC;QACnE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC;IAMD,KAAK,CAAC,YAAY,CAAC,IAA2G;QAC5H,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,IAAI,UAAU,GAAG,EAAE,CAAC;QACpB,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,UAAU,GAAG,kBAAkB,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;QACD,MAAM,KAAK,GAAa,CAAC,QAAQ,CAAC,CAAC;QACnC,IAAI,IAAI,CAAC,CAAC,EAAE,CAAC;YAIX,KAAK,CAAC,IAAI,CAAC,mIAAmI,CAAC,CAAC;YAChJ,MAAM,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QACzC,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,+DAA+D,CAAC,CAAC;YAC5E,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACjH,CAAC;QACD,MAAM,GAAG,GACP,+GAA+G;YAC/G,+DAA+D;YAC/D,4FAA4F;YAC5F,mEAAmE;YACnE,oEAAoE;YACpE,uDAAuD;YACvD,gBAAgB,GAAG,UAAU,GAAG,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;YAClE,uDAAuD,CAAC;QAC1D,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,GAAG,EAAE,MAAM,CAAC,CAAC;QACnE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtB,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC;YAC/B,KAAK,EAAG,CAAC,CAAC,KAAuB,IAAI,IAAI;YACzC,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC;YACpC,eAAe,EAAE,GAAG,CAAC,CAAC,CAAC,cAAc,CAAC;YACtC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAC7B,gBAAgB,EAAG,CAAC,CAAC,iBAAmC,IAAI,IAAI;YAChE,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;YAC5B,SAAS,EAAG,CAAC,CAAC,OAAyB,IAAI,IAAI;YAC/C,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC,CAAC;IACN,CAAC;IAoBD,KAAK,CAAC,eAAe,CAAC,SAAiB,EAAE,KAAoB;QAC3D,IAAI,MAAM,GAA6C,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QACtE,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAE3B,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAC/B,iEAAiE,EACjE,CAAC,SAAS,CAAC,CACZ,CAAC;YACF,IAAI,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;gBACd,MAAM,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC/C,OAAO;YACT,CAAC;YAED,MAAM,IAAI,CAAC,KAAK,CACd,qHAAqH,EACrH,CAAC,SAAS,EAAE,KAAK,CAAC,CACnB,CAAC;YACF,MAAM,IAAI,CAAC,KAAK,CACd,sHAAsH,EACtH,CAAC,SAAS,EAAE,KAAK,CAAC,CACnB,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAC5B,2DAA2D,EAC3D,CAAC,SAAS,EAAE,KAAK,CAAC,CACnB,CAAC;YACF,MAAM,GAAG,EAAE,OAAO,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAID,KAAK,CAAC,aAAa,CAAC,OAAe;QACjC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,kGAAkG,EAClG,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CACpB,CAAC;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAG,CAAC,CAAC,MAAwB,IAAI,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1H,CAAC;IAKD,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,IAAY,EAAE,IAAyC;QACrF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,IAAI,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;QAC1E,MAAM,KAAK,GAAG,CAAC,iBAAiB,EAAE,gBAAgB,CAAC,CAAC;QACpD,MAAM,MAAM,GAAc,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7D,IAAI,IAAI,EAAE,KAAK,KAAK,SAAS,EAAE,CAAC;YAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAAC,CAAC;QACpF,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,gEAAgE,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,8BAA8B,KAAK,GAAG,CAAC,EAAE,EAC5H,MAAM,CACP,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACtC,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACrD,OAAO;YACL,SAAS;YACT,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrB,KAAK,EAAG,CAAC,CAAC,KAAuB,IAAI,IAAI;gBACzC,WAAW,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,UAA2B,CAAC,CAAC,OAAO,EAAE;gBAC9D,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;gBACxB,GAAG,CAAC,CAAC,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,iBAAiB,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACpE,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAID,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,wDAAwD,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;QACzG,MAAM,CAAC,GAAI,IAAoC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACtD,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC;IAGD,KAAK,CAAC,SAAS,CAAC,MAAc,EAAE,QAAgB;QAC9C,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,2FAA2F,EAC3F,CAAC,MAAM,EAAE,QAAQ,CAAC,CACnB,CAAC;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtB,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC;YAClB,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,SAAS,CAAU,CAAC,CAAC,IAAI,CAAC;YAChC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;SACd,CAAC,CAAC,CAAC;IACN,CAAC;IAQD,KAAK,CAAC,SAAS,CAAC,WAAmB;QACjC,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,CAAC;QAClD,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAC5B,gGAAgG;gBAC9F,6CAA6C,EAC/C,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,CAAC,CACrB,CAAC;YACF,MAAM,GAAG,GAAG,CAAC,YAAY,CAAC;YAI1B,MAAM,IAAI,CAAC,KAAK,CACd,yIAAyI,CAC1I,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAiBD,KAAK,CAAC,aAAa,CAAC,WAAmB;QACrC,MAAM,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,CAAC;QAClD,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAC5B,oJAAoJ;gBAClJ,sDAAsD;gBACtD,yGAAyG,EAC3G,CAAC,IAAI,IAAI,EAAE,EAAE,MAAM,CAAC,CACrB,CAAC;YACF,MAAM,GAAG,GAAG,CAAC,YAAY,CAAC;YAC1B,MAAM,IAAI,CAAC,KAAK,CACd,yIAAyI,CAC1I,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAaD,KAAK,CAAC,kCAAkC;QACtC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC3B,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAC5B,oJAAoJ;gBAIlJ,kDAAkD;gBAClD,yGAAyG;gBACzG,4GAA4G,EAC9G,CAAC,IAAI,IAAI,EAAE,CAAC,CACb,CAAC;YACF,MAAM,GAAG,GAAG,CAAC,YAAY,CAAC;YAC1B,MAAM,IAAI,CAAC,KAAK,CACd,yIAAyI,CAC1I,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}

package/dist/plugins/tidb-session-policy-store.d.ts

@@ -0,0 +1,12 @@
+import { type SessionPolicyStore, type SessionPermissionRules, type StoredSessionRules, type SessionRulesRecord, type PutRulesOptions } from "@sema-agent/core";
+import type { Pool } from "mysql2/promise";
+export declare class TiDBSessionPolicyStore implements SessionPolicyStore {
+    private readonly pool;
+    constructor(pool: Pool);
+    getRules(sessionId: string, principal?: string): Promise<StoredSessionRules | null>;
+    listBySession(sessionId: string): Promise<SessionRulesRecord[]>;
+    putRules(sessionId: string, principal: string | undefined, rules: SessionPermissionRules, opts?: PutRulesOptions): Promise<StoredSessionRules>;
+    deleteBySession(sessionId: string): Promise<number>;
+    private tx;
+}
+//# sourceMappingURL=tidb-session-policy-store.d.ts.map

package/dist/plugins/tidb-session-policy-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-session-policy-store.d.ts","sourceRoot":"","sources":["../../src/plugins/tidb-session-policy-store.ts"],"names":[],"mappings":"AAyBA,OAAO,EAA+D,KAAK,kBAAkB,EAAE,KAAK,sBAAsB,EAAE,KAAK,kBAAkB,EAAE,KAAK,kBAAkB,EAAE,KAAK,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC7N,OAAO,KAAK,EAAE,IAAI,EAAkD,MAAM,gBAAgB,CAAC;AAO3F,qBAAa,sBAAuB,YAAW,kBAAkB;IACnD,OAAO,CAAC,QAAQ,CAAC,IAAI;gBAAJ,IAAI,EAAE,IAAI;IAGjC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC;IAanF,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAY/D,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,SAAS,EAAE,KAAK,EAAE,sBAAsB,EAAE,IAAI,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAkD9I,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAK3C,EAAE;CAkBjB"}

package/dist/plugins/tidb-session-policy-store.js

@@ -0,0 +1,88 @@
+import { createHash } from "node:crypto";
+import { loosenReasons, SessionPolicyError, normalizeRules, stripRev } from "@sema-agent/core";
+const policyKey = (sessionId, principal) => createHash("sha256").update(JSON.stringify([sessionId, principal ?? null])).digest("hex");
+const parseRules = (v) => (typeof v === "string" ? JSON.parse(v) : v);
+const isDupKey = (e) => e?.code === "ER_DUP_ENTRY" || e?.errno === 1062;
+export class TiDBSessionPolicyStore {
+    pool;
+    constructor(pool) {
+        this.pool = pool;
+    }
+    async getRules(sessionId, principal) {
+        const [rows] = await this.pool.query("SELECT rules, rev FROM session_policy WHERE policy_key = ?", [policyKey(sessionId, principal)]);
+        if (!rows[0])
+            return null;
+        return { ...parseRules(rows[0].rules), rev: Number(rows[0].rev) };
+    }
+    async listBySession(sessionId) {
+        const [rows] = await this.pool.query("SELECT principal, rules, rev FROM session_policy WHERE session_id = ?", [sessionId]);
+        return rows.map((r) => ({
+            principal: r.principal == null ? undefined : String(r.principal),
+            rules: { ...parseRules(r.rules), rev: Number(r.rev) },
+        }));
+    }
+    async putRules(sessionId, principal, rules, opts) {
+        const key = policyKey(sessionId, principal);
+        for (let attempt = 0;; attempt++) {
+            try {
+                return await this.tx(async (conn) => {
+                    const [rows] = await conn.query("SELECT rules, rev FROM session_policy WHERE policy_key = ? FOR UPDATE", [key]);
+                    const prior = rows[0] ? { rules: parseRules(rows[0].rules), rev: Number(rows[0].rev) } : undefined;
+                    const priorRev = prior?.rev ?? 0;
+                    if (opts?.expectedRev !== undefined && priorRev !== opts.expectedRev) {
+                        throw new SessionPolicyError("conflict", `session rules rev mismatch (expected ${opts.expectedRev}, have ${priorRev})`);
+                    }
+                    const clean = normalizeRules(stripRev(rules));
+                    if (prior && opts?.operator !== true) {
+                        const loosens = loosenReasons(prior.rules, clean);
+                        if (loosens.length > 0) {
+                            throw new SessionPolicyError("loosen_forbidden", `a non-operator session-rule write may only tighten; refused: ${loosens.join("; ")}`);
+                        }
+                    }
+                    const nextRev = priorRev + 1;
+                    const json = JSON.stringify(clean);
+                    const now = new Date();
+                    if (prior) {
+                        const [r] = await conn.query("UPDATE session_policy SET principal = ?, rules = ?, rev = ?, updated_at = ? WHERE policy_key = ? AND rev = ?", [principal ?? null, json, nextRev, now, key, priorRev]);
+                        if (r.affectedRows !== 1)
+                            throw new SessionPolicyError("conflict", `session rules rev mismatch (expected ${opts?.expectedRev ?? priorRev}, have ${priorRev})`);
+                    }
+                    else {
+                        await conn.query("INSERT INTO session_policy (policy_key, session_id, principal, rules, rev, created_at, updated_at) VALUES (?,?,?,?,?,?,?)", [key, sessionId, principal ?? null, json, nextRev, now, now]);
+                    }
+                    return structuredClone({ ...clean, rev: nextRev });
+                });
+            }
+            catch (e) {
+                if (isDupKey(e) && attempt < 4)
+                    continue;
+                throw e;
+            }
+        }
+    }
+    async deleteBySession(sessionId) {
+        const [r] = await this.pool.query("DELETE FROM session_policy WHERE session_id = ?", [sessionId]);
+        return r.affectedRows;
+    }
+    async tx(fn) {
+        const conn = await this.pool.getConnection();
+        try {
+            await conn.beginTransaction();
+            const out = await fn(conn);
+            await conn.commit();
+            return out;
+        }
+        catch (err) {
+            try {
+                await conn.rollback();
+            }
+            catch {
+            }
+            throw err;
+        }
+        finally {
+            conn.release();
+        }
+    }
+}
+//# sourceMappingURL=tidb-session-policy-store.js.map

package/dist/plugins/tidb-session-policy-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-session-policy-store.js","sourceRoot":"","sources":["../../src/plugins/tidb-session-policy-store.ts"],"names":[],"mappings":"AAwBA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,aAAa,EAAE,kBAAkB,EAAE,cAAc,EAAE,QAAQ,EAAgI,MAAM,kBAAkB,CAAC;AAG7N,MAAM,SAAS,GAAG,CAAC,SAAiB,EAAE,SAA6B,EAAU,EAAE,CAC7E,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,SAAS,EAAE,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5F,MAAM,UAAU,GAAG,CAAC,CAAU,EAA0B,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAA2B,CAAC;AACjI,MAAM,QAAQ,GAAG,CAAC,CAAU,EAAW,EAAE,CAAE,CAAuB,EAAE,IAAI,KAAK,cAAc,IAAK,CAAwB,EAAE,KAAK,KAAK,IAAI,CAAC;AAEzI,MAAM,OAAO,sBAAsB;IACJ;IAA7B,YAA6B,IAAU;QAAV,SAAI,GAAJ,IAAI,CAAM;IAAG,CAAC;IAG3C,KAAK,CAAC,QAAQ,CAAC,SAAiB,EAAE,SAAkB;QAClD,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,4DAA4D,EAC5D,CAAC,SAAS,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAClC,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1B,OAAO,EAAE,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;IACpE,CAAC;IAMD,KAAK,CAAC,aAAa,CAAC,SAAiB;QACnC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,uEAAuE,EACvE,CAAC,SAAS,CAAC,CACZ,CAAC;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtB,SAAS,EAAE,CAAC,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;YAChE,KAAK,EAAE,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE;SACtD,CAAC,CAAC,CAAC;IACN,CAAC;IAGD,KAAK,CAAC,QAAQ,CAAC,SAAiB,EAAE,SAA6B,EAAE,KAA6B,EAAE,IAAsB;QACpH,MAAM,GAAG,GAAG,SAAS,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC5C,KAAK,IAAI,OAAO,GAAG,CAAC,GAAI,OAAO,EAAE,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,OAAO,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;oBAClC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAkB,uEAAuE,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;oBACjI,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;oBACnG,MAAM,QAAQ,GAAG,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC;oBAEjC,IAAI,IAAI,EAAE,WAAW,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;wBACrE,MAAM,IAAI,kBAAkB,CAAC,UAAU,EAAE,wCAAwC,IAAI,CAAC,WAAW,UAAU,QAAQ,GAAG,CAAC,CAAC;oBAC1H,CAAC;oBAED,MAAM,KAAK,GAAG,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;oBAE9C,IAAI,KAAK,IAAI,IAAI,EAAE,QAAQ,KAAK,IAAI,EAAE,CAAC;wBACrC,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;wBAClD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BACvB,MAAM,IAAI,kBAAkB,CAAC,kBAAkB,EAAE,gEAAgE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBACzI,CAAC;oBACH,CAAC;oBACD,MAAM,OAAO,GAAG,QAAQ,GAAG,CAAC,CAAC;oBAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;oBACnC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;oBACvB,IAAI,KAAK,EAAE,CAAC;wBAGV,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAkB,8GAA8G,EAAE,CAAC,SAAS,IAAI,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC;wBAEtN,IAAI,CAAC,CAAC,YAAY,KAAK,CAAC;4BAAE,MAAM,IAAI,kBAAkB,CAAC,UAAU,EAAE,wCAAwC,IAAI,EAAE,WAAW,IAAI,QAAQ,UAAU,QAAQ,GAAG,CAAC,CAAC;oBACjK,CAAC;yBAAM,CAAC;wBAKN,MAAM,IAAI,CAAC,KAAK,CACd,2HAA2H,EAC3H,CAAC,GAAG,EAAE,SAAS,EAAE,SAAS,IAAI,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,CAAC,CAC7D,CAAC;oBACJ,CAAC;oBACD,OAAO,eAAe,CAAC,EAAE,GAAG,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,QAAQ,CAAC,CAAC,CAAC,IAAI,OAAO,GAAG,CAAC;oBAAE,SAAS;gBACzC,MAAM,CAAC,CAAC;YACV,CAAC;QACH,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,eAAe,CAAC,SAAiB;QACrC,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,iDAAiD,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QACnH,OAAO,CAAC,CAAC,YAAY,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,EAAE,CAAI,EAAwC;QAC1D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC;YAC3B,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;YACpB,OAAO,GAAG,CAAC;QACb,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACxB,CAAC;YAAC,MAAM,CAAC;YAET,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;CACF"}

package/dist/plugins/tidb-session-storage.d.ts

@@ -0,0 +1,15 @@
+import { BaseSessionStorage, type SessionMetadata, type SessionTreeEntry, type SessionWriteOptions } from "@sema-agent/core";
+import type { Pool } from "mysql2/promise";
+export declare class TiDBSessionStorage extends BaseSessionStorage {
+    private readonly pool;
+    private readonly sessionId;
+    private leafSeq;
+    constructor(pool: Pool, sessionId: string, metadata: SessionMetadata, entries: SessionTreeEntry[], leafId: string | null, leafSeq: number, floorEntryId?: string | null);
+    static wake(pool: Pool, sessionId: string): Promise<TiDBSessionStorage | null>;
+    private static computeFloor;
+    getEntry(id: string): Promise<SessionTreeEntry | undefined>;
+    appendEntry(entry: SessionTreeEntry, opts?: SessionWriteOptions): Promise<void>;
+    setLeafId(leafId: string | null, opts?: SessionWriteOptions): Promise<void>;
+    private persist;
+}
+//# sourceMappingURL=tidb-session-storage.d.ts.map

package/dist/plugins/tidb-session-storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-session-storage.d.ts","sourceRoot":"","sources":["../../src/plugins/tidb-session-storage.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,kBAAkB,EAGlB,KAAK,eAAe,EACpB,KAAK,gBAAgB,EACrB,KAAK,mBAAmB,EACzB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,IAAI,EAAkC,MAAM,gBAAgB,CAAC;AAsB3E,qBAAa,kBAAmB,SAAQ,kBAAkB;IAEtD,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,SAAS;IAI1B,OAAO,CAAC,OAAO;gBALE,IAAI,EAAE,IAAI,EACV,SAAS,EAAE,MAAM,EAClC,QAAQ,EAAE,eAAe,EACzB,OAAO,EAAE,gBAAgB,EAAE,EAC3B,MAAM,EAAE,MAAM,GAAG,IAAI,EACb,OAAO,EAAE,MAAM,EACvB,YAAY,GAAE,MAAM,GAAG,IAAW;WAmBvB,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC;mBAqC/D,YAAY;IAwB3B,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,GAAG,SAAS,CAAC;IAY3D,WAAW,CAAC,KAAK,EAAE,gBAAgB,EAAE,IAAI,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAK/E,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,EAAE,IAAI,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;YAYnE,OAAO;CAsDtB"}