1
# Copyright (C) 2005 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
"""Proxy object for access to remote branches.
20
At the moment remote branches are only for HTTP and only for read
26
from cStringIO import StringIO
31
from bzrlib.errors import BzrError, BzrCheckError
32
from bzrlib.branch import Branch, BZR_BRANCH_FORMAT
33
from bzrlib.trace import mutter
34
from bzrlib.xml import serializer_v4
37
ENABLE_URLGRABBER = False
39
from bzrlib.errors import BzrError, NoSuchRevision
41
class GetFailed(BzrError):
42
def __init__(self, url, status):
43
BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
48
import util.urlgrabber
49
import util.urlgrabber.keepalive
50
util.urlgrabber.keepalive.DEBUG = 0
51
def get_url(path, compressed=False):
56
mutter("grab url %s" % url)
57
url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
58
if url_f.status != 200:
59
raise GetFailed(url, url_f.status)
63
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
64
except urllib2.URLError, e:
65
raise BzrError("remote fetch failed: %r: %s" % (url, e))
67
def get_url(url, compressed=False):
71
mutter("get_url %s" % url)
72
url_f = urllib2.urlopen(url)
74
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
80
def _find_remote_root(url):
81
"""Return the prefix URL that corresponds to the branch root."""
85
fmt_url = url + '/.bzr/branch-format'
90
fmt = fmt.rstrip('\r\n')
91
if fmt != BZR_BRANCH_FORMAT.rstrip('\r\n'):
92
raise BzrError("sorry, branch format %r not supported at url %s"
96
except urllib2.URLError:
99
scheme, host, path = list(urlparse.urlparse(url))[:3]
100
# discard params, query, fragment
102
# strip off one component of the path component
103
idx = path.rfind('/')
104
if idx == -1 or path == '/':
105
raise BzrError('no branch root found for URL %s'
106
' or enclosing directories'
109
url = urlparse.urlunparse((scheme, host, path, '', '', ''))
113
class RemoteBranch(Branch):
114
def __init__(self, baseurl, find_root=True):
115
"""Create new proxy for a remote branch."""
117
self.base = _find_remote_root(baseurl)
122
self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
123
self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
124
self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
127
b = getattr(self, 'baseurl', 'undefined')
128
return '%s(%r)' % (self.__class__.__name__, b)
132
def controlfile(self, filename, mode):
133
if mode not in ('rb', 'rt', 'r'):
134
raise BzrError("file mode %r not supported for remote branches" % mode)
135
return get_url(self.base + '/.bzr/' + filename, False)
139
# no locking for remote branches yet
142
def lock_write(self):
143
from errors import LockError
144
raise LockError("write lock not supported for remote branch %s"
151
def relpath(self, path):
152
if not path.startswith(self.base):
153
raise BzrError('path %r is not under base URL %r'
156
return path[pl:].lstrip('/')
159
def get_revision(self, revision_id):
161
revf = self.revision_store[revision_id]
163
raise NoSuchRevision(self, revision_id)
164
r = serializer_v4.read_revision(revf)
165
if r.revision_id != revision_id:
166
raise BzrCheckError('revision stored as {%s} actually contains {%s}'
167
% (revision_id, r.revision_id))
171
class RemoteStore(object):
172
def __init__(self, baseurl):
173
self._baseurl = baseurl
176
def _path(self, name):
178
raise ValueError('invalid store id', name)
179
return self._baseurl + '/' + name
181
def __getitem__(self, fileid):
182
p = self._path(fileid)
184
return get_url(p, compressed=True)
185
except urllib2.URLError:
186
raise KeyError(fileid)
added
removed
bzrlib/remotebranch.py
