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 = True
44
import urlgrabber.keepalive
45
urlgrabber.keepalive.DEBUG = 0
46
def get_url(path, compressed=False):
51
mutter("grab url %s" % url)
52
url_f = urlgrabber.urlopen(url, keepalive=1, close_connection=0)
56
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
57
except urllib2.URLError, e:
58
raise BzrError("remote fetch failed: %r: %s" % (url, e))
60
def get_url(url, compressed=False):
64
mutter("get_url %s" % url)
65
url_f = urllib2.urlopen(url)
67
return gzip.GzipFile(fileobj=StringIO(url_f.read()))
73
def _find_remote_root(url):
74
"""Return the prefix URL that corresponds to the branch root."""
78
ff = get_url(url + '/.bzr/branch-format')
83
fmt = fmt.rstrip('\r\n')
84
if fmt != BZR_BRANCH_FORMAT.rstrip('\r\n'):
85
raise BzrError("sorry, branch format %r not supported at url %s"
89
except urllib2.URLError:
95
raise BzrError('no branch root found for URL %s' % orig_url)
101
class RemoteBranch(Branch):
102
def __init__(self, baseurl, find_root=True, lock_mode='r'):
103
"""Create new proxy for a remote branch."""
104
if lock_mode not in ('', 'r'):
105
raise BzrError('lock mode %r is not supported for remote branches'
109
self.baseurl = _find_remote_root(baseurl)
111
self.baseurl = baseurl
114
self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
115
self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
118
return '%s(%r)' % (self.__class__.__name__, self.baseurl)
122
def controlfile(self, filename, mode):
123
if mode not in ('rb', 'rt', 'r'):
124
raise BzrError("file mode %r not supported for remote branches" % mode)
125
return get_url(self.baseurl + '/.bzr/' + filename, False)
127
def _need_readlock(self):
128
# remote branch always safe for read
131
def _need_writelock(self):
132
raise BzrError("cannot get write lock on HTTP remote branch")
134
def relpath(self, path):
135
if not path.startswith(self.baseurl):
136
raise BzrError('path %r is not under base URL %r'
137
% (path, self.baseurl))
138
pl = len(self.baseurl)
139
return path[pl:].lstrip('/')
141
def get_revision(self, revision_id):
142
from revision import Revision
143
revf = get_url(self.baseurl + '/.bzr/revision-store/' + revision_id,
145
r = Revision.read_xml(revf)
146
if r.revision_id != revision_id:
147
raise BzrCheckError('revision stored as {%s} actually contains {%s}'
148
% (revision_id, r.revision_id))
153
def __init__(self, baseurl):
154
self._baseurl = baseurl
157
def _path(self, name):
159
raise ValueError('invalid store id', name)
160
return self._baseurl + '/' + name
162
def __getitem__(self, fileid):
163
p = self._path(fileid)
164
return get_url(p, compressed=True)
168
"""For experimental purposes, traverse many parts of a remote branch"""
169
from revision import Revision
170
from branch import Branch
171
from inventory import Inventory
172
from bzrlib import set
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__':