feat: smoke test trigger on edit, ui updates for validation status pa…

…ge (#135)

* feat: trigger smoke test on edit, fix maxDuration

* update maxduration

* update ui

* fix ui state management, redirect to edit on fail

apps/app/app/api/pipelines/edit.ts

@@ -4,6 +4,27 @@ import { uploadFile } from "@/app/api/pipelines/storage";
 import { pipelineSchema } from "@/lib/types";
 import { createServerClient } from "@repo/supabase/server";
 import { z } from "zod";
+import { triggerSmokeTest } from "./validation";
+import { createSmokeTestStream } from "./validation";
+
+export async function publishPipeline(pipelineId: string, userId: string) {
+  const supabase = await createServerClient();
+
+  await validateUser(userId);
+
+  const { data: updatedPipeline, error: updateError } = await supabase
+    .from("pipelines")
+    .upsert({
+      id: pipelineId,
+      is_private: false,
+    })
+    .select()
+    .single();
+
+  if (updateError) throw new Error(updateError.message);
+
+  return { pipeline: updatedPipeline };
+}
 
 export async function updatePipeline(body: any, userId: string) {
   const supabase = await createServerClient();
@@ -15,16 +36,22 @@ export async function updatePipeline(body: any, userId: string) {
   if (!validationResult.success) {
     throw new z.ZodError(validationResult.error.errors);
   }
-
-  const { data, error } = await supabase
+  const pipelineId = body.id;
+  const { data: pipeline, error } = await supabase
     .from("pipelines")
     .update(validationResult.data)
-    .eq("id", body.id)
+    .eq("id", pipelineId)
     .eq("author", userId)
-    .select();
+    .select()
+    .single();
 
   if (error) throw new Error(error.message);
-  return data;
+
+  // Create a smoke test stream using the pipeline for pre-publication validation with new config saved
+  const smokeTestStream = await createSmokeTestStream(pipelineId);
+  await triggerSmokeTest(smokeTestStream.stream_key);
+
+  return { pipeline, smokeTestStream };
 }
 
 export async function editPipelineFromFormData(
@@ -56,6 +83,9 @@ export async function editPipelineFromFormData(
     config: comfyUiJson,
   };
 
-  const pipeline = await updatePipeline(pipelineData, userId);
-  return pipeline;
+  const { pipeline, smokeTestStream } = await updatePipeline(
+    pipelineData,
+    userId
+  );
+  return { pipeline, smokeTestStream };
 }

apps/app/app/api/streams/[id]/sse/route.ts

@@ -2,6 +2,7 @@ import { getStoredStreamStatus } from "@/app/api/pipelines/validation";
 
 // Prevents this route's response from being cached on Vercel
 export const dynamic = "force-dynamic";
+export const maxDuration = 300;
 
 export async function GET(
   request: Request,

apps/app/app/pipelines/[id]/page.tsx

@@ -19,7 +19,12 @@ export default async function Page({
   }
 
   if (isValidationMode) {
-    return <ValidatePipeline streamId={searchParams.streamId} />;
+    return (
+      <ValidatePipeline
+        pipelineId={pipelineId}
+        streamId={searchParams.streamId}
+      />
+    );
   }
 
   const pipeline = await getPipeline(pipelineId);

apps/app/components/pipeline/edit-pipeline.tsx

@@ -14,13 +14,14 @@ import { Loader2, LoaderCircleIcon } from "lucide-react";
 import { useEffect, useState } from "react";
 import { toast } from "sonner";
 import FileUploadDropzone, { FileType } from "./json-upload";
-import PipelineStatus from "./status";
+import { useRouter } from "next/navigation";
 
 export default function EditPipeline({
   pipeline,
 }: {
   pipeline: PipelineSchema & { id: string };
 }) {
+  const router = useRouter();
   const { authenticated, user, ready: isAuthLoaded } = usePrivy();
   const [formData, setFormData] = useState<Record<string, unknown>>(() => {
     return {
@@ -56,9 +57,10 @@ export default function EditPipeline({
         formDataToSend.append(decamelize(key), value as any);
       });
       console.log("Form data to send::", formDataToSend, formData);
-      const pipeline = await editPipelineFromFormData(formDataToSend, user.id);
+      const { pipeline, smokeTestStream } = await editPipelineFromFormData(formDataToSend, user.id);
       toast.dismiss(toastId);
       toast.success("Pipeline saved successfully");
+      router.push(`/pipelines/${pipeline.id}?streamId=${smokeTestStream.id}&validation=true`);
     } catch (error) {
       toast.dismiss(toastId);
       const errorMessage =