|
|
# statichttprepo.py - simple http repository class for mercurial
|
|
|
#
|
|
|
# This provides read-only repo access to repositories exported via static http
|
|
|
#
|
|
|
# Copyright 2005-2007 Matt Mackall <mpm@selenic.com>
|
|
|
#
|
|
|
# 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 errno
|
|
|
|
|
|
from .i18n import _
|
|
|
from . import (
|
|
|
branchmap,
|
|
|
changelog,
|
|
|
error,
|
|
|
localrepo,
|
|
|
manifest,
|
|
|
namespaces,
|
|
|
pathutil,
|
|
|
pycompat,
|
|
|
url,
|
|
|
util,
|
|
|
vfs as vfsmod,
|
|
|
)
|
|
|
|
|
|
urlerr = util.urlerr
|
|
|
urlreq = util.urlreq
|
|
|
|
|
|
|
|
|
class httprangereader(object):
|
|
|
def __init__(self, url, opener):
|
|
|
# we assume opener has HTTPRangeHandler
|
|
|
self.url = url
|
|
|
self.pos = 0
|
|
|
self.opener = opener
|
|
|
self.name = url
|
|
|
|
|
|
def __enter__(self):
|
|
|
return self
|
|
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
|
self.close()
|
|
|
|
|
|
def seek(self, pos):
|
|
|
self.pos = pos
|
|
|
|
|
|
def read(self, bytes=None):
|
|
|
req = urlreq.request(pycompat.strurl(self.url))
|
|
|
end = b''
|
|
|
if bytes:
|
|
|
end = self.pos + bytes - 1
|
|
|
if self.pos or end:
|
|
|
req.add_header('Range', 'bytes=%d-%s' % (self.pos, end))
|
|
|
|
|
|
try:
|
|
|
f = self.opener.open(req)
|
|
|
data = f.read()
|
|
|
code = f.code
|
|
|
except urlerr.httperror as inst:
|
|
|
num = inst.code == 404 and errno.ENOENT or None
|
|
|
raise IOError(num, inst)
|
|
|
except urlerr.urlerror as inst:
|
|
|
raise IOError(None, inst.reason)
|
|
|
|
|
|
if code == 200:
|
|
|
# HTTPRangeHandler does nothing if remote does not support
|
|
|
# Range headers and returns the full entity. Let's slice it.
|
|
|
if bytes:
|
|
|
data = data[self.pos : self.pos + bytes]
|
|
|
else:
|
|
|
data = data[self.pos :]
|
|
|
elif bytes:
|
|
|
data = data[:bytes]
|
|
|
self.pos += len(data)
|
|
|
return data
|
|
|
|
|
|
def readlines(self):
|
|
|
return self.read().splitlines(True)
|
|
|
|
|
|
def __iter__(self):
|
|
|
return iter(self.readlines())
|
|
|
|
|
|
def close(self):
|
|
|
pass
|
|
|
|
|
|
|
|
|
# _RangeError and _HTTPRangeHandler were originally in byterange.py,
|
|
|
# which was itself extracted from urlgrabber. See the last version of
|
|
|
# byterange.py from history if you need more information.
|
|
|
class _RangeError(IOError):
|
|
|
"""Error raised when an unsatisfiable range is requested."""
|
|
|
|
|
|
|
|
|
class _HTTPRangeHandler(urlreq.basehandler):
|
|
|
"""Handler that enables HTTP Range headers.
|
|
|
|
|
|
This was extremely simple. The Range header is a HTTP feature to
|
|
|
begin with so all this class does is tell urllib2 that the
|
|
|
"206 Partial Content" response from the HTTP server is what we
|
|
|
expected.
|
|
|
"""
|
|
|
|
|
|
def http_error_206(self, req, fp, code, msg, hdrs):
|
|
|
# 206 Partial Content Response
|
|
|
r = urlreq.addinfourl(fp, hdrs, req.get_full_url())
|
|
|
r.code = code
|
|
|
r.msg = msg
|
|
|
return r
|
|
|
|
|
|
def http_error_416(self, req, fp, code, msg, hdrs):
|
|
|
# HTTP's Range Not Satisfiable error
|
|
|
raise _RangeError(b'Requested Range Not Satisfiable')
|
|
|
|
|
|
|
|
|
def build_opener(ui, authinfo):
|
|
|
# urllib cannot handle URLs with embedded user or passwd
|
|
|
urlopener = url.opener(ui, authinfo)
|
|
|
urlopener.add_handler(_HTTPRangeHandler())
|
|
|
|
|
|
class statichttpvfs(vfsmod.abstractvfs):
|
|
|
def __init__(self, base):
|
|
|
self.base = base
|
|
|
self.options = {}
|
|
|
|
|
|
def __call__(self, path, mode=b'r', *args, **kw):
|
|
|
if mode not in (b'r', b'rb'):
|
|
|
raise IOError(b'Permission denied')
|
|
|
f = b"/".join((self.base, urlreq.quote(path)))
|
|
|
return httprangereader(f, urlopener)
|
|
|
|
|
|
def join(self, path):
|
|
|
if path:
|
|
|
return pathutil.join(self.base, path)
|
|
|
else:
|
|
|
return self.base
|
|
|
|
|
|
return statichttpvfs
|
|
|
|
|
|
|
|
|
class statichttppeer(localrepo.localpeer):
|
|
|
def local(self):
|
|
|
return None
|
|
|
|
|
|
def canpush(self):
|
|
|
return False
|
|
|
|
|
|
|
|
|
class statichttprepository(
|
|
|
localrepo.localrepository, localrepo.revlogfilestorage
|
|
|
):
|
|
|
supported = localrepo.localrepository._basesupported
|
|
|
|
|
|
def __init__(self, ui, path):
|
|
|
self._url = path
|
|
|
self.ui = ui
|
|
|
|
|
|
self.root = path
|
|
|
u = util.url(path.rstrip(b'/') + b"/.hg")
|
|
|
self.path, authinfo = u.authinfo()
|
|
|
|
|
|
vfsclass = build_opener(ui, authinfo)
|
|
|
self.vfs = vfsclass(self.path)
|
|
|
self.cachevfs = vfsclass(self.vfs.join(b'cache'))
|
|
|
self._phasedefaults = []
|
|
|
|
|
|
self.names = namespaces.namespaces()
|
|
|
self.filtername = None
|
|
|
self._extrafilterid = None
|
|
|
|
|
|
try:
|
|
|
requirements = set(self.vfs.read(b'requires').splitlines())
|
|
|
except IOError as inst:
|
|
|
if inst.errno != errno.ENOENT:
|
|
|
raise
|
|
|
requirements = set()
|
|
|
|
|
|
# check if it is a non-empty old-style repository
|
|
|
try:
|
|
|
fp = self.vfs(b"00changelog.i")
|
|
|
fp.read(1)
|
|
|
fp.close()
|
|
|
except IOError as inst:
|
|
|
if inst.errno != errno.ENOENT:
|
|
|
raise
|
|
|
# we do not care about empty old-style repositories here
|
|
|
msg = _(b"'%s' does not appear to be an hg repository") % path
|
|
|
raise error.RepoError(msg)
|
|
|
|
|
|
supportedrequirements = localrepo.gathersupportedrequirements(ui)
|
|
|
localrepo.ensurerequirementsrecognized(
|
|
|
requirements, supportedrequirements
|
|
|
)
|
|
|
localrepo.ensurerequirementscompatible(ui, requirements)
|
|
|
|
|
|
# setup store
|
|
|
self.store = localrepo.makestore(requirements, self.path, vfsclass)
|
|
|
self.spath = self.store.path
|
|
|
self.svfs = self.store.opener
|
|
|
self.sjoin = self.store.join
|
|
|
self._filecache = {}
|
|
|
self.requirements = requirements
|
|
|
|
|
|
rootmanifest = manifest.manifestrevlog(self.svfs)
|
|
|
self.manifestlog = manifest.manifestlog(
|
|
|
self.svfs, self, rootmanifest, self.narrowmatch()
|
|
|
)
|
|
|
self.changelog = changelog.changelog(self.svfs)
|
|
|
self._tags = None
|
|
|
self.nodetagscache = None
|
|
|
self._branchcaches = branchmap.BranchMapCache()
|
|
|
self._revbranchcache = None
|
|
|
self.encodepats = None
|
|
|
self.decodepats = None
|
|
|
self._transref = None
|
|
|
|
|
|
def _restrictcapabilities(self, caps):
|
|
|
caps = super(statichttprepository, self)._restrictcapabilities(caps)
|
|
|
return caps.difference([b"pushkey"])
|
|
|
|
|
|
def url(self):
|
|
|
return self._url
|
|
|
|
|
|
def local(self):
|
|
|
return False
|
|
|
|
|
|
def peer(self):
|
|
|
return statichttppeer(self)
|
|
|
|
|
|
def wlock(self, wait=True):
|
|
|
raise error.LockUnavailable(
|
|
|
0,
|
|
|
_(b'lock not available'),
|
|
|
b'lock',
|
|
|
_(b'cannot lock static-http repository'),
|
|
|
)
|
|
|
|
|
|
def lock(self, wait=True):
|
|
|
raise error.Abort(_(b'cannot lock static-http repository'))
|
|
|
|
|
|
def _writecaches(self):
|
|
|
pass # statichttprepository are read only
|
|
|
|
|
|
|
|
|
def instance(ui, path, create, intents=None, createopts=None):
|
|
|
if create:
|
|
|
raise error.Abort(_(b'cannot create new static-http repository'))
|
|
|
return statichttprepository(ui, path[7:])
|
|
|
|
