"""Oauth Client."""

import logging

import requests
from requests_oauthlib import OAuth2Session

from product_review.lib.vapi.oauth_code_error import OauthCodeError


class OauthClient:
    """Oauth Client."""

    def __init__(
        self, client_id, client_secret, base_url, redirect_uri, user_id, user_type
    ):
        """__init__."""
        self.client_id = client_id
        self.client_secret = client_secret
        self.base_url = base_url
        self.redirect_uri = redirect_uri
        self.user_id = user_id
        self.user_type = user_type
        self.oauth_session = OAuth2Session(self.client_id)

    def auth_url(self):
        """Generate the URL to be called when fetching an auth code."""
        base_url = self.base_url + "/authorize/getoauthcode"
        auth_url, state = self.oauth_session.authorization_url(
            base_url, None, user_id=self.user_id, user_type=self.user_type
        )
        # passing this in kwargs to authorization_url doesn't seem to work...
        auth_url += "&isRedirect=0"
        return auth_url

    def fetch_code(self):
        """Fetch an authorization code for the API."""
        auth_response = requests.get(self.auth_url())
        auth_response_body = auth_response.json()
        if auth_response.status_code != requests.codes.ok:
            raise OauthCodeError(
                "Error fetching OAuth code",
                auth_response_body,
                auth_response.status_code,
            )
        return auth_response_body["query"]["code"]

    def fetch_token(self):
        """Fetch an access token for the API.

        :return: A dictionary object containing token data. Keys:
            access_token
            refresh_token
        """
        token_url = f"{self.base_url}/authorize/getaccesstoken"
        oauth_code = self.fetch_code()
        return self.oauth_session.fetch_token(
            token_url,
            oauth_code,
            client_secret=self.client_secret,
            user_type=self.user_type,
        )

    @staticmethod
    def logger():
        """Return the logger instance used internally by requests_oauthlib.

        Useful for debugging purposes. Example usage:

        import logging
        import sys

        logger = oauth_client.logger()
        logger.setLevel(logging.DEBUG)
        handler = logging.StreamHandler(stream=sys.stdout)
        handler.setLevel(logging.DEBUG)
        logger.addHandler(handler)
        """
        return logging.getLogger("requests_oauthlib")