Synnefo

# Copyright 2011-2012 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from django.conf import settings as global_settings

from django.utils.translation import ugettext as _

from django.contrib import messages

from django.views.decorators.http import require_http_methods

from django.http import HttpResponseRedirect

from astakos.im.util import login_url

from astakos.im.models import AstakosUser

from astakos.im import settings

from astakos.im.views.target import get_pending_key, \

    handle_third_party_signup, handle_third_party_login, \

    init_third_party_session

from astakos.im.views.decorators import cookie_fix, requires_auth_provider

import astakos.im.messages as astakos_messages

import logging

logger = logging.getLogger(__name__)

class Tokens:

    # these are mapped by the Shibboleth SP software

    SHIB_EPPN = "HTTP_EPPN"  # eduPersonPrincipalName

    SHIB_NAME = "HTTP_SHIB_INETORGPERSON_GIVENNAME"

    SHIB_SURNAME = "HTTP_SHIB_PERSON_SURNAME"

    SHIB_CN = "HTTP_SHIB_PERSON_COMMONNAME"

    SHIB_DISPLAYNAME = "HTTP_SHIB_INETORGPERSON_DISPLAYNAME"

    SHIB_EP_AFFILIATION = "HTTP_SHIB_EP_AFFILIATION"

    SHIB_SESSION_ID = "HTTP_SHIB_SESSION_ID"

    SHIB_MAIL = "HTTP_SHIB_MAIL"

    SHIB_REMOTE_USER = "HTTP_REMOTE_USER"

@requires_auth_provider('shibboleth')

@require_http_methods(["GET", "POST"])

@cookie_fix

def login(

    request,

    template='im/third_party_check_local.html',

    extra_context=None):

    init_third_party_session(request)

    extra_context = extra_context or {}

    tokens = request.META

    third_party_key = get_pending_key(request)

    shibboleth_headers = {}

    for token in dir(Tokens):

        if token == token.upper():

            shibboleth_headers[token] = request.META.get(getattr(Tokens,

                                                                 token),

                                                         'NOT_SET')

    # log shibboleth headers

    # TODO: info -> debug

    logger.info("shibboleth request: %r" % shibboleth_headers)

    try:

        eppn = tokens.get(Tokens.SHIB_EPPN)

        fullname, first_name, last_name, email = None, None, None, None

        if global_settings.DEBUG and not eppn:

            eppn = getattr(global_settings, 'SHIBBOLETH_TEST_EPPN', None)

            fullname = getattr(global_settings, 'SHIBBOLETH_TEST_FULLNAME',

                               None)

        if not eppn:

            raise KeyError(_(astakos_messages.SHIBBOLETH_MISSING_EPPN) % {

                'domain': settings.BASE_HOST,

                'contact_email': settings.CONTACT_EMAIL

            })

        if Tokens.SHIB_DISPLAYNAME in tokens:

            fullname = tokens[Tokens.SHIB_DISPLAYNAME]

        elif Tokens.SHIB_CN in tokens:

            fullname = tokens[Tokens.SHIB_CN]

        if Tokens.SHIB_NAME in tokens:

            first_name = tokens[Tokens.SHIB_NAME]

        if Tokens.SHIB_SURNAME in tokens:

            last_name = tokens[Tokens.SHIB_SURNAME]

        if fullname:

            splitted = fullname.split(' ', 1)

            if len(splitted) == 2:

                first_name, last_name = splitted

        fullname = '%s %s' % (first_name, last_name)

        if not any([first_name, last_name]) and \

                    settings.SHIBBOLETH_REQUIRE_NAME_INFO:

            raise KeyError(_(astakos_messages.SHIBBOLETH_MISSING_NAME))

    except KeyError, e:

        # invalid shibboleth headers, redirect to login, display message

        messages.error(request, e.message)

        return HttpResponseRedirect(login_url(request))

    affiliation = tokens.get(Tokens.SHIB_EP_AFFILIATION, 'Shibboleth')

    email = tokens.get(Tokens.SHIB_MAIL, '')

    eppn_info = tokens.get(Tokens.SHIB_EPPN)

    provider_info = {'eppn': eppn_info, 'email': email, 'name': fullname,

                     'headers': shibboleth_headers}

    userid = eppn

    try:

        return handle_third_party_login(request, 'shibboleth',

                                        eppn, provider_info,

                                        affiliation, third_party_key)

    except AstakosUser.DoesNotExist, e:

        third_party_key = get_pending_key(request)

        user_info = {'affiliation': affiliation,

                     'first_name': first_name,

                     'last_name': last_name,

                     'email': email}

        return handle_third_party_signup(request, userid, 'shibboleth',

                                         third_party_key,

                                         provider_info,

                                         user_info,

                                         template,

                                         extra_context)