@sema-agent/server 1.314.0 → 1.316.0

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

@@ -1,193 +1,2 @@
-import { captureManifest, applyManifest, DEFAULT_SNAPSHOT_BOUNDS } from "@sema-agent/core";
-import { createHash } from "node:crypto";
-import { SqlBlobBackend, sqlBlobSizes } from "./blob-backend.js";
-export class TiDBFileSnapshotStore {
-    pool;
-    bounds;
-    inFlight = new Set();
-    blobs;
-    onSkippedVanished = undefined;
-    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 };
-            if (cap.skippedVanished > 0 && this.onSkippedVanished) {
-                try {
-                    void Promise.resolve(this.onSkippedVanished(cap.skippedVanished)).catch(() => { });
-                }
-                catch { }
-            }
-            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 blobSizes(hashes) {
-        return sqlBlobSizes("tidb", this.pool, 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();
-        }
-    }
-}
+export { TiDBFileSnapshotStore } from "./file-snapshot-store-sql.js";
 //# sourceMappingURL=tidb-file-snapshot-store.js.map

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

@@ -1,41 +1,4 @@
-import type { Pool } from "mysql2/promise";
-import { type BakeStatus, type BakeState, type BakeRecord, type BakeEvent, type CreateBakeInput, type BakeTerminal } from "./store-contracts.js";
+export { TiDBImageBake, SqlImageBake } from "./image-bake-store-sql.js";
 export type { BakeStatus, BakeState, BakeErrorCode, BakeRecord, BakeEvent, CreateBakeInput, BakeTerminal, BakeRow, } from "./store-contracts.js";
 export { mapBakeRow as mapRow, BAKE_SELECT_COLS as SELECT_COLS } from "./store-contracts.js";
-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<boolean>;
-    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.js

@@ -1,350 +1,3 @@
-import { uuidv7 } from "@sema-agent/core";
-import { randomBytes } from "node:crypto";
-import { parseJsonOr as parseJson, toIso as iso } from "./sql-row-helpers.js";
-import { mapBakeRow as mapRow, BAKE_SELECT_COLS as SELECT_COLS, } from "./store-contracts.js";
+export { TiDBImageBake, SqlImageBake } from "./image-bake-store-sql.js";
 export { mapBakeRow as mapRow, BAKE_SELECT_COLS as SELECT_COLS } from "./store-contracts.js";
