pixelkiln 0.31.0 → 0.32.0

package/dist/cli.js

@@ -6431,7 +6431,7 @@ async function poll(provider, lock, lockPath, opts = {}) {
 }
 
 // src/pipeline/fetch.ts
-import { existsSync as existsSync10 } from "fs";
+import { existsSync as existsSync10, readFileSync as readFileSync4 } from "fs";
 import { mkdir as mkdir3, readFile as readFile7, rename as rename3, rm as rm4, writeFile as writeFile3 } from "fs/promises";
 import path14 from "path";
 
@@ -6487,7 +6487,12 @@ async function fetchAssets(provider, specs, lock, lockPath, opts = {}) {
     if (opts.refresh) return e.outputs.length > 0 && Boolean(e.sourceUrls?.length || e.sourceUrl);
     if (!opts.repair) return false;
     const spec = specByKey.get(key);
-    return e.outputs.length === 0 || !spec || e.outputs.some((output) => !existsSync10(resolveOutputPath(output.path, spec.root)));
+    if (e.outputs.length === 0 || !spec) return true;
+    return e.outputs.some((output) => {
+      const file = resolveOutputPath(output.path, spec.root);
+      if (!existsSync10(file)) return true;
+      return Boolean(opts.force) && sha256(readFileSync4(file)) !== output.sha256;
+    });
   });
   const concurrency = Math.min(Math.max(1, opts.concurrency ?? 8), Math.max(1, pending.length));
   let cursor = 0;
