Recover finished Comfy videos after relay restarts.

Persist queued jobs to disk, import leftover MP4s from Comfy history, and do not mark a run complete until the saved video exists.

Co-authored-by: Cursor <cursoragent@cursor.com>

server/api/generate/[id].get.ts

@@ -1,8 +1,23 @@
-export default defineEventHandler((event) => {
+export default defineEventHandler(async (event) => {
   const id = getRouterParam(event, 'id')
-  const job = id ? getJob(id) : undefined
-  if (!job) {
-    throw createError({ statusCode: 404, statusMessage: 'Job not found' })
+  const live = id ? getJob(id) : undefined
+  if (live) return jobSnapshot(live)
+
+  if (id) {
+    const pending = readPendingJob(id)
+    if (pending) {
+      const done = await completePendingIfReady(pending).catch(() => null)
+      if (done) return done
+      return {
+        type: 'snapshot',
+        status: 'running',
+        message: 'Reconnecting to ComfyUI...',
+        progress: 90,
+        promptId: pending.promptId,
+        elapsedMs: Date.now() - pending.startedAt
+      }
+    }
   }
-  return jobSnapshot(job)
+
+  throw createError({ statusCode: 404, statusMessage: 'Job not found' })
 })

server/api/generate/recover.post.ts

@@ -0,0 +1,40 @@
+export default defineEventHandler(async (event) => {
+  const body = await readBody<{ jobId?: string }>(event).catch(() => ({} as { jobId?: string }))
+  const jobId = String(body?.jobId || '')
+  const live = jobId ? getJob(jobId) : undefined
+  if (live) return jobSnapshot(live)
+
+  if (jobId) {
+    const pending = readPendingJob(jobId)
+    if (pending) {
+      const done = await completePendingIfReady(pending)
+      if (done) return done
+      return {
+        type: 'snapshot',
+        status: 'running',
+        message: 'Reconnecting to ComfyUI...',
+        progress: 90,
+        promptId: pending.promptId,
+        elapsedMs: Date.now() - pending.startedAt
+      }
+    }
+  }
+
+  const { owner } = assertLibraryOwner(event)
+  const library = publicLibrary(event)
+  const folderId = library.folders.find(folder => folder.unlocked)?.id
+  const imported = folderId ? await importMissingComfyVideos(owner, folderId) : []
+  const newest = imported[imported.length - 1]
+  if (!newest) {
+    throw createError({ statusCode: 404, statusMessage: 'No finished ComfyUI video to recover' })
+  }
+  return {
+    type: 'complete',
+    status: 'complete',
+    message: 'Recovered video from ComfyUI',
+    progress: 100,
+    clipId: newest.id,
+    filename: newest.comfyFilename,
+    hideThumbnail: newest.hideThumbnail
+  }
+})