from decimal import Decimal
from typing import Optional
from urllib.parse import urlencode, urljoin, urlunsplit

from django.conf import settings
from django.urls import reverse

from corporate.lib.stripe import RealmBillingSession
from corporate.models import get_customer_by_realm
from zerver.models import Realm, UserProfile, get_realm


def get_support_url(realm: Realm) -> str:
    support_realm_uri = get_realm(settings.STAFF_SUBDOMAIN).uri
    support_url = urljoin(
        support_realm_uri,
        urlunsplit(("", "", reverse("support"), urlencode({"q": realm.string_id}), "")),
    )
    return support_url


def get_discount_for_realm(realm: Realm) -> Optional[Decimal]:
    customer = get_customer_by_realm(realm)
    if customer is not None:
        return customer.default_discount
    return None


def attach_discount_to_realm(realm: Realm, discount: Decimal, *, acting_user: UserProfile) -> None:
    billing_session = RealmBillingSession(acting_user, realm, support_session=True)
    billing_session.attach_discount_to_customer(discount)


def approve_realm_sponsorship(realm: Realm, *, acting_user: UserProfile) -> None:
    billing_session = RealmBillingSession(acting_user, realm, support_session=True)
    billing_session.approve_sponsorship()


def update_realm_sponsorship_status(
    realm: Realm, sponsorship_pending: bool, *, acting_user: UserProfile
) -> None:
    billing_session = RealmBillingSession(acting_user, realm, support_session=True)
    billing_session.update_customer_sponsorship_status(sponsorship_pending)


def update_realm_billing_method(
    realm: Realm, charge_automatically: bool, *, acting_user: UserProfile
) -> None:
    billing_session = RealmBillingSession(acting_user, realm, support_session=True)
    billing_session.update_billing_method_of_current_plan(charge_automatically)