-function isDupKey(err) {
-    return Boolean(err) && err.code === "ER_DUP_ENTRY";
-}
-function dupKeyName(err) {
-    const msg = String(err?.sqlMessage ?? err?.message ?? "");
-    if (/uq_line/i.test(msg))
-        return "uq_line";
-    if (/PRIMARY/i.test(msg))
-        return "primary";
-    return "other";
-}
-export class TiDBImageBake {
-    pool;
-    poolName;
-    constructor(pool, poolName = "standard") {
-        this.pool = pool;
-        this.poolName = poolName;
-    }
-    async createBake(input) {
-        const bakeId = uuidv7();
-        const now = new Date();
-        try {
-            await this.pool.query("INSERT INTO image_bake (bake_id, profile, bands, base_ref, push, dry_run, logs, argv, status, " +
-                "idem_key, cancel_requested, requested_by, created_at, updated_at) " +
-                "VALUES (?,?,?,?,?,?,?,?,'queued',?,0,?,?,?)", [
-                bakeId,
-                input.profile,
-                input.bands == null ? null : JSON.stringify(input.bands),
-                input.baseRef,
-                input.push ? 1 : 0,
-                input.dryRun ? 1 : 0,
-                input.logs ? 1 : 0,
-                JSON.stringify(input.argv),
-                input.idemKey,
-                input.requestedBy,
-                now,
-                now,
-            ]);
-        }
-        catch (err) {
-            if (isDupKey(err) && input.idemKey) {
-                const existing = await this.getBakeByIdem(input.idemKey);
-                if (existing)
-                    return { created: false, bake: existing };
-            }
-            throw err;
-        }
-        const bake = await this.getBake(bakeId);
-        if (!bake)
-            throw new Error("createBake: row vanished after insert");
-        return { created: true, bake };
-    }
-    async createBakeIfIdle(input) {
-        const bakeId = uuidv7();
-        const now = new Date();
-        const conn = await this.pool.getConnection();
-        let inserted = false;
-        try {
-            await conn.query("BEGIN PESSIMISTIC");
-            await conn.query("INSERT IGNORE INTO image_bake_admit (pool) VALUES (?)", [this.poolName]);
-            await conn.query("SELECT pool FROM image_bake_admit WHERE pool = ? FOR UPDATE", [this.poolName]);
-            const [busyRows] = await conn.query("SELECT 1 FROM image_bake WHERE status IN ('queued','running') AND dry_run = 0 LIMIT 1 FOR UPDATE");
-            if (busyRows.length === 0) {
-                await conn.query("INSERT INTO image_bake (bake_id, profile, bands, base_ref, push, dry_run, logs, argv, status, " +
-                    "idem_key, cancel_requested, requested_by, created_at, updated_at) " +
-                    "VALUES (?,?,?,?,?,?,?,?,'queued',?,0,?,?,?)", [
-                    bakeId,
-                    input.profile,
-                    input.bands == null ? null : JSON.stringify(input.bands),
-                    input.baseRef,
-                    input.push ? 1 : 0,
-                    input.dryRun ? 1 : 0,
-                    input.logs ? 1 : 0,
-                    JSON.stringify(input.argv),
-                    input.idemKey,
-                    input.requestedBy,
-                    now,
-                    now,
-                ]);
-                inserted = true;
-            }
-            await conn.commit();
-        }
-        catch (err) {
-            try {
-                await conn.rollback();
-            }
-            catch {
-            }
-            throw err;
-        }
-        finally {
-            conn.release();
-        }
-        if (!inserted)
-            return { created: false, bake: null };
-        const bake = await this.getBake(bakeId);
-        if (!bake)
-            throw new Error("createBakeIfIdle: row vanished after insert");
-        return { created: true, bake };
-    }
-    async getBake(bakeId) {
-        const [rows] = await this.pool.query(`SELECT ${SELECT_COLS} FROM image_bake WHERE bake_id = ? LIMIT 1`, [bakeId]);
-        return rows[0] ? mapRow(rows[0]) : null;
-    }
-    async getBakeByIdem(idemKey) {
-        const [rows] = await this.pool.query(`SELECT ${SELECT_COLS} FROM image_bake WHERE idem_key = ? LIMIT 1`, [idemKey]);
-        return rows[0] ? mapRow(rows[0]) : null;
-    }
-    async findActiveByArgv(argv) {
-        const [rows] = await this.pool.query(`SELECT ${SELECT_COLS} FROM image_bake WHERE status IN ('queued','running') AND argv = CAST(? AS JSON) ` +
-            "ORDER BY created_at DESC LIMIT 1", [JSON.stringify(argv)]);
-        return rows[0] ? mapRow(rows[0]) : null;
-    }
-    async findActiveAny() {
-        const [rows] = await this.pool.query(`SELECT ${SELECT_COLS} FROM image_bake WHERE status IN ('queued','running') AND dry_run = 0 ` +
-            "ORDER BY created_at ASC LIMIT 1", []);
-        return rows[0] ? mapRow(rows[0]) : null;
-    }
-    async findNextQueuedId() {
-        const [rows] = await this.pool.query("SELECT bake_id FROM image_bake WHERE status = 'queued' ORDER BY created_at ASC LIMIT 1", []);
-        return rows[0] ? String(rows[0].bake_id) : null;
-    }
-    async claimBake(bakeId, runnerId, leaseMs) {
-        const conn = await this.pool.getConnection();
-        try {
-            await conn.beginTransaction();
-            const now = new Date();
-            const leaseUntil = new Date(now.getTime() + leaseMs);
-            const ingestSecret = randomBytes(24).toString("hex");
-            const [bakeRes] = await conn.query("UPDATE image_bake SET status='running', state='PENDING', runner_id=?, ingest_secret=?, " +
-                "lease_until=?, updated_at=? WHERE bake_id=? AND status='queued'", [runnerId, ingestSecret, leaseUntil, now, bakeId]);
-            if (bakeRes.affectedRows !== 1) {
-                await conn.rollback();
-                return null;
-            }
-            const [dryRows] = await conn.query("SELECT dry_run FROM image_bake WHERE bake_id=?", [bakeId]);
-            if (Boolean(dryRows[0]?.dry_run)) {
-                await conn.commit();
-                return (await this.getBake(bakeId)) ?? null;
-            }
-            const [leaseRows] = await conn.query("SELECT bake_id, expires_at FROM image_bake_lease WHERE pool=?", [this.poolName]);
-            const cur = leaseRows[0];
-            if (!cur) {
-                try {
-                    await conn.query("INSERT INTO image_bake_lease (pool, bake_id, acquired_at, expires_at) VALUES (?,?,?,?)", [this.poolName, bakeId, now, leaseUntil]);
-                }
-                catch (err) {
-                    if (isDupKey(err)) {
-                        await conn.rollback();
-                        return null;
-                    }
-                    throw err;
-                }
-            }
-            else {
-                if (new Date(cur.expires_at).getTime() >= now.getTime()) {
-                    await conn.rollback();
-                    return null;
-                }
-                const holderId = String(cur.bake_id ?? "");
-                const [holderRows] = await conn.query("SELECT status FROM image_bake WHERE bake_id=?", [holderId]);
-                const holderStatus = String(holderRows[0]?.status ?? "");
-                if (holderStatus === "queued" || holderStatus === "running") {
-                    await conn.rollback();
-                    return null;
-                }
-                const [stealRes] = await conn.query("UPDATE image_bake_lease SET bake_id=?, acquired_at=?, expires_at=? WHERE pool=? AND bake_id=? AND expires_at=?", [bakeId, now, leaseUntil, this.poolName, holderId, cur.expires_at]);
-                if (stealRes.affectedRows === 0) {
-                    await conn.rollback();
-                    return null;
-                }
-            }
-            const [check] = await conn.query("SELECT bake_id FROM image_bake_lease WHERE pool=?", [this.poolName]);
-            if (String(check[0]?.bake_id ?? "") !== bakeId) {
-                await conn.rollback();
-                return null;
-            }
-            await conn.commit();
-            return (await this.getBake(bakeId)) ?? null;
-        }
-        catch (err) {
-            try {
-                await conn.rollback();
-            }
-            catch {
-            }
-            throw err;
-        }
-        finally {
-            conn.release();
-        }
-    }
-    async appendEvent(bakeId, kind, data, opts = {}) {
-        const lineOrd = opts.lineOrd ?? null;
-        if (lineOrd !== null && (await this.lineOrdExists(bakeId, lineOrd)))
-            return null;
-        const payload = data == null ? null : JSON.stringify(data);
-        for (let attempt = 0; attempt < 50; attempt += 1) {
-            const seq = (await this.maxSeq(bakeId)) + 1;
-            try {
-                await this.pool.query("INSERT INTO image_bake_event (bake_id, seq, line_ord, kind, level, data, ts) VALUES (?,?,?,?,?,?,?)", [bakeId, seq, lineOrd, kind, opts.level ?? null, payload, new Date()]);
-                return seq;
-            }
-            catch (err) {
-                if (!isDupKey(err))
-                    throw err;
-                const which = dupKeyName(err);
-                if (which === "uq_line" && lineOrd !== null)
-                    return null;
-                await new Promise((r) => setTimeout(r, Math.floor(Math.random() * 5)));
-            }
-        }
-        throw new Error(`appendEvent: seq allocation lost the (bake_id,seq) race after retries (bakeId=${bakeId})`);
-    }
-    async lineOrdExists(bakeId, lineOrd) {
-        const [rows] = await this.pool.query("SELECT 1 FROM image_bake_event WHERE bake_id = ? AND line_ord = ? LIMIT 1", [bakeId, lineOrd]);
-        return rows.length > 0;
-    }
-    async maxSeq(bakeId) {
-        const [rows] = await this.pool.query("SELECT MAX(seq) AS m FROM image_bake_event WHERE bake_id = ?", [bakeId]);
-        const m = rows[0]?.m;
-        return m == null ? 0 : Number(m);
-    }
-    async retainedFrom(bakeId) {
-        const [rows] = await this.pool.query("SELECT MIN(seq) AS m FROM image_bake_event WHERE bake_id = ?", [bakeId]);
-        const m = rows[0]?.m;
-        return m == null ? 0 : Number(m);
-    }
-    async getEvents(bakeId, afterSeq) {
-        const [rows] = await this.pool.query("SELECT seq, line_ord, kind, level, data, ts FROM image_bake_event WHERE bake_id = ? AND seq > ? ORDER BY seq ASC", [bakeId, afterSeq]);
-        return rows.map((r) => ({
-            seq: Number(r.seq),
-            lineOrd: r.line_ord === null ? null : Number(r.line_ord),
-            kind: String(r.kind),
-            level: r.level ?? null,
-            data: parseJson(r.data, null),
-            ts: iso(r.ts),
-        }));
-    }
-    async setStatus(bakeId, status) {
-        await this.pool.query("UPDATE image_bake SET status = ?, updated_at = ? WHERE bake_id = ?", [status, new Date(), bakeId]);
-    }
-    async setState(bakeId, state) {
-        await this.pool.query("UPDATE image_bake SET state = ?, updated_at = ? WHERE bake_id = ?", [state, new Date(), bakeId]);
-    }
-    async setTerminal(bakeId, t) {
-        let affectedRows = 0;
-        await this.tx(async (conn) => {
-            const [res] = await conn.query("UPDATE image_bake SET status=?, state=?, digest=?, repo=?, ref=?, index_id=?, exit_code=?, error=?, " +
-                "error_code=?, manifest_sha=?, tag=?, ingest_secret=NULL, updated_at=? WHERE bake_id=? AND status IN ('queued','running')", [
-                t.status,
-                t.state,
-                t.digest ?? null,
-                t.repo ?? null,
-                t.ref ?? null,
-                t.indexId ?? null,
-                t.exitCode ?? null,
-                t.error ?? null,
-                t.errorCode ?? null,
-                t.manifestSha ?? null,
-                t.tag ?? null,
-                new Date(),
-                bakeId,
-            ]);
-            affectedRows = res.affectedRows;
-            await conn.query("DELETE FROM image_bake_lease WHERE pool = ? AND bake_id = ?", [this.poolName, bakeId]);
-        });
-        return affectedRows;
-    }
-    async setIndexId(bakeId, indexId) {
-        const [res] = await this.pool.query("UPDATE image_bake SET index_id = ?, updated_at = ? WHERE bake_id = ? AND status IN ('queued','running')", [indexId, new Date(), bakeId]);
-        return res.affectedRows >= 1;
-    }
-    async heartbeatLease(bakeId, runnerId, leaseMs) {
-        const now = new Date();
-        const leaseUntil = new Date(now.getTime() + leaseMs);
-        const [res] = await this.pool.query("UPDATE image_bake SET lease_until = ?, updated_at = ? WHERE bake_id = ? AND runner_id = ? AND status = 'running'", [leaseUntil, now, bakeId, runnerId]);
-        if (res.affectedRows < 1)
-            return false;
-        const [dryRows] = await this.pool.query("SELECT dry_run FROM image_bake WHERE bake_id = ?", [bakeId]);
-        if (Boolean(dryRows[0]?.dry_run))
-            return true;
-        const [leaseRes] = await this.pool.query("UPDATE image_bake_lease SET expires_at = ? WHERE pool = ? AND bake_id = ?", [leaseUntil, this.poolName, bakeId]);
-        return leaseRes.affectedRows >= 1;
-    }
-    async requestCancel(bakeId) {
-        const [r] = await this.pool.query("UPDATE image_bake SET cancel_requested = 1, updated_at = ? WHERE bake_id = ? AND status IN ('queued','running')", [new Date(), bakeId]);
-        return r.affectedRows > 0;
-    }
-    async isCancelRequested(bakeId) {
-        const [rows] = await this.pool.query("SELECT cancel_requested FROM image_bake WHERE bake_id = ?", [bakeId]);
-        return Boolean(rows[0]?.cancel_requested);
-    }
-    async reapStaleBakes(olderThanMs) {
-        const cutoff = new Date(Date.now() - olderThanMs);
-        const [victims] = await this.pool.query("SELECT bake_id FROM image_bake WHERE status = 'running' AND " +
-            "(lease_until IS NULL OR lease_until < ?) AND updated_at < ?", [cutoff, cutoff]);
-        let reaped = 0;
-        for (const v of victims) {
-            const bakeId = String(v.bake_id);
-            const [res] = await this.pool.query("UPDATE image_bake SET status='failed', state='FAILED', error='bake-runner lost (lease expired)', " +
-                "ingest_secret=NULL, index_id=NULL, updated_at=? WHERE bake_id=? AND status='running'", [new Date(), bakeId]);
-            if (res.affectedRows === 1) {
-                reaped += 1;
-                try {
-                    await this.appendEvent(bakeId, "done", {
-                        status: "FAILED",
-                        digest: null,
-                        repo: null,
-                        ref: null,
-                        indexId: null,
-                        exitCode: null,
-                        error: "bake-runner lost (lease expired)",
-                        errorCode: null,
-                    });
-                }
-                catch {
-                }
-                await this.pool.query("DELETE FROM image_bake_lease WHERE pool = ? AND bake_id = ?", [this.poolName, bakeId]);
-            }
-        }
-        return reaped;
-    }
-    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();
-        }
-    }
-}
 //# sourceMappingURL=tidb-image-bake.js.map

