@sema-agent/server 1.185.0

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

@@ -0,0 +1,196 @@
+import { createHash } from "node:crypto";
+import { CheckpointError, validatePendingSteer, checkpointVersionOf, winnerFromOutcome, summarizeCheckpoint, MAX_SUPPORTED_CHECKPOINT_VERSION } from "@sema-agent/core";
+import { redactDeep } from "../trace/redact.js";
+const MAX_TOOL_INPUT_CHARS = 8192;
+export const TERMINAL_BACKSTOP_MS = Math.max(60_000, Number(process.env.APPROVAL_TERMINAL_BACKSTOP_MS) || 30 * 86_400_000);
+export const TERMINAL_GRACE_MS = 3_600_000;
+export function tokenFingerprint(token) {
+    return "sha256:" + createHash("sha256").update(token).digest("hex").slice(0, 12);
+}
+export function boundedToolInput(args) {
+    if (args === undefined)
+        return null;
+    const redacted = redactDeep(args);
+    const json = JSON.stringify(redacted);
+    if (json === undefined)
+        return null;
+    if (json.length > MAX_TOOL_INPUT_CHARS)
+        return { truncated: true, bytes: json.length };
+    return redacted;
+}
+const DUP_ENTRY = 1062;
+export class TiDBCheckpointStore {
+    pool;
+    logger;
+    constructor(pool, logger) {
+        this.pool = pool;
+        this.logger = logger;
+    }
+    async put(token, cp) {
+        const pa = cp.pendingAction;
+        const version = checkpointVersionOf(cp);
+        const toolInput = boundedToolInput(pa?.args);
+        try {
+            await this.pool.query("INSERT INTO checkpoint (token, scope, session_id, version, status, tool_name, tool_call_id, tool_input, checkpoint, deadline, created_at, terminal_at, gate_kind, bound_input_hash, risk_descriptor) " +
+                "VALUES (?,?,?,?,'pending',?,?,?,?,?,?,?,?,?,?)", [
+                token,
+                cp.scope,
+                cp.sessionId,
+                version,
+                pa?.toolName ?? null,
+                pa?.toolCallId ?? null,
+                toolInput === null ? null : JSON.stringify(toolInput),
+                JSON.stringify(cp),
+                cp.deadline ?? null,
+                cp.createdAt,
+                Math.max(cp.createdAt + TERMINAL_BACKSTOP_MS, (cp.deadline ?? 0) + TERMINAL_GRACE_MS),
+                cp.gate?.kind ?? null,
+                pa?.boundInputHash ?? null,
+                ((g) => (g?.riskDescriptor ? JSON.stringify(g.riskDescriptor) : null))(cp.gate),
+            ]);
+        }
+        catch (e) {
+            if (e?.errno === DUP_ENTRY) {
+                throw new CheckpointError("checkpoint.already_exists", `checkpoint token already exists`);
+            }
+            throw e;
+        }
+    }
+    async get(token) {
+        const [rows] = await this.pool.query("SELECT version, status, checkpoint, outcome, reopen_reason, rev, pending_steer FROM checkpoint WHERE token = ?", [token]);
+        const r = rows[0];
+        if (!r)
+            return null;
+        const version = Number(r.version);
+        if (version > MAX_SUPPORTED_CHECKPOINT_VERSION) {
+            throw new Error(`checkpoint format version ${version} is newer than this build supports (max ${MAX_SUPPORTED_CHECKPOINT_VERSION}) — upgrade the service to resume it`);
+        }
+        const cp = parseJson(r.checkpoint);
+        cp.status = r.status;
+        const outcomeRaw = parseJson(r.outcome);
+        if (outcomeRaw)
+            cp.resolvedOutcome = winnerFromOutcome(outcomeRaw);
+        cp.reopenReason = r.reopen_reason ?? undefined;
+        cp.rev = r.rev == null ? 0 : Number(r.rev);
+        const steerRaw = parseJson(r.pending_steer);
+        if (steerRaw && cp.state)
+            cp.state.pendingSteer = steerRaw;
+        this.logger?.info?.("checkpoint_get", {
+            tokenFp: tokenFingerprint(token),
+            status: cp.status,
+            version,
+            hasWorkspaceHandle: !!cp.state?.workspaceHandle,
+            snapshotId: cp.state?.workspaceHandle?.snapshotId,
+        });
+        return cp;
+    }
+    async resolve(token, scope, outcome, expect) {
+        const params = [JSON.stringify(outcome), Date.now(), token, scope];
+        if (expect)
+            params.push(expect.rev);
+        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'resolved', outcome = ?, decided_at = ?, rev = rev + 1, reopen_reason = NULL WHERE token = ? AND scope = ? AND status = 'pending'" +
+            (expect ? " AND rev = ?" : ""), params);
+        return res.affectedRows === 1;
+    }
+    async reopen(token, scope, reason) {
+        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'pending', reopen_reason = ?, rev = rev + 1 WHERE token = ? AND scope = ? AND status = 'resolved'", [reason, token, scope]);
+        return res.affectedRows === 1;
+    }
+    async setPendingSteer(token, scope, steer) {
+        const clean = validatePendingSteer(steer);
+        const [res] = await this.pool.query("UPDATE checkpoint SET pending_steer = ? WHERE token = ? AND scope = ? AND status = 'pending'", [JSON.stringify(clean), token, scope]);
+        return res.affectedRows > 0;
+    }
+    async expire(token, scope) {
+        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'expired', decided_at = ? WHERE token = ? AND scope = ? AND status = 'pending'", [Date.now(), token, scope]);
+        return res.affectedRows === 1;
+    }
+    async reap(scope, cutoff) {
+        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'expired', decided_at = ? WHERE scope = ? AND status = 'pending' AND deadline IS NOT NULL AND deadline <= ?", [Date.now(), scope, cutoff]);
+        return res.affectedRows;
+    }
+    async reapExpired(cutoff) {
+        const [res] = await this.pool.query("UPDATE checkpoint SET status = 'expired', decided_at = ? " +
+            "WHERE status = 'pending' AND ((deadline IS NOT NULL AND deadline <= ? AND (gate_kind IS NULL OR gate_kind NOT IN ('human','irreversible_ask') OR COALESCE(tool_name,'') = 'AskUserQuestion')) OR (terminal_at IS NOT NULL AND terminal_at <= ?))", [Date.now(), cutoff, cutoff]);
+        return res.affectedRows;
+    }
+    async listExpiredApprovalGates(cutoff, limit = 100) {
+        const [rows] = await this.pool.query("SELECT session_id, scope FROM checkpoint WHERE status = 'pending' AND gate_kind IN ('human','irreversible_ask') " +
+            "AND COALESCE(tool_name,'') <> 'AskUserQuestion' AND deadline IS NOT NULL AND deadline <= ? ORDER BY deadline ASC LIMIT ?", [cutoff, limit]);
+        return rows.map((r) => ({ sessionId: String(r.session_id), scope: String(r.scope) }));
+    }
+    async listPending(scope) {
+        const base = "SELECT c.session_id, c.scope, c.tool_name, c.tool_call_id, c.tool_input, c.bound_input_hash, c.risk_descriptor, c.created_at, c.deadline, ta.task_id " +
+            "FROM checkpoint c LEFT JOIN task_active ta ON ta.session_id = c.session_id WHERE c.status='pending'";
+        const [rows] = scope
+            ? await this.pool.query(`${base} AND c.scope=? ORDER BY c.created_at ASC`, [scope])
+            : await this.pool.query(`${base} ORDER BY c.created_at ASC`);
+        const out = rows.map((r) => {
+            const toolCallId = r.tool_call_id ?? null;
+            return {
+                sessionId: String(r.session_id),
+                scope: String(r.scope),
+                toolName: r.tool_name ?? null,
+                toolCallId,
+                boundCallId: toolCallId,
+                boundInputHash: r.bound_input_hash ?? null,
+                taskId: r.task_id ?? null,
+                input: r.tool_input ?? null,
+                createdAt: Number(r.created_at),
+                deadline: r.deadline == null ? null : Number(r.deadline),
+                riskDescriptor: parseJson(r.risk_descriptor),
+            };
+        });
+        return out.sort((a, b) => (b.riskDescriptor?.severity ?? 0) - (a.riskDescriptor?.severity ?? 0) || a.createdAt - b.createdAt);
+    }
+    async listByScope(scope) {
+        const [rows] = await this.pool.query("SELECT checkpoint, status FROM checkpoint WHERE status = 'pending' AND scope = ? ORDER BY created_at ASC LIMIT 500", [scope]);
+        const out = [];
+        for (const r of rows) {
+            const cp = parseJson(r.checkpoint);
+            if (!cp)
+                continue;
+            cp.status = r.status;
+            out.push(summarizeCheckpoint(cp));
+        }
+        return out;
+    }
+    async findPendingTokenBySession(sessionId, scope) {
+        const [rows] = scope
+            ? await this.pool.query("SELECT token FROM checkpoint WHERE session_id=? AND scope=? AND status='pending' LIMIT 1", [sessionId, scope])
+            : await this.pool.query("SELECT token FROM checkpoint WHERE session_id=? AND status='pending' LIMIT 1", [sessionId]);
+        return rows[0] ? String(rows[0].token) : null;
+    }
+    async peekPendingScope(sessionId) {
+        const [rows] = await this.pool.query("SELECT scope FROM checkpoint WHERE session_id=? AND status='pending' LIMIT 1", [sessionId]);
+        if (!rows[0])
+            return undefined;
+        return rows[0].scope ?? null;
+    }
+    async putCtx(sessionId, ctx) {
+        await this.pool.query("INSERT INTO checkpoint_ctx (session_id, ctx, updated_at) VALUES (?,?,?) " +
+            "ON DUPLICATE KEY UPDATE ctx = VALUES(ctx), updated_at = VALUES(updated_at)", [sessionId, JSON.stringify(ctx), Date.now()]);
+    }
+    async getCtx(sessionId) {
+        const [rows] = await this.pool.query("SELECT ctx FROM checkpoint_ctx WHERE session_id=?", [sessionId]);
+        return rows[0] ? parseJson(rows[0].ctx) : null;
+    }
+    async reapCtx(cutoff) {
+        const [res] = await this.pool.query("DELETE FROM checkpoint_ctx WHERE updated_at < ? " +
+            "AND session_id NOT IN (SELECT session_id FROM checkpoint WHERE status='pending') " +
+            "AND session_id NOT IN (SELECT session_id FROM task_active)", [cutoff]);
+        return res.affectedRows;
+    }
+    async deleteBySession(sessionId, owner) {
+        const ownerGuard = "EXISTS (SELECT 1 FROM session_meta sm WHERE sm.session_id = ? AND sm.owner <=> ?)";
+        const [res] = await this.pool.query(`DELETE FROM checkpoint WHERE session_id = ? AND ${ownerGuard}`, [sessionId, sessionId, owner]);
+        await this.pool.query(`DELETE FROM checkpoint_ctx WHERE session_id = ? AND ${ownerGuard}`, [sessionId, sessionId, owner]);
+        return res.affectedRows;
+    }
+}
+function parseJson(v) {
+    if (v == null)
+        return null;
+    return (typeof v === "string" ? JSON.parse(v) : v);
+}
+//# sourceMappingURL=tidb-checkpoint-store.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-checkpoint-store.js","sourceRoot":"","sources":["../../src/plugins/tidb-checkpoint-store.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,gCAAgC,EAA4K,MAAM,kBAAkB,CAAC;AAClV,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAIhD,MAAM,oBAAoB,GAAG,IAAI,CAAC;AAWlC,MAAM,CAAC,MAAM,oBAAoB,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,IAAI,EAAE,GAAG,UAAU,CAAC,CAAC;AAW3H,MAAM,CAAC,MAAM,iBAAiB,GAAG,SAAS,CAAC;AAO3C,MAAM,UAAU,gBAAgB,CAAC,KAAa;IAC5C,OAAO,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACnF,CAAC;AAiCD,MAAM,UAAU,gBAAgB,CAAC,IAAa;IAC5C,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACpC,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACtC,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACpC,IAAI,IAAI,CAAC,MAAM,GAAG,oBAAoB;QAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;IACvF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,SAAS,GAAG,IAAI,CAAC;AAEvB,MAAM,OAAO,mBAAmB;IAEX;IACA;IAFnB,YACmB,IAAU,EACV,MAAqD;QADrD,SAAI,GAAJ,IAAI,CAAM;QACV,WAAM,GAAN,MAAM,CAA+C;IACrE,CAAC;IAGJ,KAAK,CAAC,GAAG,CAAC,KAAsB,EAAE,EAAc;QAC9C,MAAM,EAAE,GAAG,EAAE,CAAC,aAAuF,CAAC;QACtG,MAAM,OAAO,GAAG,mBAAmB,CAAC,EAAE,CAAC,CAAC;QAGxC,MAAM,SAAS,GAAG,gBAAgB,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACnB,uMAAuM;gBACrM,gDAAgD,EAClD;gBACE,KAAK;gBACL,EAAE,CAAC,KAAK;gBACR,EAAE,CAAC,SAAS;gBACZ,OAAO;gBACP,EAAE,EAAE,QAAQ,IAAI,IAAI;gBACpB,EAAE,EAAE,UAAU,IAAI,IAAI;gBACtB,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;gBACrD,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBAClB,EAAE,CAAC,QAAQ,IAAI,IAAI;gBACnB,EAAE,CAAC,SAAS;gBACZ,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,GAAG,oBAAoB,EAAE,CAAC,EAAE,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,iBAAiB,CAAC;gBACpF,EAAE,CAAC,IAAsC,EAAE,IAAI,IAAI,IAAI;gBACvD,EAA8C,EAAE,cAAc,IAAI,IAAI;gBACvE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,IAAgD,CAAC;aAC5H,CACF,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAK,CAAoC,EAAE,KAAK,KAAK,SAAS,EAAE,CAAC;gBAE/D,MAAM,IAAI,eAAe,CAAC,2BAA2B,EAAE,iCAAiC,CAAC,CAAC;YAC5F,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAsB;QAC9B,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,gHAAgH,EAChH,CAAC,KAAK,CAAC,CACR,CAAC;QACF,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,IAAI,CAAC,CAAC;YAAE,OAAO,IAAI,CAAC;QAKpB,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAClC,IAAI,OAAO,GAAG,gCAAgC,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,6BAA6B,OAAO,2CAA2C,gCAAgC,sCAAsC,CAAC,CAAC;QACzK,CAAC;QACD,MAAM,EAAE,GAAG,SAAS,CAAa,CAAC,CAAC,UAAU,CAAE,CAAC;QAEhD,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,MAA8B,CAAC;QAM7C,MAAM,UAAU,GAAG,SAAS,CAAgB,CAAC,CAAC,OAAO,CAAC,CAAC;QACvD,IAAI,UAAU;YAAE,EAAE,CAAC,eAAe,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACnE,EAAE,CAAC,YAAY,GAAI,CAAC,CAAC,aAAqC,IAAI,SAAS,CAAC;QACxE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAG3C,MAAM,QAAQ,GAAG,SAAS,CAAqC,CAAC,CAAC,aAA8B,CAAC,CAAC;QACjG,IAAI,QAAQ,IAAI,EAAE,CAAC,KAAK;YAAE,EAAE,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC;QAI3D,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,gBAAgB,EAAE;YACpC,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC;YAChC,MAAM,EAAE,EAAE,CAAC,MAAM;YACjB,OAAO;YACP,kBAAkB,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,EAAE,eAAe;YAC/C,UAAU,EAAE,EAAE,CAAC,KAAK,EAAE,eAAe,EAAE,UAAU;SAClD,CAAC,CAAC;QACH,OAAO,EAAE,CAAC;IACZ,CAAC;IAOD,KAAK,CAAC,OAAO,CAAC,KAAsB,EAAE,KAAa,EAAE,OAAsB,EAAE,MAA2B;QActG,MAAM,MAAM,GAAc,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAC9E,IAAI,MAAM;YAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,kKAAkK;YAChK,CAAC,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,EAChC,MAAM,CACP,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC;IAChC,CAAC;IAkBD,KAAK,CAAC,MAAM,CAAC,KAAsB,EAAE,KAAa,EAAE,MAAoB;QACtE,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,kIAAkI,EAClI,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CACvB,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC;IAChC,CAAC;IAQD,KAAK,CAAC,eAAe,CAAC,KAAsB,EAAE,KAAa,EAAE,KAAyC;QAMpG,MAAM,KAAK,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,8FAA8F,EAC9F,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,CACtC,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,GAAG,CAAC,CAAC;IAC9B,CAAC;IAWD,KAAK,CAAC,MAAM,CAAC,KAAsB,EAAE,KAAa;QAChD,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,+GAA+G,EAC/G,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,CAC3B,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,KAAK,CAAC,CAAC;IAChC,CAAC;IAGD,KAAK,CAAC,IAAI,CAAC,KAAa,EAAE,MAAc;QACtC,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,4IAA4I,EAC5I,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,CAC5B,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,CAAC;IAC1B,CAAC;IASD,KAAK,CAAC,WAAW,CAAC,MAAc;QAU9B,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,2DAA2D;YACzD,kPAAkP,EACpP,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,CAC7B,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,CAAC;IAC1B,CAAC;IAYD,KAAK,CAAC,wBAAwB,CAAC,MAAc,EAAE,KAAK,GAAG,GAAG;QACxD,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,kHAAkH;YAChH,0HAA0H,EAC5H,CAAC,MAAM,EAAE,KAAK,CAAC,CAChB,CAAC;QACF,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;IACxF,CAAC;IAOD,KAAK,CAAC,WAAW,CAAC,KAAc;QAC9B,MAAM,IAAI,GACR,uJAAuJ;YACvJ,qGAAqG,CAAC;QACxG,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK;YAClB,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,GAAG,IAAI,0CAA0C,EAAE,CAAC,KAAK,CAAC,CAAC;YACpG,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,GAAG,IAAI,4BAA4B,CAAC,CAAC;QAChF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACzB,MAAM,UAAU,GAAI,CAAC,CAAC,YAA8B,IAAI,IAAI,CAAC;YAC7D,OAAO;gBACL,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC/B,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC;gBACtB,QAAQ,EAAG,CAAC,CAAC,SAA2B,IAAI,IAAI;gBAChD,UAAU;gBAGV,WAAW,EAAE,UAAU;gBACvB,cAAc,EAAG,CAAC,CAAC,gBAAkC,IAAI,IAAI;gBAC7D,MAAM,EAAG,CAAC,CAAC,OAAyB,IAAI,IAAI;gBAE5C,KAAK,EAAE,CAAC,CAAC,UAAU,IAAI,IAAI;gBAC3B,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC;gBAC/B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC;gBACxD,cAAc,EAAE,SAAS,CAAiB,CAAC,CAAC,eAAgC,CAAC;aAC9E,CAAC;QACJ,CAAC,CAAC,CAAC;QAGH,OAAO,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,cAAc,EAAE,QAAQ,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IAChI,CAAC;IAWD,KAAK,CAAC,WAAW,CAAC,KAAa;QAG7B,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,oHAAoH,EACpH,CAAC,KAAK,CAAC,CACR,CAAC;QACF,MAAM,GAAG,GAAwB,EAAE,CAAC;QACpC,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YAGrB,MAAM,EAAE,GAAG,SAAS,CAAa,CAAC,CAAC,UAAU,CAAC,CAAC;YAC/C,IAAI,CAAC,EAAE;gBAAE,SAAS;YAClB,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,MAA8B,CAAC;YAC7C,GAAG,CAAC,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAOD,KAAK,CAAC,yBAAyB,CAAC,SAAiB,EAAE,KAAc;QAC/D,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK;YAClB,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACnB,0FAA0F,EAC1F,CAAC,SAAS,EAAE,KAAK,CAAC,CACnB;YACH,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACnB,8EAA8E,EAC9E,CAAC,SAAS,CAAC,CACZ,CAAC;QACN,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAqB,CAAC,CAAC,CAAC,IAAI,CAAC;IACrE,CAAC;IAQD,KAAK,CAAC,gBAAgB,CAAC,SAAiB;QACtC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAClC,8EAA8E,EAC9E,CAAC,SAAS,CAAC,CACZ,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAAE,OAAO,SAAS,CAAC;QAC/B,OAAQ,IAAI,CAAC,CAAC,CAAC,CAAC,KAAuB,IAAI,IAAI,CAAC;IAClD,CAAC;IAKD,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,GAAY;QAC1C,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACnB,0EAA0E;YACxE,4EAA4E,EAC9E,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAC7C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAc,SAAiB;QACzC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,mDAAmD,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QACxH,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAI,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACpD,CAAC;IAaD,KAAK,CAAC,OAAO,CAAC,MAAc;QAC1B,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CACjC,kDAAkD;YAChD,mFAAmF;YACnF,4DAA4D,EAC9D,CAAC,MAAM,CAAC,CACT,CAAC;QACF,OAAO,GAAG,CAAC,YAAY,CAAC;IAC1B,CAAC;IAcD,KAAK,CAAC,eAAe,CAAC,SAAiB,EAAE,KAAoB;QAC3D,MAAM,UAAU,GAAG,mFAAmF,CAAC;QACvG,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,mDAAmD,UAAU,EAAE,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;QACrJ,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,uDAAuD,UAAU,EAAE,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;QAC1H,OAAO,GAAG,CAAC,YAAY,CAAC;IAC1B,CAAC;CACF;AAED,SAAS,SAAS,CAAI,CAAU;IAC9B,IAAI,CAAC,IAAI,IAAI;QAAE,OAAO,IAAI,CAAC;IAC3B,OAAO,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAM,CAAC;AAC1D,CAAC"}

