o jx @sDdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZdd lmZmZdd lmZmZmZmZmZmZdd lmZdd lmZmZdd lm Z m!Z!m"Z"Gdddej#Z$Gddde Z% dJdKddZ&dLd)d*Z'dMd/d0Z(dNd5d6Z)dOd;d<Z*dNd=d>Z+  ?dPd@dAddBdCddDdEdQdHdIZ,dS)Rz*This is a python implementation of clip().) annotationsN)islice)LiteralCallableOptional NamedTuple)width)iter_graphemes) HyperlinkHyperlinkParams) _SGR_PATTERN_SGR_STATE_DEFAULT _SGRState_sgr_state_update_sgr_state_is_active_sgr_state_to_sequence)_clamp_ambiguous_width) TextSizingTextSizingParams)_SEQUENCE_CLASSIFY_HORIZONTAL_CURSOR_MOVEMENTINDETERMINATE_EFFECT_SEQUENCEc@s0eZdZdZeZeZeZeZ dS)_HyperlinkActionz.Outcome of processing an OSC 8 hyperlink unit.N) __name__ __module__ __qualname____doc__enumautoNO_CLOSEEMPTYOUTSIDEVISIBLEr$r$/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/wcwidth/_clip.pyrs  rc@sneZdZUdZded<dZded<dZded<dZd ed <dZd ed <dZ d ed <dZ ded <dZ ded<dS)_HyperlinkResultzn Result of processing an OSC 8 hyperlink. Only the fields relevant to each action are populated. ractionrint close_end inner_widthstropen_seq clipped_inner close_seq clipped_width hl_col_endN) rrrr__annotations__r)r*r-r.r/r0r1r$r$r$r%r&%s       r&resultr,captured_styleOptional[_SGRState] end_stylereturncCs<|durt|}r||}t|dur|n|r|d7}|S)aZ Apply SGR prefix/suffix around *result*. If an SGR state was captured at the first visible character, prefix the result with the corresponding SGR sequence, and suffix with a reset if any styles remain active at the end of the clipped region. *end_style* is the style in effect after the final SGR sequence emitted within the clip window, or ``None`` when no such sequence was emitted, the style at the first visible character is still in effect. This matches :func:`wcwidth.propagate_sgr`, which decides the trailing reset by SGR state at the end of each line. Nz)rr)r3r4r6prefixr$r$r%_apply_sgr_wrap6s  r9textstartr(endfillchartabsizeambiguous_width term_program bool | str control_codes$Literal['parse', 'strict', 'ignore']paramsr match_endcolc Cst|| \} } | | fdkrttjS|| | } t| ||||d}|dkr-ttj| dS| |}||ks9| |krAttj| |dStd|| }|| }t | ||||||d|d }t|||||d}ttj | || || ||dS) z Process an OSC 8 hyperlink unit. Finds the matching close sequence, measures the inner text width, and determines whether the hyperlink is empty, outside the clip window, or visible (requiring inner-text clipping). )rG)rBr>r?r@r)r))r)r*F)r=r>r?r@ propagate_sgrrB)r)r*r-r.r/r0r1) r find_closer&rr r r!r"maxclipr# make_open make_close)r:r;r<r=r>r?r@rBrDrErF close_startr) inner_textr*r1inner_clip_startinner_clip_endr.r0r$r$r%_process_hyperlinkLsR    rRcellsdict[int, tuple[str, int]] sequenceslist[tuple[int, int, str]]cCs\i}|D]\}}}||g||fq|D]} | q|r(t|nd} |r2t|nd} t| | } g} d}t| |}||kr||dD] \}}| |qL||kr_|d7}qB||vru||\}}| |||prd7}n||kr| krnn| ||d7}||ksFt|D]}||kr||D] \}}| |qqd | S)z Reconstruct the output string from painter's algorithm state. Walks columns left-to-right, interleaving escape sequences and cell content, filling gaps with *fillchar*. rGrr$rr+) setdefaultappendvaluessortrJkeysmingetsortedjoin)rSrUr;r<r= seqs_by_colcol_posorderseq_textentries max_cell_col max_seq_colmax_colpartswalk_col col_limit_ cell_textcell_wcr$r$r%_reconstruct_painters>          rorHboolstrict4tuple[str, Optional[_SGRState], Optional[_SGRState]]c sg} d} d} d} d}|rtnd}| t|krh|| }| |kr:|dvr:| dur)n?|d| d}|dkr7n1|} q|dkrt|| }|sQ| || d7} q|ddur{|r{|dur{t||}| durv| |krv| ||}|} qt |}rt |||||||| ||| d }|j t jur| ||} nS|j t jur|j} nI|j t jur| |j7} |j} n:| |j| |j| |j|r| dur|} |r|durt|jD] }t||}|}q| |j7} |j} q|d }dur|d p d }|d }|dusJttj|| d ||}||}| |krK| ||krK| ||rF| durF|} | |7} n4| |kr{| ||kr{gdfdd }t|| ||||||} | |rz| durz|} n| |7} |} q|}|rt|rt d| d|| ||} q|dkr|dkr| || |}| |kr|| kr|krnn| d|r| dur|} | d7} | |ksn| d| d7} qt!t"|| d}t#|||d}|dkr|| kr |krnnI| |nC| |kr1| ||kr1| ||r0| dur0|} n'| |krX| ||krX| |t$|| |t%|| |rX| durX|} | |7} | t|7} | t|ksd &| | |fS)z Clip text without cursor movement (simple append-to-output path). Returns ``(result, captured_style, end_style)``. The caller applies SGR wrapping. rN  rrG sgr_paramsrDrErFts_metats_textr+ts_termrBsr,_wr(_colr7Nonecs|dSN)rX)r{r|r}ts_partsr$r% _ts_writeGsz_clip_simple.._ts_write*Indeterminate cursor sequence at position ,   r;r?r@)r{r,r|r(r}r(r7r~)'rlenfindrmatchrXgrouprr<r parserRr'rr r!r)r"r*r-r.r/r finditerrr from_params display_width make_sequence_text_sizing_clipextendr ValueErrornextr r r\rJr_) r:r;r<rHr?r@r=r>rqrBoutputrFidxr4r6 current_stylecharnext_escmhl_statersgr_mrwrxrytsts_widthrseqnext_tabgrapheme grapheme_wr$rr% _clip_simples                                  "  rrr write_cellsCallable[[str, int, int], None]c s|}||kr|||kr||||S||ks%|||kr)||Std||} t||||} jjg} jjdkrnttj jjD] } | | fqOt jjt | D] } | dfqcntj D]} | | t| ||dfqsgd fdd }|| }d}| D]T\}}||}|| kr|}q|| krn@t|| t|| }||kr|dkrЈs|t|| } ||fn|||t|| }t |D] }|d ||q|}q||||S) z Emit tokens for a text-sizing (OSC 66) sequence, clipped to (start, end). Calls *write_cells(text, width, col)* for each emitted cell or sequence. Returns new column position. rr+r flush_colr(r7r~cssdSddD}tddD}tjjdkr t|ndjjjjjjjj}t |d |j  || dS)NcSsg|]}|dqS)rr$.0ur$r$r% sz4_text_sizing_clip..flush..css|]}|dVqdS)rNr$rr$r$r% sz3_text_sizing_clip..flush..rr+)sumrrDr r numerator denominatorvertical_alignhorizontal_alignrr_ terminatorrclear)rtextstotal_wrD pending_unitsscalerrr$r%flushs    z _text_sizing_clip..flushrN)rr(r7r~) rrrJr\rDrr rr r:rXranger)rrFr;r<r=r?r@rr rel_startrel_endunitsgrkr flush_col_posunit_pos unit_textunit_wunit_endoverlap abs_startir$rr%rs^    rc( sitgdd} d} dd} rtndd} d+d,fd d }| t|kr<|| }| |kr|dvr| dusLd| krJ| krpnn$t|d|d|d| kr`d} nt|| }|durldn|} | dkrdurzn|d| d}|dkrn|} q)|dkrht || }|s | |fd7| d7} q)| ddurڈrڈdurt | dur| |krՈ | | fd7} |} q)t| }rut||||||| ||| d }|jtjur | | fd7|} ne|jtjur|j} nZ|jtjur*| |j7} |j} nJ | |jfd7||j|j| dd| |j7}  | |jfd7rndurnt|jD] }t | } qa|j} |j} q)| d}dur| dpd}| d}|dusJtt j!|| d||}t"|| |||||} |} q)| }|rt# |rt$d| d|| d }dur|rt%|dnd} |} q)| d!}dur|rt%|nd}| |}| |kr||krt&t| |t'||D] }|d|q |} |} q)| d"} durU| r-t%| nd}!|rE|!| krEt$d#| d$|!d%| d&| |!8} | dkrPd} |} q) | | fd7|} q)|dkrtd} | d7} q)|dkr| dkr| d8} | d7} q)|d'kr|dkr| || |}"t&t| |t'|"|D] }#|d(d|#q|"} n  | d'fd7| d7} q)t(t)|| d)}$t*|$||d*}%|%dkr|| kr|krnnH | |$fd7n;| |kr| |%|kr||$|%| n(| |kr,| |%|kr,t|| }&t&t'|| |%|&D] }'|d|&|'q | |%7} | t|$7} | t|ks0t+||| fS)-z Clip text with cursor movement (painter's algorithm path). Returns ``(result, captured_style, end_style)``. The caller applies SGR wrapping. rNFr{r,wr( write_col is_hyperlinkrpr7r~cs|dkr|r||fd7rdurdSt|D]I}||}|dkrE|ddddkrEdf|d<|d|dddkr^df|d<|d|d|q ||f|<|rt|D] }||qvrdurdSdSdS)zIWrite *w* cells of text *s* at *write_col*, handling wide-char splitting.rrN)r+r)rXrr]discardpopadd)r{rrroffsetsrc_colr4rSrr=hyperlink_cellsrH seq_orderrUr$r% _write_cells s0        z#_clip_painter.._write_cellsrs rtrGrrurvT)rrwrxr+ryrzrrhpa_n cforward_n cbackward_nz!Cursor left movement at position z would move z cells left from column z, exceeding string startrrrr)F) r{r,rr(rr(rrpr7r~),setrrrJrfindrsearchr<rrrrXrrr rrRr'rr r!r)r"r*r-r.r0r/r rr1rrrrrrr(rr\rr r ro)(r:r;r<rHr?r@r=r>rqrBrFrr6 next_movementrrfoundrrrrrrwrxryrrrr n_forwardmove_endrr n_backwardrfill_colrr clip_startrr$rr% _clip_paintersT  "                      ;rrGrTrF)r=r>r?rHrB overtypingr@rOptional[bool]c Cst|d}|dkr|dkrtd|tj}||krdS|r+|r+|||St|}d|v} |r9| s9d}|durS|dkoQd |vpQd |vpQ| oQtt |}n|r[|dkr[d}|r_t nt } t | |||||| |||d k|d S) u Clip text to display columns (start, end) while preserving all terminal sequences. This function extracts a substring based on visible column positions rather than character indices. Terminal escape sequences are preserved in output. If a wide character (width of 2) is split at a boundary, it is replaced with ``fillchar``. TAB characters (``\t``) are expanded to spaces up to the next tab stop, controlled by the ``tabsize`` parameter. When cursor movement is detected, a "painter's algorithm" is used unless ``overtyping=False`` is set. Cursor movement control codes are parsed for their effects instead of ignored. For these operations, it is assumed that ``text`` begins at column 0. For text containing **OSC 8 hyperlinks**, the visible text inside a hyperlink is clipped to the requested column range and the hyperlink sequence is rebuilt:: >>> clip('\x1b]8;;http://example.com\x07Click This link\x1b]8;;\x07', 6, 10) '\x1b]8;;http://example.com\x07This\x1b]8;;\x07' :param text: String to clip, may contain terminal escape sequences. :param start: Absolute starting column (inclusive, 0-indexed), default ``0``. :param end: Absolute ending column (exclusive). The default value, ``-1``, signifies "to the end of the line", clipping only from *start* without requiring the caller to measure the display width of *text*. :param fillchar: Character to use when a wide character must be split at a boundary (default space). Must have display width of 1. :param tabsize: Tab stop width (default 8). Set to 0 to pass tabs through as zero-width (preserved in output but don't advance column position). :param ambiguous_width: Width to use for East Asian Ambiguous (A) characters. Default is ``1`` (narrow). Set to ``2`` for CJK contexts. :param propagate_sgr: If True (default), SGR (terminal styling) sequences are propagated, matching :func:`propagate_sgr`. The result begins with any style active at the start position, retains any style changes occurring within the clipped region at their original position, and ends with a reset sequence if styles are active at the end position. :param control_codes: How to handle control characters and sequences: - ``'parse'`` (default): Track horizontal cursor movement and clip hyperlink text. Cursor overwrite is always allowed, with best effort results; indeterminate sequences (home, clear, reset, etc.) are preserved as zero-width. - ``'strict'``: Like ``parse``, but raises :exc:`ValueError` on sequences with indeterminate effects (cursor home, clear screen, reset, vertical movement, etc.) matching :func:`width` behavior. Also raises on out-of-bounds horizontal cursor movement. - ``'ignore'``: All control characters are treated as zero-width. Cursor movement is not tracked (fastest path). :param overtyping: Whether to use the painter's algorithm for cursor movement (``\b`` backspace, ``\r`` carriage return, and CSI cursor left/right/position sequences). When ``None`` (default), auto-detects by scanning for these characters in *text*. Set to ``False`` for improved performance when the caller knows *text* contains no cursor movement characters. Set to ``True`` to force the painter's algorithm (useful for testing). Has no effect when ``control_codes='ignore'``. :param term_program: Terminal software identifier for table correction. ``False`` (default) disables override lookup. ``True`` reads the ``TERM_PROGRAM`` or ``TERM`` environment variable for auto-detection. Accepts a canonical terminal name matching :func:`list_term_programs`, such as from XTVERSION_, ENQ_, or ``TERM_PROGRAM``. .. versionadded:: 0.8.0 :returns: Substring of ``text`` spanning display columns (start, end), with all terminal sequences preserved and wide characters at boundaries replaced with ``fillchar``. :raises ValueError: If ``end`` is negative and not ``-1``, or if ``control_codes='strict'`` and an indeterminate-effect sequence or out-of-bounds cursor movement is encountered. SGR (terminal styling) sequences are propagated by default. The result begins with any active style and ends with a reset:: >>> clip('\x1b[1;34mHello world\x1b[0m', 6, 11) '\x1b[1;34mworld\x1b[0m' >>> clip('\x1b[1mbold\x1b[m normal', 1, 9) '\x1b[1mold\x1b[m norm' Set ``propagate_sgr=False`` to disable this behavior. .. versionadded:: 0.3.0 .. versionchanged:: 0.5.0 Added ``propagate_sgr`` parameter (default True). .. versionchanged:: 0.7.0 Added ``control_codes`` parameter (default 'parse'). OSC 8 hyperlink-aware clipping. OSC 66 text sizing protocol support. Added ``overtyping`` parameter (default None, auto-detect). .. versionchanged:: 0.8.4 ``start`` now defaults to ``0`` and ``end`` to ``-1``, meaning "to the end of the line":: >>> clip('\x1b[1;34mHello world\x1b[0m', 6) '\x1b[1;34mworld\x1b[0m' Example:: >>> clip('hello world', 0, 5) 'hello' >>> clip('中文字', 0, 3) # Wide char split at column 3 '中 ' >>> clip('a\tb', 0, 10) # Tab expanded to spaces 'a b' rrGz5end must be -1 (to end of line) or non-negative, got r+rtFNignorerrrq) r:r;r<rHr?r@r=r>rqrB) rJrsysmaxsizeisascii isprintablerrprrrrr9) r:r;r<r=r>r?rHrBrr@has_escfn_clipr$r$r%rKsF w   rKr)r3r,r4r5r6r5r7r,)r:r,r;r(r<r(r=r,r>r(r?r(r@rArBrCrDr rEr(rFr(r7r&) rSrTrUrVr;r(r<r(r=r,r7r,)r:r,r;r(r<r(rHrpr?r(r@rAr=r,r>r(rqrprBrCr7rr)rrrFr(r;r(r<r(r=r,r?r(r@rArrr7r()rrG)r:r,r;r(r<r(r=r,r>r(r?r(rHrprBrCrrr@rAr7r,)-r __future__rrr itertoolsrtypingrrrr_widthr rr hyperlinkr r sgr_stater rrrrr _constantsr text_sizingrrescape_sequencesrrrEnumrr&r9rRrorrrrKr$r$r$r%sH         F 7 @ X