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
"""

import logging
from dataclasses import dataclass

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


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: User | None, 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}'
    if price_info.has_discount:
        return (
            f'📅 {period_label} - '
            f'{format_price_func(price_info.base_price)} ➜ '
            f'{format_price_func(price_info.final_price)}'
        )
    return f'📅 {period_label} - {format_price_func(price_info.final_price)}'