Onboard New Users

Author: ThomasCode92

src/app/drive/page.tsx

@@ -1,14 +1,29 @@
 import { auth } from "@clerk/nextjs/server";
 import { redirect } from "next/navigation";
 
+import { Button } from "~/components/ui/button";
 import * as queries from "~/server/db/queries";
+import * as mutations from "~/server/db/mutations";
 
 export default async function DrivePage() {
   const session = await auth();
   if (!session.userId) return redirect("/sign-in");
 
   const rootFolder = await queries.getRootFolderForUser(session.userId);
-  if (!rootFolder) return redirect("/drive/create-root-folder");
+
+  if (!rootFolder) {
+    return (
+      <form
+        action={async () => {
+          "use server";
+          const rootFolderId = await mutations.onboardUser(session.userId);
+          return redirect(`/f/${rootFolderId}`);
+        }}
+      >
+        <Button>Create Folder</Button>
+      </form>
+    );
+  }
 
   return redirect(`/f/${rootFolder.id}`);
 }

src/server/db/mutations.ts

@@ -1,5 +1,26 @@
 import { db } from "~/server/db";
-import { type File, files_table as filesSchema } from "~/server/db/schema";
+import {
+  type File,
+  folders_table as folderSchema,
+  files_table as filesSchema,
+} from "~/server/db/schema";
+
+export async function onboardUser(userId: string) {
+  const [rootFolder] = await db
+    .insert(folderSchema)
+    .values({ name: "root", parent: null, ownerId: userId })
+    .$returningId();
+
+  const rootFolderId = rootFolder!.id;
+
+  await db.insert(folderSchema).values([
+    { name: "Trash", parent: rootFolderId, ownerId: userId },
+    { name: "Shared", parent: rootFolderId, ownerId: userId },
+    { name: "Documents", parent: rootFolderId, ownerId: userId },
+  ]);
+
+  return rootFolderId;
+}
 
 export function createFile(
   file: Pick<File, "name" | "size" | "url" | "parent">,