i! .UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z mZeejejejejeeddeeddhdhz ZejeZejdd pd d vZeGd d e Zejd dZ de!d<ddZ"ddZ#ddZ$GddZ%GddZ&dS)uCTransport abstraction for the tui_gateway JSON-RPC server. Historically the gateway wrote every JSON frame directly to real stdout. This module decouples the I/O sink from the handler logic so the same dispatcher can be driven over stdio (``tui_gateway.entry``) or WebSocket (``tui_gateway.ws``) without duplicating code. A :class:`Transport` is anything that can accept a JSON-serialisable dict and forward it to its peer. The active transport for the current request is tracked in a :class:`contextvars.ContextVar` so handlers — including those dispatched onto the worker pool — route their writes to the right peer. Backward compatibility ---------------------- ``tui_gateway.server.write_json`` still works without any transport bound. When nothing is on the contextvar and no session-level transport is found, it falls back to the module-level :class:`StdioTransport`, which wraps the original ``_real_stdout`` + ``_stdout_lock`` pair. Tests that monkey-patch ``server._real_stdout`` continue to work because the stdio transport resolves the stream lazily through a callback. ) annotationsN)AnyCallableOptionalProtocolruntime_checkable WSAECONNRESET WSAESHUTDOWNHERMES_TUI_GATEWAY_NO_FLUSH>1onyestruec"eZdZdZd dZd dZd S) Transportz-Minimal interface every transport implements.objdictreturnboolcdS)z"!!!!!%788MQTUVQWQWDG F9G+ F94'E,G, F99*F4#G4F99GG G cdSrBrr!s rr"zStdioTransport.closestrN)r=r>r?r@rrr#r$r%r&r'r( __slots__rCrr"rrrr9r9dsf,IBBBBHrr9c.eZdZdZdZddZdd Zdd ZdS) TeeTransportu]Mirrors writes to one primary plus N best-effort secondaries. The primary's return value (and exceptions) determine the result — secondaries swallow failures so a wedged sidecar never stalls the main IO path. Used by the PTY child so every dispatcher emit lands on stdio (Ink) AND on a back-WS feeding the dashboard sidebar. _primary _secondariesprimary 'Transport' secondariesrrc"||_||_dSrBr])rr`rbs rrCzTeeTransport.__init__s 'rrrrc|j|}|jD](} ||#t$rY%wxYw|SrB)r^rr_ Exception)rroksecs rrzTeeTransport.writesd ]  % %$  C  #     s; AAc |j|jD]'} |#t$rY$wxYwdS#|jD]'} |#t$rY$wxYwwxYwrB)r^r"r_re)rrgs rr"zTeeTransport.closes  M   ! ! !(  IIKKKK D  t(  IIKKKK D s>A : AA A=A,+A=, A9 6A=8A9 9A=N)r`rarbrarrr#r$rYrrrr\r\sa-I((((rr\)rr,)r0r,r$)'r( __future__r contextvarsrOrGloggingos threadingtypingrrrrr frozensetEPIPE ECONNRESETEBADF ESHUTDOWNgetattrrP getLoggerr%rQenvironr.striplowerrSr ContextVarr+__annotations__r/r3r7r9r\rrrr{sL,#"""""  GGGGGGGGGGGGGG I K  K O GE?B'' GE>2&& D     8 $ $*..!>CCIrPPRRXXZZ_========K"$$$$ ---- $$$$ SSSSSSSSl!!!!!!!!!!r