feat: add opening directly from comfy deploy -> machines

2024-01-23 09:33:53 +08:00
parent c0450b58d5
commit 52d6e07eeb
12 changed files with 808 additions and 405 deletions
@@ -0,0 +1,81 @@
+import { db } from "@/db/db";
+import {
+  deploymentsTable,
+  workflowTable,
+  workflowVersionTable,
+} from "@/db/schema";
+import { APIKeyUserType } from "@/server/APIKeyBodyRequest";
+import { auth } from "@clerk/nextjs";
+import { and, desc, eq, isNull } from "drizzle-orm";
+import { redirect } from "next/navigation";
+
+export async function getAllUserWorkflow() {
+  const { userId, orgId } = await auth();
+
+  if (!userId) {
+    return null;
+  }
+
+  const workflow = await db.query.workflowTable.findMany({
+    with: {
+      user: {
+        columns: {
+          name: true,
+        },
+      },
+      versions: {
+        limit: 1,
+        orderBy: desc(workflowVersionTable.version),
+        columns: {
+          id: true,
+          version: true,
+        },
+      },
+      deployments: {
+        limit: 1,
+        where: eq(deploymentsTable.environment, "public-share"),
+        columns: {
+          id: true,
+        },
+      },
+    },
+    columns: {
+      id: true,
+      updated_at: true,
+      name: true,
+    },
+    orderBy: desc(workflowTable.updated_at),
+    where:
+      orgId != undefined
+        ? eq(workflowTable.org_id, orgId)
+        : and(eq(workflowTable.user_id, userId), isNull(workflowTable.org_id)),
+  });
+
+  return workflow;
+}
+
+export async function getWorkflowVersion(
+  apiUser: APIKeyUserType,
+  version_id: string,
+) {
+  const { org_id, user_id } = apiUser;
+
+  if (!user_id) {
+    throw new Error("No user id");
+  }
+
+  const parentWorkflow = await db.query.workflowTable.findFirst({
+    where:
+      org_id != undefined
+        ? eq(workflowTable.org_id, org_id)
+        : and(eq(workflowTable.user_id, user_id), isNull(workflowTable.org_id)),
+  });
+
+  if (!parentWorkflow) {
+    throw new Error("No workflow found");
+  }
+
+  return db.query.workflowVersionTable.findFirst({
+    where: eq(workflowVersionTable.id, version_id),
+  });
+}