@atomic-chat/core 0.5.1

package/dist/diffusion/gallery.js

@@ -0,0 +1,307 @@
+/**
+ * The gallery on disk: `<outputDir>/<jobId>-<index:02>.png` with the recipe inside the PNG, a small
+ * thumbnail beside it, and a `.flags.json` for pin and archive. Port of `gallery.rs` in
+ * `tauri-plugin-atomic-diffusion` (app commit `ec1fd3ea7`).
+ *
+ * A PNG without a valid `atomic` chunk is somebody else's: it is never listed and never deleted,
+ * even when it sits in the output folder under a name that looks like ours.
+ */
+import { copyFile, mkdir, readdir, readFile, rename, rm, stat, writeFile } from 'node:fs/promises';
+import { basename, dirname, join } from 'node:path';
+import { isWithin } from './containment.js';
+import { diffusionError, ioError } from './errors.js';
+import { AsyncMutex } from './mutex.js';
+import { decodePng, isBlankRaster, readPngHeader, thumbnailPng, UnsupportedPngError } from './png.js';
+import { parseRecipe, RECIPE_KEYWORD, spliceRecipe } from './recipe.js';
+export const FLAGS_FILE = '.flags.json';
+export const THUMB_EDGE = 256;
+const THUMB_SUFFIX = '.thumb.png';
+/** How many image headers a listing reads at once. */
+const SCAN_CONCURRENCY = 16;
+/** A listing checks for blank frames only in files this small; real outputs are far larger. */
+export const BLANK_SCAN_LIMIT = 128 * 1024;
+/**
+ * sd.cpp can finish a job after a numerical overflow and return a frame whose every pixel is pure
+ * white or pure black: not a generated image. Bytes that are not a readable PNG are
+ * `INVALID_OUTPUT`; a well-formed PNG of a kind this module does not decode is not called blank.
+ */
+export async function isBlankOutput(png) {
+    let image;
+    try {
+        image = await decodePng(png);
+    }
+    catch (error) {
+        if (error instanceof UnsupportedPngError)
+            return false;
+        throw diffusionError('INVALID_OUTPUT', 'The image engine returned an unreadable image.', error instanceof Error ? error.message : String(error));
+    }
+    return isBlankRaster(image);
+}
+/** `^[a-f0-9]{32}-\d{2}$` */
+export function isValidId(id) {
+    return /^[a-f0-9]{32}-[0-9]{2}$/.test(id);
+}
+export function makeId(jobId, index) {
+    return `${jobId}-${String(index).padStart(2, '0')}`;
+}
+export function pngPath(dir, id) {
+    return join(dir, `${id}.png`);
+}
+export function thumbPath(dir, id) {
+    return join(dir, `${id}${THUMB_SUFFIX}`);
+}
+function checkedPngPath(dir, id) {
+    if (!isValidId(id))
+        throw diffusionError('INVALID_REQUEST', 'That is not a gallery image id.', id);
+    const path = pngPath(dir, id);
+    if (!isWithin(path, dir))
+        throw diffusionError('INVALID_REQUEST', 'That image is outside the gallery folder.', id);
+    return path;
+}
+const isFile = (path) => stat(path).then((s) => s.isFile(), () => false);
+export async function readFlags(dir) {
+    const flags = new Map();
+    let raw;
+    try {
+        raw = JSON.parse(await readFile(join(dir, FLAGS_FILE), 'utf8'));
+    }
+    catch {
+        return flags;
+    }
+    if (raw === null || typeof raw !== 'object' || Array.isArray(raw))
+        return flags;
+    for (const [id, value] of Object.entries(raw)) {
+        // One malformed entry invalidated the whole file for the plugin's serde; keep that.
+        if (value === null || typeof value !== 'object' || Array.isArray(value))
+            return new Map();
+        const entry = value;
+        const pinned = entry['pinned'] ?? false;
+        const archived = entry['archived'] ?? false;
+        if (typeof pinned !== 'boolean' || typeof archived !== 'boolean')
+            return new Map();
+        flags.set(id, { pinned, archived });
+    }
+    return flags;
+}
+const IMAGE_WRITE_MESSAGES = {
+    write: 'Could not write to the images folder.',
+    finish: 'Could not finish writing the image.',
+};
+/** Write a file through a hidden temporary sibling, so a reader never sees half of it. */
+export async function writeAtomic(target, bytes, messages = IMAGE_WRITE_MESSAGES) {
+    const tmp = join(dirname(target), `.${basename(target)}.tmp`);
+    try {
+        await writeFile(tmp, bytes);
+    }
+    catch (error) {
+        await rm(tmp, { force: true }).catch(() => { });
+        throw ioError(messages.write, error);
+    }
+    try {
+        await rename(tmp, target);
+    }
+    catch (error) {
+        await rm(tmp, { force: true }).catch(() => { });
+        throw ioError(messages.finish, error);
+    }
+}
+export function writeFlags(dir, flags) {
+    const sorted = Object.fromEntries([...flags.entries()].sort(([a], [b]) => (a < b ? -1 : a > b ? 1 : 0)));
+    return writeAtomic(join(dir, FLAGS_FILE), JSON.stringify(sorted, null, 2));
+}
+/** A 256 px PNG beside the image. Fails for a PNG flavour the decoder does not read; the caller goes without. */
+export async function writeThumbnail(png, path) {
+    await writeAtomic(path, await thumbnailPng(png, THUMB_EDGE));
+}
+async function itemFromPath(dir, id, path, flags, isBlank) {
+    if (await isBlank(dir, id, path))
+        return undefined;
+    const header = await readPngHeader(path);
+    const text = header?.texts.get(RECIPE_KEYWORD);
+    const recipe = text === undefined ? undefined : parseRecipe(text);
+    if (!header || !recipe)
+        return undefined;
+    const meta = await stat(path).catch(() => undefined);
+    if (!meta)
+        return undefined;
+    const thumb = thumbPath(dir, id);
+    const flag = flags.get(id);
+    return {
+        id,
+        path,
+        thumbnailPath: (await isFile(thumb)) ? thumb : null,
+        width: header.width,
+        height: header.height,
+        sizeBytes: meta.size,
+        createdAtMs: recipe.createdAtMs,
+        pinned: flag?.pinned ?? false,
+        archived: flag?.archived ?? false,
+        recipe,
+    };
+}
+/** Every owned image, newest first. */
+async function scan(dir, flags, isBlank) {
+    const names = await readdir(dir).catch(() => []);
+    const ids = names
+        .filter((name) => name.endsWith('.png'))
+        .map((name) => name.slice(0, -'.png'.length))
+        .filter(isValidId);
+    const items = [];
+    for (let at = 0; at < ids.length; at += SCAN_CONCURRENCY) {
+        const batch = ids.slice(at, at + SCAN_CONCURRENCY);
+        const found = await Promise.all(batch.map((id) => itemFromPath(dir, id, pngPath(dir, id), flags, isBlank)));
+        for (const item of found)
+            if (item)
+                items.push(item);
+    }
+    return items.sort((a, b) => b.createdAtMs - a.createdAtMs || (a.id < b.id ? 1 : a.id > b.id ? -1 : 0));
+}
+/**
+ * The gallery's operations. The flags file is read and rewritten whole, so everything that touches
+ * it goes through one lock; the images themselves have unique names and need none.
+ */
+export class Gallery {
+    log;
+    flagsLock = new AsyncMutex();
+    /** The last verdict per small file, so a listing decodes each thumbnail once rather than every time. */
+    blankFiles = new Map();
+    isBlank = (dir, id, path) => this.isBlankItem(dir, id, path);
+    constructor(log = () => { }) {
+        this.log = log;
+    }
+    /**
+     * A blank frame an older engine build saved is hidden from the listing and from lookups: judged on
+     * the thumbnail (the PNG when there is none) when that file is small enough to be one. A file
+     * that cannot be read or decoded is not blank.
+     */
+    async isBlankItem(dir, id, path) {
+        const thumb = thumbPath(dir, id);
+        const file = (await isFile(thumb)) ? thumb : path;
+        const meta = await stat(file).catch(() => undefined);
+        if (!meta || meta.size > BLANK_SCAN_LIMIT)
+            return false;
+        const known = this.blankFiles.get(file);
+        if (known && known.size === meta.size && known.mtimeMs === meta.mtimeMs)
+            return known.blank;
+        const blank = await readFile(file)
+            .then(isBlankOutput, () => false)
+            .catch(() => false);
+        this.blankFiles.set(file, { size: meta.size, mtimeMs: meta.mtimeMs, blank });
+        if (blank)
+            this.log('warn', `hiding blank gallery output ${id}`);
+        return blank;
+    }
+    /**
+     * Splice the recipe in, write `<id>.png`, write the thumbnail, and return the item with the final
+     * bytes (the OpenAI facade answers with them).
+     */
+    async save(dir, recipe, png) {
+        await mkdir(dir, { recursive: true }).catch((error) => {
+            throw ioError('Could not create the images folder.', error);
+        });
+        const id = makeId(recipe.jobId, recipe.index);
+        const path = checkedPngPath(dir, id);
+        const bytes = spliceRecipe(png, recipe);
+        await writeAtomic(path, bytes);
+        const thumb = thumbPath(dir, id);
+        const thumbnailPath = await writeThumbnail(bytes, thumb).then(() => thumb, (error) => {
+            this.log('warn', `thumbnail for ${id} failed: ${error instanceof Error ? error.message : String(error)}`);
+            return null;
+        });
+        const header = await readPngHeader(path);
+        if (!header)
+            throw diffusionError('INTERNAL', 'The saved PNG could not be read back.');
+        const flag = (await this.flagsLock.run(() => readFlags(dir))).get(id);
+        return {
+            item: {
+                id,
+                path,
+                thumbnailPath,
+                width: header.width,
+                height: header.height,
+                sizeBytes: bytes.length,
+                createdAtMs: recipe.createdAtMs,
+                pinned: flag?.pinned ?? false,
+                archived: flag?.archived ?? false,
+                recipe,
+            },
+            bytes,
+        };
+    }
+    async get(dir, id) {
+        const path = checkedPngPath(dir, id);
+        if (!(await isFile(path)))
+            return null;
+        const flags = await this.flagsLock.run(() => readFlags(dir));
+        return (await itemFromPath(dir, id, path, flags, this.isBlank)) ?? null;
+    }
+    async list(dir, options) {
+        const flags = await this.flagsLock.run(() => readFlags(dir));
+        const all = (await scan(dir, flags, this.isBlank)).filter((item) => options.includeArchived === true || !item.archived);
+        const items = all.slice(options.offset, options.offset + Math.max(options.limit, 1));
+        return { items, hasMore: options.offset + items.length < all.length, total: all.length };
+    }
+    /** Remove each PNG, its thumbnail and its flags entry. Foreign PNGs are left alone. */
+    delete(dir, ids) {
+        return this.flagsLock.run(async () => {
+            const flags = await readFlags(dir);
+            let changed = false;
+            for (const id of ids) {
+                const path = checkedPngPath(dir, id);
+                if (await isFile(path)) {
+                    const text = (await readPngHeader(path))?.texts.get(RECIPE_KEYWORD);
+                    if (text === undefined || !parseRecipe(text)) {
+                        this.log('warn', `refusing to delete foreign image ${id}`);
+                        continue;
+                    }
+                    await rm(path).catch((error) => {
+                        throw ioError('Could not delete the image.', error);
+                    });
+                }
+                await rm(thumbPath(dir, id), { force: true }).catch(() => { });
+                this.blankFiles.delete(path);
+                this.blankFiles.delete(thumbPath(dir, id));
+                if (flags.delete(id))
+                    changed = true;
+            }
+            if (changed)
+                await writeFlags(dir, flags);
+        });
+    }
+    setFlags(dir, id, update) {
+        return this.flagsLock.run(async () => {
+            const path = checkedPngPath(dir, id);
+            if (!(await isFile(path)))
+                throw diffusionError('JOB_NOT_FOUND', 'That image is no longer in the gallery.');
+            const flags = await readFlags(dir);
+            const entry = { ...(flags.get(id) ?? { pinned: false, archived: false }) };
+            if (update.pinned !== undefined)
+                entry.pinned = update.pinned;
+            if (update.archived !== undefined)
+                entry.archived = update.archived;
+            // An entry with nothing set is not worth a line in the file.
+            if (entry.pinned || entry.archived)
+                flags.set(id, entry);
+            else
+                flags.delete(id);
+            await writeFlags(dir, flags);
+            const item = await itemFromPath(dir, id, path, flags, this.isBlank);
+            if (!item)
+                throw diffusionError('JOB_NOT_FOUND', 'That image is not an Atomic Chat gallery image.');
+            return item;
+        });
+    }
+    /** Byte-for-byte copy, keeping the embedded recipe. */
+    async export(dir, id, target) {
+        const path = checkedPngPath(dir, id);
+        if (!(await isFile(path)))
+            throw diffusionError('JOB_NOT_FOUND', 'That image is no longer in the gallery.');
+        await mkdir(dirname(target), { recursive: true }).catch((error) => {
+            throw ioError('Could not create the destination folder.', error);
+        });
+        await copyFile(path, target).catch((error) => {
+            throw ioError('Could not export the image.', error);
+        });
+    }
+}
+//# sourceMappingURL=gallery.js.map

