a 'Ni@sdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZddlmZddlmZddlmZmZdd Zd d Zd d ZddZddZddZdS)a8 WebSocket lifecycle and handler functions. This module contains the core WebSocket connection handlers, lifecycle management functions, and background tasks for the BaoLife game server. Functions: - start: Handles WebSocket connection initialization and game loading - shutdown: Handles WebSocket disconnection and cleanup - error: Sends error messages to clients - every_minute: Background task for periodic game iteration - handler: Main WebSocket connection handler - initialize_dummy_user: Creates a dummy user for server initialization N) exceptions) loadGameAsync saveGameAsync playerClassgetOccupations insertGameconnect)config)USERS) sendUserInfosendDictc sddl}|j}|j}|j}h}tdd}|j}|durtdjtjIdH}|r|jt j krvt j |_d|_ d|_ t||j|td|std t}t|_j|_t|t||j|nT|}td |jjd |jjd t|jj|jt j kr(t j |_d|_ t|t|IdHtd td|ddlm}fdd} ||j| t !||IdHdS)a Initialize a WebSocket connection and load/create player game. Handles: - Loading existing games from database - Creating new games for first-time players - Reconnecting disconnected players - Initializing game controller and connection state - Sending initial player data to client - Checking daily login rewards Args: websocket: WebSocket connection with userID attribute Returns: None - Starts consumer and producer handlers via asyncio.gather rNzstarting connection...Fzattempting load for active connectedz loaded gamez3no loaded games, no saved game -- creating new gamez reconnected  z age zUSERS: z Players: )handle_daily_login_checkcstt|dS)N)asyncio create_taskr ) player_idmessage websocket:/var/www/lichun.app/lichun/ws/server/websocket_handlers.pysend_to_client_wrapperasz%start..send_to_client_wrapper)"appconsumer_handlerproducer_handler playerRecordsprintgetuserIDr gameSpeedr SPEED_QUESTION_PAUSE SPEED_DEFAULT controller connectionrsetrr occupationsidrc firstnamelastnamestrageYearsr r countsizeretention.daily_rewardsrrgather) rrrrrrplayerZ cached_playerrrrrrstart$sT  ,   r3csdddl}|j}ttjtd||j}|r`d|j_d|_ d|_ t |IdHt |dSdS)a^ Handle WebSocket disconnection and cleanup. Performs: - Resets offline tracking - Updates player connection state - Saves game to database - Removes from active users registry Args: websocket: WebSocket connection with userID attribute Returns: bool: True if cleanup successful, None otherwise rNz!Client disconnected. Do cleanup disconnectedinactiveT)rrrwebsockets_exceptionsConnectionClosedrr offlineStatsminutesOffliner%r$rr remove)rrrr2rrrshutdownks   r;cs$d|d}|t|IdHdS)z Send an error message to the client. Args: websocket: WebSocket connection to send error to message: Error message string Returns: None error)typerN)sendjsondumps)rreventrrrr<s r<c sddl}|j}ddlm}td|IdH|}td|dd|dd |d d |d d |ddd tdIdHqdS)aP Background task that runs every minute to iterate offline games. This task: - Checks for disconnected games that need updates - Iterates game state for offline players (1 game-minute per real-world minute) - Logs player cache statistics Runs indefinitely until server shutdown. Returns: None rN) iterateGamesz(every minute, checking for offline gamesz PlayerCache: r//max_sizez players (rz connected, r4z disconnected), memory_mbz.1fMB<)rrrBr get_statsrsleep)rrrBstatsrrr every_minutes   rKc szttj|IdH}t|}|ddkrf|d|_td|jt|t |IdHnT|d|_td|jt|t |IdHt |dIdH| IdHWdS| IdHWnHt j y}z,tdt|t|IdHWYd}~dSd}~00dS) a Main WebSocket connection handler. This is the entry point for all WebSocket connections. It: - Receives and validates the initial "init" message - Extracts userID from the init message - Registers the connection in the USERS registry - Delegates to start() for game initialization - Handles connection errors and cleanup Args: websocket: WebSocket connection object Returns: bool: False on error or completion Nr=initr zClient connected with userID zFirst message must be 'init'Fz)handler: Client disconnected. Do cleanup)rdatetimenowrecvr?loadsr r addr3r<closer6r7r,r;)rrrAerrrrrhandlers*      rTc sd}t|4IdH}ddd}z(|t|IdH|IdHWnDtjy}z*tdt |t |IdHWYd}~n d}~00WdIdHq1IdHs0YdS)a Create a dummy user for server initialization testing. This function connects to the local WebSocket server and sends a test message to ensure the server is properly initialized before accepting real connections. Returns: bool: False after completion zws://localhost:8001NtestZ DUMMY_USER_ID)r=r z7initialize_dummy_user: Client disconnected. Do cleanupF) websocketsrr>r?r@rRr6r7rr,r;)urirZ dummy_messagerSrrrinitialize_dummy_users  NrX)__doc__rrMr? tracebackrVrr6 functionsrrrrrrr server.websocket_registryr server.websocket_messagingr r r3r;r<rKrTrXrrrrs     G! *