initial commit

server/migrate.ts

@@ -0,0 +1,80 @@
+// Однократный перенос старых прогонов text2img в ImageBatches.
+// ComfyUI (SaveImage) записывает в каждый PNG граф запуска (текстовый чанк "prompt") — из него
+// восстанавливается пользовательский ввод; картинки одного батча — подряд идущие файлы с одинаковым графом.
+import fs from 'node:fs';
+import path from 'node:path';
+import zlib from 'node:zlib';
+import { COMFY_OUTPUT } from './config.ts';
+import { attachImages, batchOfImage, getMeta, insertTextBatch, setMeta, transaction } from './db.ts';
+import { inputFromGraph } from './workflows.ts';
+
+const KEY = 'migration.imageBatches.v1';
+const SRC_DIR = 'img2glb_src';
+/** Картинки одного батча сохраняются почти одновременно. */
+const SAME_BATCH_MS = 15_000;
+
+/** Текстовые чанки PNG (tEXt, iTXt, zTXt). */
+function pngTexts(file: string): Record<string, string> {
+  const b = fs.readFileSync(file);
+  const out: Record<string, string> = {};
+  let o = 8;
+  while (o + 8 <= b.length) {
+    const len = b.readUInt32BE(o);
+    const type = b.toString('latin1', o + 4, o + 8);
+    const d = b.subarray(o + 8, o + 8 + len);
+    const nul = d.indexOf(0);
+    if (type === 'tEXt') {
+      out[d.toString('latin1', 0, nul)] = d.toString('latin1', nul + 1);
+    } else if (type === 'zTXt') {
+      out[d.toString('latin1', 0, nul)] = zlib.inflateSync(d.subarray(nul + 2)).toString('latin1');
+    } else if (type === 'iTXt') {
+      const compressed = d[nul + 1] === 1;
+      let p = d.indexOf(0, nul + 3) + 1; // после языка
+      p = d.indexOf(0, p) + 1;           // после переведённого ключа
+      const text = d.subarray(p);
+      out[d.toString('latin1', 0, nul)] = (compressed ? zlib.inflateSync(text) : text).toString('utf8');
+    } else if (type === 'IEND') {
+      break;
+    }
+    o += 12 + len;
+  }
+  return out;
+}
+
+export function migrateOldRuns() {
+  if (getMeta(KEY)) return;
+  const dir = path.join(COMFY_OUTPUT, SRC_DIR);
+  let names: string[] = [];
+  try { names = fs.readdirSync(dir).filter(n => /\.png$/i.test(n)); } catch { /* папки нет — переносить нечего */ }
+
+  const files = names
+    .map(name => ({ name, mtime: fs.statSync(path.join(dir, name)).mtimeMs }))
+    .filter(f => batchOfImage({ filename: f.name, subfolder: SRC_DIR, type: 'output' }) == null)
+    .sort((a, b) => a.mtime - b.mtime || a.name.localeCompare(b.name));
+
+  // группировка: одинаковый граф, подряд, в пределах SAME_BATCH_MS от первой картинки
+  const groups: { graph: string; first: number; names: string[] }[] = [];
+  let skipped = 0;
+  for (const f of files) {
+    let graph: string | undefined;
+    try { graph = pngTexts(path.join(dir, f.name)).prompt; } catch { /* битый PNG */ }
+    if (!graph) { skipped++; continue; }
+    const last = groups[groups.length - 1];
+    if (last && last.graph === graph && f.mtime - last.first <= SAME_BATCH_MS) last.names.push(f.name);
+    else groups.push({ graph, first: f.mtime, names: [f.name] });
+  }
+
+  let created = 0;
+  transaction(() => {
+    for (const g of groups) {
+      let parsed;
+      try { parsed = inputFromGraph(JSON.parse(g.graph)); } catch { parsed = null; }
+      if (!parsed) { skipped += g.names.length; continue; }
+      const id = insertTextBatch(parsed.input, { created: Math.round(g.first), state: 'done', seedUsed: parsed.seed, prompt: parsed.prompt });
+      attachImages(id, g.names.map(filename => ({ filename, subfolder: SRC_DIR, type: 'output' })));
+      created++;
+    }
+    setMeta(KEY, new Date().toISOString());
+  });
+  console.log(`Перенос старых прогонов: батчей ${created}, картинок ${files.length - skipped}` + (skipped ? `, пропущено ${skipped}` : ''));
+}