^caddlZddlZddlZddlmZddlmZddlmZddl m Z m Z Gdde Z Gdd e ZGd d ZGd d ZddZGddeZdS)N)hex)open)pycompat)hashutilprocutilceZdZdS)BundleWriteExceptionN__name__ __module__ __qualname__:/usr/lib/python3/dist-packages/hgext/infinitepush/store.pyr r Drr ceZdZdS)BundleReadExceptionNr rrrrrrrrc`eZdZdZejZejdZejdZ dS)abstractbundlestorea6Defines the interface for bundle stores. A bundle store is an entity that stores raw bundle data. It is a simple key-value store. However, the keys are chosen by the store. The keys can be any Python object understood by the corresponding bundle index (see ``abstractbundleindex`` below). cdS)zWrite bundle data to the store. This function receives the raw data to be written as a str. Throws BundleWriteException The key of the written data MUST be returned. Nr)selfdatas rwritezabstractbundlestore.write'rcdS)zObtain bundle data for a key. Returns None if the bundle isn't known. Throws BundleReadException The returned object should be a file object supporting read() and close(). Nr)rkeys rreadzabstractbundlestore.read0rrN) r r r __doc__abcABCMeta __metaclass__abstractmethodrrrrrrrscKM         rrc0eZdZdZdZdZdZdZdZdS)filebundlestorezkbundle store in filesystem meant for storing bundles somewhere on disk and on network filesystems c.||_||_|dd|_|js%|jjdd|_t j|jst j |jdSdS)Ns scratchbranchs storepathsscratchbranchessfilebundlestore) uirepo configpath storepathvfsjoinospathexistsmakedirs)rr&r's r__init__zfilebundlestore.__init__As '7FF~ !Y]//"$6DNw~~dn-- ( K ' ' ' ' ' ( (rcntj|j|dd|ddS)zFirst two bytes of the hash are the name of the upper level directory, next two bytes are the name of the next level directoryr)r,r-r+r))r hashvalues r_dirpathzfilebundlestore._dirpathLs.w||DNIacNIacNKKKrchtj|||S)N)r,r-r+r5)rfilenames r _filepathzfilebundlestore._filepathRs$w||DMM(33X>>>rcttj|}||}t j|st j|t| |d5}| |dddn #1swxYwY|S)Nswb) rrsha1digestr5r,r-r.r/rr8r)rrr7dirpathfs rrzfilebundlestore.writeUsx}T**113344--))w~~g&& ! K $..**E 2 2 a GGDMMM               sCCCc t||d5}|cdddS#1swxYwYdS#t$rYdSwxYw)Nsrb)rr8rIOError)rrr=s rrzfilebundlestore.readas dnnS))511 Qvvxx                     44 s3#AA AA  A A A A! A!N) r r r rr0r5r8rrrrrr$r$;sl ( ( (LLL ???   rr$cg}|D]S}|r|dkr|| |r|dkr||>||T|S)zFormats `args` with Infinitepush replacements. Hack to get `str.format()`-ed strings working in a BC way with bytes. s {filename}s{handle})append)argsr7handleformatted_argsargs rformat_placeholders_argsrFis N''  '},,  ! !( + + + +  '{**  ! !& ) ) ) )  ! !# & & & & rc&eZdZdZdZdZdZdS)externalbundlestorec>||_||_||_||_dS)a6 `put_binary` - path to binary file which uploads bundle to external storage and prints key to stdout `put_args` - format string with additional args to `put_binary` {filename} replacement field can be used. `get_binary` - path to binary file which accepts filename and key (in that order), downloads bundle from store and saves it to file `get_args` - format string with additional args to `get_binary`. {filename} and {handle} replacement field can be used. N)put_argsget_args put_binary get_binary)rrLrJrMrKs rr0zexternalbundlestore.__init__{s$!   $$rctjtjtj|tjtjd}|\}}|j}|||fS)NT)stdoutstderr close_fds) subprocessPopenrrapplyr tonativestrPIPE communicate returncode)rrBprOrPrXs r _call_binaryz externalbundlestore._call_binarys_   OH0$ 7 7??     \ 66))rctj5}||||dt |j|j}||j g|z\}}}|dkrtd|z| }t|dkr|dcdddStd|j |fz#1swxYwYdS)Nr)r7s&Failed to upload to external store: %ssBad output from %s: %s) r namedtempfilerflushseekrFrJnamerZrLr splitlineslen)rrtemprDrXrOrP stdout_liness rrzexternalbundlestore.writes_  # % %  JJt    JJLLL IIaLLL5  N*.):):!N2** &JQ*=F",,..L<  A%%#A#        &+-&0II'          sCC<#C<<DDc0tj5}t|j|j|}||jg|z\}}}|dkrtd|z|cdddS#1swxYwYdS)N)r7rCrs*Failed to download from external store: %s) rr]rFrKr`rZrMrr)rrCrcrDrXrOrPs rrzexternalbundlestore.reads # % % 5  &N*.):):!N2** &JQ)AFJ99;;                  sA*B  BBN)r r r r0rZrrrrrrHrHzsP%%%" * * *8rrH)NN)rr,rRmercurial.nodermercurial.pycompatr mercurialrmercurial.utilsrr Exceptionr rrr$rFrHrrrrks{ ######      9        )           >++++++++\"HHHHH-HHHHHr