package/dist/plugins/tidb-cost-quota.d.ts

@@ -0,0 +1,16 @@
+import type { Pool } from "mysql2/promise";
+import type { QuotaDecision, QuotaTracker } from "../observability/cost-quota.js";
+import { type CounterDegradeHook } from "./write-behind-counter.js";
+export declare class TiDBCostQuota implements QuotaTracker {
+    private readonly limitMicroUsd;
+    private readonly counter;
+    constructor(pool: Pool, limitMicroUsd: number, windowMs: number, now?: () => number, onDegraded?: CounterDegradeHook, queryTimeoutMs?: number);
+    check(scopeKey: string): QuotaDecision;
+    add(scopeKey: string, micro: number): void;
+    flush(): Promise<void>;
+    refresh(): Promise<void>;
+    reap(): Promise<number>;
+    startRefresh(intervalMs?: number): this;
+    stop(): void;
+}
+//# sourceMappingURL=tidb-cost-quota.d.ts.map

package/dist/plugins/tidb-cost-quota.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-cost-quota.d.ts","sourceRoot":"","sources":["../../src/plugins/tidb-cost-quota.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,gCAAgC,CAAC;AAClF,OAAO,EAAsB,KAAK,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAExF,qBAAa,aAAc,YAAW,YAAY;IAGxB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAFtD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAqB;gBAEjC,IAAI,EAAE,IAAI,EAAmB,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,CAAC,EAAE,MAAM,MAAM,EAAE,UAAU,CAAC,EAAE,kBAAkB,EAAE,cAAc,CAAC,EAAE,MAAM;IAI9J,KAAK,CAAC,QAAQ,EAAE,MAAM,GAAG,aAAa;IAKtC,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAK1C,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAGtB,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAGxB,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC;IAGvB,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI;IAIvC,IAAI,IAAI,IAAI;CAGb"}