@@ -9412,6 +9417,8 @@ function renderGallery(snapshot, opts = {}) {
   .tray > button { padding:5px 11px; font-size:12.5px; }
   @media (max-width: 900px) { .sheet.review-host, .sheet.compare-host, .sheet.editor-host { width:100vw; } }
   .drawer .gen { display:flex; gap:8px; flex-wrap:wrap; margin-top:12px; }
+  .drawer .readonly-hint { margin-top:10px; font-size:12.5px; line-height:1.5; }
+  .drawer .readonly-hint code { font:12px var(--mono); color:var(--text); }
   .pair { display:grid; grid-template-columns:1fr 1fr; gap:8px; margin:8px 0 10px; }
   .pair figure { margin:0; border:1px solid var(--line); background:var(--panel-deep); background-image:var(--checker);
     background-size:12px 12px; background-position:0 0,6px 6px; display:grid; place-items:center; padding:8px; min-height:96px; }
@@ -9621,7 +9628,7 @@ function renderJobs() {
     const row = el('div', 'job');
     const ph = el('span', 'ph');
     ph.append(el('i', 'dot ' + PHASE_TONE[job.phase]), document.createTextNode(job.phase));
-    const what = (job.mode === 'refresh' ? 'pull upstream for ' : job.mode === 'resume' ? 'resume ' : 'generate ') + job.keys.length + (job.keys.length === 1 ? ' asset' : ' assets') +
+    const what = (job.mode === 'refresh' ? 'pull upstream for ' : job.mode === 'restore' ? 'restore ' : job.mode === 'resume' ? 'resume ' : 'generate ') + job.keys.length + (job.keys.length === 1 ? ' asset' : ' assets') +
       (job.project ? ' in ' + job.project : '');
     const last = el('span', 'last', what + (job.messages.length ? ' \u2014 ' + job.messages[job.messages.length - 1].trim() : ''));
     last.title = job.keys.join('\\n');
@@ -9746,13 +9753,16 @@ function budgetLine() {
 
 // The confirm step: what will be sent, what it is estimated to cost, and what
 // this session may still spend. Mirrors gen's "Spend \u2026 on N asset(s)?" prompt.
-function generateDialog(items, { project = null, force = false, resume = false, refresh = false } = {}) {
+function generateDialog(items, { project = null, force = false, resume = false, refresh = false, restore = false } = {}) {
   const host = $('dialog-host');
   host.textContent = '';
   const wrap = el('div', 'dialog');
   const form = el('form');
   const noun = items.length + (items.length === 1 ? ' asset' : ' assets');
+  const free = resume || refresh || restore;
+  const replacing = items.filter((i) => i.state === 'orphaned' && i.outputs.every((o) => o.exists) || i.state === 'untracked');
   form.append(el('h3', null, refresh ? 'Pull upstream changes for ' + noun
+    : restore ? 'Restore ' + noun
     : resume ? 'Resume ' + noun
     : (force ? 'Regenerate ' : 'Generate ') + noun));
   const table = el('table');
@@ -9762,9 +9772,9 @@ function generateDialog(items, { project = null, force = false, resume = false,
   for (const item of items) {
     const tr = el('tr');
     tr.append(el('td', 'mono', item.key), el('td', null, item.state), el('td', 'n', item.candidates ?? '\u2014'),
-      el('td', 'n', resume || refresh ? 'no cost' : fmtCost(item.costUnit, item.estimatedCost ?? 0)));
+      el('td', 'n', free ? 'no cost' : fmtCost(item.costUnit, item.estimatedCost ?? 0)));
     table.append(tr);
-    if (!resume && !refresh) {
+    if (!free) {
       const g = byProvider.get(item.provider) || { cost: 0, unit: item.costUnit };
       g.cost += item.estimatedCost ?? 0; byProvider.set(item.provider, g);
     }
@@ -9773,6 +9783,9 @@ function generateDialog(items, { project = null, force = false, resume = false,
   const sum = el('div', 'sum');
   if (refresh) {
     sum.append(el('div', null, 'Re-downloads each object from the provider and replaces the local file only when the object changed upstream \u2014 for example after editing it in the provider\u2019s own editor. Unchanged objects are left alone; a file you changed locally is refused. Nothing is submitted.'));
+  } else if (restore) {
+    sum.append(el('div', null, 'Puts the recorded bytes back on disk from the local cache or the provider object. Nothing is generated or submitted.'));
+    if (force && replacing.length) sum.append(el('div', 'warn', 'The file on disk was changed after PixelKiln wrote it. Restoring discards that change. To keep it instead, cancel and use Edit by hand or Edit in browser first: the edit is copied beside the generated file, and the record stays intact.'));
   } else if (resume) {
     sum.append(el('div', null, 'Polls, reviews, and downloads existing provider work. Nothing is submitted.'));
   } else {
@@ -9785,12 +9798,13 @@ function generateDialog(items, { project = null, force = false, resume = false,
       if (left !== null && g.cost > left) line.className = 'warn';
       sum.append(line);
     }
-    if (force) sum.append(el('div', 'warn', 'Regenerating replaces the current art. The previous file is replaced when the new result is fetched; the provider objects it came from are not deleted.'));
+    if (force && replacing.length) sum.append(el('div', 'warn', (replacing.length === 1 ? 'This file' : replacing.length + ' of these files') + ' changed on disk after PixelKiln wrote it (or was never recorded). Generating replaces it and that change is lost. To keep it, cancel and use Edit by hand or Edit in browser first.'));
+    else if (force) sum.append(el('div', 'warn', 'Regenerating replaces the current art. The previous file is replaced when the new result is fetched; the provider objects it came from are not deleted.'));
     sum.append(el('div', null, 'Candidate sets land in review; you choose from them here before anything is downloaded.'));
   }
   form.append(sum);
   const actions = el('div', 'actions');
-  const go = el('button', 'primary', refresh ? 'Pull changes' : resume ? 'Resume' : 'Generate');
+  const go = el('button', 'primary', refresh ? 'Pull changes' : restore ? (force && replacing.length ? 'Restore and discard my change' : 'Restore') : resume ? 'Resume' : force && replacing.length ? 'Generate and replace' : 'Generate');
   go.type = 'submit';
   const cancel = el('button', null, 'Cancel'); cancel.type = 'button'; cancel.onclick = () => { host.textContent = ''; };
   const msg = el('span', 'msg');
@@ -9805,6 +9819,7 @@ function generateDialog(items, { project = null, force = false, resume = false,
       if (force) body.force = true;
       if (resume) body.resume = true;
       if (refresh) body.refresh = true;
+      if (restore) body.restore = true;
       const job = await postGenerate(body);
       host.textContent = '';
       GEN.jobs.unshift(job);
@@ -10463,6 +10478,13 @@ function generateActions(item) {
   };
   if (item.state === 'missing' || item.state === 'stale' || item.state === 'failed') add('Generate' + cost, 'primary', {});
   else if (item.state === 'ok') add('Regenerate' + cost, null, { force: true });
+  else if (item.state === 'orphaned' && item.outputs.every((o) => o.exists)) {
+    // The file changed after download. Putting the recorded bytes back and
+    // regenerating both discard that change, so both say so and ask.
+    add('Restore recorded bytes \xB7 no cost', null, { restore: true, force: true });
+    add('Regenerate' + cost, null, { force: true });
+  } else if (item.state === 'orphaned') add('Restore \xB7 no cost', 'primary', { restore: true });
+  else if (item.state === 'untracked') add('Generate' + cost + ' \xB7 replaces the file', null, { force: true });
   if (item.state === 'in-flight' || item.state === 'recoverable') {
     // A job already parked this asset in review: go straight to its sheet.
     const holder = GEN.jobs.find((job) => job.phase === 'review' && job.project === item.project && job.review.includes(item.key));
@@ -11153,6 +11175,15 @@ function renderDrawer() {
   if (ui.notice && ui.notice.id === item.id) body.append(el('div', 'notice', ui.notice.text));
   body.append(stateNode(item.state, item.reason));
   if (GENERATION) body.append(generateActions(item));
+  // Writes are opt-in flags on the command line; say so where the buttons would be.
+  if ((!EDITABLE || !GENERATION) && item.declared) {
+    const missing = [];
+    if (!EDITABLE) missing.push('--edit to change its prompt, size, and tags or open it in the browser editor');
+    if (!GENERATION) missing.push('--budget <n> to generate or regenerate it from here');
+    const hint = el('div', 'state-dim readonly-hint');
+    hint.append(document.createTextNode('This gallery is read-only. Restart it with '), el('code', null, 'pixelkiln gallery'), document.createTextNode(' plus ' + missing.join(', and ') + '.'));
+    body.append(hint);
+  }
   if (canEdit && ui.editing === item.id) body.append(editForm(item));
   // Plan already quotes the error as the reason for a failed entry; only a
   // stale or superseded failure needs its own line.
@@ -11743,13 +11774,21 @@ import { z as z8 } from "zod";
 var GenerateRequestSchema = z8.object({
   project: z8.string().min(1).optional(),
   keys: z8.array(z8.string().min(1)).min(1).max(500),
-  /** Regenerate work that is up to date, as `gen --force` does. */
+  /**
+   * Regenerate work that is up to date, as `gen --force` does, and let the
+   * fetch replace a locally modified or untracked file. With `restore`, put
+   * the recorded bytes back over a modified file.
+   */
   force: z8.boolean().optional(),
   /** Advance in-flight, review, or selected work without submitting anything. */
   resume: z8.boolean().optional(),
   /** Re-download downloaded outputs and replace files whose object changed upstream. */
-  refresh: z8.boolean().optional()
-}).strict().refine((request) => !(request.resume && request.refresh), { message: "resume and refresh are exclusive" });
+  refresh: z8.boolean().optional(),
+  /** Put recorded outputs back on disk from the cache or the provider, without generating. */
+  restore: z8.boolean().optional()
+}).strict().refine((request) => [request.resume, request.refresh, request.restore].filter(Boolean).length <= 1, {
+  message: "resume, refresh, and restore are exclusive"
+});
 var GenerateRequestError = class extends Error {
   constructor(message6, status = 400) {
     super(message6);
@@ -11837,7 +11876,9 @@ function createGenerateHandlers(opts) {
     for (const [providerId, providerSpecs] of groupByRecordedProvider(specs, ctx.lock)) {
       const provider = opts.providerFor(job.project ?? void 0, providerId);
       const res = await fetchAssets(provider, providerSpecs, ctx.lock, ctx.lockPath, {
-        onProgress: (line) => say(job, line)
+        onProgress: (line) => say(job, line),
+        // A forced regeneration replaces whatever is on disk, as `gen --force`.
+        force: job.force
       });
       job.counts.downloaded += res.downloaded;
       if (res.downloaded || res.failed) {
@@ -11852,12 +11893,32 @@ function createGenerateHandlers(opts) {
       job.finishedAt = now().toISOString();
     }
   }
+  async function restore(job, ctx, specs) {
+    job.phase = "fetching";
+    for (const [providerId, providerSpecs] of groupByRecordedProvider(specs, ctx.lock)) {
+      const provider = opts.providerFor(job.project ?? void 0, providerId);
+      const res = await fetchAssets(provider, providerSpecs, ctx.lock, ctx.lockPath, {
+        onProgress: (line) => say(job, line),
+        repair: true,
+        force: job.force
+      });
+      job.counts.downloaded += res.downloaded;
+      say(job, `${providerId}: restored ${res.downloaded}, skipped ${res.skipped}, failed ${res.failed}`);
+      if (res.failed) throw new Error(`${res.failed} file(s) could not be restored`);
+    }
+    job.phase = "done";
+    job.finishedAt = now().toISOString();
+  }
   async function run(job, ctx, specs, groups) {
     try {
       if (job.mode === "refresh") {
         await refresh(job, ctx, specs);
         return;
       }
+      if (job.mode === "restore") {
+        await restore(job, ctx, specs);
+        return;
+      }
       if (job.mode === "generate") {
         job.phase = "submitting";
         for (const group of groups) {
@@ -11920,8 +11981,16 @@ function createGenerateHandlers(opts) {
         const holder = busy.get(`${request.project ?? ""}:${key}`);
         if (holder) throw new GenerateRequestError(`"${key}" is already being worked on by job ${holder}`, 409);
       }
+      if (request.restore) {
+        for (const key of request.keys) {
+          const entry = ctx.lock.entries[key];
+          if (!entry || entry.status !== "downloaded" || !entry.outputs.length) {
+            throw new GenerateRequestError(`"${key}" has no downloaded output recorded to restore`);
+          }
+        }
+      }
       let groups = [];
-      if (!request.resume && !request.refresh) {
+      if (!request.resume && !request.refresh && !request.restore) {
         const plan = await buildPlan(specs, ctx.lock, { force: request.force });
         if (!plan.actionable.length) {
           const reasons = plan.items.map((item) => `${item.key}: ${item.state} \u2014 ${item.reason}`).slice(0, 5);
@@ -11941,7 +12010,8 @@ function createGenerateHandlers(opts) {
         id: randomBytes2(8).toString("hex"),
         project: request.project ?? null,
         keys: request.keys,
-        mode: request.refresh ? "refresh" : request.resume ? "resume" : "generate",
+        mode: request.refresh ? "refresh" : request.resume ? "resume" : request.restore ? "restore" : "generate",
+        force: Boolean(request.force),
         phase: "queued",
         startedAt: now().toISOString(),
         finishedAt: null,
@@ -11953,7 +12023,7 @@ function createGenerateHandlers(opts) {
       };
       jobs.set(job.id, job);
       contexts.set(job.id, ctx);
-      say(job, request.refresh ? `checking ${request.keys.length} asset(s) upstream at no cost` : request.resume ? `resuming ${request.keys.length} asset(s) at no cost` : `generating ${groups.reduce((n, g) => n + g.actionable.length, 0)} asset(s): ` + groups.map((g) => `${g.provider} ${formatCost(g.costUnit, g.cost)}`).join("; "));
+      say(job, request.refresh ? `checking ${request.keys.length} asset(s) upstream at no cost` : request.restore ? `restoring ${request.keys.length} asset(s) at no cost${request.force ? " (replacing modified files)" : ""}` : request.resume ? `resuming ${request.keys.length} asset(s) at no cost` : `generating ${groups.reduce((n, g) => n + g.actionable.length, 0)} asset(s): ` + groups.map((g) => `${g.provider} ${formatCost(g.costUnit, g.cost)}`).join("; "));
       void run(job, ctx, specs, groups);
       return job;
     },
@@ -14016,7 +14086,7 @@ function announceGalleryReady(url, count, stdout = process.stdout, stderr = proc
   gallery of ${count} generation${count === 1 ? "" : "s"}: ${url}
 ` + (notes.length ? `  (${notes.join("; ")}; Ctrl+C to stop)
 
-` : "  (read-only; press Ctrl+C to stop)\n\n")
+` : "  (read-only \u2014 add --edit to change prompts or edit sprites, --budget <n> to generate; Ctrl+C to stop)\n\n")
   );
 }
 function sessionBudget(args) {