/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/remotebranch.py

  • Committer: Robert Collins
  • Date: 2010-05-06 11:08:10 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506110810-h3j07fh5gmw54s25
Cleaner matcher matching revised unlocking protocol.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/remotebranch.py
1
 
# Copyright (C) 2005 Canonical Ltd
2
 
 
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.
7
 
 
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.
12
 
 
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
16
 
 
17
 
 
18
 
"""Proxy object for access to remote branches.
19
 
 
20
 
At the moment remote branches are only for HTTP and only for read
21
 
access.
22
 
"""
23
 
 
24
 
 
25
 
import gzip
26
 
from cStringIO import StringIO
27
 
import os
28
 
import urllib2
29
 
import urlparse
30
 
 
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.xml5 import serializer_v5
35
 
 
36
 
# velocitynet.com.au transparently proxies connections and thereby
37
 
# breaks keep-alive -- sucks!
38
 
 
39
 
 
40
 
ENABLE_URLGRABBER = False
41
 
 
42
 
from bzrlib.errors import BzrError, NoSuchRevision
43
 
 
44
 
class GetFailed(BzrError):
45
 
    def __init__(self, url, status):
46
 
        BzrError.__init__(self, "Get %s failed with status %s" % (url, status))
47
 
        self.url = url
48
 
        self.status = status
49
 
 
50
 
if ENABLE_URLGRABBER:
51
 
    import util.urlgrabber
52
 
    import util.urlgrabber.keepalive
53
 
    util.urlgrabber.keepalive.DEBUG = 0
54
 
    def get_url(path, compressed=False):
55
 
        try:
56
 
            url = path
57
 
            if compressed:
58
 
                url += '.gz'
59
 
            mutter("grab url %s" % url)
60
 
            url_f = util.urlgrabber.urlopen(url, keepalive=1, close_connection=0)
61
 
            if url_f.status != 200:
62
 
                raise GetFailed(url, url_f.status)
63
 
            if not compressed:
64
 
                return url_f
65
 
            else:
66
 
                return gzip.GzipFile(fileobj=StringIO(url_f.read()))
67
 
        except urllib2.URLError, e:
68
 
            raise BzrError("remote fetch failed: %r: %s" % (url, e))
69
 
else:
70
 
    def get_url(url, compressed=False):
71
 
        import urllib2
72
 
        if compressed:
73
 
            url += '.gz'
74
 
        mutter("get_url %s" % url)
75
 
        url_f = urllib2.urlopen(url)
76
 
        if compressed:
77
 
            return gzip.GzipFile(fileobj=StringIO(url_f.read()))
78
 
        else:
79
 
            return url_f
80
 
 
81
 
 
82
 
 
83
 
def _find_remote_root(url):
84
 
    """Return the prefix URL that corresponds to the branch root."""
85
 
    orig_url = url
86
 
    while True:
87
 
        try:
88
 
            fmt_url = url + '/.bzr/branch-format'
89
 
            ff = get_url(fmt_url)
90
 
            fmt = ff.read()
91
 
            ff.close()
92
 
 
93
 
            if fmt != BZR_BRANCH_FORMAT_5:
94
 
                raise BzrError("sorry, branch format %r not supported at url %s"
95
 
                               % (fmt, url))
96
 
            
97
 
            return url
98
 
        except urllib2.URLError:
99
 
            pass
100
 
 
101
 
        scheme, host, path = list(urlparse.urlparse(url))[:3]
102
 
        # discard params, query, fragment
103
 
        
104
 
        # strip off one component of the path component
105
 
        idx = path.rfind('/')
106
 
        if idx == -1 or path == '/':
107
 
            raise BzrError('no branch root found for URL %s'
108
 
                           ' or enclosing directories'
109
 
                           % orig_url)
110
 
        path = path[:idx]
111
 
        url = urlparse.urlunparse((scheme, host, path, '', '', ''))
112
 
        
113
 
 
114
 
 
115
 
class RemoteBranch(LocalBranch):
116
 
    def __init__(self, baseurl, find_root=True):
117
 
        """Create new proxy for a remote branch."""
118
 
        if find_root:
119
 
            self.base = _find_remote_root(baseurl)
120
 
        else:
121
 
            self.base = baseurl
122
 
            self._check_format()
123
 
 
124
 
        self.inventory_store = RemoteStore(baseurl + '/.bzr/inventory-store/')
125
 
        self.text_store = RemoteStore(baseurl + '/.bzr/text-store/')
126
 
        self.revision_store = RemoteStore(baseurl + '/.bzr/revision-store/')
127
 
 
128
 
    def __str__(self):
129
 
        b = getattr(self, 'baseurl', 'undefined')
130
 
        return '%s(%r)' % (self.__class__.__name__, b)
131
 
 
132
 
    __repr__ = __str__
133
 
 
134
 
    def setup_caching(self, cache_root):
135
 
        """Set up cached stores located under cache_root"""
136
 
        from bzrlib.meta_store import CachedStore
137
 
        for store_name in ('inventory_store', 'text_store', 'revision_store'):
138
 
            if not isinstance(getattr(self, store_name), CachedStore):
139
 
                cache_path = os.path.join(cache_root, store_name)
140
 
                os.mkdir(cache_path)
141
 
                new_store = CachedStore(getattr(self, store_name), cache_path)
142
 
                setattr(self, store_name, new_store)
143
 
 
144
 
    def controlfile(self, filename, mode):
145
 
        if mode not in ('rb', 'rt', 'r'):
146
 
            raise BzrError("file mode %r not supported for remote branches" % mode)
147
 
        return get_url(self.base + '/.bzr/' + filename, False)
148
 
 
149
 
 
150
 
    def lock_read(self):
151
 
        # no locking for remote branches yet
152
 
        pass
153
 
 
154
 
    def lock_write(self):
155
 
        from errors import LockError
156
 
        raise LockError("write lock not supported for remote branch %s"
157
 
                        % self.base)
158
 
 
159
 
    def unlock(self):
160
 
        pass
161
 
    
162
 
 
163
 
    def relpath(self, path):
164
 
        if not path.startswith(self.base):
165
 
            raise BzrError('path %r is not under base URL %r'
166
 
                           % (path, self.base))
167
 
        pl = len(self.base)
168
 
        return path[pl:].lstrip('/')
169
 
 
170
 
 
171
 
    def get_revision(self, revision_id):
172
 
        try:
173
 
            revf = self.revision_store[revision_id]
174
 
        except KeyError:
175
 
            raise NoSuchRevision(self, revision_id)
176
 
        r = serializer_v5.read_revision(revf)
177
 
        if r.revision_id != revision_id:
178
 
            raise BzrCheckError('revision stored as {%s} actually contains {%s}'
179
 
                                % (revision_id, r.revision_id))
180
 
        return r
181
 
 
182
 
 
183
 
class RemoteStore(object):
184
 
    def __init__(self, baseurl):
185
 
        self._baseurl = baseurl
186
 
        
187
 
 
188
 
    def _path(self, name):
189
 
        if '/' in name:
190
 
            raise ValueError('invalid store id', name)
191
 
        return self._baseurl + '/' + name
192
 
        
193
 
    def __getitem__(self, fileid):
194
 
        p = self._path(fileid)
195
 
        try:
196
 
            return get_url(p, compressed=True)
197
 
        except urllib2.URLError:
198
 
            pass
199
 
        try:
200
 
            return get_url(p, compressed=False)
201
 
        except urllib2.URLError:
202
 
            raise KeyError(fileid)
203
 
    
204
 
 
205