package/dist/diffusion/gallery.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gallery.js","sourceRoot":"","sources":["../../src/diffusion/gallery.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAClG,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAQnD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAA;AAC3C,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,aAAa,CAAA;AACrD,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AACvC,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,aAAa,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAA;AAErG,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,aAAa,CAAA;AAEvE,MAAM,CAAC,MAAM,UAAU,GAAG,aAAa,CAAA;AACvC,MAAM,CAAC,MAAM,UAAU,GAAG,GAAG,CAAA;AAC7B,MAAM,YAAY,GAAG,YAAY,CAAA;AACjC,sDAAsD;AACtD,MAAM,gBAAgB,GAAG,EAAE,CAAA;AAC3B,+FAA+F;AAC/F,MAAM,CAAC,MAAM,gBAAgB,GAAG,GAAG,GAAG,IAAI,CAAA;AAE1C;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,GAAW;IAC7C,IAAI,KAAkB,CAAA;IACtB,IAAI,CAAC;QACH,KAAK,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,CAAA;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,mBAAmB;YAAE,OAAO,KAAK,CAAA;QACtD,MAAM,cAAc,CAClB,gBAAgB,EAChB,gDAAgD,EAChD,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CACvD,CAAA;IACH,CAAC;IACD,OAAO,aAAa,CAAC,KAAK,CAAC,CAAA;AAC7B,CAAC;AAWD,6BAA6B;AAC7B,MAAM,UAAU,SAAS,CAAC,EAAU;IAClC,OAAO,yBAAyB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;AAC3C,CAAC;AAED,MAAM,UAAU,MAAM,CAAC,KAAa,EAAE,KAAa;IACjD,OAAO,GAAG,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAA;AACrD,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,GAAW,EAAE,EAAU;IAC7C,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAA;AAC/B,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,GAAW,EAAE,EAAU;IAC/C,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,YAAY,EAAE,CAAC,CAAA;AAC1C,CAAC;AAED,SAAS,cAAc,CAAC,GAAW,EAAE,EAAU;IAC7C,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAAE,MAAM,cAAc,CAAC,iBAAiB,EAAE,iCAAiC,EAAE,EAAE,CAAC,CAAA;IAClG,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;IAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC;QACtB,MAAM,cAAc,CAAC,iBAAiB,EAAE,2CAA2C,EAAE,EAAE,CAAC,CAAA;IAC1F,OAAO,IAAI,CAAA;AACb,CAAC;AAED,MAAM,MAAM,GAAG,CAAC,IAAY,EAAoB,EAAE,CAChD,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CACb,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,EACjB,GAAG,EAAE,CAAC,KAAK,CACZ,CAAA;AAEH,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,GAAW;IACzC,MAAM,KAAK,GAAY,IAAI,GAAG,EAAE,CAAA;IAChC,IAAI,GAAY,CAAA;IAChB,IAAI,CAAC;QACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,CAAC,CAAA;IACjE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;IACD,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;QAAE,OAAO,KAAK,CAAA;IAC/E,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAA8B,CAAC,EAAE,CAAC;QACzE,oFAAoF;QACpF,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,OAAO,IAAI,GAAG,EAAE,CAAA;QACzF,MAAM,KAAK,GAAG,KAAgC,CAAA;QAC9C,MAAM,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAA;QACvC,MAAM,QAAQ,GAAG,KAAK,CAAC,UAAU,CAAC,IAAI,KAAK,CAAA;QAC3C,IAAI,OAAO,MAAM,KAAK,SAAS,IAAI,OAAO,QAAQ,KAAK,SAAS;YAAE,OAAO,IAAI,GAAG,EAAE,CAAA;QAClF,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAA;IACrC,CAAC;IACD,OAAO,KAAK,CAAA;AACd,CAAC;AAOD,MAAM,oBAAoB,GAAkB;IAC1C,KAAK,EAAE,uCAAuC;IAC9C,MAAM,EAAE,qCAAqC;CAC9C,CAAA;AAED,0FAA0F;AAC1F,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,MAAc,EACd,KAAsB,EACtB,WAA0B,oBAAoB;IAE9C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAC7D,IAAI,CAAC;QACH,MAAM,SAAS,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAC9C,MAAM,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;IACtC,CAAC;IACD,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;IAC3B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAC9C,MAAM,OAAO,CAAC,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC,CAAA;IACvC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,GAAW,EAAE,KAAc;IACpD,MAAM,MAAM,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IACxG,OAAO,WAAW,CAAC,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;AAC5E,CAAC;AAED,iHAAiH;AACjH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,GAAW,EAAE,IAAY;IAC5D,MAAM,WAAW,CAAC,IAAI,EAAE,MAAM,YAAY,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,CAAA;AAC9D,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,GAAW,EACX,EAAU,EACV,IAAY,EACZ,KAAc,EACd,OAAmB;IAEnB,IAAI,MAAM,OAAO,CAAC,GAAG,EAAE,EAAE,EAAE,IAAI,CAAC;QAAE,OAAO,SAAS,CAAA;IAClD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAA;IACxC,MAAM,IAAI,GAAG,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAA;IAC9C,MAAM,MAAM,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAA;IACjE,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAA;IACxC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;IACpD,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAA;IAC3B,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;IAChC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAC1B,OAAO;QACL,EAAE;QACF,IAAI;QACJ,aAAa,EAAE,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI;QACnD,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,SAAS,EAAE,IAAI,CAAC,IAAI;QACpB,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,MAAM,EAAE,IAAI,EAAE,MAAM,IAAI,KAAK;QAC7B,QAAQ,EAAE,IAAI,EAAE,QAAQ,IAAI,KAAK;QACjC,MAAM;KACP,CAAA;AACH,CAAC;AAED,uCAAuC;AACvC,KAAK,UAAU,IAAI,CAAC,GAAW,EAAE,KAAc,EAAE,OAAmB;IAClE,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAc,CAAC,CAAA;IAC5D,MAAM,GAAG,GAAG,KAAK;SACd,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;SACvC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;SAC5C,MAAM,CAAC,SAAS,CAAC,CAAA;IACpB,MAAM,KAAK,GAAuB,EAAE,CAAA;IACpC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,CAAC,MAAM,EAAE,EAAE,IAAI,gBAAgB,EAAE,CAAC;QACzD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,GAAG,gBAAgB,CAAC,CAAA;QAClD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,GAAG,CAC7B,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,EAAE,EAAE,EAAE,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,CAC3E,CAAA;QACD,KAAK,MAAM,IAAI,IAAI,KAAK;YAAE,IAAI,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACtD,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;AACxG,CAAC;AAMD;;;GAGG;AACH,MAAM,OAAO,OAAO;IAMW;IALZ,SAAS,GAAG,IAAI,UAAU,EAAE,CAAA;IAC7C,wGAAwG;IACvF,UAAU,GAAG,IAAI,GAAG,EAA6D,CAAA;IACjF,OAAO,GAAe,CAAC,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,EAAE,EAAE,IAAI,CAAC,CAAA;IAEzF,YAA6B,MAAqB,GAAG,EAAE,GAAE,CAAC;QAA7B,QAAG,GAAH,GAAG,CAA0B;IAAG,CAAC;IAE9D;;;;OAIG;IACK,KAAK,CAAC,WAAW,CAAC,GAAW,EAAE,EAAU,EAAE,IAAY;QAC7D,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QAChC,MAAM,IAAI,GAAG,CAAC,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAA;QACjD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;QACpD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,gBAAgB;YAAE,OAAO,KAAK,CAAA;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QACvC,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC,KAAK,CAAA;QAC3F,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;aAC/B,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC;aAChC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAA;QACrB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;QAC5E,IAAI,KAAK;YAAE,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,+BAA+B,EAAE,EAAE,CAAC,CAAA;QAChE,OAAO,KAAK,CAAA;IACd,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,IAAI,CACR,GAAW,EACX,MAAmB,EACnB,GAAW;QAEX,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YAC7D,MAAM,OAAO,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAA;QAC7D,CAAC,CAAC,CAAA;QACF,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAA;QAC7C,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QACpC,MAAM,KAAK,GAAG,YAAY,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;QACvC,MAAM,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAE9B,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QAChC,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,IAAI,CAC3D,GAAG,EAAE,CAAC,KAAK,EACX,CAAC,KAAc,EAAE,EAAE;YACjB,IAAI,CAAC,GAAG,CACN,MAAM,EACN,iBAAiB,EAAE,YAAY,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACxF,CAAA;YACD,OAAO,IAAI,CAAA;QACb,CAAC,CACF,CAAA;QACD,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,CAAA;QACxC,IAAI,CAAC,MAAM;YAAE,MAAM,cAAc,CAAC,UAAU,EAAE,uCAAuC,CAAC,CAAA;QACtF,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QACrE,OAAO;YACL,IAAI,EAAE;gBACJ,EAAE;gBACF,IAAI;gBACJ,aAAa;gBACb,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,SAAS,EAAE,KAAK,CAAC,MAAM;gBACvB,WAAW,EAAE,MAAM,CAAC,WAAW;gBAC/B,MAAM,EAAE,IAAI,EAAE,MAAM,IAAI,KAAK;gBAC7B,QAAQ,EAAE,IAAI,EAAE,QAAQ,IAAI,KAAK;gBACjC,MAAM;aACP;YACD,KAAK;SACN,CAAA;IACH,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,GAAW,EAAE,EAAU;QAC/B,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QACpC,IAAI,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;YAAE,OAAO,IAAI,CAAA;QACtC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAA;QAC5D,OAAO,CAAC,MAAM,YAAY,CAAC,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,IAAI,CAAA;IACzE,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,GAAW,EAAE,OAA2B;QACjD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAA;QAC5D,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CACvD,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,eAAe,KAAK,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,CAC7D,CAAA;QACD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAA;QACpF,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,CAAA;IAC1F,CAAC;IAED,uFAAuF;IACvF,MAAM,CAAC,GAAW,EAAE,GAAsB;QACxC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE;YACnC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,CAAA;YAClC,IAAI,OAAO,GAAG,KAAK,CAAA;YACnB,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;gBACrB,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;gBACpC,IAAI,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;oBACvB,MAAM,IAAI,GAAG,CAAC,MAAM,aAAa,CAAC,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAA;oBACnE,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC7C,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,oCAAoC,EAAE,EAAE,CAAC,CAAA;wBAC1D,SAAQ;oBACV,CAAC;oBACD,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;wBACtC,MAAM,OAAO,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAA;oBACrD,CAAC,CAAC,CAAA;gBACJ,CAAC;gBACD,MAAM,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;gBAC7D,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;gBAC5B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAA;gBAC1C,IAAI,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;oBAAE,OAAO,GAAG,IAAI,CAAA;YACtC,CAAC;YACD,IAAI,OAAO;gBAAE,MAAM,UAAU,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;QAC3C,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,QAAQ,CAAC,GAAW,EAAE,EAAU,EAAE,MAAoB;QACpD,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE;YACnC,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;YACpC,IAAI,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;gBACvB,MAAM,cAAc,CAAC,eAAe,EAAE,yCAAyC,CAAC,CAAA;YAClF,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,CAAA;YAClC,MAAM,KAAK,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,EAAE,CAAA;YAC1E,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS;gBAAE,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAA;YAC7D,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS;gBAAE,KAAK,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAA;YACnE,6DAA6D;YAC7D,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,QAAQ;gBAAE,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAA;;gBACnD,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;YACrB,MAAM,UAAU,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;YAC5B,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;YACnE,IAAI,CAAC,IAAI;gBAAE,MAAM,cAAc,CAAC,eAAe,EAAE,iDAAiD,CAAC,CAAA;YACnG,OAAO,IAAI,CAAA;QACb,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,uDAAuD;IACvD,KAAK,CAAC,MAAM,CAAC,GAAW,EAAE,EAAU,EAAE,MAAc;QAClD,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QACpC,IAAI,CAAC,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;YACvB,MAAM,cAAc,CAAC,eAAe,EAAE,yCAAyC,CAAC,CAAA;QAClF,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACzE,MAAM,OAAO,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAA;QAClE,CAAC,CAAC,CAAA;QACF,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;YACpD,MAAM,OAAO,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAA;QACrD,CAAC,CAAC,CAAA;IACJ,CAAC;CACF"}

