import requests
import os
import re
from datetime import datetime, timedelta
from dotenv import load_dotenv

load_dotenv()

class Spotify():
    def __init__(self):
        self.token = None
        self.valid_until = None

    def token_valid(self):
        if self.token and self.valid_until and (datetime.now() < self.valid_until):
            return True
        return False

    def get_token(self):
        if self.token_valid():
            print("Token still valid")
            return self.token
        print("New token")
        url = 'https://accounts.spotify.com/api/token'
        headers = {
            'Content-Type': 'application/x-www-form-urlencoded',
            'Authorization': 'Basic ' + os.getenv('SPOTIFY_ENCODED_ID')
        }
        data = {
            'grant_type': 'client_credentials'
        }
        r = requests.post(url, headers=headers, data=data)
        data = r.json()
        self.token = data["access_token"]
        self.valid_until = datetime.now() + timedelta(seconds=data["expires_in"]-300)
        return self.token

    def unshort(self, url):
        """Converts a spotify.link url into a normal spotify ID"""
        r = requests.get(url, allow_redirects=False)
        r.raise_for_status()
        return r.headers["Location"]

    def get_song_info(self, spotify_id):
        token = self.get_token()
        url = 'https://api.spotify.com/v1/tracks/' + spotify_id
        headers = {
            'Content-Type': 'application/json',
            'Authorization': 'Bearer ' + token
        }
        r = requests.get(url, headers=headers)
        r.raise_for_status()
        return r.json()