<reponame>allink/django-admin-sso
from django.contrib.sites.models import RequestSite
from django.core.urlresolvers import reverse
from django.contrib.auth import authenticate, login
from django.http import HttpResponseRedirect
from django.shortcuts import render
from django.views.generic import View

from openid.consumer import consumer
from openid.extensions import ax, sreg

from admin_sso import settings
from .store import DjangoOpenIDStore

openid_store = DjangoOpenIDStore()


class OpenIDMixin(object):
    return_to_url = 'admin:admin_sso_assignment_return'

    def get_url(self, url=None):
        scheme = self.request.is_secure() and 'https' or 'http'
        primary_site = RequestSite(self.request)
        if url:
            path = reverse(url)
        else:
            path = "/"
        return '%s://%s%s' % (scheme, primary_site.domain, path)

    def get_consumer(self):
        return consumer.Consumer(self.request.session, self.get_openid_store())

    def get_openid_store(self):
        return openid_store


class StartOpenIDView(View, OpenIDMixin):
    def get(self, request, *args, **kwargs):
        c = self.get_consumer()
        auth_request = c.begin(settings.DJANGO_ADMIN_SSO_OPENID_ENDPOINT)

        trust_root = self.get_url()
        return_to = self.get_url(self.return_to_url)
        if auth_request.endpoint.supportsType(ax.AXMessage.ns_uri):
            # Add Attribute Exchange request information.
            ax_request = ax.FetchRequest()
            # XXX - uses myOpenID-compatible schema values, which are
            # not those listed at axschema.org.
            for ax_name, field_name in settings.AX_MAPPING:
                ax_request.add(ax.AttrInfo(ax_name, required=True))
            auth_request.addExtension(ax_request)
        else:
            sreg_request = sreg.SRegRequest(required=['email', 'nickname'])
            auth_request.addExtension(sreg_request)

        if auth_request.shouldSendRedirect():
            url = auth_request.redirectURL(trust_root, return_to)
            return HttpResponseRedirect(url)
        form_id = 'openid_message'
        form_html = auth_request.formMarkup(trust_root, return_to,
                                            False, {'id': form_id})
        return render(
            request, 'admin_sso/request_form.html', {'html': form_html})


class FinishOpenIDView(View, OpenIDMixin):
    def get(self, request, *args, **kwargs):
        if request.REQUEST:
            c = self.get_consumer()
            return_to = self.get_url(self.return_to_url)
            response = c.complete(request.REQUEST, return_to)
            if response.status == consumer.SUCCESS:
                user = authenticate(openid_response=response)
                if user and user.is_active:
                    user.active_openid_user.update_last_login()
                    login(request, user)
        return HttpResponseRedirect(reverse('admin:index'))

    def post(self, *args, **kwargs):
        return self.get(*args, **kwargs)

from django.contrib.sites.models import RequestSite
from django.core.urlresolvers import reverse
from django.contrib.auth import authenticate, login
from django.http import HttpResponseRedirect
from django.shortcuts import render
from django.views.generic import View

from openid.consumer import consumer
from openid.extensions import ax, sreg

from admin_sso import settings
from .store import DjangoOpenIDStore

openid_store = DjangoOpenIDStore()


class OpenIDMixin(object):
    return_to_url = 'admin:admin_sso_assignment_return'

    def get_url(self, url=None):
        scheme = self.request.is_secure() and 'https' or 'http'
        primary_site = RequestSite(self.request)
        if url:
            path = reverse(url)
        else:
            path = "/"
        return '%s://%s%s' % (scheme, primary_site.domain, path)

    def get_consumer(self):
        return consumer.Consumer(self.request.session, self.get_openid_store())

    def get_openid_store(self):
        return openid_store


class StartOpenIDView(View, OpenIDMixin):
    def get(self, request, *args, **kwargs):
        c = self.get_consumer()
        auth_request = c.begin(settings.DJANGO_ADMIN_SSO_OPENID_ENDPOINT)

        trust_root = self.get_url()
        return_to = self.get_url(self.return_to_url)
        if auth_request.endpoint.supportsType(ax.AXMessage.ns_uri):
            # Add Attribute Exchange request information.
            ax_request = ax.FetchRequest()
            # XXX - uses myOpenID-compatible schema values, which are
            # not those listed at axschema.org.
            for ax_name, field_name in settings.AX_MAPPING:
                ax_request.add(ax.AttrInfo(ax_name, required=True))
            auth_request.addExtension(ax_request)
        else:
            sreg_request = sreg.SRegRequest(required=['email', 'nickname'])
            auth_request.addExtension(sreg_request)

        if auth_request.shouldSendRedirect():
            url = auth_request.redirectURL(trust_root, return_to)
            return HttpResponseRedirect(url)
        form_id = 'openid_message'
        form_html = auth_request.formMarkup(trust_root, return_to,
                                            False, {'id': form_id})
        return render(
            request, 'admin_sso/request_form.html', {'html': form_html})


class FinishOpenIDView(View, OpenIDMixin):
    def get(self, request, *args, **kwargs):
        if request.REQUEST:
            c = self.get_consumer()
            return_to = self.get_url(self.return_to_url)
            response = c.complete(request.REQUEST, return_to)
            if response.status == consumer.SUCCESS:
                user = authenticate(openid_response=response)
                if user and user.is_active:
                    user.active_openid_user.update_last_login()
                    login(request, user)
        return HttpResponseRedirect(reverse('admin:index'))

    def post(self, *args, **kwargs):
        return self.get(*args, **kwargs)
# Add Attribute Exchange request information. # XXX - uses myOpenID-compatible schema values, which are # not those listed at axschema.org.