@atomic-chat/core 0.5.1

package/dist/diffusion/args.js

@@ -0,0 +1,291 @@
+/**
+ * Pure argv and request builders for `sd-server`: no I/O and no process, so every flag decision is
+ * tested without a binary or a model file. Port of `args.rs` in `tauri-plugin-atomic-diffusion` (app
+ * commit `ec1fd3ea7`, and the host switch of `process.rs`), itself a port of Studio's `sd_cpp_args.py`.
+ *
+ * Every flag emitted here is listed in `test/fixtures/sdcpp/required-flags.txt`, which the live test
+ * holds against the pinned binary's `--help`.
+ */
+import { defaultStrength, usesInitImage, usesMask, usesReferences, workflowOf } from './workflow.js';
+/** Kill switch for the Metal text-encoder placement: `1`/`true`/`yes`/`on` keeps the encoder on Metal. */
+export const METAL_TE_GPU_ENV = 'ATOMIC_DIFFUSION_METAL_TE_GPU';
+/**
+ * Opt-in for `-M vid_gen` on a video load. `sd-server` infers its modes from the model, and the flag
+ * is documented for `sd-cli`; it is emitted only when this is `1`/`true`/`yes`/`on`, until the live
+ * test has shown whether the server wants it.
+ */
+export const VID_GEN_MODE_FLAG_ENV = 'ATOMIC_DIFFUSION_VID_GEN_MODE_FLAG';
+const truthy = (value) => ['1', 'true', 'yes', 'on'].includes((value ?? '').trim().toLowerCase());
+function dedup(flags) {
+    return [...new Set(flags)];
+}
+/**
+ * The memory policy as sd.cpp offload flags. `group` streams the model (`--offload-to-cpu`) with
+ * flash attention; `model` also puts CLIP and the VAE on the CPU and tiles the VAE.
+ */
+export function offloadFlags(policy) {
+    if (policy === 'group')
+        return ['--offload-to-cpu', '--diffusion-fa'];
+    if (policy === 'model')
+        return ['--offload-to-cpu', '--clip-on-cpu', '--vae-on-cpu', '--diffusion-fa', '--vae-tiling'];
+    return [];
+}
+/** Numerically exact speed-ups: flash attention, and direct convolution in the diffusion model. */
+export function speedFlags() {
+    return ['--diffusion-fa', '--diffusion-conv-direct'];
+}
+/**
+ * Keep the text encoder on the CPU under Apple Metal. ggml's Metal backend aborts on `RMS_NORM` for
+ * non-contiguous rows with no per-op CPU fallback, so an LLM text encoder (Qwen3 for Z-Image and
+ * FLUX.2, T5 for FLUX.1) takes the whole server down on the first prompt. The encoder runs once per
+ * prompt and the denoiser every step, so pinning only the encoder keeps Metal where it matters.
+ */
+export function metalTextEncoderFlags(isMacos, envOverride) {
+    if (!isMacos)
+        return [];
+    return truthy(envOverride) ? [] : ['--clip-on-cpu'];
+}
+/** `-M vid_gen` for a video model, only while the environment opts in. */
+export function videoModeFlags(modality, envOverride) {
+    return modality === 'video' && truthy(envOverride) ? ['-M', 'vid_gen'] : [];
+}
+/** An Apple M5 CPU brand string: `Apple M5`, `Apple M5 Max`. */
+export function isM5Brand(brand) {
+    return brand.split(/\s+/).some((part) => part.startsWith('M5'));
+}
+/**
+ * The environment `sd-server` gets for this host. ggml's Metal Tensor API is unstable on some M5 and
+ * macOS combinations: it can fail command buffers, or return NaN latents that come out as white
+ * images. On an M5 with the Metal backend it is switched off, which keeps Metal on its mature
+ * SIMD-group path; the argv is left alone, because older engine builds reject newer scaling flags.
+ */
+export function hostEnv(platform, backend, cpuBrand) {
+    if (platform === 'darwin' && backend === 'metal' && isM5Brand(cpuBrand ?? ''))
+        return { GGML_METAL_TENSOR_DISABLE: '1' };
+    return {};
+}
+/**
+ * `flags` with every `--backend <spec>` pair removed. sd.cpp concatenates repeated `--backend`
+ * values instead of replacing them, and a per-module entry beats the bare default: appending
+ * `--backend cpu` to a spec that still says `diffusion=CUDA0` would leave the denoiser on CUDA.
+ */
+export function withoutDeviceBackendFlags(flags) {
+    const out = [];
+    let skip = false;
+    for (const flag of flags) {
+        if (skip) {
+            skip = false;
+            continue;
+        }
+        if (flag === '--backend') {
+            skip = true;
+            continue;
+        }
+        out.push(flag);
+    }
+    return out;
+}
+/** The `extraArgs` of the one automatic recovery: everything on the CPU backend. */
+export function cpuBackendExtraArgs(extraArgs) {
+    return [...withoutDeviceBackendFlags(extraArgs), '--backend', 'cpu'];
+}
+const GGML_UNSUPPORTED_OP_MARKERS = ['unsupported op', 'ggml_abort'];
+/**
+ * True when the captured output carries a ggml unsupported-op abort. That signature is
+ * deterministic for the graph in question: a retry on the same backend fails identically, a CPU
+ * restart runs it. No other death may be retried automatically.
+ */
+export function isGgmlUnsupportedOpAbort(text) {
+    const lower = text.toLowerCase();
+    return GGML_UNSUPPORTED_OP_MARKERS.every((marker) => lower.includes(marker));
+}
+/**
+ * The `sd-server` argv, without the binary. Model files first, then the listener, the scratch
+ * directories sd-server insists on iterating, threads, the hardware flags (de-duplicated, stable
+ * order), `-v` so the per-step sampling lines are printed, and the caller's `extraArgs` last.
+ */
+export function buildServerArgs(spec, port, scratchDir, host) {
+    const { files } = spec;
+    const args = ['--diffusion-model', files.diffusionModel];
+    const optional = [
+        ['--vae', files.vae],
+        ['--audio-vae', files.audioVae],
+        ['--embeddings-connectors', files.embeddingsConnectors],
+        ['--clip_l', files.clipL],
+        ['--t5xxl', files.t5xxl],
+        ['--llm', files.llm],
+        ['--llm_vision', files.llmVision],
+        ['--qwen2vl', files.qwen2vl],
+    ];
+    for (const [flag, value] of optional)
+        if (value)
+            args.push(flag, value);
+    if (files.vaeFormat)
+        args.push('--vae-format', files.vaeFormat);
+    args.push('--listen-ip', '127.0.0.1', '--listen-port', String(port));
+    for (const flag of ['--lora-model-dir', '--hires-upscalers-dir', '--embd-dir'])
+        args.push(flag, scratchDir);
+    if (spec.threads !== undefined)
+        args.push('--threads', String(spec.threads));
+    args.push(...dedup([
+        ...offloadFlags(spec.offload),
+        ...speedFlags(),
+        ...metalTextEncoderFlags(host.platform === 'darwin', host.env[METAL_TE_GPU_ENV]),
+    ]));
+    args.push(...videoModeFlags(spec.modality, host.env[VID_GEN_MODE_FLAG_ENV]));
+    args.push('-v', ...spec.extraArgs);
+    return args;
+}
+/**
+ * Output area above which a request turns VAE tiling on. The VAE's compute buffer grows with the
+ * pixel count (FLUX.2: 6.7 GB to decode at 1024², 26.6 GB at 2048², where a 2x Upscale lands and
+ * where it failed on a 24 GB card). Tiled, the peak stays at the one-tile figure. Up to 1024²
+ * nothing changes.
+ */
+export const VAE_TILING_AREA = 1024 * 1024;
+/**
+ * Pixel-frames (width × height × frames) above which a video request turns VAE tiling on. A video
+ * VAE decodes the whole clip in one graph, so its buffer grows with the frame count as much as with
+ * the frame size: the Wan 2.2 VAE asked Metal for 27.6 GB at 1280×704 × 121 frames (≈250 bytes per
+ * pixel-frame, 2026-09-23) and failed on a machine with 14.9 GB free. Eight megapixel-frames is
+ * about 2 GB of that, the point past which a single-graph decode stops being safe anywhere.
+ */
+export const VIDEO_VAE_TILING_PIXEL_FRAMES = VAE_TILING_AREA * 8;
+/**
+ * The `POST /sdcpp/v1/img_gen` body. The whole batch goes in one request; guidance is split the way
+ * sd.cpp expects (CFG → `txt_cfg`, FLUX distilled → `distilled_guidance`). Only set keys are sent,
+ * so the server's own defaults apply to the rest. The workflow decides which images go in. sd.cpp
+ * resizes the init image to `width`×`height` itself, which is how Upscale works.
+ *
+ * sd.cpp reads integers with `is_number_integer()` and floats with `is_number()`, so `1` for a
+ * `cfgScale` of 1.0 is accepted (checked in its `common.cpp`).
+ */
+export function buildImgGenRequest(request, defaults, seed, inputs) {
+    const guidance = { txt_cfg: request.cfgScale };
+    const distilled = request.guidance ?? defaults.guidance;
+    if (distilled !== undefined)
+        guidance['distilled_guidance'] = distilled;
+    const sampleParams = { sample_steps: request.steps };
+    const method = request.samplingMethod ?? defaults.samplingMethod;
+    if (method)
+        sampleParams['sample_method'] = method;
+    const shift = request.flowShift ?? defaults.flowShift;
+    if (shift !== undefined)
+        sampleParams['flow_shift'] = shift;
+    sampleParams['guidance'] = guidance;
+    const body = {
+        prompt: request.prompt,
+        negative_prompt: request.negativePrompt ?? '',
+        width: request.width,
+        height: request.height,
+        batch_count: Math.max(request.batchSize, 1),
+        output_format: 'png',
+        seed,
+        sample_params: sampleParams,
+    };
+    const workflow = workflowOf(request);
+    if (usesInitImage(workflow)) {
+        if (inputs.init !== undefined) {
+            body['init_image'] = inputs.init;
+            body['strength'] = request.strength ?? defaultStrength(workflow);
+        }
+        if (usesMask(workflow) && inputs.mask !== undefined)
+            body['mask_image'] = inputs.mask;
+    }
+    if (usesReferences(workflow) && inputs.refs.length > 0)
+        body['ref_images'] = inputs.refs;
+    if (request.width * request.height > VAE_TILING_AREA)
+        body['vae_tiling_params'] = { enabled: true };
+    return body;
+}
+/**
+ * The `POST /sdcpp/v1/vid_gen` body: one clip, the frame count and rate resolved against the family,
+ * `custom_sigmas` when the family ships a fixed schedule for exactly this many steps (LTX-2 distilled),
+ * VP8 in WebM (the one container the app plays), and the first and last frames when a later workflow
+ * resolves them. Only set keys are sent, like the image body.
+ */
+export function buildVidGenRequest(request, defaults, seed, inputs) {
+    const guidance = { txt_cfg: request.cfgScale };
+    const distilled = request.guidance ?? defaults.guidance;
+    if (distilled !== undefined)
+        guidance['distilled_guidance'] = distilled;
+    const sampleParams = { sample_steps: request.steps };
+    const method = request.samplingMethod ?? defaults.samplingMethod;
+    if (method)
+        sampleParams['sample_method'] = method;
+    const shift = request.flowShift ?? defaults.flowShift;
+    if (shift !== undefined)
+        sampleParams['flow_shift'] = shift;
+    if (defaults.sigmas !== undefined && defaults.sigmas.length === request.steps)
+        sampleParams['custom_sigmas'] = [...defaults.sigmas];
+    sampleParams['guidance'] = guidance;
+    const video = defaults.video;
+    const body = {
+        prompt: request.prompt,
+        negative_prompt: request.negativePrompt ?? '',
+        width: request.width,
+        height: request.height,
+        video_frames: request.frames ?? video?.frames ?? 1,
+        fps: request.fps ?? video?.fps ?? 24,
+        output_format: 'webm',
+        seed,
+        sample_params: sampleParams,
+    };
+    if (inputs.init !== undefined)
+        body['init_image'] = inputs.init;
+    if (inputs.end !== undefined)
+        body['end_image'] = inputs.end;
+    const frames = body['video_frames'];
+    if (request.width * request.height * frames > VIDEO_VAE_TILING_PIXEL_FRAMES)
+        body['vae_tiling_params'] = { enabled: true };
+    return body;
+}
+function lastOptionValue(args, option) {
+    let value;
+    let i = 0;
+    while (i < args.length) {
+        const arg = args[i];
+        if (arg === option) {
+            value = args[i + 1] ?? value;
+            i += 2;
+            continue;
+        }
+        if (arg.startsWith(`${option}=`))
+            value = arg.slice(option.length + 1);
+        i += 1;
+    }
+    return value;
+}
+function compact(value, limit) {
+    const chars = [...value];
+    return chars.length <= limit ? value : `${chars.slice(0, Math.max(limit - 3, 0)).join('')}...`;
+}
+/** One line for the log: the model's basename and the numeric settings, never a path or a prompt. */
+export function commandSummaryForLog(args) {
+    const fields = [];
+    const model = lastOptionValue(args, '--diffusion-model');
+    if (model !== undefined) {
+        const name = model.replaceAll('\\', '/').split('/').pop() ?? '';
+        fields.push(`model=${compact(name, 48)}`);
+    }
+    const width = lastOptionValue(args, '--width');
+    const height = lastOptionValue(args, '--height');
+    if (width !== undefined && height !== undefined)
+        fields.push(`size=${compact(width, 8)}x${compact(height, 8)}`);
+    const numeric = [
+        ['steps', '--steps'],
+        ['seed', '--seed'],
+        ['port', '--listen-port'],
+        ['threads', '--threads'],
+    ];
+    for (const [label, option] of numeric) {
+        const value = lastOptionValue(args, option);
+        if (value !== undefined)
+            fields.push(`${label}=${compact(value, 16)}`);
+    }
+    const backend = lastOptionValue(args, '--backend');
+    if (backend !== undefined)
+        fields.push(`backend=${compact(backend, 32)}`);
+    return fields.join(' ');
+}
+//# sourceMappingURL=args.js.map

