2 files changed, 34 insertions, 3 deletions

diff --git a/frontend/components/drive/DriveDirectoryClient.tsx b/frontend/components/drive/DriveDirectoryClient.tsx
index d238065..330c7a0 100644
--- a/frontend/components/drive/DriveDirectoryClient.tsx
+++ b/frontend/components/drive/DriveDirectoryClient.tsx
@@ -36,6 +36,7 @@ import { formatFileSize } from "@/lib/utils"
 import { DriveMoveDialog } from "./DriveMoveDialog"
 import { StorageUsage } from "./StorageUsage"
 import type { StorageData } from "@/lib/storage"
+import type { UserAuth } from "@/lib/auth_types"
 
 function formatDate(timestamp: number): string {
   return new Date(timestamp * 1000).toISOString().split('T')[0]
@@ -64,9 +65,10 @@ interface DriveDirectoryClientProps {
   files: DriveLsEntry[]
   breadcrumbs: Breadcrumb[]
   storageData: StorageData
+  userAuth: UserAuth
 }
 
-export function DriveDirectoryClient({ path, files, breadcrumbs, storageData }: DriveDirectoryClientProps) {
+export function DriveDirectoryClient({ path, files, breadcrumbs, storageData, userAuth }: DriveDirectoryClientProps) {
   const [selectedFiles, setSelectedFiles] = useState<Set<string>>(new Set())
   const [renameDialogOpen, setRenameDialogOpen] = useState(false)
   const [infoDialogOpen, setInfoDialogOpen] = useState(false)
@@ -414,6 +416,32 @@ export function DriveDirectoryClient({ path, files, breadcrumbs, storageData }:
     }
   }
 
+  const handleUploadButtonClick = () => {
+    if (!userAuth.isLoggedIn) {
+      toast({
+        title: "Authentication required",
+        description: "User is not authenticated. Please log in to upload files.",
+        variant: "destructive"
+      })
+      return
+    }
+
+    fileInputRef.current?.click()
+  }
+
+  const handleCreateFolderButtonClick = () => {
+    if (!userAuth.isLoggedIn) {
+      toast({
+        title: "Authentication required",
+        description: "User is not authenticated. Please log in to create folders.",
+        variant: "destructive"
+      })
+      return
+    }
+
+    setCreateFolderDialogOpen(true)
+  }
+
   return (
     <div className="space-y-6">
       {/* Storage Info */}
@@ -443,14 +471,14 @@ export function DriveDirectoryClient({ path, files, breadcrumbs, storageData }:
         <div className="flex items-center gap-2 sm:gap-2">
           <Button
             variant="secondary"
-            onClick={() => setCreateFolderDialogOpen(true)}
+            onClick={handleCreateFolderButtonClick}
             className="flex-1 sm:flex-none"
           >
             <FolderPlus className="mr-2 h-4 w-4" />
             Create Folder
           </Button>
           <Button
-            onClick={() => fileInputRef.current?.click()}
+            onClick={handleUploadButtonClick}
             disabled={uploading}
             className="flex-1 sm:flex-none"
           >
diff --git a/frontend/components/drive/DriveDirectoryView.tsx b/frontend/components/drive/DriveDirectoryView.tsx
index de9d70a..b20a111 100644
--- a/frontend/components/drive/DriveDirectoryView.tsx
+++ b/frontend/components/drive/DriveDirectoryView.tsx
@@ -2,6 +2,7 @@ import { DriveLsEntry } from "@/lib/drive_types"
 import { DriveDirectoryClient } from "./DriveDirectoryClient"
 import { DriveHeader } from "./DriveHeader"
 import type { StorageData } from "@/lib/storage"
+import { Auth_get_user } from "@/lib/auth"
 
 interface DriveDirectoryViewProps {
   path: string
@@ -47,6 +48,7 @@ function sortFiles(files: DriveLsEntry[]): DriveLsEntry[] {
 export async function DriveDirectoryView({ path, files, storageData }: DriveDirectoryViewProps) {
   const sortedFiles = sortFiles(files)
   const breadcrumbs = generateBreadcrumbs(path)
+  const userAuth = await Auth_get_user()
   
   return (
     <div className="container mx-auto p-6 space-y-6">
@@ -56,6 +58,7 @@ export async function DriveDirectoryView({ path, files, storageData }: DriveDire
         files={sortedFiles} 
         breadcrumbs={breadcrumbs}
         storageData={storageData}
+        userAuth={userAuth}
       />
     </div>
   )