From f3b80e76fa9148a699c4d3140cddfb9801b6248f Mon Sep 17 00:00:00 2001
From: HugoPoi <hugo.poissonnet@gmail.com>
Date: Sat, 4 Jul 2020 17:30:25 +0200
Subject: [PATCH] feat: add myfreemp3 download source

---
 index.js | 40 +++++++++++++++++++++++++++-------------
 1 file changed, 27 insertions(+), 13 deletions(-)

diff --git a/index.js b/index.js
index 770448a..648295f 100644
--- a/index.js
+++ b/index.js
@@ -44,19 +44,33 @@ async function searchOnVkMusic(query){
     if(responseBody.error){
       throw _.assign(responseBody.error, new Error(responseBody.error.error_msg));
     } else {
-      return responseBody;
+      return responseBody.response.items;
     }
   });
 }
 
+async function searchOnMyFreeMp3(query){
+  const url = new URL('https://myfreemp3cc.com/api/search.php?callback=callback');
+
+  return await fetch(url, {
+    method: 'POST',
+    body: new URLSearchParams({
+      q: query,
+      page: 0,
+    }),
+  })
+  .then(res => res.text())
+  .then(jsonp => vm.runInNewContext(jsonp, { callback: (payload) => payload.response }));
+}
+
 function matchScore(spotifyMetas, vkmusicMetas){
   const originalArtistNames = _.map(spotifyMetas.artists, 'name').join(', ').toLowerCase();
   const originalTitle = spotifyMetas.name.toLowerCase();
   const originalDuration = Math.round(spotifyMetas.duration_ms/1000);
 
-  const matchArtistScore = 1 - (leven(originalArtistNames, vkmusicMetas.artist.toLowerCase()) / Math.max(originalArtistNames.length, vkmusicMetas.artist.length));
-  const matchTitleScore =  1 - (leven(originalTitle, vkmusicMetas.title.toLowerCase()) / Math.max(originalTitle.length, vkmusicMetas.title.length));
-  const matchDurationScore = 1 - (Math.abs(originalDuration - vkmusicMetas.duration) / originalDuration); // TODO this can return more than 1 or less than 0
+  const matchArtistScore = 1 - (leven(originalArtistNames, _.get(vkmusicMetas, 'artist', '').toLowerCase()) / Math.max(originalArtistNames.length, _.get(vkmusicMetas, 'artist', '').length));
+  const matchTitleScore =  1 - (leven(originalTitle, _.get(vkmusicMetas, 'title', '').toLowerCase()) / Math.max(originalTitle.length, _.get(vkmusicMetas, 'title', '').length));
+  const matchDurationScore = 1 - (Math.abs(originalDuration - _.get(vkmusicMetas, 'duration', 0)) / originalDuration); // TODO this can return more than 1 or less than 0
   debug('matchArtistScore=%f matchTitleScore=%f matchDurationScore=%f', matchArtistScore, matchTitleScore, matchDurationScore);
   return matchArtistScore + matchTitleScore + matchDurationScore;
 }
@@ -72,19 +86,19 @@ async function main(){
   const vkPlaylist = await Promise.map(playlist.tracks.items, async ({track}) => {
     const artistNames = _.map(track.artists, 'name').join(', ');
     debug('%s - %s', track.name, artistNames);
-    const { response } = await searchOnVkMusic(`${artistNames} ${track.name}`);
-    debug('response=%O', response);
-    const bestMatch = _.chain(response.items).map((item) => {
+    const items = await searchOnMyFreeMp3(`${artistNames} ${track.name}`);
+    debug('items=%O', items);
+    const bestMatch = _.chain(items).map((item) => {
       item.score = matchScore(track, item);
       return item;
-    }).sortBy('score').last().value();
-    debug('bestMatch=%O', bestMatch);
+    }).filter(item => item.score && item.url).sortBy('score').last().value();
     if(!bestMatch){
       console.log(`You are on your own for ${track.name} - ${artistNames}`);
       return;
     }
     bestMatch.path = `${bestMatch.artist} - ${bestMatch.title}.mp3`;
-    await fs.access(bestMatch.path).catch(async () => {
+    debug('bestMatch=%O', bestMatch);
+    await fs.access(bestMatch.path).catch(async () => { // TODO find a proper way to not re-download, lower/uppercase problems
       await fetch(bestMatch.url).then(res => {
         res.body.pipe(createWriteStream(bestMatch.path));
       });
@@ -96,11 +110,11 @@ async function main(){
   await generateM3U8Playlist(_.compact(vkPlaylist));
 }
 
-//main().catch(err => console.error(err));
+main().catch(err => console.error(err));
 
 async function test(query){
-  const { response } = await searchOnVkMusic(query);
+  const response = await searchOnMyFreeMp3(query);
   console.log(response);
 }
 
-test(process.argv[2]);
+//test(process.argv[2]);