
    Ti                         U d dl Z d dlmZ  e j                  dd      Zi Zeeef   ed<   ddedefd	Z	defd
Z
d Zd Zy)    N)Dict*asgi)cors_allowed_origins
async_modeactive_connections	socket_id	user_infoc                 Z    |xs i t         | <   t        d|  dt        t                       y )NzNew connection added: , total connections: r   printlen)r	   r
   s     R/home/jay/workspace/tools/ai-image-gen/jaaz-app/server/services/websocket_state.pyadd_connectionr      s/    $-Oy!	"9+-B3GYCZB[
\]    c                 `    | t         v r&t         | = t        d|  dt        t                       y y )NzConnection removed: r   r   )r	   s    r   remove_connectionr      s6    &&y)$YK/DSI[E\D]^_ 'r   c                  <    t        t        j                               S N)listr   keys r   r   get_all_socket_idsr      s    "'')**r   c                       t        t              S r   )r   r   r   r   r   get_connection_countr      s    !""r   r   )socketiotypingr   AsyncServersior   strdict__annotations__r   r   r   r   r   r   r   <module>r$      sf     h
 ') DdO (^c ^d ^` `
+#r   