hdZddlmZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ddl Z ddlmZmZddlmZddlmZmZdd lmZd d gZGd d e jZGd d e jZdS)zI Utility for running a prompt_toolkit application in an asyncssh server. ) annotationsN)get_running_loop)Any AwaitableCallableOptionalTextIOcast) AppSessioncreate_app_session)Size) PipeInputcreate_pipe_input) Vt100_OutputPromptToolkitSSHSessionPromptToolkitSSHServercNeZdZddZdd Zdd Zdd Zd dZd dZd!dZ d"dZ dS)#rinteract4Callable[[PromptToolkitSSHSession], Awaitable[None]] enable_cprboolreturnNonec|_|_d_d_d_d_d_Gfdd}tt|_ dS)NcHeZdZd fd Zd dZd dZedfd Zd S)0PromptToolkitSSHSession.__init__..Stdoutdatastrrrc j0j|dddSdS#t$rYdSwxYw)N z )_chanwritereplaceBrokenPipeError)srselfs ]/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/prompt_toolkit/contrib/ssh/server.pyr"z6PromptToolkitSSHSession.__init__..Stdout.write+sdz- ((dF)C)CDDDDD.-&DDs5< A  A rcdSNTr%s r'isattyz7PromptToolkitSSHSession.__init__..Stdout.isatty2stcdSNr*r+s r'flushz6PromptToolkitSSHSession.__init__..Stdout.flush5sr-cxjJtjjdS)Nr)r!r _orig_chan get_encoding)r%r&s r'encodingz9PromptToolkitSSHSession.__init__..Stdout.encoding8s5z---4:0==??BCCCr-N)rrrrrrrr)rr)__name__ __module__ __qualname__r"r,r0propertyr4r&sr'Stdoutr*s               D D D D DX D D Dr-r<) rr interact_taskr! app_session_input_outputr r stdout)r&rrr<s` r'__init__z PromptToolkitSSHSession.__init__s ! $8<!% .2 )- ,0  D D D D D D D D D D&66688,, r-r c|jtddS|j\}}}}t||S)zU Callable that returns the current `Size`, required by Vt100_Output. NO)rowscolumns)r!r get_terminal_sizer&widthheightpixwidth pixheights r' _get_sizez!PromptToolkitSSHSession._get_size?sN : R,,, ,151M1M1O1O .E68YVU333 3r-chanrc||_dSr/)r!)r&rOs r'connection_madez'PromptToolkitSSHSession.connection_madeIs  r-cdSr)r*r;s r'shell_requestedz'PromptToolkitSSHSession.shell_requestedLstr-cvt||_dSr/)r create_task _interactr=r;s r'session_startedz'PromptToolkitSSHSession.session_startedOs.-//;;DNNr BaseException traceback print_excclose)r&rZsessions r'rVz!PromptToolkitSSHSession._interactRs* : JKK K 4: / / ,DJ4F4R J $ $U + + +z++--# Kdt     (DK#$+dlKKK (w#* (------------$***')))))*J$$&&&K%%''''J$$&&&K%%''''' ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( ( (sl"!FF C('D>(DD>DD> 3F>4E22F6 FF F F FF!FrJintrKrLobjectrMcp|jr,|jjr"|jjdSdSdSr/)r>app _on_resizerIs r'terminal_size_changedz-PromptToolkitSSHSession.terminal_size_changednsN   . 0 4 .   + + - - - - - . . . .r-rrdatatypecL|jdS|j|dSr/)r? send_text)r&rrls r' data_receivedz%PromptToolkitSSHSession.data_receivedus+ ;  F d#####r-Nrrrrrr)rr )rOrrrr5r6) rJrfrKrfrLrgrMrgrr)rrrlrgrr) r7r8r9rBrNrQrSrWrVrkror*r-r'rrs'-'-'-'-R4444NNNN((((8....$$$$$$r-c0eZdZdZdddd Zdd ZddZdS)raZ Run a prompt_toolkit application over an asyncssh server. This takes one argument, an `interact` function, which is called for each connection. This should be an asynchronous function that runs the prompt_toolkit applications. This function runs in an `AppSession`, which means that we can have multiple UI interactions concurrently. Example usage: .. code:: python async def interact(ssh_session: PromptToolkitSSHSession) -> None: await yes_no_dialog("my title", "my text").run_async() prompt_session = PromptSession() text = await prompt_session.prompt_async("Type something: ") print_formatted_text('You said: ', text) server = PromptToolkitSSHServer(interact=interact) loop = get_running_loop() loop.run_until_complete( asyncssh.create_server( lambda: MySSHServer(interact), "", port, server_host_keys=["/etc/ssh/..."], ) ) loop.run_forever() :param enable_cpr: When `True`, the default, try to detect whether the SSH client runs in a terminal that responds to "cursor position requests". That way, we can properly determine how much space there is available for the UI (especially for drop down menus) to render. Trrrrrrrc"||_||_dSr/)rr)r&rrs r'rBzPromptToolkitSSHServer.__init__s ! $r-usernamercdS)NFr*)r&rts r' begin_authz!PromptToolkitSSHServer.begin_authsur-rc8t|j|jS)Nrr)rrrr;s r'session_requestedz(PromptToolkitSSHServer.session_requesteds&t}QQQQr-Nrp)rtrrr)rr)r7r8r9__doc__rBrvrxr*r-r'rr}sp##R %%%%%%RRRRRRr-)ry __future__rasynciorbrtypingrrrrr r asyncssh"prompt_toolkit.application.currentr r prompt_toolkit.data_structuresr prompt_toolkit.inputrrprompt_toolkit.output.vt100r__all__SSHServerSessionr SSHServerrr*r-r'rsY#"""""$$$$$$CCCCCCCCCCCCCCCCMMMMMMMM//////========444444 $&> ?e$e$e$e$e$h7e$e$e$P4R4R4R4R4RX/4R4R4R4R4Rr-