feat: wasm plugin switch for whether to mux metadata

the1812 · Jul 30, 2024 · 3684f7d · 3684f7d
1 parent 0638dbd
commit 3684f7d
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 11 deletions.
diff --git a/registry/lib/plugins/video/download/wasm-output/Config.vue b/registry/lib/plugins/video/download/wasm-output/Config.vue
@@ -0,0 +1,41 @@
+<template>
+  <div v-if="shouldShow" class="download-video-config-item" style="flex-wrap: wrap">
+    <div class="download-video-config-title">写入元数据：</div>
+    <SwitchBox v-model="muxWithMetadata" @change="saveOptions" />
+    <div class="download-video-config-description" style="width: 100%">
+      仅支持元数据类型「ffmetadata」
+    </div>
+  </div>
+</template>
+
+<script lang="ts">
+import { SwitchBox } from '@/ui'
+import { isComponentEnabled, getComponentSettings } from '@/core/settings'
+
+interface Options {
+  muxWithMetadata: boolean
+}
+const defaultOptions: Options = {
+  muxWithMetadata: false,
+}
+const { options: storedOptions } = getComponentSettings('downloadVideo')
+const options: Options = { ...defaultOptions, ...storedOptions }
+export default Vue.extend({
+  components: {
+    SwitchBox,
+  },
+  data() {
+    const shouldShow = isComponentEnabled('saveVideoMetadata')
+    return {
+      shouldShow,
+      muxWithMetadata: shouldShow && options.muxWithMetadata,
+    }
+  },
+  methods: {
+    saveOptions() {
+      options.muxWithMetadata = this.muxExtraAssets
+      Object.assign(storedOptions, options)
+    },
+  },
+})
+</script>
diff --git a/registry/lib/plugins/video/download/wasm-output/handler.ts b/registry/lib/plugins/video/download/wasm-output/handler.ts
@@ -87,23 +87,25 @@ async function single(
   )
 }
 
-export async function run(action: DownloadVideoAction) {
+export async function run(action: DownloadVideoAction, muxWithMetadata: boolean) {
   if (!ffmpeg.loaded) {
     await loadFFmpeg()
   }
 
   const { infos: pages, extraAssets } = action
 
   let ffmetadata: PackageEntry[]
-  const extraAssetsForBrowser = []
-  for (const { asset, instance } of extraAssets) {
-    if (!ffmetadata && asset.name === 'saveVideoMetadata' && instance.type === 'ffmetadata') {
-      ffmetadata = await asset.getAssets(pages, instance)
-    } else {
-      extraAssetsForBrowser.push({ asset, instance })
+  if (muxWithMetadata) {
+    const extraAssetsForBrowser = []
+    for (const { asset, instance } of extraAssets) {
+      if (!ffmetadata && asset.name === 'saveVideoMetadata' && instance.type === 'ffmetadata') {
+        ffmetadata = await asset.getAssets(pages, instance)
+      } else {
+        extraAssetsForBrowser.push({ asset, instance })
+      }
     }
+    action.extraAssets = extraAssetsForBrowser
   }
-  action.extraAssets = extraAssetsForBrowser
 
   const { dashAudioExtension, dashFlacAudioExtension, dashVideoExtension } =
     getComponentSettings<Options>('downloadVideo').options

diff --git a/registry/lib/plugins/video/download/wasm-output/index.ts b/registry/lib/plugins/video/download/wasm-output/index.ts
@@ -19,14 +19,15 @@ export const plugin: PluginMetadata = {
       outputs.push({
         name: 'wasm',
         displayName: 'WASM',
-        description: `${desc}，运行过程中请勿关闭页面，初次使用或清除缓存后需要加载约 30 MB 的 WASM 文件`,
-        runAction: async action => {
+        description: `${desc}。运行过程中请勿关闭页面，初次使用或清除缓存后需要加载约 30 MB 的 WASM 文件。`,
+        runAction: async (action, instance) => {
           try {
-            await run(action)
+            await run(action, instance.muxWithMetadata)
           } catch (error) {
             Toast.error(String(error), title)
           }
         },
+        component: () => import('./Config.vue').then(m => m.default),
       })
     })
   },