XBeA6hdZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z  ddl mZdZddlmZd Zd ZGd d Zd ZddededefdZGddZGddeZGdde Z!GddeZ"GddeZ#y#e$r ddlmZn #e$reZYnwxYwdZYwwxYw)zMTerminal management for exposing terminals to a web interface using Tornado. N)deque)futures)PtyProcessUnicodechtjtjtjyN)signalSIGPIPESIG_DFL4lib/python3.12/site-packages/terminado/management.py preexec_fnrs fnnfnn5r ) PtyProcess)IOLoopPYXTERM_zxterm-256colorc^eZdZddZdZej fdZej fdZddZ y) PtyWithClientsNcg|_tgd|_t||xsg|}t t|d<t j di||_tjdd|j_ y) N)maxlen)argvenvcwdrzutf-8replace)errorsr ) clientsr read_bufferdictrrspawnptyproccodecsgetincrementaldecoderdecoder)selfrrrkwargss r __init__zPtyWithClients.__init__)sp (-b(>4SYBC8  !#-F< (..88  Ev;;GDIV r cdx}}|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)rsizer getwinsize setwinsize)r$minrowsmincolsclientrowscolss r resize_to_smallestz!PtyWithClients.resize_to_smallest6s"!'ll FJD$D7ND7N   e w%/ \\,,. d $tj tj tjtjg}tj}fd}jjsy |D]>}j||d{jjr>y|rJjtj|d{jjsyyy7o7##t$r/|d{7jjsYyYywxYww)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.r5cVtjjjSr)asynciosleepr delayafterterminater$sr r?z'PtyWithClients.terminate..sleepcs==!A!AB Br TNF)r6r7rSIGINTSIGTERMSIGHUPSIGCONTrcurrentr isaliver2SIGKILLOSError)r$forcesignals_r?r3s` r terminatezPtyWithClients.terminateWs 77d?}}fnn5G}}fnnfmmV^^TG NN  C||##%   #g ||++-   &..)g ||++-     'MM<<'') slB'E:+ D? D; D?+D?,E:-,D?D=D?8E:;D?=D??E7EE72E:6E77E:)NNF) __name__ __module__ __qualname__r&r0rrCr2r:rMr r r rr(s. W6,~~ (r rch|jD]\}}||j|d|||<!y)z**--b"g61ar cdeZdZdZ ddZddZdZdZdZddZ d Z dd Z d Z d Z d Zy)TermManagerBasez"Base class for a terminal manager.Ncb||_||_|xsi|_||_t j t |_i|_|#d|_ tjd|_ nd|_ ||_ |5tjd|jj dt dyy) NF) max_workersTzSetting z!.ioloop is deprecated and ignored) stacklevel) shell_command server_url term_settings extra_envlogging getLoggerrOlog ptys_by_fd!_blocking_io_executor_is_externalrThreadPoolExecutorblocking_io_executorwarningswarn __class__DeprecationWarning)r$rprqrrrsiolooprzs r r&zTermManagerBase.__init__s+$*0b"$$X.  '5:D 2(/(B(Bq(QD %59D 2(>2233TU"  r c 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 DIMENSIONSCOLUMNSLINESURLrs) r6environcopyrrgetDEFAULT_TERM_TYPE ENV_PREFIXstrrqrsrY isinstancer) 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( + r c |jj}|j|d<|j||d}|jdi|}|j dd}t |||S)z?Make a new terminal, return a :class:`PtyWithClients` instance.rprNr )rrrrpupdaterrr)r$r%optionsrrrs r new_terminalzTermManagerBase.new_terminalsp$$))+#'#5#5 v' d  +7+kk%&dC--r c|jj}||j|<tj}|j ||j |jy)zBConnect a terminal to the tornado event loop to read data from it.N)r rZrwrrF add_handlerpty_readREAD)r$ ptywclientsrZloops r start_readingzTermManagerBase.start_readingsG  # #)~~ T]]DII6r c|jj}|jjd||j|=t j j||jjy)zCalled when the pty has closed.zEOF on FD %d; stopping readingN) r rZrvinforwrrFremove_handlerclose)r$rrZs r on_eofzTermManagerBase.on_eofs] # #  6; OOB ''+ !!#r ct|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.皙?)r[zSpurious pty_read() on fd Ni)rhrvdebugrwpre_pty_read_hookr readrappendr on_pty_readEOFErrorr on_pty_died)r$rZeventsrsr-s r rzTermManagerBase.pty_readsR% HHNN7t< = 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 r rz!TermManagerBase.pre_pty_read_hooksr ct)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 "!r cy)z=Override this to e.g. kill terminals on client disconnection.Nr r$ websockets r client_disconnectedz#TermManagerBase.client_disconnecteds r cK|jd{|js|jjddyy7/w)NFT)waitcancel_futures)kill_allrxrzshutdownrAs r rzTermManagerBase.shutdownsAmmo55  % % . .E$ . O6 sAA0AcKg}|jjD]$}|j|jd&|rt j |d{yy7w)NTrJ)rwvaluesrrMr>gather)r$rterms r rzTermManagerBase.kill_alls_OO**, 7D GNN4>>>5 6 7 ..'* * *  *sAA&A$A&)NNNN)Prrr)rOrPrQ__doc__r&rrrrrrrrrrr r r rjrjsO, !>2.7 $%$`" P +r rjc4eZdZdZfdZddZfdZxZS)SingleTermManagerz9All connections to the websocket share a common terminal.c 2t|di|d|_yNr )superr&terminal)r$r%r}s r r&zSingleTermManager.__init__#s "6" r c|j0|j|_|j|j|jSr)rrrrs r rzSingleTermManager.get_terminal's8 ==  --/DM   t}} -}}r cLKt|d{d|_y7 wr)rrrr$r}s r rzSingleTermManager.kill_all-s$g     ! $" $r)rOrPrQrr&rr __classcell__r}s@r rr sC r rceZdZdZdZy)MaxTerminalsReachedc||_yr max_terminals)r$rs r r&zMaxTerminalsReached.__init__3s *r c d|jzS)Nz$Cannot create more than %d terminalsrrAs r __str__zMaxTerminalsReached.__str__6s58J8JJJr N)rOrPrQr&rr r r rr2s +Kr rc2eZdZdZdfd ZddZdZxZS)UniqueTermManagerz-Give each websocket a unique terminal to use.c 2t|di|||_yr)rr&rr$rr%r}s r r&zUniqueTermManager.__init__=s "6"*r c|jr7t|j|jk\rt|j|j }|j ||Sr)rlenrwrrr)r$rrs r rzUniqueTermManager.get_terminalAsS   #doo"6$:L:L"L%d&8&89 9  " 4  r c`|jjd|jrtjdk(rJ|jj |j |jjjy|jjtjyy)z-Send terminal SIGHUP when client disconnects.z-Websocket closed, sending SIGHUP to terminal.r5N) rvrrr6r7r2rr rZr:rrDrs r rz%UniqueTermManager.client_disconnectedIsy  EF   ww$""'') i0088;;<    % %fmm 4 r r)rOrPrQrr&rrrrs@r rr:s7+ 5r rcteZdZdZd fd ZdZdZdZdZe jfdZ d dZ fd Z fd ZxZS) NamedTermManagerzBShare terminals between websockets connected to the same endpoint.c @t|di|||_i|_yr)rr&r terminalsrs r r&zNamedTermManager.__init__Ys" "6"*r ct|J||jvr|j|S|jr7t|j|jk\rt|j|jj d||j }||_||j|<|j||S)Nz$New terminal with specified name: %s) rrrrrvrr term_namer)r$rrs r rzNamedTermManager.get_terminal^s$$  &>>), ,   #dnn"59K9K"K%d&8&89 9  >)  r c d|vr|d}n|j}|jdi|}|jjd|||_||j |<|j |||fS)Nr7z$New terminal with automatic name: %sr )rrrvrrrr)r$r%r7rs r new_named_terminalz#NamedTermManager.new_named_terminalwsv V &>D,,.D t  *6*  r!rrtr6r_rr{ collectionsr concurrentr ptyprocessrr ImportErrorwinptyrobjecttornado.iolooprrrrrYintfloatlistrhrjr Exceptionrrrr r r rs   ,6"  %WWt c E D D+D+N$K)K558=={ #: #"#J s5 BB1BB1B)&B1(B))B10B1