package/dist/diffusion/http.d.ts

@@ -0,0 +1,16 @@
+/**
+ * The little HTTP the job runner needs against `sd-server`: a GET or a JSON POST with a deadline,
+ * status and body text back. Over the raw-socket client rather than the global `fetch`, which in the
+ * Bun-compiled binary honours `HTTP_PROXY` even for loopback (measured while planning stage 7;
+ * ADR 2026-09-15-proxied-downloads-use-a-raw-socket-client).
+ */
+export interface SdResponse {
+    status: number;
+    text: string;
+}
+export interface SdHttpClient {
+    get(url: string, timeoutMs: number): Promise<SdResponse>;
+    /** `json` undefined sends an empty body (sd.cpp's cancel takes none). */
+    post(url: string, json: unknown, timeoutMs: number): Promise<SdResponse>;
+}
+export declare function createSdHttpClient(fetchImpl?: typeof fetch): SdHttpClient;

package/dist/diffusion/http.js

@@ -0,0 +1,41 @@
+/**
+ * The little HTTP the job runner needs against `sd-server`: a GET or a JSON POST with a deadline,
+ * status and body text back. Over the raw-socket client rather than the global `fetch`, which in the
+ * Bun-compiled binary honours `HTTP_PROXY` even for loopback (measured while planning stage 7;
+ * ADR 2026-09-15-proxied-downloads-use-a-raw-socket-client).
+ */
+import { createPolicyFetch } from '../downloads/index.js';
+/** One request with a deadline that covers the whole exchange, body included. */
+async function exchange(fetchImpl, url, init, timeoutMs) {
+    const controller = new AbortController();
+    let timedOut = false;
+    const timer = setTimeout(() => {
+        timedOut = true;
+        controller.abort();
+    }, timeoutMs);
+    try {
+        const response = await fetchImpl(url, { ...init, signal: controller.signal });
+        return { status: response.status, text: await response.text() };
+    }
+    catch (error) {
+        if (timedOut)
+            throw new Error(`sd-server did not answer within ${timeoutMs} ms`);
+        throw error;
+    }
+    finally {
+        clearTimeout(timer);
+    }
+}
+export function createSdHttpClient(fetchImpl = createPolicyFetch({})) {
+    return {
+        get: (url, timeoutMs) => exchange(fetchImpl, url, { method: 'GET' }, timeoutMs),
+        post: (url, json, timeoutMs) => {
+            const body = json === undefined ? '' : JSON.stringify(json);
+            const headers = { 'content-length': String(Buffer.byteLength(body)) };
+            if (json !== undefined)
+                headers['content-type'] = 'application/json';
+            return exchange(fetchImpl, url, { method: 'POST', headers, body }, timeoutMs);
+        },
+    };
+}
+//# sourceMappingURL=http.js.map

