From 007e4126305e9b9a57496d99d5b42da53983f2d0 Mon Sep 17 00:00:00 2001
From: myl7 <myl@myl.moe>
Date: Sun, 2 Jan 2022 11:54:31 +0800
Subject: [PATCH] handle errors when traversing

---
 components/MultiFileDownloader.tsx | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/components/MultiFileDownloader.tsx b/components/MultiFileDownloader.tsx
index 1eb0bef..0fd1bd6 100644
--- a/components/MultiFileDownloader.tsx
+++ b/components/MultiFileDownloader.tsx
@@ -178,7 +178,18 @@ export async function* traverseFolder(path: string): AsyncGenerator<
     const itemLists = await Promise.all(
       folderPaths.map(fp =>
         (async fp => {
-          const data = await fetcher(`/api?path=${fp}`, hashedToken ?? undefined)
+          let data: any
+          try {
+            data = await fetcher(`/api?path=${fp}`, hashedToken ?? undefined)
+          } catch (error: any) {
+            // Skip errors caused by the client
+            if (Math.floor(error.response.status / 100) === 4) {
+              return null
+            } else {
+              throw error
+            }
+          }
+
           if (data && data.folder) {
             return data.folder.value.map((c: any) => {
               const p = `${fp === '/' ? '' : fp}/${encodeURIComponent(c.name)}`
@@ -191,7 +202,7 @@ export async function* traverseFolder(path: string): AsyncGenerator<
       )
     )
 
-    const items = itemLists.flat() as { path: string; meta: any; isFolder: boolean }[]
+    const items = itemLists.filter(Boolean).flat() as { path: string; meta: any; isFolder: boolean }[]
     yield* items
     folderPaths = items.filter(i => i.isFolder).map(i => i.path)
   }