split db and storage workers

backend/database/src/index.ts

@@ -0,0 +1,72 @@
+import type { DrizzleD1Database } from "drizzle-orm/d1";
+import { drizzle } from "drizzle-orm/d1";
+import { json } from "itty-router-extras";
+import { ZodError, z } from "zod";
+
+import { user, sandbox } from "./schema";
+import * as schema from "./schema";
+import { eq } from "drizzle-orm";
+
+const success = new Response("Success", { status: 200 });
+const notFound = new Response("Not Found", { status: 404 });
+const methodNotAllowed = new Response("Method Not Allowed", { status: 405 });
+
+export interface Env {
+	DB: D1Database;
+	R2: R2Bucket;
+}
+
+// https://github.com/drizzle-team/drizzle-orm/tree/main/examples/cloudflare-d1
+
+export default {
+	async fetch(request: Request, env: Env, ctx: ExecutionContext): Promise<Response> {
+		const url = new URL(request.url);
+		const path = url.pathname;
+		const method = request.method;
+
+		const db = drizzle(env.DB, { schema });
+
+		// ^\/api\/sandbox\/([^\/]+)\/((init|files))$
+
+		if (new RegExp("^/api/sandbox/([^/]+)/((init|files))$").test(path)) {
+			const sandboxId = path.split("/")[2];
+			const command = path.split("/")[3] as "init" | "files";
+
+			if (command === "init") {
+				await db.update(sandbox).set({ init: true }).where(eq(sandbox.id, sandboxId));
+			}
+		} else if (path === "/api/user") {
+			if (method === "GET") {
+				const params = url.searchParams;
+
+				if (params.has("id")) {
+					const id = params.get("id") as string;
+					const res = await db.query.user.findFirst({
+						where: (user, { eq }) => eq(user.id, id),
+						with: {
+							sandbox: true,
+						},
+					});
+					return json(res ?? {});
+				} else {
+					const res = await db.select().from(user).all();
+					return json(res ?? {});
+				}
+			} else if (method === "POST") {
+				const userSchema = z.object({
+					id: z.string(),
+					name: z.string(),
+					email: z.string().email(),
+				});
+
+				const body = await request.json();
+				const { id, name, email } = userSchema.parse(body);
+
+				const res = await db.insert(user).values({ id, name, email }).returning().get();
+				return json({ res });
+			} else {
+				return methodNotAllowed;
+			}
+		} else return notFound;
+	},
+};