feat: add create workflow run error server action catch, redirect workflow parse error, add key revoked col

2023-12-16 14:55:30 +08:00
parent 0cf6d97f2f
commit 21a17cb753
16 changed files with 839 additions and 188 deletions
@@ -6,102 +6,101 @@ import { ComfyAPI_Run } from "@/types/ComfyAPI_Run";
 import { eq } from "drizzle-orm";
 import { revalidatePath } from "next/cache";
 import "server-only";
+import { withServerPromise } from "./withServerPromise";

-export async function createRun(
-  origin: string,
-  workflow_version_id: string,
-  machine_id: string,
-  inputs?: Record<string, string>
-) {
-  const machine = await db.query.machinesTable.findFirst({
-    where: eq(workflowRunsTable.id, machine_id),
-  });
-
-  if (!machine) {
-    throw new Error("Machine not found");
-    // return new Response("Machine not found", {
-    //   status: 404,
-    // });
-  }
-
-  const workflow_version_data =
-    // workflow_version_id
-    //   ?
-    await db.query.workflowVersionTable.findFirst({
-      where: eq(workflowRunsTable.id, workflow_version_id),
+export const createRun = withServerPromise(
+  async (
+    origin: string,
+    workflow_version_id: string,
+    machine_id: string,
+    inputs?: Record<string, string>,
+  ) => {
+    const machine = await db.query.machinesTable.findFirst({
+      where: eq(workflowRunsTable.id, machine_id),
    });
-  // : workflow_version != undefined
-  // ? await db.query.workflowVersionTable.findFirst({
-  //     where: and(
-  //       eq(workflowVersionTable.version, workflow_version),
-  //       eq(workflowVersionTable.workflow_id)
-  //     ),
-  //   })
-  // : null;
-  if (!workflow_version_data) {
-    throw new Error("Workflow version not found");
-    // return new Response("Workflow version not found", {
-    //   status: 404,
-    // });
-  }

-  const comfyui_endpoint = `${machine.endpoint}/comfyui-deploy/run`;
-
-  const workflow_api = workflow_version_data.workflow_api;
-
-  // Replace the inputs
-  if (inputs && workflow_api) {
-    for (const key in inputs) {
-      Object.entries(workflow_api).forEach(([_, node]) => {
-        if (node.inputs["input_id"] === key) {
-          node.inputs["input_id"] = inputs[key];
-        }
-      });
+    if (!machine) {
+      throw new Error("Machine not found");
+      // return new Response("Machine not found", {
+      //   status: 404,
+      // });
    }
-  }

-  const body = {
-    workflow_api: workflow_api,
-    status_endpoint: `${origin}/api/update-run`,
-    file_upload_endpoint: `${origin}/api/file-upload`,
-  };
-  // console.log(body);
-  const bodyJson = JSON.stringify(body);
-  // console.log(bodyJson);
+    const workflow_version_data =
+      // workflow_version_id
+      //   ?
+      await db.query.workflowVersionTable.findFirst({
+        where: eq(workflowRunsTable.id, workflow_version_id),
+      });
+    // : workflow_version != undefined
+    // ? await db.query.workflowVersionTable.findFirst({
+    //     where: and(
+    //       eq(workflowVersionTable.version, workflow_version),
+    //       eq(workflowVersionTable.workflow_id)
+    //     ),
+    //   })
+    // : null;
+    if (!workflow_version_data) {
+      throw new Error("Workflow version not found");
+      // return new Response("Workflow version not found", {
+      //   status: 404,
+      // });
+    }

-  // Sending to comfyui
-  const _result = await fetch(comfyui_endpoint, {
-    method: "POST",
-    body: bodyJson,
-    cache: "no-store",
-  });
+    const comfyui_endpoint = `${machine.endpoint}/comfyui-deploy/run`;

-  if (!_result.ok) {
-    throw new Error(`Error creating run, ${_result.statusText}`);
-  }
+    const workflow_api = workflow_version_data.workflow_api;

-  console.log(_result);
+    // Replace the inputs
+    if (inputs && workflow_api) {
+      for (const key in inputs) {
+        Object.entries(workflow_api).forEach(([_, node]) => {
+          if (node.inputs["input_id"] === key) {
+            node.inputs["input_id"] = inputs[key];
+          }
+        });
+      }
+    }

-  const result = await ComfyAPI_Run.parseAsync(await _result.json());
+    const body = {
+      workflow_api: workflow_api,
+      status_endpoint: `${origin}/api/update-run`,
+      file_upload_endpoint: `${origin}/api/file-upload`,
+    };
+    // console.log(body);
+    const bodyJson = JSON.stringify(body);
+    // console.log(bodyJson);

-  console.log(result);
+    // Sending to comfyui
+    const _result = await fetch(comfyui_endpoint, {
+      method: "POST",
+      body: bodyJson,
+      cache: "no-store",
+    });

-  // Add to our db
-  const workflow_run = await db
-    .insert(workflowRunsTable)
-    .values({
-      id: result.prompt_id,
-      workflow_id: workflow_version_data.workflow_id,
-      workflow_version_id: workflow_version_data.id,
-      workflow_inputs: inputs,
-      machine_id,
-    })
-    .returning();
+    if (!_result.ok) {
+      throw new Error(`Error creating run, ${_result.statusText}`);
+    }

-  revalidatePath(`/${workflow_version_data.workflow_id}`);
+    const result = await ComfyAPI_Run.parseAsync(await _result.json());

-  return {
-    workflow_run_id: workflow_run[0].id,
-    message: "Successfully workflow run",
-  };
-}
+    // Add to our db
+    const workflow_run = await db
+      .insert(workflowRunsTable)
+      .values({
+        id: result.prompt_id,
+        workflow_id: workflow_version_data.workflow_id,
+        workflow_version_id: workflow_version_data.id,
+        workflow_inputs: inputs,
+        machine_id,
+      })
+      .returning();
+
+    revalidatePath(`/${workflow_version_data.workflow_id}`);
+
+    return {
+      workflow_run_id: workflow_run[0].id,
+      message: "Successfully workflow run",
+    };
+  },
+);