import requests

base_url = "http://tautulli:8181"
end_point = "/auth/signin"
headers = {
    "Content-Type": "application/x-www-form-urlencoded",
}

def authenticate_tautulli(data):
    try:
        response = requests.post(base_url + end_point, headers=headers, data=data, timeout=5)
        if response.status_code == 200:
            cookie_obj = None
            for key, value in response.cookies.items():
                if key.startswith("tautulli_token"):
                    cookie_obj = f"{key}={value}"
                    break
            if not cookie_obj:
                ak_logger.error("Tautulli: Cannot get tautulli_token")
                return {}
            return {
                "ak_proxy": {
                    "user_attributes": {
                        "additionalHeaders": {
                            "Cookie": cookie_obj,
                        }
                    }
                }
            }
        else:
            ak_logger.error(f"Tautulli: The request failed with: {response.text}")
            return {}
    except requests.Timeout:
        ak_logger.error("Tautulli: Request timed out")
        return {}
    except requests.RequestException as e:
        ak_logger.error(f"Tautulli: Request exception: {e}")
        return {}
    except Exception as e:
        ak_logger.error(f"Tautulli: Unexpected error: {e}")
        return {}

# Try to authenticate with Plex token first
connection = UserPlexSourceConnection.objects.filter(user=request.user).first()
if connection and connection.plex_token:
    ak_logger.info("Tautulli: Trying to authenticate with Plex token")
    data = f'token={connection.plex_token}&remember_me=1'
    result = authenticate_tautulli(data)
    if result:
        ak_logger.info("Tautulli: Successfully authenticated with Plex token")
        return result

# Fallback to username/password auth
if "tautulli_user" in request.user.attributes and "tautulli_pass" in request.user.attributes:
    ak_logger.info("Tautulli: Trying to authenticate with username and password")
    username = request.user.attributes["tautulli_user"]
    password = request.user.attributes["tautulli_pass"]
    data = f'username={username}&password={password}&remember_me=1'
    result = authenticate_tautulli(data)
    if result:
        ak_logger.info("Tautulli: Successfully authenticated with username and password")
        return result

ak_logger.info("Tautulli: Authentication failed")
return {}