3
# Copyright (C) 2005 Canonical Ltd
5
# This program is free software; you can redistribute it and/or modify
6
# it under the terms of the GNU General Public License as published by
7
# the Free Software Foundation; either version 2 of the License, or
8
# (at your option) any later version.
10
# This program is distributed in the hope that it will be useful,
11
# but WITHOUT ANY WARRANTY; without even the implied warranty of
12
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
# GNU General Public License for more details.
15
# You should have received a copy of the GNU General Public License
16
# along with this program; if not, write to the Free Software
17
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20
"""Proxy object for access to remote branches.
22
At the moment remote branches are only for HTTP and only for read
28
from cStringIO import StringIO
31
from errors import BzrError, BzrCheckError
32
from branch import Branch, BZR_BRANCH_FORMAT
33
from trace import mutter
35
# velocitynet.com.au transparently proxies connections and thereby
36
# breaks keep-alive -- sucks!
39
ENABLE_URLGRABBER = False
41
from bzrlib.errors import BzrError, NoSuchRevision
43
class GetFailed(BzrError):
44
def __init__(self, url, status):
45
BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
50
import util.urlgrabber
51
import util.urlgrabber.keepalive
52
util.urlgrabber.keepalive.DEBUG = 0
53
def get_url(path, compressed=False):
58
mutter("grab url %s" % url)
59
url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
60
if url_f.status != 200:
61
raise GetFailed(url, url_f.status)
65
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
66
except urllib2.URLError, e:
67
raise BzrError("remote fetch failed: %r: %s" % (url, e))
69
def get_url(url, compressed=False):
73
mutter("get_url %s" % url)
74
url_f = urllib2.urlopen(url)
76
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
82
def _find_remote_root(url):
83
"""Return the prefix URL that corresponds to the branch root."""
87
ff = get_url(url + '/.bzr/branch-format')
92
fmt = fmt.rstrip('\r\n')
93
if fmt != BZR_BRANCH_FORMAT.rstrip('\r\n'):
94
raise BzrError("sorry, branch format %r not supported at url %s"
98
except urllib2.URLError:
102
idx = url.rindex('/')
104
raise BzrError('no branch root found for URL %s' % orig_url)
109
class RemoteBranch(Branch):
111
def __init__(self, baseurl, find_root=True):
112
"""Create new proxy for a remote branch."""
113
# circular import protection
114
from bzrlib.store import RemoteStore
116
self.baseurl = _find_remote_root(baseurl)
118
self.baseurl = baseurl
120
self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
121
self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
122
self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
125
b = getattr(self, 'baseurl', 'undefined')
126
return '%s(%r)' % (self.__class__.__name__, b)
130
def controlfile(self, filename, mode):
131
if mode not in ('rb', 'rt', 'r'):
132
raise BzrError("file mode %r not supported for remote branches" % mode)
133
return get_url(self.baseurl + '/.bzr/' + filename, False)
137
# no locking for remote branches yet
140
def lock_write(self):
141
from errors import LockError
142
raise LockError("write lock not supported for remote branch %s"
149
def relpath(self, path):
150
if not path.startswith(self.baseurl):
151
raise BzrError('path %r is not under base URL %r'
152
% (path, self.baseurl))
153
pl = len(self.baseurl)
154
return path[pl:].lstrip('/')
157
def get_revision(self, revision_id):
158
from bzrlib.revision import Revision
159
from bzrlib.xml import unpack_xml
161
revf = self.revision_store[revision_id]
163
raise NoSuchRevision(self, revision_id)
164
r = unpack_xml(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))
added
removed
bzrlib/remotebranch.py
