qi%dZddlZddlmZddlmZddlmZmZm Z m Z m Z m Z ddl mZddlmZddlmZdd lmZmZeGd d ZeGd d ZeGddZGddeZGddZy)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) IceServerSmallWebRTCConnectioncxeZdZUdZeed<eed<dZeeed<dZee ed<dZ ee ed<e de fd Zy) 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_datadatacJd|vrd|vr|jd|d<|di|S)z@Accept both snake_case and camelCase for the request_data field. requestDatar)pop)clsrs `/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/transports/smallwebrtc/request_handler.py from_dictzSmallWebRTCRequest.from_dict*s2 D ^4%?#'88M#:D {T{)__name__ __module__ __qualname____doc__str__annotations__rr rboolrr classmethoddictrrrrrrsV H IE8C=!%J%"&L(3-&Trrc0eZdZUdZeed<eed<eed<y) 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)r r!r"r#r$r%intrrrr*r*2sN Lrr*c,eZdZUdZeed<eeed<y)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)r r!r"r#r$r%r r*rrrr0r0As J\""rr0ceZdZdZdZdZy)ConnectionModez,Enum defining the connection handling modes.singlemultipleN)r r!r"r#SINGLEMULTIPLErrrr3r3Ns6 FHrr3c eZdZdZdddej fdeeede dee deddf d Z d ee ddfd Z ddeeefd Z d edeegedfdeee e ffdZd efdZdZy)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_modereturncJ||_||_||_||_i|_y)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)selfr:r;r<r=s r__init__z"SmallWebRTCRequestHandler.__init__`s,$(% /;= rrc|jtjk7ry|jsyt t |jj }|j|k7r4|r2tjd|jd|tdd|s"tjdtddy) aCheck if the connection request satisfies single connection mode constraints. Args: pc_id: The peer connection ID from the request Raises: HTTPException: If constraints are violated in single connection mode Nz$Connection pc_id mismatch: existing=z , received=iz'PC ID mismatch with existing connection status_codedetailzMCannot create new connection: existing connection found but no pc_id receivedzSmallWebRTCRequestHandler._check_single_connection_constraintszs  N$9$9 9 }} #4 (<(<(>#?@  $ $ -% NN67J7P7P6QQ\]b\cd  C8ab b NN_  U  rc||_y)z;Update the list of ICE servers used for WebRTC connections.N)r@)rEr:s rupdate_ice_serversz,SmallWebRTCRequestHandler.update_ice_serverss 'rrequestwebrtc_connection_callbackcK |j}j||rjj|nd}|rY|}t j d||j |j|j|jxsdd{ntj}|j|j|jd{|jddtffd } ||d{t jd |j|j#}j$rd dlm} | |dj*|d<|j|d<|S777y#t$r/}t j d |jd |Yd}~d}~wwxYw#t$r6} t j d| t jd|d} ~ wwxYww)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)r:)rrclosedwebrtc_connectioncKtjd|jjj |jdyw)Nz&Discarding peer connection for pc_id: )r inforrDr)rWrEs rhandle_disconnectedzISmallWebRTCRequestHandler.handle_web_request..handle_disconnecteds>KK"HIZI`I`Ha bcMM%%&7&=&=tDsA A z;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: )rrPrDgetr rY renegotiaterrrrr@ initialize event_handlerdebug Exceptionerror get_answerrApipecat.runner.utilsr[rB) rErSrTrrOpipecat_connectionrZcallback_erroranswerr[es ` rhandle_web_requestz,SmallWebRTCRequestHandler.handle_web_requests:2 MME  5 5e <?D$--"3"3E": "%8" EeWMN(44  &11:U5 &;tGXGX%Y"(33 ',,3WWW#11(;EAVE<E 45GHHHLLUVhVnVnUop(224FH 7u tzz Ru -?DMM&/ *MEXI!LLEFXF^F^E__abpaqr  LLA!E F LL8 B C  sHBGFAGF$G F F&F 4AGHGGF G%F<7G<GG H 1G>>HHc:K|jj|j}|s tdd|jD]R}t |j }|j|_|j|_ |j|d{Ty7w)z-Handle a SmallWebRTC patch candidate request.izPeer connection not foundrHN) rDr\rr r1r r+r,sdpMidr- sdpMLineIndexadd_ice_candidate)rErSpeer_connectioncr+s rhandle_patch_requestz.SmallWebRTCRequestHandler.handle_patch_requests--++GMM:C8ST T## ?A*1;;7I yyI &'&7&7I #!33I> > >  ? ?sBBBBcK|jjDcgc]}|j}}tj|d{|jj ycc}w7$w)zClear the connection map.N)rDrM disconnectasynciogatherclear)rEpccoross rclosezSmallWebRTCRequestHandler.closesW+/==+?+?+ABRBBnne$$$ C$sA4A-A4 A2%A4)N)r r!r"r#r3r7r r rr&r$rFrPrRrrrrrrir0rprxrrrr9r9Us26 "*8*A*A =d9o.==sm = ( =  =4(3-DB(htI.G(O#O%-cUIdO-C$DO $sCx. ! Ob ?2I ?rr9)r#rs dataclassesrenumrtypingrrrrr r aiortc.sdpr fastapir logurur )pipecat.transports.smallwebrtc.connectionrrrr*r0r3r9rrrrs !AA)!V  2       # #  #Tllr