AHj:NddlmZddlmZmZddlmZddlmZddl m Z m Z m Z m Z ddlmZmZe rddlmZmZddlmZdd lmZed Z eeeZ eGd d Zed ddZddZGddeZGddeZe e e fZ! Gdde Z"y)) annotations)ABCabstractmethod) dataclass) lru_cache) TYPE_CHECKING NamedTupleTupleoverload)Literalget_args)NodeQuery)cell_len)Size)   c(eZdZUdZded< ded<y) EditResultz5Contains information about an edit that has occurred.Location end_locationstr replaced_textN)__name__ __module__ __qualname____doc____annotations__R/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/document/_document.pyrrs?:%r!ri)maxsizec$|jdS)zEncode the input text as utf-8 bytes. The returned encoded bytes may be retrieved from a cache. Args: text: The text to encode. Returns: The utf-8 bytes representing the input string. zutf-8)encodetexts r" _utf8_encoder( s ;;w r!c"d|vryd|vryd|vryy)zReturn the newline type used in this document. Args: text: The text to inspect. Returns: The Newline used in the file. rrrr r&s r"_detect_newline_styler*/s%~  r!c^eZdZdZeddZeeddZeeddZeeddZ eddZ eddZ eddZ d dd Z dd Zeedd Zeedd ZeeddZereddZeddZed dZy )! DocumentBasezzDescribes the minimum functionality a Document implementation must provide in order to be used by the TextArea widget.cy)aCReplace the text at the given range. Args: start: A tuple (row, column) where the edit starts. end: A tuple (row, column) where the edit ends. text: The text to insert between start and end. Returns: The new end location after the edit is complete. Nr )selfstartendr's r" replace_rangezDocumentBase.replace_rangeFr!cy)z'The text from the document as a string.Nr r.s r"r'zDocumentBase.textSr2r!cy)z/Return the line separator used in the document.Nr r4s r"newlinezDocumentBase.newlineXr2r!cy)zGet the lines of the document as a list of strings. The strings should *not* include newline characters. The newline character used for the document can be retrieved via the newline property. Nr r4s r"lineszDocumentBase.lines]r2r!cy)aEReturns the line with the given index from the document. This is used in rendering lines, and will be called by the TextArea for each line that is rendered. Args: index: The index of the line in the document. Returns: The str instance representing the line. Nr )r.indexs r"get_linezDocumentBase.get_linegr2r!cy)aGet the text that falls between the start and end locations. Args: start: The start location of the selection. end: The end location of the selection. Returns: The text between start (inclusive) and end (exclusive). Nr r.r/r0s r"get_text_rangezDocumentBase.get_text_rangeur2r!cy)a<Get the size of the document. The height is generally the number of lines, and the width is generally the maximum cell length of all the lines. Args: indent_width: The width to use for tab characters. Returns: The Size of the document bounding box. Nr )r. indent_widths r"get_sizezDocumentBase.get_sizer2r!NciS)aQuery the tree-sitter syntax tree. The default implementation always returns an empty list. To support querying in a subclass, this must be implemented. Args: query: The tree-sitter Query to perform. start_point: The (row, column byte) to start the query at. end_point: The (row, column byte) to end the query at. Returns: A dict mapping captured node names to lists of Nodes with that name. r )r.query start_point end_points r"query_syntax_treezDocumentBase.query_syntax_trees ( r!cyNr )r.rCs r" prepare_queryzDocumentBase.prepare_querysr!cy),Returns the number of lines in the document.Nr r4s r" line_countzDocumentBase.line_countr2r!cy)9Returns the location of the start of the document (0, 0).rrr r4s r"r/zDocumentBase.startsr!cy)0Returns the location of the end of the document.Nr r4s r"r0zDocumentBase.endr2r!cyrHr r. line_indexs r" __getitem__zDocumentBase.__getitem__s7:r!cyrHr rSs r"rUzDocumentBase.__getitem__s?Br!cy)Return the content of a line as a string, excluding newline characters. Args: line_index: The index or slice of the line(s) to retrieve. Returns: The line or list of lines requested. Nr rSs r"rUzDocumentBase.__getitem__r2r!r/rr0rr'rreturnrrZrrZNewlinerZ list[str]r:intrZrr/rr0rrZr)r@rarZr)NN)rCz'Query'rDtuple[int, int] | NonerErcrZzdict[str, list['Node']])rCrrZz'Query | None'rZrarZrrTrarZrrTslicerZr_rTz int | slicerZzstr | list[str])rrrrrr1propertyr'r6r8r;r>rArFrIrLr/r0rr rUr r!r"r,r,Bsj;    66>>              /3,0 ,*  ! ,;;?? : : B B  r!r,ceZdZdZddZeddZeddZeddZddZ ddZ ddZ edd Z edfd Z edd Zdd Zdd ZddZeddZeddZd dZxZS)!Documentz-A document which can be opened in a TextArea.ct||_ |jd|_ |j t t s|s|jjdyy)NF)keepends)r*_newline splitlines_linesendswithtupleVALID_NEWLINESappend)r.r's r"__init__zDocument.__init__sT!6t!< 3!%%!@   ==~. /t KK  r "8rr)r.r/r0r'topbottomtop_row top_column bottom_row bottom_column insert_linesr8rafter_selectionbefore_selectiondestination_columndestination_rowrs r"r1zDocument.replace_rangesCeS\* V!$*! M( ==~. /    # ++C8 U # O#J/ ?O c%j ! $W~kz:  .a@LO!$\"%5!6 +B//AL !$%5!6 ,>?L*6g Q'!C $559');< , 66r!cd||k(ryt||f\}}|\}}|\}}|j} ||k(r | |} | ||} | S| |} ||jdz kr| |nd} | |d} t|dz|D]}| |j| |zz } ||jkr| |jz } | | d|z } | S)a"Get the text that falls between the start and end locations. Returns the text between `start` and `end`, including the appropriate line separator character as specified by `Document._newline`. Note that `_newline` is set automatically to the first line separator character found in the document. Args: start: The start location of the selection. end: The end location of the selection. Returns: The text between start (inclusive) and end (exclusive). rorN)rrrrLrangerp)r.r/r0rrrrrrr8r selected_text start_lineend_linerows r"r>zDocument.get_text_range/s C<eS\* V!$*! M  j >D M:MwJ,6$//A:M,MuZ(SUH&z{3MWq[*5 <s!;;  <DOO+. .=!99 r!c,t|jS)rK)rrrr4s r"rLzDocument.line_countUs4;;r!ct|S)rN)superr/)r. __class__s r"r/zDocument.startZsw}r!cT|jd}|jdz t|fS)rQrr)rrrLr)r. last_lines r"r0z Document.end_s)KKO !#S^44r!c|\}}|t|jz|z}t|D]}|t|j|z }!|S)zGiven a location, returns the index from the document's text. Args: location: The location in the document. Returns: The index in the document's text. )rr6rr;)r.locationrcolumnr:rTs r"get_index_from_locationz Document.get_index_from_locationesU Vc$,,''&0* 4J Sz23 3E 4 r!cRd|d}|dks|t|jkDr t|d}t|j}t |j D]@}|t|j |z|z}||kr |||z fcS||k(r |dzdfcS|}Bt|)aEGiven a codepoint index in the document's text, returns the corresponding location. Args: index: The index in the document's text. Returns: The corresponding location. Raises: ValueError: If the index is doesn't correspond to a location in the document. zIndex z3 does not correspond to a location in the document.rr)rr' ValueErrorr6rrLr;)r.r: error_message column_indexnewline_lengthrTnext_column_indexs r"get_location_from_indexz Document.get_location_from_indextsUIP Q  19DII.]+ + T\\*0 -Js4==#<==N (("EL$899++"Q**,L -''r!c||}|S)zReturns the line with the given index from the document. Args: index: The index of the line in the document. Returns: The string representing the line. r )r.r: line_strings r"r;zDocument.get_lines5k r!cyrHr rSs r"rUzDocument.__getitem__s36r!cyrHr rSs r"rUzDocument.__getitem__s;>r!c |j|S)rXryrSs r"rUzDocument.__getitem__s{{:&&r!)r'rrZNoner^r[r\)rrarZrrYrbrdre)rrrZra)r:rarZrr`rfrgri)rrrrrwrjr8r'r6rAr1r>rLr/r0rrr;r rU __classcell__)rs@r"rlrls7 #// -07d$L  55 (@ 66 >> 'r!rlc^eZdZUdZdZded< dZded< ed dZe d dZ d dZ y ) SelectionaA range of characters within a document from a start point to the end point. The location of the cursor is always considered to be the `end` point of the selection. The selection is inclusive of the minimum point and exclusive of the maximum point. rOrr/r0c|||S)zCreate a Selection with the same start and end point - a "cursor". Args: location: The location to create the zero-width Selection. r )clsrs r"cursorzSelection.cursors8X&&r!c|\}}||k(S)zBReturn True if the selection has 0 width, i.e. it's just a cursor.r r=s r"is_emptyzSelection.is_emptys s|r!cnt|jd|jdf\}}||k\xr||kS)z0Check if the given line is within the selection.r)rr/r0)r.yrrs r" contains_linezSelection.contains_lines8djjmTXXa[9: VCx'AK'r!N)rrrZz 'Selection')rZbool)rrarZr) rrrrr/rr0 classmethodrrjrrr r!r"rrsV E8C '' (r!rN)r'rrZbytes)r'rrZr])# __future__rabcrr dataclassesr functoolsrtypingrr r r typing_extensionsr r tree_sitterrrtextual._cellsrtextual.geometryrr]setrurr(r*r,rlrarrr r!r"rs"#!==/'#! $ %2Xg&'. && & 4    &G 3G Tc'|c'L c?I#( #(r!