members.lumbung.space/members_lumbung_space/routes/register.py

"""Registration routes."""

import json
from datetime import datetime as dt
from datetime import timedelta

from fastapi import APIRouter, Depends, Form, Request
from pydantic import EmailStr, errors

from members_lumbung_space.dependencies import fresh_token, get_invites

router = APIRouter()


@router.get("/register/{invite}")
async def register_invite(
    request: Request, invite: str, invites=Depends(get_invites)
):
    from members_lumbung_space.config import INVITE_TIME_LIMIT

    matching, invited_by, matching_invite = False, None, None
    for username in invites:
        for _invite in invites[username]:
            if invite == _invite["link"]:
                matching = True
                invited_by = username
                matching_invite = _invite
                break
        if matching:
            break

    if not matching:
        message = "This invite does not exist, sorry."
        context = {"request": request, "message": message}
        return request.app.state.templates.TemplateResponse(
            "invalid.html", context=context
        )

    expired = (
        dt.fromisoformat(matching_invite["time"])
        + timedelta(days=INVITE_TIME_LIMIT)
    ).day > dt.now().day

    if expired:
        message = "This invite has expired, sorry."
        context = {"request": request, "message": message}
        return request.app.state.templates.TemplateResponse(
            "invalid.html", context=context
        )

    context = {"request": request, "invited_by": invited_by}
    return request.app.state.templates.TemplateResponse(
        "register.html", context=context
    )


@router.post("/form/keycloak/register", dependencies=[Depends(fresh_token)])
def form_keycloak_register(
    request: Request,
    first_name: str = Form(...),
    last_name: str = Form(...),
    username: str = Form(...),
    email: str = Form(...),
    password: str = Form(...),
    password_again: str = Form(...),
    invited_by: str = Form(...),
):
    context = {
        "request": request,
        "invited_by": invited_by,
        "first_name": first_name,
        "last_name": last_name,
        "username": username,
        "email": email,
    }

    try:
        EmailStr().validate(email)
    except errors.EmailError:
        context["exception"] = "email is not valid?"
        return request.app.state.templates.TemplateResponse(
            "register.html", context=context
        )

    if password != password_again:
        context["exception"] = "passwords don't match?"
        return request.app.state.templates.TemplateResponse(
            "register.html", context=context
        )

    payload = {
        "email": email,
        "username": username,
        "enabled": True,
        "firstName": first_name,
        "lastName": last_name,
        "credentials": [
            {
                "value": password,
                "type": "password",
            }
        ],
        "realmRoles": [
            "user_default",
        ],
        "attributes": {"invited_by": invited_by},
    }

    try:
        request.app.state.keycloak.create_user(payload, exist_ok=False)
    except Exception as exception:
        request.app.state.log.error(
            f"Keycloak user registration failed, saw: {exception}"
        )
        message = json.loads(exception.error_message).get(
            "errorMessage", "Unknown reason"
        )
        context = {"request": request, "exception": message}
        return request.app.state.templates.TemplateResponse(
            "submit.html", context=context
        )

    context = {"request": request, "first_name": first_name}
    return request.app.state.templates.TemplateResponse(
        "submit.html", context=context
    )