o g@sUddlmZmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZeZeZiZe eefed<edeejeejfd ed ed e fd d ZdS)) APIRouter WebSocketWebSocketDisconnectDepends)deps)message_service)Dict)Session) setup_loggingactive_connectionsz/ws/chat/{friend_id} websocket friend_iddbcstd|jd||IdH|t|j<z= |IdH}tj||j||d}td|j|d|IdH|tvrVt|}||j d|IdHqt ys|jtvrft|j=td |jd YdSw) z Supports real-time chat between the currently logged-in user and the user with friend_id. This example primarily functions as an echo for messages. z'WebSocket connection attempt from user z to friend NT)r sender_id receiver_idcontentzMessage saved with id zYou: z: zClient #z disconnected) loggerinfoidacceptr receive_textrcreate_message send_text first_namer)r r r current_userdatamessageZ friend_wsr$/app/app/api/v1/endpoints/ws_chat.pywebsocket_chats0    rN)fastapirrrrZapp.apirZ app.servicesrtypingrsqlalchemy.ormr app.core.logging_configr routerrr int__annotations__r get_dbget_current_ws_userrrrrrs&