package/dist/diffusion/http.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../../src/diffusion/http.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAA;AAazD,iFAAiF;AACjF,KAAK,UAAU,QAAQ,CACrB,SAAuB,EACvB,GAAW,EACX,IAAiB,EACjB,SAAiB;IAEjB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAA;IACxC,IAAI,QAAQ,GAAG,KAAK,CAAA;IACpB,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;QAC5B,QAAQ,GAAG,IAAI,CAAA;QACf,UAAU,CAAC,KAAK,EAAE,CAAA;IACpB,CAAC,EAAE,SAAS,CAAC,CAAA;IACb,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAA;QAC7E,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAA;IACjE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,SAAS,KAAK,CAAC,CAAA;QAChF,MAAM,KAAK,CAAA;IACb,CAAC;YAAS,CAAC;QACT,YAAY,CAAC,KAAK,CAAC,CAAA;IACrB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,YAA0B,iBAAiB,CAAC,EAAE,CAAC;IAChF,OAAO;QACL,GAAG,EAAE,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,SAAS,CAAC;QAC/E,IAAI,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;YAC7B,MAAM,IAAI,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;YAC3D,MAAM,OAAO,GAA2B,EAAE,gBAAgB,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,CAAA;YAC7F,IAAI,IAAI,KAAK,SAAS;gBAAE,OAAO,CAAC,cAAc,CAAC,GAAG,kBAAkB,CAAA;YACpE,OAAO,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,SAAS,CAAC,CAAA;QAC/E,CAAC;KACF,CAAA;AACH,CAAC"}

