<gh_stars>0
# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import importlib
import logging

import pyramid.security

from ziggurat_foundations.models.base import get_db_session
from ziggurat_foundations.models.services.user import UserService

CONFIG_KEY = "ziggurat_foundations"
log = logging.getLogger(__name__)


class ZigguratSignInSuccess(object):
    def __contains__(self, other):
        return True

    def __init__(self, headers, came_from, user):
        self.headers = headers
        self.came_from = came_from
        self.user = user


class ZigguratSignInBadAuth(object):
    def __contains__(self, other):
        return False

    def __init__(self, headers, came_from):
        self.headers = headers
        self.came_from = came_from


class ZigguratSignOut(object):
    def __contains__(self, other):
        return True

    def __init__(self, headers):
        self.headers = headers


def includeme(config):
    settings = config.registry.settings
    sign_in_path = settings.get("%s.sign_in.sign_in_pattern" % CONFIG_KEY, "/sign_in")
    sign_out_path = settings.get(
        "%s.sign_in.sign_out_pattern" % CONFIG_KEY, "/sign_out"
    )
    session_provider_callable_config = settings.get(
        "%s.session_provider_callable" % CONFIG_KEY
    )
    signin_came_from_key = settings.get(
        "%s.sign_in.came_from_key" % CONFIG_KEY, "came_from"
    )
    signin_username_key = settings.get("%s.sign_in.username_key" % CONFIG_KEY, "login")
    signin_password_key = settings.get(
        "%s.sign_in.password_key" % CONFIG_KEY, "password"
    )

    if not session_provider_callable_config:

        def session_provider_callable(request):
            return get_db_session()

    else:
        parts = session_provider_callable_config.split(":")
        _tmp = importlib.import_module(parts[0])
        session_provider_callable = getattr(_tmp, parts[1])

    endpoint = ZigguratSignInProvider(
        settings=settings,
        session_getter=session_provider_callable,
        signin_came_from_key=signin_came_from_key,
        signin_username_key=signin_username_key,
        signin_password_key=signin_password_key,
    )

    config.add_route(
        "ziggurat.routes.sign_in",
        sign_in_path,
        use_global_views=True,
        factory=endpoint.sign_in,
    )
    config.add_route(
        "ziggurat.routes.sign_out",
        sign_out_path,
        use_global_views=True,
        factory=endpoint.sign_out,
    )


class ZigguratSignInProvider(object):
    def __init__(self, *args, **kwargs):
        for k, v in kwargs.items():
            setattr(self, k, v)

    def sign_in(self, request):
        came_from = request.params.get(self.signin_came_from_key, "/")
        db_session = self.session_getter(request)

        user = UserService.by_user_name(
            request.params.get(self.signin_username_key), db_session=db_session
        )
        if user is None:
            # if no result, test to see if email exists
            user = UserService.by_email(
                request.params.get(self.signin_username_key), db_session=db_session
            )
        if user:
            password = request.params.get(self.signin_password_key)
            if UserService.check_password(user, password):
                headers = pyramid.security.remember(request, user.id)
                return ZigguratSignInSuccess(
                    headers=headers, came_from=came_from, user=user
                )
        headers = pyramid.security.forget(request)
        return ZigguratSignInBadAuth(headers=headers, came_from=came_from)

    def sign_out(self, request):
        headers = pyramid.security.forget(request)
        return ZigguratSignOut(headers=headers)

    def session_getter(self, request):
        raise NotImplementedError()

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import importlib
import logging

import pyramid.security

from ziggurat_foundations.models.base import get_db_session
from ziggurat_foundations.models.services.user import UserService

CONFIG_KEY = "ziggurat_foundations"
log = logging.getLogger(__name__)


class ZigguratSignInSuccess(object):
    def __contains__(self, other):
        return True

    def __init__(self, headers, came_from, user):
        self.headers = headers
        self.came_from = came_from
        self.user = user


class ZigguratSignInBadAuth(object):
    def __contains__(self, other):
        return False

    def __init__(self, headers, came_from):
        self.headers = headers
        self.came_from = came_from


class ZigguratSignOut(object):
    def __contains__(self, other):
        return True

    def __init__(self, headers):
        self.headers = headers


def includeme(config):
    settings = config.registry.settings
    sign_in_path = settings.get("%s.sign_in.sign_in_pattern" % CONFIG_KEY, "/sign_in")
    sign_out_path = settings.get(
        "%s.sign_in.sign_out_pattern" % CONFIG_KEY, "/sign_out"
    )
    session_provider_callable_config = settings.get(
        "%s.session_provider_callable" % CONFIG_KEY
    )
    signin_came_from_key = settings.get(
        "%s.sign_in.came_from_key" % CONFIG_KEY, "came_from"
    )
    signin_username_key = settings.get("%s.sign_in.username_key" % CONFIG_KEY, "login")
    signin_password_key = settings.get(
        "%s.sign_in.password_key" % CONFIG_KEY, "password"
    )

    if not session_provider_callable_config:

        def session_provider_callable(request):
            return get_db_session()

    else:
        parts = session_provider_callable_config.split(":")
        _tmp = importlib.import_module(parts[0])
        session_provider_callable = getattr(_tmp, parts[1])

    endpoint = ZigguratSignInProvider(
        settings=settings,
        session_getter=session_provider_callable,
        signin_came_from_key=signin_came_from_key,
        signin_username_key=signin_username_key,
        signin_password_key=signin_password_key,
    )

    config.add_route(
        "ziggurat.routes.sign_in",
        sign_in_path,
        use_global_views=True,
        factory=endpoint.sign_in,
    )
    config.add_route(
        "ziggurat.routes.sign_out",
        sign_out_path,
        use_global_views=True,
        factory=endpoint.sign_out,
    )


class ZigguratSignInProvider(object):
    def __init__(self, *args, **kwargs):
        for k, v in kwargs.items():
            setattr(self, k, v)

    def sign_in(self, request):
        came_from = request.params.get(self.signin_came_from_key, "/")
        db_session = self.session_getter(request)

        user = UserService.by_user_name(
            request.params.get(self.signin_username_key), db_session=db_session
        )
        if user is None:
            # if no result, test to see if email exists
            user = UserService.by_email(
                request.params.get(self.signin_username_key), db_session=db_session
            )
        if user:
            password = request.params.get(self.signin_password_key)
            if UserService.check_password(user, password):
                headers = pyramid.security.remember(request, user.id)
                return ZigguratSignInSuccess(
                    headers=headers, came_from=came_from, user=user
                )
        headers = pyramid.security.forget(request)
        return ZigguratSignInBadAuth(headers=headers, came_from=came_from)

    def sign_out(self, request):
        headers = pyramid.security.forget(request)
        return ZigguratSignOut(headers=headers)

    def session_getter(self, request):
        raise NotImplementedError()
# -*- coding: utf-8 -*- # if no result, test to see if email exists