Mh >dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZmZmZerddlmZ ddlmZd'd Zn"#e$r dd lmZn #e$reZYnwxYwdZYnwxYwdd lmZdZdZGddZ d(dZ!d)d*dZ"GddZ#Gdd e#Z$Gd!d"e%Z&Gd#d$e#Z'Gd%d&e#Z(dS)+zMTerminal management for exposing terminals to a web interface using Tornado. ) annotationsN)deque)futures) TYPE_CHECKINGAny Coroutine) TermSocket)PtyProcessUnicodereturnNonecXtjtjtjdS)z-A prexec function to set up a signal handler.N)signalSIGPIPESIG_DFLT/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/terminado/management.py preexec_fnrs fnfn55555r) PtyProcess)IOLoopPYXTERM_zxterm-256colorcfeZdZUdZded<ddd Zdd ZejfddZ ejfddZ dddZ dS)PtyWithClientsz%A pty object with associated clients. str | None term_nameNargvrenvdict[str, str] | Nonecwdcg|_tgd|_||pg|d}t t|d<t jd i||_tjdd|j_ dS) zInitialize the pty.)maxlen)rrrNrzutf-8replace)errorsr) clientsr read_bufferrr spawnptyproccodecsgetincrementaldecoderdecoder)selfrrrkwargss r__init__zPtyWithClients.__init__2s"$ ',R'='='=syb==  !#-F< (.8888  Ev;GDDIVVV rr r cdx}}|jD] }|j\}}|||kr|}|||kr|}!|dks|dkrdS|j\}}||f||fkr|j||dSdS)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"!'l  FJD$D7NND7NN e  w%// F\,,.. d $r Coroutine[Any, Any, None]c@tjjjS)z/Sleep to allow the terminal to exit gracefully.)asynciosleepr(delayafterterminater,srrKz'PtyWithClients.terminate..sleepns=!ABB BrTNF)r rH)r?r@rSIGINTSIGTERMSIGHUPSIGCONTrcurrentr(isaliver<SIGKILLrFOSError)r,rEsignals_rKr9s` r terminatezPtyWithClients.terminatebs 7d??}fn5GG}fnfmV^TG N   C C C C C C|##%% 4 4   #egg |++-- 44  8 &.)))egg  4 4 6 667775 4 4 4 %''MMMMMMMDL0022233 3 3 3  4sAD0AD00AE43E4)NN)rrrrrrr r )r9r:r r )r9r:r rF)rErFr rF) __name__ __module__ __qualname____doc____annotations__r.r8rrOr<rCrXrrrrr-s// W W W W W6666,%n!'#4#4#4#4#4#4#4rrtargetrchangescv|D]#\}}|||d|||<$dS)z***mRD"b'22GAq! HrceZdZdZ d)d*d Z d+d,dZd-dZd.dZd.dZd/d0d!Z d.d"Z d/d1d$Z d2d&Z d3d'Z d3d(ZdS)4TermManagerBasez"Base class for a terminal manager.N shell_commandstr server_url term_settingsr extra_envioloopblocking_io_executorcL||_||_|pi|_||_t jt |_i|_|"d|_ tj d|_ nd|_ ||_ |,tjd|jjdt d dSdS) Initialize the manager.NF) max_workersTzSetting z!.ioloop is deprecated and ignored) stacklevel)r|r~rrlogging getLoggerr[log ptys_by_fd!_blocking_io_executor_is_externalrThreadPoolExecutorrwarningswarn __class__DeprecationWarning)r,r|r~rrrrs rr.zTermManagerBase.__init__s+$*0b"$X..57  '5:D 2(/(Bq(Q(Q(QD % %59D 2(2UUU"        rPrheightr:width winheightwinwidthr-r dict[str, str]c tj}|jdt |d<d||fz}|r |r |d||fzz }||t dz<t||d<t||d<|jr|j|t dz<|j rt||j |d i}|r%t|trt|||S) z@Build the environment variables for the process in the terminal.typeTERMz%dx%dz;%dx%d DIMENSIONSCOLUMNSLINESURLr) r?environcopyrgetDEFAULT_TERM_TYPE ENV_PREFIXr}r~rrg isinstancedict) r,rrrrr-r dimensionsterm_envs r make_term_envzTermManagerBase.make_term_envs joo(,,V5FGGF v.  ;  ; (h %:: :J)3J %&UI6{{G ? 6&*oC U" # > 2 S$. 1 1 1::k2..  , 8T22 , S( + + + rrc |j}|j|d<|||d}|jdi|}|dd}t |||S)z?Make a new terminal, return a :class:`PtyWithClients` instance.r|rNr)rrr|updaterrr)r,r-optionsrrrs r new_terminalzTermManagerBase.new_terminals|$))++#'#5 v' d ++7++kk%&&dC---r ptywclientsr c|jj}||j|<tj}|||j|jdS)zBConnect a terminal to the tornado event loop to read data from it.N)r(rirrrR add_handlerpty_readREAD)r,rriloops r start_readingzTermManagerBase.start_readingsH  #)~ T]DI66666rc|jj}|jd||j|=t j||jdS)zCalled when the pty has closed.zEOF on FD %d; stopping readingN) r(ririnforrrRremove_handlerclose)r,rris ron_eofzTermManagerBase.on_eofsi #  6;;; OB ''+++ !!#####rrieventsct|ds|jd|dS|j|} |||jd}|j||j D]}| |dS#t$r7| ||j D]}| YdSwxYw)z>Called by the event loop when there is pty data ready to read.rh)rjzSpurious pty_read() on fd %sNi)rxrdebugrpre_pty_read_hookr(readr&appendr% on_pty_readEOFErrorr on_pty_died)r,rirrsr5s rrzTermManagerBase.pty_readsR%%%  HNN92 > > > Fob)  %  " "; / / /#((//A  # * *1 - - -%- & &""1%%%% & & % % % KK $ $ $%- % %""$$$$ % % % %sA(B''=C('C(cdS)zQHook before pty read, subclass can patch something into ptywclients when pty_readNr)r,rs rrz!TermManagerBase.pre_pty_read_hookr url_componentct)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,rs r get_terminalzTermManagerBase.get_terminals "!r websocketcdS)z=Override this to e.g. kill terminals on client disconnection.Nrr,rs rclient_disconnectedz#TermManagerBase.client_disconnectedrrcK|d{V|js|jdddSdS)zShutdown the manager.NFT)waitcancel_futures)kill_allrrshutdownrMs rrzTermManagerBase.shutdowns`mmoo5 P  % . .E$ . O O O O O P PrcKg}|jD]&}|j|d'|rt j|d{VdSdS)Kill all terminals.TrEN)rvaluesrrXrJgather)r,rterms rrzTermManagerBase.kill_all$sO**,, 7 7D GN4>>>55 6 6 6 6  +.'* * * * * * * * * * + +r)r{NNNN) r|r}r~r}rrrrrrrr)rrrr) rr:rr:rr:rr:r-rr r)r-rr rrrr r N)rir:rrr r rrr r)rrr r rY)r[r\r]r^r.rrrrrrrrrrrrrrzrzs7,, !$(D @....7777 $ $ $ $%%%%%$````"""""LLLLPPPP ++++++rrzc:eZdZdZd fd Zd dd Zdfd ZxZS)SingleTermManagerz9All connections to the websocket share a common terminal.r-rr r c Htjdi|d|_dSrNr)superr.terminal)r,r-rs rr.zSingleTermManager.__init__1s+""6"""/3 rNrrc|j3||_||j|jS)z "Get the singleton terminal.)rrrrs rrzSingleTermManager.get_terminal6s; =  --//DM   t} - - -}rcfKtd{Vd|_dS)zKill the singletone terminal.N)rrrr,rs rrzSingleTermManager.kill_all=s:gg          r)r-rr r rrrY)r[r\r]r^r.rr __classcell__rs@rrr.szCC444444 rrc"eZdZdZd dZd dZd S) MaxTerminalsReachedz;An error raised when we exceed the max number of terminals. max_terminalsr:r r c||_dS)zInitialize the error.Nr)r,rs rr.zMaxTerminalsReached.__init__Fs*rr}cd|jzS)z'The string representation of the error.z$Cannot create more than %d terminalsrrMs r__str__zMaxTerminalsReached.__str__Js58JJJrN)rr:r r )r r})r[r\r]r^r.rrrrrrCsHEE++++KKKKKKrrc8eZdZdZddfd Zddd ZddZxZS)UniqueTermManagerz-Give each websocket a unique terminal to use.Nr int | Noner-rr r c Htjdi|||_dSr)rr.rr,rr-rs rr.zUniqueTermManager.__init__Rs.""6"""*rrrc|jr1t|j|jkrt|j|}|||S)z Get a terminal from the manager.)rlenrrrr)r,rrs rrzUniqueTermManager.get_terminalWs_   :#do"6"6$:L"L"L%d&899 9  "" 4    rrr c2|jd|jrutjdkr?|j||jjjdS|j tj dSdS)z-Send terminal SIGHUP when client disconnects.z-Websocket closed, sending SIGHUP to terminal.r>N) rrrr?r@r<rr(rirCrrPrs rrz%UniqueTermManager.client_disconnected`s  EFFF   5w$"''))) i08;<<<   % %fm 4 4 4 4 4 5 5rr)rrr-rr r r)rr r r )r[r\r]r^r.rrrrs@rrrOsu77+++++++  5 5 5 5 5 5 5 5rrczeZdZdZddfd Zdd Zd Zd dZd!dZe j fd"dZ d#d$dZ d%fd Z d&fd ZxZS)'NamedTermManagerzBShare terminals between websockets connected to the same endpoint.Nrrr-r r c Vtjdi|||_i|_dSr)rr.r terminalsrs rr.zNamedTermManager.__init__ps3""6"""*46rrr}rcT|J||jvr |j|S|jr1t|j|jkrt|j|jd||}||_||j|<|||S)z!Get or create a terminal by name.Nz$New terminal with specified name: %s) rrrrrrrrr)r,rrs rrzNamedTermManager.get_terminalvs$$$  & &>), ,   :#dn"5"59K"K"K%d&899 9  )) *trtuple[str, PtyWithClients]c d|vr|dn|}|jdi|}|jd|||_||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_terminals!'6!1!1vf~~t7P7P7R7R t **6**  ~d#nn5n)))))))))))rrct||j}|jd||J|j|ddS)z*Handle end of file for a pty with clients.zTerminal %s closedN)rrrrrrrd)r,rr@rs rrzNamedTermManager.on_eofsb {###$  *D111 4&&&&&rcfKtd{Vi|_dS)rN)rrrrs rrzNamedTermManager.kill_alls:gg         rr)rrr-rr r )rr}r r)r r)r-rr r)r@r}r9r:r r rZ)r@r}rErFr r rrY)r[r\r]r^r.rrrrrrOr<rXrrrrs@rrrmsLL7777777 $M*0 ***** ''''''rrrY)r`rrarr r )rh)rir:rjrkr rl))r^ __future__rrJr)rrr?rorr collectionsr concurrentrtypingrrrterminado.websocketr ptyprocessr r ImportErrorwinptyrobjecttornado.iolooprrrrrgrxrzr Exceptionrrrrrrrs #"""""  0000000000/...... ,,,,,,66666#::::::: ###"#JJJ "!!!!!  %X4X4X4X4X4X4X4X4v      M+M+M+M+M+M+M+M+`* K K K K K) K K K55555555