Cr`F$ddlZddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z ddlmZmZmZmZddlmZd Zd Zd Zd Zd ZeeeegZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&y)N)_)getattr)errormetadatapycompat requirementsscmutilstoreutilvfs) constantsflagutilnodemapsidedata)actionsc p|jjdd}tjxs| }|sd|j t j tjtjftj||tjdtj||jS)Ns experimentalsworker.repository-upgradeT)replace)ui configboolr iswindowsregister_sidedata_computer revlogconstKIND_CHANGELOG sidedatamodSD_FILESr_get_worker_sidedata_adderrREVIDX_HASCOPIESINFOget_sidedata_helpers_wanted_sidedata)srcrepodstrepouse_w sequentials Busr/lib/python3.12/site-packages/mercurial/upgrade_utils/engine.pyr r s JJ ! !/3O PE##05yJ **  & &   ! ! #  / / A  ) ) +   + +GW5M5M NNc:t|d|}|j|}t|d|}|j}|j}|j|j}|j|j}|j|j } |j|j } ||jd5 dddt j|||jj|j } | rt j| | |jrd|j} |jjj| | r,|jjj| dddzyyy#1swYxYw)a copy all relevant files for `oldrl` into `destrepo` store Files are copied "as is" without any transformation. The copy is performed without extra checks. Callers are responsible for making sure the copied content is compatible with format of the destination repository. _revlogwNs.d)rget_revlog_instanceopenerjoin _indexfile _datafiler copyfileexists is_filelogmain_file_pathsvfsfncacheadd) trdestrepooldrlentrynewrloldvfsnewvfsoldindexnewindexolddatanewdatacopydata unencodednames r& _copyrevlogrE/sH E9e ,E  % %h /E E9e ,E \\F \\F{{5++,H{{5++,Hkk%//*Gkk%//*G   $ '  ( MM(H%||""5??3H gw' ,,.  !!-0  MM ! ! % %mCR&85&@ A  ( 's FFs changelogsmanifests all-filelogsc~|jrt|vS|jrt|vS|jsJt |vS)zcheck if a revlog is selected for cloning. In other words, are there any updates which need to be done on revlog or it can be blindly copied. The store entry is checked against the passed filter) is_changelogUPGRADE_CHANGELOGis_manifestlogUPGRADE_MANIFESTr3UPGRADE_FILELOGS) revlogfilterr;s r& matchrevlogrMYsF  L00   <//     | ++r'cd}|j} t|j|rf|jt dt || fz|j |}|j||||j|j||St d} |j| | t |fzt|||||j |}|S)z%returns the new revlog object createdNscloning %d revisions from %s ) addrevisioncb deltareuseforcedeltabothparentssidedata_helperss+blindly copying %s containing %i revisions ) r4rMrevlogs_to_processnoterlenr,clonedelta_reuse_modeforce_re_delta_both_parentsrE) rr#r8 old_revlogr; upgrade_oprRoncopiedrevisionr< revlog_pathmsgs r&_perform_cloner^hs E&&(K:00%8  / 0: , - ))'2  *!22","H"H-   L ?@ {C O445BU3))'2 Lr'c  %d}d}d}d}d} d} d} d} d} d}d}d}d}d}d}d}d}d}t|jj}i}i}i}|D]}|js|j |}|j dddd}||dxsdz }|dxsd}|dxsd}||z }||z }|j r(|||j<|t|z }||z }||z }|jr-|||j<|dz }|t|z }||z }||z }|jr.|||j<| dz } | t|z } | |z } | |z } tjd|sy |jtd || ||fz|jtd tj |tj |fzd %%fd } t#||}!|jtd | | tj | tj | fz|j$j'td| %t)|j+D]F\}"}|j |}#t-||||#|||!| }$|$j d}| |dxsdz } H|jtd| | tj | | z fz|jtd||tj |tj |fz%r%j/|j$j'td|%t)|j+D]F\}"}|j |}#t-||||#|||!| }$|$j d}||dxsdz }H|jtd||tj ||z fz|jtd|tj |tj |fz%r%j/|j$j'td|%t)|j+D]F\}"}|j |}#t-||||#|||!| }$|$j d}||dxsdz }H%j/|jtd|tj ||z fz| |z|z}|jtd|tj ||z fzy )zCopy revlogs between 2 repos.rT)exclusivefilesrevisionscount trackedsize storedsizesrevisionscounts storedsizes trackedsizersunknown revlog typeNsPmigrating %d total revisions (%d in filelogs, %d in manifests, %d in changelog) s'migrating %s in store; %s tracked data c&jy)N) increment)rlrevnodeprogresss r&r[z'_clonerevlogs..oncopiedrevisionsr'sMmigrating %d filelogs containing %d revisions (%s in store; %s tracked data) sfile revisions)total)rcsOfinished migrating %d filelog revisions across %d filelogs; change in size: %s sNmigrating %d manifests containing %d revisions (%s in store; %s tracked data) smanifest revisionssQfinished migrating %d manifest revisions across %d manifests; change in size: %s sKmigrating changelog containing %d revisions (%s in store; %s tracked data) schangelog revisionss>finished migrating %d changelog revisions; change in size: %s sFfinished migrating %d total revisions; total change in store size: %s )listr walk is_revlogr, storageinforG target_idrUrIr3rProgrammingErrorstatusrr bytecountr r makeprogresssorteditemsr^complete)&rr"r#r8rZrevcountsrcsize srcrawsizedstsizefcount frevcountfsrcsizefrawsizefdstsizemcount mrevcountmsrcsizemrawsizemdstsize crevcountcsrcsizecrawsizecdstsize alldatafiles manifests changelogsfilelogsr;rfinfodatasizerawsizer[rRror:r<ris& @r& _clonerevlogsrs+HGJG FIHHH FIHHHIHHH **,-LIJH   & &w /~~   D*+0q0 &+!~&+!8g    */Ju ' R I  H  H  ! !).Ieoo & aKF R I  H  H   (-HU__ % aKF R I  H  H  " "#9 :OR II  "  Y 9 5  6II 56 NN7 #T^^J%? @ B H,GW=II  0   NN8 $ NN8 $   zz&&q):';9&MH"8>>#34 5))'2            D 1D',1,5II  .  fdnnX-@A B  CII  0   NN8 $ NN8 $   zz&&  'H#9??#45 5))'2            D 1D',1,6II  /  fdnnX-@A B  CII  0  NN8 $ NN8 $   zz&& !'H#:#3#3#56 5))'2            D 1D',1,7 II    dnnX%89 :  ;!H,GII    T^^Gg$56 7  8r'c#Kt|jjjddD]K\}}}tj|r|j dr/|t jk7rC|dvrH|Myw)zOyields files which should be copied to destination after revlogs are clonedr'T)statsundo)slocksfncacheN)rtr r readdiris_revlog_file startswithrS_IFREG)r"pathkindsts r& _files_to_copy_post_revlog_clonerpsz!!2!2!:!:3T!:!JKdB    %  ??7 #  4<<   ( (  LsBBc |jr0tj|j|j dn|j j ddtj|j|jy)aReplace the stores after current repository is upgraded Creates a backup of current repository store at backup path Replaces upgraded store files in current repo from upgraded one Arguments: currentrepo: repo object of current repository upgradedrepo: repo object of the upgraded data backupvfs: vfs object for the backup path upgrade_op: upgrade operation object to be used to decide what all is upgraded sstoreT)forciblyN) backup_storer renamespathr.r rmtree) currentrepo upgradedrepo backupvfsrZs r&_replacestoresrs[ K%%y~~h'?@x$7KK ""K$5$56r'cy)zHook point for extensions to perform additional actions during upgrade. This function is called after revlogs and store files have been copied but before the new store is swapped into the original location. N)rr"r#r s r&finishdatamigrationrsr'c|jsJ|jsJd}d}|jtdtj|jvrR|jtdt |||dd|jj tjtj|jvrR|jtdt |||dd|jj tjtj|jvrS|jtdt|||d |jj tjnntj|jvrR|jtd t|||d |jj tj|js |jsy|jr<|jtd tj||j|St|jd k(rd|jvr|js|jtd|j!d5}|j#}|j$}t'j(||dt'j(||j*j,j.ddddtj||j|St|jd k(r|jD cgc]} | j0dk(r| c} r|js|jtd|j!d5}|j#}|j$}t'j2|||t'j2|||j*j,j.dddtj||j|S|j!d5}t5|||||dddt7|D]} |j8jtd| z|j:j<j?| } |j:j<j?| } tAjB| | dtE|||tF|jtd|jHrttKjLd|jN}tQjR|}tAjB|jRj?d|j?d|jtdtj||jFdhz|jtd|jHr|jtd|z|jtdtAjT} tW||||tAjT| z }|jtd|z|jtdtj||j|jHr|J|jYd |S#1swYxYwcc} w#1swYxYw#1swYxYw)!ayDo the low-level work of upgrading a repository. The upgrade is effectively performed as a copy between a source repository and a temporary destination repository. The source repository is unmodified for as long as possible so the upgrade can abort at any time without causing loss of service for readers and without corrupting the source repository. NsP(it is safe to interrupt this process any time before data migration completes) s!upgrading to dirstate-v2 from v1 sv1v2s#downgrading from dirstate-v2 to v1 s"create dirstate-tracked-hint file T)r7s"remove dirstate-tracked-hint file Fs"upgrading repository requirements rspersistent-nodemaps7upgrading repository to use persistent nodemap feature supgrade)forces=downgrading repository to not use persistent nodemap feature s copying %s )copystats.data fully upgraded in a temporary repository upgradebackup.prefixdirrequiress\marking source repository as being upgraded; clients will be unable to read from repository supgradeinprogresss*starting in-place swap of repository data 'replaced files will be backed up at %s sreplacing store... sCstore replacement complete; repository was inconsistent for %0.1fs sBfinalizing requirements file and making repository readable again s store/lock)- currentwlockrqrupgrade_actions dirstatev2upgrade_dirstateremoveremoved_actionsdirstatetrackedkeyupgrade_tracked_hintrequirements_onlyr writereporequirementsnew_requirementsrUupgrade_actions_names transaction unfiltered changelogrpersist_nodemap manifestlog _rootstorer)namedelete_nodemaprrrr rawvfsr.r r1rr rrmkdtemprvfsmodr timerrunlink)rr"r#rZ backuppathrr8unficlxpsrcdsttstartelapseds r&upgraders    !! !    !! !JIII  + !!Z%?%?? !9:;Wj%?""))/*D*DE!!Z%?%?? !;<=Wj%?""))/*D*DE))Z-G-GG !:;<R*$?""))/*L*LM  + +z/I/I I !:;<R*%@""))/*L*LM  & &**D*D## !:;<%%gz/J/JKD { J & &'1, !Z%E%E E** I J   ,%%'DB  # #B$ 7  # #D$$//77t - %%gz/J/JKV S J & &'1, // /vv.. / ** O P   ,%%'DB  " "2w 3  " "GT--88@@  - %%gz/J/JKl i  ,   -2':A JJ  a014 5--&&++A.C--&&++A.C MM#sT 2 ; B,? !FGH  " "!))(gllJ :.I MM   -y~~k/J  4  %% W))-A,BB  !BCD  " " II=>K  !+,-wJ?**,'        %%gz/J/JK  " " ( ((   ] + m- , - ,- ,s,:A+\5\#A)\(+\5\ (\25\?c|jrtjd|j}|j t d|zt j|}tj|jjd|jd tj|jjd|jd|jj|dk(k(sJ|dk(}|r tj||j |jj"j%||j_ ||jj"_ d|j_ |jj)d|jj+d|s!tj||j yy#t$rY wxYw#t$rYZwxYw)NrrrrsdirstaterT)rrrrrqrrr r r1r.FileNotFoundErrordirstate_use_dirstate_v2r rr_mappreload_dirtyrwrite)rr"rZoldnewrruse_v2s r&rrVs%%$',,  !?@:MNJJz*  KK  [ )9>>++F   MM   -y~~k/J     , , >> > E\F %%gz/J/JK !!#(.G%-3G*"G  ;'  4 %%gz/J/JK /!   "    s$>GG. G+*G+. G:9G:c|rNd|j_d|j_d|j_|jj d|s|jj t j||jy)NT) r_use_tracked_hintr_dirty_tracked_setrdelete_tracked_hintr rr)rr"rZr7s r&rrsn -1*"&.2+t$ ,,. !!':+F+FGr')'ri18nrrrrrr r r r r r revlogutilsrrrrrrrrr rErHrJrK frozensetUPGRADE_ALL_REVLOGSrMr^rrrrrrrrr'r&rs     ) OBB!"(*:; ,!HaH*7,pf'LT Hr'