~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/remote.py

(jameinel) Allow 'bzr serve' to interpret SIGHUP as a graceful shutdown.
 (bug #795025) (John A Meinel)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006, 2007, 2008 Canonical Ltd
 
1
# Copyright (C) 2006-2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
# TODO: At some point, handle upgrades by just passing the whole request
18
 
# across to run on the server.
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19
16
 
20
17
import bz2
21
18
 
22
19
from bzrlib import (
 
20
    bencode,
23
21
    branch,
 
22
    bzrdir as _mod_bzrdir,
 
23
    config,
 
24
    controldir,
24
25
    debug,
25
26
    errors,
26
27
    graph,
 
28
    lock,
27
29
    lockdir,
28
 
    repository,
29
 
    revision,
 
30
    repository as _mod_repository,
 
31
    revision as _mod_revision,
 
32
    static_tuple,
30
33
    symbol_versioning,
31
34
    urlutils,
32
 
)
33
 
from bzrlib.branch import BranchReferenceFormat
34
 
from bzrlib.bzrdir import BzrDir, RemoteBzrDirFormat
35
 
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
35
    vf_repository,
 
36
    )
 
37
from bzrlib.branch import BranchReferenceFormat, BranchWriteLockResult
 
38
from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises
36
39
from bzrlib.errors import (
37
40
    NoSuchRevision,
38
41
    SmartProtocolError,
39
42
    )
 
43
from bzrlib.i18n import gettext
40
44
from bzrlib.lockable_files import LockableFiles
41
 
from bzrlib.smart import client, vfs
42
 
from bzrlib.revision import ensure_null, NULL_REVISION
 
45
from bzrlib.smart import client, vfs, repository as smart_repo
 
46
from bzrlib.smart.client import _SmartClient
 
47
from bzrlib.revision import NULL_REVISION
 
48
from bzrlib.repository import RepositoryWriteLockResult, _LazyListJoin
43
49
from bzrlib.trace import mutter, note, warning
44
50
 
45
51
 
 
52
_DEFAULT_SEARCH_DEPTH = 100
 
53
 
 
54
 
46
55
class _RpcHelper(object):
47
56
    """Mixin class that helps with issuing RPCs."""
48
57
 
51
60
            return self._client.call(method, *args)
52
61
        except errors.ErrorFromSmartServer, err:
53
62
            self._translate_error(err, **err_context)
54
 
        
 
63
 
55
64
    def _call_expecting_body(self, method, *args, **err_context):
56
65
        try:
57
66
            return self._client.call_expecting_body(method, *args)
58
67
        except errors.ErrorFromSmartServer, err:
59
68
            self._translate_error(err, **err_context)
60
 
        
 
69
 
 
70
    def _call_with_body_bytes(self, method, args, body_bytes, **err_context):
 
71
        try:
 
72
            return self._client.call_with_body_bytes(method, args, body_bytes)
 
73
        except errors.ErrorFromSmartServer, err:
 
74
            self._translate_error(err, **err_context)
 
75
 
61
76
    def _call_with_body_bytes_expecting_body(self, method, args, body_bytes,
62
77
                                             **err_context):
63
78
        try:
65
80
                method, args, body_bytes)
66
81
        except errors.ErrorFromSmartServer, err:
67
82
            self._translate_error(err, **err_context)
68
 
        
69
 
# Note: RemoteBzrDirFormat is in bzrdir.py
70
 
 
71
 
class RemoteBzrDir(BzrDir, _RpcHelper):
 
83
 
 
84
 
 
85
def response_tuple_to_repo_format(response):
 
86
    """Convert a response tuple describing a repository format to a format."""
 
87
    format = RemoteRepositoryFormat()
 
88
    format._rich_root_data = (response[0] == 'yes')
 
89
    format._supports_tree_reference = (response[1] == 'yes')
 
90
    format._supports_external_lookups = (response[2] == 'yes')
 
91
    format._network_name = response[3]
 
92
    return format
 
93
 
 
94
 
 
95
# Note that RemoteBzrDirProber lives in bzrlib.bzrdir so bzrlib.remote
 
96
# does not have to be imported unless a remote format is involved.
 
97
 
 
98
class RemoteBzrDirFormat(_mod_bzrdir.BzrDirMetaFormat1):
 
99
    """Format representing bzrdirs accessed via a smart server"""
 
100
 
 
101
    supports_workingtrees = False
 
102
 
 
103
    def __init__(self):
 
104
        _mod_bzrdir.BzrDirMetaFormat1.__init__(self)
 
105
        # XXX: It's a bit ugly that the network name is here, because we'd
 
106
        # like to believe that format objects are stateless or at least
 
107
        # immutable,  However, we do at least avoid mutating the name after
 
108
        # it's returned.  See <https://bugs.launchpad.net/bzr/+bug/504102>
 
109
        self._network_name = None
 
110
 
 
111
    def __repr__(self):
 
112
        return "%s(_network_name=%r)" % (self.__class__.__name__,
 
113
            self._network_name)
 
114
 
 
115
    def get_format_description(self):
 
116
        if self._network_name:
 
117
            real_format = controldir.network_format_registry.get(self._network_name)
 
118
            return 'Remote: ' + real_format.get_format_description()
 
119
        return 'bzr remote bzrdir'
 
120
 
 
121
    def get_format_string(self):
 
122
        raise NotImplementedError(self.get_format_string)
 
123
 
 
124
    def network_name(self):
 
125
        if self._network_name:
 
126
            return self._network_name
 
127
        else:
 
128
            raise AssertionError("No network name set.")
 
129
 
 
130
    def initialize_on_transport(self, transport):
 
131
        try:
 
132
            # hand off the request to the smart server
 
133
            client_medium = transport.get_smart_medium()
 
134
        except errors.NoSmartMedium:
 
135
            # TODO: lookup the local format from a server hint.
 
136
            local_dir_format = _mod_bzrdir.BzrDirMetaFormat1()
 
137
            return local_dir_format.initialize_on_transport(transport)
 
138
        client = _SmartClient(client_medium)
 
139
        path = client.remote_path_from_transport(transport)
 
140
        try:
 
141
            response = client.call('BzrDirFormat.initialize', path)
 
142
        except errors.ErrorFromSmartServer, err:
 
143
            _translate_error(err, path=path)
 
144
        if response[0] != 'ok':
 
145
            raise errors.SmartProtocolError('unexpected response code %s' % (response,))
 
146
        format = RemoteBzrDirFormat()
 
147
        self._supply_sub_formats_to(format)
 
148
        return RemoteBzrDir(transport, format)
 
149
 
 
150
    def parse_NoneTrueFalse(self, arg):
 
151
        if not arg:
 
152
            return None
 
153
        if arg == 'False':
 
154
            return False
 
155
        if arg == 'True':
 
156
            return True
 
157
        raise AssertionError("invalid arg %r" % arg)
 
158
 
 
159
    def _serialize_NoneTrueFalse(self, arg):
 
160
        if arg is False:
 
161
            return 'False'
 
162
        if arg:
 
163
            return 'True'
 
164
        return ''
 
165
 
 
166
    def _serialize_NoneString(self, arg):
 
167
        return arg or ''
 
168
 
 
169
    def initialize_on_transport_ex(self, transport, use_existing_dir=False,
 
170
        create_prefix=False, force_new_repo=False, stacked_on=None,
 
171
        stack_on_pwd=None, repo_format_name=None, make_working_trees=None,
 
172
        shared_repo=False):
 
173
        try:
 
174
            # hand off the request to the smart server
 
175
            client_medium = transport.get_smart_medium()
 
176
        except errors.NoSmartMedium:
 
177
            do_vfs = True
 
178
        else:
 
179
            # Decline to open it if the server doesn't support our required
 
180
            # version (3) so that the VFS-based transport will do it.
 
181
            if client_medium.should_probe():
 
182
                try:
 
183
                    server_version = client_medium.protocol_version()
 
184
                    if server_version != '2':
 
185
                        do_vfs = True
 
186
                    else:
 
187
                        do_vfs = False
 
188
                except errors.SmartProtocolError:
 
189
                    # Apparently there's no usable smart server there, even though
 
190
                    # the medium supports the smart protocol.
 
191
                    do_vfs = True
 
192
            else:
 
193
                do_vfs = False
 
194
        if not do_vfs:
 
195
            client = _SmartClient(client_medium)
 
196
            path = client.remote_path_from_transport(transport)
 
197
            if client_medium._is_remote_before((1, 16)):
 
198
                do_vfs = True
 
199
        if do_vfs:
 
200
            # TODO: lookup the local format from a server hint.
 
201
            local_dir_format = _mod_bzrdir.BzrDirMetaFormat1()
 
202
            self._supply_sub_formats_to(local_dir_format)
 
203
            return local_dir_format.initialize_on_transport_ex(transport,
 
204
                use_existing_dir=use_existing_dir, create_prefix=create_prefix,
 
205
                force_new_repo=force_new_repo, stacked_on=stacked_on,
 
206
                stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
 
207
                make_working_trees=make_working_trees, shared_repo=shared_repo,
 
208
                vfs_only=True)
 
209
        return self._initialize_on_transport_ex_rpc(client, path, transport,
 
210
            use_existing_dir, create_prefix, force_new_repo, stacked_on,
 
211
            stack_on_pwd, repo_format_name, make_working_trees, shared_repo)
 
212
 
 
213
    def _initialize_on_transport_ex_rpc(self, client, path, transport,
 
214
        use_existing_dir, create_prefix, force_new_repo, stacked_on,
 
215
        stack_on_pwd, repo_format_name, make_working_trees, shared_repo):
 
216
        args = []
 
217
        args.append(self._serialize_NoneTrueFalse(use_existing_dir))
 
218
        args.append(self._serialize_NoneTrueFalse(create_prefix))
 
219
        args.append(self._serialize_NoneTrueFalse(force_new_repo))
 
220
        args.append(self._serialize_NoneString(stacked_on))
 
221
        # stack_on_pwd is often/usually our transport
 
222
        if stack_on_pwd:
 
223
            try:
 
224
                stack_on_pwd = transport.relpath(stack_on_pwd)
 
225
                if not stack_on_pwd:
 
226
                    stack_on_pwd = '.'
 
227
            except errors.PathNotChild:
 
228
                pass
 
229
        args.append(self._serialize_NoneString(stack_on_pwd))
 
230
        args.append(self._serialize_NoneString(repo_format_name))
 
231
        args.append(self._serialize_NoneTrueFalse(make_working_trees))
 
232
        args.append(self._serialize_NoneTrueFalse(shared_repo))
 
233
        request_network_name = self._network_name or \
 
234
            _mod_bzrdir.BzrDirFormat.get_default_format().network_name()
 
235
        try:
 
236
            response = client.call('BzrDirFormat.initialize_ex_1.16',
 
237
                request_network_name, path, *args)
 
238
        except errors.UnknownSmartMethod:
 
239
            client._medium._remember_remote_is_before((1,16))
 
240
            local_dir_format = _mod_bzrdir.BzrDirMetaFormat1()
 
241
            self._supply_sub_formats_to(local_dir_format)
 
242
            return local_dir_format.initialize_on_transport_ex(transport,
 
243
                use_existing_dir=use_existing_dir, create_prefix=create_prefix,
 
244
                force_new_repo=force_new_repo, stacked_on=stacked_on,
 
245
                stack_on_pwd=stack_on_pwd, repo_format_name=repo_format_name,
 
246
                make_working_trees=make_working_trees, shared_repo=shared_repo,
 
247
                vfs_only=True)
 
248
        except errors.ErrorFromSmartServer, err:
 
249
            _translate_error(err, path=path)
 
250
        repo_path = response[0]
 
251
        bzrdir_name = response[6]
 
252
        require_stacking = response[7]
 
253
        require_stacking = self.parse_NoneTrueFalse(require_stacking)
 
254
        format = RemoteBzrDirFormat()
 
255
        format._network_name = bzrdir_name
 
256
        self._supply_sub_formats_to(format)
 
257
        bzrdir = RemoteBzrDir(transport, format, _client=client)
 
258
        if repo_path:
 
259
            repo_format = response_tuple_to_repo_format(response[1:])
 
260
            if repo_path == '.':
 
261
                repo_path = ''
 
262
            if repo_path:
 
263
                repo_bzrdir_format = RemoteBzrDirFormat()
 
264
                repo_bzrdir_format._network_name = response[5]
 
265
                repo_bzr = RemoteBzrDir(transport.clone(repo_path),
 
266
                    repo_bzrdir_format)
 
267
            else:
 
268
                repo_bzr = bzrdir
 
269
            final_stack = response[8] or None
 
270
            final_stack_pwd = response[9] or None
 
271
            if final_stack_pwd:
 
272
                final_stack_pwd = urlutils.join(
 
273
                    transport.base, final_stack_pwd)
 
274
            remote_repo = RemoteRepository(repo_bzr, repo_format)
 
275
            if len(response) > 10:
 
276
                # Updated server verb that locks remotely.
 
277
                repo_lock_token = response[10] or None
 
278
                remote_repo.lock_write(repo_lock_token, _skip_rpc=True)
 
279
                if repo_lock_token:
 
280
                    remote_repo.dont_leave_lock_in_place()
 
281
            else:
 
282
                remote_repo.lock_write()
 
283
            policy = _mod_bzrdir.UseExistingRepository(remote_repo, final_stack,
 
284
                final_stack_pwd, require_stacking)
 
285
            policy.acquire_repository()
 
286
        else:
 
287
            remote_repo = None
 
288
            policy = None
 
289
        bzrdir._format.set_branch_format(self.get_branch_format())
 
290
        if require_stacking:
 
291
            # The repo has already been created, but we need to make sure that
 
292
            # we'll make a stackable branch.
 
293
            bzrdir._format.require_stacking(_skip_repo=True)
 
294
        return remote_repo, bzrdir, require_stacking, policy
 
295
 
 
296
    def _open(self, transport):
 
297
        return RemoteBzrDir(transport, self)
 
298
 
 
299
    def __eq__(self, other):
 
300
        if not isinstance(other, RemoteBzrDirFormat):
 
301
            return False
 
302
        return self.get_format_description() == other.get_format_description()
 
303
 
 
304
    def __return_repository_format(self):
 
305
        # Always return a RemoteRepositoryFormat object, but if a specific bzr
 
306
        # repository format has been asked for, tell the RemoteRepositoryFormat
 
307
        # that it should use that for init() etc.
 
308
        result = RemoteRepositoryFormat()
 
309
        custom_format = getattr(self, '_repository_format', None)
 
310
        if custom_format:
 
311
            if isinstance(custom_format, RemoteRepositoryFormat):
 
312
                return custom_format
 
313
            else:
 
314
                # We will use the custom format to create repositories over the
 
315
                # wire; expose its details like rich_root_data for code to
 
316
                # query
 
317
                result._custom_format = custom_format
 
318
        return result
 
319
 
 
320
    def get_branch_format(self):
 
321
        result = _mod_bzrdir.BzrDirMetaFormat1.get_branch_format(self)
 
322
        if not isinstance(result, RemoteBranchFormat):
 
323
            new_result = RemoteBranchFormat()
 
324
            new_result._custom_format = result
 
325
            # cache the result
 
326
            self.set_branch_format(new_result)
 
327
            result = new_result
 
328
        return result
 
329
 
 
330
    repository_format = property(__return_repository_format,
 
331
        _mod_bzrdir.BzrDirMetaFormat1._set_repository_format) #.im_func)
 
332
 
 
333
 
 
334
class RemoteBzrDir(_mod_bzrdir.BzrDir, _RpcHelper):
72
335
    """Control directory on a remote server, accessed via bzr:// or similar."""
73
336
 
74
 
    def __init__(self, transport, _client=None):
 
337
    def __init__(self, transport, format, _client=None, _force_probe=False):
75
338
        """Construct a RemoteBzrDir.
76
339
 
77
340
        :param _client: Private parameter for testing. Disables probing and the
78
341
            use of a real bzrdir.
79
342
        """
80
 
        BzrDir.__init__(self, transport, RemoteBzrDirFormat())
 
343
        _mod_bzrdir.BzrDir.__init__(self, transport, format)
81
344
        # this object holds a delegated bzrdir that uses file-level operations
82
345
        # to talk to the other side
83
346
        self._real_bzrdir = None
 
347
        self._has_working_tree = None
 
348
        # 1-shot cache for the call pattern 'create_branch; open_branch' - see
 
349
        # create_branch for details.
 
350
        self._next_open_branch_result = None
84
351
 
85
352
        if _client is None:
86
353
            medium = transport.get_smart_medium()
87
354
            self._client = client._SmartClient(medium)
88
355
        else:
89
356
            self._client = _client
90
 
            return
91
 
 
 
357
            if not _force_probe:
 
358
                return
 
359
 
 
360
        self._probe_bzrdir()
 
361
 
 
362
    def __repr__(self):
 
363
        return '%s(%r)' % (self.__class__.__name__, self._client)
 
364
 
 
365
    def _probe_bzrdir(self):
 
366
        medium = self._client._medium
92
367
        path = self._path_for_remote_call(self._client)
 
368
        if medium._is_remote_before((2, 1)):
 
369
            self._rpc_open(path)
 
370
            return
 
371
        try:
 
372
            self._rpc_open_2_1(path)
 
373
            return
 
374
        except errors.UnknownSmartMethod:
 
375
            medium._remember_remote_is_before((2, 1))
 
376
            self._rpc_open(path)
 
377
 
 
378
    def _rpc_open_2_1(self, path):
 
379
        response = self._call('BzrDir.open_2.1', path)
 
380
        if response == ('no',):
 
381
            raise errors.NotBranchError(path=self.root_transport.base)
 
382
        elif response[0] == 'yes':
 
383
            if response[1] == 'yes':
 
384
                self._has_working_tree = True
 
385
            elif response[1] == 'no':
 
386
                self._has_working_tree = False
 
387
            else:
 
388
                raise errors.UnexpectedSmartServerResponse(response)
 
389
        else:
 
390
            raise errors.UnexpectedSmartServerResponse(response)
 
391
 
 
392
    def _rpc_open(self, path):
93
393
        response = self._call('BzrDir.open', path)
94
394
        if response not in [('yes',), ('no',)]:
95
395
            raise errors.UnexpectedSmartServerResponse(response)
96
396
        if response == ('no',):
97
 
            raise errors.NotBranchError(path=transport.base)
 
397
            raise errors.NotBranchError(path=self.root_transport.base)
98
398
 
99
399
    def _ensure_real(self):
100
400
        """Ensure that there is a _real_bzrdir set.
102
402
        Used before calls to self._real_bzrdir.
103
403
        """
104
404
        if not self._real_bzrdir:
