Ozwillo » Modules CKAN Ozwillo

import logging

import ckan.plugins.toolkit as toolkit

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,

                                      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,

    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):

            value = request.cookies.get(cookie)

            response.set_cookie(cookie, value, secure=True, httponly=True)

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

                toolkit.request.headers.update(ht_args)

    def logout(self):

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

            response.delete_cookie(cookie)

                                      controller='organization',

                                      action='read',

                                      id=g.name,

                                      qualified=True)

    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)

    def callback(self):

        client = Clients.get(g)

        userinfo = client.callback(request.GET)

            if '-' in locale:

                locale, country = locale.split('-')

                                      controller="organization",

                                      action='read',

                                      id=g.name,

                                      locale=locale,

                                      qualified=True))

        if 'sub' in userinfo:

                user_dict = {'id': userinfo['sub'],

                             'name': userinfo['sub'].replace('-', ''),

                             'email': userinfo['email'],

                             'password': userinfo['sub']

                             }

                context = {'ignore_auth': True, 'model': model,

                           'session': model.Session}

                user_create(context, user_dict)

                userobj = model.User.get(userinfo['sub'])

                if client.app_admin or client.app_user:

                    member_dict = {

                        'id': g.id,

                        'object': userinfo['sub'],

                        'object_type': 'user',

                        'capacity': 'admin',

                    }

                    member_create_context = {

                        'model': model,

                        'user': userobj.name,

                        'ignore_auth': True,

                        'session': session

                    }

                    member_create(member_create_context, member_dict)

                userobj.fullname = userinfo['given_name']

            if 'family_name' in userinfo:

                userobj.fullname += ' ' + userinfo['family_name']

            userobj.save()

                userobj.name = userinfo['nickname']

            try:

                userobj.save()

            except Exception, e:

                log.warning('Error while saving user name: %s' % e)

            session.save()

        """

        Revokes the delivered access token. Logs out the user

        """

        if not request.referer or request.host not in request.referer:

            redirect_to('/')

                          controller='organization',

                          action='read',

                          id=g.name,

                          qualified=True)

            logout_url = client.end_session_endpoint

            redirect_uri = org_url + '/logout'

                self.sso(g.name)

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

            data = 'token=' + client.access_token

            data += '&token_type_hint=access_token'

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

                                data=data, headers=headers)

            # redirect to IDP logout

            logout_url += '?id_token_hint=%s&' % client.id_token

            logout_url += 'post_logout_redirect_uri=%s' % redirect_uri