o ^ji-@sUddlmZmZddlmZmZddlZddlmZiZee ee effe d<ddl Z iZ ee e j fe d<de ddfd d Zd@de d e d eddfddZiZee ee effe d<        dAde d e dededBde dBde dBde de dBde dBde de dBdee effddZde ddfddZdBd e dBdee ee efffd d!ZdBde d"e d#e dBddfd$d%Z &dCd'e dede dee effd(d)ZdBd'e dedBddfd*d+Zd'e dee efdBfd,d-Zdee ee efffd.d/Zd'e d"e defd0d1Zdee ee efffd2d3Zdeddfd4d5Zdeddfd6d7ZdBd8ed9e dBddfd:d;Z!de d8eddfdd?Z#dS)D)datetimetimezone)DictAnyN) WebSocketactive_sessionsgrace_period_timers session_idreturncCs,t|d}|r|s|dSdSdSN)rpopdonecancel)r timer_taskr6/web/html/dev/chatbot/backend/app/services/presence.pycancel_grace_period_timers   r>@site_id grace_secondscs2tfdd}t|}|t<dS)Nc s^zztIdHtvr'dtd<dtd<dtd<dtd<ddlm}|}|durydd lm}m}t t j }||dIdH||dIdH|d  d id ddd iIdH|d d id dd|diIdHt ddddIdHtWn tjyYn wWtddSWtddStdw)NofflineconnectionStatus ABANDONEDstatusassignedAgentId assignedAtr) get_database) update_session_connection_statusupdate_session_statussessions sessionIdz$set)rr conversations)rr updatedAtvisitor_timeout)typer siteIdrr)asynciosleepr app.databaser#app.repositories.session_repositoryrrrnowrutc update_onebroadcast_to_agentsremove_active_sessionCancelledErrorrr )rdbrrr*rr rrr _timer_corosL          z-start_grace_period_timer.._timer_coro)rr& create_taskr)r rrr2taskrr1rstart_grace_period_timers  ' r5connected_agentsRINGINGonline websocket user_data ip_addresscountry status_valassigned_agent_id assigned_atconnection_statusdisconnected_atc Cs>ttj} |||| | |||pd| |||d } | t|<| S)NUnknown) r r%rrdisconnectedAtuser ipAddressr< connectedAtr9rr)rr*rr+ isoformatr) r rr9r:r;r<r=r>r?r@rAnow_iso session_datarrradd_active_sessionFs  rJcCst||tvr t|=dSdSr )rr)r rrrr.es r.cs0ddtD}rfdd|DS|S)NcSs$i|]\}}|ddvr||qS)r)ENDEDrget.0siddatarrr k$z'get_active_sessions..cs$i|]\}}|dkr||qS)r%rLrNrrrrRmrS)ritems)ractiverrTrget_active_sessionsjsrW new_statusended_bycCs0|tvr|t|d<|r|t|d<dSdSdS)NrendedBy)r)r rXrYrrrupdate_active_session_statusps  r[ AVAILABLEagent_idcCs>|sd}ttj}|||dvr|nd|d}|t|<|S)z Registers an agent WebSocket connection in the multi-agent presence pool. Safely replaces stale connections if the same agent reconnects. agent_1r\BUSYOFFLINEr\)agentIdr9rrF)rr*rr+rGr6)r]r9r=rH agent_inforrradd_agent_connectionys rdcCs8|tvr|durt|d}||krdSt|=dSdS)z Removes an agent connection. If a websocket is provided, only removes if it matches the currently registered active socket for that agent identity. Nr9r6rM)r]r9 active_wsrrrremove_agent_connections rgcCs t|S)z8Returns presence info for a specific connected agent ID.rer]rrrget_agent_connections ricCstS)z1Returns a dict of all currently connected agents.r6rrrrget_connected_agentssrkcCs(|dvrdS|tvr|t|d<dSdS)zIUpdates presence status (AVAILABLE, BUSY, OFFLINE) for a connected agent.r_FrTrj)r]rXrrrset_agent_statuss  rlcCsddtDS)zAReturns all connected agents whose status is currently AVAILABLE.cSs$i|]\}}|ddkr||qS)rr\rLrOaidrQrrrrRrSz(get_available_agents..)r6rUrrrrget_available_agentssrocCtd|dSNr^)rdr9rrradd_agent_websocketrscCrprq)rgrrrrrremove_agent_websocketrtru event_datatarget_agent_idsc sg}t|}tt}|durt|fdd|D}|D]&\}}|d}|rGz ||IdHWq!tyF| ||fYq!wq!|D] \}}t ||qJdS)z Broadcasts a JSON event payload to connected agent dashboards in the pool. Optionally filters by target_agent_ids. Ncs g|] \}}|vr||fqSrrrm target_setrr s z'broadcast_to_agents..r9) jsondumpslistr6rUsetrM send_text Exceptionappendrg)rvrw to_remove message_stragents_to_notifyrnrQwsrrxrr-s&       r-c stdtd||tvr`t|d}|rZtdtdtdtdtj|dd z|t|Id HWd StyY}ztd |WYd }~d Sd }~wwtd d Std |dttdd S)zHSends a JSON event payload to a specific visitor's WebSocket connection.z0[VIDEO][TRACE][BACKEND] resolving visitor socketz$[VIDEO][TRACE][BACKEND] sessionId = r9z3[VIDEO][TRACE][BACKEND] visitor socket found = truez+[VIDEO][TRACE][BACKEND] socket state = openz([VIDEO][TRACE][BACKEND] forwarding eventz,[VIDEO][TRACE][BACKEND] sending to visitor: )indentNz9[VIDEO][TRACE][BACKEND] Error sending to visitor socket: z[[VIDEO][TRACE][BACKEND] visitor socket found = false (websocket is None in active_sessions)z6[VIDEO][TRACE][BACKEND] visitor socket found = false (z not in active_sessions: )) printrrMr{r|rrr}keys)r rvrerrrnotify_visitor_sockets& "rc st|sdSddlm}i}|durNz |d|dd}|2z3dHW}|dd||d <q6WntyM}ztd |d |WYd}~nd}~wwttD]\}}|d dkrn|d |krn|dd||<qT|D]D\}} |dur||||dIdHnd} | rt |d|tvrdt|d <dt|d<t |ddddIdHt d|| dddIdHqsdS)a. Handles agent WebSocket disconnection. 1. Finds active IN_CHAT sessions assigned to that agent in DB and in memory. 2. Requeues each IN_CHAT session back to RINGING state atomically via requeue_session(db, sid, agent_id). 3. Notifies visitor socket and remaining connected agents. Nr)requeue_sessionrIN_CHAT)rrr%r z/Error querying sessions for disconnected agent z: rrrhTr7rstatus_changedzAYour agent disconnected. We're reconnecting you to another agent.)r$rmessagevisitor_requeuedagent_disconnected)r$r r%rreason) r)rfindrMrrr}rrUr[rr-) r0r]rsessions_to_requeuecursordocrrPsdatar requeued_docrrrhandle_agent_disconnectsP      r)r)NNNr7NNr8Nr )r\)$rrtypingrrr{fastapirrstr__annotations__r&rTaskrfloatr5r6dictrJr.rWr[rdrgrirkboolrlrorsrur}r-rrrrrrs /    (