U bh1x@sddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ej dksZt e srddlZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZmZmZdd lmZddlm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddddddgZ5dZ6dZ7dZ8dZ9dZ:Gdd d e2Z;Gd!dde;ZGd%d&d&Z?e d d'd(d)d*dZ@e d d(d+d,dZAGd-ddZBGd.ddeBZCd/d0d1d2ZDdS)3) annotationsN)abstractmethod)get_running_loop)contextmanager)SPHINX_AUTODOC_RUNNINGwin32)windll)Arraybyrefpointer)DWORDHANDLE)CallableContextManagerIterableIteratorTextIOrun_in_executor_with_context)create_win32_eventwait_for_handles)KeyPress)Keys) MouseButtonMouseEventType) INPUT_RECORDKEY_EVENT_RECORDMOUSE_EVENT_RECORDSTD_INPUT_HANDLE EventTypes)REVERSE_ANSI_SEQUENCES)Input) Vt100Parser Win32InputConsoleInputReaderraw_mode cooked_modeattach_win32_inputdetach_win32_inputic@s4eZdZdZddddZeeddddZd S) _Win32InputBasez; Base class for `Win32Input` and `Win32PipeInput`. NonereturncCs t|_dSN) _Win32Handles win32_handlesselfr5>/tmp/pip-unpacked-wheel-9s95tbmv/prompt_toolkit/input/win32.py__init__Asz_Win32InputBase.__init__rcCsdSr0r5r3r5r5r6handleDsz_Win32InputBase.handleN)__name__ __module__ __qualname____doc__r7propertyrr8r5r5r5r6r,<s r,cseZdZdZd'dddfdd Zdd d d d Zd d ddZdd ddZdd ddZe dd ddZ d d ddZ d d ddZ dd ddZ dd d d!Zdd d"d#Ze d$d d%d&ZZS)(r%z< `Input` class that reads from the Windows console. Nz TextIO | Noner-)stdinr/cs2tt|_||jr&t|_nt|_dSr0)superr7_is_win_vt100_input_enabled_use_virtual_terminal_inputVt100ConsoleInputReaderconsole_input_readerr&)r4r> __class__r5r6r7Os   zWin32Input.__init__Callable[[], None]zContextManager[None])input_ready_callbackr/cCs t||S)zj Return a context manager that makes this input active in the current event loop. )r))r4rGr5r5r6attachZszWin32Input.attachr.cCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )r*r3r5r5r6detachaszWin32Input.detachlist[KeyPress]cCst|jSr0)listrCreadr3r5r5r6 read_keyshszWin32Input.read_keyscCsdSr0r5r3r5r5r6flushkszWin32Input.flushboolcCsdS)NFr5r3r5r5r6closednszWin32Input.closedcCs t|jdS)N) use_win10_virtual_terminal_input)r'rAr3r5r5r6r'rszWin32Input.raw_modecCstSr0)r(r3r5r5r6r(wszWin32Input.cooked_modeintcCs tjSr0)sysr>filenor3r5r5r6rTzszWin32Input.filenostrcCsdS)Nz win32-inputr5r3r5r5r6typeahead_hashszWin32Input.typeahead_hashcCs|jdSr0)rCcloser3r5r5r6rWszWin32Input.closercCs|jjSr0)rCr8r3r5r5r6r8szWin32Input.handle)N)r9r:r;r<r7rHrIrMrNr=rPr'r(rTrVrWr8 __classcell__r5r5rDr6r%Js c"@seZdZdZejejejejej ej ej ej ej ejejejejejejejejejejejejejejejejejejej ej!ej"ej#ej$ej%d!Z&ej'ej(ej)ej*ej+ej,ej-ej.ej/ej0ej1ej2ej3ej4ej5ej6ej7ej8ej9ej:ej;ejdZ?dZ@dZAdZBd.d d d d dZCd dddZDddddZEdddddZFddddddZGeHddd d!d"ZIeHdd d#d$d%ZJd&dd'd(d)ZKd*dd'd+d,ZLd-S)/r&z| :param recognize_paste: When True, try to discover paste actions and turn the event into a BracketedPaste. )!     )!"#$%&'(-.pqrstuvwxyz{rr!r+TrOr-)recognize_paster/cCsZd|_||_|tjr.ttjt |_ n(t dt j t jB|_tt|j|_ dS)NCONIN$)_fdconrrSr>isattyrr kernel32 GetStdHandlerr8osopenO_RDWRO_BINARYmsvcrt get_osfhandle)r4rr5r5r6r7s zConsoleInputReader.__init__r.cCs|jdk rt|jdSz Close fdcon.NrrrWr3r5r5r6rWs zConsoleInputReader.closeIterable[KeyPress]c #s2d}td}t|}|}tjgdds.dStjjt||t|t ||}fdd|D}t |}j r$ |r$t |}|D]}g}|rt|jtr|jtjtjhkr||jz t|}Wqtk rd}YqXq|rttjd|V|dk r|Vqn |EdHdS) Return a list of `KeyPress` instances. It won't return anything when there was nothing to read. (This function doesn't block.) http://msdn.microsoft.com/en-us/library/windows/desktop/ms684961(v=vs.85).aspx rtimeoutNcsg|]}|qSr5)_insert_key_data).0keyr3r5r6 sz+ConsoleInputReader.read..)r rrr8r rReadConsoleInputWr rK _get_keys_merge_paired_surrogatesr _is_pasteiter isinstancerrControlJControlMappenddatanext StopIterationrZBracketedPastejoin) r4 max_countrLarrtype input_recordsZall_keysgenkrr5r3r6rLsD     zConsoleInputReader.readr) key_pressr/cCs6|jr |St|jtr&t|jd}nd}t|j|S)z@ Insert KeyPress data, for vt100 compatibility. r)rrrrr"getr)r4rrr5r5r6r*s  z#ConsoleInputReader._insert_key_datar Array[INPUT_RECORD]zIterator[KeyPress]rLrr/ccstt|jD]d}||}|jtkr t|jt|j}t|trT|jrT| |EdHq t|t r | |EdHq dS)R Generator that yields `KeyPress` objects from the input records. N) rangevalue EventTyper getattrEventrrKeyDown_event_to_key_pressesr _handle_mouse)r4rLriirevr5r5r6r8s  zConsoleInputReader._get_keysrJ) key_pressesr/ccsd}|D]}t|jt }|o6d|jko2dkn}|oTd|jkoPdkn}|r|r|j|jddd}t||}n|Vd}|r|}q|Vq|r|VdS)zm Combines consecutive KeyPresses with high and low surrogates into single characters Nuuuuz utf-16-le surrogatepass)rrrencodedecoder)rZbuffered_high_surrogaterZis_textZis_high_surrogateZis_low_surrogateZfullcharr5r5r6rPs,  z+ConsoleInputReader._merge_paired_surrogates)keysr/cCsJd}d}|D],}t|jts$|d7}|jtjkr |d7}q |dkoH|dkS)a< Return `True` when we should consider this list of keys as a paste event. Pasted text on windows will be turned into a `Keys.BracketedPaste` event. (It's not 100% correct, but it is probably the best possible way to detect pasting of text and handle that correctly.) rr!)rrrr)rZ text_countZ newline_countrr5r5r6rqs    zConsoleInputReader._is_paster)rr/cCst|tr|jstd}|j}|jj}|dd}|dkrZ|j|j krt |j |jd}n:||j kr|j |t j krxd}t |j ||}n t ||}||j@s||j@r||j@r|rt jt jt jt jt jt jt jt jt jt jt jt jt jt jt jt j t j!t j"i }|#|j$|j$|_$||j@s2||j@r|rt jt j%t jt j&t jt j't jt j(t jt j)t jt j*t jt j+t j,t j-t jt j.t j!t j/i }|#|j$|j$|_$||j@r|rt j0t j1t jt j2t jt j3t jt j4t jt j5t jt j6t jt j7t jt j8t j,t j9t jt j:t j!t j;i }|#|j$|j$|_$||j@s6||j@rT|rT|jd|gS|r||j?@}|rt t j>d|gS|gSngSdS)zU For this `KEY_EVENT_RECORD`, return a list of `KeyPress` instances. Nzutf-8rr  )@rrrAssertionErrorZControlKeyStateuChar UnicodeCharrZVirtualKeyCodekeycodesrmappingsrrLEFT_CTRL_PRESSEDRIGHT_CTRL_PRESSED SHIFT_PRESSEDLeftZControlShiftLeftRightZControlShiftRightUpZControlShiftUpDownZControlShiftDownHomeZControlShiftHomeEndZControlShiftEndInsertZControlShiftInsertPageUpZControlShiftPageUpPageDownZControlShiftPageDownrrZ ControlLeftZ ControlRightZ ControlUpZ ControlDownZ ControlHomeZ ControlEndZ ControlInsertDeleteZ ControlDeleteZ ControlPageUpZControlPageDownZTabZBackTabZ ShiftLeftZ ShiftRightZShiftUpZ ShiftDownZ ShiftHomeZShiftEndZ ShiftInsertZ ShiftDeleteZ ShiftPageUpZ ShiftPageDownr ControlSpaceEscapeLEFT_ALT_PRESSED)r4rresultZcontrol_key_stateu_charZ ascii_charmappingZ meta_pressedr5r5r6rs            z(ConsoleInputReader._event_to_key_pressesrcCs|j}|j}d}tj}|t@r6|dkr.tj}qTtj}n|tkrFtj }n|t krTtj }|t @rbtj }|dkr|dkrztj}ntj}d|j|jt|jjt|jjg}ttj|gS)zK Handle mouse events. Return a list of KeyPress instances. Nr;)Z EventFlagsZ ButtonStaterNONE MOUSE_WHEELEDrZ SCROLL_UPZ SCROLL_DOWNFROM_LEFT_1ST_BUTTON_PRESSEDLEFTRIGHTMOST_BUTTON_PRESSEDRIGHT MOUSE_MOVEDZ MOUSE_MOVEZ MOUSE_DOWNZMOUSE_UPrrrUZ MousePositionXYrrZWindowsMouseEvent)r4rZ event_flagsZ button_stateZ event_typebuttonrr5r5r6r s4  z ConsoleInputReader._handle_mouseN)T)Mr9r:r;r<rrrZControlAZControlBZControlCZControlDZControlEZControlFZControlGZControlHZControlIrZControlKZControlLrZControlNZControlOZControlPZControlQZControlRZControlSZControlTZControlUZControlVZControlWZControlXZControlYZControlZZControlBackslashZControlSquareCloseZControlCircumflexZControlUnderscoreZ BackspacerrrrrrrrrrrZF1ZF2ZF3ZF4ZF5ZF6ZF7ZF8ZF9ZF10ZF11ZF12rrZRIGHT_ALT_PRESSEDrrrr7rWrLrr staticmethodrrrrr5r5r5r6r&s' > c@sLeZdZdZddddZddddZddd d Zd d d dddZdS)rBa Similar to `ConsoleInputReader`, but for usage when `ENABLE_VIRTUAL_TERMINAL_INPUT` is enabled. This assumes that Windows sends us the right vt100 escape sequences and we parse those with our vt100 parser. (Using this instead of `ConsoleInputReader` results in the "data" attribute from the `KeyPress` instances to be more correct in edge cases, because this responds to for instance the terminal being in application cursor keys mode.) r-r.csld_g_tfdd_tjr@ttj t _ n(t dt jt jB_ttj_ dS)Ncs j|Sr0)_bufferr)rr3r5r6Mz2Vt100ConsoleInputReader.__init__..r)rrr$ _vt100_parserrSr>rrr rrrr8rrrrrrr3r5r3r6r7Hs  z Vt100ConsoleInputReader.__init__cCs|jdk rt|jdSrrr3r5r5r6rWYs zVt100ConsoleInputReader.closercCsxd}td}t|}|}t|jgdds.gStj|jt||t||||D]}|j |qV|j }g|_ |S)rrrr) r rrr8r rrr rrfeedr)r4rrLrrZkey_datarr5r5r6rL^s"zVt100ConsoleInputReader.readr rz Iterator[str]rccs^t|jD]N}||}|jtkr t|jt|j}t|tr |jr |j j }|dkr |Vq dS)rrN) rrrr rrrrrrr)r4rLrrrrrr5r5r6rs z!Vt100ConsoleInputReader._get_keysN)r9r:r;r<r7rWrLrr5r5r5r6rB;s  %rBc@s@eZdZdZddddZddddd d Zdd d d dZdS)r1a Utility to keep track of which handles are connectod to which callbacks. `add_win32_handle` starts a tiny event loop in another thread which waits for the Win32 handle to become ready. When this happens, the callback will be called in the current asyncio event loop using `call_soon_threadsafe`. `remove_win32_handle` will stop this tiny event loop. NOTE: We use this technique, so that we don't have to use the `ProactorEventLoop` on Windows and we can wait for things like stdin in a `SelectorEventLoop`. This is important, because our inputhook mechanism (used by IPython), only works with the `SelectorEventLoop`. r-r.cCsi|_i|_dSr0)_handle_callbacks_remove_eventsr3r5r5r6r7sz_Win32Handles.__init__rrF)r8callbackr/cs~j}|dkrtd|t|j|<t|j|<ddfdd ddfdd td dS) z7 Add a Win32 handle to the event loop. NInvalid handle.r-r.c sz W5tdXdS)Nlooprr5)rrwaitr5r6readys z-_Win32Handles.add_win32_handle..readycs2tg}|kr$tjdSdSr0)rr rZ CloseHandleZcall_soon_threadsafe)r)r8rr remove_eventr5r6rs   z,_Win32Handles.add_win32_handle..waitr)r ValueErrorremove_win32_handlerrrrr)r4r8rZ handle_valuer5)rr8rrrrr6add_win32_handles     z_Win32Handles.add_win32_handlezCallable[[], None] | None)r8r/cCsn|jdkrdSz|j|j}Wntk r4YnXtj|z|j|jWStk rhYdSXdS)zt Remove a Win32 handle from the event loop. Return either the registered handler or `None`. N)rrpopKeyErrorr rZSetEventr)r4r8eventr5r5r6rs  z!_Win32Handles.remove_win32_handleN)r9r:r;r<r7rrr5r5r5r6r1s+r1rFzIterator[None])inputrr/c cs`|j}|j}|jdkrtd||}|||z dVW5|||rZ|||XdS)z Context manager that makes this input active in the current event loop. :param input: :class:`~prompt_toolkit.input.Input` object. :param input_ready_callback: Called when the input is ready to read. Nrr2r8rrrr)r rr2r8previous_callbackr5r5r6r)s      )r r/c csJ|j}|j}|jdkrtd||}z dVW5|rD|||XdS)Nrr )r r2r8rr5r5r6r*s   c@sPeZdZdZddddddd Zdd d d Zdd d dZdddddZdS)r'z :: with raw_mode(stdin): ''' the windows terminal is now in 'raw' mode. ''' The ``fileno`` attribute is ignored. This is to be compatible with the `raw_input` method of `.vt100_input`. NFz int | NonerOr-)rTrQr/cCsttjt|_||_dSr0)rr rrrr8rQ)r4rTrQr5r5r6r7,szraw_mode.__init__r.cCs,t}tj|jt|||_|dSr0)r r rGetConsoleModer8r original_mode_patch)r4rr5r5r6 __enter__2szraw_mode.__enter__cCsDd}d}d}|jj||B|B@}|jr0|tO}tj|j|dSNr+rr!)rrrQENABLE_VIRTUAL_TERMINAL_INPUTr rSetConsoleModer8)r4ENABLE_ECHO_INPUTENABLE_LINE_INPUTENABLE_PROCESSED_INPUTZnew_moder5r5r6r:s zraw_mode._patchobject)ar/cGstj|j|jdSr0)r rrr8r)r4rr5r5r6__exit__Iszraw_mode.__exit__)NF)r9r:r;r<r7rrrr5r5r5r6r'!s c@seZdZdZddddZdS)r(zx :: with cooked_mode(stdin): ''' The pseudo-terminal stdin is now used in cooked mode. ''' r-r.cCs0d}d}d}tj|j|jj||B|BBdSr)r rrr8rr)r4rrrr5r5r6rVs zcooked_mode._patchN)r9r:r;r<rr5r5r5r6r(NsrOr.c Cs^ttjt}td}tj|t|ztj|tt }|dkWStj||XdS)z_ Returns True when we're running Windows and VT100 escape sequences are supported. rr!N) rr rrrr rr rr)Zhconsolerrr5r5r6r@cs r@)E __future__rrrSabcrZasyncior contextlibrutilsrplatformrrctypesr r r r Zctypes.wintypesr rtypingrrrrrZprompt_toolkit.eventlooprZprompt_toolkit.eventloop.win32rrZ(prompt_toolkit.key_binding.key_processorrZprompt_toolkit.keysrZprompt_toolkit.mouse_eventsrrZprompt_toolkit.win32_typesrrrrr Zansi_escape_sequencesr"baser#Z vt100_parserr$__all__rrrrrr,r%r&rBr1r)r*r'r(r@r5r5r5r6sb             B2`X-