Handle Redirect to Root Folder

Author: ThomasCode92

src/app/drive/page.tsx

@@ -1,3 +1,14 @@
-export default function DrivePage() {
-  return <div />;
+import { auth } from "@clerk/nextjs/server";
+import { redirect } from "next/navigation";
+
+import * as queries from "~/server/db/queries";
+
+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");
+
+  return redirect(`/f/${rootFolder.id}`);
 }

src/app/sign-in/page.tsx

@@ -1,5 +1,7 @@
 import { SignInButton } from "@clerk/nextjs";
 
+import { Button } from "~/components/ui/button";
+
 export default function HomePage() {
   return (
     <div className="min-h-screen bg-gradient-to-br from-black via-gray-900 to-gray-800 text-white">
@@ -18,7 +20,14 @@ export default function HomePage() {
             documents from any device, anywhere in the world.
           </p>
 
-          <SignInButton />
+          <Button
+            size="lg"
+            className="cursor-pointer rounded-full bg-white px-8 py-6 text-lg font-semibold text-black transition-all duration-200 hover:scale-105 hover:bg-gray-100"
+            type="submit"
+            asChild
+          >
+            <SignInButton forceRedirectUrl="/drive" />
+          </Button>
         </div>
       </main>
     </div>

src/server/db/queries.ts

@@ -1,6 +1,6 @@
 import "server-only"; // Ensure this file is only run on the server
 
-import { eq } from "drizzle-orm";
+import { and, eq, isNull } from "drizzle-orm";
 
 import { db } from "~/server/db";
 import {
@@ -27,6 +27,14 @@ export async function getAllParentsForFolder(folderId: number) {
   return parents;
 }
 
+export async function getRootFolderForUser(ownerId: string) {
+  const folders = await db
+    .select()
+    .from(folderSchema)
+    .where(and(eq(folderSchema.ownerId, ownerId), isNull(folderSchema.parent)));
+  return folders[0];
+}
+
 export function getAllFolders(folderId: number) {
   return db
     .select()