a %i@sdZddlmZmZddlmZmZddlZddlmZddl m Z dd d d dd d d d dd d dZ e d dde e ee efdddZe edddZe ee efdddZee ee e ffdddZdS)z Energy Refill System Handles energy refill purchases using diamonds. Supports multiple tiers including unlimited energy for 24 hours. )datetime timedelta)DictAnyN)get_database_connection rate_limit )energydiamonds2d#)Zsmallmediumfull unlimited_24h<)Z max_callsZ time_window) player_id refill_typereturnc Csd}d}zz|tvrDdd|dWW|r6||rB|St|}|d}|d}t}|jdd}|d |f|}|sdd dWW|r||r|S|d|krdd |d |ddWW|r||r|Sd}|d kr0ttdd}|d} |d|| |||fn*t |d||d} |d|| ||f|j dkr| dddWW|r||r|S|d|} |d|||||d| f| t d|d|d|ddd| | |r|ndddWW|r||r&|Sty} z\|rF| t jd|d| dddd dWYd} ~ W|r||r|Sd} ~ 00W|r||r|n|r||r|0dS)!a3 Handle energy refill purchase. Rate limited to 10 purchases per minute to prevent abuse. Args: player_id: The player's ID refill_type: Type of refill ('small', 'medium', 'full', 'unlimited_24h') Returns: dict with 'success', 'message', and 'new_balance' keys NFzInvalid refill type: )successmessager r T dictionaryzVSELECT diamonds, energy, max_energy, unlimited_energy_until FROM players WHERE id = %szPlayer not foundzNot enough diamonds. Need z, have r)hoursZ max_energyzUPDATE players SET diamonds = diamonds - %s, energy = %s, unlimited_energy_until = %s WHERE id = %s AND diamonds >= %szUPDATE players SET diamonds = diamonds - %s, energy = %s WHERE id = %s AND diamonds >= %srz>Insufficient diamonds (concurrent purchase or balance changed)zINSERT INTO energy_refill_purchases (player_id, refill_type, energy_amount, diamond_cost, energy_before, energy_after) VALUES (%s, %s, %s, %s, %s, %s)zPlayer z purchased z refill for z diamondszEnergy refilled successfully)r r unlimited_until)rr new_balancezEnergy refill error for player : )exc_infozServer error. Please try again.) REFILL_TIERSclosercursorexecutefetchonernowrminrowcountrollbackcommitlogginginfo isoformat Exceptionerror) rrconnr$ZtierZ diamond_costZ energy_amountplayerrZ new_energyZ new_diamondser4UPDATE players SET unlimited_energy_until = NULL WHERE id = %sz+Error checking unlimited energy for player r ) rr$r%r&r#rr'r+r/r,r0)rr1r$r2r3r4r4r5check_unlimited_energysf     r8)r message_datacCs|d}|s$||dddddSddlm}t||}|d r||d d d |d d||d|d d |d d|d dddn||dd|dddS)z WebSocket message handler for 'purchaseEnergyRefill' messages. Args: player_id: The player's ID message_data: Message data containing 'refillType' send_to_client: Function to send message back to client Z refillTyper0ZINVALID_REQUESTzMissing refillType)type error_coderNrrrZpurchaseCompleter Tr)r:categoryrZ newBalanceZ playerUpdater r)r r r7)r:dataZPURCHASE_FAILEDr)get validationr r6)rr9send_to_clientrr resultr4r4r5handle_purchase_energy_refills:       rB)rcCstS)z~ Get all available refill tiers. Returns: Dictionary of refill tiers with their energy and diamond costs )r"copyr4r4r4r5get_refill_tierssrD)__doc__rrtypingrrr,databaserr?r r"intstrr6boolr8rBrDr4r4r4r5s    03