?\ bddlZddlZddlZddlZddlmZddlmZmZddl 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| | } t|D cgc] } | | r | c} }||||||g|i|Scc} w)Nsincludepattern=sexcludepattern=) _bundlecapsrBUNDLE2_CAPABLITY startswithlensplitr alwaysrootlist)origself changedfiles linknodes commonrevssourceargskwargscapsincludepatternexcludepatterncapmfrepos Kusr/lib/python3.12/site-packages/hgext/remotefilelog/remotefilelogserver.py generatefilesz"setupserver..generatefiles,s%2  & &$ .!N!N''-2->>"45%(-?)@)B%C%I%I%%PN^^$67%(-?)@)B%C%I%I%%PN .  AKKIIsD.. L ELq!L EFL , :v @D HN  !Fs  D-Ds generatefilesN) onetimesetupr wrapfunctionrcgpacker)uir1r3s ` r2 setupserverr8's. 2. 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 shallowremotenoflatmfrr2 streamstater?_s rrFc 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 r2rJz5onetimesetup..stream_out_shallow..gens$  &]]EK+s") getrrCr rDr r!rstreamnextrJrrN)r1protootherr,r-raw oldshallowoldmatch oldnoflatmfrJrLrMrNstates @@@r2stream_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)"rCr isenabledrrpathrawvfsreaddirosjoinpopr\S_IFREGendswithrpconvert decodedirSimpleStoreEntryst_sizeS_IFDIRappendr istreemanifest data_entries is_revlogis_manifestlogr r target_id top_entriesrDr Abortr)r#r1matcherrbrcstriplenrgvisitpr0kindstfpndentryr xrXs r2_walkstreamfilesz&onetimesetup.._walkstreamfiless9   $$T*tzz/!3**++33djjoow?@ A'.qt'< 4X\4<</#%;;u#5bkk %?%)MM"XY-$@$)OOA$6&+&<&>ad3Bi>&A.  " "4 (++EF gUs!EJ;#J;9A J;A*J;0B J;s_walkstreamfilesc |||}tj|sjddrQt|tr|j t j|j d|j d|S)N remotefilelogserverr:r=)rrd configbool isinstance _sshv1serverrrrNETWORK_CAP_LEGACY_SSH_GETFILES)r#r1rRr+r7s r2 _capabilitiesz#onetimesetup.._capabilitiessjD%    &"-- i+ %. IEEF KK- . KK( ) rs _capabilitiesc|j}tj|drJ|jr>|jj |jj |jS||g|i|S)Ns forcelinkrev)_repor safehasattr forcelinkrev_fileloglinkrevrev _filenode)r#r$r)r*r1s r2_adjustlinkrevz$onetimesetup.._adjustlinkrevsbzz   D/ 2t7H7H==(():):4>>)JK KD*4*6**rs_adjustlinkrevc|dk(ry||S)Nr<FrE)r#cmds r2_iscmdzonetimesetup.._iscmds # #Cyrsiscmd)NFF) onetimerwireprotocommand getflogheadsgetfilesgetfilecommandsr r5rr basefilectxr)r7rFrYrrrrrXs` @r2r4r4Ms$G&&w7&&37&&,7  ME&)R  945=B<|K)<>NO -/?O+. OXv>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)rhrerirexistsgetsizefilectxnodenullidrsvfscreatefileblobzlibcompressumaskdirnamemakedirsFileExistsErrorratomictempfilewriteIOErrorOSErrorcloserread) r1 cachepathrer filecachepathrtextoldumaskrr0s r2 _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|!ywrI)rr).0headr1s r2 zgetflogheads..$s!JUTddkk6Is4yUs(()fileheadsri)r1rRreflogrs` r2rr 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) rrdrr7configrhrerirstriprr)r1rRrrrs r2rr'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) _finr7rrhrerireadlinerrrr_foutflush)finrrequestrrerrRr1s r2streamerzgetfiles..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)r7rsetrevsvfsrimanifestitemsrhreradd configinttime makeprogressrupdatewalk incrementr\st_mtimeremovecomplete)r7r1 neededfilesrrrmffilenamerrdays expirationprogressr!dirsfilesrfilepathr\s r2gcserverr sx 77  . :%K II1 2E 34I $Z ""$((* HhGGLLHc(mLM OOM *#- 77  -/B CDr B 34Jq!=>XNH OOAWWY/dEDww||D$/H    ;&778$D}}z) (#0 r)'rhr\rrmercurial.i18nrmercurial.noderrmercurial.pycompatr mercurialrrr r r r r rrrrrrrrsshv1protocolhandlerrr8rr4rrrrrr rErr2rsx ## 33  F g?T&RL?("0J<6~r