o tf@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 r\/var/www/html/software/conda/envs/catlas/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.)NNNZLOG_TERMINAL_OUTPUTfalsetrue)r term_namesizeterminalZblocking_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.ZOriginrN)requestheadersgetZ 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 inforrrZ get_terminalrclientsappendsend_json_messageZ read_buffercopypopleft on_pty_read)r&r/bufferedZpreopen_bufferr  __class__rrr32s&   zTermSocket.opentextr cCs|d|gdS)z$Data read from pty; send to frontendstdoutN)r7r&r>rrrr:OszTermSocket.on_pty_readcontentcCsXt|}|||jr&|ddkr(t|dtr*|d|ddSdSdSdS)z"Send a json message on the socket.rr?zSTDOUT: N)jsondumpsZ write_messager%r r log_terminal_output)r&rAZjson_msgrrrr7Ss    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. rNstdinrB zSTDIN: rZset_size) rCloadsrstdin_to_ptyprocr%rEr!rresize_to_smallest)r&rFcommandmsg_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 r4rr5removerLrZclient_disconnectedr&rrron_closers  zTermSocket.on_closecCs |ddg|d|_dS)z9Terminal closed: tell the frontend, and close the socket.Z disconnectrBN)r7closerrQrrr 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&rUrrrrEszTermSocket.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)rZptyprocwriter@rrrrKs zTermSocket.stdin_to_ptyproc)rrr r)N)r(r)r r*)r/rr r)r>r r r)rArr r)rFr r r)r r)r)rUr r r)r __module__ __qualname____doc__r'r.r3r:r7r coroutinerOrRrTrErrK __classcell__rrr<rrs       r)r r r r )r[ __future__rrCrr#typingrrZtornado.websockettornadorZtornado.concurrentrZterminado.managementrrrZ websocketZWebSocketHandlerrrrrrs