refactor: support download flac at app

radishes-music · Dec 3, 2023 · 772a46f · 772a46f
1 parent 5760f81
commit 772a46f
Showing 1 changed file with 10 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,16 @@ const nodeDownload = async (
     })
     response.on('end', async () => {
       const buffer = Buffer.concat(chunks)
-      const path = join(downloadPath, name + suffix)
+
+      let suffixName = ''
+
+      const urlSuffix = url.replace(/.+\./, '')
+      if (['.mp3', '.flac'].includes(urlSuffix)) {
+        suffixName = urlSuffix
+      } 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)