From 59b8c857cebc5bf2a2d346fe7df5efdac54b2699 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Thu, 16 Dec 2021 11:31:41 +0100
Subject: [PATCH] initial commit

---
 .gitignore           |  2 ++
 README               |  3 +++
 secret.json.template |  1 +
 wibe.py              | 56 ++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 62 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 README
 create mode 100644 secret.json.template
 create mode 100644 wibe.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..5133d2a
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+secret*.json
+.cache-*
diff --git a/README b/README
new file mode 100644
index 0000000..babe71b
--- /dev/null
+++ b/README
@@ -0,0 +1,3 @@
+# Project wibe
+
+
diff --git a/secret.json.template b/secret.json.template
new file mode 100644
index 0000000..659c1ae
--- /dev/null
+++ b/secret.json.template
@@ -0,0 +1 @@
+[<username>,<client_id>,<client_secret>]
diff --git a/wibe.py b/wibe.py
new file mode 100644
index 0000000..f57d430
--- /dev/null
+++ b/wibe.py
@@ -0,0 +1,56 @@
+import spotipy
+from spotipy import util as spoUtil
+import requests
+import io
+import colorgram
+import json
+import time
+from pprint import pprint
+
+wledPalettes = {'colorsOnly': 5, 'colorGradient': 4, '2Colors': 3}
+
+def main(token, ip='192.168.3.42', bri=51):
+    url = 'http://'+ip+'/json/state'
+
+    sp = spotipy.Spotify(auth=token)
+
+    curTrack = None
+    requests.post(url, json={"on":True,"bri":bri,"transition":7,"mainseg":0,"seg":[{"id":0,"grp":1,"spc":0,"of":0,"on":True,"bri":255,"cct":127,"col":[[0,0,0,255],[0,0,0,0],[0,0,0,0]],"fx":38,"sx":24,"ix":16,"pal":5,"sel":True,"rev":False,"mi":False}]}).json()
+    while True:
+        res = sp.current_user_playing_track()
+        if res and res["item"] and (curTrack == None or curTrack['uri'] != res["item"]['uri']):
+            curTrack = res["item"]
+            del curTrack['available_markets']
+            del curTrack['album']['available_markets']
+            print("[*] Now Playing: "+curTrack["name"])
+            feat = sp.audio_features(curTrack["uri"])[0]
+            palette = getPalette(curTrack,3)
+            #pprint(curTrack)
+            #pprint(feat)
+            #pprint(palette)
+            speed = min(int(feat['tempo']/2),255)
+            power = 16 + int(feat['energy'])*64
+            pprint(requests.post(url, json={"seg": [{"col": palette, "pal": wledPalettes['colorGradient'], 'sx': speed, 'ix': power}]}).json())
+            time.sleep(3)
+        time.sleep(1)
+
+def getPalette(curTrack,numColors):
+    imgStream = requests.get(curTrack["album"]["images"][-1]["url"]).content #Get the lowest res cover
+    imgFile = io.BytesIO(imgStream)
+    print("[i] Generating palette")
+    palette = [normalizeColor([col.rgb[c] for c in range(3)]+[0]) for col in sorted(colorgram.extract(imgFile, numColors), key=lambda c: c.hsl.h)] #c.proportion
+    return palette
+
+def normalizeColor(rgbw):
+    s = max(rgbw)
+    fac = (1 + 255/s)/2
+    return [min(int(c*fac),255) for c in rgbw]
+
+def genSpoToken(username, client_id, client_secret):
+    return spoUtil.prompt_for_user_token(username,"user-read-currently-playing",client_id=client_id,client_secret=client_secret,redirect_uri='http://localhost/')
+
+if __name__=="__main__":
+    with open('secret.json') as f:
+        username, client_id, client_secret = json.loads(f.read())
+    token = genSpoToken(username, client_id, client_secret)
+    main(token,'192.168.3.42')