From 0b0b19163b6fb9c714b25e843d24dcc3fe2920cd Mon Sep 17 00:00:00 2001
From: MengYX <admin@mengyx.cn>
Date: Thu, 12 Sep 2019 15:51:10 +0800
Subject: [PATCH] Complete ID3 for ncm

---
 README.md          |  2 +-
 src/plugins/ncm.js | 38 +++++++++++++++++++++++++++++---------
 2 files changed, 30 insertions(+), 10 deletions(-)

diff --git a/README.md b/README.md
index a0af25f..174fae5 100644
--- a/README.md
+++ b/README.md
@@ -10,8 +10,8 @@
 - [x] Play instantly 在线播放
 - [x] Batch unlocking 批量解锁
 - [x] Progressive Web App 渐进式Web应用
+- [x] Complete ID3 for ncm 补全ncm的ID3信息
 - [ ] Multi-language 多语言
-- [ ] Write ID3 写入ID3信息
 
 ## Project setup
 ```
diff --git a/src/plugins/ncm.js b/src/plugins/ncm.js
index e36fac0..eab76f4 100644
--- a/src/plugins/ncm.js
+++ b/src/plugins/ncm.js
@@ -1,4 +1,5 @@
 const CryptoJS = require("crypto-js");
+const ID3Writer = require("browser-id3-writer");
 const CORE_KEY = CryptoJS.enc.Hex.parse("687a4852416d736f356b496e62617857");
 const META_KEY = CryptoJS.enc.Hex.parse("2331346C6A6B5F215C5D2630553C2728");
 
@@ -7,7 +8,6 @@ const audio_mime_type = {
     flac: "audio/flac"
 };
 
-
 export {Decrypt};
 
 async function Decrypt(file) {
@@ -121,15 +121,14 @@ async function Decrypt(file) {
 
     offset += dataView.getUint32(offset + 5, true) + 13;
 
-    const audioData = new Uint8Array(fileBuffer, offset);
-    const audioDataLen = audioData.length;
+    let audioData = new Uint8Array(fileBuffer, offset);
+    let audioDataLen = audioData.length;
 
 
     for (let cur = 0; cur < audioDataLen; ++cur) {
         audioData[cur] ^= keyBox[cur & 0xff];
     }
 
-
     if (musicMeta.format === undefined) {
         musicMeta.format = (() => {
             const [f, L, a, C] = audioData;
@@ -140,16 +139,37 @@ async function Decrypt(file) {
         })();
     }
     const mime = audio_mime_type[musicMeta.format];
-    const musicData = new Blob([audioData], {
-        type: mime
-    });
-
-    const musicUrl = URL.createObjectURL(musicData);
 
     const artists = [];
     musicMeta.artist.forEach(arr => {
         artists.push(arr[0]);
     });
+
+    if (musicMeta.format === "mp3") {
+        const writer = new ID3Writer(audioData);
+        writer.setFrame("TPE1", artists)
+            .setFrame("TIT2", musicMeta.musicName)
+            .setFrame("TALB", musicMeta.album);
+        if (musicMeta.albumPic !== "") {
+            try {
+                const img = await (await fetch(musicMeta.albumPic)).arrayBuffer();
+                writer.setFrame('APIC', {
+                    type: 3,
+                    data: img,
+                    description: 'Cover'
+                })
+            } catch (e) {
+                console.log("Fail to write cover image!");
+            }
+        }
+        writer.addTag();
+        audioData = writer.arrayBuffer;
+    }
+
+    const musicData = new Blob([audioData], {
+        type: mime
+    });
+    const musicUrl = URL.createObjectURL(musicData);
     const filename = artists.join(" & ") + " - " + musicMeta.musicName + "." + musicMeta.format;
     return {
         meta: musicMeta,