package/dist/plugins/tidb-cost-quota.js

@@ -0,0 +1,34 @@
+import { WriteBehindCounter } from "./write-behind-counter.js";
+export class TiDBCostQuota {
+    limitMicroUsd;
+    counter;
+    constructor(pool, limitMicroUsd, windowMs, now, onDegraded, queryTimeoutMs) {
+        this.limitMicroUsd = limitMicroUsd;
+        this.counter = new WriteBehindCounter(pool, { table: "cost_quota", keyCol: "scope_key", valCol: "micro" }, windowMs, now, onDegraded, queryTimeoutMs);
+    }
+    check(scopeKey) {
+        const used = this.counter.used(scopeKey);
+        const allowed = used < this.limitMicroUsd;
+        return { allowed, usedMicroUsd: used, limitMicroUsd: this.limitMicroUsd, retryAfterSec: allowed ? 0 : this.counter.retryAfterSec() };
+    }
+    add(scopeKey, micro) {
+        this.counter.addLocal(scopeKey, micro);
+    }
+    flush() {
+        return this.counter.flush();
+    }
+    refresh() {
+        return this.counter.refresh();
+    }
+    reap() {
+        return this.counter.reap();
+    }
+    startRefresh(intervalMs) {
+        this.counter.startRefresh(intervalMs);
+        return this;
+    }
+    stop() {
+        this.counter.stop();
+    }
+}
+//# sourceMappingURL=tidb-cost-quota.js.map

