|
|
# This software may be used and distributed according to the terms of the
|
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
|
|
from __future__ import absolute_import
|
|
|
|
|
|
import re
|
|
|
|
|
|
from mercurial.i18n import _
|
|
|
from mercurial.pycompat import getattr
|
|
|
from mercurial import (
|
|
|
error,
|
|
|
hg,
|
|
|
util,
|
|
|
)
|
|
|
|
|
|
from . import (
|
|
|
lfutil,
|
|
|
localstore,
|
|
|
wirestore,
|
|
|
)
|
|
|
|
|
|
# During clone this function is passed the src's ui object
|
|
|
# but it needs the dest's ui object so it can read out of
|
|
|
# the config file. Use repo.ui instead.
|
|
|
def openstore(repo=None, remote=None, put=False, ui=None):
|
|
|
if ui is None:
|
|
|
ui = repo.ui
|
|
|
|
|
|
if not remote:
|
|
|
lfpullsource = getattr(repo, 'lfpullsource', None)
|
|
|
if lfpullsource:
|
|
|
path = ui.expandpath(lfpullsource)
|
|
|
elif put:
|
|
|
path = ui.expandpath(b'default-push', b'default')
|
|
|
else:
|
|
|
path = ui.expandpath(b'default')
|
|
|
|
|
|
# ui.expandpath() leaves 'default-push' and 'default' alone if
|
|
|
# they cannot be expanded: fallback to the empty string,
|
|
|
# meaning the current directory.
|
|
|
if repo is None:
|
|
|
path = ui.expandpath(b'default')
|
|
|
path, _branches = hg.parseurl(path)
|
|
|
remote = hg.peer(repo or ui, {}, path)
|
|
|
elif path == b'default-push' or path == b'default':
|
|
|
remote = repo
|
|
|
else:
|
|
|
path, _branches = hg.parseurl(path)
|
|
|
remote = hg.peer(repo or ui, {}, path)
|
|
|
|
|
|
# The path could be a scheme so use Mercurial's normal functionality
|
|
|
# to resolve the scheme to a repository and use its path
|
|
|
path = util.safehasattr(remote, b'url') and remote.url() or remote.path
|
|
|
|
|
|
match = _scheme_re.match(path)
|
|
|
if not match: # regular filesystem path
|
|
|
scheme = b'file'
|
|
|
else:
|
|
|
scheme = match.group(1)
|
|
|
|
|
|
try:
|
|
|
storeproviders = _storeprovider[scheme]
|
|
|
except KeyError:
|
|
|
raise error.Abort(_(b'unsupported URL scheme %r') % scheme)
|
|
|
|
|
|
for classobj in storeproviders:
|
|
|
try:
|
|
|
return classobj(ui, repo, remote)
|
|
|
except lfutil.storeprotonotcapable:
|
|
|
pass
|
|
|
|
|
|
raise error.Abort(
|
|
|
_(b'%s does not appear to be a largefile store')
|
|
|
% util.hidepassword(path)
|
|
|
)
|
|
|
|
|
|
|
|
|
_storeprovider = {
|
|
|
b'file': [localstore.localstore],
|
|
|
b'http': [wirestore.wirestore],
|
|
|
b'https': [wirestore.wirestore],
|
|
|
b'ssh': [wirestore.wirestore],
|
|
|
}
|
|
|
|
|
|
_scheme_re = re.compile(br'^([a-zA-Z0-9+-.]+)://')
|
|
|
|
|
|
|
|
|
def getlfile(ui, hash):
|
|
|
return util.chunkbuffer(openstore(ui=ui)._get(hash))
|
|
|
|
