o whLd@sddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ej dks-Je s9ddl Z dd lmZdd lmZmZdd lmZmZdd lmZmZmZmZmZdd lmZddlmZmZddl m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.ddl/m0Z0gdZ1dZ2dZ3dZ4dZ5Gddde0Z6Gddde6Z7GdddZ8GdddZ9e d-d%d&Z:e d.d'd(Z;Gd)d*d*Z __class__r2r3r4Ks  zWin32Input.__init__input_ready_callbackCallable[[], None]ContextManager[None]cCs t||S)zj Return a context manager that makes this input active in the current event loop. )r')r1rDr2r2r3attachO zWin32Input.attachcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )r(r0r2r2r3detachVszWin32Input.detachlist[KeyPress]cCst|jSr-)listrAreadr0r2r2r3 read_keys]zWin32Input.read_keyscCr5r-r2r0r2r2r3flush`zWin32Input.flushboolcCdS)NFr2r0r2r2r3closedcszWin32Input.closedcCtSr-)r%r0r2r2r3r%gzWin32Input.raw_modecCrTr-)r&r0r2r2r3r&jrUzWin32Input.cooked_modeintcCs tjSr-)sysr>filenor0r2r2r3rXmrHzWin32Input.filenostrcCrR)Nz win32-inputr2r0r2r2r3typeahead_hashtrPzWin32Input.typeahead_hashcCs|jdSr-)rAcloser0r2r2r3r[wrNzWin32Input.closer cCs|jjSr-)rAr6r0r2r2r3r6zszWin32Input.handler-)r>r?r+r,)rDrEr+rF)r+rF)r+rJr7)r+rQ)r+rV)r+rYr8)r9r:r;r<r4rGrIrMrOr=rSr%r&rXrZr[r6 __classcell__r2r2rBr3r#Fs           r#c@s>eZdZdZidejdejdejdejdej dej dej d ej d ej d ejd ejd ejdejdejdejdejdejidejdejdejdejdejdejdejdejdejdejdej dej!dej"d ej#d!ej$d"ej%Z&id#ej'd$ej(d%ej)d&ej*d'ej+d(ej,d)ej-d*ej.d+ej/d,ej0d-ej1d.ej2d/ej3d0ej4d1ej5d2ej6d3ej7ej8ej9ej:ej;ejd6Z?d7Z@d8ZAd9ZBd`dad?d@ZCdbdAdBZDdcdDdEZEdddHdIZFdedOdPZGeHdfdSdTZIeHdgdVdWZJdhdZd[ZKdid]d^ZLd_S)jr$z| :param recognize_paste: When True, try to discover paste actions and turn the event into a BracketedPaste.      !"#$%&'(-.pqrstuv)wxyz{rr r)Trecognize_pasterQr+r,cCs\d|_||_|tjrttjt |_ dSt dt j t jB|_tt|j|_ dS)NzCONIN$)_fdconrrWr>isattyr r kernel32 GetStdHandlerr6osopenO_RDWRO_BINARYmsvcrt get_osfhandle)r1rr2r2r3r4s zConsoleInputReader.__init__cCs|jdur t|jdSdS)z Close fdcon.N)rrr[r0r2r2r3r[s zConsoleInputReader.closeIterable[KeyPress]c #sNd}td}t|}|}tjgddsdStjjt||t|t ||}fdd|D}t |}j r |rt |}|D]Q}g}|rt|jtra|jtjtjhvr||jzt|}Wn tyxd}Ynw|rt|jtra|jtjtjhvsa|rttjd|V|dur|VqLdS|EdHdS)z 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 ir)timeoutNcsg|]}|qSr2)_insert_key_data).0keyr0r2r3 sz+ConsoleInputReader.read..)r rrr6r rReadConsoleInputWr rK _get_keys_merge_paired_surrogatesr _is_pasteiter isinstancerrControlJControlMappenddatanext StopIterationrBracketedPastejoin) r1 max_countrLarrtype input_recordsall_keysgenkrr2r0r3rLsL     zConsoleInputReader.read key_pressrcCs6|jr|St|jtrt|jd}nd}t|j|S)z@ Insert KeyPress data, for vt100 compatibility. r)rrrrr!getr)r1rrr2r2r3rs   z#ConsoleInputReader._insert_key_datarLr rArray[INPUT_RECORD]Iterator[KeyPress]ccsvt|jD]2}||}|jtvr8t|jt|j}t|tr+|jr+| |EdHqt|t r8| |EdHqdS)zR Generator that yields `KeyPress` objects from the input records. N) rangevalue EventTypergetattrEventrrKeyDown_event_to_key_pressesr _handle_mouse)r1rLriirevr2r2r3r+s  zConsoleInputReader._get_keys key_pressesrJccsd}|D]K}t|jt }|od|jkodkn}|o+d|jko)dkn}|rH|rC|j|jddd}t||}n|Vd}|rM|}q|Vq|rX|VdSdS)zm Combines consecutive KeyPresses with high and low surrogates into single characters Nuuuuz utf-16-le surrogatepass)rrrencodedecoder)rbuffered_high_surrogateris_textis_high_surrogateis_low_surrogatefullcharr2r2r3rCs*   z+ConsoleInputReader._merge_paired_surrogateskeyscCsJd}d}|D]}t|jts|d7}|jtjkr|d7}q|dko$|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)r text_count newline_countrr2r2r3rds   zConsoleInputReader._is_pasterrcCst|tr|js Jd}|j}|jj}|dd}|dkr-|j|jvr,t |j|jd}n||j vrE|j |t j kr@}|rUt t j=d|gS|gSgS)zU For this `KEY_EVENT_RECORD`, return a list of `KeyPress` instances. Nzutf-8rr  )?rrrControlKeyStateuChar UnicodeCharrVirtualKeyCodekeycodesrmappingsrrLEFT_CTRL_PRESSEDRIGHT_CTRL_PRESSED SHIFT_PRESSEDLeftControlShiftLeftRightControlShiftRightUpControlShiftUpDownControlShiftDownHomeControlShiftHomeEndControlShiftEndInsertControlShiftInsertPageUpControlShiftPageUpPageDownControlShiftPageDownrr ControlLeft ControlRight ControlUp ControlDown ControlHome ControlEnd ControlInsertDelete ControlDelete ControlPageUpControlPageDownTabBackTab ShiftLeft ShiftRightShiftUp ShiftDown ShiftHomeShiftEnd ShiftInsert ShiftDelete ShiftPageUp ShiftPageDownr ControlSpaceEscapeLEFT_ALT_PRESSED)r1rresultcontrol_key_stateu_char ascii_charmapping meta_pressedr2r2r3rzs           z(ConsoleInputReader._event_to_key_pressesrcCs|j}|j}d}tj}|t@r|dkrtj}ntj}n|tkr#tj }n|t kr*tj }|t @r1tj }|dur@|dkr=tj}ntj}d|j|jt|jjt|jjg}ttj|gS)zK Handle mouse events. Return a list of KeyPress instances. Nr;) EventFlags ButtonStaterNONE MOUSE_WHEELEDr SCROLL_UP SCROLL_DOWNFROM_LEFT_1ST_BUTTON_PRESSEDLEFTRIGHTMOST_BUTTON_PRESSEDRIGHT MOUSE_MOVED MOUSE_MOVE MOUSE_DOWNMOUSE_UPrrrY MousePositionXYrrWindowsMouseEvent)r1r event_flags button_state event_typebuttonrr2r2r3rs4  z ConsoleInputReader._handle_mouseN)T)rrQr+r,r7)r+r)rrr+r)rLr rrr+r)rrJr+r)rrJr+rQ)rrr+rJ)rrr+rJ)Mr9r:r;r<rr r ControlAControlBControlCControlDControlEControlFControlGControlHControlIrControlKControlLrControlNControlOControlPControlQControlRControlSControlTControlUControlVControlWControlXControlYControlZControlBackslashControlSquareCloseControlCircumflexControlUnderscore BackspacerrrrrrrrrrrF1F2F3F4F5F6F7F8F9F10F11F12rrRIGHT_ALT_PRESSEDrrrr4r[rLrr staticmethodrrrrr2r2r2r3r$s      !%       >    r$c@s.eZdZdZdddZdd d Zdd dZdS)r.a 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|_dSr-)_handle_callbacks_remove_eventsr0r2r2r3r4>s z_Win32Handles.__init__r6r callbackrEcsvj}|dur td|t|j|<t|j|<d fdd d fdd td dS) z7 Add a Win32 handle to the event loop. NInvalid handle.r+r,c s(z WtddStdw)Nlooprr2)rYr\waitr2r3readyYsz-_Win32Handles.add_win32_handle..readycs2tg}|urtjdSdSr-)rr r CloseHandlecall_soon_threadsafe)r)r6r\r^ remove_eventr2r3r]cs  z,_Win32Handles.add_win32_handle..waitr[r7)r ValueErrorremove_win32_handlerrWrrXr)r1r6rY handle_valuer2)rYr6r\r^rar]r3add_win32_handleEs     z_Win32Handles.add_win32_handleCallable[[], None] | NonecCsf|jdurdSz |j|j}Wn tyYnwtj|z|j|jWSty2YdSw)zt Remove a Win32 handle from the event loop. Return either the registered handler or `None`. N)rrXpopKeyErrorr rSetEventrW)r1r6eventr2r2r3rcps    z!_Win32Handles.remove_win32_handleNr7)r6r rYrEr+r,)r6r r+rf)r9r:r;r<r4rercr2r2r2r3r..s   +r.inputrYrEr+Iterator[None]c cs|j}|j}|jdurtd||}|||zdVW|||r/|||dSdS|||r?|||ww)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. NrZr/r6rrbrcre)rkrYr/r6previous_callbackr2r2r3r's"      r'c cs`|j}|j}|jdurtd||}zdVW|r$|||dSdS|r/|||ww)NrZrm)rkr/r6rnr2r2r3r(s  r(c@s:eZdZdZddddZdd d Zdd d ZdddZdS)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`. NrX int | Noner+r,cCsttjt|_dSr-)r r rrrr6)r1rXr2r2r3r4szraw_mode.__init__cCs,t}tj|jt|||_|dSr-)r r rGetConsoleModer6r original_mode_patch)r1rqr2r2r3 __enter__s zraw_mode.__enter__cCs2d}d}d}tj|j|jj||B|B@dSNr)rr r rSetConsoleModer6rqrr1ENABLE_ECHO_INPUTENABLE_LINE_INPUTENABLE_PROCESSED_INPUTr2r2r3rrs zraw_mode._patchaobjectcGstj|j|jdSr-)r rrvr6rq)r1r{r2r2r3__exit__szraw_mode.__exit__r-)rXror+r,r7)r{r|r+r,)r9r:r;r<r4rsrrr}r2r2r2r3r%s    r%c@seZdZdZdddZdS)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|BBdSrtrurwr2r2r3rrs zcooked_mode._patchNr7)r9r:r;r<rrr2r2r2r3r&sr&)rkr*rYrEr+rl)rkr*r+rl)> __future__rrrWabcrasyncior contextlibrutilsrplatformrctypesr r r ctypes.wintypesr r typingrrrrrprompt_toolkit.eventlooprprompt_toolkit.eventloop.win32rr(prompt_toolkit.key_binding.key_processorrprompt_toolkit.keysrprompt_toolkit.mouse_eventsrrprompt_toolkit.win32_typesrrrrransi_escape_sequencesr!baser"__all__rrr rr*r#r$r.r'r(r%r&r2r2r2r3sP            92X  '