package/dist/plugins/tidb-image-index.d.ts

@@ -1,22 +1,4 @@
-import type { Pool } from "mysql2/promise";
-import { type ImageStatus, type ImageIndexEntry, type ImageIndexUpsert, type ImageViewer, type ImageListFilter } from "./store-contracts.js";
+export { TiDBImageIndex } from "./image-index-sql.js";
 export type { ImageNestedBuildMode, ImageCapabilities, ImagePodContract, ImageStatus, ImageVisibility, ImageIndexEntry, ImageIndexUpsert, ImageViewer, ImageListFilter, ImageRow, } from "./store-contracts.js";
 export { mapImageRow as mapRow } from "./store-contracts.js";
-export declare class TiDBImageIndex {
-    private readonly pool;
-    constructor(pool: Pool);
-    static idFor(repo: string, digest: string): string;
-    upsert(e: ImageIndexUpsert): Promise<string>;
-    registerBuilding(e: ImageIndexUpsert): Promise<string>;
-    getById(id: string): Promise<ImageIndexEntry | null>;
-    getByDigest(digest: string): Promise<ImageIndexEntry | null>;
-    latestPublished(profile: string, viewer: ImageViewer): Promise<ImageIndexEntry | null>;
-    list(filter: ImageListFilter): Promise<{
-        entries: ImageIndexEntry[];
-        nextCursor: string | null;
-    }>;
-    setStatus(id: string, status: ImageStatus, opts?: {
-        supersedes?: string | null;
-    }): Promise<void>;
-}
 //# sourceMappingURL=tidb-image-index.d.ts.map