.dZddlmZddlmZddlmZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZdd lZdd lZd ZGd deZGddeZGddeZy )a) An :class:`~.InputProcessor` receives callbacks for the keystrokes parsed from the input in the :class:`~prompt_toolkit.inputstream.InputStream` instance. The `InputProcessor` will according to the implemented keybindings call the correct callbacks when new key presses are feed through `feed`. )unicode_literals)EditReadOnlyBuffer)ViNavigationMode)KeysKey)Event) BaseRegistry)deque)rangeN)InputProcessorKeyPressc$eZdZdZddZdZdZy)rz :param key: A `Keys` instance or text (one character). :param data: The received string on stdin. (Often vt100 escape codes.) Nct|tjtfsJ|t|tjsJ|t|tr |jn|}||_||_yN) isinstancesix text_typernamekeydata)selfrrs =lib/third_party/prompt_toolkit/key_binding/input_processor.py__init__zKeyPress.__init__!sZ# s3444|z$ >>> <)#s3388D ch|jjd|jd|jdS)Nz(key=z, data=)) __class____name__rrrs r__repr__zKeyPress.__repr__+s% NN # #TXXtyy: :rcj|j|jk(xr|j|jk(Sr)rr)rothers r__eq__zKeyPress.__eq__/s'xx599$@ejj)@@rr)r __module__ __qualname____doc__rr!r$rrrrs:ArrcZeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z dd ZdZy )r aw Statemachine that receives :class:`KeyPress` instances and according to the key bindings in the given :class:`Registry`, calls the matching handlers. :: p = InputProcessor(registry) # Send keys into the processor. p.feed(KeyPress(Keys.ControlX, '')) p.feed(KeyPress(Keys.ControlC, '') # Process all the keys in the queue. p.process_keys() # Now the ControlX-ControlC callback will be called if this sequence is # registered in the registry. :param registry: `BaseRegistry` instance. :param cli_ref: weakref to `CommandLineInterface`. ct|tsJ||_||_t ||_t ||_t|_g|_ d|_ g|_ |jy)NF) rr _registry_cli_refrbeforeKeyPress afterKeyPressr input_queue key_buffer record_macromacroreset)rregistrycli_refs rrzInputProcessor.__init__Isg(L111! #Dk"4[!7"  rcd|_d|_|j|_|jj dd|_yr)_previous_key_sequence_previous_handler_process_process_coroutinesendargr s rr3zInputProcessor.reset_s=&*#!%"&--/ $$T*rc d|_g|_y)z Start recording macro. TN)r1r2r s r start_macrozInputProcessor.start_macrojs  rcd|_y)z End recording macro. FN)r1r s r end_macrozInputProcessor.end_macroos !rcH|jD]}|j|yr)r2feed)rks r call_macrozInputProcessor.call_macrossA IIaLrctd|D}|j}|jj|Dcgc]}|j |s|c}Scc}w)zw For a list of :class:`KeyPress` instances. Give the matching handlers that would handle this. c34K|]}|jywrr.0rCs r z.InputProcessor._get_matches..|0KqQUUK)tupler,r+get_bindings_for_keysfilter)r key_presseskeysclibs r _get_matcheszInputProcessor._get_matcheswsU 0K00mmo >>??EWEaRUEWWWs AActd|D}|jtd|jj |D}t fd|DS)z For a list of :class:`KeyPress` instances. Return True if there is any handler that is bound to a suffix of this keys. c34K|]}|jywrrGrHs rrJz.rKrLc34K|]}|jywr)rO)rIrSs rrJz.s](\1ahh(\rLc3.K|] }|ywrr()rIfrRs rrJz.s+7a1S67s)rMr,setr+get_bindings_starting_with_keysany)rrPrQfiltersrRs @r_is_prefix_of_longer_matchz)InputProcessor._is_prefix_of_longer_matchsU 0K00mmo ](V(VW[(\]]+7+++rc#K|j}d} |rd}n|jd|r|j|}|j|}|Dcgc]$}|j |j s#|&}}|r|}d}|s|r|j |d||dd=nb|s`|s^d}d}tt|ddD]:}|j|d|}|s|j |d|d||d|=d}n|s|dd=cc}ww)z Coroutine implementing the key match algorithm. Key strokes are sent into this generator, and it calls the appropriate handlers. FTN) key_sequencerr ) r0appendr^rTeagerr, _call_handlerr len) rbufferretryis_prefix_of_longer_matchmatchesm eager_matchesfoundis rr9zInputProcessor._processs>  u&,0,K,KF,S)++F3-4 PGqqwwt}}7OG P +G05-1W&&wr{&Hq 37 E!E#3v;26"&"3"3F2AJ"?" ..wr{PRQR.T &rr $(E! 7!"2A2JO!QsAD $D7D;AD 0D c^t|tsJ|jj|y)z Add a new :class:`KeyPress` to the input queue. (Don't forget to call `process_keys` in order to process the queue.) N)rrr/rb)r key_presss rrBzInputProcessor.feeds( )X...  *rc|jr|jj}|jtjk7r|j j |jj||jtjk7r|jj |jr|j}|r|jyy)a, Process all the keys in the `input_queue`. (To be called after `feed`.) Note: because of the `feed`/`process_keys` separation, it is possible to call `feed` from inside a key binding. This function keeps looping until the queue is empty. N) r/popleftrr CPRResponser-firer:r;r.r, invalidate)rrorRs r process_keyszInputProcessor.process_keyss((002I}} 0 00##((*  # # ( ( 3}} 0 00""'')mmo  NN  rNc|j}|j}d|_ttj||||j ||j k(}|j}|j|r|r|jj |j||j|||_||_|jr|r|jj|yyy#t$rYEwxYw)N)r<raprevious_key_sequence is_repeat)r1r< KeyPressEventweakrefrefr7r8rR save_beforecurrent_buffersave_to_undo_stackcall_fix_vi_cursor_positionrr2extend)rhandlerra was_recordingr<eventrRs rrdzInputProcessor._call_handlers)) hh KK 3\"&"="=$"8"88; ii   u %#    1 1 3  LL   ( ( /'3#!(    JJ  l +"/ "   s"C00 C<;C<c<|j}|r|j}|j}t|jrW|j j r@t|j jdkDr|xjdzc_ ||_yyyyy)z After every command, make sure that if we are in Vi navigation mode, we never put the cursor after the last character of a line. (Unless it's an empty line.) rr N) r,r}preferred_columnrrRdocumentis_cursor_at_the_end_of_linere current_linecursor_position)rrrRbuffrs rrz&InputProcessor._fix_vi_cursor_positions mmo %%D#44 " "599-MM>> 223a7$$)$)9% 8?. rr)rr%r&r'rr3r>r@rDrTr^r9rBrurdrr(rrr r 3sF*,  " X,"/'b+2 ,D9rr ceZdZdZ d dZdZedZedZedZ edZ ed Z ed Z d Z y) ryax Key press event, delivered to key bindings. :param input_processor_ref: Weak reference to the `InputProcessor`. :param arg: Repetition argument. :param key_sequence: List of `KeyPress` instances. :param previouskey_sequence: Previous list of `KeyPress` instances. :param is_repeat: True when the previous event was delivered to the same handler. NcJ||_||_||_||_||_yr)_input_processor_refrarwrx_arg)rinput_processor_refr<rarwrxs rrzKeyPressEvent.__init__'s*$7!(%:"# rcVd|jd|jd|jdS)NzKeyPressEvent(arg=z, key_sequence=z , is_repeat=r)r<rarxr s rr!zKeyPressEvent.__repr__2s!$++T^^= =rc4|jdjS)Nr`)rarr s rrzKeyPressEvent.data6s  $)))rc"|jSr)rr s rinput_processorzKeyPressEvent.input_processor:s((**rc6|jjS)z) Command line interface. )rr,r s rrRzKeyPressEvent.cli>s ##,,..rc.|jjS)z% The current buffer. )rRr}r s rr}zKeyPressEvent.current_bufferEs xx&&&rcx|jdk(ryt|jxsd}t|dk\rd}|S)z& Repetition argument. -r`r i@B)rint)rresults rr<zKeyPressEvent.argLs= 99 TYY^!$ v;' !F rc|jduS)zF True if repetition argument was explicitly provided. N)rr s r arg_presentzKeyPressEvent.arg_present\s yy$$rc|dvsJ|j}|dk(r ||dk(sJ|}n ||}n||}||j_y)zb Add digit to the input argument. :param data: the typed digit as string z -0123456789rN)rrr<)rrcurrentrs rappend_to_arg_countz!KeyPressEvent.append_to_arg_countcsZ }$$$)) 3;?gn 44F _F&-F#) r)NNNF)rr%r&r'rr!propertyrrrRr}r<rrr(rrryrysDH27 =**++// ''   %% *rry)r' __future__rprompt_toolkit.bufferrprompt_toolkit.filters.clirprompt_toolkit.keysrrprompt_toolkit.utilsrr4r collectionsr six.movesr rzr__all__objectrr ryr(rrrs\(47)&"   AvA.f9Vf9TW*FW*r