upstream-zulip/corporate/views/sponsorship.py

from typing import TYPE_CHECKING

from django.http import HttpRequest, HttpResponse, HttpResponseRedirect
from django.shortcuts import render
from django.urls import reverse

from corporate.lib.decorator import (
    authenticated_remote_realm_management_endpoint,
    authenticated_remote_server_management_endpoint,
)
from zerver.decorator import require_billing_access, zulip_login_required
from zerver.lib.response import json_success
from zerver.models import UserProfile

if TYPE_CHECKING:
    from corporate.lib.stripe import RemoteRealmBillingSession, RemoteServerBillingSession


@zulip_login_required
def sponsorship_page(request: HttpRequest) -> HttpResponse:
    from corporate.lib.stripe import RealmBillingSession

    user = request.user
    assert user.is_authenticated

    billing_session = RealmBillingSession(user)
    if billing_session.realm.demo_organization_scheduled_deletion_date is not None:
        return render(
            request,
            "corporate/billing/demo_organization_billing_disabled.html",
            context={
                "sponsorship_request": True,
            },
        )
    context = billing_session.get_sponsorship_request_context()
    if context is None or not user.has_billing_access:
        return HttpResponseRedirect(reverse("billing_page"))

    return render(request, "corporate/billing/sponsorship.html", context=context)


@authenticated_remote_realm_management_endpoint
def remote_realm_sponsorship_page(
    request: HttpRequest,
    billing_session: "RemoteRealmBillingSession",
) -> HttpResponse:  # nocoverage
    context = billing_session.get_sponsorship_request_context()
    if context is None:
        return HttpResponseRedirect(
            reverse("remote_realm_billing_page", args=(billing_session.remote_realm.uuid,))
        )

    return render(request, "corporate/billing/sponsorship.html", context=context)


@authenticated_remote_server_management_endpoint
def remote_server_sponsorship_page(
    request: HttpRequest,
    billing_session: "RemoteServerBillingSession",
) -> HttpResponse:  # nocoverage
    context = billing_session.get_sponsorship_request_context()
    if context is None:
        return HttpResponseRedirect(
            reverse("remote_server_billing_page", args=(billing_session.remote_server.uuid,))
        )

    return render(request, "corporate/billing/sponsorship.html", context=context)


@require_billing_access
def sponsorship(
    request: HttpRequest,
    user: UserProfile,
) -> HttpResponse:
    from corporate.lib.stripe import RealmBillingSession, SponsorshipRequestForm

    billing_session = RealmBillingSession(user)
    post_data = request.POST.copy()
    form = SponsorshipRequestForm(post_data)
    billing_session.request_sponsorship(form)
    return json_success(request)


@authenticated_remote_realm_management_endpoint
def remote_realm_sponsorship(
    request: HttpRequest,
    billing_session: "RemoteRealmBillingSession",
) -> HttpResponse:  # nocoverage
    from corporate.lib.stripe import SponsorshipRequestForm

    post_data = request.POST.copy()
    form = SponsorshipRequestForm(post_data)
    billing_session.request_sponsorship(form)
    return json_success(request)


@authenticated_remote_server_management_endpoint
def remote_server_sponsorship(
    request: HttpRequest,
    billing_session: "RemoteServerBillingSession",
) -> HttpResponse:  # nocoverage
    from corporate.lib.stripe import SponsorshipRequestForm

    post_data = request.POST.copy()
    form = SponsorshipRequestForm(post_data)
    billing_session.request_sponsorship(form)
    return json_success(request)