ijx HdZddlmZddlZddlZddlmZddlmZm Z m Z m Z ddl m Z ddlmZdd lmZmZdd lmZmZmZmZmZmZdd lmZdd lmZmZdd lm Z m!Z!m"Z"GddejFZ$Gdde Z% d d dZ& d!dZ' d"dZ( d#dZ) d$dZ* d#dZ+ d%dddddddd d&dZ,y)'z*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_SEQUENCEceZdZdZej Zej Zej Zej Z y)_HyperlinkActionz.Outcome of processing an OSC 8 hyperlink unit.N) __name__ __module__ __qualname____doc__enumautoNO_CLOSEEMPTYOUTSIDEVISIBLEV/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/wcwidth/_clip.pyrrs98tyy{H DIIKEdiikGdiikGr'rc~eZdZUdZded<dZded<dZded<dZd ed <dZd ed <dZ d ed <dZ ded <dZ ded<y)_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.r1r2r3r4r5r&r'r(r*r*%sU IsKHcM3IsM3Jr'r*cV|&t|x}r||z}t||n|r|dz }|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. z)rr)resultcaptured_style end_styleprefixs r(_apply_sgr_wrapr<6sA!+N; ;6 ;f_F )2C S i F Mr'c tj|| \} } | | fdk(rttjS|| | } t | ||||}|dk(rttj | S| |z}||ks| |k\rttj| |Std|| z }|| z }t| ||||||d| }t |||||}ttj| ||j||j||S) 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). )r>) control_codestabsizeambiguous_width term_programr)r-)r-r.F)fillcharr@rArB propagate_sgrr?)r-r.r1r2r3r4r5) r find_closer*rr"r r#r$maxclipr% make_open make_close)textstartendrCr@rArBr?params match_endcol close_startr- inner_textr.r5inner_clip_startinner_clip_endr2r4s r(_process_hyperlinkrTLsE*'11$ BKY8+ 0 9 9::i ,J-!K a 0 6 6)LL{"JUcSj 0 8 8I,79 91eck*3YN$n7'!# M]!M   !!####%#  r'ci}|D])\}}}|j|gj||f+|jD]} | j|rt |j nd} |rt |j nd} t | | } g} d}t | |}||kr|j|dD]\}}| j|||k\r|dz };||vr#||\}}| j|||xsdz }n$||cxkr| krnn| j||dz }||krt|j D]&}||kDs ||D]\}}| j|(dj| 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*. r>rr&r r/) setdefaultappendvaluessortrFkeysmingetsortedjoin)cells sequencesrKrLrC seqs_by_colcol_posorderseq_textentries max_cell_col max_seq_colmax_colpartswalk_col col_limit_ cell_textcell_wcs r(_reconstruct_painterrps57K$-F w+22E83DEF%%' ).3uzz|$2L-8#k&&()bK, ,GEHGS!I i &??8R8 #KAx LL " # s? MH  u  %h Iv LL #  ! #H0L0 X& MH! i &K$$& '' y=*1~ ' 8 X& '' 775>r'c  g} d} d} d} d}|rtnd}| t|kr || }| |k\r'|dvr#| n|jd| dz}|dk(rn|} @|dk(rtj|| }|s| j || dz } u|j dW|rU|St||j }| &| |kr!| j |j |}|j} tj|j x}rt|||||||| ||j|  }|jtjur1| j |j |j} n|jtjur |j } n|jtj"ur| |j$z } |j } n| j |j&| j |j(| j |j*|r| |} |rB|@t-j.|j(D]}t||j }|} | |j$z } |j } |j d x}|j d xsd }|j d }|Jt1t3j4|| ||}|j7|}| |k\r3| |z|kr+| j |j9|r| |} | |z } nD| |kr:| |z|kDr2g d fd }t;|| ||||||} | j= |r | |} n| |z } |j} |j }|r&t?j|rtAd| d|| j ||j} |dk(r^|dkDrA| || |zz z}| |krB|| cxkr|krnn| j d|r| |} | dz } | |kr0n| j d| dz } ItCtE|| }tG|||}|dk(r || cxkr|kr{nnx| j |nf| |k\r | |z|kr| j ||rF| D|} nA| |kr<| |z|kDr4| j |tI|| |ztK|| z z|r| |} | |z } | t|z } | t|kr d jM| | |fS)z Clip text without cursor movement (simple append-to-output path). Returns ``(result, captured_style, end_style)``. The caller applies SGR wrapping. rN  r r> sgr_paramsrMrNrOts_metats_textr/ts_termr?c(j|yN)rW)s_w_colts_partss r( _ts_writez_clip_simple.._ts_writeGs *r'*Indeterminate cursor sequence at position ,   rKrArB)r|r0r}r,r~r,returnNone)'rlenfindrmatchrWgrouprrLr parserTr+rr"r#r-r$r.r1r2r3rfinditerrr from_params display_width make_sequence_text_sizing_clipextendr ValueErrornextr r r[rFr^)!rJrKrLrDrArBrCr@strictr?outputrOidxr9r: current_stylecharnext_escmhl_statersgr_mrvrwrxtsts_widthrseqnext_tabgrapheme grapheme_wrs! @r( _clip_simplersN,F C C+/N&*I*7&TM D /Cy #:$n4) yyq1H2~C  6>"((s3A d#q ww|$0]}G` 1- K !-#)MM!''), -Ieeg+00;;x;&%h !#quuwC  88/888MM!''),%%'CXX!1!7!77++CXX!1!9!991==(C++CMM!**-MM!//2MM!++.$)?)6%)B%1%:%:1??%K6E,=mU[[],[M(5I61==(C++C779--:''),2''),***$00 VW&++O<%E#9*,H++CX$!#CMM(+$)?)68OCeeg'')C7==cB @Rg MM# %%'C  4<{'S7]";<Hn)c) c*(^-C-:N1HC Hn d# 1HC t3788_(46  ?!c! h' E\cJ.#5 MM( #!7!. 3Y3+e3 MM(c#sZ/?&@3uc?&RS T!7!. z s8}e D /h 776?NI 55r'c Tj|}||k\r%||z|krj||||zS||k\s||z|kr||zStd||z } t|||z|z } jj g} jj dkDrttjjj D]} | j| ftjj t| z D]} | jdfnz3_text_sizing_clip..flush..s2qad2sr/)sumrrMr r numerator denominatorvertical_alignhorizontal_alignrr^ terminatorrclear) flush_colrtextstotal_wrM pending_unitsscaler write_cellss r(flushz _text_sizing_clip..flushs ,-!1--2M22! ))//A-CJ1 II  !6!6 II $ $bii&@&@ B  vrwwu~r}} = K K M    .s C:r )rr,rr) rrrFr[rMrr rr rJrWranger)rrOrKrLrCrArBrr rel_startrel_endunitsgrlr flush_col_posunit_pos unit_textunit_wunit_endoverlap abs_startirrs` ` @@r(rrsY"0H e|h#-B$$&#6X~ czS8^u,X~Aus{#I#sX~&,G IIOOE#%E yyrww/A %A LL!U $ %ryyU34 &A LL"e % & ( ?A LLE!_'357<=> ? ? ,.M )OMH" 6f$ y H  w  h(3x+CC f ! #c(I&> >  )V!4 5 - c(I66I7^ 8HaQ7 8%( - >r'c  ()*+,-i)t+g-d,d} d} d(d} rtnd*d} d d!()*+,-fd }| t|kr|| }| |k\r|dvr| d| cxkr| krmnnjt|j d|j d|j d| krd} n*t j || }|dn|j} | dkr#(n|jd| d z}|dk(rn|} |dk(rwtj|| }|s-j| ,|f,d z ,| d z } |jd `r^*\t*|j*(.| |kr)-j| ,|jf,d z ,*} |j} mtj|jx}rt!||||||| ||j| }|j"t$j&ur9-j| ,|jf,d z ,|j} n3|j"t$j(ur|j*} n |j"t$j,ur| |j.z } |j*} nщ-j| ,|j0f,d z ,||j2|j4| d | |j4z } -j| ,|j6f,d z ,rB*@t9j:|j2D]}t*|j**} |j<} |j*} A|jdx}p|jdxsd}|jd}|Jt?tAjB|| ||}tE|| |||||} |j} |j}|r&tGj|rtId| d||jdx}$|rtK|d z nd} |j} 3|jdx}a|r tK|nd }| |z}| |kr4||kDr/tMt| |tO||D] }|d ||} |j} |jdx} I| r tK| nd }!|r|!| kDrtId| d|!d| d| |!z} | dkrd} |j} -j| ,|jf,d z ,|j} <|dk(r d} | d z } J|dk(r| dkDr| d z} | d z } `|dk(rb|dkDr=| || |zz z}"tMt| |tO|"|D] }#|dd |#|"} n-j| ,df,d z ,| d z } tQtS|| }$tU|$||}%|%dk(r(|| cxkr|kr{nnx-j| ,|$f,d z ,n^| |k\r| |%z|kr ||$|%| nF| |krA| |%z|kDr9t|| }&tMtO|| |%z|&z D]}'|d |&|'z| |%z } | t|$z } | t|krtW)-||(| fS)"z Clip text with cursor movement (painter's algorithm path). Returns ``(result, captured_style, end_style)``. The caller applies SGR wrapping. rNc|dk(r"|r j| |f dz ryt|D]}||z}|dkDr9j|dz dddk(r df|dz < j|dz j|dddk(r df|dz< j|dzj |d j|||f|<|r$t|D]} j ||z ryyy)zIWrite *w* cells of text *s* at *write_col*, handling wide-char splitting.rr N)r/r)rWrr\discardpopadd)r|w write_col is_hyperlinkoffsetsrc_colr9r_rrChyperlink_cellsrD seq_orderr`s r( _write_cellsz#_clip_painter.._write_cells sE 6  )Y!:;Q  ^%;%2N Ah -F&(G{uyy1g>qAQF&.]gk"''! 4yy'*1-2&.]gk"''! 4 IIgt $  # #G , -q6i ( 8##I$67 8 ^3*N4=r'rr rsr>r rtruT)rrvrwr/rxryrrhpa_n cforward_n cbackward_nz!Cursor left movement at position z would move z cells left from column z, exceeding string startrrrr)F) r|r0rr,rr,rboolrr),setrrrFrfindrsearchrLrrrrWrrr rrTr+rr"r#r-r$r.r1r2r4r3rrr5rrrrrrr,rr[rr r rp).rJrKrLrDrArBrCr@rr?rOrr: next_movementrrfoundrrrrrrvrwrxrrrr n_forwardmove_endrr n_backwardrfill_colrr clip_startrr9r_rrrr`s. ` ` @@@@@@r( _clip_painterrsu,)+E #O,.II C C+/N&*I*7&TM#'M+0+#'+48++: D /Cy #:$n4$](@S(@tzz&)4::d+;TZZ=OPSVV$&M7>>tSIE*/-BUYY[Mq !-99VS1W5r> 6>"((s3A  #y$!78Q q ww|$0]}G` 1- K !-#)$$c9aggi%@ANI -Ieeg+00;;x;&%h !#quuwC  88/888$$c9aggi%@ANI%%'CXX!1!7!77++CXX!1!9!991==(C++C$$c9ajj%ABNI !//3.241??*C$$c9akk%BCNI%)B%1%:%:1??%K6E,=mU[[],[M(5I6,,C++C779--:''),2''),***$00 VW&(UC?  "eeg'')C7==cB @Rg ))6(-c%j1n1eeg ggl33 @/9C Oq ?9E!1"3sE?C#4FG5$Xq!45eeg !ww}55 B1seD12 z!7Ceeg   c9aggi8 9 NI%%'C  4<C 1HC  6>Qwq 1HC  4<{'S7]";< %c#uos8S7I J3H a23  #y$!78Q 1HC t3788_(46  ?!c!  #y(!;<Q E\cJ.#5 :s 3 3Y3+e3UCJCz)9 :Z GH ?Xq*v*=> ? z s8}q D /t ! 5#x H I ''r'rTrF)rCr@rArDr? overtypingrBct|d}|dkr#|dk7rtd|tj}||kry|j r|j r|||St |}d|v} |r| sd}|6|dk7xr.d|vxs(d |vxs"| xrttj|}n |r|dk(rd}|rtnt} t| |||||| |||d k(| 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' rr>z5end must be -1 (to end of line) or non-negative, got r/rsFignorerrr) rJrKrLrDrArBrCr@rr?) rFrsysmaxsizeisascii isprintablerrrrrrr<) rJrKrLrCr@rArDr?rrBhas_escfn_clips r(rGrGs-n qME Qw "9GuMO Okk e| ||~$**,E#,_=OnGW  X % K t^Jtt|JH$:AA$GH   1 )m|G G  #'!)#   r'r{)r8r0r9Optional[_SGRState]r:rrr0)rJr0rKr,rLr,rCr0r@r,rAr,rB bool | strr?$Literal['parse', 'strict', 'ignore']rMr rNr,rOr,rr*) r_zdict[int, tuple[str, int]]r`zlist[tuple[int, int, str]]rKr,rLr,rCr0rr0)rJr0rKr,rLr,rDrrAr,rBrrCr0r@r,rrr?rrz4tuple[str, Optional[_SGRState], Optional[_SGRState]])rrrOr,rKr,rLr,rCr0rAr,rBrrzCallable[[str, int, int], None]rr,)rr>)rJr0rKr,rLr,rCr0r@r,rAr,rDrr?rrzOptional[bool]rBrrr0)-r __future__rrr itertoolsrtypingrrrr_widthr rr hyperlinkr r sgr_staterrrrrr _constantsr text_sizingrrescape_sequencesrrrEnumrr*r<rTrprrrrGr&r'r(rs<0" ::$100 /5>> tyyz$6:2>A,C C C C C  C  CC8C CC CCL4 %4)4 4 4  4  4n|6 |6 |6 |6  |6  |6|6|6|6 |68|6:|6~UU U U U  U  UU1U UpE' E' E' E'  E'  E'E'E'E' E'8E':E'Tc :A!%$c c c c  c  ccc8ccc cr'