B tZ.@s&dZddlmZmZddlZejddkr0eZn ddZeZ ddl m Z ddl Z ddl Z ddlZddlZyddlmZWn ek rdd lmZYnXdd lmZdd lmZd Zd ZGdddeZddZGdddeZGdddeZGddde Z!GdddeZ"GdddeZ#dS)zMTerminal management for exposing terminals to a web interface using Tornado. )absolute_importprint_functionNcCs|S)N)xrr3lib/python3.7/site-packages/terminado/management.py sr)deque)PtyProcessUnicode) PtyProcess)gen)IOLoopZPYXTERM_Zxtermc@sHeZdZddZddZejfddZejfddZe j d d d Z d S)PtyWithClientscCs||_g|_tgdd|_dS)N )maxlen)ptyprocclientsr read_buffer)selfrrrr__init__#szPtyWithClients.__init__cCsd}}x@|jD]6}|j\}}|dk r2||kr2|}|dk r||kr|}qW|dksZ|dkr^dS|j\}}||f||fkr|j||dS)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)rsizerZ getwinsizeZ setwinsize)rZminrowsZmincolsclientZrowsZcolsrrrresize_to_smallest+s  z!PtyWithClients.resize_to_smallestcCs|j|dS)z'Send a signal to the process in the ptyN)rkill)rsigrrrrAszPtyWithClients.killcCs4tjdkr|j|St|jj}t||dS)z.TFN)rrsignalSIGINTSIGTERMSIGHUPSIGCONTr ZcurrentrZisaliver ZReturnrSIGKILLOSError)rforceZsignalsZsleeprr)r!rr terminateLs4             zPtyWithClients.terminateN)F) __name__ __module__ __qualname__rrr"r$rr r coroutiner*rrrrr"s rcCs8x2|D]&\}}|dkr(||dq |||<q WdS)zALike dict.update(), but remove keys where the value is None. N)itemspop)targetZchangeskvrrr_update_removingvsr4c@s|eZdZdZdiddfddZdd d Zd d Zd dZddZdddZ dddZ ddZ e j ddZe j ddZdS)TermManagerBasez"Base class for a terminal manager.NcCsT||_||_||_||_tt|_i|_|dk r:||_ nddl }|j j |_ dS)Nr) shell_command server_url term_settings extra_envloggingZ getLoggerr+log ptys_by_fdiolooptornado.ioloopr instance)rr7r8r9r:r>tornadorrrrs zTermManagerBase.__init__PrcKstj}|jdt|d<d||f}|r@|r@|d||f7}||td<t||d<t||d<|jrx|j|td<|j rt ||j |S) z@Build the environment variables for the process in the terminal.typeZTERMz%dx%dz;%dx%dZ DIMENSIONSZCOLUMNSZLINESZURL) renvironcopyr9getDEFAULT_TERM_TYPE ENV_PREFIXstrr8r:r4)rZheightwidthZ winheightZwinwidthkwargsenvZ dimensionsrrr make_term_envs      zTermManagerBase.make_term_envcKsR|j}|j|d<|||d}|jf|}tj|||ddd}t|S)z?Make a new terminal, return a :class:`PtyWithClients` instance.r7cwdN)rMrO) r9rFr7updaterNr ZspawnrGr)rrLZoptionsargvrMZptyrrr new_terminals    zTermManagerBase.new_terminalcCs,|jj}||j|<|j||j|jjdS)zBConnect a terminal to the tornado event loop to read data from it.N)rfdr=r>Z add_handlerpty_readZREAD)r ptywclientsrSrrr start_readings zTermManagerBase.start_readingcCs8|jj}|jd||j|=|j||jdS)z(Called when the pty has closed. zEOF on FD %d; stopping readingN)rrSr<infor=r>Zremove_handlerclose)rrUrSrrron_eofs  zTermManagerBase.on_eofcCs||j|}y6|jd}|j|x|jD]}||q,WWn6tk rv||x|jD] }| qbWYnXdS)z>Called by the event loop when there is pty data ready to read.iN) r=rreadrappendrZ on_pty_readEOFErrorrYZ on_pty_died)rrSZeventsrUsrrrrrTs      zTermManagerBase.pty_readcCstdS)a?Override 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. N)NotImplementedError)r url_componentrrr get_terminalszTermManagerBase.get_terminalcCsdS)zFOverride this to e.g. kill terminals on client disconnection. Nr)r websocketrrrclient_disconnectedsz#TermManagerBase.client_disconnectedccs|VdS)N)kill_all)rrrrshutdownszTermManagerBase.shutdownccsBg}x$|jD]}||jddqWx|D] }|Vq0WdS)NT)r))r=valuesr[r*)rZfuturestermfrrrrcs  zTermManagerBase.kill_all)rBrCrr)N)N)r+r,r-__doc__rrNrRrVrYrTr`rbr r.rdrcrrrrr5s   r5cs<eZdZdZfddZd ddZejfddZZ S) SingleTermManagerz9All connections to the websocket share a common terminal.c stt|jf|d|_dS)N)superrirterminal)rrL) __class__rrrszSingleTermManager.__init__NcCs&|jdkr ||_||j|jS)N)rkrRrV)rr_rrrr`s   zSingleTermManager.get_terminalc#stt|Vd|_dS)N)rjrircrk)r)rlrrrcszSingleTermManager.kill_all)N) r+r,r-rhrr`r r.rc __classcell__rr)rlrris  ric@seZdZddZddZdS)MaxTerminalsReachedcCs ||_dS)N) max_terminals)rrorrrrszMaxTerminalsReached.__init__cCs d|jS)Nz$Cannot create more than %d terminals)ro)rrrr__str__szMaxTerminalsReached.__str__N)r+r,r-rrprrrrrnsrncs4eZdZdZd fdd Zd ddZddZZS) UniqueTermManagerz-Give each websocket a unique terminal to use.Nc stt|jf|||_dS)N)rjrqrro)rrorL)rlrrrszUniqueTermManager.__init__cCs6|jr t|j|jkr t|j|}|||S)N)rolenr=rnrRrV)rr_rfrrrr`s   zUniqueTermManager.get_terminalcCsL|jd|jrHtjdkr:|j||jjjdS|j t j dS)z-Send terminal SIGHUP when client disconnects.z-Websocket closed, sending SIGHUP to terminal.rN) r<rWrkrrrrTrrSr r"r%)rrarrrrbs   z%UniqueTermManager.client_disconnected)N)N)r+r,r-rhrr`rbrmrr)rlrrqs rqcszeZdZdZdfdd ZddZdZdd Zd d Ze j fd d Z e j dddZfddZe j fddZZS)NamedTermManagerzGShare terminals between websockets connected to the same endpoint. Nc s"tt|jf|||_i|_dS)N)rjrsrro terminals)rrorL)rlrrrszNamedTermManager.__init__cCst|dk s t||jkr |j|S|jr@t|j|jkr@t|j|jd||}||_||j|<| ||S)Nz$New terminal with specified name: %s) AssertionErrorrtrorrrnr<rWrR term_namerV)rrvrfrrrr`$s      zNamedTermManager.get_terminalz%dcCs2x,tjddD]}|j|}||jkr|SqWdS)N)start) itertoolscount name_templatert)rnrrrr_next_available_name7s  z%NamedTermManager._next_available_namecCs@|}|}|jd|||_||j|<||||fS)Nz$New terminal with automatic name: %s)r}rRr<rWrvrtrV)rrrfrrrnew_named_terminal=s  z#NamedTermManager.new_named_terminalcCs|j|}||dS)N)rtr)rrrrfrrrrFs zNamedTermManager.killFccs|j|}|j|dVdS)N)r))rtr*)rrr)rfrrrr*Js zNamedTermManager.terminatecs6tt|||j}|jd||j|ddS)NzTerminal %s closed)rjrsrYrvr<rWrtr0)rrUr)rlrrrYOszNamedTermManager.on_eofc#stt|Vi|_dS)N)rjrsrcrt)r)rlrrrcUszNamedTermManager.kill_all)N)F)r+r,r-rhrr`r{r}r~r"r$rr r.r*rYrcrmrr)rlrrss   rs)$rhZ __future__rrsys version_infoordZ byte_coderJZunicode collectionsr ryr;rr"Z ptyprocessr ImportErrorZwinptyr rAr r?r rIrHobjectrr4r5ri Exceptionrnrqrsrrrrs4   T j