o Zh@sdZddlmZddlZddlZddlmZddlmZmZm Z m Z m Z ddl Z ddl mZmZddlmZddlmZmZdd lmZd d gZGd d d e jZGd d d e jZdS)zI Utility for running a prompt_toolkit application in an asyncssh server. ) annotationsN)get_running_loop)AnyCallable CoroutineTextIOcast) AppSessioncreate_app_session)Size) PipeInputcreate_pipe_input) Vt100_OutputPromptToolkitSSHSessionPromptToolkitSSHServerc@s\eZdZd$ddZd%d d Zd&ddZd'ddZd(ddZd(ddZd)ddZ d*d!d"Z d#S)+rinteract>Callable[[PromptToolkitSSHSession], Coroutine[Any, Any, None]] enable_cprboolreturnNonecsN|_|_d_d_d_d_d_Gfddd}tt|_ dS)Ncs@eZdZdfdd Zddd Zdd d Zedfd d ZdS)z0PromptToolkitSSHSession.__init__..Stdoutdatastrrrcs@zjdurj|ddWdSWdStyYdSw)N z )_chanwritereplaceBrokenPipeError)srselfX/var/www/html/lang_env/lib/python3.10/site-packages/prompt_toolkit/contrib/ssh/server.pyr+s  z6PromptToolkitSSHSession.__init__..Stdout.writercSdSNTr!rr!r!r"isatty2z7PromptToolkitSSHSession.__init__..Stdout.isattycSsdSNr!r%r!r!r"flush5r'z6PromptToolkitSSHSession.__init__..Stdout.flushcs"jdusJtjjdS)Nr)rrZ _orig_chanZ get_encodingr%rr!r"encoding8sz9PromptToolkitSSHSession.__init__..Stdout.encodingN)rrrrrrrr)rr)__name__ __module__ __qualname__rr&r)propertyr*r!rr!r"Stdout*s   r1) rr interact_taskr app_session_input_outputrrstdout)r rrr1r!rr"__init__sz PromptToolkitSSHSession.__init__r cCs4|jdur tdddS|j\}}}}t||dS)zU Callable that returns the current `Size`, required by Vt100_Output. NO)rowscolumns)rr get_terminal_sizer widthheightpixwidth pixheightr!r!r" _get_size?s   z!PromptToolkitSSHSession._get_sizechanrcCs ||_dSr()r)r rCr!r!r"connection_madeIs z'PromptToolkitSSHSession.connection_madecCr#r$r!rr!r!r"shell_requestedLr'z'PromptToolkitSSHSession.shell_requestedcCst||_dSr()rZ create_task _interactr2rr!r!r"session_startedOsz'PromptToolkitSSHSession.session_startedc s.|jdur tdt|jdr|jjdur|jd|j}t|j|j||j d|_ t ]|_ t |j |j d;}||_z$z ||IdHWn tyVtYnwW|j|j n |j|j wWdn1sxwYWddSWddS1swYdS)Nz,`_interact` called before `connection_made`. set_line_modeF)termr)inputoutput)r ExceptionhasattrZ_editorrHZget_terminal_typerr6rBrr5r r4r r3r BaseException traceback print_excclose)r rIsessionr!r!r"rFRs4         "z!PromptToolkitSSHSession._interactr>intr?r@objectrAcCs&|jr|jjr|jjdSdSdSr()r3ZappZ _on_resizer=r!r!r"terminal_size_changednsz-PromptToolkitSSHSession.terminal_size_changedrrdatatypecCs|jdurdS|j|dSr()r4Z send_text)r rrVr!r!r" data_receivedus z%PromptToolkitSSHSession.data_receivedNrrrrrr)rr )rCrrrr+r,) r>rSr?rSr@rTrArTrr)rrrVrTrr) r-r.r/r7rBrDrErGrFrUrWr!r!r!r"rs  )    c@s4eZdZdZdddd d ZdddZdddZdS)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. TrrrrrrrcCs||_||_dSr()rr)r rrr!r!r"r7s zPromptToolkitSSHServer.__init__usernamercCr#)NFr!)r rZr!r!r" begin_authsz!PromptToolkitSSHServer.begin_authrcCst|j|jdS)NrY)rrrrr!r!r"session_requestedsz(PromptToolkitSSHServer.session_requestedNrX)rZrrr)rr)r-r.r/__doc__r7r[r\r!r!r!r"r}s ) )r] __future__rZasynciorOrtypingrrrrrZasyncsshZ"prompt_toolkit.application.currentr r Zprompt_toolkit.data_structuresr Zprompt_toolkit.inputr r Zprompt_toolkit.output.vt100r__all__ZSSHServerSessionrZ SSHServerrr!r!r!r"s    h