105
 
            self._real_bzrdir = BzrDir.open_from_transport(
 
405
            if 'hpssvfs' in debug.debug_flags:
 
406
                import traceback
 
407
                warning('VFS BzrDir access triggered\n%s',
 
408
                    ''.join(traceback.format_stack()))
 
409
            self._real_bzrdir = _mod_bzrdir.BzrDir.open_from_transport(
106
410
                self.root_transport, _server_formats=False)
 
411
            self._format._network_name = \
 
412
                self._real_bzrdir._format.network_name()
107
413
 
108
414
    def _translate_error(self, err, **context):
109
415
        _translate_error(err, bzrdir=self, **context)
110
416
 
111
 
    def cloning_metadir(self, stacked=False):
 
417
    def break_lock(self):
 
418
        # Prevent aliasing problems in the next_open_branch_result cache.
 
419
        # See create_branch for rationale.
 
420
        self._next_open_branch_result = None
 
421
        return _mod_bzrdir.BzrDir.break_lock(self)
 
422
 
 
423
    def _vfs_cloning_metadir(self, require_stacking=False):
112
424
        self._ensure_real()
113
 
        return self._real_bzrdir.cloning_metadir(stacked)
 
425
        return self._real_bzrdir.cloning_metadir(
 
426
            require_stacking=require_stacking)
 
427
 
 
428
    def cloning_metadir(self, require_stacking=False):
 
429
        medium = self._client._medium
 
430
        if medium._is_remote_before((1, 13)):
 
431
            return self._vfs_cloning_metadir(require_stacking=require_stacking)
 
432
        verb = 'BzrDir.cloning_metadir'
 
433
        if require_stacking:
 
434
            stacking = 'True'
 
435
        else:
 
436
            stacking = 'False'
 
437
        path = self._path_for_remote_call(self._client)
 
438
        try:
 
439
            response = self._call(verb, path, stacking)
 
440
        except errors.UnknownSmartMethod:
 
441
            medium._remember_remote_is_before((1, 13))
 
442
            return self._vfs_cloning_metadir(require_stacking=require_stacking)
 
443
        except errors.UnknownErrorFromSmartServer, err:
 
444
            if err.error_tuple != ('BranchReference',):
 
445
                raise
 
446
            # We need to resolve the branch reference to determine the
 
447
            # cloning_metadir.  This causes unnecessary RPCs to open the
 
448
            # referenced branch (and bzrdir, etc) but only when the caller
 
449
            # didn't already resolve the branch reference.
 
450
            referenced_branch = self.open_branch()
 
451
            return referenced_branch.bzrdir.cloning_metadir()
 
452
        if len(response) != 3:
 
453
            raise errors.UnexpectedSmartServerResponse(response)
 
454
        control_name, repo_name, branch_info = response
 
455
        if len(branch_info) != 2:
 
456
            raise errors.UnexpectedSmartServerResponse(response)
 
457
        branch_ref, branch_name = branch_info
 
458
        format = controldir.network_format_registry.get(control_name)
 
459
        if repo_name:
 
460
            format.repository_format = _mod_repository.network_format_registry.get(
 
461
                repo_name)
 
462
        if branch_ref == 'ref':
 
463
            # XXX: we need possible_transports here to avoid reopening the
 
464
            # connection to the referenced location
 
465
            ref_bzrdir = _mod_bzrdir.BzrDir.open(branch_name)
 
466
            branch_format = ref_bzrdir.cloning_metadir().get_branch_format()
 
467
            format.set_branch_format(branch_format)
 
468
        elif branch_ref == 'branch':
 
469
            if branch_name:
 
470
                format.set_branch_format(
 
471
                    branch.network_format_registry.get(branch_name))
 
472
        else:
 
473
            raise errors.UnexpectedSmartServerResponse(response)
 
474
        return format
114
475
 
115
476
    def create_repository(self, shared=False):
116
 
        self._ensure_real()
117
 
        self._real_bzrdir.create_repository(shared=shared)
118
 
        return self.open_repository()
 
477
        # as per meta1 formats - just delegate to the format object which may
 
478
        # be parameterised.
 
479
        result = self._format.repository_format.initialize(self, shared)
 
480
        if not isinstance(result, RemoteRepository):
 
481
            return self.open_repository()
 
482
        else:
 
483
            return result
119
484
 
120
485
    def destroy_repository(self):
121
486
        """See BzrDir.destroy_repository"""
122
487
        self._ensure_real()
123
488
        self._real_bzrdir.destroy_repository()
124
489
 
125
 
    def create_branch(self):
126
 
        self._ensure_real()
127
 
        real_branch = self._real_bzrdir.create_branch()
128
 
        return RemoteBranch(self, self.find_repository(), real_branch)
 
490
    def create_branch(self, name=None, repository=None,
 
491
                      append_revisions_only=None):
 
492
        # as per meta1 formats - just delegate to the format object which may
 
493
        # be parameterised.
 
494
        real_branch = self._format.get_branch_format().initialize(self,
 
495
            name=name, repository=repository,
 
496
            append_revisions_only=append_revisions_only)
 
497
        if not isinstance(real_branch, RemoteBranch):
 
498
            if not isinstance(repository, RemoteRepository):
 
499
                raise AssertionError(
 
500
                    'need a RemoteRepository to use with RemoteBranch, got %r'
 
501
                    % (repository,))
 
502
            result = RemoteBranch(self, repository, real_branch, name=name)
 
503
        else:
 
504
            result = real_branch
 
505
        # BzrDir.clone_on_transport() uses the result of create_branch but does
 
506
        # not return it to its callers; we save approximately 8% of our round
 
507
        # trips by handing the branch we created back to the first caller to
 
508
        # open_branch rather than probing anew. Long term we need a API in
 
509
        # bzrdir that doesn't discard result objects (like result_branch).
 
510
        # RBC 20090225
 
511
        self._next_open_branch_result = result
 
512
        return result
129
513
 
130
 
    def destroy_branch(self):
 
514
    def destroy_branch(self, name=None):
131
515
        """See BzrDir.destroy_branch"""
132
516
        self._ensure_real()
133
 
        self._real_bzrdir.destroy_branch()
 
517
        self._real_bzrdir.destroy_branch(name=name)
 
518
        self._next_open_branch_result = None
134
519
 
135
 
    def create_workingtree(self, revision_id=None, from_branch=None):
 
520
    def create_workingtree(self, revision_id=None, from_branch=None,
 
521
        accelerator_tree=None, hardlink=False):
136
522
        raise errors.NotLocalUrl(self.transport.base)
137
523
 
138
 
    def find_branch_format(self):
 
524
    def find_branch_format(self, name=None):
139
525
        """Find the branch 'format' for this bzrdir.
140
526
 
141
527
        This might be a synthetic object for e.g. RemoteBranch and SVN.
142
528
        """
143
 
        b = self.open_branch()
 
529
        b = self.open_branch(name=name)
144
530
        return b._format
145
531
 
146
 
    def get_branch_reference(self):
 
532
    def get_branch_reference(self, name=None):
147
533
        """See BzrDir.get_branch_reference()."""
 
534
        if name is not None:
 
535
            # XXX JRV20100304: Support opening colocated branches
 
536
            raise errors.NoColocatedBranchSupport(self)
 
537
        response = self._get_branch_reference()
 
538
        if response[0] == 'ref':
 
539
            return response[1]
 
540
        else:
 
541
            return None
 
542
 
 
543
    def _get_branch_reference(self):
148
544
        path = self._path_for_remote_call(self._client)
149
 
        response = self._call('BzrDir.open_branch', path)
150
 
        if response[0] == 'ok':
151
 
            if response[1] == '':
152
 
                # branch at this location.
153
 
                return None
154
 
            else:
155
 
                # a branch reference, use the existing BranchReference logic.
156
 
                return response[1]
157
 
        else:
 
545
        medium = self._client._medium
 
546
        candidate_calls = [
 
547
            ('BzrDir.open_branchV3', (2, 1)),
 
548
            ('BzrDir.open_branchV2', (1, 13)),
 
549
            ('BzrDir.open_branch', None),
 
550
            ]
 
551
        for verb, required_version in candidate_calls:
 
552
            if required_version and medium._is_remote_before(required_version):
 
553
                continue
 
554
            try:
 
555
                response = self._call(verb, path)
 
556
            except errors.UnknownSmartMethod:
 
557
                if required_version is None:
 
558
                    raise
 
559
                medium._remember_remote_is_before(required_version)
 
560
            else:
 
561
                break
 
562
        if verb == 'BzrDir.open_branch':
 
563
            if response[0] != 'ok':
 
564
                raise errors.UnexpectedSmartServerResponse(response)
 
565
            if response[1] != '':
 
566
                return ('ref', response[1])
 
567
            else:
 
568
                return ('branch', '')
 
569
        if response[0] not in ('ref', 'branch'):
158
570
            raise errors.UnexpectedSmartServerResponse(response)
 
571
        return response
159
572
 
160
 
    def _get_tree_branch(self):
 
573
    def _get_tree_branch(self, name=None):
161
574
        """See BzrDir._get_tree_branch()."""
162
 
        return None, self.open_branch()
 
575
        return None, self.open_branch(name=name)
163
576
 
164
 
    def open_branch(self, _unsupported=False):
165
 
        if _unsupported:
 
577
    def open_branch(self, name=None, unsupported=False,
 
578
                    ignore_fallbacks=False):
 
579
        if unsupported:
166
580
            raise NotImplementedError('unsupported flag support not implemented yet.')
167
 
        reference_url = self.get_branch_reference()
168
 
        if reference_url is None:
169
 
            # branch at this location.
170
 
            return RemoteBranch(self, self.find_repository())
171
 
        else:
 
581
        if self._next_open_branch_result is not None:
 
582
            # See create_branch for details.
 
583
            result = self._next_open_branch_result
 
584
            self._next_open_branch_result = None
 
585
            return result
 
586
        response = self._get_branch_reference()
 
587
        if response[0] == 'ref':
172
588
            # a branch reference, use the existing BranchReference logic.
173
589
            format = BranchReferenceFormat()
174
 
            return format.open(self, _found=True, location=reference_url)
175
 
                
 
590
            return format.open(self, name=name, _found=True,
 
591
                location=response[1], ignore_fallbacks=ignore_fallbacks)
 
592
        branch_format_name = response[1]
 
593
        if not branch_format_name:
 
594
            branch_format_name = None
 
595
        format = RemoteBranchFormat(network_name=branch_format_name)
 
596
        return RemoteBranch(self, self.find_repository(), format=format,
 
597
            setup_stacking=not ignore_fallbacks, name=name)
 
598
 
 
599
    def _open_repo_v1(self, path):
 
600
        verb = 'BzrDir.find_repository'
 
601
        response = self._call(verb, path)
 
602
        if response[0] != 'ok':
 
603
            raise errors.UnexpectedSmartServerResponse(response)
 
604
        # servers that only support the v1 method don't support external
 
605
        # references either.
 
606
        self._ensure_real()
 
607
        repo = self._real_bzrdir.open_repository()
 
608
        response = response + ('no', repo._format.network_name())
 
609
        return response, repo
 
610
 
 
611
    def _open_repo_v2(self, path):
 
612
        verb = 'BzrDir.find_repositoryV2'
 
613
        response = self._call(verb, path)
 
614
        if response[0] != 'ok':
 
615
            raise errors.UnexpectedSmartServerResponse(response)
 
616
        self._ensure_real()
 
617
        repo = self._real_bzrdir.open_repository()
 
618
        response = response + (repo._format.network_name(),)
 
619
        return response, repo
 
620
 
 
621
    def _open_repo_v3(self, path):
 
622
        verb = 'BzrDir.find_repositoryV3'
 
623
        medium = self._client._medium
 
624
        if medium._is_remote_before((1, 13)):
 
625
            raise errors.UnknownSmartMethod(verb)
 
626
        try:
 
627
            response = self._call(verb, path)
 
628
        except errors.UnknownSmartMethod:
 
629
            medium._remember_remote_is_before((1, 13))
 
630
            raise
 
631
        if response[0] != 'ok':
 
632
            raise errors.UnexpectedSmartServerResponse(response)
 
633
        return response, None
 
634
 
176
635
    def open_repository(self):
177
636
        path = self._path_for_remote_call(self._client)
178
 
        verb = 'BzrDir.find_repositoryV2'
179
 
        try:
180
 
            response = self._call(verb, path)
181
 
        except errors.UnknownSmartMethod:
182
 
            verb = 'BzrDir.find_repository'
183
 
            response = self._call(verb, path)
 
637
        response = None
 
638
        for probe in [self._open_repo_v3, self._open_repo_v2,
 
639
            self._open_repo_v1]:
 
640
            try:
 
641
                response, real_repo = probe(path)
 
642
                break
 
643
            except errors.UnknownSmartMethod:
 
644
                pass
 
645
        if response is None:
 
646
            raise errors.UnknownSmartMethod('BzrDir.find_repository{3,2,}')
184
647
        if response[0] != 'ok':
185
648
            raise errors.UnexpectedSmartServerResponse(response)
186
 
        if verb == 'BzrDir.find_repository':
187
 
            # servers that don't support the V2 method don't support external
188
 
            # references either.
189
 
            response = response + ('no', )
190
 
        if not (len(response) == 5):
 
649
        if len(response) != 6:
191
650
            raise SmartProtocolError('incorrect response length %s' % (response,))
192
651
        if response[1] == '':
193
 
            format = RemoteRepositoryFormat()
194
 
            format.rich_root_data = (response[2] == 'yes')
195
 
            format.supports_tree_reference = (response[3] == 'yes')
196
 
            # No wire format to check this yet.
197
 
            format.supports_external_lookups = (response[4] == 'yes')
 
652
            # repo is at this dir.
 
653
            format = response_tuple_to_repo_format(response[2:])
198
654
            # Used to support creating a real format instance when needed.
199
655
            format._creating_bzrdir = self
200
 
            return RemoteRepository(self, format)
 
656
            remote_repo = RemoteRepository(self, format)
 
657
            format._creating_repo = remote_repo
 
658
            if real_repo is not None:
 
659
                remote_repo._set_real_repository(real_repo)
 
660
            return remote_repo
201
661
        else:
202
662
            raise errors.NoRepositoryPresent(self)
203
663
 
 
664
    def has_workingtree(self):
 
665
        if self._has_working_tree is None:
 
666
            self._ensure_real()
 
667
            self._has_working_tree = self._real_bzrdir.has_workingtree()
 
668
        return self._has_working_tree
 
669
 
204
670
    def open_workingtree(self, recommend_upgrade=True):
205
 
        self._ensure_real()
206
 
        if self._real_bzrdir.has_workingtree():
 
671
        if self.has_workingtree():
207
672
            raise errors.NotLocalUrl(self.root_transport)
208
673
        else:
209
674
            raise errors.NoWorkingTree(self.root_transport.base)
210
675
 
211
676
    def _path_for_remote_call(self, client):
212
677
        """Return the path to be used for this bzrdir in a remote call."""
213
 
        return client.remote_path_from_transport(self.root_transport)
 
678
        return urlutils.split_segment_parameters_raw(
 
679
            client.remote_path_from_transport(self.root_transport))[0]
214
680
 
215
 
    def get_branch_transport(self, branch_format):
 
681
    def get_branch_transport(self, branch_format, name=None):
216
682
        self._ensure_real()
217
 
        return self._real_bzrdir.get_branch_transport(branch_format)
 
683
        return self._real_bzrdir.get_branch_transport(branch_format, name=name)
218
684
 
219
685
    def get_repository_transport(self, repository_format):
220
686
        self._ensure_real()
228
694
        """Upgrading of remote bzrdirs is not supported yet."""
229
695
        return False
230
696
 
231
 
    def needs_format_conversion(self, format=None):
 
697
    def needs_format_conversion(self, format):
232
698
        """Upgrading of remote bzrdirs is not supported yet."""
233
 
        if format is None:
234
 
            symbol_versioning.warn(symbol_versioning.deprecated_in((1, 13, 0))
235
 
                % 'needs_format_conversion(format=None)')
236
699
        return False
237
700
 
238
701
    def clone(self, url, revision_id=None, force_new_repo=False,
241
704
        return self._real_bzrdir.clone(url, revision_id=revision_id,
242
705
            force_new_repo=force_new_repo, preserve_stacking=preserve_stacking)
243
706
 
244
 
    def get_config(self):
245
 
        self._ensure_real()
246
 
        return self._real_bzrdir.get_config()
247
 
 
248
 
 
249
 
class RemoteRepositoryFormat(repository.RepositoryFormat):
 
707
    def _get_config(self):
 
708
        return RemoteBzrDirConfig(self)
 
709
 
 
710
 
 
711
class RemoteRepositoryFormat(vf_repository.VersionedFileRepositoryFormat):
250
712
    """Format for repositories accessed over a _SmartClient.
251
713
 
252
714
    Instances of this repository are represented by RemoteRepository
257
719
    the attributes rich_root_data and supports_tree_reference are set
258
720
    on a per instance basis, and are not set (and should not be) at
259
721
    the class level.
 
722
 
 
723
    :ivar _custom_format: If set, a specific concrete repository format that
 
724
        will be used when initializing a repository with this
 
725
        RemoteRepositoryFormat.
 
726
    :ivar _creating_repo: If set, the repository object that this
 
727
        RemoteRepositoryFormat was created for: it can be called into
 
728
        to obtain data like the network name.
260
729
    """
261
730
 
262
731
    _matchingbzrdir = RemoteBzrDirFormat()
263
 
 
264
 
    def initialize(self, a_bzrdir, shared=False):
265
 
        if not isinstance(a_bzrdir, RemoteBzrDir):
 
732
    supports_full_versioned_files = True
 
733
    supports_leaving_lock = True
 
734
 
 
735
    def __init__(self):
 
736
        _mod_repository.RepositoryFormat.__init__(self)
 
737
        self._custom_format = None
 
738
        self._network_name = None
 
739
        self._creating_bzrdir = None
 
740
        self._revision_graph_can_have_wrong_parents = None
 
741
        self._supports_chks = None
 
742
        self._supports_external_lookups = None
 
743
        self._supports_tree_reference = None
 
744
        self._supports_funky_characters = None
 
745
        self._supports_nesting_repositories = None
 
746
        self._rich_root_data = None
 
747
 
 
748
    def __repr__(self):
 
749
        return "%s(_network_name=%r)" % (self.__class__.__name__,
 
750
            self._network_name)
 
751
 
 
752
    @property
 
753
    def fast_deltas(self):
 
754
        self._ensure_real()
 
755
        return self._custom_format.fast_deltas
 
756
 
 
757
    @property
 
758
    def rich_root_data(self):
 
759
        if self._rich_root_data is None:
 
760
            self._ensure_real()
 
761
            self._rich_root_data = self._custom_format.rich_root_data
 
762
        return self._rich_root_data
 
763
 
 
764
    @property
 
765
    def supports_chks(self):
 
766
        if self._supports_chks is None:
 
767
            self._ensure_real()
 
768
            self._supports_chks = self._custom_format.supports_chks
 
769
        return self._supports_chks
 
770
 
 
771
    @property
 
772
    def supports_external_lookups(self):
 
773
        if self._supports_external_lookups is None:
 
774
            self._ensure_real()
 
775
            self._supports_external_lookups = \
 
776
                self._custom_format.supports_external_lookups
 
777
        return self._supports_external_lookups
 
778
 
 
779
    @property
 
780
    def supports_funky_characters(self):
 
781
        if self._supports_funky_characters is None:
 
782
            self._ensure_real()
 
783
            self._supports_funky_characters = \
 
784
                self._custom_format.supports_funky_characters
 
785
        return self._supports_funky_characters
 
786
 
 
787
    @property
 
788
    def supports_nesting_repositories(self):
 
789
        if self._supports_nesting_repositories is None:
 
790
            self._ensure_real()
 
791
            self._supports_nesting_repositories = \
 
792
                self._custom_format.supports_nesting_repositories
 
793
        return self._supports_nesting_repositories
 
794
 
 
795
    @property
 
796
    def supports_tree_reference(self):
 
797
        if self._supports_tree_reference is None:
 
798
            self._ensure_real()
 
799
            self._supports_tree_reference = \
 
800
                self._custom_format.supports_tree_reference
 
801
        return self._supports_tree_reference
 
802
 
 
803
    @property
 
804
    def revision_graph_can_have_wrong_parents(self):
 
805
        if self._revision_graph_can_have_wrong_parents is None:
 
806
            self._ensure_real()
 
807
            self._revision_graph_can_have_wrong_parents = \
 
808
                self._custom_format.revision_graph_can_have_wrong_parents
 
809
        return self._revision_graph_can_have_wrong_parents
 
810
 
 
811
    def _vfs_initialize(self, a_bzrdir, shared):
 
812
        """Helper for common code in initialize."""
 
813
        if self._custom_format:
 
814
            # Custom format requested
 
815
            result = self._custom_format.initialize(a_bzrdir, shared=shared)
 
816
        elif self._creating_bzrdir is not None:
 
817
            # Use the format that the repository we were created to back
 
818
            # has.
266
819
            prior_repo = self._creating_bzrdir.open_repository()
267
820
            prior_repo._ensure_real()
268
 
            return prior_repo._real_repository._format.initialize(
 
821
            result = prior_repo._real_repository._format.initialize(
269
822
                a_bzrdir, shared=shared)
270
 
        return a_bzrdir.create_repository(shared=shared)
271
 
    
 
823
        else:
 
824
            # assume that a_bzr is a RemoteBzrDir but the smart server didn't
 
825
            # support remote initialization.
 
826
            # We delegate to a real object at this point (as RemoteBzrDir
 
827
            # delegate to the repository format which would lead to infinite
 
828
            # recursion if we just called a_bzrdir.create_repository.
 
829
            a_bzrdir._ensure_real()
 
830
            result = a_bzrdir._real_bzrdir.create_repository(shared=shared)
 
831
        if not isinstance(result, RemoteRepository):
 
832
            return self.open(a_bzrdir)
 
833
        else:
 
834
            return result
 
835
 
 
836
    def initialize(self, a_bzrdir, shared=False):
 
837
        # Being asked to create on a non RemoteBzrDir:
 
838
        if not isinstance(a_bzrdir, RemoteBzrDir):
 
839
            return self._vfs_initialize(a_bzrdir, shared)
 
840
        medium = a_bzrdir._client._medium
 
841
        if medium._is_remote_before((1, 13)):
 
842
            return self._vfs_initialize(a_bzrdir, shared)
 
843
        # Creating on a remote bzr dir.
 
844
        # 1) get the network name to use.
 
845
        if self._custom_format:
 
846
            network_name = self._custom_format.network_name()
 
847
        elif self._network_name:
 
848
            network_name = self._network_name
 
849
        else:
 
850
            # Select the current bzrlib default and ask for that.
 
851
            reference_bzrdir_format = _mod_bzrdir.format_registry.get('default')()
 
852
            reference_format = reference_bzrdir_format.repository_format
 
853
            network_name = reference_format.network_name()
 
854
        # 2) try direct creation via RPC
 
855
        path = a_bzrdir._path_for_remote_call(a_bzrdir._client)
 
856
        verb = 'BzrDir.create_repository'
 
857
        if shared:
 
858
            shared_str = 'True'
 
859
        else:
 
860
            shared_str = 'False'
 
861
        try:
 
862
            response = a_bzrdir._call(verb, path, network_name, shared_str)
 
863
        except errors.UnknownSmartMethod:
 
864
            # Fallback - use vfs methods
 
865
            medium._remember_remote_is_before((1, 13))
 
866
            return self._vfs_initialize(a_bzrdir, shared)
 
867
        else:
 
868
            # Turn the response into a RemoteRepository object.
 
869
            format = response_tuple_to_repo_format(response[1:])
 
870
            # Used to support creating a real format instance when needed.
 
871
            format._creating_bzrdir = a_bzrdir
 
872
            remote_repo = RemoteRepository(a_bzrdir, format)
 
873
            format._creating_repo = remote_repo
 
874
            return remote_repo
 
875
 
272
876
    def open(self, a_bzrdir):
273
877
        if not isinstance(a_bzrdir, RemoteBzrDir):
274
878
            raise AssertionError('%r is not a RemoteBzrDir' % (a_bzrdir,))
275
879
        return a_bzrdir.open_repository()
276
880
 
 
881
    def _ensure_real(self):
 
882
        if self._custom_format is None:
 
883
            self._custom_format = _mod_repository.network_format_registry.get(
 
884
                self._network_name)
 
885
 
 
886
    @property
 
887
    def _fetch_order(self):
 
888
        self._ensure_real()
 
889
        return self._custom_format._fetch_order
 
890
 
 
891
    @property
 
892
    def _fetch_uses_deltas(self):
 
893
        self._ensure_real()
 
894
        return self._custom_format._fetch_uses_deltas
 
895
 
 
896
    @property
 
897
    def _fetch_reconcile(self):
 
898
        self._ensure_real()
 
899
        return self._custom_format._fetch_reconcile
 
900
 
277
901
    def get_format_description(self):
278
 
        return 'bzr remote repository'
 
902
        self._ensure_real()
 
903
        return 'Remote: ' + self._custom_format.get_format_description()
279
904
 
280
905
    def __eq__(self, other):
281
 
        return self.__class__ == other.__class__
282
 
 
283
 
    def check_conversion_target(self, target_format):
284
 
        if self.rich_root_data and not target_format.rich_root_data:
285
 
            raise errors.BadConversionTarget(
286
 
                'Does not support rich root data.', target_format)
287
 
        if (self.supports_tree_reference and
288
 
            not getattr(target_format, 'supports_tree_reference', False)):
289
 
            raise errors.BadConversionTarget(
290
 
                'Does not support nested trees', target_format)
291
 
 
292
 
 
293
 
class RemoteRepository(_RpcHelper):
 
906
        return self.__class__ is other.__class__
 
907
 
 
908
    def network_name(self):
 
909
        if self._network_name:
 
910
            return self._network_name
 
911
        self._creating_repo._ensure_real()
 
912
        return self._creating_repo._real_repository._format.network_name()
 
913
 
 
914
    @property
 
915
    def pack_compresses(self):
 
916
        self._ensure_real()
 
917
        return self._custom_format.pack_compresses
 
918
 
 
919
    @property
 
920
    def _serializer(self):
 
921
        self._ensure_real()
 
922
        return self._custom_format._serializer
 
923
 
 
924
 
 
925
class RemoteRepository(_RpcHelper, lock._RelockDebugMixin,
 
926
    controldir.ControlComponent):
294
927
    """Repository accessed over rpc.
295
928
 
296
929
    For the moment most operations are performed using local transport-backed
299
932
 
300
933
    def __init__(self, remote_bzrdir, format, real_repository=None, _client=None):
301
934
        """Create a RemoteRepository instance.
302
 
        
 
935
 
303
936
        :param remote_bzrdir: The bzrdir hosting this repository.
304
937
        :param format: The RemoteFormat object to use.
305
938
        :param real_repository: If not None, a local implementation of the
322
955
        self._lock_token = None
323
956
        self._lock_count = 0
324
957
        self._leave_lock = False
 
958
        # Cache of revision parents; misses are cached during read locks, and
 
959
        # write locks when no _real_repository has been set.
325
960
        self._unstacked_provider = graph.CachingParentsProvider(
326
961
            get_parent_map=self._get_parent_map_rpc)
327
962
        self._unstacked_provider.disable_cache()
337
972
        # Additional places to query for data.
338
973
        self._fallback_repositories = []
339
974
 
 
975
    @property
 
976
    def user_transport(self):
 
977
        return self.bzrdir.user_transport
 
978
 
 
979
    @property
 
980
    def control_transport(self):
 
981
        # XXX: Normally you shouldn't directly get at the remote repository
 
982
        # transport, but I'm not sure it's worth making this method
 
983
        # optional -- mbp 2010-04-21
 
984
        return self.bzrdir.get_repository_transport(None)
 
985
 
340
986
    def __str__(self):
341
987
        return "%s(%s)" % (self.__class__.__name__, self.base)
342
988
 
344
990
 
345
991
    def abort_write_group(self, suppress_errors=False):
346
992
        """Complete a write group on the decorated repository.
347
 
        
348
 
        Smart methods peform operations in a single step so this api
 
993
 
 
994
        Smart methods perform operations in a single step so this API
349
995
        is not really applicable except as a compatibility thunk
350
996
        for older plugins that don't use e.g. the CommitBuilder
351
997
        facility.
356
1002
        return self._real_repository.abort_write_group(
357
1003
            suppress_errors=suppress_errors)
358
1004
 
 
1005
    @property
 
1006
    def chk_bytes(self):
 
1007
        """Decorate the real repository for now.
 
1008
 
 
1009
        In the long term a full blown network facility is needed to avoid
 
1010
        creating a real repository object locally.
 
1011
        """
 
1012
        self._ensure_real()
 
1013
        return self._real_repository.chk_bytes
 
1014
 
359
1015
    def commit_write_group(self):
360
1016
        """Complete a write group on the decorated repository.
361
 
        
362
 
        Smart methods peform operations in a single step so this api
 
1017
 
 
1018
        Smart methods perform operations in a single step so this API
363
1019
        is not really applicable except as a compatibility thunk
364
1020
        for older plugins that don't use e.g. the CommitBuilder
365
1021
        facility.
367
1023
        self._ensure_real()
368
1024
        return self._real_repository.commit_write_group()
369
1025
 
 
1026
    def resume_write_group(self, tokens):
 
1027
        self._ensure_real()
 
1028
        return self._real_repository.resume_write_group(tokens)
 
1029
 
 
1030
    def suspend_write_group(self):
 
1031
        self._ensure_real()
 
1032
        return self._real_repository.suspend_write_group()
 
1033
 
 
1034
    def get_missing_parent_inventories(self, check_for_missing_texts=True):
 
1035
        self._ensure_real()
 
1036
        return self._real_repository.get_missing_parent_inventories(
 
1037
            check_for_missing_texts=check_for_missing_texts)
 
1038
 
 
1039
    def _get_rev_id_for_revno_vfs(self, revno, known_pair):
 
1040
        self._ensure_real()
 
1041
        return self._real_repository.get_rev_id_for_revno(
 
1042
            revno, known_pair)
 
1043
 
 
1044
    def get_rev_id_for_revno(self, revno, known_pair):
 
1045
        """See Repository.get_rev_id_for_revno."""
 
1046
        path = self.bzrdir._path_for_remote_call(self._client)
 
1047
        try:
 
1048
            if self._client._medium._is_remote_before((1, 17)):
 
1049
                return self._get_rev_id_for_revno_vfs(revno, known_pair)
 
1050
            response = self._call(
 
1051
                'Repository.get_rev_id_for_revno', path, revno, known_pair)
 
1052
        except errors.UnknownSmartMethod:
 
1053
            self._client._medium._remember_remote_is_before((1, 17))
 
1054
            return self._get_rev_id_for_revno_vfs(revno, known_pair)
 
1055
        if response[0] == 'ok':
 
1056
            return True, response[1]
 
1057
        elif response[0] == 'history-incomplete':
 
1058
            known_pair = response[1:3]
 
1059
            for fallback in self._fallback_repositories:
 
1060
                found, result = fallback.get_rev_id_for_revno(revno, known_pair)
 
1061
                if found:
 
1062
                    return True, result
 
1063
                else:
 
1064
                    known_pair = result
 
1065
            # Not found in any fallbacks
 
1066
            return False, known_pair
 
1067
        else:
 
1068
            raise errors.UnexpectedSmartServerResponse(response)
 
1069
 
370
1070
    def _ensure_real(self):
371
1071
        """Ensure that there is a _real_repository set.
372
1072
 
373
1073
        Used before calls to self._real_repository.
 
1074
 
 
1075
        Note that _ensure_real causes many roundtrips to the server which are
 
1076
        not desirable, and prevents the use of smart one-roundtrip RPC's to
 
1077
        perform complex operations (such as accessing parent data, streaming
 
1078
        revisions etc). Adding calls to _ensure_real should only be done when
 
1079
        bringing up new functionality, adding fallbacks for smart methods that
 
1080
        require a fallback path, and never to replace an existing smart method
 
1081
        invocation. If in doubt chat to the bzr network team.
374
1082
        """
375
1083
        if self._real_repository is None:
 
1084
            if 'hpssvfs' in debug.debug_flags:
 
1085
                import traceback
 
1086
                warning('VFS Repository access triggered\n%s',
 
1087
                    ''.join(traceback.format_stack()))
 
1088
            self._unstacked_provider.missing_keys.clear()
376
1089
            self.bzrdir._ensure_real()
377
1090
            self._set_real_repository(
378
1091
                self.bzrdir._real_bzrdir.open_repository())
383
1096
    def find_text_key_references(self):
384
1097
        """Find the text key references within the repository.
385
1098
 
386
 
        :return: a dictionary mapping (file_id, revision_id) tuples to altered file-ids to an iterable of
387
 
        revision_ids. Each altered file-ids has the exact revision_ids that
388
 
        altered it listed explicitly.
389
1099
        :return: A dictionary mapping text keys ((fileid, revision_id) tuples)
390
1100
            to whether they were referred to by the inventory of the
391
1101
            revision_id that they contain. The inventory texts from all present
405
1115
        self._ensure_real()
406
1116
        return self._real_repository._generate_text_key_index()
407
1117
 
408
 
    @symbol_versioning.deprecated_method(symbol_versioning.one_four)
409
 
    def get_revision_graph(self, revision_id=None):
410
 
        """See Repository.get_revision_graph()."""
411
 
        return self._get_revision_graph(revision_id)
412
 
 
413
1118
    def _get_revision_graph(self, revision_id):
414
1119
        """Private method for using with old (< 1.2) servers to fallback."""
415
1120
        if revision_id is None:
416
1121
            revision_id = ''
417
 
        elif revision.is_null(revision_id):
 
1122
        elif _mod_revision.is_null(revision_id):
418
1123
            return {}
419
1124
 
420
1125
        path = self.bzrdir._path_for_remote_call(self._client)
432
1137
        for line in lines:
433
1138
            d = tuple(line.split())
434
1139
            revision_graph[d[0]] = d[1:]
435
 
            
 
1140
 
436
1141
        return revision_graph
437
1142
 
 
1143
    def _get_sink(self):
 
1144
        """See Repository._get_sink()."""
 
1145
        return RemoteStreamSink(self)
 
1146
 
 
1147
    def _get_source(self, to_format):
 
1148
        """Return a source for streaming from this repository."""
 
1149
        return RemoteStreamSource(self, to_format)
 
1150
 
 
1151
    @needs_read_lock
 
1152
    def get_file_graph(self):
 
1153
        return graph.Graph(self.texts)
 
1154
 
 
1155
    @needs_read_lock
438
1156
    def has_revision(self, revision_id):
439
 
        """See Repository.has_revision()."""
440
 
        if revision_id == NULL_REVISION:
441
 
            # The null revision is always present.
442
 
            return True
443
 
        path = self.bzrdir._path_for_remote_call(self._client)
444
 
        response = self._call('Repository.has_revision', path, revision_id)
445
 
        if response[0] not in ('yes', 'no'):
446
 
            raise errors.UnexpectedSmartServerResponse(response)
447
 
        if response[0] == 'yes':
448
 
            return True
449
 
        for fallback_repo in self._fallback_repositories:
450
 
            if fallback_repo.has_revision(revision_id):
451
 
                return True
452
 
        return False
 
1157
        """True if this repository has a copy of the revision."""
 
1158
        # Copy of bzrlib.repository.Repository.has_revision
 
1159
        return revision_id in self.has_revisions((revision_id,))
453
1160
 
 
1161
    @needs_read_lock
454
1162
    def has_revisions(self, revision_ids):
455
 
        """See Repository.has_revisions()."""
456
 
        # FIXME: This does many roundtrips, particularly when there are
457
 
        # fallback repositories.  -- mbp 20080905
458
 
        result = set()
459
 
        for revision_id in revision_ids:
460
 
            if self.has_revision(revision_id):
461
 
                result.add(revision_id)
 
1163
        """Probe to find out the presence of multiple revisions.
 
1164
 
 
1165
        :param revision_ids: An iterable of revision_ids.
 
1166
        :return: A set of the revision_ids that were present.
 
1167
        """
 
1168
        # Copy of bzrlib.repository.Repository.has_revisions
 
1169
        parent_map = self.get_parent_map(revision_ids)
 
1170
        result = set(parent_map)
 
1171
        if _mod_revision.NULL_REVISION in revision_ids:
 
1172
            result.add(_mod_revision.NULL_REVISION)
462
1173
        return result
463
1174
 
 
1175
    def _has_same_fallbacks(self, other_repo):
 
1176
        """Returns true if the repositories have the same fallbacks."""
 
1177
        # XXX: copied from Repository; it should be unified into a base class
 
1178
        # <https://bugs.launchpad.net/bzr/+bug/401622>
 
1179
        my_fb = self._fallback_repositories
 
1180
        other_fb = other_repo._fallback_repositories
 
1181
        if len(my_fb) != len(other_fb):
 
1182
            return False
 
1183
        for f, g in zip(my_fb, other_fb):
 
1184
            if not f.has_same_location(g):
 
1185
                return False
 
1186
        return True
 
1187
 
464
1188
    def has_same_location(self, other):
465
 
        return (self.__class__ == other.__class__ and
 
1189
        # TODO: Move to RepositoryBase and unify with the regular Repository
 
1190
        # one; unfortunately the tests rely on slightly different behaviour at
 
1191
        # present -- mbp 20090710
 
1192
        return (self.__class__ is other.__class__ and
466
1193
                self.bzrdir.transport.base == other.bzrdir.transport.base)
467
1194
 
468
1195
    def get_graph(self, other_repository=None):
470
1197
        parents_provider = self._make_parents_provider(other_repository)
471
1198
        return graph.Graph(parents_provider)
472
1199
 
 
1200
    @needs_read_lock
 
1201
    def get_known_graph_ancestry(self, revision_ids):
 
1202
        """Return the known graph for a set of revision ids and their ancestors.
 
1203
        """
 
1204
        st = static_tuple.StaticTuple
 
1205
        revision_keys = [st(r_id).intern() for r_id in revision_ids]
 
1206
        known_graph = self.revisions.get_known_graph_ancestry(revision_keys)
 
1207
        return graph.GraphThunkIdsToKeys(known_graph)
 
1208
 
473
1209
    def gather_stats(self, revid=None, committers=None):
474
1210
        """See Repository.gather_stats()."""
475
1211
        path = self.bzrdir._path_for_remote_call(self._client)
476
1212
        # revid can be None to indicate no revisions, not just NULL_REVISION
477
 
        if revid is None or revision.is_null(revid):
 
1213
        if revid is None or _mod_revision.is_null(revid):
478
1214
            fmt_revid = ''
479
1215
        else:
480
1216
            fmt_revid = revid
535
1271
    def is_write_locked(self):
536
1272
        return self._lock_mode == 'w'
537
1273
 
 
1274
    def _warn_if_deprecated(self, branch=None):
 
1275
        # If we have a real repository, the check will be done there, if we
 
1276
        # don't the check will be done remotely.
 
1277
        pass
 
1278
 
538
1279
    def lock_read(self):
 
1280
        """Lock the repository for read operations.
 
1281
 
 
1282
        :return: A bzrlib.lock.LogicalLockResult.
 
1283
        """
539
1284
        # wrong eventually - want a local lock cache context
540
1285
        if not self._lock_mode:
 
1286
            self._note_lock('r')
541
1287
            self._lock_mode = 'r'
542
1288
            self._lock_count = 1
543
 
            self._unstacked_provider.enable_cache(cache_misses=False)
 
1289
            self._unstacked_provider.enable_cache(cache_misses=True)
544
1290
            if self._real_repository is not None:
545
1291
                self._real_repository.lock_read()
 
1292
            for repo in self._fallback_repositories:
 
1293
                repo.lock_read()
546
1294
        else:
547
1295
            self._lock_count += 1
 
1296
        return lock.LogicalLockResult(self.unlock)
548
1297
 
549
1298
    def _remote_lock_write(self, token):
550
1299
        path = self.bzrdir._path_for_remote_call(self._client)
561
1310
 
562
1311
    def lock_write(self, token=None, _skip_rpc=False):
563
1312
        if not self._lock_mode:
 
1313
            self._note_lock('w')
564
1314
            if _skip_rpc:
565
1315
                if self._lock_token is not None:
566
1316
                    if token != self._lock_token:
580
1330
                self._leave_lock = False
581
1331
            self._lock_mode = 'w'
582
1332
            self._lock_count = 1
583
 
            self._unstacked_provider.enable_cache(cache_misses=False)
 
1333
            cache_misses = self._real_repository is None
 
1334
            self._unstacked_provider.enable_cache(cache_misses=cache_misses)
 
1335
            for repo in self._fallback_repositories:
 
1336
                # Writes don't affect fallback repos
 
1337
                repo.lock_read()
584
1338
        elif self._lock_mode == 'r':
585
1339
            raise errors.ReadOnlyError(self)
586
1340
        else:
587
1341
            self._lock_count += 1
588
 
        return self._lock_token or None
 
1342
        return RepositoryWriteLockResult(self.unlock, self._lock_token or None)
589
1343
 
590
1344
    def leave_lock_in_place(self):
591
1345
        if not self._lock_token:
604
1358
            implemented operations.
605
1359
        """
606
1360
        if self._real_repository is not None:
607
 
            raise AssertionError('_real_repository is already set')
 
1361
            # Replacing an already set real repository.
 
1362
            # We cannot do this [currently] if the repository is locked -
 
1363
            # synchronised state might be lost.
 
1364
            if self.is_locked():
 
1365
                raise AssertionError('_real_repository is already set')
608
1366
        if isinstance(repository, RemoteRepository):
609
1367
            raise AssertionError()
610
1368
        self._real_repository = repository
 
1369
        # three code paths happen here:
 
1370
        # 1) old servers, RemoteBranch.open() calls _ensure_real before setting
 
1371
        # up stacking. In this case self._fallback_repositories is [], and the
 
1372
        # real repo is already setup. Preserve the real repo and
 
1373
        # RemoteRepository.add_fallback_repository will avoid adding
 
1374
        # duplicates.
 
1375
        # 2) new servers, RemoteBranch.open() sets up stacking, and when
 
1376
        # ensure_real is triggered from a branch, the real repository to
 
1377
        # set already has a matching list with separate instances, but
 
1378
        # as they are also RemoteRepositories we don't worry about making the
 
1379
        # lists be identical.
 
1380
        # 3) new servers, RemoteRepository.ensure_real is triggered before
 
1381
        # RemoteBranch.ensure real, in this case we get a repo with no fallbacks
 
1382
        # and need to populate it.
 
1383
        if (self._fallback_repositories and
 
1384
            len(self._real_repository._fallback_repositories) !=
 
1385
            len(self._fallback_repositories)):
 
1386
            if len(self._real_repository._fallback_repositories):
 
1387
                raise AssertionError(
 
1388
                    "cannot cleanly remove existing _fallback_repositories")
611
1389
        for fb in self._fallback_repositories:
612
1390
            self._real_repository.add_fallback_repository(fb)
613
1391
        if self._lock_mode == 'w':
619
1397
 
620
1398
    def start_write_group(self):
621
1399
        """Start a write group on the decorated repository.
622
 
        
623
 
        Smart methods peform operations in a single step so this api
 
1400
 
 
1401
        Smart methods perform operations in a single step so this API
624
1402
        is not really applicable except as a compatibility thunk
625
1403
        for older plugins that don't use e.g. the CommitBuilder
626
1404
        facility.
641
1419
        else:
642
1420
            raise errors.UnexpectedSmartServerResponse(response)
643
1421
 
 
1422
    @only_raises(errors.LockNotHeld, errors.LockBroken)
644
1423
    def unlock(self):
 
1424
        if not self._lock_count:
 
1425
            return lock.cant_unlock_not_held(self)
645
1426
        self._lock_count -= 1
646
1427
        if self._lock_count > 0:
647
1428
            return
661
1442
            # problem releasing the vfs-based lock.
662
1443
            if old_mode == 'w':
663
1444
                # Only write-locked repositories need to make a remote method
664
 
                # call to perfom the unlock.
 
1445
                # call to perform the unlock.
665
1446
                old_token = self._lock_token
666
1447
                self._lock_token = None
667
1448
                if not self._leave_lock:
668
1449
                    self._unlock(old_token)
 
1450
        # Fallbacks are always 'lock_read()' so we don't pay attention to
 
1451
        # self._leave_lock
 
1452
        for repo in self._fallback_repositories:
 
1453
            repo.unlock()
669
1454
 
670
1455
    def break_lock(self):
671
1456
        # should hand off to the network
674
1459
 
675
1460
    def _get_tarball(self, compression):
676
1461
        """Return a TemporaryFile containing a repository tarball.
677
 
        
 
1462
 
678
1463
        Returns None if the server does not support sending tarballs.
679
1464
        """
680
1465
        import tempfile
714
1499
 
715
1500
    def get_commit_builder(self, branch, parents, config, timestamp=None,
716
1501
                           timezone=None, committer=None, revprops=None,
717
 
                           revision_id=None):
 
1502
                           revision_id=None, lossy=False):
718
1503
        # FIXME: It ought to be possible to call this without immediately
719
1504
        # triggering _ensure_real.  For now it's the easiest thing to do.
720
1505
        self._ensure_real()
721
1506
        real_repo = self._real_repository
722
1507
        builder = real_repo.get_commit_builder(branch, parents,
723
1508
                config, timestamp=timestamp, timezone=timezone,
724
 
                committer=committer, revprops=revprops, revision_id=revision_id)
 
1509
                committer=committer, revprops=revprops,
 
1510
                revision_id=revision_id, lossy=lossy)
725
1511
        return builder
726
1512
 
727
1513
    def add_fallback_repository(self, repository):
728
1514
        """Add a repository to use for looking up data not held locally.
729
 
        
 
1515
 
730
1516
        :param repository: A repository.
731
1517
        """
732
 
        # XXX: At the moment the RemoteRepository will allow fallbacks
733
 
        # unconditionally - however, a _real_repository will usually exist,
734
 
        # and may raise an error if it's not accommodated by the underlying
735
 
        # format.  Eventually we should check when opening the repository
736
 
        # whether it's willing to allow them or not.
737
 
        #
 
1518
        if not self._format.supports_external_lookups:
 
1519
            raise errors.UnstackableRepositoryFormat(
 
1520
                self._format.network_name(), self.base)
738
1521
        # We need to accumulate additional repositories here, to pass them in
739
1522
        # on various RPC's.
 
1523
        #
 
1524
        # Make the check before we lock: this raises an exception.
 
1525
        self._check_fallback_repository(repository)
 
1526
        if self.is_locked():
 
1527
            # We will call fallback.unlock() when we transition to the unlocked
 
1528
            # state, so always add a lock here. If a caller passes us a locked
 
1529
            # repository, they are responsible for unlocking it later.
 
1530
            repository.lock_read()
740
1531
        self._fallback_repositories.append(repository)
741
 
        # They are also seen by the fallback repository.  If it doesn't exist
742
 
        # yet they'll be added then.  This implicitly copies them.
743
 
        self._ensure_real()
 
1532
        # If self._real_repository was parameterised already (e.g. because a
 
1533
        # _real_branch had its get_stacked_on_url method called), then the
 
1534
        # repository to be added may already be in the _real_repositories list.
 
1535
        if self._real_repository is not None:
 
1536
            fallback_locations = [repo.user_url for repo in
 
1537
                self._real_repository._fallback_repositories]
 
1538
            if repository.user_url not in fallback_locations:
 
1539
                self._real_repository.add_fallback_repository(repository)
 
1540
 
 
1541
    def _check_fallback_repository(self, repository):
 
1542
        """Check that this repository can fallback to repository safely.
 
1543
 
 
1544
        Raise an error if not.
 
1545
 
 
1546
        :param repository: A repository to fallback to.
 
1547
        """
 
1548
        return _mod_repository.InterRepository._assert_same_model(
 
1549
            self, repository)
744
1550
 
745
1551
    def add_inventory(self, revid, inv, parents):
746
1552
        self._ensure_real()
747
1553
        return self._real_repository.add_inventory(revid, inv, parents)
748
1554
 
749
1555
    def add_inventory_by_delta(self, basis_revision_id, delta, new_revision_id,
750
 
                               parents):
 
1556
            parents, basis_inv=None, propagate_caches=False):
751
1557
        self._ensure_real()
752
1558
        return self._real_repository.add_inventory_by_delta(basis_revision_id,
753
 
            delta, new_revision_id, parents)
 
1559
            delta, new_revision_id, parents, basis_inv=basis_inv,
 
1560
            propagate_caches=propagate_caches)
754
1561
 
755
1562
    def add_revision(self, rev_id, rev, inv=None, config=None):
756
1563
        self._ensure_real()
762
1569
        self._ensure_real()
763
1570
        return self._real_repository.get_inventory(revision_id)
764
1571
 
765
 
    def iter_inventories(self, revision_ids):
 
1572
    def iter_inventories(self, revision_ids, ordering=None):
766
1573
        self._ensure_real()
767
 
        return self._real_repository.iter_inventories(revision_ids)
 
1574
        return self._real_repository.iter_inventories(revision_ids, ordering)
768
1575
 
769
1576
    @needs_read_lock
770
1577
    def get_revision(self, revision_id):
785
1592
        self._ensure_real()
786
1593
        return self._real_repository.make_working_trees()
787
1594
 
 
1595
    def refresh_data(self):
 
1596
        """Re-read any data needed to synchronise with disk.
 
1597
 
 
1598
        This method is intended to be called after another repository instance
 
1599
        (such as one used by a smart server) has inserted data into the
 
1600
        repository. On all repositories this will work outside of write groups.
 
1601
        Some repository formats (pack and newer for bzrlib native formats)
 
1602
        support refresh_data inside write groups. If called inside a write
 
1603
        group on a repository that does not support refreshing in a write group
 
1604
        IsInWriteGroupError will be raised.
 
1605
        """
 
1606
        if self._real_repository is not None:
 
1607
            self._real_repository.refresh_data()
 
1608
 
788
1609
    def revision_ids_to_search_result(self, result_set):
789
1610
        """Convert a set of revision ids to a graph SearchResult."""
790
1611
        result_parents = set()
799
1620
        return result
800
1621
 
801
1622
    @needs_read_lock
802
 
    def search_missing_revision_ids(self, other, revision_id=None, find_ghosts=True):
 
1623
    def search_missing_revision_ids(self, other,
 
1624
            revision_id=symbol_versioning.DEPRECATED_PARAMETER,
 
1625
            find_ghosts=True, revision_ids=None, if_present_ids=None,
 
1626
            limit=None):
803
1627
        """Return the revision ids that other has that this does not.
804
 
        
 
1628
 
805
1629
        These are returned in topological order.
806
1630
 
807
1631
        revision_id: only return revision ids included by revision_id.
808
1632
        """
809
 
        return repository.InterRepository.get(
810
 
            other, self).search_missing_revision_ids(revision_id, find_ghosts)
 
1633
        if symbol_versioning.deprecated_passed(revision_id):
 
1634
            symbol_versioning.warn(
 
1635
                'search_missing_revision_ids(revision_id=...) was '
 
1636
                'deprecated in 2.4.  Use revision_ids=[...] instead.',
 
1637
                DeprecationWarning, stacklevel=2)
 
1638
            if revision_ids is not None:
 
1639
                raise AssertionError(
 
1640
                    'revision_ids is mutually exclusive with revision_id')
 
1641
            if revision_id is not None:
 
1642
                revision_ids = [revision_id]
 
1643
        inter_repo = _mod_repository.InterRepository.get(other, self)
 
1644
        return inter_repo.search_missing_revision_ids(
 
1645
            find_ghosts=find_ghosts, revision_ids=revision_ids,
 
1646
            if_present_ids=if_present_ids, limit=limit)
811
1647
 
812
 
    def fetch(self, source, revision_id=None, pb=None, find_ghosts=False):
813
 
        # Not delegated to _real_repository so that InterRepository.get has a
814
 
        # chance to find an InterRepository specialised for RemoteRepository.
815
 
        if self.has_same_location(source):
 
1648
    def fetch(self, source, revision_id=None, find_ghosts=False,
 
1649
            fetch_spec=None):
 
1650
        # No base implementation to use as RemoteRepository is not a subclass
 
1651
        # of Repository; so this is a copy of Repository.fetch().
 
1652
        if fetch_spec is not None and revision_id is not None:
 
1653
            raise AssertionError(
 
1654
                "fetch_spec and revision_id are mutually exclusive.")
 
1655
        if self.is_in_write_group():
 
1656
            raise errors.InternalBzrError(
 
1657
                "May not fetch while in a write group.")
 
1658
        # fast path same-url fetch operations
 
1659
        if (self.has_same_location(source)
 
1660
            and fetch_spec is None
 
1661
            and self._has_same_fallbacks(source)):
816
1662
            # check that last_revision is in 'from' and then return a
817
1663
            # no-operation.
818
1664
            if (revision_id is not None and
819
 
                not revision.is_null(revision_id)):
 
1665
                not _mod_revision.is_null(revision_id)):
820
1666
                self.get_revision(revision_id)
821
1667
            return 0, []
822
 
        inter = repository.InterRepository.get(source, self)
823
 
        try:
824
 
            return inter.fetch(revision_id=revision_id, pb=pb, find_ghosts=find_ghosts)
825
 
        except NotImplementedError:
826
 
            raise errors.IncompatibleRepositories(source, self)
 
1668
        # if there is no specific appropriate InterRepository, this will get
 
1669
        # the InterRepository base class, which raises an
 
1670
        # IncompatibleRepositories when asked to fetch.
 
1671
        inter = _mod_repository.InterRepository.get(source, self)
 
1672
        return inter.fetch(revision_id=revision_id,
 
1673
            find_ghosts=find_ghosts, fetch_spec=fetch_spec)
827
1674
 
828
1675
    def create_bundle(self, target, base, fileobj, format=None):
829
1676
        self._ensure_real()
830
1677
        self._real_repository.create_bundle(target, base, fileobj, format)
831
1678
 
832
1679
    @needs_read_lock
 
1680
    @symbol_versioning.deprecated_method(
 
1681
        symbol_versioning.deprecated_in((2, 4, 0)))
833
1682
    def get_ancestry(self, revision_id, topo_sorted=True):
834
1683
        self._ensure_real()
835
1684
        return self._real_repository.get_ancestry(revision_id, topo_sorted)
842
1691
        self._ensure_real()
843
1692
        return self._real_repository._get_versioned_file_checker(
844
1693
            revisions, revision_versions_cache)
845
 
        
 
1694
 
846
1695
    def iter_files_bytes(self, desired_files):
847
1696
        """See Repository.iter_file_bytes.
848
1697
        """
849
1698
        self._ensure_real()
850
1699
        return self._real_repository.iter_files_bytes(desired_files)
851
1700
 
852
 
    @property
853
 
    def _fetch_order(self):
854
 
        """Decorate the real repository for now.
855
 
 
856
 
        In the long term getting this back from the remote repository as part
857
 
        of open would be more efficient.
858
 
        """
859
 
        self._ensure_real()
860
 
        return self._real_repository._fetch_order
861
 
 
862
 
    @property
863
 
    def _fetch_uses_deltas(self):
864
 
        """Decorate the real repository for now.
865
 
 
866
 
        In the long term getting this back from the remote repository as part
867
 
        of open would be more efficient.
868
 
        """
869
 
        self._ensure_real()
870
 
        return self._real_repository._fetch_uses_deltas
871
 
 
872
 
    @property
873
 
    def _fetch_reconcile(self):
874
 
        """Decorate the real repository for now.
875
 
 
876
 
        In the long term getting this back from the remote repository as part
877
 
        of open would be more efficient.
878
 
        """
879
 
        self._ensure_real()
880
 
        return self._real_repository._fetch_reconcile
 
1701
    def get_cached_parent_map(self, revision_ids):
 
1702
        """See bzrlib.CachingParentsProvider.get_cached_parent_map"""
 
1703
        return self._unstacked_provider.get_cached_parent_map(revision_ids)
881
1704
 
882
1705
    def get_parent_map(self, revision_ids):
883
1706
        """See bzrlib.Graph.get_parent_map()."""
890
1713
            # We already found out that the server can't understand
891
1714
            # Repository.get_parent_map requests, so just fetch the whole
892
1715
            # graph.
893
 
            # XXX: Note that this will issue a deprecation warning. This is ok
894
 
            # :- its because we're working with a deprecated server anyway, and
895
 
            # the user will almost certainly have seen a warning about the
896
 
            # server version already.
897
 
            rg = self.get_revision_graph()
898
 
            # There is an api discrepency between get_parent_map and
 
1716
            #
 
1717
            # Note that this reads the whole graph, when only some keys are
 
1718
            # wanted.  On this old server there's no way (?) to get them all
 
1719
            # in one go, and the user probably will have seen a warning about
 
1720
            # the server being old anyhow.
 
1721
            rg = self._get_revision_graph(None)
 
1722
            # There is an API discrepancy between get_parent_map and
899
1723
            # get_revision_graph. Specifically, a "key:()" pair in
900
1724
            # get_revision_graph just means a node has no parents. For
901
1725
            # "get_parent_map" it means the node is a ghost. So fix up the
931
1755
        # TODO: Manage this incrementally to avoid covering the same path
932
1756
        # repeatedly. (The server will have to on each request, but the less
933
1757
        # work done the better).
 
1758
        #
 
1759
        # Negative caching notes:
 
1760
        # new server sends missing when a request including the revid
 
1761
        # 'include-missing:' is present in the request.
 
1762
        # missing keys are serialised as missing:X, and we then call
 
1763
        # provider.note_missing(X) for-all X
934
1764
        parents_map = self._unstacked_provider.get_cached_map()
935
1765
        if parents_map is None:
936
1766
            # Repository is not locked, so there's no cache.
937
1767
            parents_map = {}
938
 
        start_set = set(parents_map)
939
 
        result_parents = set()
940
 
        for parents in parents_map.itervalues():
941
 
            result_parents.update(parents)
942
 
        stop_keys = result_parents.difference(start_set)
943
 
        included_keys = start_set.intersection(result_parents)
944
 
        start_set.difference_update(included_keys)
945
 
        recipe = (start_set, stop_keys, len(parents_map))
 
1768
        if _DEFAULT_SEARCH_DEPTH <= 0:
 
1769
            (start_set, stop_keys,
 
1770
             key_count) = graph.search_result_from_parent_map(
 
1771
                parents_map, self._unstacked_provider.missing_keys)
 
1772
        else:
 
1773
            (start_set, stop_keys,
 
1774
             key_count) = graph.limited_search_result_from_parent_map(
 
1775
                parents_map, self._unstacked_provider.missing_keys,
 
1776
                keys, depth=_DEFAULT_SEARCH_DEPTH)
 
1777
        recipe = ('manual', start_set, stop_keys, key_count)
946
1778
        body = self._serialise_search_recipe(recipe)
947
1779
        path = self.bzrdir._path_for_remote_call(self._client)
948
1780
        for key in keys:
950
1782
                raise ValueError(
951
1783
                    "key %r not a plain string" % (key,))
952
1784
        verb = 'Repository.get_parent_map'
953
 
        args = (path,) + tuple(keys)
 
1785
        args = (path, 'include-missing:') + tuple(keys)
954
1786
        try:
955
1787
            response = self._call_with_body_bytes_expecting_body(
956
1788
                verb, args, body)
966
1798
            # To avoid having to disconnect repeatedly, we keep track of the
967
1799
            # fact the server doesn't understand remote methods added in 1.2.
968
1800
            medium._remember_remote_is_before((1, 2))
969
 
            return self.get_revision_graph(None)
 
1801
            # Recurse just once and we should use the fallback code.
 
1802
            return self._get_parent_map_rpc(keys)
970
1803
        response_tuple, response_handler = response
971
1804
        if response_tuple[0] not in ['ok']:
972
1805
            response_handler.cancel_read_body()
983
1816
                if len(d) > 1:
984
1817
                    revision_graph[d[0]] = d[1:]
985
1818
                else:
986
 
                    # No parents - so give the Graph result (NULL_REVISION,).
987
 
                    revision_graph[d[0]] = (NULL_REVISION,)
 
1819
                    # No parents:
 
1820
                    if d[0].startswith('missing:'):
 
1821
                        revid = d[0][8:]
 
1822
                        self._unstacked_provider.note_missing_key(revid)
 
1823
                    else:
 
1824
                        # no parents - so give the Graph result
 
1825
                        # (NULL_REVISION,).
 
1826
                        revision_graph[d[0]] = (NULL_REVISION,)
988
1827
            return revision_graph
989
1828
 
990
1829
    @needs_read_lock
993
1832
        return self._real_repository.get_signature_text(revision_id)
994
1833
 
995
1834
    @needs_read_lock
996
 
    @symbol_versioning.deprecated_method(symbol_versioning.one_three)
997
 
    def get_revision_graph_with_ghosts(self, revision_ids=None):
998
 
        self._ensure_real()
999
 
        return self._real_repository.get_revision_graph_with_ghosts(
1000
 
            revision_ids=revision_ids)
1001
 
 
1002
 
    @needs_read_lock
1003
 
    def get_inventory_xml(self, revision_id):
1004
 
        self._ensure_real()
1005
 
        return self._real_repository.get_inventory_xml(revision_id)
1006
 
 
1007
 
    def deserialise_inventory(self, revision_id, xml):
1008
 
        self._ensure_real()
1009
 
        return self._real_repository.deserialise_inventory(revision_id, xml)
 
1835
    def _get_inventory_xml(self, revision_id):
 
1836
        self._ensure_real()
 
1837
        return self._real_repository._get_inventory_xml(revision_id)
1010
1838
 
1011
1839
    def reconcile(self, other=None, thorough=False):
1012
1840
        self._ensure_real()
1013
1841
        return self._real_repository.reconcile(other=other, thorough=thorough)
1014
 
        
 
1842
 
1015
1843
    def all_revision_ids(self):
1016
1844
        self._ensure_real()
1017
1845
        return self._real_repository.all_revision_ids()
1018
 
    
1019
 
    @needs_read_lock
1020
 
    def get_deltas_for_revisions(self, revisions):
1021
 
        self._ensure_real()
1022
 
        return self._real_repository.get_deltas_for_revisions(revisions)
1023
 
 
1024
 
    @needs_read_lock
1025
 
    def get_revision_delta(self, revision_id):
1026
 
        self._ensure_real()
1027
 
        return self._real_repository.get_revision_delta(revision_id)
 
1846
 
 
1847
    @needs_read_lock
 
1848
    def get_deltas_for_revisions(self, revisions, specific_fileids=None):
 
1849
        self._ensure_real()
 
1850
        return self._real_repository.get_deltas_for_revisions(revisions,
 
1851
            specific_fileids=specific_fileids)
 
1852
 
 
1853
    @needs_read_lock
 
1854
    def get_revision_delta(self, revision_id, specific_fileids=None):
 
1855
        self._ensure_real()
 
1856
        return self._real_repository.get_revision_delta(revision_id,
 
1857
            specific_fileids=specific_fileids)
1028
1858
 
1029
1859
    @needs_read_lock
1030
1860
    def revision_trees(self, revision_ids):
1037
1867
        return self._real_repository.get_revision_reconcile(revision_id)
1038
1868
 
1039
1869
    @needs_read_lock
1040
 
    def check(self, revision_ids=None):
 
1870
    def check(self, revision_ids=None, callback_refs=None, check_repo=True):
1041
1871
        self._ensure_real()
1042
 
        return self._real_repository.check(revision_ids=revision_ids)
 
1872
        return self._real_repository.check(revision_ids=revision_ids,
 
1873
            callback_refs=callback_refs, check_repo=check_repo)
1043
1874
 
1044
1875
    def copy_content_into(self, destination, revision_id=None):
1045
1876
        self._ensure_real()
1052
1883
        from bzrlib import osutils
1053
1884
        import tarfile
1054
1885
        # TODO: Maybe a progress bar while streaming the tarball?
1055
 
        note("Copying repository content as tarball...")
 
1886
        note(gettext("Copying repository content as tarball..."))
1056
1887
        tar_file = self._get_tarball('bz2')
1057
1888
        if tar_file is None:
1058
1889
            return None
1063
1894
            tmpdir = osutils.mkdtemp()
1064
1895
            try:
1065
1896
                _extract_tar(tar, tmpdir)
1066
 
                tmp_bzrdir = BzrDir.open(tmpdir)
 
1897
                tmp_bzrdir = _mod_bzrdir.BzrDir.open(tmpdir)
1067
1898
                tmp_repo = tmp_bzrdir.open_repository()
1068
1899
                tmp_repo.copy_content_into(destination, revision_id)
1069
1900
            finally:
1085
1916
        return self._real_repository.inventories
1086
1917
 
1087
1918
    @needs_write_lock
1088
 
    def pack(self):
 
1919
    def pack(self, hint=None, clean_obsolete_packs=False):
1089
1920
        """Compress the data within the repository.
1090
1921
 
1091
1922
        This is not currently implemented within the smart server.
1092
1923
        """
1093
1924
        self._ensure_real()
1094
 
        return self._real_repository.pack()
 
1925
        return self._real_repository.pack(hint=hint, clean_obsolete_packs=clean_obsolete_packs)
1095
1926
 
1096
1927
    @property
1097
1928
    def revisions(self):
1106
1937
        return self._real_repository.revisions
1107
1938
 
1108
1939
    def set_make_working_trees(self, new_value):
1109
 
        self._ensure_real()
1110
 
        self._real_repository.set_make_working_trees(new_value)
 
1940
        if new_value:
 
1941
            new_value_str = "True"
 
1942
        else:
 
1943
            new_value_str = "False"
 
1944
        path = self.bzrdir._path_for_remote_call(self._client)
 
1945
        try:
 
1946
            response = self._call(
 
1947
                'Repository.set_make_working_trees', path, new_value_str)
 
1948
        except errors.UnknownSmartMethod:
 
1949
            self._ensure_real()
 
1950
            self._real_repository.set_make_working_trees(new_value)
 
1951
        else:
 
1952
            if response[0] != 'ok':
 
1953
                raise errors.UnexpectedSmartServerResponse(response)
1111
1954
 
1112
1955
    @property
1113
1956
    def signatures(self):
1140
1983
        return self._real_repository.get_revisions(revision_ids)
1141
1984
 
1142
1985
    def supports_rich_root(self):
1143
 
        self._ensure_real()
1144
 
        return self._real_repository.supports_rich_root()
 
1986
        return self._format.rich_root_data
1145
1987
 
 
1988
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4, 0)))
1146
1989
    def iter_reverse_revision_history(self, revision_id):
1147
1990
        self._ensure_real()
1148
1991
        return self._real_repository.iter_reverse_revision_history(revision_id)
1149
1992
 
1150
1993
    @property
1151
1994
    def _serializer(self):
1152
 
        self._ensure_real()
1153
 
        return self._real_repository._serializer
 
1995
        return self._format._serializer
1154
1996
 
1155
1997
    def store_revision_signature(self, gpg_strategy, plaintext, revision_id):
1156
1998
        self._ensure_real()
1170
2012
        return self._real_repository.item_keys_introduced_by(revision_ids,
1171
2013
            _files_pb=_files_pb)
1172
2014
 
1173
 
    def revision_graph_can_have_wrong_parents(self):
1174
 
        # The answer depends on the remote repo format.
1175
 
        self._ensure_real()
1176
 
        return self._real_repository.revision_graph_can_have_wrong_parents()
1177
 
 
1178
 
    def _find_inconsistent_revision_parents(self):
1179
 
        self._ensure_real()
1180
 
        return self._real_repository._find_inconsistent_revision_parents()
 
2015
    def _find_inconsistent_revision_parents(self, revisions_iterator=None):
 
2016
        self._ensure_real()
 
2017
        return self._real_repository._find_inconsistent_revision_parents(
 
2018
            revisions_iterator)
1181
2019
 
1182
2020
    def _check_for_inconsistent_revision_parents(self):
1183
2021
        self._ensure_real()
1187
2025
        providers = [self._unstacked_provider]
1188
2026
        if other is not None:
1189
2027
            providers.insert(0, other)
1190
 
        providers.extend(r._make_parents_provider() for r in
1191
 
                         self._fallback_repositories)
1192
 
        return graph._StackedParentsProvider(providers)
 
2028
        return graph.StackedParentsProvider(_LazyListJoin(
 
2029
            providers, self._fallback_repositories))
1193
2030
 
1194
2031
    def _serialise_search_recipe(self, recipe):
1195
2032
        """Serialise a graph search recipe.
1197
2034
        :param recipe: A search recipe (start, stop, count).
1198
2035
        :return: Serialised bytes.
1199
2036
        """
1200
 
        start_keys = ' '.join(recipe[0])
1201
 
        stop_keys = ' '.join(recipe[1])
1202
 
        count = str(recipe[2])
 
2037
        start_keys = ' '.join(recipe[1])
 
2038
        stop_keys = ' '.join(recipe[2])
 
2039
        count = str(recipe[3])
1203
2040
        return '\n'.join((start_keys, stop_keys, count))
1204
2041
 
 
2042
    def _serialise_search_result(self, search_result):
 
2043
        parts = search_result.get_network_struct()
 
2044
        return '\n'.join(parts)
 
2045
 
1205
2046
    def autopack(self):
1206
2047
        path = self.bzrdir._path_for_remote_call(self._client)
1207
2048
        try:
1210
2051
            self._ensure_real()
1211
2052
            self._real_repository._pack_collection.autopack()
1212
2053
            return
1213
 
        if self._real_repository is not None:
1214
 
            # Reset the real repository's cache of pack names.
1215
 
            # XXX: At some point we may be able to skip this and just rely on
1216
 
            # the automatic retry logic to do the right thing, but for now we
1217
 
            # err on the side of being correct rather than being optimal.
1218
 
            self._real_repository._pack_collection.reload_pack_names()
 
2054
        self.refresh_data()
1219
2055
        if response[0] != 'ok':
1220
2056
            raise errors.UnexpectedSmartServerResponse(response)
1221
2057
 
1222
2058
 
 
2059
class RemoteStreamSink(vf_repository.StreamSink):
 
2060
 
 
2061
    def _insert_real(self, stream, src_format, resume_tokens):
 
2062
        self.target_repo._ensure_real()
 
2063
        sink = self.target_repo._real_repository._get_sink()
 
2064
        result = sink.insert_stream(stream, src_format, resume_tokens)
 
2065
        if not result:
 
2066
            self.target_repo.autopack()
 
2067
        return result
 
2068
 
 
2069
    def insert_stream(self, stream, src_format, resume_tokens):
 
2070
        target = self.target_repo
 
2071
        target._unstacked_provider.missing_keys.clear()
 
2072
        candidate_calls = [('Repository.insert_stream_1.19', (1, 19))]
 
2073
        if target._lock_token:
 
2074
            candidate_calls.append(('Repository.insert_stream_locked', (1, 14)))
 
2075
            lock_args = (target._lock_token or '',)
 
2076
        else:
 
2077
            candidate_calls.append(('Repository.insert_stream', (1, 13)))
 
2078
            lock_args = ()
 
2079
        client = target._client
 
2080
        medium = client._medium
 
2081
        path = target.bzrdir._path_for_remote_call(client)
 
2082
        # Probe for the verb to use with an empty stream before sending the
 
2083
        # real stream to it.  We do this both to avoid the risk of sending a
 
2084
        # large request that is then rejected, and because we don't want to
 
2085
        # implement a way to buffer, rewind, or restart the stream.
 
2086
        found_verb = False
 
2087
        for verb, required_version in candidate_calls:
 
2088
            if medium._is_remote_before(required_version):
 
2089
                continue
 
2090
            if resume_tokens:
 
2091
                # We've already done the probing (and set _is_remote_before) on
 
2092
                # a previous insert.
 
2093
                found_verb = True
 
2094
                break
 
2095
            byte_stream = smart_repo._stream_to_byte_stream([], src_format)
 
2096
            try:
 
2097
                response = client.call_with_body_stream(
 
2098
                    (verb, path, '') + lock_args, byte_stream)
 
2099
            except errors.UnknownSmartMethod:
 
2100
                medium._remember_remote_is_before(required_version)
 
2101
            else:
 
2102
                found_verb = True
 
2103
                break
 
2104
        if not found_verb:
 
2105
            # Have to use VFS.
 
2106
            return self._insert_real(stream, src_format, resume_tokens)
 
2107
        self._last_inv_record = None
 
2108
        self._last_substream = None
 
2109
        if required_version < (1, 19):
 
2110
            # Remote side doesn't support inventory deltas.  Wrap the stream to
 
2111
            # make sure we don't send any.  If the stream contains inventory
 
2112
            # deltas we'll interrupt the smart insert_stream request and
 
2113
            # fallback to VFS.
 
2114
            stream = self._stop_stream_if_inventory_delta(stream)
 
2115
        byte_stream = smart_repo._stream_to_byte_stream(
 
2116
            stream, src_format)
 
2117
        resume_tokens = ' '.join(resume_tokens)
 
2118
        response = client.call_with_body_stream(
 
2119
            (verb, path, resume_tokens) + lock_args, byte_stream)
 
2120
        if response[0][0] not in ('ok', 'missing-basis'):
 
2121
            raise errors.UnexpectedSmartServerResponse(response)
 
2122
        if self._last_substream is not None:
 
2123
            # The stream included an inventory-delta record, but the remote
 
2124
            # side isn't new enough to support them.  So we need to send the
 
2125
            # rest of the stream via VFS.
 
2126
            self.target_repo.refresh_data()
 
2127
            return self._resume_stream_with_vfs(response, src_format)
 
2128
        if response[0][0] == 'missing-basis':
 
2129
            tokens, missing_keys = bencode.bdecode_as_tuple(response[0][1])
 
2130
            resume_tokens = tokens
 
2131
            return resume_tokens, set(missing_keys)
 
2132
        else:
 
2133
            self.target_repo.refresh_data()
 
2134
            return [], set()
 
2135
 
 
2136
    def _resume_stream_with_vfs(self, response, src_format):
 
2137
        """Resume sending a stream via VFS, first resending the record and
 
2138
        substream that couldn't be sent via an insert_stream verb.
 
2139
        """
 
2140
        if response[0][0] == 'missing-basis':
 
2141
            tokens, missing_keys = bencode.bdecode_as_tuple(response[0][1])
 
2142
            # Ignore missing_keys, we haven't finished inserting yet
 
2143
        else:
 
2144
            tokens = []
 
2145
        def resume_substream():
 
2146
            # Yield the substream that was interrupted.
 
2147
            for record in self._last_substream:
 
2148
                yield record
 
2149
            self._last_substream = None
 
2150
        def resume_stream():
 
2151
            # Finish sending the interrupted substream
 
2152
            yield ('inventory-deltas', resume_substream())
 
2153
            # Then simply continue sending the rest of the stream.
 
2154
            for substream_kind, substream in self._last_stream:
 
2155
                yield substream_kind, substream
 
2156
        return self._insert_real(resume_stream(), src_format, tokens)
 
2157
 
 
2158
    def _stop_stream_if_inventory_delta(self, stream):
 
2159
        """Normally this just lets the original stream pass-through unchanged.
 
2160
 
 
2161
        However if any 'inventory-deltas' substream occurs it will stop
 
2162
        streaming, and store the interrupted substream and stream in
 
2163
        self._last_substream and self._last_stream so that the stream can be
 
2164
        resumed by _resume_stream_with_vfs.
 
2165
        """
 
2166
 
 
2167
        stream_iter = iter(stream)
 
2168
        for substream_kind, substream in stream_iter:
 
2169
            if substream_kind == 'inventory-deltas':
 
2170
                self._last_substream = substream
 
2171
                self._last_stream = stream_iter
 
2172
                return
 
2173
            else:
 
2174
                yield substream_kind, substream
 
2175
 
 
2176
 
 
2177
class RemoteStreamSource(vf_repository.StreamSource):
 
2178
    """Stream data from a remote server."""
 
2179
 
 
2180
    def get_stream(self, search):
 
2181
        if (self.from_repository._fallback_repositories and
 
2182
            self.to_format._fetch_order == 'topological'):
 
2183
            return self._real_stream(self.from_repository, search)
 
2184
        sources = []
 
2185
        seen = set()
 
2186
        repos = [self.from_repository]
 
2187
        while repos:
 
2188
            repo = repos.pop(0)
 
2189
            if repo in seen:
 
2190
                continue
 
2191
            seen.add(repo)
 
2192
            repos.extend(repo._fallback_repositories)
 
2193
            sources.append(repo)
 
2194
        return self.missing_parents_chain(search, sources)
 
2195
 
 
2196
    def get_stream_for_missing_keys(self, missing_keys):
 
2197
        self.from_repository._ensure_real()
 
2198
        real_repo = self.from_repository._real_repository
 
2199
        real_source = real_repo._get_source(self.to_format)
 
2200
        return real_source.get_stream_for_missing_keys(missing_keys)
 
2201
 
 
2202
    def _real_stream(self, repo, search):
 
2203
        """Get a stream for search from repo.
 
2204
        
 
2205
        This never called RemoteStreamSource.get_stream, and is a heler
 
2206
        for RemoteStreamSource._get_stream to allow getting a stream 
 
2207
        reliably whether fallback back because of old servers or trying
 
2208
        to stream from a non-RemoteRepository (which the stacked support
 
2209
        code will do).
 
2210
        """
 
2211
        source = repo._get_source(self.to_format)
 
2212
        if isinstance(source, RemoteStreamSource):
 
2213
            repo._ensure_real()
 
2214
            source = repo._real_repository._get_source(self.to_format)
 
2215
        return source.get_stream(search)
 
2216
 
 
2217
    def _get_stream(self, repo, search):
 
2218
        """Core worker to get a stream from repo for search.
 
2219
 
 
2220
        This is used by both get_stream and the stacking support logic. It
 
2221
        deliberately gets a stream for repo which does not need to be
 
2222
        self.from_repository. In the event that repo is not Remote, or
 
2223
        cannot do a smart stream, a fallback is made to the generic
 
2224
        repository._get_stream() interface, via self._real_stream.
 
2225
 
 
2226
        In the event of stacking, streams from _get_stream will not
 
2227
        contain all the data for search - this is normal (see get_stream).
 
2228
 
 
2229
        :param repo: A repository.
 
2230
        :param search: A search.
 
2231
        """
 
2232
        # Fallbacks may be non-smart
 
2233
        if not isinstance(repo, RemoteRepository):
 
2234
            return self._real_stream(repo, search)
 
2235
        client = repo._client
 
2236
        medium = client._medium
 
2237
        path = repo.bzrdir._path_for_remote_call(client)
 
2238
        search_bytes = repo._serialise_search_result(search)
 
2239
        args = (path, self.to_format.network_name())
 
2240
        candidate_verbs = [
 
2241
            ('Repository.get_stream_1.19', (1, 19)),
 
2242
            ('Repository.get_stream', (1, 13))]
 
2243
 
 
2244
        found_verb = False
 
2245
        for verb, version in candidate_verbs:
 
2246
            if medium._is_remote_before(version):
 
2247
                continue
 
2248
            try:
 
2249
                response = repo._call_with_body_bytes_expecting_body(
 
2250
                    verb, args, search_bytes)
 
2251
            except errors.UnknownSmartMethod:
 
2252
                medium._remember_remote_is_before(version)
 
2253
            except errors.UnknownErrorFromSmartServer, e:
 
2254
                if isinstance(search, graph.EverythingResult):
 
2255
                    error_verb = e.error_from_smart_server.error_verb
 
2256
                    if error_verb == 'BadSearch':
 
2257
                        # Pre-2.4 servers don't support this sort of search.
 
2258
                        # XXX: perhaps falling back to VFS on BadSearch is a
 
2259
                        # good idea in general?  It might provide a little bit
 
2260
                        # of protection against client-side bugs.
 
2261
                        medium._remember_remote_is_before((2, 4))
 
2262
                        break
 
2263
                raise
 
2264
            else:
 
2265
                response_tuple, response_handler = response
 
2266
                found_verb = True
 
2267
                break
 
2268
        if not found_verb:
 
2269
            return self._real_stream(repo, search)
 
2270
        if response_tuple[0] != 'ok':
 
2271
            raise errors.UnexpectedSmartServerResponse(response_tuple)
 
2272
        byte_stream = response_handler.read_streamed_body()
 
2273
        src_format, stream = smart_repo._byte_stream_to_stream(byte_stream,
 
2274
            self._record_counter)
 
2275
        if src_format.network_name() != repo._format.network_name():
 
2276
            raise AssertionError(
 
2277
                "Mismatched RemoteRepository and stream src %r, %r" % (
 
2278
                src_format.network_name(), repo._format.network_name()))
 
2279
        return stream
 
2280
 
 
2281
    def missing_parents_chain(self, search, sources):
 
2282
        """Chain multiple streams together to handle stacking.
 
2283
 
 
2284
        :param search: The overall search to satisfy with streams.
 
2285
        :param sources: A list of Repository objects to query.
 
2286
        """
 
2287
        self.from_serialiser = self.from_repository._format._serializer
 
2288
        self.seen_revs = set()
 
2289
        self.referenced_revs = set()
 
2290
        # If there are heads in the search, or the key count is > 0, we are not
 
2291
        # done.
 
2292
        while not search.is_empty() and len(sources) > 1:
 
2293
            source = sources.pop(0)
 
2294
            stream = self._get_stream(source, search)
 
2295
            for kind, substream in stream:
 
2296
                if kind != 'revisions':
 
2297
                    yield kind, substream
 
2298
                else:
 
2299
                    yield kind, self.missing_parents_rev_handler(substream)
 
2300
            search = search.refine(self.seen_revs, self.referenced_revs)
 
2301
            self.seen_revs = set()
 
2302
            self.referenced_revs = set()
 
2303
        if not search.is_empty():
 
2304
            for kind, stream in self._get_stream(sources[0], search):
 
2305
                yield kind, stream
 
2306
 
 
2307
    def missing_parents_rev_handler(self, substream):
 
2308
        for content in substream:
 
2309
            revision_bytes = content.get_bytes_as('fulltext')
 
2310
            revision = self.from_serialiser.read_revision_from_string(
 
2311
                revision_bytes)
 
2312
            self.seen_revs.add(content.key[-1])
 
2313
            self.referenced_revs.update(revision.parent_ids)
 
2314
            yield content
 
2315
 
 
2316
 
1223
2317
class RemoteBranchLockableFiles(LockableFiles):
1224
2318
    """A 'LockableFiles' implementation that talks to a smart server.
1225
 
    
 
2319
 
1226
2320
    This is not a public interface class.
1227
2321
    """
1228
2322
 
1242
2336
 
1243
2337
class RemoteBranchFormat(branch.BranchFormat):
1244
2338
 
1245
 
    def __init__(self):
 
2339
    def __init__(self, network_name=None):
1246
2340
        super(RemoteBranchFormat, self).__init__()
1247
2341
        self._matchingbzrdir = RemoteBzrDirFormat()
1248
2342
        self._matchingbzrdir.set_branch_format(self)
 
2343
        self._custom_format = None
 
2344
        self._network_name = network_name
1249
2345
 
1250
2346
    def __eq__(self, other):
1251
 
        return (isinstance(other, RemoteBranchFormat) and 
 
2347
        return (isinstance(other, RemoteBranchFormat) and
1252
2348
            self.__dict__ == other.__dict__)
1253
2349
 
 
2350
    def _ensure_real(self):
 
2351
        if self._custom_format is None:
 
2352
            self._custom_format = branch.network_format_registry.get(
 
2353
                self._network_name)
 
2354
 
1254
2355
    def get_format_description(self):
1255
 
        return 'Remote BZR Branch'
1256
 
 
1257
 
    def get_format_string(self):
1258
 
        return 'Remote BZR Branch'
1259
 
 
1260
 
    def open(self, a_bzrdir):
1261
 
        return a_bzrdir.open_branch()
1262
 
 
1263
 
    def initialize(self, a_bzrdir):
1264
 
        return a_bzrdir.create_branch()
 
2356
        self._ensure_real()
 
2357
        return 'Remote: ' + self._custom_format.get_format_description()
 
2358
 
 
2359
    def network_name(self):
 
2360
        return self._network_name
 
2361
 
 
2362
    def open(self, a_bzrdir, name=None, ignore_fallbacks=False):
 
2363
        return a_bzrdir.open_branch(name=name, 
 
2364
            ignore_fallbacks=ignore_fallbacks)
 
2365
 
 
2366
    def _vfs_initialize(self, a_bzrdir, name, append_revisions_only):
 
2367
        # Initialisation when using a local bzrdir object, or a non-vfs init
 
2368
        # method is not available on the server.
 
2369
        # self._custom_format is always set - the start of initialize ensures
 
2370
        # that.
 
2371
        if isinstance(a_bzrdir, RemoteBzrDir):
 
2372
            a_bzrdir._ensure_real()
 
2373
            result = self._custom_format.initialize(a_bzrdir._real_bzrdir,
 
2374
                name, append_revisions_only=append_revisions_only)
 
2375
        else:
 
2376
            # We assume the bzrdir is parameterised; it may not be.
 
2377
            result = self._custom_format.initialize(a_bzrdir, name,
 
2378
                append_revisions_only=append_revisions_only)
 
2379
        if (isinstance(a_bzrdir, RemoteBzrDir) and
 
2380
            not isinstance(result, RemoteBranch)):
 
2381
            result = RemoteBranch(a_bzrdir, a_bzrdir.find_repository(), result,
 
2382
                                  name=name)
 
2383
        return result
 
2384
 
 
2385
    def initialize(self, a_bzrdir, name=None, repository=None,
 
2386
                   append_revisions_only=None):
 
2387
        # 1) get the network name to use.
 
2388
        if self._custom_format:
 
2389
            network_name = self._custom_format.network_name()
 
2390
        else:
 
2391
            # Select the current bzrlib default and ask for that.
 
2392
            reference_bzrdir_format = _mod_bzrdir.format_registry.get('default')()
 
2393
            reference_format = reference_bzrdir_format.get_branch_format()
 
2394
            self._custom_format = reference_format
 
2395
            network_name = reference_format.network_name()
 
2396
        # Being asked to create on a non RemoteBzrDir:
 
2397
        if not isinstance(a_bzrdir, RemoteBzrDir):
 
2398
            return self._vfs_initialize(a_bzrdir, name=name,
 
2399
                append_revisions_only=append_revisions_only)
 
2400
        medium = a_bzrdir._client._medium
 
2401
        if medium._is_remote_before((1, 13)):
 
2402
            return self._vfs_initialize(a_bzrdir, name=name,
 
2403
                append_revisions_only=append_revisions_only)
 
2404
        # Creating on a remote bzr dir.
 
2405
        # 2) try direct creation via RPC
 
2406
        path = a_bzrdir._path_for_remote_call(a_bzrdir._client)
 
2407
        if name is not None:
 
2408
            # XXX JRV20100304: Support creating colocated branches
 
2409
            raise errors.NoColocatedBranchSupport(self)
 
2410
        verb = 'BzrDir.create_branch'
 
2411
        try:
 
2412
            response = a_bzrdir._call(verb, path, network_name)
 
2413
        except errors.UnknownSmartMethod:
 
2414
            # Fallback - use vfs methods
 
2415
            medium._remember_remote_is_before((1, 13))
 
2416
            return self._vfs_initialize(a_bzrdir, name=name,
 
2417
                    append_revisions_only=append_revisions_only)
 
2418
        if response[0] != 'ok':
 
2419
            raise errors.UnexpectedSmartServerResponse(response)
 
2420
        # Turn the response into a RemoteRepository object.
 
2421
        format = RemoteBranchFormat(network_name=response[1])
 
2422
        repo_format = response_tuple_to_repo_format(response[3:])
 
2423
        repo_path = response[2]
 
2424
        if repository is not None:
 
2425
            remote_repo_url = urlutils.join(a_bzrdir.user_url, repo_path)
 
2426
            url_diff = urlutils.relative_url(repository.user_url,
 
2427
                    remote_repo_url)
 
2428
            if url_diff != '.':
 
2429
                raise AssertionError(
 
2430
                    'repository.user_url %r does not match URL from server '
 
2431
                    'response (%r + %r)'
 
2432
                    % (repository.user_url, a_bzrdir.user_url, repo_path))
 
2433
            remote_repo = repository
 
2434
        else:
 
2435
            if repo_path == '':
 
2436
                repo_bzrdir = a_bzrdir
 
2437
            else:
 
2438
                repo_bzrdir = RemoteBzrDir(
 
2439
                    a_bzrdir.root_transport.clone(repo_path), a_bzrdir._format,
 
2440
                    a_bzrdir._client)
 
2441
            remote_repo = RemoteRepository(repo_bzrdir, repo_format)
 
2442
        remote_branch = RemoteBranch(a_bzrdir, remote_repo,
 
2443
            format=format, setup_stacking=False, name=name)
 
2444
        if append_revisions_only:
 
2445
            remote_branch.set_append_revisions_only(append_revisions_only)
 
2446
        # XXX: We know this is a new branch, so it must have revno 0, revid
 
2447
        # NULL_REVISION. Creating the branch locked would make this be unable
 
2448
        # to be wrong; here its simply very unlikely to be wrong. RBC 20090225
 
2449
        remote_branch._last_revision_info_cache = 0, NULL_REVISION
 
2450
        return remote_branch
 
2451
 
 
2452
    def make_tags(self, branch):
 
2453
        self._ensure_real()
 
2454
        return self._custom_format.make_tags(branch)
1265
2455
 
1266
2456
    def supports_tags(self):
1267
2457
        # Remote branches might support tags, but we won't know until we
1268
2458
        # access the real remote branch.
1269
 
        return True
1270
 
 
1271
 
 
1272
 
class RemoteBranch(branch.Branch, _RpcHelper):
 
2459
        self._ensure_real()
 
2460
        return self._custom_format.supports_tags()
 
2461
 
 
2462
    def supports_stacking(self):
 
2463
        self._ensure_real()
 
2464
        return self._custom_format.supports_stacking()
 
2465
 
 
2466
    def supports_set_append_revisions_only(self):
 
2467
        self._ensure_real()
 
2468
        return self._custom_format.supports_set_append_revisions_only()
 
2469
 
 
2470
    def _use_default_local_heads_to_fetch(self):
 
2471
        # If the branch format is a metadir format *and* its heads_to_fetch
 
2472
        # implementation is not overridden vs the base class, we can use the
 
2473
        # base class logic rather than use the heads_to_fetch RPC.  This is
 
2474
        # usually cheaper in terms of net round trips, as the last-revision and
 
2475
        # tags info fetched is cached and would be fetched anyway.
 
2476
        self._ensure_real()
 
2477
        if isinstance(self._custom_format, branch.BranchFormatMetadir):
 
2478
            branch_class = self._custom_format._branch_class()
 
2479
            heads_to_fetch_impl = branch_class.heads_to_fetch.im_func
 
2480
            if heads_to_fetch_impl is branch.Branch.heads_to_fetch.im_func:
 
2481
                return True
 
2482
        return False
 
2483
 
 
2484
class RemoteBranch(branch.Branch, _RpcHelper, lock._RelockDebugMixin):
1273
2485
    """Branch stored on a server accessed by HPSS RPC.
1274
2486
 
1275
2487
    At the moment most operations are mapped down to simple file operations.
1276
2488
    """
1277
2489
 
1278
2490
    def __init__(self, remote_bzrdir, remote_repository, real_branch=None,
1279
 
        _client=None):
 
2491
        _client=None, format=None, setup_stacking=True, name=None):
1280
2492
        """Create a RemoteBranch instance.
1281
2493
 
1282
2494
        :param real_branch: An optional local implementation of the branch
1283
2495
            format, usually accessing the data via the VFS.
1284
2496
        :param _client: Private parameter for testing.
 
2497
        :param format: A RemoteBranchFormat object, None to create one
 
2498
            automatically. If supplied it should have a network_name already
 
2499
            supplied.
 
2500
        :param setup_stacking: If True make an RPC call to determine the
 
2501
            stacked (or not) status of the branch. If False assume the branch
 
2502
            is not stacked.
 
2503
        :param name: Colocated branch name
1285
2504
        """
1286
2505
        # We intentionally don't call the parent class's __init__, because it
1287
2506
        # will try to assign to self.tags, which is a property in this subclass.
1288
2507
        # And the parent's __init__ doesn't do much anyway.
1289
 
        self._revision_id_to_revno_cache = None
1290
 
        self._partial_revision_id_to_revno_cache = {}
1291
 
        self._revision_history_cache = None
1292
 
        self._last_revision_info_cache = None
1293
 
        self._merge_sorted_revisions_cache = None
1294
2508
        self.bzrdir = remote_bzrdir
1295
2509
        if _client is not None:
1296
2510
            self._client = _client
1309
2523
            self._real_branch.repository = self.repository
1310
2524
        else:
1311
2525
            self._real_branch = None
1312
 
        # Fill out expected attributes of branch for bzrlib api users.
1313
 
        self._format = RemoteBranchFormat()
1314
 
        self.base = self.bzrdir.root_transport.base
 
2526
        # Fill out expected attributes of branch for bzrlib API users.
 
2527
        self._clear_cached_state()
 
2528
        # TODO: deprecate self.base in favor of user_url
 
2529
        self.base = self.bzrdir.user_url
 
2530
        self._name = name
1315
2531
        self._control_files = None
1316
2532
        self._lock_mode = None
1317
2533
        self._lock_token = None
1318
2534
        self._repo_lock_token = None
1319
2535
        self._lock_count = 0
1320
2536
        self._leave_lock = False
 
2537
        # Setup a format: note that we cannot call _ensure_real until all the
 
2538
        # attributes above are set: This code cannot be moved higher up in this
 
2539
        # function.
 
2540
        if format is None:
 
2541
            self._format = RemoteBranchFormat()
 
2542
            if real_branch is not None:
 
2543
                self._format._network_name = \
 
2544
                    self._real_branch._format.network_name()
 
2545
        else:
 
2546
            self._format = format
 
2547
        # when we do _ensure_real we may need to pass ignore_fallbacks to the
 
2548
        # branch.open_branch method.
 
2549
        self._real_ignore_fallbacks = not setup_stacking
 
2550
        if not self._format._network_name:
 
2551
            # Did not get from open_branchV2 - old server.
 
2552
            self._ensure_real()
 
2553
            self._format._network_name = \
 
2554
                self._real_branch._format.network_name()
 
2555
        self.tags = self._format.make_tags(self)
1321
2556
        # The base class init is not called, so we duplicate this:
1322
2557
        hooks = branch.Branch.hooks['open']
1323
2558
        for hook in hooks:
1324
2559
            hook(self)
1325
 
        self._setup_stacking()
 
2560
        self._is_stacked = False
 
2561
        if setup_stacking:
 
2562
            self._setup_stacking()
1326
2563
 
1327
2564
    def _setup_stacking(self):
1328
2565
        # configure stacking into the remote repository, by reading it from
1332
2569
        except (errors.NotStacked, errors.UnstackableBranchFormat,
1333
2570
            errors.UnstackableRepositoryFormat), e:
1334
2571
            return
1335
 
        # it's relative to this branch...
1336
 
        fallback_url = urlutils.join(self.base, fallback_url)
1337
 
        transports = [self.bzrdir.root_transport]
1338
 
        if self._real_branch is not None:
1339
 
            transports.append(self._real_branch._transport)
1340
 
        stacked_on = branch.Branch.open(fallback_url,
1341
 
                                        possible_transports=transports)
1342
 
        self.repository.add_fallback_repository(stacked_on.repository)
 
2572
        self._is_stacked = True
 
2573
        self._activate_fallback_location(fallback_url)
 
2574
 
 
2575
    def _get_config(self):
 
2576
        return RemoteBranchConfig(self)
1343
2577
 
1344
2578
    def _get_real_transport(self):
1345
2579
        # if we try vfs access, return the real branch's vfs transport
1363
2597
                raise AssertionError('smart server vfs must be enabled '
1364
2598
                    'to use vfs implementation')
1365
2599
            self.bzrdir._ensure_real()
1366
 
            self._real_branch = self.bzrdir._real_bzrdir.open_branch()
 
2600
            self._real_branch = self.bzrdir._real_bzrdir.open_branch(
 
2601
                ignore_fallbacks=self._real_ignore_fallbacks, name=self._name)
1367
2602
            if self.repository._real_repository is None:
1368
2603
                # Give the remote repository the matching real repo.
1369
2604
                real_repo = self._real_branch.repository
1398
2633
        too, in fact doing so might harm performance.
1399
2634
        """
1400
2635
        super(RemoteBranch, self)._clear_cached_state()
1401
 
        
 
2636
 
1402
2637
    @property
1403
2638
    def control_files(self):
1404
2639
        # Defer actually creating RemoteBranchLockableFiles until its needed,
1408
2643
                self.bzrdir, self._client)
1409
2644
        return self._control_files
1410
2645
 
1411
 
    def _get_checkout_format(self):
 
2646
    def _get_checkout_format(self, lightweight=False):
1412
2647
        self._ensure_real()
1413
 
        return self._real_branch._get_checkout_format()
 
2648
        if lightweight:
 
2649
            format = RemoteBzrDirFormat()
 
2650
            self.bzrdir._format._supply_sub_formats_to(format)
 
2651
            format.workingtree_format = self._real_branch._get_checkout_format(
 
2652
                lightweight=lightweight).workingtree_format
 
2653
            return format
 
2654
        else:
 
2655
            return self._real_branch._get_checkout_format(lightweight=False)
1414
2656
 
1415
2657
    def get_physical_lock_status(self):
1416
2658
        """See Branch.get_physical_lock_status()."""
1443
2685
            raise errors.UnexpectedSmartServerResponse(response)
1444
2686
        return response[1]
1445
2687
 
 
2688
    def set_stacked_on_url(self, url):
 
2689
        branch.Branch.set_stacked_on_url(self, url)
 
2690
        if not url:
 
2691
            self._is_stacked = False
 
2692
        else:
 
2693
            self._is_stacked = True
 
2694
 
 
2695
    def _vfs_get_tags_bytes(self):
 
2696
        self._ensure_real()
 
2697
        return self._real_branch._get_tags_bytes()
 
2698
 
 
2699
    @needs_read_lock
 
2700
    def _get_tags_bytes(self):
 
2701
        if self._tags_bytes is None:
 
2702
            self._tags_bytes = self._get_tags_bytes_via_hpss()
 
2703
        return self._tags_bytes
 
2704
 
 
2705
    def _get_tags_bytes_via_hpss(self):
 
2706
        medium = self._client._medium
 
2707
        if medium._is_remote_before((1, 13)):
 
2708
            return self._vfs_get_tags_bytes()
 
2709
        try:
 
2710
            response = self._call('Branch.get_tags_bytes', self._remote_path())
 
2711
        except errors.UnknownSmartMethod:
 
2712
            medium._remember_remote_is_before((1, 13))
 
2713
            return self._vfs_get_tags_bytes()
 
2714
        return response[0]
 
2715
 
 
2716
    def _vfs_set_tags_bytes(self, bytes):
 
2717
        self._ensure_real()
 
2718
        return self._real_branch._set_tags_bytes(bytes)
 
2719
 
 
2720
    def _set_tags_bytes(self, bytes):
 
2721
        if self.is_locked():
 
2722
            self._tags_bytes = bytes
 
2723
        medium = self._client._medium
 
2724
        if medium._is_remote_before((1, 18)):
 
2725
            self._vfs_set_tags_bytes(bytes)
 
2726
            return
 
2727
        try:
 
2728
            args = (
 
2729
                self._remote_path(), self._lock_token, self._repo_lock_token)
 
2730
            response = self._call_with_body_bytes(
 
2731
                'Branch.set_tags_bytes', args, bytes)
 
2732
        except errors.UnknownSmartMethod:
 
2733
            medium._remember_remote_is_before((1, 18))
 
2734
            self._vfs_set_tags_bytes(bytes)
 
2735
 
1446
2736
    def lock_read(self):
 
2737
        """Lock the branch for read operations.
 
2738
 
 
2739
        :return: A bzrlib.lock.LogicalLockResult.
 
2740
        """
1447
2741
        self.repository.lock_read()
1448
2742
        if not self._lock_mode:
 
2743
            self._note_lock('r')
1449
2744
            self._lock_mode = 'r'
1450
2745
            self._lock_count = 1
1451
2746
            if self._real_branch is not None:
1452
2747
                self._real_branch.lock_read()
1453
2748
        else:
1454
2749
            self._lock_count += 1
 
2750
        return lock.LogicalLockResult(self.unlock)
1455
2751
 
1456
2752
    def _remote_lock_write(self, token):
1457
2753
        if token is None:
1458
2754
            branch_token = repo_token = ''
1459
2755
        else:
1460
2756
            branch_token = token
1461
 
            repo_token = self.repository.lock_write()
 
2757
            repo_token = self.repository.lock_write().repository_token
1462
2758
            self.repository.unlock()
1463
2759
        err_context = {'token': token}
1464
 
        response = self._call(
1465
 
            'Branch.lock_write', self._remote_path(), branch_token,
1466
 
            repo_token or '', **err_context)
 
2760
        try:
 
2761
            response = self._call(
 
2762
                'Branch.lock_write', self._remote_path(), branch_token,
 
2763
                repo_token or '', **err_context)
 
2764
        except errors.LockContention, e:
 
2765
            # The LockContention from the server doesn't have any
 
2766
            # information about the lock_url. We re-raise LockContention
 
2767
            # with valid lock_url.
 
2768
            raise errors.LockContention('(remote lock)',
 
2769
                self.repository.base.split('.bzr/')[0])
1467
2770
        if response[0] != 'ok':
1468
2771
            raise errors.UnexpectedSmartServerResponse(response)
1469
2772
        ok, branch_token, repo_token = response
1470
2773
        return branch_token, repo_token
1471
 
            
 
2774
 
1472
2775
    def lock_write(self, token=None):
1473
2776
        if not self._lock_mode:
 
2777
            self._note_lock('w')
1474
2778
            # Lock the branch and repo in one remote call.
1475
2779
            remote_tokens = self._remote_lock_write(token)
1476
2780
            self._lock_token, self._repo_lock_token = remote_tokens
1489
2793
            self._lock_mode = 'w'
1490
2794
            self._lock_count = 1
1491
2795
        elif self._lock_mode == 'r':
1492
 
            raise errors.ReadOnlyTransaction
 
2796
            raise errors.ReadOnlyError(self)
1493
2797
        else:
1494
2798
            if token is not None:
1495
2799
                # A token was given to lock_write, and we're relocking, so
1500
2804
            self._lock_count += 1
1501
2805
            # Re-lock the repository too.
1502
2806
            self.repository.lock_write(self._repo_lock_token)
1503
 
        return self._lock_token or None
 
2807
        return BranchWriteLockResult(self.unlock, self._lock_token or None)
1504
2808
 
1505
2809
    def _unlock(self, branch_token, repo_token):
1506
2810
        err_context = {'token': str((branch_token, repo_token))}
1511
2815
            return
1512
2816
        raise errors.UnexpectedSmartServerResponse(response)
1513
2817
 
 
2818
    @only_raises(errors.LockNotHeld, errors.LockBroken)
1514
2819
    def unlock(self):
1515
2820
        try:
1516
2821
            self._lock_count -= 1
1529
2834
                    self._real_branch.unlock()
1530
2835
                if mode != 'w':
1531
2836
                    # Only write-locked branched need to make a remote method
1532
 
                    # call to perfom the unlock.
 
2837
                    # call to perform the unlock.
1533
2838
                    return
1534
2839
                if not self._lock_token:
1535
2840
                    raise AssertionError('Locked, but no token!')
1556
2861
            raise NotImplementedError(self.dont_leave_lock_in_place)
1557
2862
        self._leave_lock = False
1558
2863
 
1559
 
    def _last_revision_info(self):
 
2864
    @needs_read_lock
 
2865
    def get_rev_id(self, revno, history=None):
 
2866
        if revno == 0:
 
2867
            return _mod_revision.NULL_REVISION
 
2868
        last_revision_info = self.last_revision_info()
 
2869
        ok, result = self.repository.get_rev_id_for_revno(
 
2870
            revno, last_revision_info)
 
2871
        if ok:
 
2872
            return result
 
2873
        missing_parent = result[1]
 
2874
        # Either the revision named by the server is missing, or its parent
 
2875
        # is.  Call get_parent_map to determine which, so that we report a
 
2876
        # useful error.
 
2877
        parent_map = self.repository.get_parent_map([missing_parent])
 
2878
        if missing_parent in parent_map:
 
2879
            missing_parent = parent_map[missing_parent]
 
2880
        raise errors.RevisionNotPresent(missing_parent, self.repository)
 
2881
 
 
2882
    def _read_last_revision_info(self):
1560
2883
        response = self._call('Branch.last_revision_info', self._remote_path())
1561
2884
        if response[0] != 'ok':
1562
2885
            raise SmartProtocolError('unexpected response code %s' % (response,))
1566
2889
 
1567
2890
    def _gen_revision_history(self):
1568
2891
        """See Branch._gen_revision_history()."""
 
2892
        if self._is_stacked:
 
2893
            self._ensure_real()
 
2894
            return self._real_branch._gen_revision_history()
1569
2895
        response_tuple, response_handler = self._call_expecting_body(
1570
2896
            'Branch.revision_history', self._remote_path())
1571
2897
        if response_tuple[0] != 'ok':
1580
2906
 
1581
2907
    def _set_last_revision_descendant(self, revision_id, other_branch,
1582
2908
            allow_diverged=False, allow_overwrite_descendant=False):
 
2909
        # This performs additional work to meet the hook contract; while its
 
2910
        # undesirable, we have to synthesise the revno to call the hook, and
 
2911
        # not calling the hook is worse as it means changes can't be prevented.
 
2912
        # Having calculated this though, we can't just call into
 
2913
        # set_last_revision_info as a simple call, because there is a set_rh
 
2914
        # hook that some folk may still be using.
 
2915
        old_revno, old_revid = self.last_revision_info()
 
2916
        history = self._lefthand_history(revision_id)
 
2917
        self._run_pre_change_branch_tip_hooks(len(history), revision_id)
1583
2918
        err_context = {'other_branch': other_branch}
1584
2919
        response = self._call('Branch.set_last_revision_ex',
1585
2920
            self._remote_path(), self._lock_token, self._repo_lock_token,
1590
2925
            raise errors.UnexpectedSmartServerResponse(response)
1591
2926
        new_revno, new_revision_id = response[1:]
1592
2927
        self._last_revision_info_cache = new_revno, new_revision_id
 
2928
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
1593
2929
        if self._real_branch is not None:
1594
2930
            cache = new_revno, new_revision_id
1595
2931
            self._real_branch._last_revision_info_cache = cache
1596
2932
 
1597
2933
    def _set_last_revision(self, revision_id):
 
2934
        old_revno, old_revid = self.last_revision_info()
 
2935
        # This performs additional work to meet the hook contract; while its
 
2936
        # undesirable, we have to synthesise the revno to call the hook, and
 
2937
        # not calling the hook is worse as it means changes can't be prevented.
 
2938
        # Having calculated this though, we can't just call into
 
2939
        # set_last_revision_info as a simple call, because there is a set_rh
 
2940
        # hook that some folk may still be using.
 
2941
        history = self._lefthand_history(revision_id)
 
2942
        self._run_pre_change_branch_tip_hooks(len(history), revision_id)
1598
2943
        self._clear_cached_state()
1599
2944
        response = self._call('Branch.set_last_revision',
1600
2945
            self._remote_path(), self._lock_token, self._repo_lock_token,
1601
2946
            revision_id)
1602
2947
        if response != ('ok',):
1603
2948
            raise errors.UnexpectedSmartServerResponse(response)
 
2949
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
1604
2950
 
 
2951
    @symbol_versioning.deprecated_method(symbol_versioning.deprecated_in((2, 4, 0)))
1605
2952
    @needs_write_lock
1606
2953
    def set_revision_history(self, rev_history):
 
2954
        """See Branch.set_revision_history."""
 
2955
        self._set_revision_history(rev_history)
 
2956
 
 
2957
    @needs_write_lock
 
2958
    def _set_revision_history(self, rev_history):
1607
2959
        # Send just the tip revision of the history; the server will generate
1608
2960
        # the full history from that.  If the revision doesn't exist in this
1609
2961
        # branch, NoSuchRevision will be raised.
1612
2964
        else:
1613
2965
            rev_id = rev_history[-1]
1614
2966
        self._set_last_revision(rev_id)
 
2967
        for hook in branch.Branch.hooks['set_rh']:
 
2968
            hook(self, rev_history)
1615
2969
        self._cache_revision_history(rev_history)
1616
2970
 
1617
 
    def get_parent(self):
1618
 
        self._ensure_real()
1619
 
        return self._real_branch.get_parent()
1620
 
        
1621
 
    def set_parent(self, url):
1622
 
        self._ensure_real()
1623
 
        return self._real_branch.set_parent(url)
1624
 
        
1625
 
    def set_stacked_on_url(self, stacked_location):
1626
 
        """Set the URL this branch is stacked against.
1627
 
 
1628
 
        :raises UnstackableBranchFormat: If the branch does not support
1629
 
            stacking.
1630
 
        :raises UnstackableRepositoryFormat: If the repository does not support
1631
 
            stacking.
1632
 
        """
1633
 
        self._ensure_real()
1634
 
        return self._real_branch.set_stacked_on_url(stacked_location)
1635
 
 
1636
 
    def sprout(self, to_bzrdir, revision_id=None):
1637
 
        branch_format = to_bzrdir._format._branch_format
1638
 
        if (branch_format is None or
1639
 
            isinstance(branch_format, RemoteBranchFormat)):
1640
 
            # The to_bzrdir specifies RemoteBranchFormat (or no format, which
1641
 
            # implies the same thing), but RemoteBranches can't be created at
1642
 
            # arbitrary URLs.  So create a branch in the same format as
1643
 
            # _real_branch instead.
1644
 
            # XXX: if to_bzrdir is a RemoteBzrDir, this should perhaps do
1645
 
            # to_bzrdir.create_branch to create a RemoteBranch after all...
1646
 
            self._ensure_real()
1647
 
            result = self._real_branch._format.initialize(to_bzrdir)
1648
 
            self.copy_content_into(result, revision_id=revision_id)
1649
 
            result.set_parent(self.bzrdir.root_transport.base)
1650
 
        else:
1651
 
            result = branch.Branch.sprout(
1652
 
                self, to_bzrdir, revision_id=revision_id)
1653
 
        return result
 
2971
    def _get_parent_location(self):
 
2972
        medium = self._client._medium
 
2973
        if medium._is_remote_before((1, 13)):
 
2974
            return self._vfs_get_parent_location()
 
2975
        try:
 
2976
            response = self._call('Branch.get_parent', self._remote_path())
 
2977
        except errors.UnknownSmartMethod:
 
2978
            medium._remember_remote_is_before((1, 13))
 
2979
            return self._vfs_get_parent_location()
 
2980
        if len(response) != 1:
 
2981
            raise errors.UnexpectedSmartServerResponse(response)
 
2982
        parent_location = response[0]
 
2983
        if parent_location == '':
 
2984
            return None
 
2985
        return parent_location
 
2986
 
 
2987
    def _vfs_get_parent_location(self):
 
2988
        self._ensure_real()
 
2989
        return self._real_branch._get_parent_location()
 
2990
 
 
2991
    def _set_parent_location(self, url):
 
2992
        medium = self._client._medium
 
2993
        if medium._is_remote_before((1, 15)):
 
2994
            return self._vfs_set_parent_location(url)
 
2995
        try:
 
2996
            call_url = url or ''
 
2997
            if type(call_url) is not str:
 
2998
                raise AssertionError('url must be a str or None (%s)' % url)
 
2999
            response = self._call('Branch.set_parent_location',
 
3000
                self._remote_path(), self._lock_token, self._repo_lock_token,
 
3001
                call_url)
 
3002
        except errors.UnknownSmartMethod:
 
3003
            medium._remember_remote_is_before((1, 15))
 
3004
            return self._vfs_set_parent_location(url)
 
3005
        if response != ():
 
3006
            raise errors.UnexpectedSmartServerResponse(response)
 
3007
 
 
3008
    def _vfs_set_parent_location(self, url):
 
3009
        self._ensure_real()
 
3010
        return self._real_branch._set_parent_location(url)
1654
3011
 
1655
3012
    @needs_write_lock
1656
3013
    def pull(self, source, overwrite=False, stop_revision=None,
1662
3019
            _override_hook_target=self, **kwargs)
1663
3020
 
1664
3021
    @needs_read_lock
1665
 
    def push(self, target, overwrite=False, stop_revision=None):
 
3022
    def push(self, target, overwrite=False, stop_revision=None, lossy=False):
1666
3023
        self._ensure_real()
1667
3024
        return self._real_branch.push(
1668
 
            target, overwrite=overwrite, stop_revision=stop_revision,
 
3025
            target, overwrite=overwrite, stop_revision=stop_revision, lossy=lossy,
1669
3026
            _override_hook_source_branch=self)
1670
3027
 
1671
3028
    def is_locked(self):
1678
3035
 
1679
3036
    @needs_write_lock
1680
3037
    def set_last_revision_info(self, revno, revision_id):
1681
 
        revision_id = ensure_null(revision_id)
 
3038
        # XXX: These should be returned by the set_last_revision_info verb
 
3039
        old_revno, old_revid = self.last_revision_info()
 
3040
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
 
3041
        if not revision_id or not isinstance(revision_id, basestring):
 
3042
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
1682
3043
        try:
1683
3044
            response = self._call('Branch.set_last_revision_info',
1684
3045
                self._remote_path(), self._lock_token, self._repo_lock_token,
1692
3053
        if response == ('ok',):
1693
3054
            self._clear_cached_state()
1694
3055
            self._last_revision_info_cache = revno, revision_id
 
3056
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
1695
3057
            # Update the _real_branch's cache too.
1696
3058
            if self._real_branch is not None:
1697
3059
                cache = self._last_revision_info_cache
1704
3066
                                  other_branch=None):
1705
3067
        medium = self._client._medium
1706
3068
        if not medium._is_remote_before((1, 6)):
 
3069
            # Use a smart method for 1.6 and above servers
1707
3070
            try:
1708
3071
                self._set_last_revision_descendant(revision_id, other_branch,
1709
3072
                    allow_diverged=True, allow_overwrite_descendant=True)
1711
3074
            except errors.UnknownSmartMethod:
1712
3075
                medium._remember_remote_is_before((1, 6))
1713
3076
        self._clear_cached_state_of_remote_branch_only()
1714
 
        self._ensure_real()
1715
 
        self._real_branch.generate_revision_history(
1716
 
            revision_id, last_rev=last_rev, other_branch=other_branch)
1717
 
 
1718
 
    @property
1719
 
    def tags(self):
1720
 
        self._ensure_real()
1721
 
        return self._real_branch.tags
 
3077
        self._set_revision_history(self._lefthand_history(revision_id,
 
3078
            last_rev=last_rev,other_branch=other_branch))
1722
3079
 
1723
3080
    def set_push_location(self, location):
1724
3081
        self._ensure_real()
1725
3082
        return self._real_branch.set_push_location(location)
1726
3083
 
1727
 
    @needs_write_lock
1728
 
    def update_revisions(self, other, stop_revision=None, overwrite=False,
1729
 
                         graph=None):
1730
 
        """See Branch.update_revisions."""
1731
 
        other.lock_read()
1732
 
        try:
1733
 
            if stop_revision is None:
1734
 
                stop_revision = other.last_revision()
1735
 
                if revision.is_null(stop_revision):
1736
 
                    # if there are no commits, we're done.
1737
 
                    return
1738
 
            self.fetch(other, stop_revision)
1739
 
 
1740
 
            if overwrite:
1741
 
                # Just unconditionally set the new revision.  We don't care if
1742
 
                # the branches have diverged.
1743
 
                self._set_last_revision(stop_revision)
1744
 
            else:
1745
 
                medium = self._client._medium
1746
 
                if not medium._is_remote_before((1, 6)):
1747
 
                    try:
1748
 
                        self._set_last_revision_descendant(stop_revision, other)
1749
 
                        return
1750
 
                    except errors.UnknownSmartMethod:
1751
 
                        medium._remember_remote_is_before((1, 6))
1752
 
                # Fallback for pre-1.6 servers: check for divergence
1753
 
                # client-side, then do _set_last_revision.
1754
 
                last_rev = revision.ensure_null(self.last_revision())
1755
 
                if graph is None:
1756
 
                    graph = self.repository.get_graph()
1757
 
                if self._check_if_descendant_or_diverged(
1758
 
                        stop_revision, last_rev, graph, other):
1759
 
                    # stop_revision is a descendant of last_rev, but we aren't
1760
 
                    # overwriting, so we're done.
1761
 
                    return
1762
 
                self._set_last_revision(stop_revision)
1763
 
        finally:
1764
 
            other.unlock()
 
3084
    def heads_to_fetch(self):
 
3085
        if self._format._use_default_local_heads_to_fetch():
 
3086
            # We recognise this format, and its heads-to-fetch implementation
 
3087
            # is the default one (tip + tags).  In this case it's cheaper to
 
3088
            # just use the default implementation rather than a special RPC as
 
3089
            # the tip and tags data is cached.
 
3090
            return branch.Branch.heads_to_fetch(self)
 
3091
        medium = self._client._medium
 
3092
        if medium._is_remote_before((2, 4)):
 
3093
            return self._vfs_heads_to_fetch()
 
3094
        try:
 
3095
            return self._rpc_heads_to_fetch()
 
3096
        except errors.UnknownSmartMethod:
 
3097
            medium._remember_remote_is_before((2, 4))
 
3098
            return self._vfs_heads_to_fetch()
 
3099
 
 
3100
    def _rpc_heads_to_fetch(self):
 
3101
        response = self._call('Branch.heads_to_fetch', self._remote_path())
 
3102
        if len(response) != 2:
 
3103
            raise errors.UnexpectedSmartServerResponse(response)
 
3104
        must_fetch, if_present_fetch = response
 
3105
        return set(must_fetch), set(if_present_fetch)
 
3106
 
 
3107
    def _vfs_heads_to_fetch(self):
 
3108
        self._ensure_real()
 
3109
        return self._real_branch.heads_to_fetch()
 
3110
 
 
3111
 
 
3112
class RemoteConfig(object):
 
3113
    """A Config that reads and writes from smart verbs.
 
3114
 
 
3115
    It is a low-level object that considers config data to be name/value pairs
 
3116
    that may be associated with a section. Assigning meaning to the these
 
3117
    values is done at higher levels like bzrlib.config.TreeConfig.
 
3118
    """
 
3119
 
 
3120
    def get_option(self, name, section=None, default=None):
 
3121
        """Return the value associated with a named option.
 
3122
 
 
3123
        :param name: The name of the value
 
3124
        :param section: The section the option is in (if any)
 
3125
        :param default: The value to return if the value is not set
 
3126
        :return: The value or default value
 
3127
        """
 
3128
        try:
 
3129
            configobj = self._get_configobj()
 
3130
            section_obj = None
 
3131
            if section is None:
 
3132
                section_obj = configobj
 
3133
            else:
 
3134
                try:
 
3135
                    section_obj = configobj[section]
 
3136
                except KeyError:
 
3137
                    pass
 
3138
            if section_obj is None:
 
3139
                value = default
 
3140
            else:
 
3141
                value = section_obj.get(name, default)
 
3142
        except errors.UnknownSmartMethod:
 
3143
            value = self._vfs_get_option(name, section, default)
 
3144
        for hook in config.OldConfigHooks['get']:
 
3145
            hook(self, name, value)
 
3146
        return value
 
3147
 
 
3148
    def _response_to_configobj(self, response):
 
3149
        if len(response[0]) and response[0][0] != 'ok':
 
3150
            raise errors.UnexpectedSmartServerResponse(response)
 
3151
        lines = response[1].read_body_bytes().splitlines()
 
3152
        conf = config.ConfigObj(lines, encoding='utf-8')
 
3153
        for hook in config.OldConfigHooks['load']:
 
3154
            hook(self)
 
3155
        return conf
 
3156
 
 
3157
 
 
3158
class RemoteBranchConfig(RemoteConfig):
 
3159
    """A RemoteConfig for Branches."""
 
3160
 
 
3161
    def __init__(self, branch):
 
3162
        self._branch = branch
 
3163
 
 
3164
    def _get_configobj(self):
 
3165
        path = self._branch._remote_path()
 
3166
        response = self._branch._client.call_expecting_body(
 
3167
            'Branch.get_config_file', path)
 
3168
        return self._response_to_configobj(response)
 
3169
 
 
3170
    def set_option(self, value, name, section=None):
 
3171
        """Set the value associated with a named option.
 
3172
 
 
3173
        :param value: The value to set
 
3174
        :param name: The name of the value to set
 
3175
        :param section: The section the option is in (if any)
 
3176
        """
 
3177
        medium = self._branch._client._medium
 
3178
        if medium._is_remote_before((1, 14)):
 
3179
            return self._vfs_set_option(value, name, section)
 
3180
        if isinstance(value, dict):
 
3181
            if medium._is_remote_before((2, 2)):
 
3182
                return self._vfs_set_option(value, name, section)
 
3183
            return self._set_config_option_dict(value, name, section)
 
3184
        else:
 
3185
            return self._set_config_option(value, name, section)
 
3186
 
 
3187
    def _set_config_option(self, value, name, section):
 
3188
        try:
 
3189
            path = self._branch._remote_path()
 
3190
            response = self._branch._client.call('Branch.set_config_option',
 
3191
                path, self._branch._lock_token, self._branch._repo_lock_token,
 
3192
                value.encode('utf8'), name, section or '')
 
3193
        except errors.UnknownSmartMethod:
 
3194
            medium = self._branch._client._medium
 
3195
            medium._remember_remote_is_before((1, 14))
 
3196
            return self._vfs_set_option(value, name, section)
 
3197
        if response != ():
 
3198
            raise errors.UnexpectedSmartServerResponse(response)
 
3199
 
 
3200
    def _serialize_option_dict(self, option_dict):
 
3201
        utf8_dict = {}
 
3202
        for key, value in option_dict.items():
 
3203
            if isinstance(key, unicode):
 
3204
                key = key.encode('utf8')
 
3205
            if isinstance(value, unicode):
 
3206
                value = value.encode('utf8')
 
3207
            utf8_dict[key] = value
 
3208
        return bencode.bencode(utf8_dict)
 
3209
 
 
3210
    def _set_config_option_dict(self, value, name, section):
 
3211
        try:
 
3212
            path = self._branch._remote_path()
 
3213
            serialised_dict = self._serialize_option_dict(value)
 
3214
            response = self._branch._client.call(
 
3215
                'Branch.set_config_option_dict',
 
3216
                path, self._branch._lock_token, self._branch._repo_lock_token,
 
3217
                serialised_dict, name, section or '')
 
3218
        except errors.UnknownSmartMethod:
 
3219
            medium = self._branch._client._medium
 
3220
            medium._remember_remote_is_before((2, 2))
 
3221
            return self._vfs_set_option(value, name, section)
 
3222
        if response != ():
 
3223
            raise errors.UnexpectedSmartServerResponse(response)
 
3224
 
 
3225
    def _real_object(self):
 
3226
        self._branch._ensure_real()
 
3227
        return self._branch._real_branch
 
3228
 
 
3229
    def _vfs_set_option(self, value, name, section=None):
 
3230
        return self._real_object()._get_config().set_option(
 
3231
            value, name, section)
 
3232
 
 
3233
 
 
3234
class RemoteBzrDirConfig(RemoteConfig):
 
3235
    """A RemoteConfig for BzrDirs."""
 
3236
 
 
3237
    def __init__(self, bzrdir):
 
3238
        self._bzrdir = bzrdir
 
3239
 
 
3240
    def _get_configobj(self):
 
3241
        medium = self._bzrdir._client._medium
 
3242
        verb = 'BzrDir.get_config_file'
 
3243
        if medium._is_remote_before((1, 15)):
 
3244
            raise errors.UnknownSmartMethod(verb)
 
3245
        path = self._bzrdir._path_for_remote_call(self._bzrdir._client)
 
3246
        response = self._bzrdir._call_expecting_body(
 
3247
            verb, path)
 
3248
        return self._response_to_configobj(response)
 
3249
 
 
3250
    def _vfs_get_option(self, name, section, default):
 
3251
        return self._real_object()._get_config().get_option(
 
3252
            name, section, default)
 
3253
 
 
3254
    def set_option(self, value, name, section=None):
 
3255
        """Set the value associated with a named option.
 
3256
 
 
3257
        :param value: The value to set
 
3258
        :param name: The name of the value to set
 
3259
        :param section: The section the option is in (if any)
 
3260
        """
 
3261
        return self._real_object()._get_config().set_option(
 
3262
            value, name, section)
 
3263
 
 
3264
    def _real_object(self):
 
3265
        self._bzrdir._ensure_real()
 
3266
        return self._bzrdir._real_bzrdir
 
3267
 
1765
3268
 
1766
3269
 
1767
3270
def _extract_tar(tar, to_dir):
1811
3314
        raise NoSuchRevision(find('branch'), err.error_args[0])
1812
3315
    elif err.error_verb == 'nosuchrevision':
1813
3316
        raise NoSuchRevision(find('repository'), err.error_args[0])
1814
 
    elif err.error_tuple == ('nobranch',):
1815
 
        raise errors.NotBranchError(path=find('bzrdir').root_transport.base)
 
3317
    elif err.error_verb == 'nobranch':
 
3318
        if len(err.error_args) >= 1:
 
3319
            extra = err.error_args[0]
 
3320
        else:
 
3321
            extra = None
 
3322
        raise errors.NotBranchError(path=find('bzrdir').root_transport.base,
 
3323
            detail=extra)
1816
3324
    elif err.error_verb == 'norepository':
1817
3325
        raise errors.NoRepositoryPresent(find('bzrdir'))
1818
 
    elif err.error_verb == 'LockContention':
1819
 
        raise errors.LockContention('(remote lock)')
1820
3326
    elif err.error_verb == 'UnlockableTransport':
1821
3327
        raise errors.UnlockableTransport(find('bzrdir').root_transport)
1822
 
    elif err.error_verb == 'LockFailed':
1823
 
        raise errors.LockFailed(err.error_args[0], err.error_args[1])
1824
3328
    elif err.error_verb == 'TokenMismatch':
1825
3329
        raise errors.TokenMismatch(find('token'), '(remote token)')
1826
3330
    elif err.error_verb == 'Diverged':
1827
3331
        raise errors.DivergedBranches(find('branch'), find('other_branch'))
1828
 
    elif err.error_verb == 'TipChangeRejected':
1829
 
        raise errors.TipChangeRejected(err.error_args[0].decode('utf8'))
1830
 
    elif err.error_verb == 'UnstackableBranchFormat':
1831
 
        raise errors.UnstackableBranchFormat(*err.error_args)
1832
 
    elif err.error_verb == 'UnstackableRepositoryFormat':
1833
 
        raise errors.UnstackableRepositoryFormat(*err.error_args)
1834
3332
    elif err.error_verb == 'NotStacked':
1835
3333
        raise errors.NotStacked(branch=find('branch'))
1836
3334
    elif err.error_verb == 'PermissionDenied':
1846
3344
    elif err.error_verb == 'NoSuchFile':
1847
3345
        path = get_path()
1848
3346
        raise errors.NoSuchFile(path)
 
3347
    _translate_error_without_context(err)
 
3348
 
 
3349
 
 
3350
def _translate_error_without_context(err):
 
3351
    """Translate any ErrorFromSmartServer values that don't require context"""
 
3352
    if err.error_verb == 'IncompatibleRepositories':
 
3353
        raise errors.IncompatibleRepositories(err.error_args[0],
 
3354
            err.error_args[1], err.error_args[2])
 
3355
    elif err.error_verb == 'LockContention':
 
3356
        raise errors.LockContention('(remote lock)')
 
3357
    elif err.error_verb == 'LockFailed':
 
3358
        raise errors.LockFailed(err.error_args[0], err.error_args[1])
 
3359
    elif err.error_verb == 'TipChangeRejected':
 
3360
        raise errors.TipChangeRejected(err.error_args[0].decode('utf8'))
 
3361
    elif err.error_verb == 'UnstackableBranchFormat':
 
3362
        raise errors.UnstackableBranchFormat(*err.error_args)
 
3363
    elif err.error_verb == 'UnstackableRepositoryFormat':
 
3364
        raise errors.UnstackableRepositoryFormat(*err.error_args)
1849
3365
    elif err.error_verb == 'FileExists':
1850
3366
        raise errors.FileExists(err.error_args[0])
1851
3367
    elif err.error_verb == 'DirectoryNotEmpty':
1870
3386
            raise UnicodeEncodeError(encoding, val, start, end, reason)
1871
3387
    elif err.error_verb == 'ReadOnlyError':
1872
3388
        raise errors.TransportNotPossible('readonly transport')
 
3389
    elif err.error_verb == 'MemoryError':
 
3390
        raise errors.BzrError("remote server out of memory\n"
 
3391
            "Retry non-remotely, or contact the server admin for details.")
1873
3392
    raise errors.UnknownErrorFromSmartServer(err)