a 4_böã@s„dZddlmZmZddlZzddlmZWneyJddlmZYn0ddlZddl Z ddl Z dd„Z Gdd„de j jƒZdS) z9Tornado websocket handler to serve a terminal interface. é)Úabsolute_importÚprint_functionN)ÚurlparsecCst|tƒr| d¡S|S)Nzutf-8)Ú isinstanceÚbytesÚdecode)Ús©r ú2lib/python3.9/site-packages/terminado/websocket.pyÚ _cast_unicodes  r csleZdZdZdd„Zddd„Zd‡fdd„ Zd d „Zd d „Zd d„Z dd„Z dd„Z de dœdd„Z ‡ZS)Ú TermSocketz Handler for a terminal websocketcCsF||_d|_d|_d|_t t¡|_d|_t   t   dd¡¡dk|_ dS)NÚ)NNZLOG_TERMINAL_OUTPUTZfalseÚtrue)Ú term_managerÚ term_nameÚsizeÚterminalÚloggingZ getLoggerÚ__name__Ú_loggerÚ _user_commandÚstrÚlowerÚosÚgetenvÚ_enable_output_logging)Úselfrr r r Ú initialize s zTermSocket.initializeNcCs| |p|jj d¡¡S)z1Deprecated: backward-compat for terminado <= 0.5.ZOrigin)Z check_originZrequestZheadersÚget)rÚoriginr r r Ú origin_check,szTermSocket.origin_checkcs¤tt|ƒ |¡|j d|¡t|ƒ}|p,d|_|j |¡|_ |j j   |¡|  dig¡|j d|j¡d}|j j  ¡}|s€q’| ¡}||7}qz|r | |¡dS)z‚Websocket connection opened. Call our terminal manager to get a terminal, and connect to it as a client. zTermSocket.open: %sZttyZsetupzTermSocket.open: Opened %sr N)Úsuperr ÚopenrÚinfor rrZ get_terminalrÚclientsÚappendÚsend_json_messageZ read_bufferÚcopyÚpopleftÚ on_pty_read)rZ url_componentZbufferedZpreopen_bufferr©Ú __class__r r r"0s    zTermSocket.opencCs| d|g¡dS)z$Data read from pty; send to frontendÚstdoutN)r&)rÚtextr r r r)MszTermSocket.on_pty_readcCsLt |¡}| |¡|jrH|ddkrHt|dtƒrH| d|d›¡dS)Nrr,ézSTDOUT: )ÚjsonÚdumpsZ write_messagerrrÚlog_terminal_output)rZcontentZjson_msgr r r r&Qs   zTermSocket.send_json_messagecCst |¡}|d}|dkrl|jj |d¡|jrŒ|ddkrX| d|j›¡d|_qŒ|j|d7_n |dkrŒ|dd…|_|j  ¡d S) zÇHandle 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. rÚstdinr.ú zSTDIN: r Zset_sizeéN) r/ÚloadsrZptyprocÚwriterr1rrÚresize_to_smallest)rÚmessageZcommandZmsg_typer r r Ú on_messageYs  zTermSocket.on_messagecCs:|j d¡|jr*|jj |¡|j ¡|j |¡dS)z„Handle websocket closing. Disconnect from our terminal, and tell the terminal manager we're disconnecting. zWebsocket closedN)rr#rr$Úremover7rZclient_disconnected©rr r r Úon_closeos   zTermSocket.on_closecCs | ddg¡| ¡d|_dS)zBTerminal closed: tell the frontend, and close the socket. Z disconnectr.N)r&Úcloserr;r r r Ú on_pty_died{szTermSocket.on_pty_diedr )ÚlogcCs|j |¡dS)zg Logs the terminal input/output :param log: log line to write :return: N)rÚdebug)rr?r r r r1‚szTermSocket.log_terminal_output)N)N)r )rÚ __module__Ú __qualname__Ú__doc__rr r"r)r&r9r<r>rr1Ú __classcell__r r r*r r s  r )rCZ __future__rrrZ urllib.parserÚ ImportErrorr/rZtornado.websocketZtornador Z websocketZWebSocketHandlerr r r r r Ús