^cBddlZddlmZddlmZmZmZddlmZm 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'ejZ(e()ed dZ*e()e ddZ+e()eddZ,dZ-dZ.dZ/d:dZ0ddej1fdZ2dZ3dZ4e()eddZ5e()ej6ddZ7e()ej6ddZ8d Z9e()e d!d"Z:e()ed#d$Z;e()ej<d%d&Z=d'Z>d(Z?d)Z@d*ZAd+ZBe()ed,d-ZCe()e d.d/ZDd0ZEd;d2ZFd;d3ZGd4ZHe()ed5d6ZIe()ed7e()ed8d9ZJdS)<N)_)binhexshort)getattrsetattr) bundle2 changegroupcmdutilcontexterrorexchange exthelper localreporevlogscmutilutilvfswireprotov1server)actionsengine) repository) storageutil stringutil)lfutil) blobstorepointersmakefilestoragec ^d|vr|tj|d||d|S)Nlfs) requirementsfeatures)addrREPO_FEATURE_LFS)origr"r#kwargss 3/usr/lib/python3/dist-packages/hgext/lfs/wrapper.pylocalrepomakefilestorager*8s?  Z0111 4 G\H G G G GGsallsupportedversionscF||}|d|S)Ns03)r%)r'uiversionss r)allsupportedversionsr/@s&tBxxH LL Or+s _capabilitiesc|||}tj|jdr3d|jvr|d|d|S)z5Wrap server command to announce lfs server capabilitylfslocalblobstorer!s lfs-serve)r safehasattrsvfsr"append)r'repoprotocapss r) _capabilitiesr8Gsf 4e  D  #788 T& & & KK % % % F Kr+cdSNFr$)selftexts r)bypasscheckhashr=Ws 5r+ctj|}|}|jj}||s-|j|_|jj|g|| |d}i}| D]9}| dr"|tdd}||||<:|s| drtj||}|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 K )E 99S>>=]  &00!e<<< ::c%: ( (DF VVXX   << ! ! S]]__%DQ4F4L 2))2#FD11 $<r+ctj|\}}|r|dkr ||d}ttj|}|jj||d|z}tj |dt|z}tj |sd|d<|"|D] \}}||d|z<|} | dfS) Nrs sha256:%ss%d)rAsize0 x-is-binarysx-hg-%sF)r parsemetarhashlibsha256digestrBrCwriter gitlfspointerrKrbinaryitems serialize) r;r<rOoffsetrAlongoidmetadatarPvrawtexts r) writetostorerexs *400NFF &1**FGG} gnT""))++ , ,CK!''T222S G$us4yy7HIIIH  T " "(#' LLNN ) )DAq'(HZ!^ $ $  ""G U r+c||dS||}n||}||jkrdS||}t |t jzSr:)revnodenullidflagsboolrREVIDX_EXTSTORED)rlogrhrgrjs r)_islfsrnsh { <5hhtnnyy~~ t{u JJsOOE // 0 00r+c  |jjjd} | rRt |} t j|\} }|r| |z} | |jj| r| tj z} |||||||f||| d| S)Nslfstrack) cachedeltarhrj) _revlogrBoptionsgetrKrrWrErrl)r'r;r< transactionlinkp1p2rprhrjkwdslfstracktextlenmetar`s r)filelogaddrevisionr|s|"*..{;;H-d))",T22 f   v G 8DL)7 3 3 - V, ,E 4            r+ct|j|rY|j|}|sdStj|}d|vr!d|vr|dt |dfSdS|||S)NFs x-hg-copys x-hg-copyrev)rnrqrawdatarr@r)r'r;rhrdrbs r)filelogrenamedrs dlD!!,&&t,, 5&w// 8 # #8(C(CL)3x/H+I+II I5 4d  r+ct|j|rC|j|}tj|}t |dS|||S)N)rgssize)rnrqr~rr@int)r'r;rgrdrbs r) filelogsizersa dl$$$&,&&s++&w//8G$%%% 4c??r+s_verify_revisioncnt||r||}tj|}|jj|r|tj z}n*|tj zr|d ||||||dS)N)rhs safe_renamed) rnr~rr@rBrCrDrArrlr%)r'rl skipflagsstaterhrdrbs r)_verify_revisionrs bt -**T""&w// 9 & * *8<<>> : : - &11 1II 0 0 - / " & &t , , ,DYt$$$$$r+scmpcb|rt|ddrvtj|}tj|}||kS|||S)z+returns True if text is different than fctxislfscdSr:r$r$r+r)zfilectxcmp..sur+)rrrr@r~rA)r'r;fctxrvrws r) filectxcmprs zz||$=g}}==??$   0 0   0 0vvxx26688## 4d  r+sisbinaryc|rVtj|}t t |ddS||S)NrVr)rrr@r~rkrrs)r'r;rbs r)filectxisbinaryrs\ zz||:&t||~~66C ^Q7788999 4::r+ctt|j|SN)rnfilelogrqfilenode)r;s r) filectxislfsrs% $,,..($--// : ::r+s_updatecatformatterc|||||||||dS)N)r~)datar~)r'fmctxmatcherpathdecodes r)_updatecatformatterrsDDS'4(((GGCI%%''G(((((r+swrapconvertsinkcl||}|jdkrGfdd|j|_|S)Nshgc"eZdZfdZxZS)convertsink..lfssinkc 4 t |j} | ||||||||} d|jjvr`|j|  t fd|Dr8|jjdt j|j| S)Nr!c3ZK|]%\}}|vo|V&dSr)r).0fnrs r) z9convertsink..lfssink.putcommit..;s<II41a186A IIIIIIr+)super putcommitr5r"anyr%rwritereporequirements)r;filescopiesparentscommitsourcerevmapfullcleanp2pcrhr __class__lfssinks @r)rz&convertsink..lfssink.putcommits7D))3r  !777)D/CIIII5IIIIIA .226:::5di@@@ r+)__name__ __module__ __qualname__r __classcell__)rrs@r)rrs>          r+r)repotyper)r'sinkrs @r) convertsinkrsb 4::D }! ! ! ! ! ! ! dn! ! ! F! Kr+s__init__c  ||||jD]$\}}|dr ||j|<%dD]6}tj||rt ||t ||7dS)Nr!)r1lfsremoteblobstore)rrr^rJrr2rr)r'r;othervfsrPrcrQs r)vfsinitrHsDx &&((  1 <<   DLO>99  Hd + + 9 D$$ 7 7 8 8 899r+cBtj|jdsdSg}t}|jj}|D]\}}||}||D]}t ||} | r| |vrj|| sC|| _ | | | | |r*tj |||dSdS)zSEnsure that required LFS blobs are present, fetching them as a group if needed.r1N)rr2r3setrCwalkpointerfromctxrArDrEr4r%rremoterG) r5 revmatchespointersoids localstorergmatchrrrMs r)_prefetchfilesrVs&  DI'; < <H 55D,J "" U3i% " "AsA&&A "QUUWWD((1H1H( """!!!  "? ((:>>>>>??r+ctj|jdsdSt|jjt jS)NrTrr2r3 isinstancerFr _nullremoter5s r)_canskipuploadros9  DI'< = =t di2I4I J JJr+ctj|jdsdSt|jjt j S)NrFrrs r) candownloadrxs<  DI'< = =u$)6 8MNN NNr+cht|rdSt||}t||dS)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+c@t|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{FO,C D DDr+spushcd|jvr|sJ|ds5td}t j|dztd|jj} tj || |j_|||g|Ri|||j_S#||j_wxYw|||g|Ri|S)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&<&< OPPA+F #L!M!M i2  7+4+;D&**,,+O+ODI (4f6t666v66+6DI ( (;DI ( 6 6 6 6tD&242226222s 5=B??C swritenewbundlecNt||j|||||||g|Ri|S)z;upload LFS blobs added by outgoing revisions on 'hg bundle')rr) r'r-r5rrE bundletyperrr(s r)writenewbundlersR h./// 4 D&(J ;?   CI  r+c|jdi}|jj}|tdtdt |5}|D]Y}||}t |D]}|||<|Zt|dcdddS#1swxYwYdS)z1return a list of lfs pointers added by given revss&lfs: computing set of blobs to upload s lfs searchs changesetsc*|Sr)rA)rMs r)rz!extractpointers..squuwwr+)keyN) r-debug makeprogressrrKpointersfromctxvaluesrA incrementsorted)r5rrrprogressrrrMs r)rrsMGMM<===H7'L  -!M**CII  @  ! !Aq'C$S))0022 & &$%!!    hoo''->->???@@@@@@@@@@@@@@@@@@sBC**C.1C.Fc h|}||vr\|sdS||vr|}n-||vr|}ndS||}t|j|sdS t j|}||kr|SiS#t j $rK}tj td|t||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|| 4 ==6688DD #&&((]]6688DD4 7D $,,..($--// : :t     / / $;;H  !   k 5 6 6% $$b) *    s'-CCD1&AD,,D1ci}|}|D]'}||st|||}||||<(|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)rrsv F    AYY[[qtt   37 3 3 3 =F1I Mr+cf|sdS|jj}|||jjdS)z*upload given pointers from local blobstoreN)r3rF writebatchrC)r5r remoteblobs r)rrs: -J(DI$?@@@@@r+sfinishdatamigrationc|||||tj|jdrtj|jdr|jjj}|jjj}|D]o\}}} | D]b} |td| ztj | | | | cldSdSdS)Nr1scopying lfs blob %s ) rr2r3rCrrr[rrrujoin) r'r-srcrepodstrepor" srclfsvfs dstlfsvfsdirpathdirsrrAs r)upgradefinishdatamigrationr sDWg|,,,  * F  7<)= > > FL26 L26 $-NN$4$4 F F GT5 F F344s:;;; INN3//1D1DEEEE F F F F F  F Fr+spreservedrequirementsssupporteddestrequirementscX||}d|jvr|d|S)Nr!)r"r%)r'r5reqss r)upgraderequirementsr s6 4::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)rsC**********   ,+++++   Y.//HH0/H56676 "$455  65 :> 1 1 1 1.  %%%%%R   ,--%%.-%$$f--.-$k2232;;;011))21) ,--)).-)\#[11 9 921 9???2KKKOOO   EEE7##33$#3:+,,-,@@@"    B&AAA!788 F F98 F ":;;">??@?<;r+