o i%@sdZddlZddlmZddlmZddlmZmZm Z m Z m Z m Z ddl mZddlmZddlmZdd lmZmZeGd d d ZeGd d d ZeGdddZGdddeZGdddZdS)zSmallWebRTC request handler for managing peer connections. This module provides a client for handling web requests and managing WebRTC connections. N) dataclass)Enum)Any AwaitableCallableDictListOptional)candidate_from_sdp) HTTPException)logger) IceServerSmallWebRTCConnectionc@sdeZdZUdZeed<eed<dZeeed<dZee ed<dZ ee ed<e de fd d ZdS) SmallWebRTCRequestaSmall WebRTC transport session arguments for the runner. Parameters: sdp: The SDP string (Session Description Protocol). type: The type of the SDP, either "offer" or "answer". pc_id: Optional identifier for the peer connection. restart_pc: Optional whether to restart the peer connection. request_data: Optional custom data sent by the customer. sdptypeNpc_id restart_pc request_datadatacCs,d|vrd|vr|d|d<|di|S)z@Accept both snake_case and camelCase for the request_data field. requestDatarN)pop)clsrrrb/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/transports/smallwebrtc/request_handler.py from_dict*szSmallWebRTCRequest.from_dict)__name__ __module__ __qualname____doc__str__annotations__rr rboolrr classmethoddictrrrrrrs  rc@s*eZdZUdZeed<eed<eed<dS) IceCandidatea*The remote ice candidate object received from the peer connection. Parameters: candidate: The ice candidate patch SDP string (Session Description Protocol). sdp_mid: The SDP mid for the candidate patch. sdp_mline_index: The SDP mline index for the candidate patch. candidatesdp_midsdp_mline_indexN)rrrrr r!intrrrrr%2s  r%c@s&eZdZUdZeed<eeed<dS)SmallWebRTCPatchRequestzSmall WebRTC transport session arguments for the runner. Parameters: pc_id: Identifier for the peer connection. candidates: A list of ICE candidate patches. r candidatesN)rrrrr r!rr%rrrrr*As r*c@seZdZdZdZdZdS)ConnectionModez,Enum defining the connection handling modes.singlemultipleN)rrrrSINGLEMULTIPLErrrrr,Nsr,c @seZdZdZdddejfdeeede dee deddf d d Z d ee ddfd d Z ddeeefddZ dedeegedfdeee e ffddZdefddZddZdS)SmallWebRTCRequestHandleraSmallWebRTC request handler for managing peer connections. This class is responsible for: - Handling incoming SmallWebRTC requests. - Creating and managing WebRTC peer connections. - Supporting ESP32-specific SDP munging if enabled. - Invoking callbacks for newly initialized connections. - Supporting both single and multiple connection modes. NF ice_servers esp32_modehostconnection_modereturncCs"||_||_||_||_i|_dS)a7Initialize a SmallWebRTC request handler. Args: ice_servers (Optional[List[IceServer]]): List of ICE servers to use for WebRTC connections. esp32_mode (bool): If True, enables ESP32-specific SDP munging. host (Optional[str]): Host address used for SDP munging in ESP32 mode. Ignored if `esp32_mode` is False. connection_mode (ConnectionMode): Mode of operation for handling connections. SINGLE allows only one active connection, MULTIPLE allows several. N) _ice_servers _esp32_mode_host_connection_mode_pcs_map)selfr2r3r4r5rrr__init__`s  z"SmallWebRTCRequestHandler.__init__rcCs||jtjkrdS|js dStt|j}|j|kr/|r/t d|jd|t ddd|sSmallWebRTCRequestHandler._check_single_connection_constraintscCs ||_dS)z;Update the list of ICE servers used for WebRTC connections.N)r7)r<r2rrrupdate_ice_serverss z,SmallWebRTCRequestHandler.update_ice_serversrequestwebrtc_connection_callbackc sz|j}||rj|nd}|r1|}td||j|j|j|j p*ddIdHnQt j d}|j |j|jdIdH| ddt ffd d }z||IdHtd |jWnty}ztd |jd |WYd}~nd}~ww|}jrddlm} | |dj|d<|j|d<|WSty} ztd| td|d} ~ ww)aBHandle a SmallWebRTC request and resolve the pending answer. This method will: - Reuse an existing WebRTC connection if `pc_id` exists. - Otherwise, create a new `SmallWebRTCConnection`. - Invoke the provided callback with the connection. - Manage ESP32-specific munging if enabled. - Enforce single/multiple connection mode constraints. Args: request (SmallWebRTCRequest): The incoming WebRTC request, containing SDP, type, and optionally a `pc_id`. webrtc_connection_callback (Callable[[Any], Awaitable[None]]): An asynchronous callback function that is invoked with the WebRTC connection. Returns: Dictionary containing SDP answer, type, and peer connection ID, or None if no answer is available. Raises: HTTPException: If connection mode constraints are violated Exception: Any exception raised during request handling or callback execution will be logged and propagated. Nz'Reusing existing connection for pc_id: F)rrr)r2)rrclosedwebrtc_connectioncs(td|jj|jddS)Nz&Discarding peer connection for pc_id: )r inforr;r)rKr<rrhandle_disconnectedszISmallWebRTCRequestHandler.handle_web_request..handle_disconnectedz;webrtc_connection_callback executed successfully for peer: z+webrtc_connection_callback failed for peer z: r)smallwebrtc_sdp_mungingrrz&Error processing SmallWebRTC request: zSmallWebRTC request details: )rrFr;getr rL renegotiaterrrrr7 initialize event_handlerdebug Exceptionerror get_answerr8pipecat.runner.utilsrOr9) r<rHrIrrEpipecat_connectionrNcallback_erroranswerrOerrMrhandle_web_requestsP    z,SmallWebRTCRequestHandler.handle_web_requestcsZ|j|j}|stddd|jD]}t|j}|j|_|j |_ | |IdHqdS)z-Handle a SmallWebRTC patch candidate request.izPeer connection not foundr>N) r;rPrr r+r r&r'sdpMidr( sdpMLineIndexadd_ice_candidate)r<rHpeer_connectioncr&rrrhandle_patch_requests   z.SmallWebRTCRequestHandler.handle_patch_requestcs4dd|jD}tj|IdH|jdS)zClear the connection map.cSsg|]}|qSr) disconnect).0pcrrr sz3SmallWebRTCRequestHandler.close..N)r;rCasynciogatherclear)r<corosrrrcloseszSmallWebRTCRequestHandler.close)N)rrrrr,r0r rr r"r r=rFrGrrrrrr]r*rcrlrrrrr1Us8   ! Q r1)rrh dataclassesrenumrtypingrrrrrr aiortc.sdpr fastapir logurur )pipecat.transports.smallwebrtc.connectionr rrr%r*r,r1rrrrs"