b;@ddlZddlmZddlmZmZmZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZddlmZddl m!Z!m"Z"d d l#m$Z$d d l%m&Z&m'Z'ej"Z(e(jSed dZ*e(jSe ddZ+e(jSeddZ,dZ-dZ.dZ/d9dZ0ddejbfdZ2dZ3dZ4e(jSeddZ5e(jSejlddZ7e(jSejlddZ8d Z9e(jSe d!d"Z:e(jSed#d$Z;e(jSejxd%d&Z=d'Z>d(Z?d)Z@d*ZAd+ZBe(jSed,d-ZCe(jSe d.d/ZDd0ZEd:d1ZFd:d2ZGd3ZHe(jSed4d5ZIe(jSed6e(jSed7d8ZJy);N)_)binhexshort)getattrsetattr) bundle2 changegroupcmdutilcontexterrorexchange exthelper localreporevlogscmutilutilvfswireprotov1server)actionsengine) repository) storageutil stringutil)lfutil) blobstorepointersmakefilestoragec ^d|vr|jtj|d||d|S)Nlfs) requirementsfeatures)addrREPO_FEATURE_LFS)origr"r#kwargss 5usr/lib/python3.12/site-packages/hgext/lfs/wrapper.pylocalrepomakefilestorager*8s1  Z001  G\H G GGsallsupportedversionsc8||}|jd|S)Ns03)r%)r'uiversionss r)allsupportedversionsr/@sBxH LL Or+s _capabilitiesc|||}tj|jdr0d|jvr|j d|j d|S)z5Wrap server command to announce lfs server capabilitylfslocalblobstorer!s lfs-serve)r safehasattrsvfsr"append)r'repoprotocapss r) _capabilitiesr8GsR e D  #78 T&& & KK % F Kr+cyNFr$)selftexts r)bypasscheckhashr=Ws r+ctj|}|j}|jj}|j |s8|j |_|jjj|g||j|d}i}|jD]*}|jds|tdd}||||<,|s|jdrtj||}|dfS)a%Read filelog content from local blobstore transform for flagprocessor. Default tranform for flagprocessor, returning contents from blobstore. Returns a 2-typle (text, validatehash) where validatehash is True as the contents of the blobstore should be checked using checkhash. F)verifysx-hg-Ns T)r deserializeoidopenerlfslocalblobstorehasfilenamelfsremoteblobstore readbatchreadkeys startswithlenrpackmeta)r;r<prAstorehgmetaknames r) readfromstorerR[s D!A %%'C KK ) )E 99S>]]  &&00!e< ::c%: (DF VVX << !S]_%DQ4F4L)##FD1 $<r+ctj|\}}|r |dkDr||d}ttj|j }|j jj||d|z}tj|dt|z}tj|sd|d<| |jD] \}}||d|z<|j} | dfS) Nrs sha256:%ss%d)rAsize0 x-is-binarysx-hg-%sF)r parsemetarhashlibsha256digestrBrCwriter gitlfspointerrKrbinaryitems serialize) r;r<rOoffsetrAlongoidmetadatarPvrawtexts r) writetostorerexs **40NFF &1*FG} gnnT"))+ ,CKK!!''T2S G$$us4y7HIH   T "#' LLNDAq'(HZ!^ $#  "G U r+c||y|j|}n|j|}||jk(ry|j|}t |t j zSr:)revnodenullidflagsboolrREVIDX_EXTSTORED)rlogrhrgrjs r)_islfsrns[ { <hhtnyy~ t{{ JJsOE /// 00r+c  <|jjjjd} | rZt |} t j |\} }|r| |z} | |jj| r| tjz} |||||||f||| d| S)Nslfstrack) cachedeltarhrj) _revlogrBoptionsgetrKrrWrErrl)r'r;r< transactionlinkp1p2rprhrjkwdslfstracktextlenmetar`s r)filelogaddrevisionr|s||""**..{;Hd)",,T2 f  v G DLL))7 3 V,, ,E           r+ct|j|rO|jj|}|sytj|}d|vrd|vr|dt |dfSy|||S)NFs x-hg-copys x-hg-copyrev)rnrqrawdatarr@r)r'r;rhrdrbs r)filelogrenamedrss dllD!,,&&t,&&w/ 8 #8(CL)3x/H+II I d r+ct|j|r>|jj|}tj|}t |dS|||S)N)rgssize)rnrqr~rr@int)r'r;rgrdrbs r) filelogsizersP dll$,,&&s+&&w/8G$%% c?r+s_verify_revisionc^t||r|j|}tj|}|jj j |jr|tjz}n'|tjzr|dj||||||y)N)rhs safe_renamed) rnr~rr@rBrCrDrArrlr%)r'rl skipflagsstaterhrdrbs r)_verify_revisionrs bt**T"&&w/ 99 & & * *8<<> : &111 1I 00 0 / " & &t ,Yt$r+scmpc(|jrzt|ddrgtj|j }tj|j }|j |j k7S|||S)z+returns True if text is different than fctxislfscyr:r$r$r+r)zfilectxcmp..sur+)rrrr@r~rA)r'r;fctxrvrws r) filectxcmprsn zz|=g}=?   0   0vvx2668## d r+sisbinaryc|jrGtj|j}t t |j ddS||S)NrVr)rrr@r~rkrrs)r'r;rbs r)filectxisbinaryrsF zz|&&t||~6C ^Q7899 :r+cft|jj|jSN)rnfilelogrqfilenode)r;s r) filectxislfsrs! $,,.(($--/ ::r+s_updatecatformattercb|||||||j||jy)N)r~)datar~)r'fmctxmatcherpathdecodes r)_updatecatformatterrs-S'4(GGCI%%'G(r+swrapconvertsinkcr||}|jdk(rGfdd|j|_|S)Nshgc"eZdZfdZxZS)convertsink..lfssinkc : t |} | ||||||||} d|jjvrg|j|  t fd|DrD|jjj dt j|j| S)Nr!c3VK|] \}}|vxr|j"ywr)r).0fnrs r) z9convertsink..lfssink.putcommit..;s,I541a186A 65s&))super putcommitr5r"anyr%rwritereporequirements)r;filescopiesparentscommitsourcerevmapfullcleanp2pcrhr __class__lfssinks @r)rz&convertsink..lfssink.putcommits7D3 !7!77))D/CI5II ..226:55dii@ r+)__name__ __module__ __qualname__r __classcell__)rrs@r)rrs   r+r)repotyper)r'sinkrs @r) convertsinkrs7 :D }}! dnn! F! Kr+s__init__c ||||jjD]&\}}|jds||j|<(dD]0}tj||st ||t ||2y)Nr!)r1lfsremoteblobstore)rrr^rJrr2rr)r'r;othervfsrPrcrQs r)vfsinitrHsqx  &&(1 << DLLO) >   Hd + D$$ 7 8>r+ctj|jdsyg}t}|jj}|D]\}}||}|j |D]{}t ||} | s| j|vs%|j| jrE|| _ |j| |j| j}|r&tj|j||yy)zSEnsure that required LFS blobs are present, fetching them as a group if needed.r1N)rr2r3setrCwalkpointerfromctxrArDrEr4r%rremoterG) r5 revmatchespointersoids localstorergmatchrrrMs r)_prefetchfilesrVs   DII'; <H 5D,,J U3i%AsA&AQUUWD(1H "! !! ((:>r+ctj|jdsyt|jjt j S)NrTrr2r3 isinstancerFr _nullremoter5s r)_canskipuploadros9   DII'< = dii22I4I4I JJr+ctj|jdsyt|jjt j  S)NrFrrs r) candownloadrxs<   DII'< =$))66 8M8MN NNr+cLt|ryt||}t||y)zuupload lfs blobs introduced by revs Note: also used by other extensions e. g. infinitepush. avoid renaming. N)rextractpointers uploadblobs)r5revsrs r)uploadblobsfromrevsrs% dtT*Hhr+cVt|j|jjS)zPrepush hook. Read through the revisions to push, looking for filelog entries that can be deserialized into metadata so that we can block the push on their upload to the remote blobstore. )rr5outgoingmissing)pushops r)prepushrs v{{FOO,C,C DDr+spushcd|jvr|js?|jds.td}t j |dztd|j j} tj||j|j _|||g|i|||j _S|||g|i|S#||j _wxYw)zbail on push if the extension isn't enabled on remote when needed, and update the remote store based on the destination path.r!s:required features are not supported in the destination: %ss&enable the lfs extension on the server)hint) r"localcapablerr Abortr3rFrrurl)r'r5rargsr(m remotestores r)pushrs"""||~fnnV&<OPA++F #L!M ii22  7+4+;+;D&**,+ODII (f6t6v6+6DII (D&242622,7DII (s 5?CC&swritenewbundlecPt||j|||||||g|i|S)z;upload LFS blobs added by outgoing revisions on 'hg bundle')rr) r'r-r5rrE bundletyperrr(s r)writenewbundlers> h../  D&(J ;? CI r+c|jjdi}|jj}|tdtdt |5}|D]H}||}t |j D]}|||j<|jJt|j dcdddS#1swYyxYw)z1return a list of lfs pointers added by given revss&lfs: computing set of blobs to upload s lfs searchs changesetsc"|jSr)rA)rMs r)rz!extractpointers..s quuwr+)keyN) r-debug makeprogressrrKpointersfromctxvaluesrA incrementsorted)r5rrrprogressrrrMs r)rrsGGMM<=H77''L  -!M*CI  Aq'C$S)002$%!3     hoo'->?   s A)C  Cc |}||vrJ|sy||jvr|j}n$||jvr|j}ny||}t|jj|j sy t j|j}||k(r|SiS#t j$r@}tjtd|t|j|fzd}~wwxYw)a?return a pointer for the named file from the given changectx, or None if the file isn't LFS. Optionally, the pointer for a file deleted from the context can be returned. Since no such pointer is actually stored, and to distinguish from a non LFS file, this pointer is represented by an empty dict. Ns#lfs: corrupted pointer (%s@%s): %s )rvrwrnrrqrrr@r~InvalidPointerr rrrrh)rrremoved_ctxrrMexs r)rrs D| =668D #&&(]668D 7D $,,.(($--/ :      / $;H  ! ! kk 5 6% $b) *   s )B64B66D  ;DD ci}|jj}|jD]!}||s t|||}||||<#|S)zreturn a dict {path: pointer} for given single changectx. If ``removed`` == True and the LFS file was removed from ``ctx``, the value stored for the path is an empty dict. )r)r5 narrowmatchrr)rrresultrrrMs r)rrs\ F   AYY[t  37 3 =F1I  Mr+c|sy|jj}|j||jjy)z*upload given pointers from local blobstoreN)r3rF writebatchrC)r5r remoteblobs r)rrs1 --J(DII$?$?@r+sfinishdatamigrationc|||||tj|jdrtj|jdr|jjj}|jjj}|j D]^\}}} | D]S} |j td| ztj|j| |j| U`yyy)Nr1scopying lfs blob %s ) rr2r3rCrrr[rrrujoin) r'r-srcrepodstrepor" srclfsvfs dstlfsvfsdirpathdirsrrAs r)upgradefinishdatamigrationr sWg|,  *   7<<)= >LL2266 LL2266 $-NN$4 GT534s:; INN3/1DE%5 ?r+spreservedrequirementsssupporteddestrequirementscT||}d|jvr|jd|S)Nr!)r"r%)r'r5reqss r)upgraderequirementsr s, :D """  Kr+)NN)F)KrXmercurial.i18nrmercurial.noderrrmercurial.pycompatrr mercurialr r r r r rrrrrrrvfsmodrmercurial.upgrade_utilsrupgrade_actionsrupgrade_enginemercurial.interfacesrmercurial.utilsrr largefilesrrreh wrapfunctionr*r/r8r=rRrernREVIDX_DEFAULT_FLAGSr|rrr basefilectxrrrrr readonlyvfsrrrrrrrrrrrrrr r$r+r)rs**   ,   Y./H0H567 "$45 6 :> 1.  % %%R ,-%.%$$$f-.$$k23;01)2) ,-).)\##[1 92 9?2KO E7#3$3:+,-@" B&A!78 F9 F ":;">?@<r+