q&fB,ddlZddlmZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZmZddlmZddlmZmZdd lm Z d d l!m"Z"m#Z#ejZ$e$jKed d Z&e$jKe ddZ'e$jKeddZ(dZ)dZ*dZ+d8dZ,ddejZfdZ.dZ/dZ0e$jKeddZ1e$jKe jdddZ3e$jKe jdddZ4dZ5e$jKe d d!Z6e$jKed"d#Z7e$jKejpd$d%Z9d&Z:d'Z;d(Ze$jKe d+d,Z?e$jKed-d.Z@d/ZAd9d0ZBd9d1ZCd2ZDe$jKed3d4ZEe$jKed5e$jKed6d7ZFy):N)_)binhexshort) bundle2 changegroupcmdutilcontexterrorexchange exthelper localreporevlogscmutilvfswireprotov1server)actionsengine) repository) storageutil stringutil)lfutil) blobstorepointermakefilestoragec ^d|vr|jtj|d||d|S)Nlfs) requirementsfeatures)addrREPO_FEATURE_LFS)origr r!kwargss 3/usr/lib/python3/dist-packages/hgext/lfs/wrapper.pylocalrepomakefilestorager(3s1  Z001  G\H G GGallsupportedversionsc8||}|jd|S)Ns03)r#)r%uiversionss r'r*r*;sBxH LL Or) _capabilitiesc|||}t|jdr0d|jvr|jd|jd|S)z5Wrap server command to announce lfs server capabilitylfslocalblobstorers lfs-serve)hasattrsvfsr append)r%repoprotocapss r'r.r.BsL e Dtyy-. T&& & KK % F Kr)cyNFr")selftexts r'bypasscheckhashr;Rs 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 deserializeoidopenerr0hasfilenamelfsremoteblobstore readbatchreadkeys startswithlenrpackmeta)r9r:pr?storehgmetaknames r' readfromstorerOVs 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)r?size0 x-is-binarysx-hg-%sF)r parsemetarhashlibsha256digestr@r0writer gitlfspointerrHrbinaryitems serialize) r9r:rLoffsetr?longoidmetadatarMvrawtexts r' writetostorerbss **40NFF &1*FG} gnnT"))+ ,CKK!!''T2S G$$us4y7HIH   T "#' LLN )DAq'(HZ!^ $ )  "G U r)c||y|j|}n|j|}||jk(ry|j|}t |t j zSr8)revnodenullidflagsboolrREVIDX_EXTSTORED)rlogrerdrgs r'_islfsrks[ { <hhtnyy~ t{{ JJsOE /// 00r)c  <|jjjjd} | rZt |} t j |\} }|r| |z} | |jj| r| tjz} |||||||f||| d| S)Nslfstrack) cachedeltarerg) _revlogr@optionsgetrHrrTrBrri)r%r9r: transactionlinkp1p2rmrergkwdslfstracktextlenmetar]s r'filelogaddrevisionrys||""**..{;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)rkrnrawdatarr>r)r%r9rerar_s r'filelogrenamedr|ss dllD!,,&&t,&&w/ 8 #8(CL)3x/H+II I d r)ct|j|r>|jj|}tj|}t |dS|||S)N)rdssize)rkrnr{rr>int)r%r9rdrar_s r' filelogsizersP dll$,,&&s+&&w/8G$%% c?r)_verify_revisionc^t||r|j|}tj|}|jj j |jr|tjz}n'|tjzr|dj||||||y)N)res safe_renamed) rkr{rr>r@r0rAr?rrir#)r%rl skipflagsstatererar_s r'rrs bt**T"&&w/ 99 & & * *8<<> : &111 1I 00 0 / " & &t ,Yt$r)cmpc(|jrzt|ddrgtj|j }tj|j }|j |j k7S|||S)z+returns True if text is different than fctxislfscyr8r"r"r)r'zfilectxcmp..sr))rgetattrrr>r{r?)r%r9fctxrsrts r' filectxcmprsn zz|=g}=?   0   0vvx2668## d r)isbinaryc|jrGtj|j}t t |j ddS||S)NrSr)rrr>r{rhr~rp)r%r9r_s r'filectxisbinaryrsF zz|&&t||~6C ^Q7899 :r)cft|jj|jSN)rkfilelogrnfilenode)r9s r' filectxislfsr s! $,,.(($--/ ::r)_updatecatformattercb|||||||j||jy)N)r{)datar{)r%fmctxmatcherpathdecodes r'rrs-S'4(GGCI%%'G(r)wrapconvertsinkcr||}|jdk(rGfdd|j|_|S)Nshgc"eZdZfdZxZS)convertsink..lfssinkc : t |} | ||||||||} d|jjvrg|j|  t fd|DrD|jjj dt j|j| S)Nrc3VK|] \}}|vxr|j"ywr)r).0fnrs r' z9convertsink..lfssink.putcommit..6s+I41a186A 6Is&))super putcommitr4r anyr#rwritereporequirements)r9filescopiesparentscommitsourcerevmapfullcleanp2pcrer __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)__init__c ||||jjD]&\}}|jds||j|<(dD]&}t||st ||t ||(y)Nr)r0rC)ror[rGr1setattrr)r%r9othervfsrMr`rNs r'vfsinitrCsux  &&( 1 << DLLO <9 8T " D$$ 7 89r)ct|jdsyg}t}|jj}|D]\}}||}|j |D]{}t ||} | s| j |vs%|j| j rE|| _|j| |j| j }|r&tj|j||yy)zSEnsure that required LFS blobs are present, fetching them as a group if needed.r0N)r1r2setr0walkpointerfromctxr?rArBr3r#rremoterD) r4 revmatchespointersoids localstorerdmatchrrrJs r'_prefetchfilesrQs 4991 2H 5D,,J " U3i% "AsA&AQUUWD(1H "!  "" ((:>r)ct|jdsyt|jjtj S)NrCTr1r2 isinstancerCr _nullremoter4s r'_canskipuploadrjs3 4992 3 dii22I4I4I JJr)ct|jdsyt|jjtj  S)NrCFrrs r' candownloadrss6 4992 3$))66 8M8MN NNr)cLt|ryt||}t||y)z#upload lfs blobs introduced by revsN)rextractpointers uploadblobs)r4revsrs r'uploadblobsfromrevsr|s#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. )rr4outgoingmissing)pushops r'prepushrs v{{FOO,C,C DDr)pushcd|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.rs:required features are not supported in the destination: %ss&enable the lfs extension on the server)hint) r localcapablerr Abortr2rCrrurl)r%r4rargsr&m remotestores r'rrs"""||~fnnV&<OPA++F #L!M ii22  7+4+;+;D&**,+ODII (f6t6v6+6DII (D&242622,7DII (s 5?CC&writenewbundlecPt||j|||||||g|i|S)z;upload LFS blobs added by outgoing revisions on 'hg bundle')rr) r%r,r4rrB bundletyperrr&s r'rrs> 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)r?)rJs r'rz!extractpointers..s quuwr))keyN) r,debug makeprogressrrHpointersfromctxvaluesr? incrementsorted)r4rrrprogressrrrJs r'rrsGGMM<=H77''L  -!M*CI @  !Aq'C$S)002 &$%! &     ! 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 )rsrtrkrrnrrr>r{InvalidPointerr rrrre)rrremoved_ctxrrJexs 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)r4 narrowmatchrr)rrresultrrrJs r'rrsa F   AYY[t  37 3 =F1I  Mr)c|sy|jj}|j||jjy)z*upload given pointers from local blobstoreN)r2rC writebatchr0)r4r remoteblobs r'rrs1 --J(DII$?$?@r)finishdatamigrationc|||||t|jdrt|jdr|jjj}|jjj}|j D]^\}}} | D]S} |j t d| ztj|j| |j| U`yyy)Nr0scopying lfs blob %s ) r1r2r0rrrXrrrrjoin) r%r,srcrepodstrepor srclfsvfs dstlfsvfsdirpathdirsrr?s r'upgradefinishdatamigrationr sWg|,w||01g )7LL2266 LL2266 $-NN$4 F GT5 F34s:; INN3/1DE F F 71r)preservedrequirementssupporteddestrequirementscT||}d|jvr|jd|S)Nr)r r#)r%r4reqss r'upgraderequirementsrs, :D """  Kr))NN)F)GrUmercurial.i18nrmercurial.noderrr mercurialrrr r r r r rrrrvfsmodrmercurial.upgrade_utilsrupgrade_actionsrupgrade_enginemercurial.interfacesrmercurial.utilsrr largefilesrrreh wrapfunctionr(r*r.r;rOrbrkREVIDX_DEFAULT_FLAGSryr|rr basefilectxrrrrr readonlyvfsrrrrrrrrrrrrr rr"r)r'r!s**     ,   Y-.H/H456 "O4 5 :> 1.  % %%R +,%-%$$$e,-$$j12;/0)1) +,)-)\##Z0 91 9?2KO E6"3#3:*+,@" B&A!67 F8 F "9:"=>?;r)