o j @sdZddlmZddlZddlmZmZddlmZddl m Z dd l m Z d gZ ed ed d Zed ed dZed ed dZed ed dZGdddZGdddeeefZeZGdd d ZdS)z Parser for VT100 input stream. ) annotationsN)Callable Generator)KeyPress)Keys)ANSI_SEQUENCES Vt100Parser^z \d+;\d+R\Zz(9sz:_IsPrefixOfLongerMatchCache.__missing__..)_cpr_response_prefix_rematch_mouse_event_prefix_reanyr items)selfrresultrrr __missing__0s z'_IsPrefixOfLongerMatchCache.__missing__N)rrrr)r rrrr&rrrrr*src@sleZdZdZd$ddZd%d&d d Zd'd dZd(ddZd)ddZd*ddZ d+ddZ d'dd Z d+d!d"Z d#S),r aZ Parser for VT100 input stream. Data can be fed through the `feed` method and the given callback will be called with KeyPress objects. :: def callback(key): pass i = Vt100Parser(callback) i.feed('data...') :attr feed_key_callback: Function that will be called when a key is parsed. feed_key_callbackCallable[[KeyPress], None]rNonecCs||_|dSr)r'reset)r$r'rrr__init__Y zVt100Parser.__init__FrequestrcCsd|_|dS)NF)_in_bracketed_paste _start_parser)r$r-rrrr*]r,zVt100Parser.resetcCs||_|jddS)z- Start the parser coroutine. N)_input_parser_generator _input_parsersendr$rrrr/as zVt100Parser._start_parserrrNone | Keys | tuple[Keys, ...]cCs@t|rtjSt|rtjSzt|WStyYdSw)zD Return the key (or keys) that maps to this prefix. N)_cpr_response_rer r CPRResponse_mouse_event_reVt100MouseEventr KeyError)r$rrrr _get_matchhs    zVt100Parser._get_match#Generator[None, str | _Flush, None]c csd}d}d} d}|rd}ndV}t|trd}n||7}|r|t|}||}|s-|s8|r8|||d}nD|s<|s||s|d}d}tt|ddD]}||d|}|ri|||d|||d}d}qJ|s|||d|d|dd}q)zA Coroutine (state machine) for the input parser. FTNrr) isinstancer _IS_PREFIX_OF_LONGER_MATCH_CACHEr: _call_handlerrangelen) r$rretryflushcis_prefix_of_longer_matchr foundirrrr0{s@       z#Vt100Parser._input_parser_generatorkeystr | Keys | tuple[Keys, ...] insert_textcCsft|trt|D]\}}|||dkr|ndq dS|tjkr)d|_d|_dS|t ||dS)z& Callback to handler. rr<TN) r>tuple enumerater@rBracketedPaster. _paste_bufferr'r)r$rIrKrHrrrrr@s   zVt100Parser._call_handlerdatacCs|jrA|j|7_d}||jvr?|j|}|jd|}|ttj|d|_|j|t|d}d|_||dSdSt |D]\}}|jrX|||ddS|j |qEdS)zV Feed the input stream. :param data: Input string (unicode). z[201~NFr<) r.rOindexr'rrrNrBfeedrMr1r2)r$rPend_mark end_index paste_content remainingrHrErrrrRs$   zVt100Parser.feedcCs|jtdS)a= Flush the buffer of the input stream. This will allow us to handle the escape key (or maybe meta) sooner. The input received by the escape key is actually the same as the first characters of e.g. Arrow-Up, so without knowing what follows the escape sequence, we don't know whether escape has been pressed, or whether it's something else. This flush function should be called after a timeout, and processes everything that's still in the buffer as-is, so without assuming any characters will follow. N)r1r2r r3rrrrDs zVt100Parser.flushcCs|||dS)z8 Wrapper around ``feed`` and ``flush``. N)rRrD)r$rPrrrfeed_and_flushs  zVt100Parser.feed_and_flushN)r'r(rr))F)r-rrr))rr))rrrr4)rr;)rIrJrKrrr))rPrrr)) r rrrr+r*r/r:r0r@rRrDrWrrrrr Fs      2  &)r __future__rrecollections.abcrrkey_binding.key_processorrkeysransi_escape_sequencesr __all__compileescaper5r7rr!r dictrrrr?r rrrrs"