B tZ @sdZddlmZmZyddlmZWn ek rDddlmZYnXddlZddlZddl Z ddl Z ddZ Gddde j jZdS) z9Tornado websocket handler to serve a terminal interface. )absolute_importprint_function)urlparseNcCst|tr|dS|S)Nzutf-8) isinstancebytesdecode)sr 2lib/python3.7/site-packages/terminado/websocket.py _cast_unicodes  r cs\eZdZdZddZdddZdfdd Zd d Zd d Zd dZ ddZ ddZ Z S) TermSocketz Handler for a terminal websocketcCs(||_d|_d|_d|_tt|_dS)N)NN) term_manager term_namesizeterminalloggingZ getLogger__name___logger)selfrr r r initializes zTermSocket.initializeNcCs||p|jjdS)z1Deprecated: backward-compat for terminado <= 0.5.ZOrigin)Z check_originZrequestZheadersget)roriginr r r origin_check$szTermSocket.origin_checkcstt|||jd|t|}|p,d|_|j||_ x|j j D]}| |qHW|j j ||dig|jd|jdS)zWebsocket connection opened. Call our terminal manager to get a terminal, and connect to it as a client. zTermSocket.open: %sZttyZsetupzTermSocket.open: Opened %sN)superr openrinfor rrZ get_terminalrZ read_buffer on_pty_readclientsappendsend_json_message)rZ url_componentr) __class__r r r(s zTermSocket.opencCs|d|gdS)z$Data read from pty; send to frontendstdoutN)r )rtextr r r r>szTermSocket.on_pty_readcCst|}||dS)N)jsondumpsZ write_message)rZcontentZjson_msgr r r r Bs zTermSocket.send_json_messagecCsRt|}|d}|dkr.|jj|dn |dkrN|dd|_|jdS)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. rstdinZset_sizeN)r$loadsrZptyprocwriterresize_to_smallest)rmessageZcommandZmsg_typer r r on_messageFs 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)rrrrremover+rZclient_disconnected)rr r r on_closeVs   zTermSocket.on_closecCs|ddg|dS)zBTerminal closed: tell the frontend, and close the socket. Z disconnectr'N)r close)rr r r on_pty_diedbszTermSocket.on_pty_died)N)N) r __module__ __qualname____doc__rrrrr r-r/r1 __classcell__r r )r!r r s  r )r4Z __future__rrZ urllib.parser ImportErrorr$rZ tornado.webZtornadoZtornado.websocketr Z websocketZWebSocketHandlerr r r r r s