
    pi 3                       U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
Z
d dlZd dlmZmZ d dlmZ d dlmZ d dlmZmZ d dlZd dlmZ d dlmZ er1d d	lmZ d d
lmZ d dlm Z  d dl!m"Z" d dl#m$Z$ eez  e$z  e z  e"z  Z%de&d<   ejN                  ejP                  fZ)e	j                  dk(  re)ejT                  fz  Z) ejV                  d      Z, G d d      Z- G d d      Z.y)    )annotationsN)	GeneratorSequence)
formatdate)	FrameType)TYPE_CHECKING	TypeAlias)asyncio_run)Config)H11Protocol)HttpToolsProtocol)WebSocketProtocol)WebSocketsSansIOProtocol)
WSProtocolr	   	Protocolswin32zuvicorn.errorc                      e Zd ZdZddZy)ServerStatezP
    Shared servers state that is available between all protocol instances.
    c                \    d| _         t               | _        t               | _        g | _        y )Nr   )total_requestssetconnectionstasksdefault_headers)selfs    P/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/uvicorn/server.py__init__zServerState.__init__/   s%    +.5.1e
:<    NreturnNone)__name__
__module____qualname____doc__r    r   r   r   r   *   s    =r   r   c                      e Zd ZddZdddZdddZdddZdddZddZddZ	dd	Z
ddd
ZddZej                  dd       ZddZy)Serverc                v    || _         t               | _        d| _        d| _        d| _        d| _        g | _        y )NFg        )configr   server_statestartedshould_exit
force_exitlast_notified_captured_signals)r   r*   s     r   r   zServer.__init__7   s9    'M  ,.r   Nc                l    t        | j                  |      | j                  j                               S )Nsockets)loop_factory)r
   server*   get_loop_factoryr   r3   s     r   runz
Server.runB   s'    4::g:6T[[EaEaEcddr   c                   K   | j                         5  | j                  |       d {    d d d        y 7 # 1 sw Y   y xY wwN)capture_signals_server7   s     r   r5   zServer.serveE   sA     !!# 	'++g&&&	' 	'&	' 	's$   A868	A8AAc                R  K   t        j                         }| j                  }|j                  s|j	                          |j                  |      | _        d}dt        j                  dd      z   dz   }t        j                  ||d|i       | j                  |	       d {    | j                  ry | j                          d {    | j                  |	       d {    d
}dt        j                  dd      z   dz   }t        j                  ||d|i       y 7 }7 Z7 Bw)NzStarted server process [%d]zStarted server process [z%dcyan)fg]color_messageextrar2   zFinished server process [%d]zFinished server process [)osgetpidr*   loadedloadlifespan_classlifespanclickstyleloggerinfostartupr-   	main_loopshutdown)r   r3   
process_idr*   messagerA   s         r   r<   zServer._serveI   s     YY[
}}KKM--f5/2U[[&5QQTWWGZ/OPll7l+++nnmmGm,,,03ekk$66RRUXXGZ/OP 	, 	,s6   B!D'#D!$$D'D#	D'"D%#?D'#D'%D'c                   K    j                   j                          d {     j                   j                  rd _        y  j                  	 d
	 	 	 d fd}t	        j
                         }|	 	 	 	 dd}g  _        |D ]~  }t        j                         dk(  }j                  dkD  r
|r ||      }|j                  ||j                  j                         d {   } j                  j                  |        |}nj                  t        j                   j                  t        j"                  t        j$                        }|j                  ||j                  j                         d {   }|j&                  J |j&                  }|g _        nGj(                  d}	t*        j,                  j/                  j(                        r)t+        j0                  j(                        j2                  }	|j5                  |j(                  j                  j                         d {   }t+        j6                  j(                  |	       |j&                  J |j&                  }|g _        ni	 |j                  |j8                  j:                  j                  j                  	       d {   }j&                  J |j&                  }|g _        | jI                  |       d _%        y 	 d _%        y 7 7 27 7 7 U# t<        $ rW}
t>        jA                  |
        j                   jC                          d {  7   tE        jF                  d       Y d }