package/dist/plugins/tidb-cost-quota.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-cost-quota.js","sourceRoot":"","sources":["../../src/plugins/tidb-cost-quota.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,kBAAkB,EAA2B,MAAM,2BAA2B,CAAC;AAExF,MAAM,OAAO,aAAa;IAGiB;IAFxB,OAAO,CAAqB;IAE7C,YAAY,IAAU,EAAmB,aAAqB,EAAE,QAAgB,EAAE,GAAkB,EAAE,UAA+B,EAAE,cAAuB;QAArH,kBAAa,GAAb,aAAa,CAAQ;QAC5D,IAAI,CAAC,OAAO,GAAG,IAAI,kBAAkB,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;IACxJ,CAAC;IAED,KAAK,CAAC,QAAgB;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC;QAC1C,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;IACvI,CAAC;IACD,GAAG,CAAC,QAAgB,EAAE,KAAa;QACjC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACzC,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;IACD,IAAI;QACF,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IAC7B,CAAC;IACD,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"}

package/dist/plugins/tidb-file-snapshot-store.d.ts

@@ -0,0 +1,26 @@
+import { type FileSnapshotBounds, type FileSnapshotStore, type FileSnapshotResult, type ExecutionEnv } from "@sema-agent/core";
+import type { Pool } from "mysql2/promise";
+import { type BlobBackend } from "./blob-backend.js";
+export declare class TiDBFileSnapshotStore implements FileSnapshotStore {
+    private readonly pool;
+    private readonly bounds;
+    private readonly inFlight;
+    private readonly blobs;
+    constructor(pool: Pool, blobBackend?: BlobBackend, bounds?: FileSnapshotBounds);
+    private putBlobRow;
+    putBlob(hash: string, bytes: Uint8Array): Promise<FileSnapshotResult>;
+    snapshot(scope: string, key: string, env: ExecutionEnv, root: string, signal?: AbortSignal): Promise<FileSnapshotResult>;
+    importManifest(scope: string, key: string, manifest: Map<string, string>, srcGetBlob: (hash: string) => Promise<Uint8Array | undefined>): Promise<FileSnapshotResult>;
+    restore(scope: string, key: string, env: ExecutionEnv, root: string, signal?: AbortSignal): Promise<FileSnapshotResult>;
+    has(scope: string, key: string): Promise<boolean>;
+    listKeys(scope: string): Promise<string[]>;
+    exportManifest(scope: string, key: string): Promise<Map<string, string> | null>;
+    getBlob(hash: string): Promise<Uint8Array | undefined>;
+    hasBlobs(hashes: string[]): Promise<Set<string>>;
+    reap(scope: string, keepKeys: string[]): Promise<number>;
+    deleteBySession(scope: string): Promise<number>;
+    private gcOrphanBlobs;
+    sweepOrphanBlobs(): Promise<number>;
+    private tx;
+}
+//# sourceMappingURL=tidb-file-snapshot-store.d.ts.map

