#!/usr/bin/env python3

import urllib.request
import json
import sys

ALBUMS = [
    324660713,
    15427733,
    464178301,
    75115890,
    410197513,
    418729278,
    504004321,
    510893864,
    325723583,
    336178142,
    106369871,
    423471869,
]

TOKEN = "eyJraWQiOiJ2OU1GbFhqWSIsImFsZyI6IkVTMjU2In0.eyJ0eXBlIjoibzJfYWNjZXNzIiwic2NvcGUiOiIiLCJnVmVyIjowLCJzVmVyIjowLCJjaWQiOjEzNTU3LCJhdCI6IklOVEVSTkFMIiwiZXhwIjoxNzc1MTI4ODUzLCJpc3MiOiJodHRwczovL2F1dGgudGlkYWwuY29tL3YxIn0.qRoN8BRLM3R5WAXM3kS2hkWyaGk5tWF0FaHWJmkrWNvI48hKyS9lhVOTSnP1XkFEfdXv6aTzGUNUewyp-O_d3w"

HEADERS = {
    "accept": "*/*",
    "authorization": f"Bearer {TOKEN}",
}

def fetch_album(album_id):
    url = f"https://api.tidal.com/v1/albums/{album_id}?countryCode=US"
    req = urllib.request.Request(url, headers=HEADERS)
    try:
        with urllib.request.urlopen(req) as resp:
            return json.loads(resp.read().decode())
    except Exception as e:
        print(f"Error fetching {album_id}: {e}", file=sys.stderr)
        return None

def transform_album(api_data):
    return {
        "type": "album",
        "id": api_data.get("id"),
        "title": api_data.get("title"),
        "artist": {
            "id": api_data.get("artist", {}).get("id"),
            "name": api_data.get("artist", {}).get("name"),
        },
        "releaseDate": api_data.get("releaseDate"),
        "cover": api_data.get("cover"),
        "explicit": api_data.get("explicit"),
        "audioQuality": api_data.get("audioQuality"),
        "mediaMetadata": api_data.get("mediaMetadata"),
    }

picks = []
for album_id in ALBUMS:
    data = fetch_album(album_id)
    if data:
        picks.append(transform_album(data))

with open("public/editors-picks.json", "w") as f:
    json.dump(picks, f, indent=4)

print(f"Written {len(picks)} albums to public/editors-picks.json")