main.py

"""Firebase Cloud Functions for blocking auth functions example."""
from firebase_functions import identity_fn


@identity_fn.before_user_created(
    id_token=True,
    access_token=True,
    refresh_token=True,
)
def beforeusercreated(
    event: identity_fn.AuthBlockingEvent
) -> identity_fn.BeforeCreateResponse | None:
    print(event)
    if not event.data or not event.data.email:
        return None
    if "@cats.com" in event.data.email:
        return identity_fn.BeforeCreateResponse(display_name="Meow!",)
    if "@dogs.com" in event.data.email:
        return identity_fn.BeforeCreateResponse(display_name="Woof!",)
    return None


@identity_fn.before_user_signed_in(
    id_token=True,
    access_token=True,
    refresh_token=True,
)
def beforeusersignedin(
    event: identity_fn.AuthBlockingEvent
) -> identity_fn.BeforeSignInResponse | None:
    print(event)
    if not event.data or not event.data.email:
        return None

    if "@cats.com" in event.data.email:
        return identity_fn.BeforeSignInResponse(session_claims={"emoji": "🐈"})

    if "@dogs.com" in event.data.email:
        return identity_fn.BeforeSignInResponse(session_claims={"emoji": "🐕"})

    return None


@identity_fn.before_email_sent()
# pylint: disable=useless-return
def beforeemailsent(
    event: identity_fn.AuthBlockingEvent
) -> identity_fn.BeforeEmailSentResponse | None:
    print(event)
    return None


@identity_fn.before_sms_sent()
# pylint: disable=useless-return
def beforesmssent(
    event: identity_fn.AuthBlockingEvent
) -> identity_fn.BeforeSmsSentResponse | None:
    print(event)
    return None