import os,json,base64,requests,time,subprocess,sys,concurrent.futures
from mutagen.easyid3 import EasyID3;from mutagen.mp3 import MP3
from mutagen.id3 import ID3,APIC,error

CONFIG_FILE="config.json";DOWNLOAD_DIR="downloads"
SPOTIFY_TRACK_API="https://api.spotify.com/v1/tracks/{}"
SPOTIFY_PLAYLIST_API="https://api.spotify.com/v1/playlists/{}/tracks"
INVIDIOUS_API=None

def load_config(): # opens the config file
 with open(CONFIG_FILE) as f:return json.load(f)

def ensure_valid_token():
 global INVIDIOUS_API
 c=load_config()
 INVIDIOUS_API=c.get("invidious_url","https://invidious.snopyta.org/api/v1/search?type=video&q={query}")
 if "access_token"not in c or int(time.time())>=c.get("expires_at",0):
  print("no token running oauth")
  subprocess.run(["python3","spotify_oauth.py"],check=True)
  c=load_config()
 return c["access_token"]

def get_liked_tracks(tok): # gets liked songs i guess
 l,r=[],0;h={"Authorization":f"Bearer {tok}"}
 while 1:
  u=f"https://api.spotify.com/v1/me/tracks?limit=50&offset={r}"
  d=requests.get(u,headers=h)
  if d.status_code==401:print("nah bro token's dead");break
  j=d.json();i=j.get("items",[])
  for it in i:l.append(it["track"]["id"])
  if len(i)<50:break
  r+=50;time.sleep(0.5)
 return l

def get_playlist_tracks(tok,pid): # does same thing but like for playlists
 l,r=[],0;h={"Authorization":f"Bearer {tok}"}
 while 1:
  u=f"{SPOTIFY_PLAYLIST_API.format(pid)}?limit=100&offset={r}"
  d=requests.get(u,headers=h)
  if d.status_code==401:print("nope, bad token");break
  if d.status_code==404:print("what playlist");break
  j=d.json();i=j.get("items",[])
  for t in i:
   tr=t.get("track")
   if tr and tr.get("id"):l.append(tr["id"])
  if len(i)<100:break
  r+=100;time.sleep(0.5)
 return l

def fetch_metadata(tid,tok): # gets info about a track
 return requests.get(SPOTIFY_TRACK_API.format(tid),headers={"Authorization":f"Bearer {tok}"}).json()

def sanitize_filename(n): # removes stuff that windows would cry about
 return"".join(c for c in n if c not in r'\/:*?"<>|').strip()

def search_invidious(q): # looks up stuff on invidious, yeah
 try:
  r=requests.get(INVIDIOUS_API.format(query=q),timeout=10);r.raise_for_status()
  for v in r.json():
   if v.get("videoId"):return f"https://www.youtube.com/watch?v={v['videoId']}"
 except:print("couldn't search :(");return None

def download_from_youtube(url,f): # yoink an mp3
 os.makedirs(DOWNLOAD_DIR,exist_ok=True)
 p=os.path.join(DOWNLOAD_DIR,f)
 if os.path.exists(p):print(f"{f}? already here bro");return True
 c=[];copt=["--cookies","cookies.txt"] if os.path.exists("cookies.txt") else []
 cmd=["yt-dlp",*copt,"-x","--audio-format","mp3","--no-playlist","--no-live-from-start","-o",p,url]
 try:
  subprocess.run(cmd,check=True);print(f"yanked {f}");return True
 except:print(f"nope, didn't work: {f}");return False

def embed_album_art(path,img): # slap on some pic
 try:
  i=requests.get(img).content
  a=MP3(path,ID3=ID3)
  try:a.add_tags()
  except:error()
  a.tags.add(APIC(encoding=3,mime='image/jpeg',type=3,desc='Cover',data=i))
  a.save();print(f"art slapped on {path}")
 except Exception as e:print(f"eh: {e}")

def tag_mp3_with_spotify_metadata(p,m): # add title n stuff
 try:
  a=MP3(p,ID3=EasyID3)
  a["title"]=m.get("name","")
  a["artist"]=", ".join(z["name"]for z in m.get("artists",[]))
  a["album"]=m.get("album",{}).get("name","")
  a["date"]=m.get("album",{}).get("release_date","")
  a.save();print(f"tagged: {p}")
  i=m.get("album",{}).get("images",[])
  if i:embed_album_art(p,i[0]["url"])
 except Exception as e:print(f"tag fail on {p}: {e}")

def process_track(tid,tok): # the whole shabang
 m=fetch_metadata(tid,tok)
 if not m:print(f"nah nothing for {tid}");return False
 a=m["artists"][0]["name"];t=m["name"]
 s=f"{a} - {t} audio";f=sanitize_filename(f"{a} - {t}.mp3")
 u=search_invidious(s)
 if not u:print(f"nothing on invidious for {s}");return False
 if not download_from_youtube(u,f):print(f"skipping {f}");return False
 print(f"yoinked {f}");tag_mp3_with_spotify_metadata(os.path.join(DOWNLOAD_DIR,f),m)
 return True

if __name__=="__main__":
 p=sys.argv[1]if len(sys.argv)>1 else None
 c=load_config();t=ensure_valid_token()
 if p:
  print(f"playlist: {p}");ids=get_playlist_tracks(t,p)
 else:
  print("liked songs");ids=get_liked_tracks(t)
 print(f"{len(ids)} songs found, whew")
 with concurrent.futures.ThreadPoolExecutor(max_workers=8)as x:
  fut={x.submit(process_track,i,t):i for i in ids}
  for f in concurrent.futures.as_completed(fut):
   try:f.result()
   except Exception as e:print(f"welp: {e}")