o i4@srdZddlmZddlZddlZddlZddlZddlmZddlm 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5dgZ6d$ddZ7d%ddZ8GdddZ9Gdd d Z:d&d!d"Z;Gd#ddZr6joinr@OSErrorrwarning)rBer.r.r/rJbs z_ConnectionStdout.flushcCs d|_dSrN)rArOr.r.r/closekr1z_ConnectionStdout.closecC|jSr-)r=rOr.r.r/r:nz_ConnectionStdout.encodingcCrWr-)r?rOr.r.r/rFrrXz_ConnectionStdout.errorsN)r2r3r:r;r+r4)rDr;r+r4)r+rMr+r4)r+r;) __name__ __module__ __qualname____doc__rCrLrPrJrVpropertyr:rFr.r.r.r/r9Ms      r9c@sdeZdZdZ d,d-ddZd.ddZd/ddZd.ddZd0d!d"Zd0d#d$Z d1d'd(Z d.d)d*Z d+S)2TelnetConnectionz6 Class that represents one Telnet connection. Tconnr3addrtuple[str, int]interact-Callable[[TelnetConnection], Awaitable[None]]serverr(r:r;styleBaseStyle | None vt100_inputr enable_cprrMr+r4c s|_|_|_|_|_|_d_t_ |_ _ d_ t ddd_t|dfdd ttt||d _dfd d } dfdd } dfdd } t| | | _d_dS)NF(Orowscolumnsr+rcsjSr-)sizer.rOr.r/get_sizesz+TelnetConnection.__init__..get_size)r:rDr,r4csj|dS)z-TelnetProtocolParser 'data_received' callbackN)rh send_bytesrDrOr.r/ data_receivedsz0TelnetConnection.__init__..data_receivedrmr*rncs:t||d_jdurjrjdddSdSdS)z-TelnetProtocolParser 'size_received' callbackrlNcSs tSr-)r _on_resizer.r.r.r/ zBTelnetConnection.__init__..size_received..)rro vt100_outputcontextrunrlrOr.r/ size_receivedsz0TelnetConnection.__init__..size_receivedttyper;cs"tj|d_jdS)z.TelnetProtocolParser 'ttype_received' callback)termriN)rstdoutrw_readyset)r{rirprBr.r/ttype_receiveds z1TelnetConnection.__init__..ttype_received)r+rrDr,r+r4)rmr*rnr*r+r4)r{r;r+r4)r`rarcrer:rfrAasyncioEventr~rhrirwrror8r r r9r}r'parserrx) rBr`rarcrer:rfrhrirsrzrr.rr/rC|s(   zTelnetConnection.__init__csdfdd }t}|j|z;jIdHtjjdt _ IdHWdn1s;wYW dSW dS w)z" Run application. r+r4cs:jd}|r|dStdjdS)Niz"Connection closed by client. %r %r)r`recvfeedrr5rarVrrrOr.r/handle_incoming_datas  z>TelnetConnection.run_application..handle_incoming_dataN)inputoutputrY) r add_readerr`r~waitr rhrw contextvars copy_contextrxrcrV)rBrloopr.rOr/run_applications   z TelnetConnection.run_applicationrDr,cCs|j|dS)zF Handler for incoming data. (Called by TelnetServer.) N)rrrKr.r.r/rszTelnetConnection.feedcCs@|jsd|_|jt|j|j|jdSdS)z# Closed by client. TN)rArhrVr remove_readerr`r}rOr.r.r/rVs  zTelnetConnection.closeformatted_textrcCs0|jdurdSt|}t|j||jptdS)z* Send text to the client. N)rwrrrfrrBrr.r.r/r6s zTelnetConnection.sendcstfddS)z\ Send text to the client. This is asynchronous, returns a `Future`. cs Sr-)r6r.rrBr.r/rurvz4TelnetConnection.send_above_prompt..)r_run_in_terminalrr.rr/send_above_promptsz"TelnetConnection.send_above_promptfuncCallable[[], None]cCs |jr |jt|dStd)Nz2Called _run_in_terminal outside `run_application`.)rxryr RuntimeError)rBrr.r.r/rsz!TelnetConnection._run_in_terminalcCs4|jdurdS|j|jdd|jdS)zB Erase the screen and move the cursor to the top. Nr)rw erase_screen cursor_gotorJrOr.r.r/rs  zTelnetConnection.erase_screenN)T)r`r3rarbrcrdrer(r:r;rfrgrhrrirMr+r4rYr)rrr+r4)rrr+r4) rZr[r\r]rCrrrVr6rrrr.r.r.r/r_ws 7   r_csdSr-r.r7r.r.r/_dummy_interactsrc@s`eZdZdZddedddfd$ddZed%ddZd&d'ddZd(ddZ d(dd Z d)d"d#Z 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-8NThostr;portr*rcrdr:rfrgrirMr+r4cCs<||_||_||_||_||_||_d|_g|_t|_ dSr-) rrrcr:rfri _run_task_application_tasksr connections)rBrrrcr:rfrir.r.r/rCs  zTelnetServer.__init__r3cCs>ttjtj}|tjtjd|||f|d|S)Nr)socketAF_INET SOCK_STREAM setsockopt SOL_SOCKET SO_REUSEADDRbindlisten)clsrrsr.r.r/_create_socket0s  zTelnetServer._create_socketready_cbCallable[[], None] | Nonec sjjtdjjtfdd|r#|z4tIdHWt  j D]}| q9t j dkrVtjj dtjdIdHdSdSt  j D]}| qet j dkrtjj dtjdIdHww)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.rBrr.r/ruFrvz"TelnetServer.run..Nr)timeout return_when)rrrrr5rrrFuturerrVrcancellenr ALL_COMPLETED)rBrtr.rr/ry:s@       zTelnetServer.runcCs$|jdurdSt||_dS)z} Deprecated: Use `.run()` instead. Start the telnet server (stop by calling and awaiting `stop()`). N)rr create_taskryrOr.r.r/startas zTelnetServer.startcsD|jdur |jz |jIdHWdStjyYdSwdS)z Deprecated: Use `.run()` instead. Stop a telnet server that was started using `.start()` and wait for the cancellation to complete. N)rrrCancelledErrorrOr.r.r/stopms  zTelnetServer.stop listen_socketcsP|\tjdgRdfdd }t|jdS)z1 Accept new incoming connection. zNew connection %r %rr+r4c szz]tP}tjjj|jd}j|tj dgRz| IdHWj |tj dgRnj |tj dgRwWdn1sYwYWn>t ymt dYn:t yzt dYn6ty}ztdt|j|ddl}|WYd}~n d}~wwWj dSWj dSWj dSWj dSj w)N)r:rfrhrizStarting interaction %r %rzStopping interaction %r %rz)Unhandled EOFError in telnet application.z2Unhandled KeyboardInterrupt in telnet application.zGot %sr)rr_rcr:rfriraddrr5rremoveEOFErrorKeyboardInterrupt BaseExceptionprinttyperZ traceback print_excr)rhr2rUrrar`rBtaskr.r/rysR      z!TelnetServer._accept..runNrY)acceptrr5rrrrH)rBrryr.rr/r{s &zTelnetServer._accept)rr;rr*rcrdr:r;rfrgrirMr+r4)rr;rr*r+r3r-)rrr+r4rY)rr3r+r4) rZr[r\r]rrC classmethodrryrrrr.r.r.r/r( s   ' )r)r*r+r,)r2r3r+r4)r2r_r+r4)=r] __future__rrrrsysrtypingrrrrrr r r r "prompt_toolkit.application.currentr r*prompt_toolkit.application.run_in_terminalrprompt_toolkit.data_structuresrprompt_toolkit.formatted_textrrprompt_toolkit.inputrrprompt_toolkit.output.vt100rprompt_toolkit.rendererrprompt_toolkit.stylesrrlogrprotocolrrrrrr r!r"r#r$r%r&r'__all__r0r8r9r_rr(r.r.r.r/s6  ,     <  *