o i@sdZddlmZddlZddlZddlZddlmZmZddl Z ddl m Z ddl m Z er6ddlmZmZdd d ZGddde jjZdS)z9Tornado websocket handler to serve a terminal interface. ) annotationsN) TYPE_CHECKINGAny)gen)run_on_executor)PtyWithClientsTermManagerBases str | bytesreturnstrcCst|tr |dS|S)Nzutf-8) isinstancebytesdecode)r rN/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/terminado/websocket.py _cast_unicodes  rcseZdZdZd(ddZd)d*d d Zd)d+fdd Zd,ddZd-ddZe j d.ddZ d/ddZ d/ddZ d0d1d"d#Zed$d%d,d&d'ZZS)2 TermSocketz Handler for a terminal websocket term_managerrr NonecCsN||_d|_d|_d|_|j|_tt|_ d|_ t t dddk|_dS)zInitialize the handler.)NNNLOG_TERMINAL_OUTPUTfalsetrue)r term_namesizeterminalblocking_io_executor_blocking_io_executorlogging getLogger__name___logger _user_commandr lowerosgetenv_enable_output_logging)selfrrrr initializes zTermSocket.initializeNorigin str | NoneboolcCs*|p |jjdd}|dusJ||S)z1Deprecated: backward-compat for terminado <= 0.5.OriginrN)requestheadersget check_origin)r(r*rrr origin_check,s  zTermSocket.origin_check url_componentrcst||jd|t|}|pd|_|j||_|jj || dig|jd|jd}|jj } |s?n |}||7}q<|rQ||dSdS)zWebsocket connection opened. Call our terminal manager to get a terminal, and connect to it as a client. zTermSocket.open: %sttysetupzTermSocket.open: Opened %srTN)superopenr"inforrr get_terminalrclientsappendsend_json_message read_buffercopypopleft on_pty_read)r(r3bufferedpreopen_bufferr  __class__rrr72s&   zTermSocket.opentextr cCs|d|gdS)z$Data read from pty; send to frontendstdoutN)r<r(rErrrr@OszTermSocket.on_pty_readcontentcCsXt|}|||jr&|ddkr(t|dtr*|d|ddSdSdSdS)z"Send a json message on the socket.rrFzSTDOUT: N)jsondumps write_messager'r r log_terminal_output)r(rHjson_msgrrrr<Ss    zTermSocket.send_json_messagemessageccst|}|d}|jdusJ|dkrA||dV|jr?|ddkr4|d|jd|_dS|j|d7_dSdS|dkrS|dd |_|jdSdS) 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. rNstdinrI zSTDIN: rset_size) rJloadsrstdin_to_ptyprocr'rMr#rresize_to_smallest)r(rOcommandmsg_typerrr on_message[s    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)r"r8rr:removerVrclient_disconnectedr(rrron_closers  zTermSocket.on_closecCs |ddg|d|_dS)z9Terminal closed: tell the frontend, and close the socket. disconnectrIN)r<closerr\rrr on_pty_died~s zTermSocket.on_pty_diedrlogcCs|j|dS)zg Logs the terminal input/output :param log: log line to write :return: N)r"debug)r(rarrrrMszTermSocket.log_terminal_outputr)executorcCs |jdur|jj|dSdS)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)rptyprocwriterGrrrrUs zTermSocket.stdin_to_ptyproc)rrr r)N)r*r+r r,)r3rr r)rEr r r)rHrr r)rOr r r)r r)r)rar r r)r! __module__ __qualname____doc__r)r2r7r@r<r coroutinerYr]r`rMrrU __classcell__rrrCrrs       r)r r r r )rh __future__rrJrr%typingrrtornado.websockettornadortornado.concurrentrterminado.managementrrr websocketWebSocketHandlerrrrrrs