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, LocalBranch, BZR_BRANCH_FORMAT_5
33
from bzrlib.trace import mutter
34
from bzrlib.weavestore import WeaveStore
35
from bzrlib.xml5 import serializer_v5
38
# velocitynet.com.au transparently proxies connections and thereby
39
# breaks keep-alive -- sucks!
42
ENABLE_URLGRABBER = False
44
from bzrlib.errors import BzrError, NoSuchRevision
46
class GetFailed(BzrError):
47
def __init__(self, url, status):
48
BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
53
import util.urlgrabber
54
import util.urlgrabber.keepalive
55
util.urlgrabber.keepalive.DEBUG = 0
56
def get_url(path, compressed=False):
61
mutter("grab url %s" % url)
62
url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
63
if url_f.status != 200:
64
raise GetFailed(url, url_f.status)
68
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
69
except urllib2.URLError, e:
70
raise BzrError("remote fetch failed: %r: %s" % (url, e))
72
def get_url(url, compressed=False):
76
mutter("get_url %s" % url)
77
url_f = urllib2.urlopen(url)
79
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
84
def _find_remote_root(url):
85
"""Return the prefix URL that corresponds to the branch root."""
89
fmt_url = url + '/.bzr/branch-format'
94
if fmt != BZR_BRANCH_FORMAT_5:
95
raise BzrError("sorry, branch format %r not supported at url %s"
99
except urllib2.URLError:
102
scheme, host, path = list(urlparse.urlparse(url))[:3]
103
# discard params, query, fragment
105
# strip off one component of the path component
106
idx = path.rfind('/')
107
if idx == -1 or path == '/':
108
raise BzrError('no branch root found for URL %s'
109
' or enclosing directories'
112
url = urlparse.urlunparse((scheme, host, path, '', '', ''))
116
class RemoteBranch(LocalBranch):
118
def __init__(self, baseurl, find_root=True):
119
"""Create new proxy for a remote branch."""
120
# circular import protection
121
from bzrlib.store import RemoteStore
123
self.base = _find_remote_root(baseurl)
126
self._check_format(False)
127
# is guaranteed to be a v5 store
129
cfn = self.controlfilename
130
assert self._branch_format == 5
131
self.control_weaves = WeaveStore(cfn([]), get_url)
132
self.weave_store = WeaveStore(cfn('weaves'), get_url)
133
self.revision_store = RemoteStore(cfn('revision-store'))
136
b = getattr(self, 'baseurl', 'undefined')
137
return '%s(%r)' % (self.__class__.__name__, b)
141
def setup_caching(self, cache_root):
142
"""Set up cached stores located under cache_root"""
143
from bzrlib.meta_store import CachedStore
144
for store_name in ('revision_store',):
145
if not isinstance(getattr(self, store_name), CachedStore):
146
cache_path = os.path.join(cache_root, store_name)
148
new_store = CachedStore(getattr(self, store_name), cache_path)
149
setattr(self, store_name, new_store)
151
def controlfile(self, filename, mode):
152
if mode not in ('rb', 'rt', 'r'):
153
raise BzrError("file mode %r not supported for remote branches" % mode)
154
return get_url(self.base + '/.bzr/' + filename, False)
158
# no locking for remote branches yet
161
def lock_write(self):
162
from errors import LockError
163
raise LockError("write lock not supported for remote branch %s"
170
def relpath(self, path):
171
if not path.startswith(self.base):
172
raise BzrError('path %r is not under base URL %r'
175
return path[pl:].lstrip('/')
178
def get_revision(self, revision_id):
180
revf = self.revision_store[revision_id]
182
raise NoSuchRevision(self, revision_id)
183
r = serializer_v5.read_revision(revf)
184
if r.revision_id != revision_id:
185
raise BzrCheckError('revision stored as {%s} actually contains {%s}'
186
% (revision_id, r.revision_id))
added
removed
bzrlib/remotebranch.py
