:vj >dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZmZmZerddlmZ ddlmZdd Zdd lmZd Zd ZGddZ ddZ!dddZ"GddZ#Gdde#Z$Gdde%Z&Gdde#Z'Gdde#Z(y#e$r dd lmZn #e$reZYnwxYwdZYrwxYw) zMTerminal management for exposing terminals to a web interface using Tornado. ) annotationsN)deque)futures) TYPE_CHECKINGAny Coroutine) TermSocket)PtyProcessUnicodechtjtjtjy)z-A prexec function to set up a signal handler.N)signalSIGPIPESIG_DFLe/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/terminado/management.py preexec_fnrs fnnfnn5r) PtyProcess)IOLoopPYXTERM_zxterm-256colorcxeZdZUdZded<d d dZd dZejfd dZ ejfddZ ddd Z y)PtyWithClientsz%A pty object with associated clients. str | None term_nameNcg|_tgd|_||xsg|d}t t|d<t j d i||_tjdd|j _ y) zInitialize the pty.)maxlen)argvenvcwdNrzutf-8replace)errorsr) clientsr read_bufferrr spawnptyproccodecsgetincrementaldecoderdecoder)selfrrrkwargss r__init__zPtyWithClients.__init__2sn"$ ',R'=syb=  !#-F< (..88  Ev;;GDIV rcdx}}|jD]%}|j\}}|||kr|}|||ks$|}'|dk(s|dk(ry|jj\}}||f||fk7r|jj ||yy)aSet the terminal size to that of the smallest client dimensions. A terminal not using the full space available is much nicer than a terminal trying to use more than the available space, so we keep it sized to the smallest client. i'N)r"sizer% getwinsize setwinsize)r)minrowsmincolsclientrowscolss rresize_to_smallestz!PtyWithClients.resize_to_smallest@s"!'llFJD$D7ND7N # e w%/ \\,,. d $tj tj tjtjg}tj}dfd }jjsy |D]>}j||d{jjr>y|rRjtj|d{tjj Sy7w7+#t$r7|d{7tjj cYSwxYww)zThis forces a child process to terminate. It starts nicely with SIGHUP and SIGINT. If "force" is True then moves onto SIGKILL. This returns True if the child was terminated. This returns False if the child could not be terminated.r:cVtjjjS)z/Sleep to allow the terminal to exit gracefully.)asynciosleepr%delayafterterminater)srrDz'PtyWithClients.terminate..sleepns==!A!AB BrTNF)returnzCoroutine[Any, Any, None])r;r<r SIGINTSIGTERMSIGHUPSIGCONTrcurrentr%isaliver7SIGKILLboolOSError)r)forcesignals_rDr8s` r terminatezPtyWithClients.terminatebs 77d?}}fnn5G}}fnnfmmV^^TG NN  C||##% 4 #g ||++-   &..)g  4 4 6677  4 'MMDLL00223 3  4slB(F , E E E,E-F .,EE'EF EEFE)FF FF )NN)rrrzdict[str, str] | NonerrrGNone)r8intrGrV)r8rWrGrF)rQrOrGrO) __name__ __module__ __qualname____doc____annotations__r+r5r rIr7r?rTrrrrr-s7/ W6,%nn!'#4rrch|jD]\}}||j|d|||<!y)z**mmRD"b'2GAq! HrceZdZdZ d ddZ d ddZddZddZddZdddZ dd Z ddd Z dd Z dd Z dd Zy)TermManagerBasez"Base class for a terminal manager.Ncb||_||_|xsi|_||_t j t |_i|_|#d|_ tjd|_ nd|_ ||_ |5tjd|jj dt d yy) Initialize the manager.NF) max_workersTzSetting z!.ioloop is deprecated and ignored) stacklevel) shell_command server_url term_settings extra_envlogging getLoggerrYlog ptys_by_fd!_blocking_io_executor_is_externalrThreadPoolExecutorblocking_io_executorwarningswarn __class__DeprecationWarning)r)r|r}r~riolooprs rr+zTermManagerBase.__init__s+$*0b"$$X.57  '5:D 2(/(B(Bq(QD %59D 2(>2233TU"  rc tjj}|jj dt |d<d||fz}|r |r |d||fzz }||t dz<t||d<t||d<|jr|j|t dz<|jrt||j|j d i}|rt|tr t|||S) z@Build the environment variables for the process in the terminal.typeTERMz%dx%dz;%dx%d DIMENSIONSCOLUMNSLINESURLr) r;environcopyr~getDEFAULT_TERM_TYPE ENV_PREFIXstrr}rre isinstancedict) r)heightwidth winheightwinwidthr*r dimensionsterm_envs r make_term_envzTermManagerBase.make_term_envsjjoo((,,V5FGF v.  (h %:: :J)3J %&UI6{G ??&*ooC U" # >> S$.. 1::k2.  8T2 S( + rc |jj}|j|d<|j||d}|jdi|}|j dd}t |||S)z?Make a new terminal, return a :class:`PtyWithClients` instance.r|rNr)r~rr|updaterrr)r)r*optionsrrrs r new_terminalzTermManagerBase.new_terminalsp$$))+#'#5#5 v' d  +7+kk%&dC--rc|jj}||j|<tj}|j ||j |jy)zBConnect a terminal to the tornado event loop to read data from it.N)r%rorrrL add_handlerpty_readREAD)r) ptywclientsroloops r start_readingzTermManagerBase.start_readingsG  # #)~~ T]]DII6rc|jj}|jjd||j|=t j j||jjy)zCalled when the pty has closed.zEOF on FD %d; stopping readingN) r%rorinforrrLremove_handlerclose)r)rros ron_eofzTermManagerBase.on_eofs] # #  6; OOB ''+ !!#rct|ds|jjd|y|j|} |j ||j j d}|jj||jD]}|j|y#t$r5|j||jD]}|jYywxYw)z>Called by the event loop when there is pty data ready to read.皙?)rpzSpurious pty_read() on fd %sNi)rsrdebugrpre_pty_read_hookr%readr#appendr" on_pty_readEOFErrorr on_pty_died)r)roeventsrsr2s rrzTermManagerBase.pty_readsR% HHNN92 > oob)  %  " "; /##((/A  # # * *1 -%--""1%. % KK $%--""$. %sA)B%%;C#"C#cy)zQHook before pty read, subclass can patch something into ptywclients when pty_readNr)r)rs rrz!TermManagerBase.pre_pty_read_hookrct)a7Override in a subclass to give a terminal to a new websocket connection The :class:`TermSocket` handler works with zero or one URL components (capturing groups in the URL spec regex). If it receives one, it is passed as the ``url_component`` parameter; otherwise, this is None. )NotImplementedErrorr) url_components r get_terminalzTermManagerBase.get_terminals "!rcy)z=Override this to e.g. kill terminals on client disconnection.Nrr) websockets rclient_disconnectedz#TermManagerBase.client_disconnectedrrcK|jd{|js|jjddyy7/w)zShutdown the manager.NFT)waitcancel_futures)kill_allrrshutdownrFs rrzTermManagerBase.shutdownsAmmo55  % % . .E$ . O6 sAA0AcKg}|jjD]$}|j|jd&|rt j |d{yy7w)Kill all terminals.TrQN)rvaluesrrTrCgather)r)rterms rrzTermManagerBase.kill_all$sZOO**,D GNN4>>>5 6- ..'* * *  *sAA&A$A&)NNNN) r|rr}rr~rrrrrrr)Prr) rrWrrWrrWrrWr*rrGzdict[str, str])r*rrGrrrrGrVN)rorWrrrGrVrrrGr)rrrGrVrU)rYrZr[r\r+rrrrrrrrrrrrrrurus, !$(     "D       @.7 $%$`"LP +rruc:eZdZdZdfd ZdddZdfd ZxZS) SingleTermManagerz9All connections to the websocket share a common terminal.c 2t|di|d|_yrwNr)superr+terminal)r)r*rs rr+zSingleTermManager.__init__1s "6"/3 rc|j0|j|_|j|j|jS)z "Get the singleton terminal.)rrrrs rrzSingleTermManager.get_terminal6s8 ==  --/DM   t}} -}}rcLKt|d{d|_y7 w)zKill the singletone terminal.N)rrrr)rs rrzSingleTermManager.kill_all=s$g     ! $" $)r*rrGrVrrrU)rYrZr[r\r+rr __classcell__rs@rrr.sC4 rrc eZdZdZddZddZy)MaxTerminalsReachedz;An error raised when we exceed the max number of terminals.c||_y)zInitialize the error.N max_terminals)r)rs rr+zMaxTerminalsReached.__init__Fs *rc d|jzS)z'The string representation of the error.z$Cannot create more than %d terminalsrrFs r__str__zMaxTerminalsReached.__str__Js58J8JJJrN)rrWrGrV)rGr)rYrZr[r\r+rrrrrrCsE+Krrc8eZdZdZddfd ZdddZddZxZS) UniqueTermManagerz-Give each websocket a unique terminal to use.c 2t|di|||_yr)rr+rr)rr*rs rr+zUniqueTermManager.__init__Rs "6"*rc|jr7t|j|jk\rt|j|j }|j ||S)z Get a terminal from the manager.)rlenrrrr)r)rrs rrzUniqueTermManager.get_terminalWsS   #doo"6$:L:L"L%d&8&89 9  " 4  rc`|jjd|jrtjdk(rJ|jj |j |jjjy|jjtjyy)z-Send terminal SIGHUP when client disconnects.z-Websocket closed, sending SIGHUP to terminal.r:N) rrrr;r<r7rr%ror?r rJrs rrz%UniqueTermManager.client_disconnected`sy  EF   ww$""'') i0088;;<    % %fmm 4 rr)rz int | Noner*rrGrVr)rr rGrV)rYrZr[r\r+rrrrs@rrrOs7+  5rrceZdZdZd d fd Zd dZdZddZddZe jfddZ dddZ dfd Z dfd ZxZS)NamedTermManagerzBShare terminals between websockets connected to the same endpoint.c @t|di|||_i|_yr)rr+r terminalsrs rr+zNamedTermManager.__init__ps" "6"*46rct|J||jvr|j|S|jr7t|j|jk\rt|j|jj d||j }||_||j|<|j||S)z!Get or create a terminal by name.z$New terminal with specified name: %s) rrrrrrrrr)r)rrs rrzNamedTermManager.get_terminalvs$$$  &>>), ,   #dnn"59K9K"K%d&8&89 9  >) *rc d|vr|dn|j}|jdi|}|jjd|||_||j |<|j |||fS)z3Create a new named terminal with an automatic name.r<z$New terminal with automatic name: %sr)rrrrrrr)r)r*r<rs rnew_named_terminalz#NamedTermManager.new_named_terminalsr!'6!1vf~t7P7P7R t  *6*  rs #  00. ,6"  %X4X4v  M+M+`* K) K55