refactor: support download flac at app

radishes-music · Dec 3, 2023 · 2cd4da8 · 2cd4da8
1 parent 5760f81
commit 2cd4da8
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/electron/event/ipc-main/download.ts b/src/electron/event/ipc-main/download.ts
@@ -99,7 +99,20 @@ const nodeDownload = async (
     })
     response.on('end', async () => {
       const buffer = Buffer.concat(chunks)
-      const path = join(downloadPath, name + suffix)
+
+      let suffixName = ''
+      // eslint-disable-next-line no-useless-escape
+      const match_result = url.match(/\.[^\.]+$/)
+      if (match_result) {
+        const suffixStr = match_result[0]
+        if (['.mp3', '.flac'].includes(suffixStr)) {
+          suffixName = suffixStr
+        } else {
+          suffixName = suffix
+        }
+      }
+
+      const path = join(downloadPath, name + suffixName)
       const comm = Buffer.from(String(id)).toString('base64')
       const { renderBuffer, addTag } = writeBufferID3(buffer)
       addTag('TIT2', name)