@vibes.diy/base 14.3.45 → 14.3.47

package/hooks/img-gen/picture-bytes.js

@@ -0,0 +1,312 @@
+import { exception2Result } from "@adviser/cement";
+import { type } from "arktype";
+import { FILES_ASSET_STATE_HEADER, FILES_UPLOAD_ID_HEADER, KEEPABLE_ASSET_STATES } from "@vibes.diy/vibe-types";
+import { vibe } from "@vibes.diy/vibe-runtime";
+function defaultWarn(event, fields) {
+    vibe.log("warn", event, fields);
+}
+const RASTER_IMAGE_TYPES = ["image/png", "image/jpeg", "image/webp", "image/gif", "image/avif", "image/bmp"];
+export function isRasterImageType(mime) {
+    return RASTER_IMAGE_TYPES.includes(mime.split(";")[0].trim().toLowerCase());
+}
+const DEFAULT_MEMORY_BYTES = 48 * 1024 * 1024;
+const DEFAULT_LOCAL_LOOKUP_MS = 1500;
+class MemoryTier {
+    max;
+    entries = new Map();
+    total = 0;
+    constructor(max) {
+        this.max = max;
+    }
+    get(id) {
+        const b = this.entries.get(id);
+        if (b === undefined)
+            return undefined;
+        this.entries.delete(id);
+        this.entries.set(id, b);
+        return b;
+    }
+    delete(id) {
+        const b = this.entries.get(id);
+        if (b === undefined)
+            return;
+        this.entries.delete(id);
+        this.total -= b.size;
+    }
+    set(id, b) {
+        const prior = this.entries.get(id);
+        if (prior !== undefined) {
+            this.total -= prior.size;
+            this.entries.delete(id);
+        }
+        if (b.size > this.max)
+            return;
+        this.entries.set(id, b);
+        this.total += b.size;
+        for (const [oldId, old] of this.entries) {
+            if (this.total <= this.max)
+                break;
+            this.entries.delete(oldId);
+            this.total -= old.size;
+        }
+    }
+}
+const TIMED_OUT = Symbol("timed-out");
+function withinMs(p, ms) {
+    return new Promise((resolve, reject) => {
+        const timer = setTimeout(() => resolve(TIMED_OUT), ms);
+        p.then((v) => {
+            clearTimeout(timer);
+            resolve(v);
+        }, (e) => {
+            clearTimeout(timer);
+            reject(e);
+        });
+    });
+}
+export function createPictureStore(opts) {
+    const fetcher = opts.fetcher ?? defaultFetch;
+    const warn = opts.warn ?? defaultWarn;
+    const memory = new MemoryTier(opts.memoryBytes ?? DEFAULT_MEMORY_BYTES);
+    const lookupMs = opts.localLookupMs ?? DEFAULT_LOCAL_LOOKUP_MS;
+    const filesPrefix = `${opts.assetOrigin}/_files/`;
+    const inFlight = new Map();
+    const notKeepable = new Map();
+    const unconfirmed = new Set();
+    const confirmInFlight = new Map();
+    async function keep(uploadId, blob) {
+        if (isRasterImageType(blob.type) === false) {
+            warn("picture-bytes-not-raster", { uploadId, type: blob.type });
+            return;
+        }
+        memory.set(uploadId, blob);
+        const rPut = await exception2Result(() => opts.bytes.put(uploadId, blob));
+        if (rPut.isErr())
+            warn("picture-bytes-put-failed", { uploadId, message: rPut.Err().message });
+    }
+    async function seed(uploadId, blob) {
+        if (isRasterImageType(blob.type) === false) {
+            warn("picture-bytes-not-raster", { uploadId, type: blob.type });
+            return;
+        }
+        memory.set(uploadId, blob);
+        unconfirmed.add(uploadId);
+    }
+    function dropUnconfirmed(uploadId) {
+        if (unconfirmed.delete(uploadId))
+            memory.delete(uploadId);
+    }
+    function judge(uploadId, res, type) {
+        const servedUpload = res.headers.get(FILES_UPLOAD_ID_HEADER);
+        const state = res.headers.get(FILES_ASSET_STATE_HEADER);
+        const named = servedUpload === uploadId;
+        const rasterOk = type === undefined ? true : isRasterImageType(type);
+        return { named, keepable: named && state !== null && KEEPABLE_ASSET_STATES.includes(state) && rasterOk };
+    }
+    async function headConfirm(uploadId, url) {
+        if (unconfirmed.has(uploadId) === false)
+            return "none";
+        if (url.startsWith(filesPrefix) === false)
+            return "unknown";
+        const rRes = await exception2Result(() => fetcher(url, { method: "HEAD", credentials: "include" }));
+        if (rRes.isErr()) {
+            warn("picture-bytes-confirm-failed", { uploadId, message: rRes.Err().message });
+            return "unknown";
+        }
+        const res = rRes.Ok();
+        if (res.ok === false)
+            return "unknown";
+        const seeded = memory.get(uploadId);
+        const verdict = judge(uploadId, res, seeded?.type);
+        if (verdict.keepable === false || seeded === undefined) {
+            dropUnconfirmed(uploadId);
+            return "dropped";
+        }
+        unconfirmed.delete(uploadId);
+        await keep(uploadId, seeded);
+        return "kept";
+    }
+    function confirm(ref) {
+        const pending = confirmInFlight.get(ref.uploadId);
+        if (pending !== undefined)
+            return pending;
+        const p = headConfirm(ref.uploadId, ref.url).finally(() => confirmInFlight.delete(ref.uploadId));
+        confirmInFlight.set(ref.uploadId, p);
+        return p;
+    }
+    async function lookup(uploadId) {
+        const hot = memory.get(uploadId);
+        if (hot !== undefined)
+            return hot;
+        const rStored = await exception2Result(() => withinMs(opts.bytes.get(uploadId), lookupMs));
+        if (rStored.isErr()) {
+            warn("picture-bytes-get-failed", { uploadId, message: rStored.Err().message });
+            return undefined;
+        }
+        const stored = rStored.Ok();
+        if (stored === TIMED_OUT) {
+            warn("picture-bytes-get-slow", { uploadId, ms: lookupMs });
+            return undefined;
+        }
+        if (stored === undefined)
+            return undefined;
+        if (isRasterImageType(stored.type) === false)
+            return undefined;
+        memory.set(uploadId, stored);
+        return stored;
+    }
+    async function download(uploadId, url) {
+        if (url.startsWith(filesPrefix) === false)
+            return { outcome: "unavailable" };
+        const rRes = await exception2Result(() => fetcher(url, { credentials: "include" }));
+        if (rRes.isErr()) {
+            warn("picture-bytes-fetch-failed", { uploadId, message: rRes.Err().message });
+            return { outcome: "unavailable" };
+        }
+        const res = rRes.Ok();
+        if (res.ok === false) {
+            warn("picture-bytes-fetch-status", { uploadId, status: res.status });
+            return { outcome: "unavailable" };
+        }
+        const servedUpload = res.headers.get(FILES_UPLOAD_ID_HEADER);
+        if (servedUpload !== null && servedUpload !== uploadId) {
+            warn("picture-bytes-other-upload", { uploadId, servedUpload });
+            dropUnconfirmed(uploadId);
+            return { outcome: "other-upload" };
+        }
+        const rBlob = await exception2Result(() => res.blob());
+        if (rBlob.isErr()) {
+            warn("picture-bytes-body-failed", { uploadId, message: rBlob.Err().message });
+            return { outcome: "unavailable" };
+        }
+        const blob = rBlob.Ok();
+        const verdict = judge(uploadId, res, blob.type);
+        if (verdict.keepable === false) {
+            notKeepable.set(uploadId, verdict.named);
+            dropUnconfirmed(uploadId);
+            return { outcome: "not-kept", blob, confirmed: verdict.named };
+        }
+        unconfirmed.delete(uploadId);
+        await keep(uploadId, blob);
+        return { outcome: "kept", blob };
+    }
+    function fill(ref) {
+        const pending = inFlight.get(ref.uploadId);
+        if (pending !== undefined)
+            return pending;
+        const known = notKeepable.get(ref.uploadId);
+        if (known !== undefined)
+            return Promise.resolve({ outcome: "known-not-keepable", confirmed: known });
+        const p = download(ref.uploadId, ref.url).finally(() => inFlight.delete(ref.uploadId));
+        inFlight.set(ref.uploadId, p);
+        return p;
+    }
+    return {
+        peek: (uploadId) => memory.get(uploadId),
+        lookup,
+        fill,
+        seed,
+        isUnconfirmed: (uploadId) => unconfirmed.has(uploadId),
+        confirm,
+    };
+}
+function defaultFetch(input, init) {
+    return fetch(input, init);
+}
+export const PICTURE_BYTES_DB = "vibes-picture-bytes";
+const BYTES = "bytes";
+const META = "meta";
+const DEFAULT_MAX_BYTES = 160 * 1024 * 1024;
+const bytesRow = type({ buf: type.instanceOf(ArrayBuffer), type: "string" });
+const metaRow = type({ uploadId: "string", size: "number", lastUsed: "number" });
+const metaRowId = type({ uploadId: "string" });
+function req(r) {
+    return new Promise((resolve, reject) => {
+        r.onsuccess = () => resolve(r.result);
+        r.onerror = () => reject(r.error ?? new Error("indexeddb request failed"));
+    });
+}
+function done(tx) {
+    return new Promise((resolve, reject) => {
+        tx.oncomplete = () => resolve();
+        tx.onerror = () => reject(tx.error ?? new Error("indexeddb transaction failed"));
+        tx.onabort = () => reject(tx.error ?? new Error("indexeddb transaction aborted"));
+    });
+}
+export function idbPictureBytes(opts = {}) {
+    const factory = opts.indexedDB ?? (typeof indexedDB === "undefined" ? undefined : indexedDB);
+    const maxBytes = opts.maxBytes ?? DEFAULT_MAX_BYTES;
+    const now = opts.now ?? Date.now;
+    let dbp;
+    function open() {
+        if (factory === undefined)
+            return Promise.reject(new Error("indexedDB unavailable"));
+        if (dbp === undefined) {
+            const r = factory.open(opts.dbName ?? PICTURE_BYTES_DB, 1);
+            r.onupgradeneeded = () => {
+                const db = r.result;
+                if (db.objectStoreNames.contains(BYTES) === false)
+                    db.createObjectStore(BYTES);
+                if (db.objectStoreNames.contains(META) === false)
+                    db.createObjectStore(META, { keyPath: "uploadId" });
+            };
+            const opening = req(r);
+            dbp = opening;
+            opening.then(undefined, forgetFailedOpen);
+        }
+        return dbp;
+    }
+    function forgetFailedOpen() {
+        dbp = undefined;
+    }
+    async function get(uploadId) {
+        const db = await open();
+        const tx = db.transaction([BYTES, META], "readwrite");
+        const row = bytesRow(await req(tx.objectStore(BYTES).get(uploadId)));
+        if (row instanceof type.errors) {
+            await done(tx);
+            return undefined;
+        }
+        tx.objectStore(META).put({ uploadId, size: row.buf.byteLength, lastUsed: now() });
+        await done(tx);
+        return new Blob([row.buf], { type: row.type });
+    }
+    async function put(uploadId, blob) {
+        if (blob.size > maxBytes)
+            return;
+        const buf = await blob.arrayBuffer();
+        const db = await open();
+        const tx = db.transaction([BYTES, META], "readwrite");
+        const bytes = tx.objectStore(BYTES);
+        const meta = tx.objectStore(META);
+        bytes.put({ buf, type: blob.type }, uploadId);
+        meta.put({ uploadId, size: buf.byteLength, lastUsed: now() });
+        const rows = [];
+        for (const raw of await req(meta.getAll())) {
+            const m = metaRow(raw);
+            if (m instanceof type.errors) {
+                const id = metaRowId(raw);
+                if (id instanceof type.errors === false) {
+                    bytes.delete(id.uploadId);
+                    meta.delete(id.uploadId);
+                }
+                continue;
+            }
+            rows.push(m);
+        }
+        rows.sort((a, b) => a.lastUsed - b.lastUsed);
+        let total = rows.reduce((n, m) => n + m.size, 0);
+        for (const m of rows) {
+            if (total <= maxBytes)
+                break;
+            if (m.uploadId === uploadId)
+                continue;
+            bytes.delete(m.uploadId);
+            meta.delete(m.uploadId);
+            total -= m.size;
+        }
+        await done(tx);
+    }
+    return { get, put };
+}
+//# sourceMappingURL=picture-bytes.js.map

