"""
    authlib.rfc6750.errors
    ~~~~~~~~~~~~~~~~~~~~~~

    OAuth Extensions Error Registration. When a request fails,
    the resource server responds using the appropriate HTTP
    status code and includes one of the following error codes
    in the response.

    https://tools.ietf.org/html/rfc6750#section-6.2

    :copyright: (c) 2017 by Hsiaoming Yang.
"""
from ..base import OAuth2Error
from ..rfc6749.errors import InvalidRequestError

__all__ = [
    'InvalidRequestError', 'InvalidTokenError', 'InsufficientScopeError'
]


class InvalidTokenError(OAuth2Error):
    """The access token provided is expired, revoked, malformed, or
    invalid for other reasons. The resource SHOULD respond with
    the HTTP 401 (Unauthorized) status code.  The client MAY
    request a new access token and retry the protected resource
    request.

    https://tools.ietf.org/html/rfc6750#section-3.1
    """
    error = 'invalid_token'
    status_code = 401

    def __init__(self, description=None, uri=None, status_code=None,
                 state=None, realm=None):
        super(InvalidTokenError, self).__init__(
            description, uri, status_code, state)
        self.realm = realm

    def get_error_description(self):
        return self.gettext(
            'The access token provided is expired, revoked, malformed, '
            'or invalid for other reasons.'
        )

    def get_headers(self):
        """If the protected resource request does not include authentication
        credentials or does not contain an access token that enables access
        to the protected resource, the resource server MUST include the HTTP
        "WWW-Authenticate" response header field; it MAY include it in
        response to other conditions as well.

        https://tools.ietf.org/html/rfc6750#section-3
        """
        headers = super(InvalidTokenError, self).get_headers()

        extras = []
        if self.realm:
            extras.append('realm="{}"'.format(self.realm))
        extras.append('error="{}"'.format(self.error))
        error_description = self.get_error_description()
        extras.append('error_description="{}"'.format(error_description))
        headers.append(
            ('WWW-Authenticate', 'Bearer ' + ', '.join(extras))
        )
        return headers


class InsufficientScopeError(OAuth2Error):
    """The request requires higher privileges than provided by the
    access token. The resource server SHOULD respond with the HTTP
    403 (Forbidden) status code and MAY include the "scope"
    attribute with the scope necessary to access the protected
    resource.

    https://tools.ietf.org/html/rfc6750#section-3.1
    """
    error = 'insufficient_scope'
    status_code = 403

    def __init__(self, token_scope, required_scope):
        super(InsufficientScopeError, self).__init__()
        self.token_scope = token_scope
        self.required_scope = required_scope

    def get_error_description(self):
        return self.gettext(
            'The request requires higher privileges than '
            'provided by the access token. '
            'Required: "%(required)s", Provided:"%(provided)s"'
        )  % dict(required=self.required_scope, provided=self.token_scope)