package/dist/diffusion/args.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"args.js","sourceRoot":"","sources":["../../src/diffusion/args.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAWH,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,QAAQ,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,eAAe,CAAA;AAEpG,0GAA0G;AAC1G,MAAM,CAAC,MAAM,gBAAgB,GAAG,+BAA+B,CAAA;AAE/D;;;;GAIG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,oCAAoC,CAAA;AAEzE,MAAM,MAAM,GAAG,CAAC,KAAyB,EAAW,EAAE,CACpD,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC,QAAQ,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAA;AAQzE,SAAS,KAAK,CAAC,KAAwB;IACrC,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,CAAA;AAC5B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,MAA8B;IACzD,IAAI,MAAM,KAAK,OAAO;QAAE,OAAO,CAAC,kBAAkB,EAAE,gBAAgB,CAAC,CAAA;IACrE,IAAI,MAAM,KAAK,OAAO;QACpB,OAAO,CAAC,kBAAkB,EAAE,eAAe,EAAE,cAAc,EAAE,gBAAgB,EAAE,cAAc,CAAC,CAAA;IAChG,OAAO,EAAE,CAAA;AACX,CAAC;AAED,mGAAmG;AACnG,MAAM,UAAU,UAAU;IACxB,OAAO,CAAC,gBAAgB,EAAE,yBAAyB,CAAC,CAAA;AACtD,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,qBAAqB,CAAC,OAAgB,EAAE,WAA+B;IACrF,IAAI,CAAC,OAAO;QAAE,OAAO,EAAE,CAAA;IACvB,OAAO,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAA;AACrD,CAAC;AAED,0EAA0E;AAC1E,MAAM,UAAU,cAAc,CAAC,QAA2B,EAAE,WAA+B;IACzF,OAAO,QAAQ,KAAK,OAAO,IAAI,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;AAC7E,CAAC;AAED,gEAAgE;AAChE,MAAM,UAAU,SAAS,CAAC,KAAa;IACrC,OAAO,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAA;AACjE,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,OAAO,CACrB,QAAyB,EACzB,OAAyB,EACzB,QAA4B;IAE5B,IAAI,QAAQ,KAAK,QAAQ,IAAI,OAAO,KAAK,OAAO,IAAI,SAAS,CAAC,QAAQ,IAAI,EAAE,CAAC;QAC3E,OAAO,EAAE,yBAAyB,EAAE,GAAG,EAAE,CAAA;IAC3C,OAAO,EAAE,CAAA;AACX,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,yBAAyB,CAAC,KAAwB;IAChE,MAAM,GAAG,GAAa,EAAE,CAAA;IACxB,IAAI,IAAI,GAAG,KAAK,CAAA;IAChB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,GAAG,KAAK,CAAA;YACZ,SAAQ;QACV,CAAC;QACD,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACzB,IAAI,GAAG,IAAI,CAAA;YACX,SAAQ;QACV,CAAC;QACD,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAChB,CAAC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,oFAAoF;AACpF,MAAM,UAAU,mBAAmB,CAAC,SAA4B;IAC9D,OAAO,CAAC,GAAG,yBAAyB,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,CAAA;AACtE,CAAC;AAED,MAAM,2BAA2B,GAAG,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAA;AAEpE;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CAAC,IAAY;IACnD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAA;IAChC,OAAO,2BAA2B,CAAC,KAAK,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAA;AAC9E,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe,CAC7B,IAAgB,EAChB,IAAY,EACZ,UAAkB,EAClB,IAAc;IAEd,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAA;IACtB,MAAM,IAAI,GAAG,CAAC,mBAAmB,EAAE,KAAK,CAAC,cAAc,CAAC,CAAA;IACxD,MAAM,QAAQ,GAAwC;QACpD,CAAC,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC;QACpB,CAAC,aAAa,EAAE,KAAK,CAAC,QAAQ,CAAC;QAC/B,CAAC,yBAAyB,EAAE,KAAK,CAAC,oBAAoB,CAAC;QACvD,CAAC,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC;QACzB,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC;QACxB,CAAC,OAAO,EAAE,KAAK,CAAC,GAAG,CAAC;QACpB,CAAC,cAAc,EAAE,KAAK,CAAC,SAAS,CAAC;QACjC,CAAC,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC;KAC7B,CAAA;IACD,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,QAAQ;QAAE,IAAI,KAAK;YAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;IACvE,IAAI,KAAK,CAAC,SAAS;QAAE,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,SAAS,CAAC,CAAA;IAC/D,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAA;IACpE,KAAK,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,uBAAuB,EAAE,YAAY,CAAC;QAAE,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;IAC3G,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS;QAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAA;IAC5E,IAAI,CAAC,IAAI,CACP,GAAG,KAAK,CAAC;QACP,GAAG,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC;QAC7B,GAAG,UAAU,EAAE;QACf,GAAG,qBAAqB,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;KACjF,CAAC,CACH,CAAA;IACD,IAAI,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5E,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,CAAA;IAClC,OAAO,IAAI,CAAA;AACb,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,GAAG,IAAI,CAAA;AAE1C;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,6BAA6B,GAAG,eAAe,GAAG,CAAC,CAAA;AAEhE;;;;;;;;GAQG;AACH,MAAM,UAAU,kBAAkB,CAChC,OAA6B,EAC7B,QAAiC,EACjC,IAAY,EACZ,MAAsB;IAEtB,MAAM,QAAQ,GAA4B,EAAE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAA;IACvE,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,CAAA;IACvD,IAAI,SAAS,KAAK,SAAS;QAAE,QAAQ,CAAC,oBAAoB,CAAC,GAAG,SAAS,CAAA;IAEvE,MAAM,YAAY,GAA4B,EAAE,YAAY,EAAE,OAAO,CAAC,KAAK,EAAE,CAAA;IAC7E,MAAM,MAAM,GAAG,OAAO,CAAC,cAAc,IAAI,QAAQ,CAAC,cAAc,CAAA;IAChE,IAAI,MAAM;QAAE,YAAY,CAAC,eAAe,CAAC,GAAG,MAAM,CAAA;IAClD,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAA;IACrD,IAAI,KAAK,KAAK,SAAS;QAAE,YAAY,CAAC,YAAY,CAAC,GAAG,KAAK,CAAA;IAC3D,YAAY,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAA;IAEnC,MAAM,IAAI,GAA4B;QACpC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,eAAe,EAAE,OAAO,CAAC,cAAc,IAAI,EAAE;QAC7C,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;QAC3C,aAAa,EAAE,KAAK;QACpB,IAAI;QACJ,aAAa,EAAE,YAAY;KAC5B,CAAA;IACD,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,CAAA;IACpC,IAAI,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5B,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC9B,IAAI,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,IAAI,CAAA;YAChC,IAAI,CAAC,UAAU,CAAC,GAAG,OAAO,CAAC,QAAQ,IAAI,eAAe,CAAC,QAAQ,CAAC,CAAA;QAClE,CAAC;QACD,IAAI,QAAQ,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS;YAAE,IAAI,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,IAAI,CAAA;IACvF,CAAC;IACD,IAAI,cAAc,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC;QAAE,IAAI,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,IAAI,CAAA;IACxF,IAAI,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,eAAe;QAAE,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAA;IACnG,OAAO,IAAI,CAAA;AACb,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAChC,OAA6B,EAC7B,QAAiC,EACjC,IAAY,EACZ,MAAsB;IAEtB,MAAM,QAAQ,GAA4B,EAAE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAA;IACvE,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC,QAAQ,CAAA;IACvD,IAAI,SAAS,KAAK,SAAS;QAAE,QAAQ,CAAC,oBAAoB,CAAC,GAAG,SAAS,CAAA;IAEvE,MAAM,YAAY,GAA4B,EAAE,YAAY,EAAE,OAAO,CAAC,KAAK,EAAE,CAAA;IAC7E,MAAM,MAAM,GAAG,OAAO,CAAC,cAAc,IAAI,QAAQ,CAAC,cAAc,CAAA;IAChE,IAAI,MAAM;QAAE,YAAY,CAAC,eAAe,CAAC,GAAG,MAAM,CAAA;IAClD,MAAM,KAAK,GAAG,OAAO,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAA;IACrD,IAAI,KAAK,KAAK,SAAS;QAAE,YAAY,CAAC,YAAY,CAAC,GAAG,KAAK,CAAA;IAC3D,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,IAAI,QAAQ,CAAC,MAAM,CAAC,MAAM,KAAK,OAAO,CAAC,KAAK;QAC3E,YAAY,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAA;IACtD,YAAY,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAA;IAEnC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAA;IAC5B,MAAM,IAAI,GAA4B;QACpC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,eAAe,EAAE,OAAO,CAAC,cAAc,IAAI,EAAE;QAC7C,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,YAAY,EAAE,OAAO,CAAC,MAAM,IAAI,KAAK,EAAE,MAAM,IAAI,CAAC;QAClD,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,KAAK,EAAE,GAAG,IAAI,EAAE;QACpC,aAAa,EAAE,MAAM;QACrB,IAAI;QACJ,aAAa,EAAE,YAAY;KAC5B,CAAA;IACD,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS;QAAE,IAAI,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,IAAI,CAAA;IAC/D,IAAI,MAAM,CAAC,GAAG,KAAK,SAAS;QAAE,IAAI,CAAC,WAAW,CAAC,GAAG,MAAM,CAAC,GAAG,CAAA;IAC5D,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAW,CAAA;IAC7C,IAAI,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,MAAM,GAAG,6BAA6B;QACzE,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAA;IAC/C,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAS,eAAe,CAAC,IAAuB,EAAE,MAAc;IAC9D,IAAI,KAAyB,CAAA;IAC7B,IAAI,CAAC,GAAG,CAAC,CAAA;IACT,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAW,CAAA;QAC7B,IAAI,GAAG,KAAK,MAAM,EAAE,CAAC;YACnB,KAAK,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAA;YAC5B,CAAC,IAAI,CAAC,CAAA;YACN,SAAQ;QACV,CAAC;QACD,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,MAAM,GAAG,CAAC;YAAE,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;QACtE,CAAC,IAAI,CAAC,CAAA;IACR,CAAC;IACD,OAAO,KAAK,CAAA;AACd,CAAC;AAED,SAAS,OAAO,CAAC,KAAa,EAAE,KAAa;IAC3C,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,CAAA;IACxB,OAAO,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAA;AAChG,CAAC;AAED,qGAAqG;AACrG,MAAM,UAAU,oBAAoB,CAAC,IAAuB;IAC1D,MAAM,MAAM,GAAa,EAAE,CAAA;IAC3B,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAA;IACxD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,CAAA;QAC/D,MAAM,CAAC,IAAI,CAAC,SAAS,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;IAC3C,CAAC;IACD,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,EAAE,SAAS,CAAC,CAAA;IAC9C,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,EAAE,UAAU,CAAC,CAAA;IAChD,IAAI,KAAK,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS;QAC7C,MAAM,CAAC,IAAI,CAAC,QAAQ,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,CAAA;IAChE,MAAM,OAAO,GAA4B;QACvC,CAAC,OAAO,EAAE,SAAS,CAAC;QACpB,CAAC,MAAM,EAAE,QAAQ,CAAC;QAClB,CAAC,MAAM,EAAE,eAAe,CAAC;QACzB,CAAC,SAAS,EAAE,WAAW,CAAC;KACzB,CAAA;IACD,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;QACtC,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;QAC3C,IAAI,KAAK,KAAK,SAAS;YAAE,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,IAAI,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;IACxE,CAAC;IACD,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,EAAE,WAAW,CAAC,CAAA;IAClD,IAAI,OAAO,KAAK,SAAS;QAAE,MAAM,CAAC,IAAI,CAAC,WAAW,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,CAAA;IACzE,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AACzB,CAAC"}

