import requests
import os

CLIENT_ID = os.environ.get('OSU_CLIENT_ID', '5')
CLIENT_SECRET = os.environ.get('OSU_CLIENT_SECRET', 'FGc9GAtyHzeQDshWP5Ah7dega8hJACAJpQtw6OXk')
API_URL = os.environ.get('OSU_API_URL', 'https://osu.ppy.sh')


def authenticate(username: str, password: str):
    """Authenticate via OAuth password flow and return the token dict."""
    url = f"{API_URL}/oauth/token"
    data = {
        "grant_type": "password",
        "username": username,
        "password": password,
        "client_id": CLIENT_ID,
        "client_secret": CLIENT_SECRET,
        "scope": "*",
    }
    response = requests.post(url, data=data)
    response.raise_for_status()
    return response.json()


def refresh_token(refresh: str):
    """Refresh the OAuth token."""
    url = f"{API_URL}/oauth/token"
    data = {
        "grant_type": "refresh_token",
        "refresh_token": refresh,
        "client_id": CLIENT_ID,
        "client_secret": CLIENT_SECRET,
        "scope": "*",
    }
    response = requests.post(url, data=data)
    response.raise_for_status()
    return response.json()


def get_current_user(access_token: str, ruleset: str = "osu"):
    """Retrieve the authenticated user's data."""
    url = f"{API_URL}/api/v2/me/{ruleset}"
    headers = {"Authorization": f"Bearer {access_token}"}
    response = requests.get(url, headers=headers)
    response.raise_for_status()
    return response.json()


if __name__ == "__main__":
    import getpass

    username = input("osu! username: ")
    password = getpass.getpass()

    token = authenticate(username, password)
    print("Access Token:", token["access_token"])
    user = get_current_user(token["access_token"])

    print(user)