package/dist/diffusion/idle.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Unload the model after `idleUnloadSecs` without a job. The deadline is re-armed by every job
+ * completion and by `touchIdle`. `idle_task` in `lib.rs` of `tauri-plugin-atomic-diffusion` (app
+ * commit `767ff6350`): a tick, never a wait behind a load.
+ */
+import type { AsyncMutex } from './mutex.js';
+import type { DiffusionState } from './state.js';
+/** How often the idle task checks the deadline. */
+export declare const IDLE_TICK_MS = 30000;
+export interface IdleDeps {
+    state: DiffusionState;
+    loadLock: AsyncMutex;
+    /** Unload the resident session; called with the load lock held. */
+    unload: (reason: string) => Promise<void>;
+    log: (level: 'info' | 'warn', msg: string) => void;
+    tickMs?: number;
+}
+/** One look at the deadline: whether an unload was started. Exposed for the tests and the service. */
+export declare function idleTick(deps: IdleDeps): Promise<boolean>;
+/** Start the ticking; the timer never keeps the process alive. */
+export declare function startIdleTask(deps: IdleDeps): {
+    stop(): void;
+};

package/dist/diffusion/idle.js

@@ -0,0 +1,45 @@
+/**
+ * Unload the model after `idleUnloadSecs` without a job. The deadline is re-armed by every job
+ * completion and by `touchIdle`. `idle_task` in `lib.rs` of `tauri-plugin-atomic-diffusion` (app
+ * commit `767ff6350`): a tick, never a wait behind a load.
+ */
+/** How often the idle task checks the deadline. */
+export const IDLE_TICK_MS = 30_000;
+/** One look at the deadline: whether an unload was started. Exposed for the tests and the service. */
+export async function idleTick(deps) {
+    const { state } = deps;
+    if (!state.idleExpired() || state.activeJobId !== undefined)
+        return false;
+    const release = deps.loadLock.tryAcquire();
+    if (!release)
+        return false;
+    try {
+        if (!state.session) {
+            state.clearIdle();
+            return false;
+        }
+        deps.log('info', 'unloading the image model after idling');
+        await deps.unload('idle');
+        return true;
+    }
+    catch (error) {
+        deps.log('warn', `idle unload failed: ${error instanceof Error ? error.message : String(error)}`);
+        return false;
+    }
+    finally {
+        release();
+    }
+}
+/** Start the ticking; the timer never keeps the process alive. */
+export function startIdleTask(deps) {
+    let running = false;
+    const timer = setInterval(() => {
+        if (running)
+            return;
+        running = true;
+        void idleTick(deps).finally(() => (running = false));
+    }, deps.tickMs ?? IDLE_TICK_MS);
+    timer.unref();
+    return { stop: () => clearInterval(timer) };
+}
+//# sourceMappingURL=idle.js.map

