fix pdfile over size

JiangShuuu · Nov 17, 2023 · 933844d · 933844d
1 parent 76d0279
commit 933844d
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/app/dashboard/[fileid]/page.tsx b/src/app/dashboard/[fileid]/page.tsx
@@ -23,7 +23,7 @@ const Page = async ({ params }: PageProps) => {
     //   title: 'need Signin'
     // })
 
-    redirect('/dashboard')
+    redirect('/')
   }
 
   const file = await db.file.findFirst({

diff --git a/src/components/button/UploadButton.tsx b/src/components/button/UploadButton.tsx
@@ -51,12 +51,26 @@ const UploadDropzone = ({ isSubscribed }: { isSubscribed: boolean }) => {
       onDrop={async (acceptedFile) => {
         setIsUploading(true)
 
-        const progressInterval = startSimulatedProgress()
+        const fileSize = acceptedFile[0].size / 1024 / 1024
 
+        if (fileSize > (isSubscribed ? 16 : 4)) {
+          setIsUploading(false)
+          acceptedFile.splice(0, 1) // remove file
+          return toast({
+            title: 'The file size exceeds the limit.',
+            description: `Files cannot exceed ${isSubscribed ? 16 : 4} MB or Upgrade the plan.`,
+            variant: 'destructive'
+          })
+        }
+
+        const progressInterval = startSimulatedProgress()
         // handle file uploading
         const res = await startUpload(acceptedFile)
 
         if (!res) {
+          clearInterval(progressInterval)
+          setUploadProgress(0)
+          setIsUploading(false)
           return toast({
             title: 'Something went wrong',
             description: 'Please try again later',
@@ -69,6 +83,9 @@ const UploadDropzone = ({ isSubscribed }: { isSubscribed: boolean }) => {
         const key = fileResponse?.key
 
         if (!key) {
+          clearInterval(progressInterval)
+          setUploadProgress(0)
+          setIsUploading(false)
           return toast({
             title: 'Something went wrong',
             description: 'Please try again later',