o @boj+o@sUddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZmZddlmZddlmZmZdd lmZeeZeZeeed <ehd Ze d ej!Z"eedBe#eefgedBfZ$eegedfZ%Gd ddZ&dS)N) AwaitableCallable)Any) AriClient) log_ari_eventsummarize_channel) CallDirection CallState) CallRegistry)Settings get_settings) MediaManagerPROCESSED_CHANNELS>hagentmediainboundcustomeroutboundz>^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$c@seZdZ      dSdedededBdedBdedBdege dBfdBdege dBfdBd e dBd dfd d Z d e e e fd edBfddZede d e fddZede d efddZdddee de dBd eee dBe dBe dBffddZde e e fd e fddZd e e e fd e dBfddZd e e e fd e dBfd d!Zd e e e fd efd"d#Zd$ed dfd%d&Zd$ed'e d dfd(d)Zd$ed*ed dfd+d,Zd$ed dfd-d.Z d e e e fd edBfd/d0Z!d e e e fd edBfd1d2Z"d3e d'e d e e e fd efd4d5Z#d'e d e e e fd edBfd6d7Z$d e e e fd e fd8d9Z%d$ed e fd:d;Z&d$ed'e d dfdd?Z(d$ed dfd@dAZ)d$ed dfdBdCZ*d$ed'e d e e e fd dfdDdEZ+d'e dBd e e e fd edBfdFdGZ,de e e fd$ed efdHdIZ-d$edJe d dfdKdLZ.d e e e fd edBfdMdNZ/d e e e fd edBfdOdPZ0d e e e fd edBfdQdRZ1dS)TCallStateMachineNariregistrysettings media_manager call_resolverget_agent_endpointget_registered_extension on_bot_readyreturnc Cs:||_||_|p t|_||_||_||_||_||_dSN) rrr rr_call_resolver_get_agent_endpoint_get_registered_extension _on_bot_ready) selfrrrrrrrrr%=/var/www/html/bot-asterisk-ari/backend/calls/state_machine.py__init__s   zCallStateMachine.__init__eventcsF|d}|j|j|j|j|jd}||}|r!||IdHSdS)Ntype) StasisStart StasisEndChannelStateChangeChannelDestroyedPlaybackFinished)get_on_stasis_start_on_stasis_end_on_channel_state_change_on_channel_destroyed_on_playback_finished)r$r( event_typehandlershandlerr%r%r& handle_event1s  zCallStateMachine.handle_eventrawcCst|dS)uEQuita espacios y comillas del dialplan (ej. '21601605' → 21601605).z'")strstrip)r9r%r%r&_normalize_arg?szCallStateMachine._normalize_argvaluecCstt|S)z/UUID de la API; nunca un DNI u otra cifra pura.)bool_UUID_REmatch)r=r%r%r&_looks_like_call_idDsz$CallStateMachine._looks_like_call_iddialplan_extenargsrCc sBd}d}g}|D]5}||}|sq|dvrd}q|dkr d}q|tvr%q|r/||q||r7|}qtd|qd}d} |rI||ndrg|vrg}fdd|D} | rd| d nd} n4t|d krw|d |d }} n$t|d krr|d kr|d }nr}|d } n|d }|||| fS) u Devuelve (direction, call_id, dialed_ext, document_id). Dialplan bot típico: Stasis(App,${EXTEN},${DOCUMENTO}) API saliente: Stasis args [uuid, customer] rN)rrrz$Arg Stasis ignorado como call_id: %rcg|]}|kr|qSr%r%).0ndpr%r& pz7CallStateMachine._parse_stasis_args..r)r<_STASIS_KEYWORDSisdigitappendrAloggerdebuglen) r$rDrC direction call_id_argnumericsr9r= dialed_ext document_idrestr%rIr&_parse_stasis_argsIsF         z#CallStateMachine._parse_stasis_argschannelcCst|dpi}|dp i}|dp|d}|rt|S|dp)|dp)d}d|vr6|dddS|p9d S) N connectedcallernumbernamerE/rN Desconocido)r/r:rsplit)r$r\r]r^r_r`r%r%r&_caller_numberszCallStateMachine._caller_numbercCs6|dpi}|dp id}|durt|SdS)Nr\dialplanexten)r/r:)r$r(r\rgr%r%r&_dialplan_extensz CallStateMachine._dialplan_extencCsT|dpg}|jt|||d\}}}}|r|S||}|r(|r(|SdS)NrDrB)r/r[listrhrP)r$r(rD_rXrgr%r%r&_extract_dialed_extensions  z*CallStateMachine._extract_dialed_extensioncs|dpg}jt||d\}}}}|rdSjs dS}|s(dSd|vr9tfdd|Ds9dS|}|rF||krFdSdS)NrDrBTFrc3s|] }|VqdSr)r<rP)rGar$r%r& s z4CallStateMachine._inbound_allowed..)r/r[rirhr"anyrk)r$r(rDrjrY registereddialedr%rmr&_inbound_alloweds&  z!CallStateMachine._inbound_allowedcallcs|jdkr td|jdvrtd|jr|jdnd}|s#td||IdH|||IdH|j|IdHd|_||IdHdS)NrzNo es una llamada entrante)ringinguLa llamada no está sonandorzCanal entrante no disponibleanswered) rU ValueErrorstatus channel_ids_ensure_bridge_add_to_call_bridgeranswer_attach_webrtc_or_agent)r$rs channel_idr%r%r&answer_inbound_calls  z$CallStateMachine.answer_inbound_callr}c s|js|jr||IdHdSz |j|IdHWnty;}ztd||j|d|_ WYd}~dSd}~wwd|_ t d|j||j z| |IdH| ||IdHWntyz}ztd|j|d|_ WYd}~dSd}~ww d|_|jjr|jrz|j||j|jIdHd|_t d|jWnty}ztd |j|WYd}~dSd}~ww||IdHdS) zFContesta, reproduce el documento y prepara RTP del bot conversacional.Nz+No se pudo contestar canal %s (call=%s): %sfailedruz-Bot auto-answer call=%s canal=%s documento=%sz'No se pudo armar puente bot call=%s: %sTzMedia RTP del bot lista call=%sz-No se pudo adjuntar media del bot call=%s: %s) bot_startedbot_media_ready_maybe_start_botrr{ ExceptionrRerrorcall_idrwinforYryrzdocument_playback_doner bot_enabledrattach_bot_external_mediar)r$rsr}excr%r%r&_confirm_document_playbacksp     z+CallStateMachine._confirm_document_playbacktimeoutcsNt|IdH|jdvs|jrdSd|_td|j||IdHdS)zCSi no llega PlaybackFinished, arranca el bot igual tras el timeout.NendedrTu4Timeout readout documento call=%s — arrancando bot)asynciosleeprwrrRwarningrr)r$rsrr%r%r&_document_playback_timeoutsz+CallStateMachine._document_playback_timeoutc s|jjr|jr dS|jdvrdS|jr|jsdS|js%td|j dSd|_d|_ z ||IdHWdSt yW}zd|_d|_ t d|j |WYd}~dSd}~ww) uHArranca la conversación cuando hay media y terminó el readout del DNI.Nrz0Bot listo call=%s pero sin callback on_bot_readyTstartingFrz/Fallo al iniciar bot conversacional call=%s: %s) rrrrwrrr#rRrr agent_staterrr$rsrr%r%r&r's6  z!CallStateMachine._maybe_start_botcsz|dpi}|d}|sdSd}|jD] }|j|kr#|}nq|s(dSd|_td|j|||IdH|S)NplaybackidTz1Playback documento finalizado call=%s playback=%s) r/rlist_alldocument_playback_idrrRrrr)r$r(r playback_idrs candidater%r%r&r4Bs*  z&CallStateMachine._on_playback_finishedc s4|di}|d}td||sdS|dpg}|jt|||d\}}}}||} |tvr{|||} | r^| dkr^| j s^| j rJ| Std| j || | ||IdH| S| rs| dkrs| j ss| | ||IdH| St d || St||||} | s|r||r||||} | r|r| j s|| _ |j| || j r| dkr|| |IdH| S| jd ks| d vrtd | j || | | ||IdH| Std |||||} |rt|gdd| |d} |j| || |IdH| S|dkr@||s@td||z |j|IdHWnty8} z td| WYd} ~ nd} ~ wwt|dS|g|d|dkrK| n | pV|didd}|r`||d<tdi|} |j| |dkrtd||p|d| j| j | S|d ks| dkr| | ||IdH| S)Nr\rzEvento StasisStart: %srDrBrz(Reintento setup media call=%s channel=%srzCanal ya procesado: %sr)rrrz&Setup media call=%s channel=%s role=%sz5StasisStart nueva llamada (channel=%s args=%s doc=%s)rrt)rxrUrwr_rYz/Llamada entrante rechazada (channel=%s args=%s)z'No se pudo colgar inbound rechazada: %srfrg)rxrUrwr_rz2Llamada entrante para ext %s desde %s (call_id=%s)?r%)r/rRrr[rirh _stasis_roler_resolve_existing_callexternal_media_attachedrYr_setup_call_mediarSaddrA _recover_callr link_channelrrUrer rrrhanguprrdiscardrkr_)r$r(r\r}rDrUrV _dialed_extrYroleexisting caller_numberrsr call_fieldsr%r%r&r0Ys                  z!CallStateMachine._on_stasis_startrc Cs|dpg}d|vsd|vrdnd}|di}|di}|dp,|did }t||g|d |d } |j| td || S) zLRe-vincula Stasis con la llamada originada (p. ej. tras reload del proceso).rDrrrr\r^r_rfrgrt)rrxrUrwr_z?Llamada %s recuperada desde StasisStart (no estaba en registry))r/r rrrRr) r$rr}r(rDrUr\r^r_rsr%r%r&rs&   zCallStateMachine._recover_callcCsz|jr |||}|r |S|j|}|r|S|dpg}|D]}||}|r:||r:|j|}|r:|Sq dS)NrD)r rget_by_channelr/r<rA)r$r}r(found by_channelrDr9r=r%r%r&rs     z'CallStateMachine._resolve_existing_callcCsd|dpg}d|vr dSd|vrdSd|vsd|vrdS|dp#idp(d}|d r0dSdS) NrDrrrrr\r`rE UnicastRTP/)r/ startswith)r$r(rD channel_namer%r%r&rs zCallStateMachine._stasis_rolecsB|jr|jS|jIdH}|d|_td|d|j|jS)Nrz Puente %s creado para llamada %s) bridge_idr create_bridgerRrr)r$rsbridger%r%r&ry s zCallStateMachine._ensure_bridgecsD||IdH}|j||IdHtd|||jt|jdS)Nu7Canal %s añadido al puente %s (llamada %s, %d canales))ryr add_to_bridgerRrrrTrx)r$rsr}rr%r%r&rzsz$CallStateMachine._add_to_call_bridgecCs"|jr |}|r |S|jjpdSr)r!ragent_endpoint)r$endpointr%r%r&_resolve_agent_endpoints  z(CallStateMachine._resolve_agent_endpointc s|}|r |jr dSd|_z)|jj||jjd|jdgdIdH}|j||dt d|d||jWdSt yV}zd|_t d|j|WYd}~dSd}~ww)NTr) caller_id use_stasisapp_argsru-Pata agente originada: %s → %s (llamada %s)Fz+No se pudo originar pata agente para %s: %s) ragent_leg_originatedroriginate_channelroutbound_caller_idrrrrRrrr)r$rsrr\rr%r%r&_originate_agent_leg$s8   z%CallStateMachine._originate_agent_legc s|jjr7|jr7|js7z|j||j|jIdHWdSty6}zt d|j |WYd}~dSd}~ww| |IdHdS)Nu externalMedia falló para %s: %s) rwebrtc_enabledrrattach_external_mediarrrrRrrrrr%r%r&r|?s,   z(CallStateMachine._attach_webrtc_or_agentc s||}|di}|dkrGz|||IdHd|_td|j|jWdStyF}zt d|j|d|_ WYd}~dSd}~ww|dkrvz |||IdHWdStyu}zt d |j|d|_ WYd}~dSd}~ww|j r|jr|jrdSd|_ |j j rtd |jt|z|j|IdH}td t|Wnty}z td ||WYd}~nd}~wwz |j|IdHWnty}ztd ||j|WYd}~nd}~wwz |||IdHWnty}zt d|j|d|_ WYd}~dSd}~ww|j dkrd|_ ||IdH|j js7|s7td|j|d|j jrmz|j|IdH}td||dWdStyl}z t d|WYd}~dSd}~wwdS)z:Puente: destino + WebRTC (externalMedia) o AGENT_ENDPOINT.r\rNTuNexternalMedia en puente %s — audio navegador↔PJSIP habilitado (llamada %s)u9Fallo al añadir externalMedia al puente (llamada %s): %srru2Fallo al añadir agente al puente (llamada %s): %sz*StasisStart outbound call_id=%s channel=%szCanal en vivo (ARI): %sz!No se pudo consultar canal %s: %szAnswer canal %s (call=%s): %su3Fallo al añadir cliente al puente (llamada %s): %srtruu`Llamada %s: solo hay 1 canal en el puente (%s). Activá WEBRTC_ENABLED o definí AGENT_ENDPOINT.r`u7Reproducción de prueba iniciada en %s (playback_id=%s)rz*No se pudo reproducir sonido de prueba: %s)rr/rzrrRrrrrrrwoutbound_stasis_setupr ari_debugrr get_channelrr{r|rrari_debug_play_sound play_sound) r$rsr}r(rr\rliverr%r%r&rSs        z"CallStateMachine._setup_call_mediacCsT|sdS|||}|r|S|jD]}||jvr|S||jkr'|SqdSr)rrrrxexternal_media_channel_id)r$r}r(rsrr%r%r&_find_call_for_channels   z'CallStateMachine._find_call_for_channelcCsT|dpd}|drdS|d}|sdS||jkrdS|dr%dS||jvS)Nr`rEzPJSIP/TrFr)r/rrrx)r$r\rsr`r}r%r%r&_is_customer_channels     z%CallStateMachine._is_customer_channelreasoncCs<|jdvrdS|r|js||_|dtd|j|dS)NrrzLlamada %s finalizada (%s))rwdisconnect_reasonfinalizerRrr)r$rsrr%r%r& _end_calls   zCallStateMachine._end_callcsD|di}|d}|||}|r |||r ||d|S)Nr\rzStasisEnd cliente)r/rrr)r$r(r\r}rsr%r%r&r1s    zCallStateMachine._on_stasis_endcs~|di}|d}|d}|sdS|j|}|sdS|dkr3|jdvr3d|_|js1d|_|S|dkr=|jd kr= |S) Nr\rstateUp)rtrutalkingspeakingRingingrt)r/rrrwrYr)r$r(r\r}rrsr%r%r&r2s"    z)CallStateMachine._on_channel_state_changecs|di}|drt||}|sdS|jvr-fdd|jD|_|jkr8d|_d|_|||rJ||d|d|S)Nr\rcrFr%r%)rGcr}r%r&rKrLz:CallStateMachine._on_channel_destroyed..Fzcanal destruido r`) r/rrrrxrrrr)r$r(r\rsr%rr&r3s        z&CallStateMachine._on_channel_destroyed)NNNNNN)2__name__ __module__ __qualname__rr r r CallResolverrr: BotReadyFnr'dictrr r8 staticmethodr<r>rArituplerr[rerhrkrrr~rfloatrrr4r0rrrryrzrrr|rrrrr1r2r3r%r%r%r&rs   6   Y v        &l  "r)'loggingrercollections.abcrrtypingr ari.clientr ari.debug_logrr calls.modelsrr calls.registryr configr r media.managerr getLoggerrrRsetrr:__annotations__ frozensetrOcompile IGNORECASEr?rrrrr%r%r%r&s*