zulip/corporate/views/event_status.py

import logging
from typing import Optional

from django.http import HttpRequest, HttpResponse
from django.shortcuts import render

from corporate.lib.decorator import (
    authenticated_remote_realm_management_endpoint,
    authenticated_remote_server_management_endpoint,
    self_hosting_management_endpoint,
)
from corporate.lib.stripe import (
    EventStatusRequest,
    RealmBillingSession,
    RemoteRealmBillingSession,
    RemoteServerBillingSession,
)
from zerver.decorator import require_organization_member, zulip_login_required
from zerver.lib.response import json_success
from zerver.lib.typed_endpoint import typed_endpoint
from zerver.models import UserProfile

billing_logger = logging.getLogger("corporate.stripe")


@require_organization_member
@typed_endpoint
def event_status(
    request: HttpRequest,
    user: UserProfile,
    *,
    stripe_session_id: Optional[str] = None,
    stripe_invoice_id: Optional[str] = None,
) -> HttpResponse:
    event_status_request = EventStatusRequest(
        stripe_session_id=stripe_session_id, stripe_invoice_id=stripe_invoice_id
    )
    billing_session = RealmBillingSession(user)
    data = billing_session.get_event_status(event_status_request)
    return json_success(request, data)


@authenticated_remote_realm_management_endpoint
@typed_endpoint
def remote_realm_event_status(
    request: HttpRequest,
    billing_session: RemoteRealmBillingSession,
    *,
    stripe_session_id: Optional[str] = None,
    stripe_invoice_id: Optional[str] = None,
) -> HttpResponse:
    event_status_request = EventStatusRequest(
        stripe_session_id=stripe_session_id, stripe_invoice_id=stripe_invoice_id
    )
    data = billing_session.get_event_status(event_status_request)
    return json_success(request, data)


@authenticated_remote_server_management_endpoint
@typed_endpoint
def remote_server_event_status(
    request: HttpRequest,
    billing_session: RemoteServerBillingSession,
    *,
    stripe_session_id: Optional[str] = None,
    stripe_invoice_id: Optional[str] = None,
) -> HttpResponse:  # nocoverage
    event_status_request = EventStatusRequest(
        stripe_session_id=stripe_session_id, stripe_invoice_id=stripe_invoice_id
    )
    data = billing_session.get_event_status(event_status_request)
    return json_success(request, data)


@zulip_login_required
@typed_endpoint
def event_status_page(
    request: HttpRequest,
    *,
    stripe_session_id: str = "",
    stripe_invoice_id: str = "",
) -> HttpResponse:
    context = {
        "stripe_session_id": stripe_session_id,
        "stripe_invoice_id": stripe_invoice_id,
        "billing_base_url": "",
    }
    return render(request, "corporate/billing/event_status.html", context=context)


@self_hosting_management_endpoint
@typed_endpoint
def remote_realm_event_status_page(
    request: HttpRequest,
    *,
    realm_uuid: str = "",
    server_uuid: str = "",
    stripe_session_id: str = "",
    stripe_invoice_id: str = "",
) -> HttpResponse:  # nocoverage
    context = {
        "stripe_session_id": stripe_session_id,
        "stripe_invoice_id": stripe_invoice_id,
        "billing_base_url": f"/realm/{realm_uuid}",
    }
    return render(request, "corporate/billing/event_status.html", context=context)


@self_hosting_management_endpoint
@typed_endpoint
def remote_server_event_status_page(
    request: HttpRequest,
    *,
    realm_uuid: str = "",
    server_uuid: str = "",
    stripe_session_id: str = "",
    stripe_invoice_id: str = "",
) -> HttpResponse:  # nocoverage
    context = {
        "stripe_session_id": stripe_session_id,
        "stripe_invoice_id": stripe_invoice_id,
        "billing_base_url": f"/server/{server_uuid}",
    }
    return render(request, "corporate/billing/event_status.html", context=context)