a lcyd@sddlZddlZddlmZddlmZddlmZddlm Z ej dksNJe sfddl 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/m0Z0gdZ1dZ2dZ3dZ4dZ5Gddde0Z6Gddde6Z7GdddZ8GdddZ9ee6egdfeddd d!Z:ee6edd"d#d$Z;Gd%d&d&ZdZ?dZ@dZAdZBd%eCd d d d ZDd dddZEeFeGdddZHeGeGdddZIeJdeKeGdddZLeMeNeGeKeGdddZOeMeNeGeCdddZPeQeNeGd d!d"ZReSeNeGd d#d$ZTd S)&r&z| :param recognize_paste: When True, try to discover paste actions and turn the event into a BracketedPaste. )!     )!"#$%&'(-.pqrstuvwxyz{rr"r+TN)recognize_paster.cCsZd|_||_|tjr.ttjt |_ n(t dt j t jB|_tt|j|_ dS)NzCONIN$)_fdconrrJr=isattyr rkernel32 GetStdHandler r7osopenO_RDWRO_BINARYmsvcrtZ get_osfhandle)r3rr4r4r5r6s zConsoleInputReader.__init__r-cCs|jdurt|jdS)z Close fdcon.N)rrrMr2r4r4r5rMs zConsoleInputReader.closec #s0d}td}t|}|}tjgdds.dStjjt||t|t ||}fdd|D}t |}j r" |r"t |}|D]}g}|rt|jtr|jtjtjhvr||jz t|}Wqtyd}Yq0q|rttjd|V|dur|Vqn |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|]}|qSr4)_insert_key_data).0keyr2r4r5 z+ConsoleInputReader.read..)r rrr7rrZReadConsoleInputWr rE _get_keys_merge_paired_surrogatesr _is_pasteiter isinstancerrControlJControlMappenddatanext StopIterationrZBracketedPastejoin) r3 max_countrFZarrtype input_recordsall_keysgenkrr4r2r5rFs>      zConsoleInputReader.read) key_pressr.cCs6|jr |St|jtr&t|jd}nd}t|j|S)z@ Insert KeyPress data, for vt100 compatibility. r)rrrrr#getr)r3rrr4r4r5r%s  z#ConsoleInputReader._insert_key_datazArray[INPUT_RECORD])rFrr.ccsxt|jD]h}||}|jtvr t|jt|j}t|tkrV|jrV| |EdHq t|t kr | |EdHq dS)zR Generator that yields `KeyPress` objects from the input records. N) rangevalueZ EventTyper!getattrEventtyperKeyDown_event_to_key_pressesr _handle_mouse)r3rFriZirevr4r4r5r3s  zConsoleInputReader._get_keys) 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_surrogateZfullcharr4r4r5rKs*  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_countrr4r4r5rls    zConsoleInputReader._is_paste)rr.cCst|tkr|jsJd}|j}|jj}|dd}|dkr\|j|jvrt |j|jd}n:||j vr|j |t j krzd}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 jt j t j!i }|"|j#|j#|_#||j @s4||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 @s8||j@rV|rV|j;dkrVt t j@}|rt t j=d|gS|gSngSdS)zU For this `KEY_EVENT_RECORD`, return a list of `KeyPress` instances. Nzutf-8rr  )?rrrZControlKeyStateZuCharZ 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)r3rresultZcontrol_key_stateZu_charZ ascii_charmappingZ meta_pressedr4r4r5rs            z(ConsoleInputReader._event_to_key_pressescCs|j}|j}d}tj}|t@r6|dkr.tj}qTtj}n|tkrFtj }n|t krTtj }|t @rbtj }|dur|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_UPrrrPZ MousePositionXYrrZWindowsMouseEvent)r3rZ event_flagsZ button_stateZ event_typeZbuttonrr4r4r5rs4  z ConsoleInputReader._handle_mouse)T)Ur8r9r:r;rrrZControlAZControlBZControlCZControlDZControlEZControlFZControlGZControlHZControlIrZControlKZControlLrZControlNZControlOZControlPZControlQZControlRZControlSZControlTZControlUZControlVZControlWZControlXZControlYZControlZZControlBackslashZControlSquareCloseZControlCircumflexZControlUnderscoreZ BackspacerrrrrrrrrrrZF1ZF2ZF3ZF4ZF5ZF6ZF7ZF8ZF9ZF10ZF11ZF12rrZRIGHT_ALT_PRESSEDrrrrNr6rMrrrFrr rr staticmethodrrrrrrrr4r4r4r5r&s' >  r&c@sTeZdZdZddddZeegdfddddZeeegdfd d d Z dS) r0a 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`. Nr-cCsi|_i|_dSr/)_handle_callbacks_remove_eventsr2r4r4r5r6Fsz_Win32Handles.__init__)r7callbackr.cs~j}|durtd|t|j|<t|j|<ddfdd ddfdd tddS) z7 Add a Win32 handle to the event loop. NInvalid handle.r-c s*zWtdntd0dS)Nlooprr4)rrwaitr4r5readyasz-_Win32Handles.add_win32_handle..readycs2tg}|ur$tjdSdSr/)rrrZ CloseHandlecall_soon_threadsafe)r)r7rr remove_eventr4r5rks   z,_Win32Handles.add_win32_handle..waitr)r ValueErrorremove_win32_handlerrrrr)r3r7rZ handle_valuer4)rr7rrrrr5add_win32_handleMs     z_Win32Handles.add_win32_handle)r7r.cCsj|jdurdSz|j|j}Wnty2Yn0tj|z|j|jWStydYdS0dS)zt Remove a Win32 handle from the event loop. Return either the registered handler or `None`. N)rrpopKeyErrorrrZSetEventr)r3r7eventr4r4r5rxs    z!_Win32Handles.remove_win32_handle) r8r9r:r;r6r r rrrr4r4r4r5r06s+r0)inputrr.c csz|j}|j}|jdurtd||}|||z$dVW|||rv|||n|||rt|||0dS)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. Nrr1r7rrrr)rrr1r7previous_callbackr4r4r5r)s      r))rr.c csZ|j}|j}|jdurtd||}zdVW|rV|||n|rT|||0dS)Nrr)rr1r7rr4r4r5r*s  r*c@sReZdZdZdeeddddZddddZddd d Ze dd d 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`. N)rKr.cCsttjt|_dSr/)r rrrr r7)r3rKr4r4r5r6szraw_mode.__init__r-cCs,t}tj|jt|||_|dSr/)r rrZGetConsoleModer7r original_mode_patch)r3rr4r4r5 __enter__szraw_mode.__enter__cCs2d}d}d}tj|j|jj||B|B@dSNr+rr"rrSetConsoleModer7rrr3ZENABLE_ECHO_INPUTZENABLE_LINE_INPUTZENABLE_PROCESSED_INPUTr4r4r5rs zraw_mode._patch)ar.cGstj|j|jdSr/)rrrr7r)r3rr4r4r5__exit__szraw_mode.__exit__)N) r8r9r:r;rrOr6rrobjectrr4r4r4r5r's   r'c@seZdZdZddddZdS)r(zx :: with cooked_mode(stdin): ''' The pseudo-terminal stdin is now used in cooked mode. ''' Nr-cCs0d}d}d}tj|j|jj||B|BBdSrrrr4r4r5rs zcooked_mode._patch)r8r9r:r;rr4r4r4r5r(sr()>rrJabcr contextlibrZprompt_toolkit.eventlooprutilsrplatformrctypesrr r Zctypes.wintypesr r typingr rrrrrrrrZprompt_toolkit.eventloop.win32rrZ(prompt_toolkit.key_binding.key_processorrZprompt_toolkit.keysrZprompt_toolkit.mouse_eventsrrZprompt_toolkit.win32_typesrrrr r!Zansi_escape_sequencesr#baser$__all__rrrrr,r%r&r0r)r*r'r(r4r4r4r5sP     (      92X '