U Hif@s`ddlZddlmZmZmZddlmZddlmZddl m Z GdddZ Gdd d Z dS) N)OptionalCallable Awaitable) WebSocket)WebSocketDisconnect)Logc@s4eZdZdZddZddddZdddd 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__szConnectionState.__init__NreturncCsdS)z%Reset state when a new stream starts.Nrr rrrreset_stream_statesz"ConnectionState.reset_stream_statecCsdS)z2Clear response-related state during interruptions.Nrr rrrclear_response_statesz$ConnectionState.clear_response_state)__name__ __module__ __qualname____doc__rrrrrrrr s rc@seZdZdZedddZeddddZeege dfeege dfee ge dfdd d d Z e dd d dZ ddddZ deee ddddZdS)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-sz#WebSocketConnectionManager.__init__N)messagerc sz*|jjjdkrWdS|j|IdHWntk r}zft|}d|ksd|ksd|krbn|jjr:ddlm}|}||jj|}||IdHdS)z4Send a checkpoint event to Mcube using AudioService.r AudioServiceN)rr services.audio_servicer?create_checkpoint_messager,)r r"r? audio_serviceZcheckpoint_eventrrrsend_checkpoint_to_mcubes  z3WebSocketConnectionManager.send_checkpoint_to_mcubercs<|jjr8ddlm}|}||jj}||IdHdS)z/Clear audio buffer in Mcube using AudioService.rr>N)rr r@r?create_clear_messager,)r r?rBZ clear_eventrrrclear_mcube_audios  z,WebSocketConnectionManager.clear_mcube_audio)codereasonrc s`z(|jj||pddIdHtdWn2tk rZ}ztd|W5d}~XYnXdS)zeClose the Mcube WebSocket connection gracefully. This ends the call on Mcube's side. znormal closure)rGrHNz!Closed Mcube WebSocket connectionz!Failed to close Mcube WebSocket: )rcloserr;r$r:)r rGrHr*rrrclose_mcube_connections z1WebSocketConnectionManager.close_mcube_connection)rFN)rrrrrrdictr,rrr%r=rCrEintrrJrrrrr"s  B  r) r7typingrrrfastapirfastapi.websocketsrservices.log_utilsrrrrrrrs