package/dist/plugins/tidb-file-snapshot-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-file-snapshot-store.d.ts","sourceRoot":"","sources":["../../src/plugins/tidb-file-snapshot-store.ts"],"names":[],"mappings":"AAaA,OAAO,EAA2D,KAAK,kBAAkB,EAAE,KAAK,iBAAiB,EAAE,KAAK,kBAAkB,EAAE,KAAK,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAExL,OAAO,KAAK,EAAE,IAAI,EAAkD,MAAM,gBAAgB,CAAC;AAC3F,OAAO,EAAkB,KAAK,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAErE,qBAAa,qBAAsB,YAAW,iBAAiB;IAajD,OAAO,CAAC,QAAQ,CAAC,IAAI;IAAmC,OAAO,CAAC,QAAQ,CAAC,MAAM;IAN3F,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAqB;IAK9C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAc;gBACP,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,WAAW,EAAmB,MAAM,GAAE,kBAA4C;YAQ3H,UAAU;IAalB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAmBrE,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA6DxH,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,UAAU,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,kBAAkB,CAAC;IA+CrK,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAcvH,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAOjD,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAQ1C,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,IAAI,CAAC;IAQ/E,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC;IAMtD,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAQhD,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAaxD,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAWvC,aAAa;IAiBrB,gBAAgB,IAAI,OAAO,CAAC,MAAM,CAAC;YAI3B,EAAE;CAkBjB"}

package/dist/plugins/tidb-file-snapshot-store.js

