:vjdZddlmZddlZddlZddlZddlmZmZddl Z ddl m Z ddl m Z erddlmZmZd dZGd d e j$j&Zy) z9Tornado websocket handler to serve a terminal interface. ) annotationsN) TYPE_CHECKINGAny)gen)run_on_executor)PtyWithClientsTermManagerBasecHt|tr|jdS|S)Nzutf-8) isinstancebytesdecode)ss d/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/terminado/websocket.py _cast_unicoders !Uxx  HceZdZdZddZdddZddfd ZddZddZe jddZ ddZ dd Z ddd Zed dd ZxZS) TermSocketz Handler for a terminal websocketc ||_d|_d|_d|_|j|_t jt|_ d|_ tjtjdddk(|_y)zInitialize the handler.)NNNLOG_TERMINAL_OUTPUTfalsetrue) term_manager term_namesizeterminalblocking_io_executor_blocking_io_executorlogging getLogger__name___logger _user_commandstrlowerosgetenv_enable_output_logging)selfrs r initializezTermSocket.initializesm(  /3 %1%F%F"((2 '*ii :OQX0Y&Z^d&d#rc|xs&|jjjdd}|J|j|S)z1Deprecated: backward-compat for terminado <= 0.5.Originr)requestheadersget check_origin)r)origins r origin_checkzTermSocket.origin_check,s@A4<<//33HbA!!!  ((rc,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)superopenr"inforrr get_terminalrclientsappendsend_json_message read_buffercopypopleft on_pty_read)r) url_componentbufferedpreopen_bufferr __class__s rr7zTermSocket.open2s  ]# /?%m4 &/%))66}E  $$T* }- 6G22779!&&(A MH     X & rc*|jd|gy)z$Data read from pty; send to frontendstdoutN)r<r)texts rr@zTermSocket.on_pty_readOs $/0rctj|}|j||jr5|ddk(r,t |dt r|j d|dyyyy)z"Send a json message on the socket.rrFzSTDOUT: N)jsondumps write_messager(r r$log_terminal_output)r)contentjson_msgs rr<zTermSocket.send_json_messageSsi::g& 8$  & &71:+AjQXYZQ[]`Fa  $ $x |%< =Gb+A &rc#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. rNstdinrJ zSTDIN: rset_size) rKloadsrstdin_to_ptyprocr(rNr#rresize_to_smallest)r)messagecommandmsg_types r on_messagezTermSocket.on_message[s**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)r"r8rr:removerXrclient_disconnectedr)s ron_closezTermSocket.on_closersZ ,- == MM ! ! ( ( . MM , , . --d3rcX|jddg|jd|_y)z9Terminal closed: tell the frontend, and close the socket. disconnectrJN)r<closerr`s r on_pty_diedzTermSocket.on_pty_died~s%  a01  rc:|jj|y)zg Logs the terminal input/output :param log: log line to write :return: N)r"debug)r)logs rrNzTermSocket.log_terminal_outputs 3rr)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)rptyprocwriterGs rrWzTermSocket.stdin_to_ptyprocs+ == $ MM ! ! ' ' - %r)rr returnNone)N)r1z str | Nonermbool)rArrmrn)rHr$rmrn)rOrrmrn)rYr$rmrn)rmrn)r)rhr$rmrn)r! __module__ __qualname____doc__r*r2r7r@r<r coroutiner\rarerNrrW __classcell__)rDs@rrrsa* e) ':1> ]]//, 4  56.7.rr)rz str | bytesrmr$)rr __future__rrKrr&typingrrtornado.websockettornadortornado.concurrentrterminado.managementrr r websocketWebSocketHandlerrrrr~sI #  %.D z.""33z.r