From 1b3809033863ae4d90b54499cd93910f9b880bc5 Mon Sep 17 00:00:00 2001
From: a1mer <onlylovesnow14@gmail.com>
Date: Thu, 30 Nov 2023 16:12:58 +0800
Subject: [PATCH] fix: fetch whole list

---
 src/App.vue         |  4 ++++
 src/utils/aliyun.ts | 35 ++++++++++++++++++++++++-----------
 2 files changed, 28 insertions(+), 11 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index d91e994..d05753a 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -456,6 +456,10 @@ onUnmounted(() => {
           />
           可将其填充到“剧名”
         </div>
+
+        <div v-if="showList.length" class="ml-auto text-xs font-sans text-gray-600">
+          共 <span class="font-bold text-purple-600">{{ showList.length }}</span> 个文件
+        </div>
       </div>
       <ul
         v-if="showList.length" class="grid grid-cols-[20px_auto_30px_minmax(200px,1fr)] items-center gap-x-2 gap-y-1 text-xs"
diff --git a/src/utils/aliyun.ts b/src/utils/aliyun.ts
index ebb4955..e803125 100644
--- a/src/utils/aliyun.ts
+++ b/src/utils/aliyun.ts
@@ -1,8 +1,10 @@
-const listJsonMask = 'ext_marker,items(name,file_id,drive_id,type,size,created_at,updated_at,category,file_extension,parent_file_id,mime_type,starred,thumbnail,url,streams_info,content_hash,user_tags,user_meta,trashed,video_media_metadata,video_preview_metadata,sync_meta,sync_device_flag,sync_flag,punish_flag)'
+const listJsonMask = 'next_marker,items(name,file_id,drive_id,type,size,created_at,updated_at,category,file_extension,parent_file_id,mime_type,starred,thumbnail,url,streams_info,content_hash,user_tags,user_meta,trashed,video_media_metadata,video_preview_metadata,sync_meta,sync_device_flag,sync_flag,punish_flag)'
 
 // 接口调用太频繁会被拒
 export const API_DELAY = 200
 
+const PAGE_SIZE = 100
+
 function getToken() {
   const raw = window.localStorage.getItem('token')
   if (!raw)
@@ -15,20 +17,31 @@ async function getDriveId() {
   return location.pathname.startsWith('/drive/file/resource') ? res.resource_drive_id : res.backup_drive_id
 }
 
+const INITIAL_MARKER = 'INITIAL'
 export async function getFileListOfCurrentDir(parentId = getParentId()) {
   const listApi = new URL('https://api.aliyundrive.com/adrive/v3/file/list')
   listApi.searchParams.append('jsonmask', listJsonMask)
   const driveId = await getDriveId()
-  const res = await post(listApi, {
-    all: true,
-    drive_id: driveId,
-    fields: '*',
-    order_by: 'name',
-    order_direction: 'ASC',
-    parent_file_id: parentId,
-    url_expire_sec: 14400,
-  })
-  return res.items.filter((x: any) => !x.sync_device_flag) as Resource[]
+  const result = []
+  let marker = INITIAL_MARKER
+
+  while (marker) {
+    const { items, next_marker } = await post(listApi, {
+      all: true,
+      limit: PAGE_SIZE,
+      drive_id: driveId,
+      fields: '*',
+      order_by: 'name',
+      order_direction: 'ASC',
+      parent_file_id: parentId,
+      url_expire_sec: 14400,
+      marker: marker === INITIAL_MARKER ? '' : marker,
+    })
+    result.push(...items)
+    marker = next_marker
+  }
+
+  return result.filter((x: any) => !x.sync_device_flag) as Resource[]
 }
 
 async function rename(driveId: string, fileId: string, newName: string) {