C|ddlZddlZddlmZddlmZmZmZddlm Z dZ dZ dZ d Z d Zd Zgd Zdd Z ddZGddej&ZedZdZ ddZy)N)_)errorpycompatutil) stringutilsalls prefinalizes postfinalizecfd}|S)Ncd|jdk(rtjd|g|i|S)Nrs;cannot use transaction when it is already committed/aborted)_countrProgrammingError)selfargskwdsfuncs 9usr/lib/python3.12/site-packages/mercurial/transaction.py_activezactive.._active!s< ;;! ((N D(4(4(()rrs` ractiver s) Nr%s.backupfiles))stores %s.narrowspec)plains%s.narrowspec.dirstate)rs %s.branch)rs %s.bookmarks)rs %s.phaseroots)rs %s.dirstate)rs%s.desc)rs%sc8t|z}g}g}|d} ||5}t||}ddd|D]%\} }} } | |vs | s|j|| | f'|j||ftD]\} } |j|| | |zf |D]\}} |j|y#1swYxYw#t$rc}|jtjk7r(6t^,-:)6(+Y-DEF ;'  NN8 $(" !  77ell "./C DIIn-z/F/Fq/IJ JC 3K   ww%,,&23 X. 0G0G0JKKs  sGB> B2B>D-2B;7B>> D*AD%%D*- F6AFFcTgfd}i} |D]?} | \} } } }|| }|jd|jdk(}|s5| s8| s;| | | <At}tt|j D]\}}| j |}|-|\} } } }d}|r| | f|v}||| | ||j ||s|s}|xr|df|v} ||d|}|j|kr2tjtd|||jfz|j||j |j||D]Y} | |vr| \}}}}||vr|r d ||fz||} |xr||f|v}|r|r |||||n|xs|} |j|[d |z}|j#|r|j||j| D](\}}|j#|s|j|*y#t$rtd|zwxYw#t$rYwxYw#t$rYwxYw#tt tjf$r|sY(wxYw#tt tjf$rYywxYw) zrollback a transaction : - truncate files that have been appended to - restore file backups - delete temporary files c|j|}|j|} tj|||j||fy#t$r1}t j |} td||fzd}~wwxYw)N checkambigsfailed to recover %s (%s) )r rcopyfiler"IOErrorrr!r) vfsr,br9filepath backuppathexce_msg backupfilesr%s rrestore_one_backupz%_playback..restore_one_backupss~88A;XXa[   MM*h: F   Qx ( ++C0E 1345zA B  s+A B ,BB rNFar8sBattempted to truncate %s to %d bytes, but it was already %d bytes sfailed to truncate %s s.couldn't handle %s: unknown cache location %s r)r setsorteddictitemsgetaddtellrAbortrtruncatecloser;r$FileNotFoundErrorrexists)journalr%openerr&entries backupentriesr$checkambigfilesrC store_backupentryr/ file_pathr0cacher<is_store copy_doner,o bck_entryr9fplr=r1targetr?rBs ` @r _playbackrabsK L27/)[%X88C=FKK$44  k&+L #  ItG}**,-1!$$Q'  6? 3HieJ'2oE vy+z J MM) $ F(HaX-HJ At ;779q=++2a+ , A   a A.H I   1a F?q EAN OQi (FaV-FJQ"31j9a JJv&@#W,J }}Z  j! MM' !FCzz!} 1 "_ q34q89 %  $)  %++.   Wekk *   sgA1HH? I#I/J JH<? I  I  IIII!JJJ'&J'ceZdZ d'dZdZdZedZedZ edZ edZ d Z ed(d Z d Zed)d Ze d*dZedZd efdZedZedZedZed+dZdZdZdZedZedZedZedZedZedZ edZ!edZ"ed Z#ed!Z$d"Z%d#Z&d$Z'd%Z(d&Z)y), transactionNc d|_d|_||_||_|j }||d<||_||_i|_t|_ ||_ g|_ ||_ g|_ | d} | |_t|_| r|jj!| | g|_i|_i|_|j)|jd|_g|_i|_d|jz|_|j)|j0d|_|j2j5dt6zt|_|>|j;|j|d z|j;|j0|d zi|_i|_d |_ i|_!i|_"|||jDd <i|_#i|_$y) a Begin a new transaction Begins a new transaction that allows rolling back writes in the event of an exception. * `after`: called after the transaction has been committed * `createmode`: the mode of the journal file that will be created * `releasefn`: called after releasing (with transaction and result) `checkambigfiles` is a set of (path, vfs-location) tuples, which determine whether file stat ambiguity should be avoided for corresponded files. rrNcyNr)trsuccesss rz&transaction.__init__.. sDrsw+rw%d iFs 001-userhooks)%r _usages_report_openercopy_vfsmap_after _offsetmaprE _newfiles_journal_journal_files _undoname_queue _releasefn_checkambigfilesupdate_nameschangeshookargsopen_file_backupentries _backupmap_backupjournal _backupsfilewriteversion _tmp_fileschmod_filegenerators_pendingcallback _anypending_finalizecallback_validatecallback_postclosecallback_abortcallback) rr%rRr& journalnameundonameafter createmode validator releasefnrUnames r__init__ztransaction.__init__s6    s   #  !  0I# #   ! ! ( ( 9f   [[6 !/$--?"KK(;(;TB ' 12%  ! LL U(: ; LL,,j5.@ A " " !#"$  7@D " "#3 4"$ rcpdj|j}d||j|jfzS)N/z*)r r{r rlrrs r__repr__ztransaction.__repr__>s7yy%;  KK LL?   rc>|jr|jyyrf)rt_abortrs r__del__ztransaction.__del__Fs == KKM rc|jduSrfrrs r finalizedztransaction.finalizedJs%%--rc:|jjgy)zdelay registration of file entry This is used by strip to delay vision of strip offset. The transaction sees either none or all of the strip actions to be done.N)rwr"rs r startgroupztransaction.startgroupNs 2rcp|jj}|D]\}}|j||y)zapply delayed registration of file entry. This is used by strip to delay vision of strip offset. The transaction sees either none or all of the strip actions to be done.N)rwpop _addentry)rqr,r\s rendgroupztransaction.endgroupVs0 KKOO DAq NN1a rc||jvs*||jvs||jvs||jvry|jr!|jdj ||fy|j ||y)z5record the state of an append-only file before updateN)rsrrrrrwr"rrfileoffsets rrJztransaction.add`si DNN "t&t&t&  ;; KKO " "D&> 2  tV$rcD||jvs*||jvs||jvs||jvry|r||j|<n|jj ||j j d||fz|j jy)z3add a append-only entry to memory and on-disk stateN%s%d )rsrrrrrJrrflushrs rrztransaction._addentryps DNN "t&t&t&  $*DOOD ! NN  t $ f~56 rrc|jrd}tj|||jvs||jvry||j vr|sy|r,||j vrd}|||fz}tj||j |}|j|\}}d|j|fz} |j|| } |j|r||j|} |j| } tjj| } tj | |j"dtj$| | |nd} |j'||| d fy) aAdds a backup of the file to the transaction Calling addbackup() creates a hardlink backup of the specified file that is used to recover the file in the event of the transaction aborting. * `file`: the file path, relative to .hg/store * `hardlink`: use a hardlink to quickly create the backup If `for_offset` is set, we expect a offset for this file to have been previously recorded s/cannot use transaction.addbackup inside "group"NzKcalling `addbackup` with `for_offmap=True`, but no offset recorded: [%r] %rs%s.backup.%s.bckT)mode notindexedhardlinkrF)rwrr rsrrrrpsplitrtreljoinrPr ospathdirnamermakedirsrr:_addbackupentry)rrrr/ for_offsetr.r<rfilenamebackupfilename backupfiler>r?final_dir_names r addbackupztransaction.addbackupsD ;;DC((- - 4>> !TT__%<  T__ $Z  D72  Hd# #C((- -ll8$IIdO, x/HH[[.9 ::d xx~H*-J WW__Z8N MM.s~~$ O MM(J BJ hj%@Arc|jj|t|jdz |j|d<|jj d|z|jj y)z0register a new backup entry and write it to diskr %s%s%s%d N)rr"lenrrrr)rrWs rrztransaction._addbackupentrysd ""5)$'(;(;$E9gx]-,,x(CE %DFND(((A%* th?&*H%59N9N%N LLD$4JO&AGGI!HAIIK=>@ AIIKs 2BDD)cV||jvry|jj|SNr)rsrrrI)rrs r findoffsetztransaction.findoffsets& 4>> !""4((rc|jjdg}|jjD]2}|jd\}}|j |t |f4|S)Nr)rseek readlinesrr"int)rrSr_rtroffsets r readjournalztransaction.readjournal!s[ %%'AWWU^ND( NND#h-0 1(rc||jvr.|sy|jj|||j|<nT||jvr;|s)|j|=|jj|n||j|<n t ||j j d||fz|j jy)zn replace can only replace already committed entries that are not pending in the queue Nr)rsremoverrrJKeyErrorrrrrs rreplaceztransaction.replace*s 4>> ! NN ! !$ '$*DOOD ! T__ $OOD)""4((.%4.  f~56 rc|xjdz c_|xjdz c_|jj||SNr)r rlr{r"rs rnestztransaction.nest@s5 q    4  rc|jdkDr|xjdzc_|jr|jj|jdkDr!|jdk(r|j yyy)Nrr)r rlr{rrrs rreleaseztransaction.releaseGsV ;;? LLA L ;; KKOO  ;;?t||q0 KKM 1?rc |jdkDSr)r rs rrunningztransaction.runningPs{{Qrc"||j|<y)zadd a callback to be called when the transaction is pending The transaction will be given as callback's first argument. Category is a unique identifier to allow overwriting an old callback with a newer callback. N)rrcategorycallbacks r addpendingztransaction.addpendingSs+3h'rct|j}|D]8}|jj||}|jxs||_:|xj|j dzc_|jS)zpwrite pending file to temporary version This is used to allow hooks to view a transaction before commits.pending)r)rFrrrr)r categoriescatrs r writependingztransaction.writepending]s{ D112 C0$''++C06C#//63D  D//{/CCrc||jvS)z0check is a callback already exist for a categoryrrrs r hasfinalizeztransaction.hasfinalizejs41111rc"||j|<y)zadd a callback to be called when the transaction is closed The transaction will be given as callback's first argument. Category is a unique identifier to allow overwriting old callbacks with newer callbacks. Nrrs r addfinalizeztransaction.addfinalizeos,4x(rc"||j|<y)zadd or replace a callback to be called after the transaction closed The transaction will be given as callback's first argument. Category is a unique identifier to allow overwriting an old callback with a newer callback. N)rrs r addpostcloseztransaction.addpostclosezs-5)rc:|jj|dS)z1return a postclose callback added before, or NoneN)rrIrs r getpostcloseztransaction.getpostcloses&&**8T::rc"||j|<y)aadd a callback to be called when the transaction is aborted. The transaction will be given as the first argument to the callback. Category is a unique identifier to allow overwriting an old callback with a newer callback. N)rrs raddabortztransaction.addaborts)1H%rc"||j|<y)zadds a callback to be called when validating the transaction. The transaction will be given as the first argument to the callback. callback should raise exception if to abort transactionN)rrs r addvalidatorztransaction.addvalidators,4x(rc|jdk(rt|jD]}|j||d|_|jt|j r=|j }i|_t|}|D] }||||j r=d|_|jt |xjdzc_|jdk7ry|jj|jj|jD]g\}}}}||jvr|r|jd||fz/|j|} |rA|sD| j|sV | j|ii|_t+|_|j/|j0r|j1d|_|j2j5|j6r%|j2j|j6|j2j5|j8r%|j2j|j8|jD]d\}} }}||jvr|r|jd||fz/|j|} |sA| j|sS | j|fg|_ d|_|j;|dd|_t|j<}|D]}|j<||d|_y#tt t"j$f$r3} |s|jd| j'|| fzYd} ~ fd} ~ wwxYw#tt t"j$f$r3} |s|jd| j'|| fzYd} ~ gd} ~ wwxYw) zcommit the transactionrN)rrs.couldn't remove %s: unknown cache location %s scouldn't remove %s: %s -couldn't remove %s: unknown cache location%s T)r rFrrrrrrrNrrrprmrPr$r;rrrLr rrrErs _writeundorqrnisfilerrtrxr) rr callbacksrrr_r,r=r1r<inst_fs rrNztransaction.closes[ ;;! "4#9#9:0&&x06;%)D "   &<  =(( 22 )+&#I. %C"IcN4(& ((&*D "   &=  > q ;;!    !--JAq!Q $ F!QO,,q/Cszz!}JJqM."  ;; KKMDK <<  t22 3 LL   3 3 4 <<  t}} - LL   ...KAr1a $  !f%,,q/CSZZ]JJqM/$!  d#D334 C (D # #C ( .#'] %++6LL3sxx{D6II 8 %++6LL3sxx{D6II s0>MNN(N  NO&3(O!!O&c$|jy)zabort the transaction (generally called on error, or when the transaction is not explicitly committed before going out of scope)N)rrs rabortztransaction.aborts rc>|jj||fyrf)rur")rvfs_idrs r add_journalztransaction.add_journals ""FD>2rc4jytjjjdtdtffd }djz}j j |d}|jdtzjD]\}}}}|s |sd }nt|jvr|rjd ||fz7j|} ||}tj| j|| j|d |jd ||||fz|jjD]1\} } || } | j|  | j!| | 3y#t"$rY@wxYw)z4write transaction data for possible future undo callN)r'fnreturnctjj|\}}|jjsJ|j jj d}tjj||Sr)rrr startswithrtrrvr )rbasernew_namers rrz(transaction._writeundo..undonames^r*JD$??4==1 11||DMM4>>1EH77<<h/ /rrrjrkrr Trr)rvr5rmrpbytesrnr~rrrrr:r rNru tryunlinkrenamerO) rrundo_backup_pathundobackupfiler_r,r=r1ur<srcdests ` rrztransaction._writeundos >> !  LL LL  0 05 0 -t~~=**+;TBWw./--JAq!QDLL(QLL #$a&)ll1oQK chhqk388A;F  !41a|!C D. ++HCC=D MM$   3%,%  s6F  FFcD|j}d|_d|_|jj |j j |j |} |s|jtdt|jD]}|j||d|_ |r|j|n|j|d|_ |j|dd|_y#d|_ |j|dd|_wxYw)Nrstransaction abort! F)rr rlrrNr_can_quick_abortrmrrFr_do_quick_abort_do_full_abortrtrx)rrSquickrs rrztransaction._abort*s""$    !%%g. # Q678d112(##C(.3#'D $$W-##G, DM OOD% ("DO!DM OOD% ("DOs %A7C=="Dc:|ry|jD] }|ds yy)zFalse if any semantic content have been written on disk True if nothing, except temporary files has been writen on disk.FrT)r)rrSr-s rr(ztransaction._can_quick_abortBs) $$At%rc|j|sJ|jDcgc] }|dr | }}|D](\}}}}|j|} |j|*|j r%|j j|j |jr&|j j|jyycc}w#t$rYwxYw)z4(Silently) do a quick cleanup (see _can_quick_abort)rN)r(rrpr$rOrrnrt) rrSr- tmp_filesrold_pathtmp_pathxxxr<s rr)ztransaction._do_quick_abortMs$$W--- $ 3 3@ 311Q4Q 3 @/8 +FHh,,v&C  8$09    LL   3 3 4 == LL   . A %  s CC C  CCc  t|j|j|j|j||j d|j |jtdy#t$rT}|jtd|jtdtj|zYd}~yd}~wwxYw)z?(Noisily) rollback all the change introduced by the transactionFrUsrollback completed s(rollback failed - please run hg recover s(failure reason: %s) N) rartrmrnrprryr BaseExceptionrr!)rrSr@s rr*ztransaction._do_full_abort\s      ## $ 5 5  LL23 4  LLGH I LL+,z/F/Fs/KK   sA)A,, C 5A CC )NNNNNN )TrF)r)rrF)r5)*__name__ __module__ __qualname__rrrpropertyrrrrrJrrrrrrrrrrrrrrrrrrrrr r rNrrrrr(r)r*rrrrcrcs4 `!D ..   ! !  %  %" ,B ,B\" > > %, %,N , , %(}*X ) )     *   3       2 2 4 4 5 5 ; ; 1 1 4 4 O' O'b    3 3-^#0  /rrcs8journal was created by a different version of Mercurial c |j}g}|rh|ddd}|dtzk7r|t|S|ddD]>}|s|dd}|jd\}}}} |j |||t | f@|S)aZparse an (already open) backup file an return contained backup entries entries are in the form: (location, file, backupfile, xxx) :location: the vfs identifier (vfsmap's key) :file: original file path (in the vfs) :backupfile: path of the backup (in the vfs) :cache: a boolean currently always set to False rNrs%drr)rrBAD_VERSION_MSGrr"bool) r%r^linesrTverliner_r,r=r1s rrrvs LLNEM Ahsm 57? # ? #  ab 9D!%E!2JAq!Q!((!Q47);< " rc Rg}g}|j|5}|j} ddd D]3} | jd\} } |j| t | f5d|z} |j| r&|j| 5}t||}dddfd}|Dcgc] }||s |}}t|||||||y#1swYxYw#t $r*|t dtj| zYwxYw#1swY}xYwcc}w)aRolls back the transaction contained in the given file Reads the entries in the specified file, and the corresponding '*.backupfiles' file, to recover from an incomplete transaction. * `file`: a file containing a list of entries, specifying where to truncate each file. The file should contain a list of fileoffset pairs, delimited by newlines. The corresponding '*.backupfiles' file should contain a list of filebackupfile pairs, delimited by . `checkambigfiles` is a set of (path, vfs-location) tuples, which determine whether file stat ambiguity should be avoided at restoring corresponded files. Nrs couldn't read journal entry %r! rc.j|d Sr)match)xskip_journal_patterns rrizrollback..s177!==rr3) r~rrr"r ValueErrorrrbytestrrPrra)rRr&rr%rUrDrSrTr^r=r_r,r\ backupjournalkeeprCs ` rrollbackrIs!.GM T b    775>DAq NNAs1v; '&,M }}]# [[ '2-fb9M('=$1=MqT!WM =  ''     67(:J:J1:MM  ( '>s4C0C" D/ D$=D$C"0DDD!)sundo)TN)NN)rri18nrrrrutilsrrrrrrrr#r5ra transactionalrcr;rrIrrrrNs    ')   "X v rS $$$S l@ > 5r