XBedZddlZddlZddlZddlZddlmZddlmZdZ GddejjZ y)z9Tornado websocket handler to serve a terminal interface. N)gen)run_on_executorcHt|tr|jdS|S)Nzutf-8) isinstancebytesdecode)ss 3lib/python3.12/site-packages/terminado/websocket.py _cast_unicoder s !Uxx  HceZdZdZdZddZdfd ZdZdZe jdZ d Z d Z dd ed dfd ZeddZxZS) TermSocketz Handler for a terminal websocketc ||_d|_d|_d|_|j|_t jt|_ d|_ tjtjdddk(|_y)N)NNLOG_TERMINAL_OUTPUTfalsetrue) term_manager term_namesizeterminalblocking_io_executor_blocking_io_executorlogging getLogger__name___logger _user_commandstrlowerosgetenv_enable_output_logging)selfrs r initializezTermSocket.initializesm(   %1%F%F"((2 '*ii :OQX0Y&Z^d&d#r Ncr|j|xs%|jjjdS)z1Deprecated: backward-compat for terminado <= 0.5.Origin) check_originrequestheadersget)r$origins r origin_checkzTermSocket.origin_check&s-  !M4<<+?+?+C+CH+MNNr c,t|||jjd|t |}|xsd|_|j j||_|jjj||jdig|jjd|j d}|jjj} |sn|j}||z }|r|j|yy)zWebsocket connection opened. Call our terminal manager to get a terminal, and connect to it as a client. zTermSocket.open: %sttysetupzTermSocket.open: Opened %srN)superopenrinfor rr get_terminalrclientsappendsend_json_message read_buffercopypopleft on_pty_read)r$ url_componentbufferedpreopen_bufferr __class__s r r2zTermSocket.open*s  ]# /?%m4 &/%))66}E  $$T* }- 6G22779!&&(A MH     X & r c*|jd|gy)z$Data read from pty; send to frontendstdoutN)r7r$texts r r;zTermSocket.on_pty_readGs $/0r ctj|}|j||jr5|ddk(r,t |dt r|j d|dyyyy)NrrAzSTDOUT: )jsondumps write_messager#rrlog_terminal_output)r$contentjson_msgs r r7zTermSocket.send_json_messageKsg::g& 8$  & &qzX%*WQZ*E((8GAJ<)@A+F% 'r c#Ktj|}|d}|jJ|dk(rj|j|d|jrG|ddk(r&|j d|j d|_y|xj |dz c_yy|dk(r%|dd |_|jjyyw) 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. rNstdinrE zSTDIN: rset_size) rFloadsrstdin_to_ptyprocr#rIrrresize_to_smallest)r$messagecommandmsg_types r on_messagezTermSocket.on_messageSs**W%1:}}(( w '' 3 3**1:%,,wt7I7I6J-KL)+D&&&'!*4& +  #! DI MM , , .$sCCc|jjd|jr?|jjj ||jj |j j|y)zHandle websocket closing. Disconnect from our terminal, and tell the terminal manager we're disconnecting. zWebsocket closedN)rr3rr5removerSrclient_disconnectedr$s r on_closezTermSocket.on_closejsZ ,- == MM ! ! ( ( . MM , , . --d3r cX|jddg|jd|_y)z9Terminal closed: tell the frontend, and close the socket. disconnectrEN)r7closerr[s r on_pty_diedzTermSocket.on_pty_diedvs%  a01  r logreturnc:|jj|y)zg Logs the terminal input/output :param log: log line to write :return: N)rdebug)r$ras r rIzTermSocket.log_terminal_output|s 3r r)executorch|j&|jjj|yy)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)rptyprocwriterBs r rRzTermSocket.stdin_to_ptyprocs+ == $ MM ! ! ' ' - %r )N)r)r __module__ __qualname____doc__r%r-r2r;r7r coroutinerWr\r`rrIrrR __classcell__)r?s@r rrsq* eO':1B ]]//, 4  s D 56.7.r r) rkrFrr!tornado.websockettornadortornado.concurrentrr websocketWebSocketHandlerrr r rts=   . w.""33w.r