feat(api): add file upload api, get a upload url, then upload file, max 50mb for now

2024-01-16 00:49:50 +08:00
parent 478859d9b5
commit 1ced8095dc
6 changed files with 358 additions and 216 deletions
@@ -0,0 +1,106 @@
+import { createRun } from "../server/createRun";
+import { db } from "@/db/db";
+import { deploymentsTable } from "@/db/schema";
+import type { App } from "@/routes/app";
+import { authError } from "@/routes/authError";
+import { z, createRoute } from "@hono/zod-openapi";
+import { eq } from "drizzle-orm";
+
+const createRunRoute = createRoute({
+  method: "post",
+  path: "/run",
+  tags: ["workflows"],
+  summary: "Run a workflow via deployment_id",
+  request: {
+    body: {
+      content: {
+        "application/json": {
+          schema: z.object({
+            deployment_id: z.string(),
+            inputs: z.record(z.string()).optional(),
+          }),
+        },
+      },
+    },
+  },
+  responses: {
+    200: {
+      content: {
+        "application/json": {
+          schema: z.object({
+            run_id: z.string(),
+          }),
+        },
+      },
+      description: "Workflow queued",
+    },
+    500: {
+      content: {
+        "application/json": {
+          schema: z.object({
+            error: z.string(),
+          }),
+        },
+      },
+      description: "Error creating run",
+    },
+    ...authError,
+  },
+});
+
+export const registerCreateRunRoute = (app: App) => {
+  app.openapi(createRunRoute, async (c) => {
+    const data = c.req.valid("json");
+    const origin = new URL(c.req.url).origin;
+    const apiKeyTokenData = c.get("apiKeyTokenData")!;
+
+    const { deployment_id, inputs } = data;
+
+    try {
+      const deploymentData = await db.query.deploymentsTable.findFirst({
+        where: eq(deploymentsTable.id, deployment_id),
+        with: {
+          machine: true,
+          version: {
+            with: {
+              workflow: {
+                columns: {
+                  org_id: true,
+                  user_id: true,
+                },
+              },
+            },
+          },
+        },
+      });
+
+      if (!deploymentData) throw new Error("Deployment not found");
+
+      const run_id = await createRun({
+        origin,
+        workflow_version_id: deploymentData.version,
+        machine_id: deploymentData.machine,
+        inputs,
+        isManualRun: false,
+        apiUser: apiKeyTokenData,
+      });
+
+      if ("error" in run_id) throw new Error(run_id.error);
+
+      return c.json({
+        run_id: "workflow_run_id" in run_id ? run_id.workflow_run_id : "",
+      });
+    } catch (error: unknown) {
+      const errorMessage =
+        error instanceof Error ? error.message : "Unknown error";
+      return c.json(
+        {
+          error: errorMessage,
+        },
+        {
+          status: 500,
+        }
+      );
+    }
+  });
+};