From 0bee4b1ade53c76ddf7080e0301b4bc9566a7d40 Mon Sep 17 00:00:00 2001
From: wukko <me@wukko.me>
Date: Thu, 6 Mar 2025 17:04:47 +0600
Subject: [PATCH] web/queue/createSavePipeline: store original request & allow
 to retry

---
 web/src/lib/api/saving-handler.ts | 2 +-
 web/src/lib/queen-bee/queue.ts    | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/web/src/lib/api/saving-handler.ts b/web/src/lib/api/saving-handler.ts
index 800f904a..1e03b465 100644
--- a/web/src/lib/api/saving-handler.ts
+++ b/web/src/lib/api/saving-handler.ts
@@ -117,7 +117,7 @@ export const savingHandler = async ({ url, request }: { url?: string, request?:
         console.log(response);
 
         downloadButtonState.set("done");
-        return createSavePipeline(response);
+        return createSavePipeline(response, selectedRequest);
     }
 
     if (response.status === "picker") {
diff --git a/web/src/lib/queen-bee/queue.ts b/web/src/lib/queen-bee/queue.ts
index a6c96238..12f4684e 100644
--- a/web/src/lib/queen-bee/queue.ts
+++ b/web/src/lib/queen-bee/queue.ts
@@ -2,7 +2,7 @@ import mime from "mime";
 
 import { addItem } from "$lib/state/queen-bee/queue";
 import type { CobaltPipelineItem } from "$lib/types/workers";
-import type { CobaltLocalProcessingResponse } from "$lib/types/api";
+import type { CobaltLocalProcessingResponse, CobaltSaveRequestBody } from "$lib/types/api";
 
 export const getMediaType = (type: string) => {
     const kind = type.split('/')[0];
@@ -50,7 +50,7 @@ export const createRemuxPipeline = (file: File) => {
     }
 }
 
-export const createSavePipeline = (info: CobaltLocalProcessingResponse) => {
+export const createSavePipeline = (info: CobaltLocalProcessingResponse, request: CobaltSaveRequestBody) => {
     const parentId = crypto.randomUUID();
     const pipeline: CobaltPipelineItem[] = [];
 
@@ -89,6 +89,8 @@ export const createSavePipeline = (info: CobaltLocalProcessingResponse) => {
         id: parentId,
         state: "waiting",
         pipeline,
+        canRetry: true,
+        originalRequest: request,
         filename: info.filename,
         mimeType: mime.getType(info.filename) || undefined,
         mediaType: "video",