remnawave-bedolaga-telegram-bot/app/utils/price_display.py

"""
Unified price display system for all subscription and balance pricing.

This module provides a centralized way to:
- Calculate prices with all applicable discounts (promo groups, promo offers)
- Format price buttons consistently across all flows
- Ensure uniform discount display throughout the application
"""

from dataclasses import dataclass
from typing import Optional
import logging

from app.database.models import User
from app.config import settings

logger = logging.getLogger(__name__)


@dataclass
class PriceInfo:
    """Container for pricing information with discounts."""

    base_price: int  # Original price without any discounts (kopeks)
    final_price: int  # Final price after all discounts (kopeks)
    discount_percent: int  # Total discount percentage

    @property
    def has_discount(self) -> bool:
        """Check if there's any discount applied."""
        return self.base_price > self.final_price and self.discount_percent > 0

    @property
    def discount_value(self) -> int:
        """Get the absolute discount value in kopeks."""
        return self.base_price - self.final_price


def calculate_user_price(
    user: Optional[User],
    base_price: int,
    period_days: int,
    category: str = "period"
) -> PriceInfo:
    """
    Calculate final price for a user with all applicable discounts.

    Args:
        user: User object (None for base/default pricing from settings)
        base_price: Base price without discounts (kopeks)
        period_days: Subscription period in days
        category: Discount category ("period", "servers", "devices", "traffic")

    Returns:
        PriceInfo with base_price, final_price, and discount_percent

    Example:
        >>> user = get_user_from_db(123)
        >>> price_info = calculate_user_price(user, 100000, 30, "period")
        >>> print(f"{price_info.base_price} -> {price_info.final_price} ({price_info.discount_percent}%)")
        100000 -> 80000 (20%)

        >>> # For base pricing (no user)
        >>> price_info = calculate_user_price(None, 100000, 30, "period")
        >>> # Uses BASE_PROMO_GROUP_PERIOD_DISCOUNTS from settings
    """
    if not base_price or base_price <= 0:
        return PriceInfo(base_price=base_price or 0, final_price=base_price or 0, discount_percent=0)

    # Get discount percentage
    if user:
        # Get user's promo group discount for this category
        discount_percent = user.get_promo_discount(category, period_days)
    else:
        # For None user, use base settings discount
        discount_percent = settings.get_base_promo_group_period_discount(period_days)

    logger.debug(
        f"calculate_user_price: user={user.telegram_id if user else 'None'}, "
        f"base_price={base_price}, period_days={period_days}, category={category}, "
        f"discount_percent={discount_percent}"
    )

    if discount_percent <= 0:
        return PriceInfo(base_price=base_price, final_price=base_price, discount_percent=0)

    # Calculate discounted price
    discount_value = (base_price * discount_percent) // 100
    final_price = base_price - discount_value

    logger.debug(
        f"Calculated price for user {user.telegram_id if user else 'None'}: "
        f"{base_price} -> {final_price} (-{discount_percent}%) "
        f"[category={category}, period={period_days}]"
    )

    return PriceInfo(
        base_price=base_price,
        final_price=final_price,
        discount_percent=discount_percent
    )


def format_price_button(
    period_label: str,
    price_info: PriceInfo,
    format_price_func,
    emphasize: bool = False,
    add_exclamation: bool = True
) -> str:
    """
    Format a price button text with unified discount display.

    Args:
        period_label: Label for the period (e.g., "30 дней", "1 месяц")
        price_info: PriceInfo object with pricing details
        format_price_func: Function to format price (usually texts.format_price)
        emphasize: Add fire emojis for emphasis (for best deals)
        add_exclamation: Add exclamation mark after discount percent

    Returns:
        Formatted button text

    Examples:
        With discount and price > 0:
            "📅 30 дней - 990₽ ➜ 693₽ (-30%)!"

        With final price = 0:
            "📅 30 дней"

        With emphasis:
            "🔥 📅 30 дней - 8990₽ ➜ 6293₽ (-30%)! 🔥"

        Without discount:
            "📅 30 дней - 990₽"
    """
    # Format button text differently if final price is 0
    if price_info.final_price == 0:
        button_text = f"📅 {period_label}"
    elif price_info.has_discount:
        exclamation = "!" if add_exclamation else ""
        button_text = (
            f"📅 {period_label} - "
            f"{format_price_func(price_info.base_price)} ➜ "
            f"{format_price_func(price_info.final_price)} "
            f"(-{price_info.discount_percent}%){exclamation}"
        )
    else:
        button_text = f"📅 {period_label} - {format_price_func(price_info.final_price)}"

    # Add emphasis for best deals
    if emphasize:
        button_text = f"🔥 {button_text} 🔥"

    logger.debug(f"Formatted button: {button_text}")
    return button_text


def format_price_text(
    period_label: str,
    price_info: PriceInfo,
    format_price_func
) -> str:
    """
    Format a price for message text (not button) with unified discount display.

    Args:
        period_label: Label for the period (e.g., "30 дней")
        price_info: PriceInfo object with pricing details
        format_price_func: Function to format price (usually texts.format_price)

    Returns:
        Formatted price text for messages

    Examples:
        With discount:
            "📅 30 дней - 990₽ ➜ 693₽"

        Without discount:
            "📅 30 дней - 990₽"
            
        With zero price:
            "📅 30 дней"
    """
    if price_info.final_price == 0:
        return f"📅 {period_label}"
    elif price_info.has_discount:
        return (
            f"📅 {period_label} - "
            f"{format_price_func(price_info.base_price)} ➜ "
            f"{format_price_func(price_info.final_price)}"
        )
    else:
        return f"📅 {period_label} - {format_price_func(price_info.final_price)}"