From 54f3451456625144dec8d0bcc6db0e6299282007 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Mon, 28 Aug 2023 18:06:59 +0800
Subject: [PATCH] =?UTF-8?q?feat=20=E6=89=8B=E5=8A=A8=E6=95=B4=E7=90=86?=
 =?UTF-8?q?=E8=BF=9B=E5=BA=A6=E6=98=BE=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/filebrowser/List.vue        | 86 +++++++++++++++++++---
 src/views/discover/TorrentCardListView.vue |  2 +-
 2 files changed, 77 insertions(+), 11 deletions(-)

diff --git a/src/components/filebrowser/List.vue b/src/components/filebrowser/List.vue
index 41069ee1..9e0750b5 100644
--- a/src/components/filebrowser/List.vue
+++ b/src/components/filebrowser/List.vue
@@ -50,6 +50,18 @@ const renamePopper = ref(false)
 // 整理弹窗
 const transferPopper = ref(false)
 
+// 整理进度条
+const progressDialog = ref(false)
+
+// 整理进度文本
+const progressText = ref('请稍候 ...')
+
+// 整理进度
+const progressValue = ref(0)
+
+// 加载进度SSE
+const progressEventSource = ref<EventSource>()
+
 // 新名称
 const newName = ref('')
 
@@ -223,21 +235,33 @@ function showTransfer(item: FileItem) {
 
 // 整理文件
 async function transfer() {
-  transferForm.path = currentItem.value?.path || ''
+  transferForm.path = currentItem.value?.path ?? ''
   // 开始整理文件
   try {
+    // 关闭弹窗
     transferPopper.value = false
-    const result: { [key: string]: any } = await api.post('transfer/manual', {}, {
+    // 显示进度条
+    progressDialog.value = true
+    // 开始监听进度
+    startLoadingProgress()
+    // 异步调API，结束后关闭进度条
+    api.post('transfer/manual', {}, {
       params: transferForm,
+    }).then((res: any) => {
+      // 关闭进度条
+      progressDialog.value = false
+      // 停止监听进度
+      stopLoadingProgress()
+      // 显示结果
+      if (res.success) {
+        $toast.success(`${currentItem.value?.name} 整理成功！`)
+        // 重新加载
+        load()
+      }
+      else {
+        $toast.error(`${currentItem.value?.name} 整理失败：${res.message}！`)
+      }
     })
-    if (result.success) {
-      $toast.success(`${currentItem.value?.name} 整理成功！`)
-      // 重新加载
-      load()
-    }
-    else {
-      $toast.error(`${currentItem.value?.name} 整理失败：${result.message}！`)
-    }
   }
   catch (e) {
     console.log(e)
@@ -264,6 +288,29 @@ watch(
   },
 )
 
+// 使用SSE监听加载进度
+function startLoadingProgress() {
+  progressText.value = '请稍候 ...'
+
+  const token = store.state.auth.token
+
+  progressEventSource.value = new EventSource(
+    `${import.meta.env.VITE_API_BASE_URL}system/progress/filetransfer?token=${token}`,
+  )
+  progressEventSource.value.onmessage = (event) => {
+    const progress = JSON.parse(event.data)
+    if (progress) {
+      progressText.value = progress.text
+      progressValue.value = progress.value
+    }
+  }
+}
+
+// 停止监听加载进度
+function stopLoadingProgress() {
+  progressEventSource.value?.close()
+}
+
 // 弹出菜单
 const dropdownItems = ref([
   {
@@ -623,6 +670,25 @@ onMounted(() => {
       </VCardActions>
     </VCard>
   </VDialog>
+  <!-- 手动整理进度框 -->
+  <vDialog
+    v-model="progressDialog"
+    :scrim="false"
+    width="400"
+  >
+    <vCard
+      color="primary"
+    >
+      <vCardText class="text-center">
+        {{ progressText }}
+        <vProgressLinear
+          color="white"
+          class="mb-0 mt-1"
+          :value="progressValue"
+        />
+      </vCardText>
+    </vCard>
+  </vDialog>
 </template>
 
 <style lang="scss" scoped>
diff --git a/src/views/discover/TorrentCardListView.vue b/src/views/discover/TorrentCardListView.vue
index 230df4b1..c285c9f9 100644
--- a/src/views/discover/TorrentCardListView.vue
+++ b/src/views/discover/TorrentCardListView.vue
@@ -134,7 +134,7 @@ async function fetchData(): Promise<Array<Context>> {
     }
     else {
       startLoadingProgress()
-      const qualify = props.keyword?.startsWith('tmdb:') || props.keyword?.startsWith('douban:')
+      const qualify = props.keyword?.startsWith('tmdb:') ?? props.keyword?.startsWith('douban:')
       // 优先按TMDBID精确查询
       if (qualify) {
         searchData = await api.get(`search/media/${props.keyword}`, {