From 4128d9ac3b0a389bb27731e5eae5c0f31ba51e79 Mon Sep 17 00:00:00 2001
From: njzy <njzydark@gmail.com>
Date: Wed, 9 Dec 2020 19:53:19 +0800
Subject: [PATCH] feat: support cache songs (#80)

fix: cache unblock song

refactor: play first and then cache it
---
 src/store/actions.js | 66 ++++++++++++++++++++++++++++++--------------
 src/utils/db.js      | 34 ++++++++++-------------
 2 files changed, 61 insertions(+), 39 deletions(-)

diff --git a/src/store/actions.js b/src/store/actions.js
index 76305b7..4200e0c 100644
--- a/src/store/actions.js
+++ b/src/store/actions.js
@@ -1,5 +1,5 @@
 import { updateMediaSessionMetaData } from "@/utils/mediaSession";
-import { getTrackDetail, scrobble, getMP3 } from "@/api/track";
+import { getTrackDetail, scrobble, getMP3 as getMP3Api } from "@/api/track";
 import { isAccountLoggedIn } from "@/utils/auth";
 import { updateHttps } from "@/utils/common";
 import localforage from "localforage";
@@ -40,17 +40,18 @@ export default {
       updateMediaSessionMetaData(track);
       document.title = `${track.name} · ${track.ar[0].name} - YesPlayMusic`;
 
+      let unblockSongUrl = null;
       if (track.playable === false) {
         let res = undefined;
         if (process.env.IS_ELECTRON === true) {
           res = ipcRenderer.sendSync("unblock-music", track);
         }
         if (res?.url) {
-          commitMP3(res.url);
+          unblockSongUrl = res.url;
         } else {
           dispatch("nextTrack");
+          return;
         }
-        return;
       }
 
       function commitMP3(mp3) {
@@ -59,31 +60,56 @@ export default {
           dispatch("nextTrack");
         });
       }
+
+      function getMP3(id) {
+        return getMP3Api(id).then((data) => {
+          // 未知情况下会没有返回数据导致报错，增加防范逻辑
+          if (data.data[0]) {
+            const url = updateHttps(data.data[0].url);
+            commitMP3(url);
+            return url;
+          }
+        });
+      }
+
       if (isAccountLoggedIn()) {
         if (store.state.settings.automaticallyCacheSongs === true) {
           let tracks = localforage.createInstance({
             name: "tracks",
           });
-          tracks.getItem(`${track.id}`).then((t) => {
-            if (t !== null) {
-              commitMP3(URL.createObjectURL(t.mp3));
-            } else {
-              cacheTrack(`${track.id}`).then((t) => {
-                commitMP3(URL.createObjectURL(t.mp3));
-              });
-            }
-          });
+          tracks
+            .getItem(`${track.id}`)
+            .then((t) => {
+              if (t !== null) {
+                const blob = new Blob([t.mp3]);
+                commitMP3(URL.createObjectURL(blob));
+              } else {
+                if (unblockSongUrl) {
+                  commitMP3(unblockSongUrl);
+                  cacheTrack(`${track.id}`, unblockSongUrl);
+                } else {
+                  getMP3(track.id).then((url) => {
+                    cacheTrack(`${track.id}`, url);
+                  });
+                }
+              }
+            })
+            .catch((err) => {
+              console.log(err.messaeg);
+              if (unblockSongUrl) {
+                commitMP3(unblockSongUrl);
+              } else {
+                getMP3(track.id);
+              }
+            });
         } else {
-          getMP3(track.id).then((data) => {
-            // 未知情况下会没有返回数据导致报错，增加防范逻辑
-            if (data.data[0]) {
-              const url = updateHttps(data.data[0].url);
-              commitMP3(url);
-            }
-          });
+          getMP3(track.id);
         }
       } else {
-        commitMP3(`https://music.163.com/song/media/outer/url?id=${track.id}`);
+        commitMP3(
+          unblockSongUrl ||
+            `https://music.163.com/song/media/outer/url?id=${track.id}`
+        );
       }
     });
   },
diff --git a/src/utils/db.js b/src/utils/db.js
index 7c8ffc6..e7bb5bf 100644
--- a/src/utils/db.js
+++ b/src/utils/db.js
@@ -1,11 +1,10 @@
-// import axios from "axios";
+import axios from "axios";
 import localforage from "localforage";
-import { getMP3 } from "@/api/track";
 
-export function cacheTrack(id) {
-  // let tracks = localforage.createInstance({
-  //   name: "tracks",
-  // });
+export function cacheTrack(id, url) {
+  let tracks = localforage.createInstance({
+    name: "tracks",
+  });
 
   // TODO: limit cache songs number
   // tracks.length().then(function (length) {
@@ -17,17 +16,14 @@ export function cacheTrack(id) {
   // });
 
   // TODO: cache track details
-  return getMP3(id).then((data) => {
-    // return axios
-    //   .get(data.data[0].url.replace(/^http:/, "https:"), {
-    //     responseType: "blob",
-    //   })
-    //   .then((data) => {
-    //     tracks.setItem(`${id}`, { mp3: data.data });
-    //     return { mp3: data.data };
-    //   });
-    return { mp3: data.data[0].url.replace(/^http:/, "https:") };
-  });
+  return axios
+    .get(url, {
+      responseType: "arraybuffer",
+    })
+    .then((response) => {
+      tracks.setItem(`${id}`, { mp3: response.data });
+      return { mp3: response.data };
+    });
 }
 
 export function countDBSize(dbName) {
@@ -37,11 +33,11 @@ export function countDBSize(dbName) {
   let trackSizes = [];
   return db
     .iterate((value) => {
-      trackSizes.push(value.mp3.size);
+      trackSizes.push(value.mp3.byteLength);
     })
     .then(() => {
       return {
-        bytes: trackSizes.reduce((s1, s2) => s1 + s2),
+        bytes: trackSizes.reduce((s1, s2) => s1 + s2, 0),
         length: trackSizes.length,
       };
     })