U d4@sdZddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZddlmZdd lmZmZdd lmZmZdd lm Z dd l!m"Z"dd l#m$Z$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5dgZ6dddddZ7dddddZ8GdddZ9GdddZ:dddd d!Z;Gd"ddZflushrArDr.r.r/writeZs z_ConnectionStdout.writeboolr,cCsdSNTr.rAr.r.r/isatty_sz_ConnectionStdout.isattyc CsZz|js|jd|jWn0tk rN}ztd|W5d}~XYnXg|_dS)Nz"Couldn't send data over socket: %s)r@r=r5joinr?OSErrorrwarning)rAer.r.r/rJbs  z_ConnectionStdout.flushcCs d|_dSrO)r@rPr.r.r/closeksz_ConnectionStdout.closecCs|jSr-)r<rPr.r.r/r:nsz_ConnectionStdout.encodingcCs|jSr-)r>rPr.r.r/rFrsz_ConnectionStdout.errorsN) __name__ __module__ __qualname____doc__rBrLrQrJrWpropertyr:rFr.r.r.r/r8Ms r8c @seZdZdZd%ddddddd d d d d dZd dddZdd dddZd dddZdd dddZdd dddZ dd dd d!Z d dd"d#Z d$S)&TelnetConnectionz6 Class that represents one Telnet connection. Tr1ztuple[str, int]-Callable[[TelnetConnection], Awaitable[None]]r(r9BaseStyle | NonerrMr2) connaddrinteractserverr:style vt100_input enable_cprr,c s|_|_|_|_|_|_d_t_ |_ _ d_ t ddd_t|ddfdd ttt||d _d d d fd d } ddd dfdd } dd dfdd } t| | | _d_dS)NF(OrowscolumnsrrNcsjSr-)sizer.rPr.r/get_sizesz+TelnetConnection.__init__..get_size)r:r*r2rCcsj|dS)z-TelnetProtocolParser 'data_received' callbackN)reZ send_bytesrDrPr.r/ data_receivedsz0TelnetConnection.__init__..data_receivedr))rjrkr,cs2t||d_jdk r.jr.jdddS)z-TelnetProtocolParser 'size_received' callbackriNcSs tSr-)rZ _on_resizer.r.r.r/rRzBTelnetConnection.__init__..size_received..)rrl vt100_outputcontextrunrirPr.r/ size_receivedsz0TelnetConnection.__init__..size_receivedr9)ttyper,cs"tj|d_jdS)z.TelnetProtocolParser 'ttype_received' callback)ZtermrfN)rstdoutrq_readyset)rurfrmrAr.r/ttype_receivedsz1TelnetConnection.__init__..ttype_received)r`rarbrcr:rdr@asyncioEventrwrerfrqrrlr7r r r8rvr'parserrr) rAr`rarbrcr:rdrerfrortrzr.ryr/rB|s(  zTelnetConnection.__init__rNc sddfdd }t}|j|zJjIdHtjjd t _ IdHW5QRXW5XdS)z" Run application. r2rNcs8jd}|r|ntdjdS)Niz"Connection closed by client. %r %r)r`recvfeedrr4rarWrnrPr.r/handle_incoming_datas   z>TelnetConnection.run_application..handle_incoming_dataN)inputoutput) r add_readerr`rWrwwaitr rerq contextvarsZ copy_contextrrrb)rArZloopr.rPr/run_applications  z TelnetConnection.run_applicationr*rCcCs|j|dS)zF Handler for incoming data. (Called by TelnetServer.) N)r}rrKr.r.r/rszTelnetConnection.feedcCs<|js8d|_|jt|j|j|jdS)z# Closed by client. TN)r@rerWr remove_readerr`rvrPr.r.r/rWs   zTelnetConnection.closer)formatted_textr,cCs0|jdkrdSt|}t|j||jp(tdS)z* Send text to the client. N)rqrrrdrrArr.r.r/r5s  zTelnetConnection.sendcstfddS)z\ Send text to the client. This is asynchronous, returns a `Future`. cs Sr-)r5r.rrAr.r/rprRz4TelnetConnection.send_above_prompt..)r_run_in_terminalrr.rr/send_above_promptsz"TelnetConnection.send_above_promptzCallable[[], None])funcr,cCs"|jr|jt|ntddS)Nz2Called _run_in_terminal outside `run_application`.)rrrsr RuntimeError)rArr.r.r/rsz!TelnetConnection._run_in_terminalcCs4|jdkrdS|j|jdd|jdS)zB Erase the screen and move the cursor to the top. Nr)rq erase_screenZ cursor_gotorJrPr.r.r/rs   zTelnetConnection.erase_screenN)T) rXrYrZr[rBrrrWr5rrrr.r.r.r/r]ws 7  r]csdSr-r.r6r.r.r/_dummy_interactsrc @seZdZdZddedddfddd dd d d d ddZeddddddZd dd dddZd dddZ d dddZ dd dddZ dS)!r(a Telnet server implementation. Example:: async def interact(connection): connection.send("Welcome") session = PromptSession() result = await session.prompt_async(message="Say something: ") connection.send(f"You said: {result} ") async def main(): server = TelnetServer(interact=interact, port=2323) await server.run() z 127.0.0.1zutf-8NTr9r)r^r_rMr2)hostportrbr:rdrfr,cCs<||_||_||_||_||_||_d|_g|_t|_ dSr-) rrrbr:rdrf _run_task_application_tasksrx connections)rArrrbr:rdrfr.r.r/rBs zTelnetServer.__init__r1)rrr,cCs>ttjtj}|tjtjd|||f|d|S)Nr)socketAF_INET SOCK_STREAM setsockopt SOL_SOCKET SO_REUSEADDRbindlisten)clsrrsr.r.r/_create_socket0s  zTelnetServer._create_socketzCallable[[], None] | None)ready_cbr,c sjjtdjjtfdd|rD|zt IdHW5tj D] }| qrt j dkrt j j dt jdIdHXdS)z Run the telnet server, until this gets cancelled. :param ready_cb: Callback that will be called at the point that we're actually listening. z.Listening for telnet connections on %s port %rcs Sr-)_acceptr.rArr.r/rpFrRz"TelnetServer.run..rN)timeoutZ return_when)rrrrr4rrrrWrcancellenr{rZ ALL_COMPLETEDZFuture)rArtr.rr/rs:s*   zTelnetServer.runrNcCs$|jdk rdSt||_dS)z} Deprecated: Use `.run()` instead. Start the telnet server (stop by calling and awaiting `stop()`). N)rr create_taskrsrPr.r.r/startas zTelnetServer.startcs@|jdk r<|jz|jIdHWntjk r:YnXdS)z Deprecated: Use `.run()` instead. Stop a telnet server that was started using `.start()` and wait for the cancellation to complete. N)rrr{ZCancelledErrorrPr.r.r/stopms   zTelnetServer.stop) listen_socketr,csP|\tjdddfdd }t|jdS)z1 Accept new incoming connection. New connection %r %rr2rNc szztp}tjjj|jd}j |t j dz| IdHW5j|t j d XW5QRXWnt k rt dYndtk rt dYnHtk r}z(tdt|j|ddl}|W5d}~XYnXW5jXdS) N)r:rdrerfStarting interaction %r %rStopping interaction %r %rz)Unhandled EOFError in telnet application.z2Unhandled KeyboardInterrupt in telnet application.zGot %sr)r)r)rremoverr]rbr:rdrfraddrr4rEOFErrorKeyboardInterrupt BaseExceptionprinttyperX traceback print_exc)rer3rVrrar`rAZtaskr.r/rss6  z!TelnetServer._accept..runN)r)acceptrr4rrrrH)rArrsr.rr/r{s  &zTelnetServer._accept)N) rXrYrZr[rrB classmethodrrsrrrr.r.r.r/r( s ' )=r[ __future__rr{rrsysrtypingrrrrrr r r r Z"prompt_toolkit.application.currentr rZ*prompt_toolkit.application.run_in_terminalrZprompt_toolkit.data_structuresrZprompt_toolkit.formatted_textrrZprompt_toolkit.inputrrZprompt_toolkit.output.vt100rZprompt_toolkit.rendererrZprompt_toolkit.stylesrrlogrprotocolrrrrrr r!r"r#r$r%r&r'__all__r0r7r8r]rr(r.r.r.r/s4  ,     <*