From fecd248a53cfa9f378cd2f5f3df11aed71a876ef Mon Sep 17 00:00:00 2001
From: uh wot <uhwot@protonmail.com>
Date: Wed, 15 Sep 2021 17:43:53 +0200
Subject: [PATCH] made code slightly less shitty

---
 dzunlock.user.js | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/dzunlock.user.js b/dzunlock.user.js
index b5b6b3c..4ffb231 100644
--- a/dzunlock.user.js
+++ b/dzunlock.user.js
@@ -3,7 +3,7 @@
 // @namespace   io.github.uhwot.dzunlock
 // @description enables deezer hifi features lol
 // @author      uh wot
-// @version     1.3
+// @version     1.3.1
 // @license     GPL-3.0-only
 // @homepageURL https://git.freezer.life/uhwot/dzunlock
 // @downloadURL https://git.freezer.life/uhwot/dzunlock/raw/branch/master/dzunlock.user.js
@@ -159,16 +159,11 @@ fetchIntercept.register({
             const track = unsafeWindow.dzPlayer.getCurrentSong()
             const id = parseInt(track.SNG_ID)
 
-            let user_status_quality = quality
-            if (user_status_quality === 'high') {
-                user_status_quality = 'hq'
-            }
-
             let is_subbed = !unsafeWindow.dzPlayer.user_status.can_subscribe
-            let is_quality_available = unsafeWindow.dzPlayer.user_status[user_status_quality]
+            let is_quality_available = unsafeWindow.dzPlayer.user_status.audio_qualities.wifi_download.includes(quality)
             // STREAM_ADS_AVAILABLE is used to check if track is restricted to premium/hifi
-            if (quality === 'standard' && (track.RIGHTS.STREAM_ADS_AVAILABLE === true || is_subbed)) {
-                is_quality_available = true
+            if (track.RIGHTS.STREAM_ADS_AVAILABLE !== true && !is_subbed) {
+                is_quality_available = false
             }
 
             if (id >= 0 && !is_quality_available) {