package/hooks/img-gen/picture-bytes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"picture-bytes.js","sourceRoot":"","sources":["../../../jsr/hooks/img-gen/picture-bytes.ts"],"names":[],"mappings":"AA4BA,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,wBAAwB,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAChH,OAAO,EAAE,IAAI,EAAE,MAAM,yBAAyB,CAAC;AAO/C,SAAS,WAAW,CAAC,KAAa,EAAE,MAA+B;IACjE,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;AAClC,CAAC;AAKD,MAAM,kBAAkB,GAAsB,CAAC,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,CAAC,CAAC;AAChI,MAAM,UAAU,iBAAiB,CAAC,IAAY;IAC5C,OAAO,kBAAkB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;AAC9E,CAAC;AAoED,MAAM,oBAAoB,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AAC9C,MAAM,uBAAuB,GAAG,IAAI,CAAC;AAGrC,MAAM,UAAU;IACG,GAAG,CAAS;IACZ,OAAO,GAAG,IAAI,GAAG,EAAgB,CAAC;IAC3C,KAAK,GAAG,CAAC,CAAC;IAClB,YAAY,GAAW;QACrB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IACD,GAAG,CAAC,EAAU;QACZ,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QACtC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACxB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACxB,OAAO,CAAC,CAAC;IACX,CAAC;IACD,MAAM,CAAC,EAAU;QACf,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC/B,IAAI,CAAC,KAAK,SAAS;YAAE,OAAO;QAC5B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACxB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC;IACvB,CAAC;IACD,GAAG,CAAC,EAAU,EAAE,CAAO;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACnC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC;YACzB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG;YAAE,OAAO;QAC9B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACxB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC;QACrB,KAAK,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACxC,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG;gBAAE,MAAM;YAClC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC3B,IAAI,CAAC,KAAK,IAAI,GAAG,CAAC,IAAI,CAAC;QACzB,CAAC;IACH,CAAC;CACF;AAED,MAAM,SAAS,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;AACtC,SAAS,QAAQ,CAAI,CAAa,EAAE,EAAU;IAC5C,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,IAAI,CACJ,CAAC,CAAC,EAAE,EAAE;YACJ,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO,CAAC,CAAC,CAAC,CAAC;QACb,CAAC,EACD,CAAC,CAAU,EAAE,EAAE;YACb,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,CAAC,CAAC,CAAC;QACZ,CAAC,CACF,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,IAA4B;IAC7D,MAAM,OAAO,GAAmB,IAAI,CAAC,OAAO,IAAI,YAAY,CAAC;IAC7D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,WAAW,CAAC;IACtC,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,WAAW,IAAI,oBAAoB,CAAC,CAAC;IACxE,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,IAAI,uBAAuB,CAAC;IAC/D,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,WAAW,UAAU,CAAC;IAClD,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAgC,CAAC;IAGzD,MAAM,WAAW,GAAG,IAAI,GAAG,EAAmB,CAAC;IAE/C,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IACtC,MAAM,eAAe,GAAG,IAAI,GAAG,EAAmC,CAAC;IAInE,KAAK,UAAU,IAAI,CAAC,QAAgB,EAAE,IAAU;QAC9C,IAAI,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,KAAK,EAAE,CAAC;YAC3C,IAAI,CAAC,0BAA0B,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChE,OAAO;QACT,CAAC;QACD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;QAC1E,IAAI,IAAI,CAAC,KAAK,EAAE;YAAE,IAAI,CAAC,0BAA0B,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;IAChG,CAAC;IAED,KAAK,UAAU,IAAI,CAAC,QAAgB,EAAE,IAAU;QAC9C,IAAI,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,KAAK,EAAE,CAAC;YAC3C,IAAI,CAAC,0BAA0B,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChE,OAAO;QACT,CAAC;QACD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC3B,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC5B,CAAC;IAID,SAAS,eAAe,CAAC,QAAgB;QACvC,IAAI,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC;YAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC5D,CAAC;IAGD,SAAS,KAAK,CACZ,QAAgB,EAChB,GAAa,EACb,IAAwB;QAExB,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACxD,MAAM,KAAK,GAAG,YAAY,KAAK,QAAQ,CAAC;QACxC,MAAM,QAAQ,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACrE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,IAAI,KAAK,KAAK,IAAI,IAAI,qBAAqB,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,QAAQ,EAAE,CAAC;IAC3G,CAAC;IAED,KAAK,UAAU,WAAW,CAAC,QAAgB,EAAE,GAAW;QACtD,IAAI,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,KAAK;YAAE,OAAO,MAAM,CAAC;QACvD,IAAI,GAAG,CAAC,UAAU,CAAC,WAAW,CAAC,KAAK,KAAK;YAAE,OAAO,SAAS,CAAC;QAC5D,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QACpG,IAAI,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;YACjB,IAAI,CAAC,8BAA8B,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YAChF,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAEtB,IAAI,GAAG,CAAC,EAAE,KAAK,KAAK;YAAE,OAAO,SAAS,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QACnD,IAAI,OAAO,CAAC,QAAQ,KAAK,KAAK,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACvD,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC1B,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7B,MAAM,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC7B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,SAAS,OAAO,CAAC,GAAwD;QACvE,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,OAAO,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC;QAC1C,MAAM,CAAC,GAAG,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACjG,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACrC,OAAO,CAAC,CAAC;IACX,CAAC;IAED,KAAK,UAAU,MAAM,CAAC,QAAgB;QACpC,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAI,GAAG,KAAK,SAAS;YAAE,OAAO,GAAG,CAAC;QAClC,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC;QAC3F,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;YACpB,IAAI,CAAC,0BAA0B,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/E,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,MAAM,GAAG,OAAO,CAAC,EAAE,EAAE,CAAC;QAC5B,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,IAAI,CAAC,wBAAwB,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3D,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAE3C,IAAI,iBAAiB,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,KAAK;YAAE,OAAO,SAAS,CAAC;QAC/D,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC7B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,UAAU,QAAQ,CAAC,QAAgB,EAAE,GAAW;QACnD,IAAI,GAAG,CAAC,UAAU,CAAC,WAAW,CAAC,KAAK,KAAK;YAAE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QAI7E,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QACpF,IAAI,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;YACjB,IAAI,CAAC,4BAA4B,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9E,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QACpC,CAAC;QACD,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACtB,IAAI,GAAG,CAAC,EAAE,KAAK,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,4BAA4B,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;YACrE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QACpC,CAAC;QACD,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;QAC7D,IAAI,YAAY,KAAK,IAAI,IAAI,YAAY,KAAK,QAAQ,EAAE,CAAC;YACvD,IAAI,CAAC,4BAA4B,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC;YAC/D,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;QACrC,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;QACvD,IAAI,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;YAClB,IAAI,CAAC,2BAA2B,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9E,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;QACpC,CAAC;QACD,MAAM,IAAI,GAAG,KAAK,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,OAAO,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;YAC/B,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YACzC,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC1B,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;QACjE,CAAC;QACD,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC7B,MAAM,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC3B,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACnC,CAAC;IAED,SAAS,IAAI,CAAC,GAAwD;QACpE,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3C,IAAI,OAAO,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC;QAC1C,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,oBAAoB,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QACrG,MAAM,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QACvF,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAC9B,OAAO,CAAC,CAAC;IACX,CAAC;IAED,OAAO;QACL,IAAI,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;QACxC,MAAM;QACN,IAAI;QACJ,IAAI;QACJ,aAAa,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC;QACtD,OAAO;KACR,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,IAAkB;IACrD,OAAO,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;AAC5B,CAAC;AAYD,MAAM,CAAC,MAAM,gBAAgB,GAAG,qBAAqB,CAAC;AACtD,MAAM,KAAK,GAAG,OAAO,CAAC;AACtB,MAAM,IAAI,GAAG,MAAM,CAAC;AACpB,MAAM,iBAAiB,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;AAI5C,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;AAC7E,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;AAGjF,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;AAE/C,SAAS,GAAG,CAAI,CAAgB;IAC9B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,CAAC,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC;AACD,SAAS,IAAI,CAAC,EAAkB;IAC9B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,EAAE,CAAC,UAAU,GAAG,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;QAChC,EAAE,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC,CAAC;QACjF,EAAE,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;AACL,CAAC;AAUD,MAAM,UAAU,eAAe,CAAC,IAAI,GAAwB,EAAE;IAG5D,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,IAAI,CAAC,OAAO,SAAS,KAAK,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC7F,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,iBAAiB,CAAC;IACpD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;IACjC,IAAI,GAAqC,CAAC;IAE1C,SAAS,IAAI;QACX,IAAI,OAAO,KAAK,SAAS;YAAE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;QACrF,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACtB,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,gBAAgB,EAAE,CAAC,CAAC,CAAC;YAC3D,CAAC,CAAC,eAAe,GAAG,GAAG,EAAE;gBACvB,MAAM,EAAE,GAAG,CAAC,CAAC,MAAM,CAAC;gBACpB,IAAI,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,KAAK;oBAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBAC/E,IAAI,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,KAAK;oBAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;YACxG,CAAC,CAAC;YACF,MAAM,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;YACvB,GAAG,GAAG,OAAO,CAAC;YAGd,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IACD,SAAS,gBAAgB;QACvB,GAAG,GAAG,SAAS,CAAC;IAClB,CAAC;IAED,KAAK,UAAU,GAAG,CAAC,QAAgB;QACjC,MAAM,EAAE,GAAG,MAAM,IAAI,EAAE,CAAC;QACxB,MAAM,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,WAAW,CAAC,CAAC;QACtD,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QACrE,IAAI,GAAG,YAAY,IAAI,CAAC,MAAM,EAAE,CAAC;YAC/B,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;YACf,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,EAAE,GAAG,EAAE,EAAoB,CAAC,CAAC;QACpG,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,OAAO,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,UAAU,GAAG,CAAC,QAAgB,EAAE,IAAU;QAC7C,IAAI,IAAI,CAAC,IAAI,GAAG,QAAQ;YAAE,OAAO;QACjC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,EAAE,GAAG,MAAM,IAAI,EAAE,CAAC;QACxB,MAAM,EAAE,GAAG,EAAE,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,WAAW,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,IAAI,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAClC,KAAK,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC9C,IAAI,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,CAAC,UAAU,EAAE,QAAQ,EAAE,GAAG,EAAE,EAAoB,CAAC,CAAC;QAGhF,MAAM,IAAI,GAAc,EAAE,CAAC;QAC3B,KAAK,MAAM,GAAG,IAAI,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;YAC3C,MAAM,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;YACvB,IAAI,CAAC,YAAY,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM,EAAE,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;gBAC1B,IAAI,EAAE,YAAY,IAAI,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;oBACxC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;oBAC1B,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;gBAC3B,CAAC;gBACD,SAAS;YACX,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACf,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACjD,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrB,IAAI,KAAK,IAAI,QAAQ;gBAAE,MAAM;YAC7B,IAAI,CAAC,CAAC,QAAQ,KAAK,QAAQ;gBAAE,SAAS;YACtC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YACzB,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YACxB,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC;QAClB,CAAC;QACD,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;IACjB,CAAC;IAED,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;AACtB,CAAC"}

