B .d@s\dZddlZddlZddlZddlZddlmZddlmZddZ Gdddej j Z dS) z9Tornado websocket handler to serve a terminal interface. N)gen)run_on_executorcCst|tr|dS|S)Nzutf-8) isinstancebytesdecode)sr`/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/terminado/websocket.py _cast_unicodes  r cseZdZdZddZdddZdfdd Zd d Zd d Ze j d dZ ddZ ddZ deddddZeddddZZS) TermSocketz Handler for a terminal websocketcCsN||_d|_d|_d|_|j|_tt|_ d|_ t t dddk|_dS)N)NNZLOG_TERMINAL_OUTPUTfalsetrue) term_manager term_namesizeterminalZblocking_io_executor_blocking_io_executorlogging getLogger__name___logger _user_commandstrlowerosgetenv_enable_output_logging)selfrrrr initializes zTermSocket.initializeNcCs||p|jjdS)z1Deprecated: backward-compat for terminado <= 0.5.ZOrigin)Z check_originrequestheadersget)roriginrrr origin_check&szTermSocket.origin_checkcst||jd|t|}|p(d|_|j||_|jj || dig|jd|jd}|jj }x|s~P|}||7}qxW|r||dS)zWebsocket connection opened. Call our terminal manager to get a terminal, and connect to it as a client. zTermSocket.open: %sttysetupzTermSocket.open: Opened %sr N)superopenrinfor rrZ get_terminalrclientsappendsend_json_messageZ read_buffercopypopleft on_pty_read)rZ url_componentbufferedZpreopen_bufferr) __class__rr r(*s"    zTermSocket.opencCs|d|gdS)z$Data read from pty; send to frontendstdoutN)r,)rtextrrr r/GszTermSocket.on_pty_readcCsLt|}|||jrH|ddkrHt|dtrH|d|ddS)Nrr2zSTDOUT: )jsondumpsZ write_messagerrrlog_terminal_output)rcontentZjson_msgrrr r,Ks   zTermSocket.send_json_messageccst|}|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. rNstdinr4 zSTDIN: r Zset_size) r5loadsrAssertionErrorstdin_to_ptyprocrr7rrresize_to_smallest)rmessagecommandZmsg_typerrr on_messageSs  zTermSocket.on_messagecCs:|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*remover?rZclient_disconnected)rrrr on_closejs   zTermSocket.on_closecCs |ddg|d|_dS)z9Terminal closed: tell the frontend, and close the socket.Z disconnectr4N)r,closer)rrrr on_pty_diedvszTermSocket.on_pty_diedr )logreturncCs|j|dS)zg Logs the terminal input/output :param log: log line to write :return: N)rdebug)rrGrrr r7|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)rZptyprocwrite)rr3rrr r>s zTermSocket.stdin_to_ptyproc)N)N)r )r __module__ __qualname____doc__rr$r(r/r,r coroutinerBrDrFrr7rr> __classcell__rr)r1r r s  r ) rNr5rrZtornado.websockettornadorZtornado.concurrentrr Z websocketZWebSocketHandlerr rrrr s