~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/remote.py

merge bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
2152
2152
        repo_format = response_tuple_to_repo_format(response[3:])
2153
2153
        repo_path = response[2]
2154
2154
        if repository is not None:
2155
 
            remote_repo_url = urlutils.join(medium.base, repo_path)
 
2155
            remote_repo_url = urlutils.join(a_bzrdir.user_url, repo_path)
2156
2156
            url_diff = urlutils.relative_url(repository.user_url,
2157
2157
                    remote_repo_url)
2158
2158
            if url_diff != '.':
2159
2159
                raise AssertionError(
2160
2160
                    'repository.user_url %r does not match URL from server '
2161
2161
                    'response (%r + %r)'
2162
 
                    % (repository.user_url, medium.base, repo_path))
 
2162
                    % (repository.user_url, a_bzrdir.user_url, repo_path))
2163
2163
            remote_repo = repository
2164
2164
        else:
2165
2165
            if repo_path == '':
2195
2195
        self._ensure_real()
2196
2196
        return self._custom_format.supports_set_append_revisions_only()
2197
2197
 
 
2198
    def _use_default_local_heads_to_fetch(self):
 
2199
        # If the branch format is a metadir format *and* its heads_to_fetch
 
2200
        # implementation is not overridden vs the base class, we can use the
 
2201
        # base class logic rather than use the heads_to_fetch RPC.  This is
 
2202
        # usually cheaper in terms of net round trips, as the last-revision and
 
2203
        # tags info fetched is cached and would be fetched anyway.
 
2204
        self._ensure_real()
 
2205
        if isinstance(self._custom_format, branch.BranchFormatMetadir):
 
2206
            branch_class = self._custom_format._branch_class()
 
2207
            heads_to_fetch_impl = branch_class.heads_to_fetch.im_func
 
2208
            if heads_to_fetch_impl is branch.Branch.heads_to_fetch.im_func:
 
2209
                return True
 
2210
        return False
2198
2211
 
2199
2212
class RemoteBranch(branch.Branch, _RpcHelper, lock._RelockDebugMixin):
2200
2213
    """Branch stored on a server accessed by HPSS RPC.
2782
2795
        self._ensure_real()
2783
2796
        return self._real_branch.set_push_location(location)
2784
2797
 
 
2798
    def heads_to_fetch(self):
 
2799
        if self._format._use_default_local_heads_to_fetch():
 
2800
            # We recognise this format, and its heads-to-fetch implementation
 
2801
            # is the default one (tip + tags).  In this case it's cheaper to
 
2802
            # just use the default implementation rather than a special RPC as
 
2803
            # the tip and tags data is cached.
 
2804
            return branch.Branch.heads_to_fetch(self)
 
2805
        medium = self._client._medium
 
2806
        if medium._is_remote_before((2, 4)):
 
2807
            return self._vfs_heads_to_fetch()
 
2808
        try:
 
2809
            return self._rpc_heads_to_fetch()
 
2810
        except errors.UnknownSmartMethod:
 
2811
            medium._remember_remote_is_before((2, 4))
 
2812
            return self._vfs_heads_to_fetch()
 
2813
 
 
2814
    def _rpc_heads_to_fetch(self):
 
2815
        response = self._call('Branch.heads_to_fetch', self._remote_path())
 
2816
        if len(response) != 2:
 
2817
            raise errors.UnexpectedSmartServerResponse(response)
 
2818
        must_fetch, if_present_fetch = response
 
2819
        return set(must_fetch), set(if_present_fetch)
 
2820
 
 
2821
    def _vfs_heads_to_fetch(self):
 
2822
        self._ensure_real()
 
2823
        return self._real_branch.heads_to_fetch()
 
2824
 
2785
2825
 
2786
2826
class RemoteConfig(object):
2787
2827
    """A Config that reads and writes from smart verbs.
2974
3014
                    'Missing key %r in context %r', key_err.args[0], context)
2975
3015
                raise err
2976
3016
 
2977
 
    if err.error_verb == 'IncompatibleRepositories':
