"""passbook core authentication views""" from typing import Dict, Optional from django.contrib import messages from django.contrib.auth import login, logout from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin from django.forms.utils import ErrorList from django.http import HttpRequest, HttpResponse from django.shortcuts import get_object_or_404, redirect, reverse from django.utils.translation import ugettext as _ from django.views import View from django.views.generic import FormView from structlog import get_logger from passbook.core.forms.authentication import LoginForm, SignUpForm from passbook.core.models import Invitation, Nonce, Source, User from passbook.core.signals import invitation_used, user_signed_up from passbook.factors.password.exceptions import PasswordPolicyInvalid from passbook.factors.view import AuthenticationView, _redirect_with_qs from passbook.lib.config import CONFIG LOGGER = get_logger() class LoginView(UserPassesTestMixin, FormView): """Allow users to sign in""" template_name = "login/form.html" form_class = LoginForm success_url = "." # Allow only not authenticated users to login def test_func(self): return self.request.user.is_authenticated is False def handle_no_permission(self): if "next" in self.request.GET: return redirect(self.request.GET.get("next")) return redirect(reverse("passbook_core:overview")) def get_context_data(self, **kwargs): kwargs["config"] = CONFIG.y("passbook") kwargs["is_login"] = True kwargs["title"] = _("Log in to your account") kwargs["primary_action"] = _("Log in") kwargs["show_sign_up_notice"] = CONFIG.y("passbook.sign_up.enabled") kwargs["sources"] = [] sources = Source.objects.filter(enabled=True).select_subclasses() for source in sources: login_button = source.login_button if login_button: kwargs["sources"].append(login_button) if kwargs["sources"]: self.template_name = "login/with_sources.html" return super().get_context_data(**kwargs) def get_user(self, uid_value) -> Optional[User]: """Find user instance. Returns None if no user was found.""" for search_field in CONFIG.y("passbook.uid_fields"): # Workaround for E-Mail -> email if search_field == "e-mail": search_field = "email" users = User.objects.filter(**{search_field: uid_value}) if users.exists(): LOGGER.debug("Found user", user=users.first(), uid_field=search_field) return users.first() return None def form_valid(self, form: LoginForm) -> HttpResponse: """Form data is valid""" pre_user = self.get_user(form.cleaned_data.get("uid_field")) if not pre_user: # No user found return self.invalid_login(self.request) # self.request.session.flush() self.request.session[AuthenticationView.SESSION_PENDING_USER] = pre_user.pk return _redirect_with_qs("passbook_core:auth-process", self.request.GET) def invalid_login( self, request: HttpRequest, disabled_user: User = None ) -> HttpResponse: """Handle login for disabled users/invalid login attempts""" LOGGER.debug("invalid_login", user=disabled_user) messages.error(request, _("Failed to authenticate.")) return self.render_to_response(self.get_context_data()) class LogoutView(LoginRequiredMixin, View): """Log current user out""" def dispatch(self, request): """Log current user out""" logout(request) messages.success(request, _("You've successfully been logged out.")) return redirect(reverse("passbook_core:auth-login")) class SignUpView(UserPassesTestMixin, FormView): """Sign up new user, optionally consume one-use invitation link.""" template_name = "login/form.html" form_class = SignUpForm success_url = "." # Invitation instance, if invitation link was used _invitation = None # Instance of newly created user _user = None # Allow only not authenticated users to login def test_func(self): return self.request.user.is_authenticated is False def handle_no_permission(self): return redirect(reverse("passbook_core:overview")) def dispatch(self, request, *args, **kwargs): """Check if sign-up is enabled or invitation link given""" allowed = False if "invitation" in request.GET: invitations = Invitation.objects.filter(uuid=request.GET.get("invitation")) allowed = invitations.exists() if allowed: self._invitation = invitations.first() if CONFIG.y("passbook.sign_up.enabled"): allowed = True if not allowed: messages.error(request, _("Sign-ups are currently disabled.")) return redirect(reverse("passbook_core:auth-login")) return super().dispatch(request, *args, **kwargs) def get_initial(self): if self._invitation: initial = {} if self._invitation.fixed_username: initial["username"] = self._invitation.fixed_username if self._invitation.fixed_email: initial["email"] = self._invitation.fixed_email return initial return super().get_initial() def get_context_data(self, **kwargs): kwargs["config"] = CONFIG.y("passbook") kwargs["is_login"] = True kwargs["title"] = _("Sign Up") kwargs["primary_action"] = _("Sign up") return super().get_context_data(**kwargs) def form_valid(self, form: SignUpForm) -> HttpResponse: """Create user""" try: self._user = SignUpView.create_user(form.cleaned_data, self.request) except PasswordPolicyInvalid as exc: # Manually inject error into form # pylint: disable=protected-access errors = form._errors.setdefault("password", ErrorList()) for error in exc.messages: errors.append(error) return self.form_invalid(form) self.consume_invitation() messages.success(self.request, _("Successfully signed up!")) LOGGER.debug("Successfully signed up", email=form.cleaned_data.get("email")) return redirect(reverse("passbook_core:auth-login")) def consume_invitation(self): """Consume invitation if an invitation was used""" if self._invitation: invitation_used.send( sender=self, request=self.request, invitation=self._invitation, user=self._user, ) self._invitation.delete() @staticmethod def create_user(data: Dict, request: HttpRequest = None) -> User: """Create user from data Args: data: Dictionary as returned by SignUpForm's cleaned_data request: Optional current request. Returns: The user created Raises: PasswordPolicyInvalid: if any policy are not fulfilled. This also deletes the created user. """ # Create user new_user = User.objects.create( username=data.get("username"), email=data.get("email"), name=data.get("name"), ) new_user.is_active = True try: new_user.set_password(data.get("password")) new_user.save() request.user = new_user # Send signal for other auth sources user_signed_up.send(sender=SignUpView, user=new_user, request=request) return new_user except PasswordPolicyInvalid as exc: new_user.delete() raise exc class SignUpConfirmView(View): """Confirm registration from Nonce""" def get(self, request, nonce): """Verify UUID and activate user""" nonce = get_object_or_404(Nonce, uuid=nonce) nonce.user.is_active = True nonce.user.save() # Workaround: hardcoded reference to ModelBackend, needs testing nonce.user.backend = "django.contrib.auth.backends.ModelBackend" login(request, nonce.user) nonce.delete() messages.success(request, _("Successfully confirmed registration.")) return redirect("passbook_core:overview") class PasswordResetView(View): """Temporarily authenticate User and allow them to reset their password""" def get(self, request, nonce): """Authenticate user with nonce and redirect to password change view""" # 3. (Optional) Trap user in password change view nonce = get_object_or_404(Nonce, uuid=nonce) # Workaround: hardcoded reference to ModelBackend, needs testing nonce.user.backend = "django.contrib.auth.backends.ModelBackend" login(request, nonce.user) nonce.delete() messages.success( request, _(("Temporarily authenticated with Nonce, " "please change your password")), ) return redirect("passbook_core:user-change-password")