|+jy/dZddlmZddlZddlZddlZddlZddlmZddl m Z ej e Z dZdZ ddlmZn #e$reZYnwxYwGd d ZddZddZddZdS)aWebSocket transport for the tui_gateway JSON-RPC server. Reuses :func:`tui_gateway.server.dispatch` verbatim so every RPC method, every slash command, every approval/clarify/sudo flow, and every agent event flows through the same handlers whether the client is Ink over stdio or an iOS / web client over WebSocket. Wire protocol ------------- Identical to stdio: newline-delimited JSON-RPC in both directions. The server emits a ``gateway.ready`` event immediately after connection accept, then echoes responses/events for inbound requests. No framing differences. Mounting -------- from fastapi import WebSocket from tui_gateway.ws import handle_ws @app.websocket("/api/ws") async def ws(ws: WebSocket): await handle_ws(ws) ) annotationsN)Any)serverg$@)WebSocketDisconnectc@eZdZdZdddd ZddZddZddZddZdS) WSTransportaPer-connection WS transport. ``write`` is safe to call from any thread *other than* the event loop thread that owns the socket. Pool workers (the only real caller) run in their own threads, so marshalling onto the loop via :func:`asyncio.run_coroutine_threadsafe` + ``future.result()`` is correct and deadlock-free there. When called from the loop thread itself (e.g. by ``handle_ws`` for an inline response) the same call would deadlock: we'd schedule work onto the loop we're currently blocking. We detect that case and fire-and- forget instead. Callers that need to know when the bytes are on the wire should use :meth:`write_async` from the loop thread. unknownpeerwsrloopasyncio.AbstractEventLoopr strreturnNonec>||_||_||_d|_dS)NF)_ws_loop_peer_closed)selfr rr s -/usr/local/lib/hermes-agent/tui_gateway/ws.py__init__zWSTransport.__init__Bs$   objdictboolcV|jrdStj|d} tj|ju}n#t $rd}YnwxYw|r/|j||dS ddl m }||||j}| d|_dS| t|j S#t$rF}d|_td|jt#|j|Yd}~dSd}~wwxYw)NF ensure_asciiTr)safe_schedule_threadsafe)timeoutz.ws write failed peer=%s error_type=%s error=%s)rjsondumpsasyncioget_running_loopr RuntimeError create_task _safe_sendagent.async_utilsr"result_WS_WRITE_TIMEOUT_S Exception_logwarningrtype__name__)rrlineon_loopr"futexcs rwritezWSTransport.writeNs^ < 5z#E222 .00DJ>GG   GGG    J " "4??4#8#8 9 9 94  B B B B B B**4??4+@+@$*MMC{# u JJ2J 3 3 3|# #   DL LL@ DII.   55555  s-< A  A 3C5"C D(";D##D(cK|jrdS|tj|dd{V|j S)zGSend from the owning event loop. Awaits until the frame is on the wire.Fr N)rr*r$r%)rrs r write_asynczWSTransport.write_asyncnsS < 5oodj5AAABBBBBBBBB<rr3cK |j|d{VdS#t$rF}d|_td|jt|j|Yd}~dSd}~wwxYw)NTz-ws send failed peer=%s error_type=%s error=%s) r send_textr.rr/r0rr1r2)rr3r6s rr*zWSTransport._safe_sendus ($$T** * * * * * * * * *   DL LL? DII.          s & A6;A11A6cd|_dS)NT)r)rs rclosezWSTransport.closes  rN)r rrrr rrr)rrrr)r3rrr)rr) r2 __module__ __qualname____doc__rr7r9r*r=rrr r 2s  (      @    rr r rrrct|dd}|dSt|ddpd}t|dd}||d|n|S)z?Return ``host:port`` when available, else a stable placeholder.clientNr hostport:)getattr)r rCrDrEs r_ws_peer_labelrHsb R4 ( (F ~y 664 ( ( 5ID 664 ( (D#/d  T   T9rrc t|ddpi}|dpidpt|dd}||dnd}|-|tjtjddSdS#t$r&}t d|Yd}~dSd}~wwxYw)u}Disable Nagle so streamed JSON-RPC frames go out individually. Without it the kernel coalesces the small per-token frames, so a burst after the model's think-pause lands on the client in one tick and no client-side smoothing can recover the cadence. GUI/WS only; chat platforms don't hit this path. Best-effort — skip silently if the socket isn't reachable. scopeN extensions transportsocketzws TCP_NODELAY skip: %s) rGgetget_extra_info setsockoptrM IPPROTO_TCP TCP_NODELAYr.r/debug)r rJrLsockr6s r_disable_naglerVs3GT**0bYY|,,277 DDfPRT_aeHfHf 5>5Jy''111PT   OOF.0BA F F F F F   333 ,c2222222223sBB C $CC c N Kt|}d}d}d}d}d}d} |d{Vd}t|td|t |t j|}|ddd d tj id d d{V}|sd }|dz }t d| d} d} |k|  t j tj|dd{V\} } n+#t$rtd|YnwxYw | d{Vn3#t$r&} td|| Yd} ~ nd} ~ wwxYwtd||||||| | dS |d{V} nh#t&$r2} dt)| dddt)| ddd}Yd} ~ nAd} ~ wt$r!d}td|YnwxYw| } | s|dz } t-j| }n#t,j$r} |dz }td||| | dt4|dddd dd!d{V}|s(d"}|dz }td#|Yd} ~ nSYd} ~ Hd} ~ wwxYwt7|t8r|d$nd}t7|t8r|d%nd} t j tj||d{V}n#t$rt|dz }td&||||dd'd(d ||ndd!d{V}|s&d)}|dz }td*|||YnKYMwxYw|@||d{Vs%d+}|dz }td,|||nd} d} |k|  t j tj|dd{V\} } n+#t$rtd|YnwxYw | d{Vn3#t$r&} td|| Yd} ~ nd} ~ wwxYwtd||||||| | dS#d} d} |k|  t j tj|dd{V\} } n+#t$rtd|YnwxYw | d{Vn3#t$r&} td|| Yd} ~ nd} ~ wwxYwtd||||||| | wxYw)-zFRun one WebSocket session. Wire-compatible with ``tui_gateway.entry``.Nr not_connected connectedzws accepted peer=%sr z2.0eventz gateway.readyskin)r1payload)jsonrpcmethodparamsready_send_failedrNz"ws ready frame send failed peer=%s ws_disconnect) end_reasonz$ws transport teardown failed peer=%sz ws close failed peer=%s error=%szws closed peer=%s reason=%s messages=%d parse_errors=%d dispatch_crashes=%d send_failures=%d reaped_sessions=%d detached_sessions=%dTzclient_disconnect(code=codez,reason=reason)receive_failedzws receive failed peer=%sz3ws parse error peer=%s index=%d error=%s payload=%riDz parse error)rcmessage)r]erroridsend_failed_after_parse_errorz(ws parse-error reply send failed peer=%srir^z)ws dispatch crash peer=%s id=%s method=%sizinternal error send_failed_after_dispatch_crashz;ws dispatch-crash reply send failed peer=%s id=%s method=%ssend_failed_after_responsez/ws response send failed peer=%s id=%s method=%s)rHacceptrVr/infor r&r'r9r resolve_skinrhr= to_thread_close_sessions_for_transportr. exceptionrT receive_text_WebSocketDisconnectrGstripr$loadsJSONDecodeErrorr0_WS_LOG_PAYLOAD_PREVIEW isinstancerrOdispatch)r r rLmessages parse_errorsdispatch_crashes send_failuresdisconnect_reasonready_okreaped_sessionsdetached_sessionsr6rawr3reqokreq_id req_methodresps r handle_wsrs "  D$(IHLM'] iikk' r '...G$<$>$>TJJJ ".. !+ &(;(=(=>             3  Q M JJ;T B B B z  OO    M;B;L8.<<<6666662!2!2  M M MEtLLLLL M F((**         F F F JJ94 E E E E E E E E F  [         qY  OO--------'   >#C66>>%c8T::>>>"    $4!:DAAA  99;;D  MH j&&'   !  I1112 %00#(*0]!K!K"(G%!Q&MLL!KTRRREEEE+ 8'1d&;&;ESWWT]]]F.8d.C.CM***J $.vYOOOOOOOO    A% ?  %00#(*0=M!N!N(.(:ff (J%!Q&MLLU"  E3 4i.C.CD.I.I(I(I(I(I(I(I$@!"  E  sY v  OO    M;B;L8.<<<6666662!2!2  M M MEtLLLLL M F((**         F F F JJ94 E E E E E E E E F  [         ;  OO    M;B;L8.<<<6666662!2!2  M M MEtLLLLL M F((**         F F F JJ94 E E E E E E E E F  [         s<CU>*D))%EEE00 F :FF U G$#U$ I .'HU*I UI  U)I>=U>L A6L ULAU0&NUA8PUPAU7*R""%S  S S)) T3TTX$*VX$%V/,X$.V//X$3W X$ W>W94X$9W>>&X$)r rrr)r rrr)r@ __future__rr&r$loggingrMtypingr tui_gatewayr getLoggerr2r/r-rxstarlette.websocketsrrt ImportErrorr.r rHrVrrArrrsI.#"""""  w"" %PPPPPPP%%%$%NNNNNNNNb::::3333$g g g g g g sAA  A