From dd5e3897392a9edea21a4a744bc115cd090c38c6 Mon Sep 17 00:00:00 2001
From: "Renato \"Lond\" Cerqueira" <renato@lond.com.br>
Date: Sun, 17 Jan 2021 12:05:16 +0100
Subject: [PATCH] Add check for streamUrl

---
 lib/services/ytmusic/index.js | 3 ++-
 test/services/ytmusic.js      | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/lib/services/ytmusic/index.js b/lib/services/ytmusic/index.js
index 85335b3..b2fe78b 100644
--- a/lib/services/ytmusic/index.js
+++ b/lib/services/ytmusic/index.js
@@ -130,7 +130,7 @@ async function lookupTrack(id) {
     type: 'track',
     id: song_meta.videoId,
     name: song_meta.title,
-    streamUrl: null,
+    streamUrl: `https://music.youtube.com/watch?v=${song_meta.videoId}`,
     purchaseUrl: null,
     artwork,
     artist: {
@@ -184,6 +184,7 @@ async function lookupAlbum(id) {
     id,
     name: album_data.title,
     streamUrl: null,
+    streamUrl: `https://music.youtube.com/browse/${id}`,
     purchaseUrl: null,
     artwork,
     artist: {
diff --git a/test/services/ytmusic.js b/test/services/ytmusic.js
index 5d27a4b..1579123 100644
--- a/test/services/ytmusic.js
+++ b/test/services/ytmusic.js
@@ -37,10 +37,12 @@ describe('ytmusic', function(){
       it('should parse track url into ID', async function (){
         const result = await ytmusic.parseUrl('https://music.youtube.com/watch?v=YLp2cW7ICCU&feature=share');
         result.id.should.equal("YLp2cW7ICCU");
+        result.streamUrl.should.equal("https://music.youtube.com/watch?v=YLp2cW7ICCU");
       });
       it('should parse album url into ID', async function (){
         const result = await ytmusic.parseUrl('https://music.youtube.com/browse/MPREb_9C36yscfgmJ');
         result.id.should.equal("MPREb_9C36yscfgmJ");
+        result.streamUrl.should.equal("https://music.youtube.com/browse/MPREb_9C36yscfgmJ");
       });
       it('should parse alternative album url into ID', async function (){
         const result = await ytmusic.parseUrl('https://music.youtube.com/playlist?list=OLAK5uy_lx9K5RpiBEwd3E4C1GKqY7e06qTlwydvs');