From 06a2ea704ebe6b6e782b94201666a81762d046dc Mon Sep 17 00:00:00 2001
From: HoshinoSuzumi <master@uniiem.com>
Date: Fri, 28 Jun 2024 21:31:48 +0800
Subject: [PATCH] wip

---
 .../aigc/course-generate/CGTaskCard.vue       | 101 +++++++++++++++---
 .../aigc/course-generate/CourseGenerate.vue   |  46 ++++++--
 composables/useDownload.ts                    |  54 +++++++---
 package.json                                  |   1 +
 typings/types.d.ts                            |   2 +-
 yarn.lock                                     |   5 +
 6 files changed, 175 insertions(+), 34 deletions(-)

diff --git a/components/aigc/course-generate/CGTaskCard.vue b/components/aigc/course-generate/CGTaskCard.vue
index 5512bea..bbbdf5c 100644
--- a/components/aigc/course-generate/CGTaskCard.vue
+++ b/components/aigc/course-generate/CGTaskCard.vue
@@ -3,6 +3,8 @@ import type { PropType } from 'vue'
 import dayjs from 'dayjs'
 import { useDownload } from '~/composables/useDownload'
 
+const toast = useToast()
+
 const props = defineProps({
   course: {
     type: Object as PropType<resp.gen.CourseGenItem>,
@@ -21,9 +23,61 @@ const stateDisplay = computed(() => {
   }
   return {
     color: 'blue',
-    text: '进行中',
+    text: !!props.course.progress ? `${ props.course.progress }%` : '队列中',
   }
 })
+const isFailed = computed(() => props.course.progress === -1)
+const isDownloadable = computed(() => !isFailed.value && props.course.progress === 100)
+
+const downloadProgress = ref(0)
+
+const startDownload = async (
+  url: string,
+  filename: string,
+) => {
+  downloadProgress.value = 0
+
+  const {
+    download,
+    progressEmitter,
+  } = useDownload(url, filename)
+
+  progressEmitter.on('done', () => {
+    downloadProgress.value = 100
+    toast.add({
+      title: '下载完成',
+      description: '资源下载已完成',
+      color: 'green',
+      icon: 'i-tabler-check',
+    })
+  })
+
+  progressEmitter.on('progress', (progress) => {
+    downloadProgress.value = progress
+  })
+
+  progressEmitter.on('error', err => {
+    downloadProgress.value = 0
+    toast.add({
+      title: '下载失败',
+      description: err.message || '下载失败，未知错误',
+      color: 'red',
+      icon: 'i-tabler-alert-triangle',
+    })
+  })
+
+  download()
+}
+
+const copyTaskId = () => {
+  navigator.clipboard.writeText(props.course.task_id)
+  toast.add({
+    title: '复制成功',
+    description: '已复制任务 ID',
+    color: 'green',
+    icon: 'i-tabler-check',
+  })
+}
 </script>
 
 <template>
@@ -32,27 +86,33 @@ const stateDisplay = computed(() => {
   >
     <div class="relative w-full aspect-video">
       <NuxtImg
+        class="w-full aspect-video object-cover pointer-events-none absolute inset-0"
         v-if="!!course.video_cover"
         :src="course.video_cover"
         alt="image"
-        class="w-full h-full object-cover pointer-events-none absolute inset-0"
+        loading="lazy"
       />
       <div
         v-else
         class="absolute inset-0 bg-gradient-to-br from-purple-400 to-primary-400 flex justify-center items-center"
       >
-        <Icon class="text-white text-[64px] animate-pulse" name="i-tabler-photo-video"/>
+        <Icon
+          v-if="isFailed"
+          class="text-white text-[64px] opacity-50"
+          name="i-tabler-alert-triangle"
+        />
+        <Icon v-else class="text-white text-[64px] animate-pulse" name="i-tabler-photo-video"/>
       </div>
       <div class="absolute inset-2 flex justify-end items-start">
         <UTooltip :prevent="course.progress > -1" :text="course.message || ''">
           <UBadge
             :color="stateDisplay.color"
-            :variant="stateDisplay.color === 'red' ? 'solid' : 'subtle'"
+            :variant="isFailed ? 'solid' : 'subtle'"
             class="shadow"
             size="sm"
           >
             <Icon
-              v-if="stateDisplay.color === 'red'"
+              v-if="isFailed"
               class="text-base mr-0.5"
               name="i-tabler-alert-triangle"
             />
@@ -63,8 +123,10 @@ const stateDisplay = computed(() => {
     </div>
     <div class="px-2 pt-1 pb-2 flex justify-between">
       <div class="flex-1 overflow-hidden pt-1">
-        <h1 class="text-sm font-medium overflow-hidden text-ellipsis text-nowrap"
-            title="课程名字课程名字课程名字课程名字课程名字课程名字课程名字课程名字">
+        <h1
+          :title="course.title"
+          class="text-sm font-medium overflow-hidden text-ellipsis text-nowrap"
+        >
           <Icon class="-mt-0.5 -ml-0.5 text-base" name="i-tabler-book-2"/>
           <span class="pl-0.5">{{ course.title }}</span>
         </h1>
@@ -74,6 +136,7 @@ const stateDisplay = computed(() => {
             class="hover:text-primary font-medium"
             tabindex="-1"
             :title="course.task_id"
+            @click="copyTaskId"
           >
             复制ID
           </button>
@@ -83,13 +146,11 @@ const stateDisplay = computed(() => {
         <UButtonGroup>
           <UButton
             color="white"
-            label="下载"
+            :disabled="!isDownloadable"
+            :label="downloadProgress > 0 && downloadProgress < 100 ? `${downloadProgress.toFixed(0)}%` : '下载'"
             leading-icon="i-tabler-download"
             size="xs"
-            @click="() => {
-              const {download} = useDownload(course.video_url, `眩生花微课_${course.title}_${course.task_id}.mp4`)
-              download()
-            }"
+            @click="startDownload(course.video_url, `眩生花微课_${ props.course.title }_${ props.course.task_id }.mp4`)"
           />
           <UDropdown
             :items="[
@@ -97,19 +158,31 @@ const stateDisplay = computed(() => {
                 label: '预览课程',
                 icon: 'i-tabler-play',
                 shortcuts: ['空格'],
+                disabled: !isDownloadable,
+              }, {
+                label: '查看字幕',
+                icon: 'i-solar-subtitles-linear',
+                shortcuts: ['Alt', 'D'],
+                disabled: !isDownloadable,
               }, {
                 label: '下载字幕',
                 icon: 'i-tabler-file-download',
                 shortcuts: ['Alt', 'S'],
+                disabled: !isDownloadable,
               }], [{
                 label: '删除记录',
                 icon: 'i-tabler-trash-x',
                 shortcuts: ['Delete'],
               }],
             ]"
-            :popper="{ placement: 'bottom-start' }"
+            :popper="{ placement: 'bottom-end' }"
           >
-            <UButton color="white" size="xs" trailing-icon="i-tabler-dots"/>
+            <UButton
+              :disabled="course.progress > 1 && course.progress < 100"
+              color="white"
+              size="xs"
+              trailing-icon="i-tabler-dots"
+            />
           </UDropdown>
         </UButtonGroup>
       </div>
diff --git a/components/aigc/course-generate/CourseGenerate.vue b/components/aigc/course-generate/CourseGenerate.vue
index a09ae2a..8082f51 100644
--- a/components/aigc/course-generate/CourseGenerate.vue
+++ b/components/aigc/course-generate/CourseGenerate.vue
@@ -9,6 +9,7 @@ const loginState = useLoginState()
 const toast = useToast()
 
 const isCreateCourseModalOpen = ref(false)
+const creationPending = ref(false)
 
 const {
   data: courseList,
@@ -29,6 +30,7 @@ const {
   },
 )
 
+
 const onCreateCourseClick = () => {
   isCreateCourseModalOpen.value = true
 }
@@ -60,24 +62,55 @@ const onCreateCourseSubmit = async (event: FormSubmitEvent<CreateCourseSchema>)
     })
     return
   }
+  creationPending.value = true
   // upload PPTX file
   useFileGo(selected_file.value[0]).then(url => {
-    useFetchWrapped<req.gen.CourseGenCreate & AuthedRequest, resp.gen.CourseGenCreate>('App.Digital_Convert.Create', {
+    useFetchWrapped<req.gen.CourseGenCreate & AuthedRequest, BaseResponse<resp.gen.CourseGenCreate>>('App.Digital_Convert.Create', {
       token: loginState.token!,
       user_id: loginState.user.id,
       task_title: event.data.task_title,
       gen_server: event.data.gen_server as 'main' | 'standby1',
-      speed: event.data.speed,
+      speed: 2 - event.data.speed,
       ppt_url: url,
-      digital_human_id: '40696',
+      digital_human_id: 40696,
       custom_video: '[]',
       opening_url: '',
       ending_url: '',
     }).then(res => {
-      console.log(res)
+      if (res.data.record_status === 1) {
+        toast.add({
+          title: '创建成功',
+          description: '微课视频已开始生成',
+          color: 'green',
+          icon: 'i-tabler-check',
+        })
+        refreshCourseList()
+        isCreateCourseModalOpen.value = false
+      } else {
+        toast.add({
+          title: '创建失败',
+          description: '未知错误',
+          color: 'red',
+          icon: 'i-tabler-alert-triangle',
+        })
+      }
+      creationPending.value = false
+    }).catch(e => {
+      creationPending.value = false
+      toast.add({
+        title: '创建失败',
+        description: e.message || '未知错误',
+        color: 'red',
+        icon: 'i-tabler-alert-triangle',
+      })
     })
   })
 }
+
+onMounted(() => {
+  const i = setInterval(refreshCourseList, 1000 * 5)
+  onBeforeUnmount(() => clearInterval(i))
+})
 </script>
 
 <template>
@@ -94,10 +127,10 @@ const onCreateCourseSubmit = async (event: FormSubmitEvent<CreateCourseSchema>)
       />
     </div>
     <div class="p-4">
-      <div class="relative grid grid-cols-1 sm:grid-cols-2 2xl:grid-cols-4 gap-4">
+      <div class="relative grid grid-cols-1 sm:grid-cols-2 xl:grid-cols-3 2xl:grid-cols-4 gap-4">
         <CGTaskCard v-for="(course, index) in courseList" :key="index" :course="course"/>
       </div>
-      <pre class="overflow-hidden">{{ JSON.stringify(courseList, null, 2) }}</pre>
+      <!--      <pre class="overflow-hidden">{{ JSON.stringify(courseList, null, 2) }}</pre>-->
     </div>
 
     <UModal
@@ -177,6 +210,7 @@ const onCreateCourseSubmit = async (event: FormSubmitEvent<CreateCourseSchema>)
               label="提交"
               type="submit"
               variant="solid"
+              :loading="creationPending"
             />
           </div>
         </UForm>
diff --git a/composables/useDownload.ts b/composables/useDownload.ts
index 301ddc0..7ab99a2 100644
--- a/composables/useDownload.ts
+++ b/composables/useDownload.ts
@@ -1,16 +1,44 @@
-export const useDownload = (url: string, filename: string) => {
+import { EventEmitter } from 'events'
+
+export const useDownload = (url: string, filename: string): {
+  download: () => void
+  progressEmitter: EventEmitter
+} => {
+  const progressEmitter = new EventEmitter()
+
   const download = () => {
-    fetch(url)
-    .then((response) => response.blob())
-    .then((blob) => {
-      const url = window.URL.createObjectURL(new Blob([blob]))
-      const link = document.createElement('a')
-      link.href = url
-      link.setAttribute('download', filename)
-      document.body.appendChild(link)
-      link.click()
-      link.parentNode?.removeChild(link)
-    })
+    const xhr = new XMLHttpRequest()
+    xhr.open('GET', url, true)
+    xhr.responseType = 'blob'
+    xhr.onprogress = (event) => {
+      if (event.lengthComputable) {
+        const percentComplete = (event.loaded / event.total) * 100
+        progressEmitter.emit('progress', percentComplete)
+      }
+    }
+    xhr.onload = function () {
+      if (this.status === 200) {
+        const blob = new Blob([this.response], { type: 'application/octet-stream' })
+        const url = window.URL.createObjectURL(blob)
+        const link = document.createElement('a')
+        link.href = url
+        link.setAttribute('download', filename)
+        document.body.appendChild(link)
+        link.click()
+        link.parentNode?.removeChild(link)
+        progressEmitter.emit('done')
+      } else {
+        progressEmitter.emit('error', new Error('资源已过期或不存在'))
+      }
+    }
+    xhr.onerror = function () {
+      progressEmitter.emit('error', new Error('网络错误，下载失败'))
+    }
+    xhr.send()
+  }
+
+  return {
+    download,
+    progressEmitter,
   }
-  return { download }
 }
\ No newline at end of file
diff --git a/package.json b/package.json
index 1c208ea..acaaf88 100644
--- a/package.json
+++ b/package.json
@@ -18,6 +18,7 @@
     "@nuxt/ui": "^2.14.1",
     "@uniiem/object-trim": "^0.2.0",
     "@uniiem/uuid": "^0.2.1",
+    "events": "^3.3.0",
     "highlight.js": "^11.9.0",
     "idb-keyval": "^6.2.1",
     "markdown-it": "^14.1.0",
diff --git a/typings/types.d.ts b/typings/types.d.ts
index d26c433..875069f 100644
--- a/typings/types.d.ts
+++ b/typings/types.d.ts
@@ -71,7 +71,7 @@ namespace req {
      * @param digital_human_id 数字人物 ID (string || number 存疑)
      */
     interface CourseGenCreate {
-      digital_human_id: string
+      digital_human_id: number
       ppt_url: string
       opening_url: string
       ending_url: string
diff --git a/yarn.lock b/yarn.lock
index 979d8e6..e4a8879 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -4118,6 +4118,11 @@ etag@^1.8.1, etag@~1.8.1:
   resolved "https://registry.npmmirror.com/etag/-/etag-1.8.1.tgz#41ae2eeb65efa62268aebfea83ac7d79299b0887"
   integrity sha512-aIL5Fx7mawVa300al2BnEE4iNvo1qETxLrPI/o05L7z6go7fCw1J6EQmbK4FmJ2AS7kgVF/KEZWufBfdClMcPg==
 
+events@^3.3.0:
+  version "3.3.0"
+  resolved "https://registry.yarnpkg.com/events/-/events-3.3.0.tgz#31a95ad0a924e2d2c419a813aeb2c4e878ea7400"
+  integrity sha512-mQw+2fkQbALzQ7V0MY0IqdnXNOeTtP4r0lN9z7AAawCXgqea7bDii20AYrIBrFd/Hx0M2Ocz6S111CaFkUcb0Q==
+
 execa@^5.1.1:
   version "5.1.1"
   resolved "https://registry.npmmirror.com/execa/-/execa-5.1.1.tgz#f80ad9cbf4298f7bd1d4c9555c21e93741c411dd"