o if@s`ddlZddlmZmZmZddlmZddlmZddl m Z GdddZ Gdd d Z dS) N)OptionalCallable Awaitable) WebSocket)WebSocketDisconnect)Logc@s,eZdZdZddZd ddZd dd ZdS) ConnectionStateay Tracks the current Mcube call session and manages connection state. - Holds the current stream ID and call ID for the active Mcube call session. - Provides methods to reset or clear state when a new call starts or an interruption occurs. Audio-related state is now managed by AudioService, so this class focuses only on connection/session state. cCsd|_d|_dSN) stream_idcall_idselfr@/var/www/html/live_calls/homebook/services/connection_manager.py__init__s zConnectionState.__init__returnNcCdS)z%Reset state when a new stream starts.Nrr rrrreset_stream_statez"ConnectionState.reset_stream_statecCr)z2Clear response-related state during interruptions.Nrr rrrclear_response_staterz$ConnectionState.clear_response_staterN)__name__ __module__ __qualname____doc__rrrrrrrr s  rc @seZdZdZdefddZdeddfdd Zd eege dfd eege dfd ee ge dfddfd dZ de ddfddZ dddZ ddedee ddfddZdS)WebSocketConnectionManagera Orchestrates WebSocket communication between Mcube and ElevenLabs for Service Type 4. - Establishes, maintains, and closes WebSocket connections to Mcube (FastAPI). - Routes incoming messages to appropriate event handlers for media, start, and playedStream events. - Sends and receives messages, manages connection state, and coordinates with AudioService for checkpoint/clear events. This is the main interface for real-time, bidirectional communication between Mcube and ElevenLabs. mcube_wscCs||_t|_dSr )rrstate)r rrrrr-s z#WebSocketConnectionManager.__init__messagerNc sz|jjjdkr WdS|j|IdHWdStyb}z?t|}d|vs1d|vs1d|vr2n%ddl}t d|t d|t d | WYd}~dSWYd}~dSd}~ww) z"Send a message to Mcube WebSocket. CONNECTEDN not connectedclosedacceptru'⚠️ Error sending message to Mcube: z Message: zFull traceback: ) r client_statename send_json Exceptionstrlower tracebackrwarning format_exc)r re error_strr)rrr send_to_mcube1s  "z(WebSocketConnectionManager.send_to_mcubeon_mediaon_starton_played_streamc sz|jjjdkr WdS|j2z3dHW}|jjjdkr"WdSt|}|ddkr5||IdHq|ddkr|di}|d}|d}||j_||j_ |r_t d||d nzt d |Wnt yut d Ynw|j||IdHq|dd kr|d } | r|| IdHq6WdStyt dYdSt y} z)t| } d| vsd| vsd| vrnt d| WYd} ~ dSWYd} ~ dSd} ~ ww)a& Receive messages from Mcube and route them to appropriate handlers. Args: on_media: Handler for media events on_start: Handler for call start events (now receives full data) on_played_stream: Handler for playedStream events rNeventmediastartstreamIdcallIdz Mcube Start)r5r6zMcube Start (no callId)zCMcube start payload (no callId) and failed to serialize start_info. playedStreamr$z'Mcube WebSocket disconnected (expected)r r"r!zError in receive_from_mcube: )rr#r$ iter_textjsonloadsgetrr r rr2r&errorrrinfor'r() r r/r0r1rdata start_infor r r$r,r-rrrreceive_from_mcubeDsX            $ z-WebSocketConnectionManager.receive_from_mcuber$csD|jjr ddlm}|}||jj|}||IdHdSdS)z4Send a checkpoint event to Mcube using AudioService.r AudioServiceN)rr services.audio_servicerBcreate_checkpoint_messager.)r r$rB audio_servicecheckpoint_eventrrrsend_checkpoint_to_mcubes z3WebSocketConnectionManager.send_checkpoint_to_mcubecsB|jjrddlm}|}||jj}||IdHdSdS)z/Clear audio buffer in Mcube using AudioService.rrAN)rr rCrBcreate_clear_messager.)r rBrE clear_eventrrrclear_mcube_audios z,WebSocketConnectionManager.clear_mcube_audiocodereasonc sdz|jj||p ddIdHtdWdSty1}ztd|WYd}~dSd}~ww)zeClose the Mcube WebSocket connection gracefully. This ends the call on Mcube's side. znormal closure)rLrMNz!Closed Mcube WebSocket connectionz!Failed to close Mcube WebSocket: )rcloserr=r&r<)r rLrMr,rrrclose_mcube_connectionsz1WebSocketConnectionManager.close_mcube_connectionr)rKN)rrrrrrdictr.rrr'r@rGrJintrrOrrrrr"s   B r) r9typingrrrfastapirfastapi.websocketsrservices.log_utilsrrrrrrrs