@@ -0,0 +1,183 @@
+import { captureManifest, applyManifest, DEFAULT_SNAPSHOT_BOUNDS } from "@sema-agent/core";
+import { createHash } from "node:crypto";
+import { SqlBlobBackend } from "./blob-backend.js";
+export class TiDBFileSnapshotStore {
+    pool;
+    bounds;
+    inFlight = new Set();
+    blobs;
+    constructor(pool, blobBackend, bounds = DEFAULT_SNAPSHOT_BOUNDS) {
+        this.pool = pool;
+        this.bounds = bounds;
+        this.blobs = blobBackend ?? new SqlBlobBackend("tidb", pool);
+    }
+    async putBlobRow(hash, bytes) {
+        await this.blobs.putBlob(hash, bytes);
+    }
+    async putBlob(hash, bytes) {
+        if (createHash("sha256").update(bytes).digest("hex") !== hash) {
+            return { ok: false, error: { code: "read_failed", message: `putBlob ${hash} fails content-address integrity — nothing stored` } };
+        }
+        try {
+            await this.putBlobRow(hash, bytes);
+            return { ok: true };
+        }
+        catch (err) {
+            return { ok: false, error: { code: "restore_failed", message: `putBlob ${hash} write/upload failed: ${err.message}` } };
+        }
+    }
+    async snapshot(scope, key, env, root, signal) {
+        const flightKey = `${scope}\x00${key}`;
+        if (this.inFlight.has(flightKey))
+            return { ok: true };
+        this.inFlight.add(flightKey);
+        try {
+            if (await this.has(scope, key))
+                return { ok: true };
+            let cap;
+            try {
+                cap = await captureManifest(env, root, this.bounds, signal, async (hash, bytes) => {
+                    await this.putBlobRow(hash, bytes);
+                });
+            }
+            catch (err) {
+                return { ok: false, error: { code: "restore_failed", message: `snapshot blob write failed: ${err.message}` } };
+            }
+            if (!cap.ok)
+                return { ok: false, error: cap.error };
+            await this.tx(async (conn) => {
+                const [exists] = await conn.query("SELECT 1 FROM snapshot_manifest WHERE scope = ? AND snap_key = ? LIMIT 1", [scope, key]);
+                if (exists[0])
+                    return;
+                for (const [relPath, hash] of cap.value) {
+                    await conn.query("INSERT INTO snapshot_manifest (scope, snap_key, rel_path, blob_hash) VALUES (?,?,?,?)", [scope, key, relPath, hash]);
+                }
+            });
+            return { ok: true };
+        }
+        catch (err) {
+            return { ok: false, error: { code: "restore_failed", message: `snapshot persist failed: ${err.message}` } };
+        }
+        finally {
+            this.inFlight.delete(flightKey);
+        }
+    }
+    async importManifest(scope, key, manifest, srcGetBlob) {
+        const flightKey = `${scope}\x00${key}`;
+        if (this.inFlight.has(flightKey))
+            return { ok: true };
+        this.inFlight.add(flightKey);
+        try {
+            if (await this.has(scope, key))
+                return { ok: true };
+            for (const hash of new Set(manifest.values())) {
+                let bytes;
+                try {
+                    bytes = await srcGetBlob(hash);
+                }
+                catch (err) {
+                    return { ok: false, error: { code: "read_failed", message: `source blob ${hash} fetch failed: ${err.message}` } };
+                }
+                if (!bytes)
+                    return { ok: false, error: { code: "read_failed", message: `missing source blob ${hash}` } };
+                if (createHash("sha256").update(bytes).digest("hex") !== hash) {
+                    return { ok: false, error: { code: "read_failed", message: `source blob ${hash} fails content-address integrity` } };
+                }
+                await this.putBlobRow(hash, bytes);
+            }
+            await this.tx(async (conn) => {
+                const [exists] = await conn.query("SELECT 1 FROM snapshot_manifest WHERE scope = ? AND snap_key = ? LIMIT 1", [scope, key]);
+                if (exists[0])
+                    return;
+                for (const [relPath, hash] of manifest) {
+                    await conn.query("INSERT INTO snapshot_manifest (scope, snap_key, rel_path, blob_hash) VALUES (?,?,?,?)", [scope, key, relPath, hash]);
+                }
+            });
+            return { ok: true };
+        }
+        catch (err) {
+            return { ok: false, error: { code: "restore_failed", message: `import persist failed: ${err.message}` } };
+        }
+        finally {
+            this.inFlight.delete(flightKey);
+        }
+    }
+    async restore(scope, key, env, root, signal) {
+        try {
+            const [rows] = await this.pool.query("SELECT rel_path, blob_hash FROM snapshot_manifest WHERE scope = ? AND snap_key = ?", [scope, key]);
+            if (rows.length === 0)
+                return { ok: false, error: { code: "not_found", message: `no file snapshot for ${scope}/${key}` } };
+            const manifest = new Map(rows.map((r) => [String(r.rel_path), String(r.blob_hash)]));
+            return await applyManifest(env, root, this.bounds, signal, manifest, (hash) => this.blobs.getBlob(hash));
+        }
+        catch (err) {
+            return { ok: false, error: { code: "restore_failed", message: `restore blob read failed: ${err.message}` } };
+        }
+    }
+    async has(scope, key) {
+        const [rows] = await this.pool.query("SELECT 1 FROM snapshot_manifest WHERE scope = ? AND snap_key = ? LIMIT 1", [scope, key]);
+        return Boolean(rows[0]);
+    }
+    async listKeys(scope) {
+        const [rows] = await this.pool.query("SELECT DISTINCT snap_key FROM snapshot_manifest WHERE scope = ?", [scope]);
+        return rows.map((r) => String(r.snap_key));
+    }
+    async exportManifest(scope, key) {
+        const [rows] = await this.pool.query("SELECT rel_path, blob_hash FROM snapshot_manifest WHERE scope = ? AND snap_key = ?", [scope, key]);
+        if (rows.length === 0)
+            return null;
+        return new Map(rows.map((r) => [String(r.rel_path), String(r.blob_hash)]));
+    }
+    async getBlob(hash) {
+        return this.blobs.getBlob(hash);
+    }
+    async hasBlobs(hashes) {
+        return this.blobs.hasBlobs(hashes);
+    }
+    async reap(scope, keepKeys) {
+        return this.tx(async (conn) => {
+            const [keys] = await conn.query("SELECT DISTINCT snap_key FROM snapshot_manifest WHERE scope = ?", [scope]);
+            const drop = keys.map((k) => String(k.snap_key)).filter((k) => !keepKeys.includes(k));
+            for (const k of drop)
+                await conn.query("DELETE FROM snapshot_manifest WHERE scope = ? AND snap_key = ?", [scope, k]);
+            return drop.length;
+        });
+    }
+    async deleteBySession(scope) {
+        const [r] = await this.pool.query("DELETE FROM snapshot_manifest WHERE scope = ?", [scope]);
+        return r.affectedRows;
+    }
+    async gcOrphanBlobs() {
+        const [rows] = await this.pool
+            .query("SELECT blob_hash FROM snapshot_blob WHERE blob_hash NOT IN (SELECT blob_hash FROM snapshot_manifest)")
+            .catch(() => [[]]);
+        const orphans = rows.map((r) => String(r.blob_hash));
+        if (orphans.length === 0)
+            return 0;
+        return this.blobs.deleteBlobs(orphans);
+    }
+    async sweepOrphanBlobs() {
+        return this.gcOrphanBlobs();
+    }
+    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-file-snapshot-store.js.map

package/dist/plugins/tidb-file-snapshot-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-file-snapshot-store.js","sourceRoot":"","sources":["../../src/plugins/tidb-file-snapshot-store.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,uBAAuB,EAA+F,MAAM,kBAAkB,CAAC;AACxL,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,cAAc,EAAoB,MAAM,mBAAmB,CAAC;AAErE,MAAM,OAAO,qBAAqB;IAaH;IAAwD;IANpE,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IAK7B,KAAK,CAAc;IACpC,YAA6B,IAAU,EAAE,WAAyB,EAAmB,SAA6B,uBAAuB;QAA5G,SAAI,GAAJ,IAAI,CAAM;QAA8C,WAAM,GAAN,MAAM,CAA8C;QACvI,IAAI,CAAC,KAAK,GAAG,WAAW,IAAI,IAAI,cAAc,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC/D,CAAC;IAMO,KAAK,CAAC,UAAU,CAAC,IAAY,EAAE,KAAiB;QACtD,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACxC,CAAC;IAWD,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,KAAiB;QAI3C,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;YAC9D,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,WAAW,IAAI,mDAAmD,EAAE,EAAE,CAAC;QACpI,CAAC;QACD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACnC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YAIb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,WAAW,IAAI,yBAA0B,GAAa,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;QACrI,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,GAAW,EAAE,GAAiB,EAAE,IAAY,EAAE,MAAoB;QAC9F,MAAM,SAAS,GAAG,GAAG,KAAK,OAAO,GAAG,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7B,IAAI,CAAC;YAGH,IAAI,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;YAQpD,IAAI,GAAG,CAAC;YACR,IAAI,CAAC;gBACH,GAAG,GAAG,MAAM,eAAe,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;oBAGhF,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gBACrC,CAAC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,+BAAgC,GAAa,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;YAC5H,CAAC;YACD,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC;YAEpD,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;gBAC3B,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAkB,0EAA0E,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;gBAC7I,IAAI,MAAM,CAAC,CAAC,CAAC;oBAAE,OAAO;gBACtB,KAAK,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;oBACxC,MAAM,IAAI,CAAC,KAAK,CAAC,uFAAuF,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;gBACzI,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YAEb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,4BAA6B,GAAa,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;QACzH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAoBD,KAAK,CAAC,cAAc,CAAC,KAAa,EAAE,GAAW,EAAE,QAA6B,EAAE,UAA6D;QAG3I,MAAM,SAAS,GAAG,GAAG,KAAK,OAAO,GAAG,EAAE,CAAC;QACvC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC;YAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7B,IAAI,CAAC;YAEH,IAAI,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;YAIpD,KAAK,MAAM,IAAI,IAAI,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;gBAC9C,IAAI,KAA6B,CAAC;gBAClC,IAAI,CAAC;oBACH,KAAK,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;gBACjC,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBAEb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,IAAI,kBAAmB,GAAa,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;gBAC/H,CAAC;gBACD,IAAI,CAAC,KAAK;oBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,uBAAuB,IAAI,EAAE,EAAE,EAAE,CAAC;gBAEzG,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;oBAC9D,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,eAAe,IAAI,kCAAkC,EAAE,EAAE,CAAC;gBACvH,CAAC;gBACD,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACrC,CAAC;YAGD,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;gBAC3B,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAkB,0EAA0E,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;gBAC7I,IAAI,MAAM,CAAC,CAAC,CAAC;oBAAE,OAAO;gBACtB,KAAK,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,QAAQ,EAAE,CAAC;oBACvC,MAAM,IAAI,CAAC,KAAK,CAAC,uFAAuF,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;gBACzI,CAAC;YACH,CAAC,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YAGb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,0BAA2B,GAAa,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;QACvH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAGD,KAAK,CAAC,OAAO,CAAC,KAAa,EAAE,GAAW,EAAE,GAAiB,EAAE,IAAY,EAAE,MAAoB;QAI7F,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,oFAAoF,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;YAC1J,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,wBAAwB,KAAK,IAAI,GAAG,EAAE,EAAE,EAAE,CAAC;YAC3H,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAiB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YACrG,OAAO,MAAM,aAAa,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3G,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,6BAA8B,GAAa,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;QAC1H,CAAC;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAa,EAAE,GAAW;QAClC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,0EAA0E,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;QAChJ,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1B,CAAC;IAID,KAAK,CAAC,QAAQ,CAAC,KAAa;QAC1B,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,iEAAiE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAClI,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC7C,CAAC;IAKD,KAAK,CAAC,cAAc,CAAC,KAAa,EAAE,GAAW;QAC7C,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,oFAAoF,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;QAC1J,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACnC,OAAO,IAAI,GAAG,CAAiB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7F,CAAC;IAID,KAAK,CAAC,OAAO,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;IAID,KAAK,CAAC,QAAQ,CAAC,MAAgB;QAC7B,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAMD,KAAK,CAAC,IAAI,CAAC,KAAa,EAAE,QAAkB;QAC1C,OAAO,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;YAC5B,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAkB,iEAAiE,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;YAC7H,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YACtF,KAAK,MAAM,CAAC,IAAI,IAAI;gBAAE,MAAM,IAAI,CAAC,KAAK,CAAC,gEAAgE,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;YACrH,OAAO,IAAI,CAAC,MAAM,CAAC;QACrB,CAAC,CAAC,CAAC;IACL,CAAC;IAMD,KAAK,CAAC,eAAe,CAAC,KAAa;QACjC,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAkB,+CAA+C,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAC7G,OAAO,CAAC,CAAC,YAAY,CAAC;IACxB,CAAC;IAQO,KAAK,CAAC,aAAa;QAIzB,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,IAAI,CAAC,IAAI;aAC3B,KAAK,CAAkB,sGAAsG,CAAC;aAC9H,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAqB,CAAsB,CAAC,CAAC;QAC7D,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QACrD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QACnC,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;IAOD,KAAK,CAAC,gBAAgB;QACpB,OAAO,IAAI,CAAC,aAAa,EAAE,CAAC;IAC9B,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-image-bake.d.ts

@@ -0,0 +1,102 @@
+import type { Pool } from "mysql2/promise";
+export type BakeStatus = "queued" | "running" | "done" | "failed";
+export type BakeState = "PENDING" | "BUILDING" | "PUSHING" | "VERIFYING" | "REGISTERING" | "COMPLETE" | "FAILED" | "CANCELLED";
+export type BakeErrorCode = "band-conflict" | "duplicate" | "disk" | "timeout" | "cancelled" | null;
+export interface BakeRecord {
+    bakeId: string;
+    profile: string;
+    bands: string[] | null;
+    baseRef: string | null;
+    push: boolean;
+    dryRun: boolean;
+    logs: boolean;
+    argv: string[];
+    status: BakeStatus;
+    state: BakeState | null;
+    digest: string | null;
+    repo: string | null;
+    ref: string | null;
+    indexId: string | null;
+    exitCode: number | null;
+    error: string | null;
+    errorCode: BakeErrorCode;
+    manifestSha: string | null;
+    tag: string | null;
+    idemKey: string | null;
+    ingestSecret: string | null;
+    runnerId: string | null;
+    leaseUntil: string | null;
+    cancelRequested: boolean;
+    requestedBy: string | null;
+    createdAt: string;
+    updatedAt: string;
+}
+export interface BakeEvent {
+    seq: number;
+    lineOrd: number | null;
+    kind: string;
+    level: string | null;
+    data: unknown;
+    ts: string;
+}
+export interface CreateBakeInput {
+    profile: string;
+    bands: string[] | null;
+    baseRef: string | null;
+    push: boolean;
+    dryRun: boolean;
+    logs: boolean;
+    argv: string[];
+    idemKey: string | null;
+    requestedBy: string | null;
+}
+export interface BakeTerminal {
+    status: "done" | "failed";
+    state: BakeState;
+    digest?: string | null;
+    repo?: string | null;
+    ref?: string | null;
+    indexId?: string | null;
+    exitCode?: number | null;
+    error?: string | null;
+    errorCode?: BakeErrorCode;
+    manifestSha?: string | null;
+    tag?: string | null;
+}
+export declare class TiDBImageBake {
+    private readonly pool;
+    private readonly poolName;
+    constructor(pool: Pool, poolName?: string);
+    createBake(input: CreateBakeInput): Promise<{
+        created: boolean;
+        bake: BakeRecord;
+    }>;
+    createBakeIfIdle(input: CreateBakeInput): Promise<{
+        created: boolean;
+        bake: BakeRecord | null;
+    }>;
+    getBake(bakeId: string): Promise<BakeRecord | null>;
+    getBakeByIdem(idemKey: string): Promise<BakeRecord | null>;
+    findActiveByArgv(argv: string[]): Promise<BakeRecord | null>;
+    findActiveAny(): Promise<BakeRecord | null>;
+    findNextQueuedId(): Promise<string | null>;
+    claimBake(bakeId: string, runnerId: string, leaseMs: number): Promise<BakeRecord | null>;
+    appendEvent(bakeId: string, kind: string, data: unknown, opts?: {
+        lineOrd?: number | null;
+        level?: string | null;
+    }): Promise<number | null>;
+    private lineOrdExists;
+    maxSeq(bakeId: string): Promise<number>;
+    retainedFrom(bakeId: string): Promise<number>;
+    getEvents(bakeId: string, afterSeq: number): Promise<BakeEvent[]>;
+    setStatus(bakeId: string, status: BakeStatus): Promise<void>;
+    setState(bakeId: string, state: BakeState): Promise<void>;
+    setTerminal(bakeId: string, t: BakeTerminal): Promise<number>;
+    setIndexId(bakeId: string, indexId: string): Promise<void>;
+    heartbeatLease(bakeId: string, runnerId: string, leaseMs: number): Promise<boolean>;
+    requestCancel(bakeId: string): Promise<boolean>;
+    isCancelRequested(bakeId: string): Promise<boolean>;
+    reapStaleBakes(olderThanMs: number): Promise<number>;
+    private tx;
+}
+//# sourceMappingURL=tidb-image-bake.d.ts.map

package/dist/plugins/tidb-image-bake.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tidb-image-bake.d.ts","sourceRoot":"","sources":["../../src/plugins/tidb-image-bake.ts"],"names":[],"mappings":"AAqBA,OAAO,KAAK,EAAE,IAAI,EAAkD,MAAM,gBAAgB,CAAC;AAG3F,MAAM,MAAM,UAAU,GAAG,QAAQ,GAAG,SAAS,GAAG,MAAM,GAAG,QAAQ,CAAC;AAElE,MAAM,MAAM,SAAS,GACjB,SAAS,GACT,UAAU,GACV,SAAS,GACT,WAAW,GACX,aAAa,GACb,UAAU,GACV,QAAQ,GACR,WAAW,CAAC;AAGhB,MAAM,MAAM,aAAa,GAAG,eAAe,GAAG,WAAW,GAAG,MAAM,GAAG,SAAS,GAAG,WAAW,GAAG,IAAI,CAAC;AAEpG,MAAM,WAAW,UAAU;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IACvB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,IAAI,EAAE,OAAO,CAAC;IACd,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,EAAE,OAAO,CAAC;IACd,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,MAAM,EAAE,UAAU,CAAC;IACnB,KAAK,EAAE,SAAS,GAAG,IAAI,CAAC;IACxB,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,SAAS,EAAE,aAAa,CAAC;IACzB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,GAAG,EAAE,MAAM,GAAG,IAAI,CAAC;IACnB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,UAAU,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,eAAe,EAAE,OAAO,CAAC;IACzB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,SAAS;IACxB,GAAG,EAAE,MAAM,CAAC;IAEZ,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,IAAI,EAAE,MAAM,CAAC;IAEb,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,IAAI,EAAE,OAAO,CAAC;IACd,EAAE,EAAE,MAAM,CAAC;CACZ;AAGD,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IACvB,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,IAAI,EAAE,OAAO,CAAC;IACd,MAAM,EAAE,OAAO,CAAC;IAChB,IAAI,EAAE,OAAO,CAAC;IAEd,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,WAAW,EAAE,MAAM,GAAG,IAAI,CAAC;CAC5B;AAGD,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,MAAM,GAAG,QAAQ,CAAC;IAC1B,KAAK,EAAE,SAAS,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,GAAG,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,SAAS,CAAC,EAAE,aAAa,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,GAAG,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACrB;AAmGD,qBAAa,aAAa;IAEtB,OAAO,CAAC,QAAQ,CAAC,IAAI;IAErB,OAAO,CAAC,QAAQ,CAAC,QAAQ;gBAFR,IAAI,EAAE,IAAI,EAEV,QAAQ,SAAa;IASlC,UAAU,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,EAAE,UAAU,CAAA;KAAE,CAAC;IAuDnF,gBAAgB,CAAC,KAAK,EAAE,eAAe,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,EAAE,UAAU,GAAG,IAAI,CAAA;KAAE,CAAC;IA4DhG,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAQnD,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAc1D,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAe5D,aAAa,IAAI,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAe3C,gBAAgB,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAgB1C,SAAS,CACb,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC;IAkHvB,WAAW,CACf,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,OAAO,EACb,IAAI,GAAE;QAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;KAAO,GAC5D,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;YA8BX,aAAa;IASrB,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOvC,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAO7C,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAgBjE,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ5D,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBzD,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC;IAgC7D,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAU1D,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IA2BnF,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAS/C,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAoBnD,cAAc,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YA2C5C,EAAE;CAiBjB"}