package/dist/diffusion/compat.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Which installed engine build may serve a model family. `check_engine_compatibility` and
+ * `select_model_install` in `session.rs` of `tauri-plugin-atomic-diffusion` (app commit
+ * `ec1fd3ea7`); messages verbatim.
+ */
+import type { DiffusionBackendInstallRecord, DiffusionEngineId } from '../contracts/index.js';
+/** The first upstream build that runs Qwen Image 2.1 and Krea 2 Turbo. */
+export declare const MIN_MODERN_FAMILY_BUILD = 883;
+/**
+ * Qwen Image 2.1 and Krea 2 Turbo need build 883 or newer; a tag that does not parse fails closed.
+ * Every other family keeps whatever engine it has.
+ */
+export declare function checkEngineCompatibility(family: string, tag: string): void;
+/**
+ * The install a load uses. The newest install of `engine` picks the backend; among that backend's
+ * installs (newest first, as `listInstalledBackends` returns them) the first one compatible with
+ * `family` wins, so an older compatible tree is used rather than the newest incompatible one. The
+ * backend is never switched to find a compatible build.
+ */
+export declare function selectModelInstall(installed: readonly DiffusionBackendInstallRecord[], engine: DiffusionEngineId, family: string): DiffusionBackendInstallRecord;

package/dist/diffusion/compat.js