package/dist/diffusion/idle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"idle.js","sourceRoot":"","sources":["../../src/diffusion/idle.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,mDAAmD;AACnD,MAAM,CAAC,MAAM,YAAY,GAAG,MAAM,CAAA;AAWlC,sGAAsG;AACtG,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,IAAc;IAC3C,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAA;IACtB,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,KAAK,CAAC,WAAW,KAAK,SAAS;QAAE,OAAO,KAAK,CAAA;IACzE,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAA;IAC1C,IAAI,CAAC,OAAO;QAAE,OAAO,KAAK,CAAA;IAC1B,IAAI,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACnB,KAAK,CAAC,SAAS,EAAE,CAAA;YACjB,OAAO,KAAK,CAAA;QACd,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,wCAAwC,CAAC,CAAA;QAC1D,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;QACzB,OAAO,IAAI,CAAA;IACb,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,uBAAuB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;QACjG,OAAO,KAAK,CAAA;IACd,CAAC;YAAS,CAAC;QACT,OAAO,EAAE,CAAA;IACX,CAAC;AACH,CAAC;AAED,kEAAkE;AAClE,MAAM,UAAU,aAAa,CAAC,IAAc;IAC1C,IAAI,OAAO,GAAG,KAAK,CAAA;IACnB,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,IAAI,OAAO;YAAE,OAAM;QACnB,OAAO,GAAG,IAAI,CAAA;QACd,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,OAAO,GAAG,KAAK,CAAC,CAAC,CAAA;IACtD,CAAC,EAAE,IAAI,CAAC,MAAM,IAAI,YAAY,CAAC,CAAA;IAC/B,KAAK,CAAC,KAAK,EAAE,CAAA;IACb,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAA;AAC7C,CAAC"}

