q&f/9ddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZddlmZmZej8ZdZd ad Z d Z!d Z"d Z#dZ$dZ%dZ&y)N)_)binhex)open) changegroup changelogcontexterror extensionsmatchscmutilstore streamcloneutilwireprotoserverwireprototypeswireprotov1server) constants shallowutilcpt|fd}tjtjd|y)zGSets up a normal Mercurial repo so it can serve files to shallow repos.c|jxsg}tj|vrd} d} |jxsgD]`} | jdr| t ddj d} 2| jdsD| t ddj d} bt j} | s| r#t j jdd| | } |D cgc] } | | r | }} ||||||g|i|Scc} w)Nsincludepattern=sexcludepattern=) _bundlecapsrBUNDLE2_CAPABLITY startswithlensplitr alwaysroot)origself changedfiles linknodes commonrevssourceargskwargscapsincludepatternexcludepatterncapmfrepos I/usr/lib/python3/dist-packages/hgext/remotefilelog/remotefilelogserver.py generatefilesz"setupserver..generatefiles,s%2  & &$ .!N!N''-2 Q>>"45%(-?)@)B%C%I%I%%PN^^$67%(-?)@)B%C%I%I%%PN  Q  AKKIIsD..(4@!1Q4A@L@ , :v @D HN  As  C>(C>r2N) onetimesetupr wrapfunctionrcgpacker)uir0r2s ` r1 setupserverr7's- 2o}rFcLtrydatjdddttjdddttjd d dt Gd d }|fd }|dftj d< dfd }tjtd|fd}tjtd|d}tjtjd|d}tjtd|y)z9Configures the wireprotocol for both clients and servers.NTx_rfl_getflogheadsspathspull) permissionx_rfl_getfilesr x_rfl_getfiles file nodeceZdZdZdZdZy)!onetimesetup..streamstateNF)__name__ __module__ __qualname__r shallowremotenoflatmfrr1 streamstater>_s rrEc d}d}|jd}|r|jd}|jd}|r|jd} j} j} j} d _tj _|jddk( _|s|r(tj|j dd|| _tj|| t j t j fd} tj| | _| _| _S#| _| _| _wxYw) NsincludepatternrsexcludepatternTsnoflatmanifestsTruerc3HKjD]}|ywN)gen)valuefirstsecond streamress r1rIz5onetimesetup..stream_out_shallow..gens)  &]] EK s") getrrBr rCr r!rstreamnextrIrrM)r0protootherr+r,raw oldshallowoldmatch oldnoflatmfrIrKrLrMstates @@@r1stream_out_shallowz(onetimesetup..stream_out_shallowfs7ii)*  YYu-Nii)*  YYu-N(( ;;nn  )"&E ,,.EK"YY'89WDEN#kkIIsD.. *00u=I'E)--(F "++CE2",E "EK(EN#-E "EK(ENs 6CEE(*sstream_out_shallowc3zKjratj|rRt|jj dz}|jj j}tj j|jj dg}|r|j}||dD]\} } } |dz| z} | tjk(rs| jdsb| jdsQtj| |d} t j | }t j"|d| j$ | tj&k(s|j)| |rt+j,|r=|jj/D] }|j0s|j2s|"j4}|r]|j6sL|jj/D]/}|j0sj5|j8r,|1|jj;D]}j<r |ddd d k(r| ytj|rt?j@tCd ||||| D]}|yw)NrsdataT)stat/s.is.dF) entry_path is_volatile file_size s 00manifest.s0Cannot clone from a shallow repo to a full repo.)phase obsolescence)"rBr isenabledrrpathrawvfsreaddirosjoinpopr[S_IFREGendswithrpconvert decodedirSimpleStoreEntryst_sizeS_IFDIRappendr istreemanifest data_entries is_revlogis_manifestlogr r target_id top_entriesrCr Abortr)r"r0matcherrarbstriplenrfvisitpr/kindstfpndentryr xrWs r1_walkstreamfilesz&onetimesetup.._walkstreamfilessR   $$T*tzz/!3**++33djjoow?@ A'.qt'<- 4X\4<</#%;;u#5bkk %?%)MM"XY-$@$)OOA$6&+&<&>ad3Bi>&A   " "4 (++EF gU  s!EJ;#J;9A J;A*J;0B J;rc |||}tj|sjddrQt|tr|j t j|j d|j d|S)N remotefilelogserverr9r<)rrc configbool isinstance _sshv1serverrqrNETWORK_CAP_LEGACY_SSH_GETFILES)r"r0rQr*r6s r1 _capabilitiesz#onetimesetup.._capabilitiessjD%    &"-- i+ %. IEEF KK- . KK( ) rrc|j}t|drJ|jr>|jj |jj |j S||g|i|S)N forcelinkrev)_repohasattrr_fileloglinkrevrev _filenode)r"r#r(r)r0s r1_adjustlinkrevz$onetimesetup.._adjustlinkrevs\zz 4 (T->->==(():):4>>)JK KD*4*6**rrc|dk(ry||S)Nr;FrD)r"cmds r1_iscmdzonetimesetup.._iscmds # #Cyriscmd)NFF) onetimerwireprotocommand getflogheadsgetfilesgetfilecommandsr r4rr basefilectxr)r6rErXrrrrrWs` @r1r3r3Ms#G&&w7&&37&&,7  ME&)R  945=B<|K);=MN - N+-~ OWf=rcxtjj||t|}tjj |r#tjj |dk(rE|j ||}|j|jk(r7tj|j|_ |j ||}t|}tj|}tjd} tjj|}tjj |s tj |d} t%j&|d} | j)|| r| j/ tj||St1|d5} | j3}ddd|S#t"$rYwxYw#t*t,f$rYqwxYw#| r| j/wwxYw#tj|wxYw#1swYSxYw)Nr)fileidswbsrb)rgrdrhrexistsgetsizefilectxnodenullidrsvfscreatefileblobzlibcompressumaskdirnamemakedirsFileExistsErrorratomictempfilewriteIOErrorOSErrorcloserread) r0 cachepathrdr filecachepathrtextoldumaskrr/s r1 _loadfileblobrsGGLLD#d)>- (BGGOOM,Ja,O,,tD,1 <<>T[[ (&00;DNll4l5Gg&}}T"88E? ggoom4G77>>'*KK(A '' u= GGI HHX  K- ' 1668D  K''W%   GGI HHX   Ksl >H GH!'G(H>H/ G%"H$G%%H(G:7G=9G::G==HHH,/H9czj|}|j}djfd|DS)z-A server api for requesting a filelog's heads c3TK|]}|jk7st|!ywrH)rr).0headr0s r1 zgetflogheads..$s JTddkk6Is4yJs(()fileheadsrh)r0rQrdflogrs` r1rr s0 99T?D JJLE ::JUJ KKrcLtj|rdtdzS|jj dd}|s*t j j|j d}t|j}||jk(rydt||||zS)afA server api for requesting a particular version of a file. Can be used in batches to request many files at once. The return protocol is: where is 0 for success or non-zero for an error. data is a compressed blob with revlog flag and ancestors information. See createfileblob for its content. s1+cannot fetch remote files from shallow reporservercachepathremotefilelogcaches0) rrcrr6configrgrdrhrstriprr)r0rQrrrs r1rr'sT"HIII/1CDI GGLL,AB tzz| D t{{ M$ 4> >>rctjrtjt dt t stjt dfd}tj|S)zDA server api for requesting particular versions of particular files.rs/cannot fetch remote files over non-ssh protocolc3Kj}jjdd}|s*tjj jd} |j dd}|syt|dd}|jk(rd9|dd}t|||}dt||fzjjxw)Nrrr(s0 s%d %s) _finr6rrgrdrhreadlinerrrr_foutflush)finrrequestrrdrrQr0s r1streamerzgetfiles..streamerBsjjGGNN#35GH  TYY0EFIllnSb)Gws|$Dt{{" 23F F Fc |jjddsyt}|jd}|jj d}|D]h}||j }|jD]@\}}tjj ||t|} |j| Bj|jjdd} tj| dzdzdzz } |jtdd } | j!d tj"|D]|\} }}|D]q}tjj | |}| j%||vr8tj&|}|j(| ks]tj*|s~| j-y) Nrrsheads(tip~25000:) - nullrsserverexpiration<sremoving old server cachesfiles)unitr)r6rsetrevsvfsrhmanifestitemsrgrdradd configinttime makeprogressrupdatewalk incrementr[st_mtimeremovecomplete)r6r0 neededfilesrrrmffilenamerrdays expirationprogressr!dirsfilesrfilepathr[s r1gcserverr s 77  . :%K II1 2E 34I+ $Z ""$((* + HhGGLLHc(mLM OOM * ++ 77  -/B CDr B 34Jq!=>XNH OOAWWY/ $dE $Dww||D$/H    ;&778$D}}z) (# $ $ r)'rgr[rrmercurial.i18nrmercurial.noderrmercurial.pycompatr mercurialrrr r r r r rrrrrrrrsshv1protocolhandlerrr7rr3rrrrrr rDrr1rsx ## 33  F g>T&RL?("0J<6~r