zulip/corporate/views/session.py

import logging

from django.http import HttpRequest, HttpResponse
from pydantic import Json

from corporate.lib.decorator import (
    authenticated_remote_realm_management_endpoint,
    authenticated_remote_server_management_endpoint,
)
from corporate.lib.stripe import (
    RealmBillingSession,
    RemoteRealmBillingSession,
    RemoteServerBillingSession,
)
from zerver.decorator import require_billing_access, require_organization_member
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_billing_access
def start_card_update_stripe_session(request: HttpRequest, user: UserProfile) -> HttpResponse:
    billing_session = RealmBillingSession(user)
    session_data = billing_session.create_card_update_session()
    return json_success(
        request,
        data=session_data,
    )


@authenticated_remote_realm_management_endpoint
def start_card_update_stripe_session_for_remote_realm(
    request: HttpRequest, billing_session: RemoteRealmBillingSession
) -> HttpResponse:  # nocoverage
    session_data = billing_session.create_card_update_session()
    return json_success(
        request,
        data=session_data,
    )


@authenticated_remote_server_management_endpoint
def start_card_update_stripe_session_for_remote_server(
    request: HttpRequest, billing_session: RemoteServerBillingSession
) -> HttpResponse:  # nocoverage
    session_data = billing_session.create_card_update_session()
    return json_success(
        request,
        data=session_data,
    )


@require_organization_member
@typed_endpoint
def start_card_update_stripe_session_for_realm_upgrade(
    request: HttpRequest,
    user: UserProfile,
    *,
    manual_license_management: Json[bool] = False,
    tier: Json[int],
) -> HttpResponse:
    billing_session = RealmBillingSession(user)
    session_data = billing_session.create_card_update_session_for_upgrade(
        manual_license_management, tier
    )
    return json_success(
        request,
        data=session_data,
    )


@authenticated_remote_realm_management_endpoint
@typed_endpoint
def start_card_update_stripe_session_for_remote_realm_upgrade(
    request: HttpRequest,
    billing_session: RemoteRealmBillingSession,
    *,
    manual_license_management: Json[bool] = False,
    tier: Json[int],
) -> HttpResponse:
    session_data = billing_session.create_card_update_session_for_upgrade(
        manual_license_management, tier
    )
    return json_success(
        request,
        data=session_data,
    )


@authenticated_remote_server_management_endpoint
@typed_endpoint
def start_card_update_stripe_session_for_remote_server_upgrade(
    request: HttpRequest,
    billing_session: RemoteServerBillingSession,
    *,
    manual_license_management: Json[bool] = False,
    tier: Json[int],
) -> HttpResponse:
    session_data = billing_session.create_card_update_session_for_upgrade(
        manual_license_management, tier
    )
    return json_success(
        request,
        data=session_data,
    )