MhdZddlmZddlZddlZddlZddlmZmZddl Z ddl m Z ddl m Z erddlmZmZdd ZGd de jjZdS)z9Tornado websocket handler to serve a terminal interface. ) annotationsN) TYPE_CHECKINGAny)gen)run_on_executor)PtyWithClientsTermManagerBases str | bytesreturnstrcZt|tr|dS|S)Nzutf-8) isinstancebytesdecode)r s S/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/terminado/websocket.py _cast_unicoders+!U!xx   HceZdZdZddZdd d Zdd!fd Zd"dZd#dZe j d$dZ d%dZ d%dZ d&d'dZedd"dZxZS)( TermSocketz Handler for a terminal websocket term_managerr r Nonec ||_d|_d|_d|_|j|_t jt|_ d|_ t tjdddk|_dS)zInitialize the handler.)NNNLOG_TERMINAL_OUTPUTfalsetrue)r term_namesizeterminalblocking_io_executor_blocking_io_executorlogging getLogger__name___logger _user_commandr lowerosgetenv_enable_output_logging)selfrs r initializezTermSocket.initializest(  /3 %1%F"(22 '*ii :OQX0Y0Y&Z&Z^d&d###rNorigin str | Noneboolcx|p|jjdd}|J||S)z1Deprecated: backward-compat for terminado <= 0.5.Originr)requestheadersget check_origin)r,r.s r origin_checkzTermSocket.origin_check,sAA4</33HbAA!!!  (((r url_componentrcHt||jd|t |}|pd|_|j||_|jj || dig|jd|jd}|jj } |sn|}||z }|r||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_terminalr clientsappendsend_json_message read_buffercopypopleft on_pty_read)r,r8bufferedpreopen_bufferr __class__s rr=zTermSocket.open2s)  ]### /???%m44 &/%)66}EE  $$T*** }--- 6GGG27799 ! &&((A MH    '   X & & & & & ' 'rtextr c4|d|gdS)z$Data read from pty; send to frontendstdoutN)rBr,rJs rrFzTermSocket.on_pty_readOs! $/00000rcontentctj|}|||jrG|ddkr=t |dt r$|d|ddSdSdSdS)z"Send a json message on the socket.rrLzSTDOUT: N)jsondumps write_messager+rr log_terminal_output)r,rNjson_msgs rrBzTermSocket.send_json_messageSs:g&& 8$$$  & >71:+A+AjQXYZQ[]`FaFa+A  $ $%< %<%< = = = = = > >+A+A+A+Armessagec#Ktj|}|d}|jJ|dkrp||dV|jrJ|ddkr&|d|jd|_dS|xj|dz c_dSdS|dkr*|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. rNstdinrP zSTDIN: rset_size) rQloadsr stdin_to_ptyprocr+rTr'rresize_to_smallest)r,rVcommandmsg_types r on_messagezTermSocket.on_message[s*W%%1:}((( w  '' 33 3 3 3* 51:%%,,-Kt7I-K-KLLL)+D&&&&&'!*4&&&&  5 5  # #! DI M , , . . . . .$ #rc|jd|jr8|jj||j|j|dS)zHandle websocket closing. Disconnect from our terminal, and tell the terminal manager we're disconnecting. zWebsocket closedN)r&r>r r@remover^rclient_disconnectedr,s ron_closezTermSocket.on_closerso ,--- = / M ! ( ( . . . M , , . . . --d33333rcj|ddg|d|_dS)z9Terminal closed: tell the frontend, and close the socket. disconnectrPN)rBcloser res r on_pty_diedzTermSocket.on_pty_died~s3  a0111  rrlogc:|j|dS)zg Logs the terminal input/output :param log: log line to write :return: N)r&debug)r,rks rrTzTermSocket.log_terminal_outputs 3rr")executorcV|j!|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)r ptyprocwriterMs rr]zTermSocket.stdin_to_ptyprocs3 = $ M ! ' ' - - - - - % $r)rr r r)N)r.r/r r0)r8rr r)rJr r r)rNrr r)rVr r r)r r)r)rkr r r)r% __module__ __qualname____doc__r-r7r=rFrBr coroutinerarfrjrTrr] __classcell__)rIs@rrrs3** e e e e))))) ''''''':1111>>>> ]///]/, 4 4 4 4      _5666...76.....rr)r r r r )rt __future__rrQr#r)typingrrtornado.websockettornadortornado.concurrentrterminado.managementrr r websocketWebSocketHandlerrrrrs #"""""  %%%%%%%%......EDDDDDDDD    z.z.z.z.z."3z.z.z.z.z.r