o ˷eyd@sddlZddlZddlmZddlmZddlmZddlm Z ej dks'Je s3ddl Z ddl m Z dd l mZmZdd lmZmZdd lmZmZmZmZmZmZmZmZdd lmZdd lmZmZddl 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/m0Z0gdZ1dZ2dZ3dZ4dZ5Gddde0Z6Gddde6Z7GdddZ8GdddZ9ede6d egdfd!edfd"d#Z:ede6d!edfd$d%Z;Gd&d'd'Z __class__r3r4r5Ss  zWin32Input.__init__input_ready_callbackcCs t||S)zj Return a context manager that makes this input active in the current event loop. )r))r2rCr3r3r4attachW zWin32Input.attachcCst|S)z{ Return a context manager that makes sure that this input is not active in the current event loop. )r*r1r3r3r4detach^szWin32Input.detachcCst|jSr.)listr@readr1r3r3r4 read_keysezWin32Input.read_keyscCr6r.r3r1r3r3r4flushhzWin32Input.flushcCdS)NFr3r1r3r3r4closedkszWin32Input.closedcCtSr.)r'r1r3r3r4r'ozWin32Input.raw_modecCrOr.)r(r1r3r3r4r(rrPzWin32Input.cooked_modecCs tjSr.)sysr>filenor1r3r3r4rRurEzWin32Input.filenocCrM)Nz win32-inputr3r1r3r3r4typeahead_hash|rLzWin32Input.typeahead_hashcCs|jdSr.)r@closer1r3r3r4rTrJzWin32Input.closecCs|jjSr.)r@r7r1r3r3r4r7szWin32Input.handler.r8)r9r:r;r<rrr5r rrDrFrrrIrKr=boolrNr'r(intrRstrrSrTr r7 __classcell__r3r3rAr4r%Ns   r%c@seZdZdZidejdejdejdejdej 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@d8ZAd9ZBdWd;eCdd?ZDdXd@dAZEdisattyr rkernel32 GetStdHandler r7osopenO_RDWRO_BINARYmsvcrt get_osfhandle)r2rr3r3r4r5s zConsoleInputReader.__init__cCs|jdur t|jdSdS)z Close fdcon.N)rrrTr1r3r3r4rTs zConsoleInputReader.closec #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|]}|qSr3)_insert_key_data).0keyr1r3r4 sz+ConsoleInputReader.read..)r rrr7rrReadConsoleInputWr rG _get_keys_merge_paired_surrogatesr _is_pasteiter isinstancerrControlJControlMappenddatanext StopIterationrBracketedPastejoin) r2 max_countrHarrtype input_recordsall_keysgenkrr3r1r4rHsL     zConsoleInputReader.read key_presscCs6|jr|St|jtrt|jd}nd}t|j|S)z@ Insert KeyPress data, for vt100 compatibility. r)rrrrr#getr)r2rrr3r3r4r%s   z#ConsoleInputReader._insert_key_datarHrzArray[INPUT_RECORD]ccszt|jD]4}||}|jtvr:t|jt|j}t|tkr,|jr,| |EdHqt|t kr:| |EdHqdS)zR Generator that yields `KeyPress` objects from the input records. N) rangevalue EventTyper!getattrEventtyperKeyDown_event_to_key_pressesr _handle_mouse)r2rHriirevr3r3r4r3s  zConsoleInputReader._get_keys key_pressesccsd}|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_surrogatefullcharr3r3r4rKs*   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_countrr3r3r4rls   zConsoleInputReader._is_pastercCst|tkr |js Jd}|j}|jj}|dd}|dkr.|j|jvr-t |j|jd}n||j vrF|j |t j kr=d}t |j ||}nt ||}||j @sU||j@r||j@r|rt jt jt jt jt jt jt jt jt jt jt jt jt jt jt jt jt j t j!i }|"|j#|j#|_#||j @s||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 |r t j/t j0t jt j1t jt j2t jt j3t jt j4t jt j5t jt j6t jt j7t j+t j8t jt j9t j t j:i }|"|j#|j#|_#||j @s||j@r%|r%|j;dkr%t t j@}|rVt 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)r2rresultcontrol_key_stateu_char ascii_charmapping meta_pressedr3r3r4rs           z(ConsoleInputReader._event_to_key_pressescCs|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_UPrrrW MousePositionXYrrWindowsMouseEvent)r2r event_flags button_state event_typebuttonrr3r3r4rs4  z ConsoleInputReader._handle_mouse)Tr8)Ur9r:r;r<rrrControlAControlBControlCControlDControlEControlFControlGControlHControlIrControlKControlLrControlNControlOControlPControlQControlRControlSControlTControlUControlVControlWControlXControlYControlZControlBackslashControlSquareCloseControlCircumflexControlUnderscore BackspacerrrrrrrrrrrF1F2F3F4F5F6F7F8F9F10F11F12rrRIGHT_ALT_PRESSEDrrrrUr5rTrrrHrr rr staticmethodrrrrrrrr3r3r3r4r&s       !%     >  r&c@sVeZdZdZd ddZdedegdfddfdd Zdedeegdffd d Z dS) 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-NcCsi|_i|_dSr.)_handle_callbacks_remove_eventsr1r3r3r4r5Fs z_Win32Handles.__init__r7callbackcsvj}|dur td|t|j|<t|j|<d fdd d fdd tddS) z7 Add a Win32 handle to the event loop. NInvalid handle.r-c s(z WtddStdw)Nlooprr3)rSrVwaitr3r4readyasz-_Win32Handles.add_win32_handle..readycs2tg}|urtjdSdSr.)rrr CloseHandlecall_soon_threadsafe)r )r7rVrX remove_eventr3r4rWks  z,_Win32Handles.add_win32_handle..waitrUr8)r ValueErrorremove_win32_handlerrQrrRr)r2r7rS handle_valuer3)rSr7rVrXr[rWr4add_win32_handleMs     z_Win32Handles.add_win32_handlecCsf|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)rrRpopKeyErrorrrSetEventrQ)r2r7eventr3r3r4r]xs    z!_Win32Handles.remove_win32_handler8) r9r:r;r<r5r r r_rr]r3r3r3r4r/6s  "+r/inputrSr-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. NrTr0r7rr\r]r_)rdrSr0r7previous_callbackr3r3r4r)s"      r)c cs`|j}|j}|jdurtd||}zdVW|r$|||dSdS|r/|||ww)NrTre)rdr0r7rfr3r3r4r*s  r*c@sNeZdZdZddeeddfddZdddZdd d Zd e ddfd d Z dS)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`. NrRr-cCsttjt|_dSr.)r rrrr r7)r2rRr3r3r4r5szraw_mode.__init__cCs,t}tj|jt|||_|dSr.)r rrGetConsoleModer7r original_mode_patch)r2rhr3r3r4 __enter__s zraw_mode.__enter__cCs2d}d}d}tj|j|jj||B|B@dSNr+rr"rrSetConsoleModer7rhrr2ENABLE_ECHO_INPUTENABLE_LINE_INPUTENABLE_PROCESSED_INPUTr3r3r4ris zraw_mode._patchacGstj|j|jdSr.)rrrmr7rh)r2rrr3r3r4__exit__szraw_mode.__exit__r.r8) r9r:r;r<rrVr5rjriobjectrsr3r3r3r4r's    r'c@seZdZdZdddZdS)r(zx :: with cooked_mode(stdin): ''' The pseudo-terminal stdin is now used in cooked mode. ''' r-NcCs0d}d}d}tj|j|jj||B|BBdSrkrlrnr3r3r4ris zcooked_mode._patchr8)r9r:r;r<rir3r3r3r4r(sr()>rrQabcr contextlibrprompt_toolkit.eventlooprutilsrplatformrctypesrr r ctypes.wintypesr r typingr rrrrrrrrprompt_toolkit.eventloop.win32rr(prompt_toolkit.key_binding.key_processorrprompt_toolkit.keysrprompt_toolkit.mouse_eventsrrprompt_toolkit.win32_typesrrrr r!ansi_escape_sequencesr#baser$__all__rrrrr,r%r&r/r)r*r'r(r3r3r3r4sZ     (      92X  '