o i"+@sdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z m Z ddl mZejGdd d ZejGd d d ZdddZddddd ddZdS)!zNexportglobalobject__all__ty-map import-objectimport-sectionN)r )selfrT/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/tvm_ffi/stub/file_utils.py __post_init__)szCodeBlock.__post_init__cCs*|jsdS|jd}t|t|dS)zECalculate the indentation level of the block based on the first line.r )rlenlstrip)r first_linerrrindent6s zCodeBlock.indentlineolinestrcCs|tjr|ttjd}td|||gdS|tjrQ|ttjd}dd|dD}t|dkrF|dgdt|7}td t|||gdS|tj sYJ|ttj d}|d rd }|td d}d |vrt|d }nW|df}nR|d rd}|td d}n@|drd}|tdd}n.|dkrd}d}n%|drd}|tdd}n|dkrd}d}n t d|d|t|||dgdS)zFParse a line to create a CodeBlock if it contains a stub begin marker.Nrr r r rrcSsg|]}|qSr)strip).0prrr Lsz-CodeBlock.from_begin_line..;rzglobal/r@zobject/rzty-map/rzexport/rrzUnknown stub type `z ` at line ) startswithC STUB_TY_MAPrr&r STUB_IMPORT_OBJECTsplittuple STUB_BEGIN ValueError)r"r#splitsstubr r rrrfrom_begin_line>sh        zCodeBlock.from_begin_lineNrr)rr )r"r r#r$rr ) __name__ __module__ __qualname____doc____annotations__rpropertyr! staticmethodr8rrrrr s   r c@sNeZdZUdZded<ded<ded<dd d ZeddddZdddZdS)FileInfoz)Information about a file being processed.rpathztuple[str, ...]rzlist[CodeBlock] code_blocksverbosebooldry_runrcCs tdd|jD}|j|kr|rttjdtjdS|r{tj|j|ddD]Q}| ds5| dr6q)| d rM| dsMttj |tjq)| d rd| dsdttj |tjq)| d rvttj |tjq)t|q)||_|s|j jd |jd d ddS)zTUpdate the file's lines based on the current code blocks and optionally show a diff.css |] }|jD]}|VqqdSN)r)r'blockr#rrr sz"FileInfo.update..z-----> UnchangedFr,)linetermz---z+++-+? utf-8encodingT)r3rCrprintr/ TERM_CYAN TERM_RESETdifflib unified_diffr.TERM_RED TERM_GREEN TERM_YELLOWrB write_textjoin)rrDrF new_linesr#rrrupdates(   zFileInfo.updateFfile include_emptyFileInfo | Nonec Cs|s Jd||}d}|jdd}t|ddD]\}}|tjr.dS|tj r8d}q|s?|s?dS~g}d}t|dD]\}}|} | tj rr|durbt d |t || }||_|j|qI| tjr|durt d tjd |||_|j|||d}qI| tjrt || } || _| j||| ~ qI| tjrt || } || _| j||| ~ qI| tj rt d |d | |dur|t dd|||gdqI|j|qI|durt dt|t||dS)z:Parse a file to extract code blocks based on stub markers.zExpected a file, but got: FrOrPr)startNTz"Nested stub not permitted at line z Unmatched `z` found at line zUnknown stub type at line z: r,r%z"Unclosed stub block at end of file)rBrrC)is_fileresolve read_text splitlines enumerater&r.r/STUB_SKIP_FILE STUB_PREFIXr4r5r r8r rappendSTUB_ENDrr0r1rAr3) r^r_ has_markerr_r#codescodelineno clean_linety_codeimp_coderrr from_filest               zFileInfo.from_filercCs0t|j}|dusJd|j|j|_dS)zCReload the code blocks from disk while preserving original `lines`.Nz File no longer exists or valid: )rArsrBrC)rsourcerrrreloads  zFileInfo.reloadN)rDrErFrErrE)F)r^rr_rErr`r9) r:r;r<r=r>r]r@rsrurrrrrAzs  CrApaths list[Path]rlist[FileInfo]c sddddfdd }t|}t|d d d }g}|D])}zt|}Wnty=}z |WYd }~qd }~ww|d urG||q|S)zLCollect all files from the given paths and parse them into FileInfo objects.e ExceptionrrcSs(ttjdttjddddS)Nz[Error] r,T)endflush)rRr/rW traceback format_excrT)ryrrr _on_errors  z collect_files.._on_errorGenerator[Path, None, None]c3shD].}|r |Vqt|ddD]\}}}|D]}t||}|jtjvr,q|VqqqdS)NF)follow_symlinkson_error)rb path_walkrsuffixlowerr/DEFAULT_SOURCE_EXTS)r(root_dirsfilesr^frrvrr_walk_recursives z&collect_files.._walk_recursivecSst|SrG)r$)rrrrszcollect_files..)keyN)ryrzrr)rr)listsortedrArsrzri)rvr filenamesrr^contentryrrr collect_filess   rTFtop_downrrr(rrrEr"Callable[[Exception], None] | Noner+Iterable[tuple[Path, list[str], list[str]]]ccsh|sdSt|dr|j|||dEdHdStj||||dD]\}}}t|}|||fVq"dS)z@Compat wrapper for Path.walk (3.12+) with a fallback for < 3.12.Nwalkr)topdownonerror followlinks)existshasattrrosr)r(rrrroot_strdirnamesrrrrrr s&  r)rvrwrrx) r(rrrErrrrErr)r= __future__r dataclassesrUrr}pathlibrtypingrrrr,rr/ dataclassr rArrrrrrs$   Z k'