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
29
from cStringIO import StringIO
32
from errors import BzrError, BzrCheckError
33
from branch import Branch, BZR_BRANCH_FORMAT
34
from trace import mutter
36
# velocitynet.com.au transparently proxies connections and thereby
37
# breaks keep-alive -- sucks!
40
ENABLE_URLGRABBER = True
45
import urlgrabber.keepalive
46
urlgrabber.keepalive.DEBUG = 0
47
def get_url(path, compressed=False):
52
mutter("grab url %s" % url)
53
url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
57
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
58
except urllib2.URLError, e:
59
raise BzrError("remote fetch failed: %r: %s" % (url, e))
61
def get_url(url, compressed=False):
65
mutter("get_url %s" % url)
66
url_f = urllib2.urlopen(url)
68
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
74
def _find_remote_root(url):
75
"""Return the prefix URL that corresponds to the branch root."""
79
ff = get_url(url + '/.bzr/branch-format')
84
fmt = fmt.rstrip('\r\n')
85
if fmt != BZR_BRANCH_FORMAT.rstrip('\r\n'):
86
raise BzrError("sorry, branch format %r not supported at url %s"
90
except urllib2.URLError:
96
raise BzrError('no branch root found for URL %s' % orig_url)
102
class RemoteBranch(Branch):
103
def __init__(self, baseurl, find_root=True, lock_mode='r'):
104
"""Create new proxy for a remote branch."""
105
if lock_mode not in ('', 'r'):
106
raise BzrError('lock mode %r is not supported for remote branches'
110
self.baseurl = _find_remote_root(baseurl)
112
self.baseurl = baseurl
115
self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
116
self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
119
return '%s(%r)' % (self.__class__.__name__, self.baseurl)
123
def controlfile(self, filename, mode):
124
if mode not in ('rb', 'rt', 'r'):
125
raise BzrError("file mode %r not supported for remote branches" % mode)
126
return get_url(self.baseurl + '/.bzr/' + filename, False)
128
def _need_readlock(self):
129
# remote branch always safe for read
132
def _need_writelock(self):
133
raise BzrError("cannot get write lock on HTTP remote branch")
135
def relpath(self, path):
136
if not path.startswith(self.baseurl):
137
raise BzrError('path %r is not under base URL %r'
138
% (path, self.baseurl))
139
pl = len(self.baseurl)
140
return path[pl:].lstrip('/')
142
def get_revision(self, revision_id):
143
from revision import Revision
144
revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
146
r = Revision.read_xml(revf)
147
if r.revision_id != revision_id:
148
raise BzrCheckError('revision stored as {%s} actually contains {%s}'
149
% (revision_id, r.revision_id))
154
def __init__(self, baseurl):
155
self._baseurl = baseurl
158
def _path(self, name):
160
raise ValueError('invalid store id', name)
161
return self._baseurl + '/' + name
163
def __getitem__(self, fileid):
164
p = self._path(fileid)
165
return get_url(p, compressed=True)
169
"""For experimental purposes, traverse many parts of a remote branch"""
170
from revision import Revision
171
from branch import Branch
172
from inventory import Inventory
178
history = get_url('/.bzr/revision-history').readlines()
179
num_revs = len(history)
180
for i, rev_id in enumerate(history):
181
rev_id = rev_id.rstrip()
182
print 'read revision %d/%d' % (i, num_revs)
184
# python gzip needs a seekable file (!!) but the HTTP response
185
# isn't, so we need to buffer it
187
rev_f = get_url('/.bzr/revision-store/%s' % rev_id,
190
rev = Revision.read_xml(rev_f)
192
inv_id = rev.inventory_id
193
if inv_id not in got_invs:
194
print 'get inventory %s' % inv_id
195
inv_f = get_url('/.bzr/inventory-store/%s' % inv_id,
197
inv = Inventory.read_xml(inv_f)
198
print '%4d inventory entries' % len(inv)
200
for path, ie in inv.iter_entries():
204
if text_id in got_texts:
206
print ' fetch %s text {%s}' % (path, text_id)
207
text_f = get_url('/.bzr/text-store/%s' % text_id,
209
got_texts.add(text_id)
217
BASE_URL = 'http://bazaar-ng.org/bzr/bzr.dev/'
218
b = RemoteBranch(BASE_URL)
219
## print '\n'.join(b.revision_history())
220
from log import show_log
224
if __name__ == '__main__':