2978
 
        raise errors.IncompatibleRepositories(err.error_args[0],
2979
 
            err.error_args[1], err.error_args[2])
2980
 
    elif err.error_verb == 'NoSuchRevision':
 
3017
    if err.error_verb == 'NoSuchRevision':
2981
3018
        raise NoSuchRevision(find('branch'), err.error_args[0])
2982
3019
    elif err.error_verb == 'nosuchrevision':
2983
3020
        raise NoSuchRevision(find('repository'), err.error_args[0])
2990
3027
            detail=extra)
2991
3028
    elif err.error_verb == 'norepository':
2992
3029
        raise errors.NoRepositoryPresent(find('bzrdir'))
2993
 
    elif err.error_verb == 'LockContention':
2994
 
        raise errors.LockContention('(remote lock)')
2995
3030
    elif err.error_verb == 'UnlockableTransport':
2996
3031
        raise errors.UnlockableTransport(find('bzrdir').root_transport)
2997
 
    elif err.error_verb == 'LockFailed':
2998
 
        raise errors.LockFailed(err.error_args[0], err.error_args[1])
2999
3032
    elif err.error_verb == 'TokenMismatch':
3000
3033
        raise errors.TokenMismatch(find('token'), '(remote token)')
3001
3034
    elif err.error_verb == 'Diverged':
3002
3035
        raise errors.DivergedBranches(find('branch'), find('other_branch'))
3003
 
    elif err.error_verb == 'TipChangeRejected':
3004
 
        raise errors.TipChangeRejected(err.error_args[0].decode('utf8'))
3005
 
    elif err.error_verb == 'UnstackableBranchFormat':
3006
 
        raise errors.UnstackableBranchFormat(*err.error_args)
3007
 
    elif err.error_verb == 'UnstackableRepositoryFormat':
3008
 
        raise errors.UnstackableRepositoryFormat(*err.error_args)
3009
3036
    elif err.error_verb == 'NotStacked':
3010
3037
        raise errors.NotStacked(branch=find('branch'))
3011
3038
    elif err.error_verb == 'PermissionDenied':
3021
3048
    elif err.error_verb == 'NoSuchFile':
3022
3049
        path = get_path()
3023
3050
        raise errors.NoSuchFile(path)
 
3051
    _translate_error_without_context(err)
 
3052
 
 
3053
 
 
3054
def _translate_error_without_context(err):
 
3055
    """Translate any ErrorFromSmartServer values that don't require context"""
 
3056
    if err.error_verb == 'IncompatibleRepositories':
 
3057
        raise errors.IncompatibleRepositories(err.error_args[0],
 
3058
            err.error_args[1], err.error_args[2])
 
3059
    elif err.error_verb == 'LockContention':
 
3060
        raise errors.LockContention('(remote lock)')
 
3061
    elif err.error_verb == 'LockFailed':
 
3062
        raise errors.LockFailed(err.error_args[0], err.error_args[1])
 
3063
    elif err.error_verb == 'TipChangeRejected':
 
3064
        raise errors.TipChangeRejected(err.error_args[0].decode('utf8'))
 
3065
    elif err.error_verb == 'UnstackableBranchFormat':
 
3066
        raise errors.UnstackableBranchFormat(*err.error_args)
 
3067
    elif err.error_verb == 'UnstackableRepositoryFormat':
 
3068
        raise errors.UnstackableRepositoryFormat(*err.error_args)
3024
3069
    elif err.error_verb == 'FileExists':
3025
3070
        raise errors.FileExists(err.error_args[0])
3026
3071
    elif err.error_verb == 'DirectoryNotEmpty':
3045
3090
            raise UnicodeEncodeError(encoding, val, start, end, reason)
3046
3091
    elif err.error_verb == 'ReadOnlyError':
3047
3092
        raise errors.TransportNotPossible('readonly transport')
 
3093
    elif err.error_verb == 'MemoryError':
 
3094
        raise errors.BzrError("remote server out of memory\n"
 
3095
            "Retry non-remotely, or contact the server admin for details.")
3048
3096
    raise errors.UnknownErrorFromSmartServer(err)