import logging

from django.utils.translation import gettext as _
from django.shortcuts import render
from saml2 import BINDING_HTTP_POST, BINDING_HTTP_REDIRECT

from .utils import repr_saml


logger = logging.getLogger(__name__)

_not_valid_saml_msg = _('Not a valid SAML Session, Probably your request is '
                        'expired or you refreshed your page getting in a stale '
                        'request. Please come back to your SP and renew '
                        'the authentication request')


def store_params_in_session(request):
    """ Entrypoint view for SSO. Gathers the parameters from the
        HTTP request and stores them in the session

        It do not return anything because request come as pointer
    """
    if request.method == 'POST':
        passed_data = request.POST
        binding = BINDING_HTTP_POST
    else:
        passed_data = request.GET
        binding = BINDING_HTTP_REDIRECT
    saml_request = passed_data.get('SAMLRequest')
    if saml_request:
        msg = "SAML request [\n{}]"
        logger.debug(msg.format(repr_saml(saml_request, b64=True)))
    else:  # pragma: no cover
        msg = _('not a valid SAMLRequest: {}').format(
            _('AuthnRequest is missing. Please Retry'))
        logger.info('SAML Request absent from {}'.format(request))
        return render(request, 'error.html',
                      {'exception_type': msg,
                       'exception_msg': _('Please renew your SAML Request'),
                       'extra_message': _not_valid_saml_msg},
                      status=403)

    request.saml_session['SAMLRequest'] = saml_request
    request.saml_session['Binding'] = binding
    request.saml_session['RelayState'] = passed_data.get('RelayState', '')


def store_params_in_session_func(func_to_decorate):
    """ store_params_in_session as a funcion decorator
    """
    def new_func(*original_args, **original_kwargs):
        request = original_args[0]
        try:
            store_params_in_session(request)
            return func_to_decorate(*original_args, **original_kwargs)
        except Exception as e:  # pragma: no cover
            msg = _('not a valid SAMLRequest: {}').format(e)
            return render(request, 'error.html',
                          {'exception_type': msg,
                           'exception_msg': _('Please renew your SAML Request'),
                           'extra_message': _not_valid_saml_msg},
                          status=403)
    return new_func


def require_saml_request(func_to_decorate):
    """ store_params_in_session as a funcion decorator
    """
    def new_func(*original_args, **original_kwargs):
        request = original_args[0]
        if not request.saml_session.get('SAMLRequest'):
            return render(request, 'error.html',
                          {'exception_type': _("You cannot access to this service directly"),
                           'exception_msg': _('Please renew your SAML Request'),
                           'extra_message': _not_valid_saml_msg},
                          status=403)
        return func_to_decorate(*original_args, **original_kwargs)
    return new_func

import logging

from django.utils.translation import gettext as _
from django.shortcuts import render
from saml2 import BINDING_HTTP_POST, BINDING_HTTP_REDIRECT

from .utils import repr_saml


logger = logging.getLogger(__name__)

_not_valid_saml_msg = _('Not a valid SAML Session, Probably your request is '
                        'expired or you refreshed your page getting in a stale '
                        'request. Please come back to your SP and renew '
                        'the authentication request')


def store_params_in_session(request):
    """ Entrypoint view for SSO. Gathers the parameters from the
        HTTP request and stores them in the session

        It do not return anything because request come as pointer
    """
    if request.method == 'POST':
        passed_data = request.POST
        binding = BINDING_HTTP_POST
    else:
        passed_data = request.GET
        binding = BINDING_HTTP_REDIRECT
    saml_request = passed_data.get('SAMLRequest')
    if saml_request:
        msg = "SAML request [\n{}]"
        logger.debug(msg.format(repr_saml(saml_request, b64=True)))
    else:  # pragma: no cover
        msg = _('not a valid SAMLRequest: {}').format(
            _('AuthnRequest is missing. Please Retry'))
        logger.info('SAML Request absent from {}'.format(request))
        return render(request, 'error.html',
                      {'exception_type': msg,
                       'exception_msg': _('Please renew your SAML Request'),
                       'extra_message': _not_valid_saml_msg},
                      status=403)

    request.saml_session['SAMLRequest'] = saml_request
    request.saml_session['Binding'] = binding
    request.saml_session['RelayState'] = passed_data.get('RelayState', '')


def store_params_in_session_func(func_to_decorate):
    """ store_params_in_session as a funcion decorator
    """
    def new_func(*original_args, **original_kwargs):
        request = original_args[0]
        try:
            store_params_in_session(request)
            return func_to_decorate(*original_args, **original_kwargs)
        except Exception as e:  # pragma: no cover
            msg = _('not a valid SAMLRequest: {}').format(e)
            return render(request, 'error.html',
                          {'exception_type': msg,
                           'exception_msg': _('Please renew your SAML Request'),
                           'extra_message': _not_valid_saml_msg},
                          status=403)
    return new_func


def require_saml_request(func_to_decorate):
    """ store_params_in_session as a funcion decorator
    """
    def new_func(*original_args, **original_kwargs):
        request = original_args[0]
        if not request.saml_session.get('SAMLRequest'):
            return render(request, 'error.html',
                          {'exception_type': _("You cannot access to this service directly"),
                           'exception_msg': _('Please renew your SAML Request'),
                           'extra_message': _not_valid_saml_msg},
                          status=403)
        return func_to_decorate(*original_args, **original_kwargs)
    return new_func

Entrypoint view for SSO. Gathers the parameters from the         HTTP request and stores them in the session          It do not return anything because request come as pointer # pragma: no cover store_params_in_session as a funcion decorator # pragma: no cover store_params_in_session as a funcion decorator