U Qe@sdZddlmZddlZddlZddlZddlmZmZddl Z ddl m Z ddl m Z erlddlmZmZdd d d d ZGd dde jjZdS)z9Tornado websocket handler to serve a terminal interface. ) annotationsN) TYPE_CHECKINGAny)gen)run_on_executor)PtyWithClientsTermManagerBasez str | bytesstr)sreturncCst|tr|dS|S)Nzutf-8) isinstancebytesdecode)r r2lib/python3.8/site-packages/terminado/websocket.py _cast_unicodes  rcseZdZdZdddddZd(dd d d d Zd)d ddfdd ZdddddZd ddddZe j dddddZ ddddZ ddddZ d*ddd!d"d#Zed$d%dddd&d'ZZS)+ TermSocketz Handler for a terminal websocketrNone) term_managerr cCsN||_d|_d|_d|_|j|_tt|_ d|_ t t dddk|_dS)zInitialize the handler.)NNNZLOG_TERMINAL_OUTPUTZfalsetrue)r term_namesizeterminalZblocking_io_executor_blocking_io_executorloggingZ getLogger__name___logger _user_commandr lowerosgetenv_enable_output_logging)selfrrrr initializes zTermSocket.initializeNz str | Nonebool)originr cCs*|p|jjdd}|dk s t||S)z1Deprecated: backward-compat for terminado <= 0.5.ZOriginrN)ZrequestZheadersgetAssertionErrorZ check_origin)r#r&rrr origin_check,s zTermSocket.origin_checkr) url_componentr cst||jd|t|}|p(d|_|j||_|jj || dig|jd|jd}|jj }|s|q|}||7}qv|r||dS)zWebsocket connection opened. Call our terminal manager to get a terminal, and connect to it as a client. zTermSocket.open: %sZttyZsetupzTermSocket.open: Opened %srN)superopenrinforrrZ get_terminalrclientsappendsend_json_messageZ read_buffercopypopleft on_pty_read)r#r*ZbufferedZpreopen_bufferr  __class__rrr,2s     zTermSocket.openr )textr cCs|d|gdS)z$Data read from pty; send to frontendstdoutN)r0r#r6rrrr3OszTermSocket.on_pty_read)contentr cCsLt|}|||jrH|ddkrHt|dtrH|d|ddS)z"Send a json message on the socket.rr7zSTDOUT: N)jsondumpsZ write_messager"r r log_terminal_output)r#r9Zjson_msgrrrr0Ss   zTermSocket.send_json_message)messager ccst|}|d}|jdk s t|dkrx||dV|jr|ddkrd|d|jd|_q|j|d7_n |dkr|dd |_|j dS) zHandle incoming websocket message We send JSON arrays, where the first element is a string indicating what kind of message this is. Data associated with the message follows. rNstdinr: zSTDIN: rZset_size) r;loadsrr(stdin_to_ptyprocr"r=rrresize_to_smallest)r#r>ZcommandZmsg_typerrr on_message[s  zTermSocket.on_message)r cCs:|jd|jr*|jj||j|j|dS)zHandle websocket closing. Disconnect from our terminal, and tell the terminal manager we're disconnecting. zWebsocket closedN)rr-rr.removerDrZclient_disconnectedr#rrron_closers   zTermSocket.on_closecCs |ddg|d|_dS)z9Terminal closed: tell the frontend, and close the socket.Z disconnectr:N)r0closerrGrrr on_pty_died~szTermSocket.on_pty_diedr)logr cCs|j|dS)zg Logs the terminal input/output :param log: log line to write :return: N)rdebug)r#rKrrrr=szTermSocket.log_terminal_outputr)executorcCs|jdk r|jj|dS)aHandles stdin messages sent on the websocket. This is a blocking call that should NOT be performed inside the server primary event loop thread. Messages must be handled asynchronously to prevent blocking on the PTY buffer. N)rZptyprocwriter8rrrrCs zTermSocket.stdin_to_ptyproc)N)N)r)r __module__ __qualname____doc__r$r)r,r3r0r coroutinerErHrJr=rrC __classcell__rrr4rrs r)rQZ __future__rr;rr typingrrZtornado.websocketZtornadorZtornado.concurrentrZterminado.managementrrrZ websocketZWebSocketHandlerrrrrrs