o i @s&ddlZddlmZGdddZdS)N)Optionalc@s^eZdZdZddefddZddd Zdefd d Zdefd d Z dddZ defddZ dS)SmallWebRTCSessionManageraY Manages the waiting logic between PipecatSession and SmallWebRTCSession. This class provides: - A mechanism to wait for a WebRTC connection within a timeout period. - Automatic cancellation if no connection is established in time. - A way to complete the waiting future once the session has successfully started. xtimeout_secondscCsd|_d|_||_dS)N)_pending_future _timeout_task_timeout_seconds)selfrr \/home/ubuntu/.local/lib/python3.10/site-packages/pipecatcloud/smallwebrtc/session_manager.py__init__s z"SmallWebRTCSessionManager.__init__returnNcs^jdur tdt_fdd}t|_z jIdHWdSw)a Creates and waits for a future to be completed when WebRTC connection arrives. Raises: TimeoutError: If WebRTC connection is not received within the timeout period RuntimeError: If another waiting operation is already in progress Nz%Already waiting for WebRTC connectioncsJtjIdHjr!js#jtdjddSdSdS)Nz&WebRTC connection not received within z seconds)asynciosleeprrdone set_exception TimeoutErrorr r r r timeout_handler!s zBSmallWebRTCSessionManager.wait_for_webrtc..timeout_handler)r RuntimeErrorrFuture create_taskr_cleanup)r rr rr wait_for_webrtcs    z)SmallWebRTCSessionManager.wait_for_webrtccCs,|jr|js||jddSdS)z Completes the waiting future. Returns: bool: True if the future was found and completed, False otherwise TF)rrcancel_timeout set_resultrr r r complete_session2s  z*SmallWebRTCSessionManager.complete_sessioncCs"|jr|js|jdSdS)z Cancels the timeout handler without completing the future. Returns: bool: True if the timeout task was found and cancelled, False otherwise TF)rrcancelrr r r rAs z(SmallWebRTCSessionManager.cancel_timeoutcCsd|_d|_dS)zCleans up all resources.N)rrrr r r rMs z"SmallWebRTCSessionManager._cleanupcCs|jduo |j S)z Checks if currently waiting for WebRTC connection. Returns: bool: True if waiting for connection, False otherwise N)rrrr r r is_waitingRsz$SmallWebRTCSessionManager.is_waiting)r)r N) __name__ __module__ __qualname____doc__intr rboolrrrrr r r r rs  r)rtypingrrr r r r s