a lc @sdZddlZddlmZmZmZmZmZddlm Z ddl m Z ddl m Z d gZed ed d Zed ed d Zed ed dZed ed dZGdddZGdddeeefZeZGdd d ZdS)z Parser for VT100 input stream. N)CallableDict GeneratorTupleUnion)KeyPress)Keys)ANSI_SEQUENCES Vt100Parser^z \d+;\d+R\Zz(6sz:_IsPrefixOfLongerMatchCache.__missing__..)_cpr_response_prefix_rematch_mouse_event_prefix_reanyr items)selfrresultrrr __missing__-s z'_IsPrefixOfLongerMatchCache.__missing__N)rrrrstrboolr'rrrrr'src@seZdZdZeegdfddddZdedddd Zdd d d Z e e de e e d ffdddZede e efdfd ddZe e e e e d ffe ddddZe ddddZdd ddZe ddddZdS)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. N)feed_key_callbackrcCs||_|dSr)r*reset)r%r*rrr__init__VszVt100Parser.__init__F)requestrcCsd|_|dS)NF)_in_bracketed_paste _start_parser)r%r-rrrr+ZszVt100Parser.reset)rcCs||_|jddS)z- Start the parser coroutine. N)_input_parser_generator _input_parsersendr%rrrr/^s zVt100Parser._start_parser.rcCsDt|rtjSt|r tjSz t|WSty>YdS0dS)zD Return the key (or keys) that maps to this prefix. N)_cpr_response_rer!r Z CPRResponse_mouse_event_reZVt100MouseEventr KeyError)r%rrrr _get_matches    zVt100Parser._get_matchc csd}d}d}d}|rd}ndV}t|tr0d}n||7}|r t|}||}|sV|sl|rl|||d}q |st|s |s d}d}tt|ddD]>}||d|}|r|||d|||d}d}q|s ||d|d|dd}q dS)zA Coroutine (state machine) for the input parser. FNTrr ) isinstancer _IS_PREFIX_OF_LONGER_MATCH_CACHEr7 _call_handlerrangelen) r%rretryflushcZis_prefix_of_longer_matchr!foundirrrr0xs8      z#Vt100Parser._input_parser_generator)key insert_textrcCsbt|tr6t|D] \}}|||dkr,|ndqn(|tjkrNd|_d|_n|t ||dS)z& Callback to handler. rr8TN) r:tuple enumerater<r BracketedPaster. _paste_bufferr*r)r%rDrErCrrrrr<s  zVt100Parser._call_handler)datarcCs|jr||j|7_d}||jvr|j|}|jd|}|ttj|d|_|j|t|d}d|_||nfeedrGr1r2)r%rJZend_markZ end_indexZ paste_content remainingrCrArrrrLs    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)r1r2rr3rrrr@s zVt100Parser.flushcCs|||dS)z8 Wrapper around ``feed`` and ``flush``. N)rLr@)r%rJrrrfeed_and_flushs zVt100Parser.feed_and_flush)F)rrrrrrr,r)r+r/r(rr rr7rrr0r<rLr@rNrrrrr Cs" 3 &)rretypingrrrrrZkey_binding.key_processorrkeysr Zansi_escape_sequencesr __all__compileescaper4r5r r"rr(r)rr;r rrrrs