Ozwillo » Modules CKAN Ozwillo

import logging

import ckan.plugins.toolkit as toolkit

from ckan import model

from pylons import config, request

from oidc import create_client

plugin_config_prefix = 'ckanext.ozwillo_pyoidc.'

log = logging.getLogger(__name__)

class Clients(object):

    @classmethod

    def get(cls, g):

        global _CLIENTS

        if g.id in _CLIENTS:

            return _CLIENTS.get(g.id)

        client = cls().get_client(g)

        _CLIENTS.update({g.id: client})

        return client

    def get_client(self, g):

        params = conf.CLIENT.copy()

        params['client_registration'].update({

            'client_id': g._extras['client_id'].value,

            'client_secret': g._extras['client_secret'].value,

            'redirect_uris': [toolkit.url_for(host=request.host,

                                              controller=plugin_controller,

                                              action='callback',

                                              id=g.name,

                                              qualified=True)]

        })

        return create_client(**params)

class OzwilloPyoidcPlugin(plugins.SingletonPlugin):

    plugins.implements(plugins.IConfigurer)

    plugins.implements(plugins.IAuthenticator, inherit=True)

                    controller=plugin_controller,

                    action='sso')

        map.connect('/organization/{id:.*}/callback',

                    controller=plugin_controller,

                    action='callback')

                    controller=plugin_controller,

                    action='slo')

        map.redirect('/organization/{id:.*}/logout', '/user/_logout')

    def after_map(self, map):

        return map

    def identify(self):

        if user and not toolkit.c.userobj:

            userobj = model.User.get(user)

            toolkit.c.user = userobj.name

            toolkit.c.userobj = userobj

    def login(self):

            g = model.Group.get(session['organization_id'])

            url, ht_args = client.create_authn_request(session, conf.ACR_VALUES)

                toolkit.request.headers.update(ht_args)

            toolkit.redirect_to(url)

        else:

            toolkit.redirect_to('/')

    def logout(self):

        session.save()

    def update_config(self, config_):

        toolkit.add_template_directory(config_, 'templates')

        toolkit.add_public_directory(config_, 'public')

        toolkit.add_resource('fanstatic', 'ozwillo_pyoidc')

class OpenidController(base.BaseController):

        log.info('SSO for organization "%s"' % id)

        session['organization_id'] = id

        session.save()

        log.info('redirecting to login page')

        login_url = toolkit.url_for(host=request.host,

                                    controller='user',

                                    action='login',

                                    qualified=True)

        toolkit.redirect_to(login_url)

    def callback(self):

        client = Clients.get(g)

        userinfo = client.callback(request.GET)

        log.info('Received userinfo: %s' % userinfo)

            if 'given_name' in userinfo:

                userobj.fullname = userinfo['given_name']

            if 'family_name' in userinfo:

                userobj.fullname += userinfo['family_name']

            userobj.save()

            session['user'] = userobj.id

            session.save()

        org_url = toolkit.url_for(host=request.host,

                                  controller="organization",

                                  action='read',

                                  id=g.id,

                                  qualified=True)

        toolkit.redirect_to(org_url)

    def slo(self):

        """

        Revokes the delivered access token. Logs out the user

        """

        client = Clients.get(g)

        logout_url = client.end_session_endpoint

                                  controller='organization',

                                  action='read',

                                  id=session['organization_id'],

                                  qualified=True)

        redirect_uri = org_url + '/logout'

        # revoke the access token

        headers = {'Content-Type': 'application/x-www-form-urlencoded'}

        client.http_request(client.revocation_endpoint, 'POST',

        # redirect to IDP logout