from __future__ import annotations

from django.contrib import messages
from django.contrib.auth import login
from django.http import HttpRequest, HttpResponse
from django.shortcuts import redirect, render
from django.utils.translation import gettext_lazy as _

from apps.organization.services.auth_service import AUTO_REGISTER_BOOTSTRAP_USERNAME, AuthService

from .forms import LawyerRegistrationForm

_auth_service = AuthService()


def register(request: HttpRequest) -> HttpResponse:
    is_first_user = _auth_service.is_first_user()
    show_auto_register = _auth_service.should_show_auto_register()

    if request.method == "POST":
        if request.POST.get("action") == "auto_register":
            try:
                result = _auth_service.auto_register_superadmin()
            except Exception as e:
                messages.error(request, str(e))
                return redirect("admin_register")

            user = result.user
            login(request, user)
            messages.success(
                request,
                _("已自动创建超级管理员账户“%(name)s”，并为您完成登录。")
                % {"name": user.real_name or AUTO_REGISTER_BOOTSTRAP_USERNAME},
            )
            return redirect("admin:index")

        form = LawyerRegistrationForm(request.POST)
        if form.is_valid():
            username: str = form.cleaned_data["username"]
            password: str = form.cleaned_data["password1"]
            try:
                result = _auth_service.register(
                    username=username,
                    password=password,
                    real_name=username,
                )
            except Exception as e:
                messages.error(request, str(e))
            else:
                user = result.user
                if user.is_admin:
                    login(request, user)
                    messages.success(
                        request,
                        _("注册成功！您是第一个用户，已自动获得管理员权限。欢迎 %(name)s")
                        % {"name": user.real_name or user.username},
                    )
                    return redirect("admin:index")
                else:
                    messages.info(request, _("注册成功！请等待管理员开通权限后再登录。"))
                    return redirect("admin:login")
    else:
        form = LawyerRegistrationForm()

    return render(
        request,
        "admin/register.html",
        {
            "form": form,
            "title": _("用户注册"),
            "is_first_user": is_first_user,
            "show_auto_register": show_auto_register,
        },
    )