added create root folder if user has no drive

Author: CoderFaris

src/app/(home)/drive/page.tsx

@@ -0,0 +1,37 @@
+import { auth } from "@clerk/nextjs/server"
+import { MUTATIONS, QUERIES } from "~/server/db/queries"
+import { redirect } from "next/navigation"
+import { Button } from "~/components/ui/button"
+import { root } from ".eslintrc.cjs"
+
+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 (
+            <form action={async ()=>{
+                "use server";
+                const session = await auth()
+
+                if (!session.userId) {
+                    return redirect("/sign-in")
+                }
+
+                const rootFolderId = await MUTATIONS.onboardUser(session.userId)
+                
+                return redirect(`/f/${rootFolderId}`)
+            }}>
+                <Button>Create New Drive</Button>
+            </form>
+        )
+    }
+
+    return redirect(`/f/${rootFolder.id}`)
+}

src/app/(home)/layout.tsx

@@ -0,0 +1,7 @@
+export default function HomePage(props: {children: React.ReactNode}) {
+    return (
+      <div className="flex min-h-screen flex-col items-center justify-center bg-gradient-to-br from-black via-neutral-900 to-neutral-800 p-4 text-white">
+        <main className="text-center">{props.children}</main>
+      </div>
+    );
+}

src/app/(home)/page.tsx

@@ -4,8 +4,7 @@ import { Button } from "~/components/ui/button";
 
 export default function HomePage() {
   return (
-    <div className="flex min-h-screen flex-col items-center justify-center bg-gradient-to-br from-black via-neutral-900 to-neutral-800 p-4 text-white">
-      <main className="text-center">
+    <>
         <h1 className="mb-4 bg-gradient-to-r from-neutral-200 to-neutral-400 bg-clip-text text-5xl font-bold text-transparent md:text-6xl">
           DriveV2
         </h1>
@@ -30,10 +29,10 @@ export default function HomePage() {
             Get Started
           </Button>
         </form>
-      </main>
+      
       <footer className="mt-16 text-sm text-neutral-500">
         © {new Date().getFullYear()} DriveV2. All rights reserved.
       </footer>
-    </div>
+    </>
   );
 }

src/app/(home)/sign-in/page.tsx

@@ -0,0 +1,11 @@
+import { SignInButton } from "@clerk/nextjs";
+export default function HomePage() {
+  return (
+      <>
+        <SignInButton forceRedirectUrl={"/drive"} />
+        <footer className="mt-16 text-sm text-neutral-500">
+          © {new Date().getFullYear()} T3 Drive. All rights reserved.
+        </footer>
+      </>
+  );
+}

src/app/drive/page.tsx

@@ -3,6 +3,7 @@ import "server-only"
 import { db } from "~/server/db"
 import { files_table as filesSchema, folders_table as foldersSchema } from "~/server/db/schema"
 import { eq, and, isNull } from "drizzle-orm"
+import { root } from ".eslintrc.cjs"
 
 export const QUERIES = {
     getFiles : function(folderId: number) {
@@ -75,6 +76,35 @@ export const MUTATIONS = {
             ownerId: input.userId,
         })
 
+    },
+
+    onboardUser : async function(userId : string) {
+        const rootFolder = await db.insert(foldersSchema).values({
+            name: "Root",
+            parent: null,
+            ownerId: userId,
+        }).$returningId()
+
+        const rootFolderId = rootFolder[0]!.id
+
+        await db.insert(foldersSchema).values([{
+            name: "Trash",
+            parent: rootFolderId,
+            ownerId: userId
+        },
+        {
+            name: "Shared",
+            parent: rootFolderId,
+            ownerId: userId,
+        },
+        {
+            name: "Documents",
+            parent: rootFolderId,
+            ownerId: userId,
+        }
+        ])
+
+        return rootFolderId
     }
 }