Reattach Comfy watchers after refresh so pending jobs finish and save.

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

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

@@ -8,14 +8,8 @@ export default defineEventHandler(async (event) => {
     if (pending) {
       const done = await resolvePendingJob(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
-      }
+      const job = ensurePendingWatch(pending)
+      return jobSnapshot(job)
     }
   }
 

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

@@ -1,6 +1,10 @@
 export default defineEventHandler(async (event) => {
   const id = getRouterParam(event, 'id')
-  const job = id ? getJob(id) : undefined
+  let job = id ? getJob(id) : undefined
+  if (!job && id) {
+    const pending = readPendingJob(id)
+    if (pending) job = ensurePendingWatch(pending)
+  }
   if (!job) {
     throw createError({ statusCode: 404, statusMessage: 'Job not found' })
   }

server/api/generate/active.get.ts

@@ -35,17 +35,13 @@ export default defineEventHandler(async (event) => {
           ...resolved
         }
       } else {
+        const job = ensurePendingWatch(pending)
         video = {
           jobId: pending.jobId,
           kind: 'video',
-          type: 'snapshot',
-          status: 'running',
-          message: 'Reconnecting to ComfyUI...',
-          progress: Math.max(8, Math.min(90, Math.round((Date.now() - pending.startedAt) / 1000))),
-          promptId: pending.promptId,
-          elapsedMs: Date.now() - pending.startedAt,
           hideThumbnail: pending.hideThumbnail === true,
-          folderLocked: pending.folderLocked === true
+          folderLocked: pending.folderLocked === true,
+          ...jobSnapshot(job)
         }
       }
     }

server/api/generate/recover.post.ts

@@ -9,17 +9,7 @@ export default defineEventHandler(async (event) => {
     if (pending) {
       const done = await resolvePendingJob(pending)
       if (done) return done
-      return {
-        type: 'snapshot',
-        status: 'running',
-        jobId,
-        message: 'Reconnecting to ComfyUI...',
-        progress: 90,
-        promptId: pending.promptId,
-        elapsedMs: Date.now() - pending.startedAt,
-        hideThumbnail: pending.hideThumbnail,
-        folderLocked: pending.folderLocked
-      }
+      return jobSnapshot(ensurePendingWatch(pending))
     }
     throw createError({ statusCode: 404, statusMessage: 'Job not found' })
   }