@@ -0,0 +1,58 @@
+/**
+ * Which installed engine build may serve a model family. `check_engine_compatibility` and
+ * `select_model_install` in `session.rs` of `tauri-plugin-atomic-diffusion` (app commit
+ * `ec1fd3ea7`); messages verbatim.
+ */
+import { diffusionError } from './errors.js';
+/** The first upstream build that runs Qwen Image 2.1 and Krea 2 Turbo. */
+export const MIN_MODERN_FAMILY_BUILD = 883;
+const MODERN_FAMILIES = new Set(['qwen-image-2.1', 'krea-2-turbo']);
+/** `master-<build>-<hash>` → the build number; anything else → undefined. */
+function buildOf(tag) {
+    if (!tag.startsWith('master-'))
+        return undefined;
+    const rest = tag.slice('master-'.length);
+    const dash = rest.indexOf('-');
+    if (dash < 0 || dash === rest.length - 1)
+        return undefined;
+    const build = rest.slice(0, dash);
+    return /^\d+$/.test(build) ? Number(build) : undefined;
+}
+/**
+ * Qwen Image 2.1 and Krea 2 Turbo need build 883 or newer; a tag that does not parse fails closed.
+ * Every other family keeps whatever engine it has.
+ */
+export function checkEngineCompatibility(family, tag) {
+    if (!MODERN_FAMILIES.has(family))
+        return;
+    const build = buildOf(tag);
+    if (build !== undefined && build >= MIN_MODERN_FAMILY_BUILD)
+        return;
+    throw diffusionError('ENGINE_UPDATE_REQUIRED', `${family} requires an image engine update. Update the engine, then retry loading the model.`, `installed=${tag}; required=master-883-137f740 or newer`);
+}
+/**
+ * The install a load uses. The newest install of `engine` picks the backend; among that backend's
+ * installs (newest first, as `listInstalledBackends` returns them) the first one compatible with
+ * `family` wins, so an older compatible tree is used rather than the newest incompatible one. The
+ * backend is never switched to find a compatible build.
+ */
+export function selectModelInstall(installed, engine, family) {
+    const selected = installed.find((r) => r.engine === engine);
+    if (!selected)
+        throw diffusionError('ENGINE_MISSING', 'Install the image engine first.');
+    const record = installed.find((candidate) => candidate.engine === engine &&
+        candidate.backendId === selected.backendId &&
+        isCompatible(family, candidate.tag)) ?? selected;
+    checkEngineCompatibility(family, record.tag);
+    return record;
+}
+function isCompatible(family, tag) {
+    try {
+        checkEngineCompatibility(family, tag);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=compat.js.map

package/dist/diffusion/compat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compat.js","sourceRoot":"","sources":["../../src/diffusion/compat.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAE5C,0EAA0E;AAC1E,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAG,CAAA;AAC1C,MAAM,eAAe,GAAwB,IAAI,GAAG,CAAC,CAAC,gBAAgB,EAAE,cAAc,CAAC,CAAC,CAAA;AAExF,6EAA6E;AAC7E,SAAS,OAAO,CAAC,GAAW;IAC1B,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAA;IAChD,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;IACxC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAA;IAC9B,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,KAAK,IAAI,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,SAAS,CAAA;IAC1D,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAA;IACjC,OAAO,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;AACxD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,MAAc,EAAE,GAAW;IAClE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC;QAAE,OAAM;IACxC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAA;IAC1B,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,IAAI,uBAAuB;QAAE,OAAM;IACnE,MAAM,cAAc,CAClB,wBAAwB,EACxB,GAAG,MAAM,oFAAoF,EAC7F,aAAa,GAAG,wCAAwC,CACzD,CAAA;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAChC,SAAmD,EACnD,MAAyB,EACzB,MAAc;IAEd,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAA;IAC3D,IAAI,CAAC,QAAQ;QAAE,MAAM,cAAc,CAAC,gBAAgB,EAAE,iCAAiC,CAAC,CAAA;IACxF,MAAM,MAAM,GACV,SAAS,CAAC,IAAI,CACZ,CAAC,SAAS,EAAE,EAAE,CACZ,SAAS,CAAC,MAAM,KAAK,MAAM;QAC3B,SAAS,CAAC,SAAS,KAAK,QAAQ,CAAC,SAAS;QAC1C,YAAY,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,CACtC,IAAI,QAAQ,CAAA;IACf,wBAAwB,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAA;IAC5C,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,YAAY,CAAC,MAAc,EAAE,GAAW;IAC/C,IAAI,CAAC;QACH,wBAAwB,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;QACrC,OAAO,IAAI,CAAA;IACb,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAA;IACd,CAAC;AACH,CAAC"}

package/dist/diffusion/containment.d.ts

@@ -0,0 +1,15 @@
+/**
+ * "Is this path really inside that folder?" for everything the diffusion surface deletes or writes
+ * on a caller's word: engine trees, model files, gallery images. `is_within` and the
+ * `canonicalize_existing_prefix` callers of `install.rs` (app commit `767ff6350`).
+ *
+ * Both sides are resolved through symlinks as far as they exist first (on Fedora Silverblue `/home`
+ * is a link to `/var/home`), then compared component-wise, ignoring case on Windows.
+ */
+/** Lexical containment of two absolute, already-resolved paths. A path is within itself. */
+export declare function isWithin(path: string, root: string, platform?: NodeJS.Platform): boolean;
+export type Containment = 'inside' | 'root' | 'outside';
+/** Where `path` lies relative to `root`, once the existing part of both is resolved through symlinks. */
+export declare function locate(path: string, root: string, platform?: NodeJS.Platform): Promise<Containment>;
+/** Whether two spellings name the same directory or file. */
+export declare function samePath(a: string, b: string, platform?: NodeJS.Platform): Promise<boolean>;

package/dist/diffusion/containment.js

@@ -0,0 +1,32 @@
+/**
+ * "Is this path really inside that folder?" for everything the diffusion surface deletes or writes
+ * on a caller's word: engine trees, model files, gallery images. `is_within` and the
+ * `canonicalize_existing_prefix` callers of `install.rs` (app commit `767ff6350`).
+ *
+ * Both sides are resolved through symlinks as far as they exist first (on Fedora Silverblue `/home`
+ * is a link to `/var/home`), then compared component-wise, ignoring case on Windows.
+ */
+import { isAbsolute, relative, resolve, sep } from 'node:path';
+import { canonicalizeExistingPrefix } from '../downloads/index.js';
+const fold = (path, platform) => platform === 'win32' ? path.toLowerCase() : path;
+/** Lexical containment of two absolute, already-resolved paths. A path is within itself. */
+export function isWithin(path, root, platform = process.platform) {
+    const rel = relative(fold(resolve(root), platform), fold(resolve(path), platform));
+    return rel === '' || (rel !== '..' && !rel.startsWith(`..${sep}`) && !isAbsolute(rel));
+}
+/** Where `path` lies relative to `root`, once the existing part of both is resolved through symlinks. */
+export async function locate(path, root, platform = process.platform) {
+    const [resolvedPath, resolvedRoot] = await Promise.all([
+        canonicalizeExistingPrefix(path),
+        canonicalizeExistingPrefix(root),
+    ]);
+    if (fold(resolvedPath, platform) === fold(resolvedRoot, platform))
+        return 'root';
+    return isWithin(resolvedPath, resolvedRoot, platform) ? 'inside' : 'outside';
+}
+/** Whether two spellings name the same directory or file. */
+export async function samePath(a, b, platform = process.platform) {
+    const [left, right] = await Promise.all([canonicalizeExistingPrefix(a), canonicalizeExistingPrefix(b)]);
+    return fold(left, platform) === fold(right, platform);
+}
+//# sourceMappingURL=containment.js.map

package/dist/diffusion/containment.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"containment.js","sourceRoot":"","sources":["../../src/diffusion/containment.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAA;AAC9D,OAAO,EAAE,0BAA0B,EAAE,MAAM,uBAAuB,CAAA;AAElE,MAAM,IAAI,GAAG,CAAC,IAAY,EAAE,QAAyB,EAAU,EAAE,CAC/D,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAA;AAElD,4FAA4F;AAC5F,MAAM,UAAU,QAAQ,CAAC,IAAY,EAAE,IAAY,EAAE,WAA4B,OAAO,CAAC,QAAQ;IAC/F,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAA;IAClF,OAAO,GAAG,KAAK,EAAE,IAAI,CAAC,GAAG,KAAK,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAA;AACxF,CAAC;AAID,yGAAyG;AACzG,MAAM,CAAC,KAAK,UAAU,MAAM,CAC1B,IAAY,EACZ,IAAY,EACZ,WAA4B,OAAO,CAAC,QAAQ;IAE5C,MAAM,CAAC,YAAY,EAAE,YAAY,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QACrD,0BAA0B,CAAC,IAAI,CAAC;QAChC,0BAA0B,CAAC,IAAI,CAAC;KACjC,CAAC,CAAA;IACF,IAAI,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,KAAK,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC;QAAE,OAAO,MAAM,CAAA;IAChF,OAAO,QAAQ,CAAC,YAAY,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAA;AAC9E,CAAC;AAED,6DAA6D;AAC7D,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,CAAS,EACT,CAAS,EACT,WAA4B,OAAO,CAAC,QAAQ;IAE5C,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,EAAE,0BAA0B,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IACvG,OAAO,IAAI,CAAC,IAAI,EAAE,QAAQ,CAAC,KAAK,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;AACvD,CAAC"}

package/dist/diffusion/errors.d.ts

@@ -0,0 +1,28 @@
+/**
+ * Image-generation failures. Port of `error.rs` in `tauri-plugin-atomic-diffusion` (app commit
+ * `767ff6350`): the same codes and messages, carried by `AtomicCoreError` so the control server's
+ * envelope and the job records show one shape.
+ */
+import { AtomicCoreError } from '../contracts/index.js';
+import type { DiffusionErrorBody, DiffusionErrorCode } from '../contracts/index.js';
+export declare function diffusionError(code: DiffusionErrorCode, message: string, details?: string): AtomicCoreError;
+export declare function internalError(message: string, details?: string): AtomicCoreError;
+export declare function notConfiguredError(): AtomicCoreError;
+export declare function modelNotLoadedError(): AtomicCoreError;
+export declare function cancelledError(): AtomicCoreError;
+/** libuv reports a full disk as `ENOSPC` everywhere, Windows' two "disk full" codes included. */
+export declare function isDiskFull(error: unknown): boolean;
+/**
+ * An I/O failure on the closest code: a full disk is something the user can act on, everything else
+ * is internal, with `context` as the message.
+ */
+export declare function ioError(context: string, error: unknown): AtomicCoreError;
+export declare function isDiffusionErrorCode(code: unknown): code is DiffusionErrorCode;
+/**
+ * Any failure as a job record or an event carries it. A code from outside this surface (an
+ * `IO_ERROR` from a shared helper, a plain `Error`) becomes `INTERNAL`, which is what the web app
+ * knows how to show; its original code is kept in the details.
+ */
+export declare function errorBody(error: unknown): DiffusionErrorBody;
+/** The thrown form of `errorBody`: what a caller awaiting the operation receives. */
+export declare function toDiffusionError(error: unknown): AtomicCoreError;

package/dist/diffusion/errors.js

@@ -0,0 +1,63 @@
+/**
+ * Image-generation failures. Port of `error.rs` in `tauri-plugin-atomic-diffusion` (app commit
+ * `767ff6350`): the same codes and messages, carried by `AtomicCoreError` so the control server's
+ * envelope and the job records show one shape.
+ */
+import { AtomicCoreError, DIFFUSION_ERROR_CODES } from '../contracts/index.js';
+export function diffusionError(code, message, details) {
+    return new AtomicCoreError(code, message, details);
+}
+export function internalError(message, details) {
+    return diffusionError('INTERNAL', message, details);
+}
+export function notConfiguredError() {
+    return diffusionError('NOT_CONFIGURED', 'Image generation has not been configured yet.');
+}
+export function modelNotLoadedError() {
+    return diffusionError('MODEL_NOT_LOADED', 'Load an image model first.');
+}
+export function cancelledError() {
+    return diffusionError('CANCELLED', 'Generation was cancelled.');
+}
+/** libuv reports a full disk as `ENOSPC` everywhere, Windows' two "disk full" codes included. */
+export function isDiskFull(error) {
+    return error?.code === 'ENOSPC';
+}
+/**
+ * An I/O failure on the closest code: a full disk is something the user can act on, everything else
+ * is internal, with `context` as the message.
+ */
+export function ioError(context, error) {
+    const reason = error instanceof Error ? error.message : String(error);
+    if (isDiskFull(error))
+        return diffusionError('DISK_FULL', 'The disk is full.', `${context}: ${reason}`);
+    return internalError(context, reason);
+}
+export function isDiffusionErrorCode(code) {
+    return typeof code === 'string' && DIFFUSION_ERROR_CODES.includes(code);
+}
+/**
+ * Any failure as a job record or an event carries it. A code from outside this surface (an
+ * `IO_ERROR` from a shared helper, a plain `Error`) becomes `INTERNAL`, which is what the web app
+ * knows how to show; its original code is kept in the details.
+ */
+export function errorBody(error) {
+    if (error instanceof AtomicCoreError) {
+        if (isDiffusionErrorCode(error.code)) {
+            return error.details === undefined
+                ? { code: error.code, message: error.message }
+                : { code: error.code, message: error.message, details: error.details };
+        }
+        const details = error.details === undefined ? error.code : `${error.code}: ${error.details}`;
+        return { code: 'INTERNAL', message: error.message, details };
+    }
+    return { code: 'INTERNAL', message: error instanceof Error ? error.message : String(error) };
+}
+/** The thrown form of `errorBody`: what a caller awaiting the operation receives. */
+export function toDiffusionError(error) {
+    if (error instanceof AtomicCoreError && isDiffusionErrorCode(error.code))
+        return error;
+    const body = errorBody(error);
+    return diffusionError(body.code, body.message, body.details);
+}
+//# sourceMappingURL=errors.js.map

package/dist/diffusion/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/diffusion/errors.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAA;AAG9E,MAAM,UAAU,cAAc,CAAC,IAAwB,EAAE,OAAe,EAAE,OAAgB;IACxF,OAAO,IAAI,eAAe,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;AACpD,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,OAAe,EAAE,OAAgB;IAC7D,OAAO,cAAc,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;AACrD,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,OAAO,cAAc,CAAC,gBAAgB,EAAE,+CAA+C,CAAC,CAAA;AAC1F,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,OAAO,cAAc,CAAC,kBAAkB,EAAE,4BAA4B,CAAC,CAAA;AACzE,CAAC;AAED,MAAM,UAAU,cAAc;IAC5B,OAAO,cAAc,CAAC,WAAW,EAAE,2BAA2B,CAAC,CAAA;AACjE,CAAC;AAED,iGAAiG;AACjG,MAAM,UAAU,UAAU,CAAC,KAAc;IACvC,OAAQ,KAA2C,EAAE,IAAI,KAAK,QAAQ,CAAA;AACxE,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,OAAO,CAAC,OAAe,EAAE,KAAc;IACrD,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IACrE,IAAI,UAAU,CAAC,KAAK,CAAC;QAAE,OAAO,cAAc,CAAC,WAAW,EAAE,mBAAmB,EAAE,GAAG,OAAO,KAAK,MAAM,EAAE,CAAC,CAAA;IACvG,OAAO,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,CAAA;AACvC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,IAAa;IAChD,OAAO,OAAO,IAAI,KAAK,QAAQ,IAAK,qBAA2C,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;AAChG,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,SAAS,CAAC,KAAc;IACtC,IAAI,KAAK,YAAY,eAAe,EAAE,CAAC;QACrC,IAAI,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,OAAO,KAAK,CAAC,OAAO,KAAK,SAAS;gBAChC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE;gBAC9C,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAA;QAC1E,CAAC;QACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAA;QAC5F,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,CAAA;IAC9D,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAA;AAC9F,CAAC;AAED,qFAAqF;AACrF,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,IAAI,KAAK,YAAY,eAAe,IAAI,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC;QAAE,OAAO,KAAK,CAAA;IACtF,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,CAAA;IAC7B,OAAO,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;AAC9D,CAAC"}

package/dist/diffusion/gallery.d.ts

@@ -0,0 +1,76 @@
+/**
+ * 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 type { GalleryFlags, GalleryImageItem, GalleryListOptions, GalleryPage, ImageRecipe } from '../contracts/index.js';
+export declare const FLAGS_FILE = ".flags.json";
+export declare const THUMB_EDGE = 256;
+/** A listing checks for blank frames only in files this small; real outputs are far larger. */
+export declare const BLANK_SCAN_LIMIT: number;
+/**
+ * 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 declare function isBlankOutput(png: Buffer): Promise<boolean>;
+export interface FlagEntry {
+    pinned: boolean;
+    archived: boolean;
+}
+export type FlagMap = Map<string, FlagEntry>;
+/** `^[a-f0-9]{32}-\d{2}$` */
+export declare function isValidId(id: string): boolean;
+export declare function makeId(jobId: string, index: number): string;
+export declare function pngPath(dir: string, id: string): string;
+export declare function thumbPath(dir: string, id: string): string;
+export declare function readFlags(dir: string): Promise<FlagMap>;
+/** What a failed write is called; the video gallery names its own folder. */
+export interface WriteMessages {
+    write: string;
+    finish: string;
+}
+/** Write a file through a hidden temporary sibling, so a reader never sees half of it. */
+export declare function writeAtomic(target: string, bytes: Buffer | string, messages?: WriteMessages): Promise<void>;
+export declare function writeFlags(dir: string, flags: FlagMap): Promise<void>;
+/** A 256 px PNG beside the image. Fails for a PNG flavour the decoder does not read; the caller goes without. */
+export declare function writeThumbnail(png: Buffer, path: string): Promise<void>;
+export interface GalleryLogger {
+    (level: 'warn', msg: string): void;
+}
+/**
+ * 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 declare class Gallery {
+    private readonly log;
+    private readonly flagsLock;
+    /** The last verdict per small file, so a listing decodes each thumbnail once rather than every time. */
+    private readonly blankFiles;
+    private readonly isBlank;
+    constructor(log?: GalleryLogger);
+    /**
+     * 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.
+     */
+    private isBlankItem;
+    /**
+     * Splice the recipe in, write `<id>.png`, write the thumbnail, and return the item with the final
+     * bytes (the OpenAI facade answers with them).
+     */
+    save(dir: string, recipe: ImageRecipe, png: Buffer): Promise<{
+        item: GalleryImageItem;
+        bytes: Buffer;
+    }>;
+    get(dir: string, id: string): Promise<GalleryImageItem | null>;
+    list(dir: string, options: GalleryListOptions): Promise<GalleryPage>;
+    /** Remove each PNG, its thumbnail and its flags entry. Foreign PNGs are left alone. */
+    delete(dir: string, ids: readonly string[]): Promise<void>;
+    setFlags(dir: string, id: string, update: GalleryFlags): Promise<GalleryImageItem>;
+    /** Byte-for-byte copy, keeping the embedded recipe. */
+    export(dir: string, id: string, target: string): Promise<void>;
+}