jYGdZddlZddlZddlmZddlmZddlmZddl m Z ejZ dZ dZd Zd ZeGd d Zd edefdZdedefdZdededefdZdedededzfdZdedeeefdedzdefdZ d/dededeeefdedzddf dZ d/dededeeefdedzddf dZdeddfdZ d/d ed!ededeeefdedzdeef d"Zd#eede ejBedzeeefdzffd$Z"d%ed&ejBedzd'eeefdzd(edef d)Z#d!ed*eed#eeddfd+Z$d!ed#eeddfd,Z%e fdd-d!ededeeefdzdefd.Z&y)0a MDL material-directory relocation pass for the install pipeline. For each MDL in the staged mod tree, rewrites its materialDirectories[] so the mod's material trees live under a the console prefix, while preserving the un-relocated original paths as fallbacks. Pipeline (run in order): 1. relocate_mdl_dirs - rewrite each MDL's material_dirs (prefixed at original index, original appended for vanilla fallback) and physically move materials// -> materials/// 2. relocate_material_names - re-prefix path-rooted material name entries (some MDLs bake the path into the name when $cdmaterials is blank) 3. update_vmt_refs - rewrite $basetexture / $bumpmap / etc. references inside every VMT to point at the new location when the relocated VTF/VMT exists there N) dataclass)Path)MDLFile)moveconsole)rzvgui/replay/thumbnailsz[A-Za-z0-9_./\-]z[^A-Za-z0-9_./\-]c"eZdZUeed<eed<y) Relocationold_dirnew_dirN)__name__ __module__ __qualname__str__annotations__OC:\Users\pivot\Documents\cueki\casual-preloader\core\operations\mdl_relocate.pyr r 0s L Lrr pathreturncB|jddjS)N\/)replacelower)rs r_normr6s <<c " ( ( **rdir_strcr|jdjtfdtDS)Nrc3FK|]}j|dzyw)rN) startswith).0root normalizeds r z&_is_under_safe_root..<s!M_Tz$$TCZ0_s!)lstripranySAFE_PATH_ROOTS)rr"s @r_is_under_safe_rootr':s+$**,J M_M MMrprefixc|r@|dk(s;t|s0|jj|djr|S|jd}|d|S)aePrepend prefix to a material dir. Empty and lone-root entries (`/`) are left untouched. Prefixing them would (a) be meaningless as a search path, (b) scoop up the entire materials/ tree on disk if we tried to move it, and (c) generate a text-replace rewrite of `/` -> `console/` that catastrophically corrupts every VMT it touches. r)r'rrr$)rr(strippeds r add_prefixr+?s]  c> w ' ==? % %l&8&8&: ;~~c"HXQxj !!rr!rel_pathct|jd}|s|jr|SdS|}|jdD]o}|r|dk(r |dk(r&||k(ry|j}|jsy6|j  t fd|jDd}|y|}q|S#ttf$rYywxYw)a(Resolve `rel_path` under `root` case-insensitively, the way the Source engine does on Windows. Linux is case-sensitive, but a mod authored on Windows can ship files whose case differs from the MDL's references; plain `Path.exists()` would say 'missing' and trigger spurious work. rN.z..c3^K|]$}|jjk(s!|&ywN)namer)r etargets rr#zresolve_ci..is%S%6!&&,,.F:R!%6s"--) rstripexistssplitparentrnextiterdirOSErrorNotADirectoryError)r!r,relcurrentpartmatchr3s @r resolve_cir@Ss /   $C {{}t.$.G #ts{  4<$nnG>>#  SW__%6SUYZE =#$ N +,  s #B99C  C src_file file_origin mdl_addon_idxc@|y|j|}|duxs||k(S)zDTrue if src_file should move alongside the MDL being relocated. NT)get)rArBrCorigins r_origin_matchesrGrs/  __X &F T> 4V}44rsrcdstcp|g}|jdD]J}|js|j|d}|)|j|j ||fLt |||D] \}}||||z <y|j ddt|jdD]z}|jst|||s"|j |}||z } | jj dd|j|d}t || |v||| <|t|y)zuMove src tree to dst (which must not exist) and migrate file_origin keys from the old paths to the new ones. N*T)parentsexist_ok) rglobis_filepopappend relative_tormkdirlistrGr7_sweep_empty_subdirs) rHrIrBrCcapturedfidxr<rAdst_files r_move_with_originrZ}s++-3Ayy{!ooa.?OOQ]]3%7$=>  S# HC%(Kc "!IIdTI*3(! xmD ""3'9dT:ooh- Xx ?$'K !)rct|jdD]}|jst|||s"|j |}||z }|j |d}|j rJ|j|d}||kDr"|jt|||||<|jt|||dk7s|||<t|y)aFile-by-file merge of src tree into an existing dst tree. Per-file collisions are resolved by load order via file_origin: the file from the later-loaded addon wins. Files without an origin entry (e.g. config files copied outside the addon staging loop) are treated as lowest priority. This is the scenario where mod A ships content under `materials/console/...` and mod B ships overlapping content under the un-prefixed path; without merging here, mod B's content would be stranded at the un-prefixed location. When mdl_addon_idx is set, only files belonging to that addon (or with no recorded origin) are considered for merging; cross-addon files stay at src to be reached via the MDL's un-prefixed fallback dir. rKN) rTrNrOrGrRrPr5rEunlinkrrU) rHrIrBrCrAr<rYsrc_idxdst_idxs r _merge_intor`s$3(! xmD ""3'9//(B/ ?? !ooh3G !Xx((/ H%! 8 $"}(/ H%)),rc|jdDcgc]}|js|}}t|ddD]} |jycc}w#t$rY'wxYw)zRemove now-empty subdirectories of src, deepest first, so the caller's rmdir of src succeeds when all files moved out. Non-empty branches (e.g. cross-addon files left behind by scoped moves) are left intact. rKc,t|jSr0)lenpartsps rz&_sweep_empty_subdirs..s S\rTkeyreverseN)rNis_dirsortedrmdirr:)rHrfsubdirssubs rrUrUsd ))C.7.QAHHJq.G7g#94H  IIKI8   sAAA A'&A'mdl_path working_rootc t|}|dz }g}g}g} |j|} tjd| d|jD]} t | j d} t| |} | | k(r|j| A| jd}| jd}t||}t||}||z }|-|+|j| tjd|d|j| | |vr|j| | jt| jddz| jddztjd| d | d | d |tjd |d Q|+t||||tjd|d|dn*t||||tjd|d|d |j|j D]}||k(r |j|j#||z| S#t$rYNwxYw#t$rYwxYw)aRewrite one MDL's material_dirs to use the prefix and move on-disk material trees to match. For each relocated entry we put the prefixed path *at the original index* (so the engine searches mod content first within that slot) and append the un-relocated original to the end of the list. Keeping the original gives the engine a path to reach vanilla VMTs that the mod doesn't ship + the mod sometimes populates a directory only partially and relies on engine fallback for shared textures (eyeballs, masks, invuln effects, etc.). Without the fallback path in the search list, those vanilla materials become unreachable. materials[]rzskip: materials/z/ not present in mod)r r zrewrote material_dir:  -> z (kept z as fallback)zon-disk: materials/z'/ already exists (moved by another MDL)zmerged on-disk: materials/z/ -> materials/zmoved on-disk: materials/)rrRlogdebug material_dirsrr$r+rQrstripr@r r`rZrmr:rLrewrite_material_dirs)rprqr(rBrCmdlmaterials_rootnew_dirs fallback_dirs relocationsrel_mdlrawoldproposedold_relnew_relrHrI canonical_dstr7s r_relocate_one_mdlrss& ( C!K/NH!M$&K""<0GII'!n  Cj$c6* (? OOC **S///#&11&0 ;3; OOC II( 1EF G ! m #   %:JJsOc)OOC(3.   *3'h\#P]^_ ; II+G94[\ ]  ? ]K G II27)?7)STU V c=+} M II1'/'RST U  IIKkkF'   "o!~h67      s$6I I II I&%I&rc @t|Dchc]}|j|jfc}dd}|sydjd|D}t j dt dtd |d td tj}t|}||fScc}w) a;Build a single-pass regex + mapping from a list of relocations. Prefixes are sorted longest-first so a more specific rule wins. The path-char lookbehind anchors the match to a path-token boundary; without it, an already-prefixed path would match an inner rule and get a doubled prefix. The extra `(?<=materials/)` alternative carves out the one place where matching after a `/` is desirable: Patch VMT `include "materials/..."` directives. Without that carve-out, those includes never get rewritten when their target file gets relocated. ct|dSNrrcres rrgz&_build_rewrite_regex..J c!A$irTrh)NN|c3LK|]\}}tj|ywr0)reescape)r r_s rr#z'_build_rewrite_regex..Os=uVS!"))C.us"$z(?:(?<=materials/)|(?<=z/)|(?z )(?Pz*)) rlr r joinrcompile_PATH_BOUNDARY_CHARS_PATH_TAIL_CHARS IGNORECASEdict)rrpairs prefix_altpatternmappings r_build_rewrite_regexr;s )45A!))QYY 5  E =u==Jjj !"6!7w?O>PQ \,<+=R A G 5kG G  6sBtextrrr}c~||Sdtjtdtffd }|j||S)zRewrite path refs whose target file exists at the new location; leave refs that don't (they're relying on engine fallback to vanilla). mrc|jd}|jd}|j}|s|S||z}|jddd}d|vr|f}n|dzf}tfd|Dr||zS|jd S) Nr(tailrr\r.z.vtfc3:K|]}t|duywr0)r@)r cr}s rr#z8_apply_path_rewrites..replace..qsM*Qz.!,D8*sr)grouprrsplitr%) rr(r new_prefixfull_new last_segment candidatesrr}s rrz%_apply_path_rewrites..replacees"wwvV\\^,  ${{3*2. , "J"V+-J M*M M$ $wwqzr)rMatchrro)rrrr}rs `` r_apply_path_rewritesrYsA'/ 288C=S ;;w %%r mdl_pathsc  |syt|Dchc]}|j|jfc}dd |dz dtdtf fd }|D]}t |}|j Dcgc] }|| }}||j k(r=|j |t|j |D]3\} } | | k7s tjd |jd | d | 5ycc}wcc}w) aRewrite per-material name strings whose path-rooted prefix matches a relocation. Some MDLs reference materials by full path (e.g. material name 'models/weapons/c_items/c_buffbanner' paired with an empty material_dir); the engine resolves these via 'materials/.vmt'. When that on-disk location has been moved, the name itself must be re-prefixed. Each rewrite is gated on the candidate VMT existing at the new location. Without that check, an MDL that references both a mod-shipped material *and* a vanilla material under the same prefix would have both names re-prefixed; the vanilla one would then point at materials/console/... where nothing was shipped, and the engine has no path back to the original vanilla VMT. Nct|dSrrres rrgz*_relocate_material_names..rrTrhrsr1rct|jd}d|vr|SD]M\}}|j|s||t|dzdz}t ||cS||t|dzcS|S)Nrz.vmt)rr$rrcr@)r1r"rnew candidater}rs r rewrite_namez._relocate_material_names..rewrite_names4['', j KHC$$S)*SXY"77&@ ni8@KZC 222   rrtz] rewrote material name: rv) rlr r rrrsrewrite_materialsziprwrxr1) rqrrrrrpr|r new_namesrrr}rs @@r_relocate_material_namesrxs  )45A!))QYY 5  E "K/N 3 3 h.1mm3R2ST U -r)rBc @t|jd}|sy|i}tjdt |d|dg}|D]2}|j |} |j t|||||4 t||| t||t |S#t$rtjd|dYwxYw#t$rtjdY`wxYw#t$r"tjd Yt |SwxYw) aRun the full MDL-relocation pipeline against a staged mod tree. `file_origin` maps each staged file path to its source addon's load-order index; the merge step uses it to resolve per-file collisions when two mods ship overlapping content (one pre-prefixed under `console/`, the other not). If omitted, collisions silently keep the dst file. Returns the number of MDLs found. z*.mdlrzRelocating material paths for z MDL(s) under prefix ''zMDL relocation failed for z; leaving as-isz MDL material-name rewrite failedzVMT reference rewrite failed) rlrNrwinforcrEextendr Exception exceptionrr)rqr(rBrall_relocationsrprCs rrelocate_mdl_pathsrs)|))'23I  HH-c)n-==STZS[[\ ]^(*O#1  R  " "!(L&+}] : y/J67 y> R MM6xjP Q R : 89: 6 45 y>6s6 B& C C2&"C  C C/.C/2DDr0)'__doc__loggingr dataclassesrpathlibr valve_parsersrcore.util.filer getLoggerrwDEFAULT_PREFIXr&rrr rrboolr'r+r@rintrGrZr`rUrTrtuplePatternrrrrrrrrrs ( !!g 8 ',   +++NNN ""S"S"(TSTD[>55!%dCi5ADt5 5!%# ##'+D#I#:# #P!%( (('+D#I(:( (V d t $!% ]]] ]dCi ] : ]  * ]@j! 2::c?T !4S>D#8 89<& & ZZ_t #&#s(^d "& &  &>+Z4+ZDJ+ZUYZdUe+Zjn+Z\V4Vd:6FV4V,!,+/ ,, ,dCi4' ,  ,r