package/hooks/img-gen/use-img-gen.d.ts

@@ -3,6 +3,8 @@ export type UseFireproofHook = typeof useFireproof;
 export type UseVibeContextHook = typeof useVibeContext;
 import type { ImgGenRequestOptions, ImgGenResult, UseImgGenOptions, UseImgGenResult } from "@vibes.diy/vibe-types";
 import { Result } from "@adviser/cement";
+import { type LocalFileUrl } from "./local-pictures.js";
+import type { PictureStore } from "./picture-bytes.js";
 export declare function __setImgGenProbeDeadlineMsForTests(ms: number | undefined): void;
 export declare function __setImgGenCommitReprobeDelaysMsForTests(ms: readonly number[] | undefined): void;
 interface InjectedDeps {
@@ -10,7 +12,13 @@ interface InjectedDeps {
     imgGen?: (prompt: string, opts?: ImgGenRequestOptions) => Promise<Result<ImgGenResult>>;
     confirmedMediaProbe?: ConfirmedMediaProbe | null;
     useFireproof?: UseFireproofHook;
+    localFileUrl?: LocalFileUrl | null;
+    pictures?: PictureStore | null;
     useVibeContext?: UseVibeContextHook;
 }
-export declare function useImgGen(opts: Partial<UseImgGenOptions> & InjectedDeps): UseImgGenResult;
+export interface UseImgGenDisplayResult extends UseImgGenResult {
+    readonly mintedUrls: ReadonlySet<string>;
+    readonly reportUnavailable: (uploadId: string) => void;
+}
+export declare function useImgGen(opts: Partial<UseImgGenOptions> & InjectedDeps): UseImgGenDisplayResult;
 export {};

package/hooks/img-gen/use-img-gen.js

@@ -1,7 +1,9 @@
-import { useState, useEffect, useMemo, useRef } from "react";
+import { useState, useEffect, useMemo, useRef, useCallback } from "react";
 import { useFireproof, imgGen as defaultImgGen, confirmedMediaProbe as defaultConfirmedMediaProbe, useVibeContext, getRegisteredVibeApi, vibe, decorateFiles, IMG_GEN_SOURCE_UNUSABLE, } from "@vibes.diy/vibe-runtime";
 import { IMG_GEN_CANCELLED, IMG_GEN_NEEDS_TAP, IMG_GEN_UNREACHABLE, IMG_GEN_UNREACHABLE_MESSAGE, buildMediaPointerDoc, buildMediaVersionDoc, mediaDbName, mediaPointerDocId, mediaVersionDocId, } from "@vibes.diy/vibe-types";
 import { exception2Result, Result } from "@adviser/cement";
+import { type } from "arktype";
+import { registeredLocalFileUrl, registeredPictureStore } from "./local-pictures.js";
 const PROMPT_KEY = "p1";
 function isValidVersionDoc(d) {
     if (d.type !== "version")
@@ -40,6 +42,41 @@ function withLocalPreviews(doc, previews) {
     }
     return changed ? { ...doc, _files: files } : doc;
 }
+function withLocalUrls(doc, localUrls) {
+    const minted = new Set();
+    if (doc === null || localUrls.size === 0 || doc._files === undefined)
+        return { doc, minted };
+    const files = {};
+    for (const [key, f] of Object.entries(doc._files)) {
+        const url = f.url === undefined && f.uploadId !== undefined ? localUrls.get(localUrlKey(key, f.uploadId)) : undefined;
+        if (url !== undefined) {
+            files[key] = { ...f, url };
+            minted.add(url);
+        }
+        else {
+            files[key] = f;
+        }
+    }
+    return minted.size > 0 ? { doc: decorateFiles({ ...doc, _files: files }), minted } : { doc, minted };
+}
+function localUrlKey(fileKey, uploadId) {
+    return `${fileKey}\u0000${uploadId}`;
+}
+const localRow = type({ _id: "string", _files: "Record<string, unknown>" });
+const localFileEntry = type({ uploadId: "string > 0", "url?": "unknown" });
+function noteLocalUrls(args) {
+    for (const raw of args.docs) {
+        const d = localRow(raw);
+        if (d instanceof type.errors)
+            continue;
+        for (const [key, rawEntry] of Object.entries(d._files)) {
+            const f = localFileEntry(rawEntry);
+            if (f instanceof type.errors || f.url !== undefined)
+                continue;
+            args.into.set(localUrlKey(key, f.uploadId), args.mint({ dbName: args.dbName, docId: d._id, key, uploadId: f.uploadId }));
+        }
+    }
+}
 function synthesizeMediaDocument(hostDocId, versionDocs, pointer) {
     const versions = versionDocs.map((vd) => {
         const fileKey = Object.keys(vd._files)[0];
@@ -182,7 +219,9 @@ async function readBackFromServer(args) {
 const IMG_GEN_MAX_REKEEPS = 3;
 const IMG_GEN_UNKEPT_MESSAGE = "This picture wasn't saved — another device added a version at the same moment. Please try again.";
 export function useImgGen(opts) {
-    const { prompt, _id, database = "ImgGen", skip = false, generationId, inputImage, model, imgGen = defaultImgGen, temporaryDraft = getRegisteredVibeApi()?.temporaryData === true, confirmedMediaProbe = defaultConfirmedMediaProbe, useFireproof: useFireproofImpl = useFireproof, useVibeContext: useVibeContextImpl = useVibeContext, tapGenerationId, } = opts;
+    const { prompt, _id, database = "ImgGen", skip = false, generationId, inputImage, model, imgGen = defaultImgGen, temporaryDraft = getRegisteredVibeApi()?.temporaryData === true, confirmedMediaProbe = defaultConfirmedMediaProbe, useFireproof: useFireproofImpl = useFireproof, useVibeContext: useVibeContextImpl = useVibeContext, tapGenerationId, localFileUrl: localFileUrlOpt = registeredLocalFileUrl(), pictures: picturesOpt = registeredPictureStore(), } = opts;
+    const mintLocal = temporaryDraft || localFileUrlOpt === null ? undefined : localFileUrlOpt;
+    const pictures = picturesOpt ?? undefined;
     const [loading, setLoading] = useState(false);
     const [progress, setProgress] = useState(0);
     const [error, setError] = useState(null);
@@ -222,6 +261,9 @@ export function useImgGen(opts) {
         unmountSignalRef.current = ac.signal;
         return () => ac.abort();
     }, []);
+    const localUrlsRef = useRef(new Map());
+    const retryDisplayReadBackRef = useRef(null);
+    const seededRef = useRef(new Set());
     const previewUrlsRef = useRef(new Map());
     useEffect(() => {
         const previews = previewUrlsRef.current;
@@ -282,6 +324,9 @@ export function useImgGen(opts) {
         const hostDocId = _id;
         async function loadMediaDocs() {
             const res = await mediaDb.query("hostDocId", { key: hostDocId });
+            if (mintLocal !== undefined) {
+                noteLocalUrls({ docs: res.docs, dbName: mediaDbName(dbName), mint: mintLocal, into: localUrlsRef.current });
+            }
             return res.docs;
         }
         let readStartedPending = false;
@@ -302,9 +347,14 @@ export function useImgGen(opts) {
                     tapSpentRef.current = generationId;
                 return;
             }
-            await decide(rMediaDocs.Ok(), rLegacy.isOk() ? rLegacy.Ok() : null, false);
+            const localLegacy = rLegacy.isOk() ? rLegacy.Ok() : null;
+            if (mintLocal !== undefined && localLegacy !== null) {
+                noteLocalUrls({ docs: [localLegacy], dbName, mint: mintLocal, into: localUrlsRef.current });
+            }
+            await decide(rMediaDocs.Ok(), localLegacy, false);
         }
         async function readBackForDisplay(shown, fileKey, localDocs, legacyDoc) {
+            const ourVersionId = sortVersionDocs(localDocs).find((vd) => vd._files[fileKey]?.uploadId === shown._files?.[fileKey]?.uploadId)?._id;
             const uploadId = shown._files?.[fileKey]?.uploadId;
             if (confirmedMediaProbe === undefined || confirmedMediaProbe === null)
                 return;
@@ -333,8 +383,17 @@ export function useImgGen(opts) {
                 logEvent: "imggen-display-probe-failed",
                 examine: (probedDocs, probedLegacy) => {
                     const probedVersions = sortVersionDocs(probedDocs);
-                    if (holdsUpload(probedVersions, uploadId, true) === false)
+                    if (holdsUpload(probedVersions, uploadId, true) === false) {
+                        const taken = ourVersionId !== undefined &&
+                            probedVersions.some((vd) => vd._id === ourVersionId && holdsUpload([vd], uploadId, false) === false);
+                        if (taken) {
+                            vibe.log("warn", "imggen-version-unkept", { docId: hostDocId, dbName, uploadId, lane: "display" });
+                            setError(new Error(IMG_GEN_UNKEPT_MESSAGE));
+                            setPending(false);
+                            return { done: true, value: true };
+                        }
                         return { done: false };
+                    }
                     const composed = composeDocument({
                         hostDocId,
                         versionDocs: probedVersions,
@@ -389,12 +448,24 @@ export function useImgGen(opts) {
                 currentVer?.id &&
                 existingDoc._files?.[currentVer.id]) {
                 const urlPending = existingDoc._files[currentVer.id]?.url === undefined;
-                setPending(urlPending);
+                const currentUpload = existingDoc._files[currentVer.id]?.uploadId;
+                const shownLocally = urlPending && currentUpload !== undefined && localUrlsRef.current.has(localUrlKey(currentVer.id, currentUpload));
+                setPending(urlPending && shownLocally === false);
                 setDocument(existingDoc);
                 setNeedsTap(false);
                 if (urlPending) {
                     currentGenRef.current = null;
-                    await readBackForDisplay(existingDoc, currentVer.id, mediaDocs, legacyDoc);
+                    if (shownLocally === false)
+                        await readBackForDisplay(existingDoc, currentVer.id, mediaDocs, legacyDoc);
+                    else if (currentUpload !== undefined) {
+                        const fileKey = currentVer.id;
+                        const restart = () => {
+                            displayReadBackRef.current = null;
+                            void readBackForDisplay(existingDoc, fileKey, mediaDocs, legacyDoc);
+                        };
+                        retryDisplayReadBackRef.current = { uploadId: currentUpload, run: restart };
+                        void readBackForDisplay(existingDoc, fileKey, mediaDocs, legacyDoc);
+                    }
                 }
                 else {
                     const committed = committedGenRef.current;
@@ -513,7 +584,7 @@ export function useImgGen(opts) {
                     if (earlyPreviewRef.current !== null)
                         return;
                     const url = URL.createObjectURL(blob);
-                    earlyPreviewRef.current = { genKey, url };
+                    earlyPreviewRef.current = { genKey, url, blob };
                     setPreviewUrl(url);
                 },
             }));
@@ -561,6 +632,11 @@ export function useImgGen(opts) {
             else if (file.preview instanceof Blob && previewUrlsRef.current.has(file.uploadId) === false) {
                 previewUrlsRef.current.set(file.uploadId, URL.createObjectURL(file.preview));
             }
+            const ownBytes = early?.blob ?? (file.preview instanceof Blob ? file.preview : undefined);
+            if (ownBytes !== undefined && pictures !== undefined) {
+                void pictures.seed(file.uploadId, ownBytes);
+                seededRef.current.add(file.uploadId);
+            }
             const rRun = await exception2Result(async () => {
                 setProgress(90);
                 const showCommitted = async (uploadId, clientWrite) => {
@@ -832,12 +908,35 @@ export function useImgGen(opts) {
         model,
         imgGen,
         confirmedMediaProbe,
+        mintLocal,
+        pictures,
         mediaFromCache,
         hostFromCache,
         deferTick,
         arrivalTick,
     ]);
-    const displayed = useMemo(() => withLocalPreviews(document, previewUrlsRef.current), [document]);
+    const { doc: displayed, minted: mintedUrls } = useMemo(() => withLocalUrls(withLocalPreviews(document, previewUrlsRef.current), localUrlsRef.current), [document]);
+    useEffect(() => {
+        if (pictures === undefined || seededRef.current.size === 0 || document?._files === undefined)
+            return;
+        for (const [key, f] of Object.entries(document._files)) {
+            const uploadId = f.uploadId;
+            if (uploadId === undefined || seededRef.current.has(uploadId) === false)
+                continue;
+            const url = f.url ?? localUrlsRef.current.get(localUrlKey(key, uploadId));
+            if (url === undefined)
+                continue;
+            void pictures.confirm({ uploadId, url }).then((answer) => {
+                if (answer !== "unknown")
+                    seededRef.current.delete(uploadId);
+            });
+        }
+    }, [document, pictures]);
+    const reportUnavailable = useCallback((uploadId) => {
+        const retry = retryDisplayReadBackRef.current;
+        if (retry !== null && retry.uploadId === uploadId)
+            retry.run();
+    }, []);
     const previewSettled = previewUrl !== undefined && (error !== null || Object.values(displayed?._files ?? {}).some((f) => f.url === previewUrl));
     useEffect(() => {
         if (previewSettled)
@@ -851,6 +950,8 @@ export function useImgGen(opts) {
         pending,
         needsTap,
         ...(previewUrl !== undefined ? { previewUrl } : {}),
+        mintedUrls,
+        reportUnavailable,
     };
 }
 //# sourceMappingURL=use-img-gen.js.map