qiDdZddlZddlZddlZddlmZmZmZmZm Z ddl Z ddl m Z ddl mZmZddlmZmZmZmZmZmZGddZy) zWhatsApp API Client. This module provides a client for communicating with the WhatsApp Cloud API, handling webhook requests, managing WebRTC connections, and processing WhatsApp call events. N) AwaitableCallableDictListOptional)logger) IceServerSmallWebRTCConnection) WhatsAppApiWhatsAppConnectCallWhatsAppConnectCallValueWhatsAppTerminateCallWhatsAppTerminateCallValueWhatsAppWebhookRequestc PeZdZdZ ddededej deee deeddf d Z ddeee fd Z ddeefd Z defd Z defd ZddZdeeefdedefdZdedefdZ d dedeeegedfdeedeedef dZdedefdZdedefdZdedefdZy)!WhatsAppClientaWhatsApp Cloud API client for handling calls and webhook requests. This client manages WhatsApp call connections using WebRTC, processes webhook events from WhatsApp, and maintains ongoing call state. It supports both incoming call handling and call termination through the WhatsApp Cloud API. Attributes: _whatsapp_api: WhatsApp API instance for making API calls _ongoing_calls_map: Dictionary mapping call IDs to WebRTC connections _ice_servers: List of ICE servers for WebRTC connections Nwhatsapp_tokenphone_number_idsession ice_serverswhatsapp_secretreturnc~t||||_||_i|_|t dg|_y||_y)aInitialize the WhatsApp client. Args: whatsapp_token: WhatsApp API access token phone_number_id: WhatsApp phone number ID for the business account session: aiohttp session for making HTTP requests ice_servers: List of ICE servers for WebRTC connections. If None, defaults to Google's public STUN server whatsapp_secret: WhatsApp APP secret for validating that the webhook request came from WhatsApp. )rrrNzstun:stun.l.google.com:19302)urls)r _whatsapp_api_whatsapp_secret_ongoing_calls_mapr _ice_servers)selfrrrrrs T/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/transports/whatsapp/client.py__init__zWhatsAppClient.__init__.sL$))?T[ !0DF  !*0N!O PD  +D c||_y)z;Update the list of ICE servers used for WebRTC connections.N)r)rrs r update_ice_serversz!WhatsAppClient.update_ice_serversLs 'r"c||_y)zZUpdate the WhatsApp APP secret for validating that the webhook request came from WhatsApp.N)r)rrs r update_whatsapp_secretz%WhatsAppClient.update_whatsapp_secretPs /r"c:|jj|y)z%Update the WhatsApp API access token.N)rupdate_whatsapp_token)rrs r r(z$WhatsAppClient.update_whatsapp_tokenTs 00@r"c:|jj|y)z7Update the WhatsApp phone number ID for authentication.N)rupdate_whatsapp_phone_number_id)rrs r r*z.WhatsAppClient.update_whatsapp_phone_number_idXs ::?Kr"cKtjd|jstjdytjdt|jdg}|jj D]r\}}tjd||j r*|j |j j||j |jttj|ddid{|jjtjd y74w) a,Terminate all ongoing WhatsApp calls. This method will: 1. Send termination requests to WhatsApp API for each ongoing call 2. Disconnect all WebRTC connections 3. Clear the ongoing calls map All terminations are executed concurrently for efficiency. z)Will terminate all ongoing WhatsApp callszNo ongoing calls to terminateNz Terminating z ongoing callszTerminating call return_exceptionsTz!All calls terminated successfully) rdebugrlenitemsrappendterminate_call_to_whatsapp disconnectasynciogatherclear)rtermination_taskscall_idpipecat_connections r terminate_all_callsz"WhatsAppClient.terminate_all_calls\s  @A&& LL8 9  |C(?(?$@#APQ+/+B+B+H+H+J F 'G' LL,WI6 7!!!((););)V)VW^)_`  $ $%7%B%B%D E  Fnn/H4HHH %%' 89 IsD EE5Eparamsexpected_verification_tokencK|jd}|jd}|jd}|r|r|s td|dk7rtd|d||k7r tdt|Sw) aHandle a verify webhook request from WhatsApp. Args: params: Dictionary containing webhook parameters from query string expected_verification_token: The expected verification token to validate against Returns: int: The challenge value if verification succeeds Raises: ValueError: If verification fails due to missing parameters or invalid token zhub.modez hub.challengezhub.verify_tokenz0Missing required webhook verification parameters subscribez-Invalid hub mode: expected 'subscribe', got ''z#Webhook verification token mismatch)get ValueErrorint)rr:r;mode challenge verify_tokens r handle_verify_webhook_requestz,WhatsAppClient.handle_verify_webhook_requestszz*%JJ/ zz"45 9LOP P ; LTFRSTU U 6 6BC C9~sA4A6raw_bodysha256_signaturecTKtj|jjd|tj j }|s td|jdd}tj||s tdtjdyw) z6Common handler for both /start and /connect endpoints.zutf-8)keymsg digestmodz"Missing X-Hub-Signature-256 headerzsha256=zInvalid webhook signaturezWebhook signature verified!N) hmacnewrencodehashlibsha256 hexdigest Exceptionsplitcompare_digestrr-)rrFrGexpected_signaturereceived_signatures r "_validate_whatsapp_webhook_requestz1WhatsAppClient._validate_whatsapp_webhook_requests"XX%%,,W5nn  )+  @A A-33I>rB""#57IJ78 8 24sB&B(requestconnection_callbackc `K |jr|j||d{|jD]Y}|jD]F}t |j t r|j jD]}|jdk(stjd|j |j|d{}|r5|r3 ||d{tjd|jyt |j ts|j jD]S}|jd k(stjd |j |j!|d{cccSI\d } tj"| d|t%| 777#t$r/} tjd|jd| Yd} ~ d} ~ wwxYw#t$r+} tjd|jd| d} ~ wwxYw7#t$r+} tjd |jd| d} ~ wwxYw#t$r6} tjd | tjd|d} ~ wwxYww)aHandle a webhook request from WhatsApp. This method processes incoming webhook requests and handles both connect and terminate events. For connect events, it establishes a WebRTC connection and optionally invokes a callback with the new connection. Args: request: The webhook request from WhatsApp containing call events connection_callback: Optional callback function to invoke when a new WebRTC connection is established. The callback receives the SmallWebRTCConnection instance. raw_body: Optional bytes containing the raw request body. sha256_signature: Optional X-Hub-Signature-256 header value from the request. Returns: bool: True if the webhook request was handled successfully, False otherwise Raises: ValueError: If the webhook request contains no supported events Exception: If connection establishment or API calls fail Nconnectz"Processing connect event for call z3Connection callback executed successfully for call z$Connection callback failed for call : T(Failed to handle connect event for call terminatez$Processing terminate event for call z*Failed to handle terminate event for call z+No supported event found in webhook requestz"Error processing webhook request: zWebhook request details: )rrXentrychanges isinstancevaluer callseventrr-id_handle_connect_eventrSerrorr_handle_terminate_eventwarningr@)rrYrZrFrGr`changecall connectioncallback_error connect_errorterminate_error error_msges r handle_webhook_requestz%WhatsAppClient.handle_webhook_requests:6 $$==hHXYYY ) *#mm(*F!&,,0HI$*LL$6$6*D#zzY6 & /QRVRYRYQZ-[ \!*7;7Q7QRV7W1WJ(;z).2Ej2Q,Q,Q,2LL2efjfmfmen0o-.,0'*6$FLL2LM$*LL$6$6 *D#zz[8 & /STXT[T[S\-] ^!*151M1Md1S+S$S *?(*) *XFI NNi[7)4 5Y' '_Z2X -R09).,2LL2VW[W^W^V__abpaq0r-.-.).(1!*$*LL*RSWSZSZR[[]^k]l(m%&%* !*,T'0!*$*LL*TUYU\U\T]]_`o_p(q%&%* !*  LL=aSA B LL4WI> ?  sJ.!I,F:A&I, "I,0G<F= G< GF? &GI,J.I,!(I, "I,-H5H3 H5I, J. /I,=G<?G G9 %G4 /G<4G9 9G<< H0 &H+ +H0 0I,3H55 I) >&I$ $I) )I,, J+51J&&J++J.sdpc|j}g}|D]6}|jdr|jds&|j|8dj|dzS)a>Filter SDP to be compatible with WhatsApp requirements. WhatsApp only supports SHA-256 fingerprints, so this method removes other fingerprint types from the SDP. Args: sdp: The original SDP string Returns: Filtered SDP string compatible with WhatsApp za=fingerprint:za=fingerprint:sha-256z ) splitlines startswithr0join)rrtlinesfilteredlines r _filter_sdp_for_whatsappz'WhatsAppClient._filter_sdp_for_whatsappsb  "D/0I`9a OOD ! "{{8$v--r"rlcJKtjdjdjd} t j }|j jjjjd{|jjd}j|}tjdj jjjd|jd{}|jdd s3tjd jd |t!d |tjd j jjjd|jd{}|jdd s3tjdjd |t!d|tjdj|j"j<|j%ddtffd }tjdj|S77o#t $r8}tjdjd |t!d |d}~wwxYw7#t $r8}tjdjd |t!d|d}~wwxYw#t $r}|rV |j'd{7n;#t $r/}tjdjd |Yd}~nd}~wwxYwtjdjd |d}~wwxYww)aHandle a CONNECT event by establishing WebRTC connection and accepting the call. This method: 1. Creates a new WebRTC connection using configured ICE servers 2. Initializes the connection with the provided SDP 3. Generates an SDP answer and filters it for WhatsApp compatibility 4. Pre-accepts the call with WhatsApp API 5. Accepts the call with WhatsApp API 6. Stores the connection for later management Args: call: WhatsApp connect call event Returns: The established SmallWebRTCConnection instance Raises: Exception: If pre-accept or accept API calls fail zIncoming call from , call_id: N)rttypertzSDP answer generated for call pre_acceptsuccessFzFailed to pre-accept call r]zFailed to pre-accept call: zPre-accept successful for call z$Pre-accept API call failed for call acceptzFailed to accept call zFailed to accept call: zAccept successful for call z Accept API call failed for call closedwebrtc_connectioncKtjd|jdjjj jdyw)NzPeer connection closed: z for call )rr-pc_idrfrpop)rrlrs r handle_disconnectedzAWhatsAppClient._handle_connect_event..handle_disconnectedZsO ./@/F/F.GzRVRYRYQZ[''++DGGT:sAAz4WebRTC connection established successfully for call z&Failed to cleanup connection for call r^)rr-from_rfr r initializerrtsdp_type get_answerr?r|ranswer_call_to_whatsapprhrSr event_handlerr2) rrlr8 sdp_answerpre_accept_resprr accept_respr cleanup_errors `` r rgz$WhatsAppClient._handle_connect_eventsK(  *4::,k$''KL!? !6t7H7H!I $//DLL4D4D4<twwiHI ?$($6$6$N$NGGXz4::% #y%8LL#9$''"[M!RS#&=k]$KLL :477)DE 0BD # #DGG , - -h 7 ;=R ;8 ; LLOPTPWPWyY Z% %_ b# C CDGG9BqcRS"=aS ABB C  ? ?y1#NO"9! =>> ?& !,77999 LL@ M?[ LLCDGG9BqcR S  s2N#AL J AL"5J JA+J 5K9K :A+K%ALN#LJ K 3KK  L K L3L  LL N N L:3L64L:9N: M2%M-(N-M22)NN  N#cKtjd|jd|jtjd|j|j r#tjd|j d |j|j vr|j |j}tjd|j |jd{tjd|j|j j|jdy tjd |jd y 7q#t$r/}tjd |jd |Yd}~d}~wwxYw#t$r/}tjd|jd |Yd}~yd}~wwxYww)aHandle a TERMINATE event by cleaning up resources and logging call completion. This method: 1. Logs call termination details including duration if available 2. Disconnects the associated WebRTC connection 3. Removes the call from the ongoing calls map Args: call: WhatsApp terminate call event Returns: bool: True if the call was terminated successfully, False otherwise zCall terminated from r~z Call status: zCall duration: z secondsz)Disconnecting WebRTC connection for call Nz5WebRTC connection disconnected successfully for call z0Failed to disconnect WebRTC connection for call r]zCall z not found in ongoing calls mapTz(Error handling terminate event for call F) rr-rrfstatusdurationrr2rSrhrrj)rrlr8disconnect_errorrrs r riz&WhatsAppClient._handle_terminate_eventrs  ,TZZL DGG9MN }T[[M23 == LL?4==/B C ww$111%)%<%rs: <<Vzzr"