package/dist/diffusion/image-job.d.ts

@@ -0,0 +1,25 @@
+/**
+ * The image side of the job runner: what an image job validates, sends, decodes and saves. Moved
+ * out of `jobs.ts` when video arrived; the functions are the plugin's (`jobs.rs` at app commit
+ * `ec1fd3ea7`), the `JobKind` wrapper is what the shared runner sees.
+ */
+import type { GalleryImageItem, ImageGenerateRequest, ImageJob, ImageJobProgress, ImageSource } from '../contracts/index.js';
+import type { JobKind, SaveContext } from './job-kind.js';
+import type { JobDeps } from './jobs.js';
+import type { ResolvedInputs } from './types.js';
+/** One source as the base64 `sd-server` takes: inline bytes as they are, a file read from disk. */
+export declare function resolveSource(source: ImageSource, deps: Pick<JobDeps, 'readSource'>): Promise<string>;
+/** The request's images as the base64 `sd-server` takes, once per job (not per retry). Only what the workflow uses is read. */
+export declare function resolveInputs(request: ImageGenerateRequest, deps: Pick<JobDeps, 'readSource'>): Promise<ResolvedInputs>;
+/** A source as a job record carries it: the path stays, inline bytes are blanked. */
+export declare const redactSource: (source: ImageSource) => ImageSource;
+/** The request as the job record and every job event carry it: file paths stay, inline bytes are blanked. */
+export declare function withoutSources(request: ImageGenerateRequest): ImageGenerateRequest;
+/** The images of a completed job, in index order. */
+export declare function decodeImages(job: Record<string, unknown>): Buffer[];
+/** Refuse a blank batch, then write every PNG with its recipe and thumbnail into the gallery. */
+export declare function saveImageOutputs(deps: JobDeps, ctx: SaveContext<ImageGenerateRequest>, pngs: Buffer[]): Promise<{
+    items: GalleryImageItem[];
+    bytes: Buffer[];
+}>;
+export declare const IMAGE_JOB_KIND: JobKind<ImageGenerateRequest, ImageJob, GalleryImageItem, ImageJobProgress, Buffer[]>;