from typing import Callable

from ..._sender import Request, Response
from ..._sender.error import get_error
from .token import Token


def handle_errors(request: Request, response: Response) -> None:
    """Examine response and raise errors accordingly."""
    if response.status_code < 400:
        return

    if response.status_code < 500:
        error_str = f"{response.status_code} {response.content['error']}"
        description = response.content.get("error_description", None)
        if description is not None:
            error_str += ": " + description
    else:
        error_str = "Unexpected error!"

    error_cls = get_error(response.status_code)
    raise error_cls(error_str, request=request, response=response)


def parse_token(uses_pkce: bool) -> Callable:
    """Wrap token parsing conditional to PKCE usage."""

    def func(request: Request, response: Response) -> Token:
        """Parse token object from response."""
        handle_errors(request, response)
        return Token(response.content, uses_pkce)

    return func


def parse_refreshed_token(uses_pkce: bool) -> Callable:
    """Wrap token parsing conditional to PKCE usage."""

    def func(request: Request, response: Response, refresh_token: str) -> Token:
        """Replace new refresh token with old value if empty."""
        handle_errors(request, response)
        refreshed = Token(response.content, uses_pkce)

        if refreshed.refresh_token is None:
            refreshed._refresh_token = refresh_token

        return refreshed

    return func