~
d }
~
ww xY ww)NTc                j    j                  j                  j                  j                  |       S )N)r*   r+   	app_state_loop)http_protocol_classr+   rI   state)rV   r*   r   s    r   create_protocolz'Server.startup.<locals>.create_protocolh   s9     --!..----	 .  r   c                d    ddl m} | j                  t        j                               } ||      S )Nr   )	fromshare)socketr[   sharerD   rE   )sockr[   	sock_datas      r   _share_socketz%Server.startup.<locals>._share_sockety   s&    
 - JJryy{3	 ++r   Windows   )r^   sslbacklogi  )pathrc   rd   )hostportrc   rd   r:   )rV   z asyncio.AbstractEventLoop | Noner    zasyncio.Protocol)r^   socket.SocketTyper    rh   )&rI   rN   r-   r*   asyncioget_running_loopserversplatformsystemworkerscreate_serverrc   rd   appendfdr\   fromfdAF_UNIXSOCK_STREAMr3   udsrD   re   existsstatst_modecreate_unix_serverchmodrf   rg   OSErrorrL   errorrP   sysexit_log_started_messager,   )r   r3   rY   loopr`   r^   
is_windowsserver	listeners	uds_permsexcr*   s   `          @r   rN   zServer.startup`   s    mm##%%%==$$#D 7;	3		 '') ,',", >@DL ,%__.);
>>A%*(.D#11/RXR\R\flftft1uu##F+,  IYY"==FNNF<N<NOD--oDfjjbhbpbp-qqF>>---I"8DLZZ#Iww~~fjj)GGFJJ/77	22fjjfjj&.. 3  F HHVZZ+>>---I"8DL#11#

"NN  2    >>---I"8DL?%%i0  u 	&N v r  S!mm,,...s   NLC N#L$BN>L?B>N=L>ANAL LL ANNNNL 	M?(2M:MM:5N:M??Nc                   | j                   }|j                  *|d   }t        j                  d|j	                                y |j
                  !t        j                  d|j
                         y d}|j                  dn|j                  }d|v rd}|j                  }|dk(  r|d   j	                         d   }|j                  rd	nd
}d| d}dt        j                  |d      z   dz   }	t        j                  ||||d|	i       y )Nr   z3Uvicorn running on socket %s (Press CTRL+C to quit)z8Uvicorn running on unix socket %s (Press CTRL+C to quit)z
%s://%s:%dz0.0.0.0:z%s://[%s]:%drb   httpshttpzUvicorn running on z (Press CTRL+C to quit)T)boldrA   rB   )r*   rq   rL   rM   getsocknameru   rf   rg   rc   rJ   rK   )
r   r   r*   r^   addr_formatrf   rg   protocol_namerR   rA   s
             r   r   zServer._log_started_message   s   99 Q<DKKE  "
 ZZ#KKRTZT^T^_ 'K & 39Dd{,;;Dqy |//1!4'-zzGvM+K=8OPG1EKKRV4WWZssMKK&6  r   c                   K   d}| j                  |       d {   }|sD|dz  }|dz  }t        j                  d       d {    | j                  |       d {   }|sCy y 7 K7 $7 w)Nr   rb   i / 皙?)on_tickri   sleep)r   counterr-   s      r   rO   zServer.main_loop   si      LL11qLG&G--$$$ $W 55K	  2 %5s9   A+A%(A+A'A+A)A+#A+'A+)A+c                  K   |dz  dk(  rt        j                          }t        |d      j                         }| j                  j                  rd|fg}ng }|| j                  j
                  z   | j                  _        | j                  j                  O|| j                  z
  | j                  j                  kD  r)|| _	        | j                  j                          d {    | j                  ry| j                  j                  }|3| j                  j                  |k\  rt        j                  d| d       yy7 ]w)	N
   r   T)usegmts   datezMaximum request limit of z exceeded. Terminating process.F)timer   encoder*   date_headerencoded_headersr+   r   callback_notifyr/   timeout_notifyr-   limit_max_requestsr   rL   warning)r   r   current_timecurrent_dater   max_requestss         r   r   zServer.on_tick   s    R<199;L%l4@GGIL{{&& '67 0;dkk>Y>Y0YD- {{**6$"4"44t{{7Q7QQ)5D&++55777 {{55#(9(9(H(HL(XNN6|nDcde 8s   CE  D>!AE c                0  K   t         j                  d       | j                  D ]  }|j                           |xs g D ]  }|j                           t	        | j
                  j                        D ]  }|j                           t        j                  d       d {    	 t        j                  | j                         | j                  j                         d {    | j&                  s#| j(                  j                          d {    y y 7 v7 6# t        j                  $ rc t         j                  dt!        | j
                  j"                               | j
                  j"                  D ]  }|j%                  d        Y w xY w7 w)NzShutting downr   )timeoutz=Cancel %s running task(s), timeout graceful shutdown exceededz2Task cancelled, timeout graceful shutdown exceeded)msg)rL   rM   rk   closelistr+   r   rP   ri   r   wait_for_wait_tasks_to_completer*   timeout_graceful_shutdownTimeoutErrorr|   lenr   cancelr.   rI   )r   r3   r   r^   
connectionts         r   rP   zServer.shutdown  sa    O$ ll 	FLLN	Mr 	DJJL	 t00<<= 	"J!	"mmC   	S"",,.==   --((*** # 	! ## 	SLLOD%%++, &&,, SQRS	S +sU   BF D!F&<D "D#D ')FFFD A3FFFFc                  K   | j                   j                  r| j                  syd}t        j	                  |       | j                   j                  rL| j                  s@t        j                  d       d {    | j                   j                  r| j                  s@| j                   j                  r| j                  syd}t        j	                  |       | j                   j                  rL| j                  s@t        j                  d       d {    | j                   j                  r| j                  s@| j                  D ]  }|j                          d {     y 7 7 S7 w)Nz8Waiting for connections to close. (CTRL+C to force quit)r   z@Waiting for background tasks to complete. (CTRL+C to force quit))
r+   r   r.   rL   rM   ri   r   r   rk   wait_closed)r   r   r   s      r   r   zServer._wait_tasks_to_complete'  s    ((LCKK##//mmC((( ##// ""4??TCKK##))$//mmC((( ##))$// ll 	'F$$&&&	' ) ) 'sC   A4E)6E#7&E)A3E)E%&E)9"E)E'E)%E)'E)c              #     K   t        j                         t        j                         urd  y t        D ci c]#  }|t	        j                  || j
                        % }}	 d  |j                         D ]  \  }}t	        j                  ||        	 t        | j                        D ]  }t	        j                  |        y c c}w # |j                         D ]  \  }}t	        j                  ||        w xY wwr:   )
	threadingcurrent_threadmain_threadHANDLED_SIGNALSsignalhandle_exititemsreversedr0   raise_signal)r   sigoriginal_handlershandlercaptured_signals        r   r;   zServer.capture_signals9  s      ##%Y-B-B-DD Sbb3S&--T5E5E"FFbb	, 1 7 7 9 ,Wc7+,
  ((>(>? 	1O0	1 c !2 7 7 9 ,Wc7+,s)   6C>(C C>#C 'A$C>0C;;C>c                    | j                   j                  |       | j                  r|t        j                  k(  rd| _        y d| _        y )NT)r0   rp   r-   r   SIGINTr.   )r   r   frames      r   r   zServer.handle_exitM  s9    %%c*v}} 4"DO#Dr   )r*   r   r    r!   r:   )r3   zlist[socket.socket] | Noner    r!   )r   zSequence[socket.SocketType]r    r!   r   )r   intr    bool)r    zGenerator[None, None, None])r   r   r   zFrameType | Noner    r!   )r"   r#   r$   r   r8   r5   r<   rN   r   rO   r   rP   r   
contextlibcontextmanagerr;   r   r&   r   r   r(   r(   6   sY    	/e'Q.[z!F6<+@'$ 1 1&$r   r(   )/
__future__r   ri   r   loggingrD   rl   r   r\   r}   r   r   collections.abcr   r   email.utilsr   typesr   typingr   r	   rJ   uvicorn._compatr
   uvicorn.configr   uvicorn.protocols.http.h11_implr   %uvicorn.protocols.http.httptools_implr   ,uvicorn.protocols.websockets.websockets_implr   3uvicorn.protocols.websockets.websockets_sansio_implr   )uvicorn.protocols.websockets.wsproto_implr   r   __annotations__r   SIGTERMr   SIGBREAK	getLoggerrL   r   r(   r&   r   r   <module>r      s    "    	    
   / "  +  ' !;GN\D&)::ZGJ[[^vvIyv MM
NN <<7))O			?	+	= 	=\$ \$r   