~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2006-2010 Canonical Ltd

    controldir,

    repository as _mod_repository,

    static_tuple,

from bzrlib.branch import BranchReferenceFormat, BranchWriteLockResult

from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises

from bzrlib.repository import RepositoryWriteLockResult

    def __init__(self, transport, format, _client=None, _force_probe=False):

        self._has_working_tree = None

            if not _force_probe:

                return

 

        self._probe_bzrdir()

 

    def __repr__(self):

        return '%s(%r)' % (self.__class__.__name__, self._client)

 

    def _probe_bzrdir(self):

        medium = self._client._medium

        if medium._is_remote_before((2, 1)):

            self._rpc_open(path)

            return

        try:

            self._rpc_open_2_1(path)

            return

        except errors.UnknownSmartMethod:

            medium._remember_remote_is_before((2, 1))

            self._rpc_open(path)

 

    def _rpc_open_2_1(self, path):

        response = self._call('BzrDir.open_2.1', path)

        if response == ('no',):

            raise errors.NotBranchError(path=self.root_transport.base)

        elif response[0] == 'yes':

            if response[1] == 'yes':

                self._has_working_tree = True

            elif response[1] == 'no':

                self._has_working_tree = False

            else:

                raise errors.UnexpectedSmartServerResponse(response)

        else:

            raise errors.UnexpectedSmartServerResponse(response)

 

    def _rpc_open(self, path):

            raise errors.NotBranchError(path=self.root_transport.base)

            if 'hpssvfs' in debug.debug_flags:

                import traceback

                warning('VFS BzrDir access triggered\n%s',

                    ''.join(traceback.format_stack()))

    def create_branch(self, name=None):

        real_branch = self._format.get_branch_format().initialize(self,

            name=name)

            result = RemoteBranch(self, self.find_repository(), real_branch,

                                  name=name)

    def destroy_branch(self, name=None):

        self._real_bzrdir.destroy_branch(name=name)

    def find_branch_format(self, name=None):

        b = self.open_branch(name=name)

    def get_branch_reference(self, name=None):

        if name is not None:

            # XXX JRV20100304: Support opening colocated branches

            raise errors.NoColocatedBranchSupport(self)

        candidate_calls = [

            ('BzrDir.open_branchV3', (2, 1)),

            ('BzrDir.open_branchV2', (1, 13)),

            ('BzrDir.open_branch', None),

            ]

        for verb, required_version in candidate_calls:

            if required_version and medium._is_remote_before(required_version):

                continue

                response = self._call(verb, path)

                if required_version is None:

                    raise

                medium._remember_remote_is_before(required_version)

            else:

                break

        if verb == 'BzrDir.open_branch':

            if response[0] != 'ok':

                raise errors.UnexpectedSmartServerResponse(response)

            if response[1] != '':

                return ('ref', response[1])

            else:

                return ('branch', '')

        if response[0] not in ('ref', 'branch'):

        return response

    def _get_tree_branch(self, name=None):

        return None, self.open_branch(name=name)

    def open_branch(self, name=None, unsupported=False,

                    ignore_fallbacks=False):

        if unsupported:

            return format.open(self, name=name, _found=True,

                location=response[1], ignore_fallbacks=ignore_fallbacks)

            setup_stacking=not ignore_fallbacks, name=name)

    def has_workingtree(self):

        if self._has_working_tree is None:

            self._ensure_real()

            self._has_working_tree = self._real_bzrdir.has_workingtree()

        return self._has_working_tree

 

        if self.has_workingtree():

    def get_branch_transport(self, branch_format, name=None):

        return self._real_bzrdir.get_branch_transport(branch_format, name=name)

        self._supports_chks = None

    def __repr__(self):

        return "%s(_network_name=%r)" % (self.__class__.__name__,

            self._network_name)

 

    def supports_chks(self):

        if self._supports_chks is None:

            self._ensure_real()

            self._supports_chks = self._custom_format.supports_chks

        return self._supports_chks

 

    @property

        self._ensure_real()

        return 'Remote: ' + self._custom_format.get_format_description()

class RemoteRepository(_RpcHelper, lock._RelockDebugMixin,

    controldir.ControlComponent):

    @property

    def user_transport(self):

        return self.bzrdir.user_transport

 

    @property

    def control_transport(self):

        # XXX: Normally you shouldn't directly get at the remote repository

        # transport, but I'm not sure it's worth making this method

        # optional -- mbp 2010-04-21

        return self.bzrdir.get_repository_transport(None)

        

        # <https://bugs.launchpad.net/bzr/+bug/401622>

    @needs_read_lock

    def get_known_graph_ancestry(self, revision_ids):

        """Return the known graph for a set of revision ids and their ancestors.

        """

        st = static_tuple.StaticTuple

        revision_keys = [st(r_id).intern() for r_id in revision_ids]

        known_graph = self.revisions.get_known_graph_ancestry(revision_keys)

        return graph.GraphThunkIdsToKeys(known_graph)

 

    def _warn_if_deprecated(self, branch=None):

        # If we have a real repository, the check will be done there, if we

        # don't the check will be done remotely.

        pass

 

        """Lock the repository for read operations.

 

        :return: A bzrlib.lock.LogicalLockResult.

        """

            self._note_lock('r')

        return lock.LogicalLockResult(self.unlock)

            self._note_lock('w')

        return RepositoryWriteLockResult(self.unlock, self._lock_token or None)

    @only_raises(errors.LockNotHeld, errors.LockBroken)

        self._check_fallback_repository(repository)

            fallback_locations = [repo.user_url for repo in

            if repository.user_url not in fallback_locations:

    def _check_fallback_repository(self, repository):

        """Check that this repository can fallback to repository safely.

 

        Raise an error if not.

 

        :param repository: A repository to fallback to.

        """

        return _mod_repository.InterRepository._assert_same_model(

            self, repository)

 

            parents, basis_inv=None, propagate_caches=False):

            delta, new_revision_id, parents, basis_inv=basis_inv,

            propagate_caches=propagate_caches)

    def iter_inventories(self, revision_ids, ordering=None):

        return self._real_repository.iter_inventories(revision_ids, ordering)

        """Re-read any data needed to synchronise with disk.

        repository. On all repositories this will work outside of write groups.

        Some repository formats (pack and newer for bzrlib native formats)

        support refresh_data inside write groups. If called inside a write

        group on a repository that does not support refreshing in a write group

        IsInWriteGroupError will be raised.

    def _get_inventory_xml(self, revision_id):

        self._ensure_real()

        return self._real_repository._get_inventory_xml(revision_id)

    def pack(self, hint=None, clean_obsolete_packs=False):

        return self._real_repository.pack(hint=hint, clean_obsolete_packs=clean_obsolete_packs)

        candidate_calls = [('Repository.insert_stream_1.19', (1, 19))]

            candidate_calls.append(('Repository.insert_stream_locked', (1, 14)))

            lock_args = (target._lock_token or '',)

            candidate_calls.append(('Repository.insert_stream', (1, 13)))

            lock_args = ()

        # Probe for the verb to use with an empty stream before sending the

        # real stream to it.  We do this both to avoid the risk of sending a

        # large request that is then rejected, and because we don't want to

        # implement a way to buffer, rewind, or restart the stream.

        found_verb = False

        for verb, required_version in candidate_calls:

            if medium._is_remote_before(required_version):

                continue

            if resume_tokens:

                # We've already done the probing (and set _is_remote_before) on

                # a previous insert.

                found_verb = True

                break

                    (verb, path, '') + lock_args, byte_stream)

            else:

                found_verb = True

                break

        if not found_verb:

            # Have to use VFS.

            return self._insert_real(stream, src_format, resume_tokens)

        self._last_inv_record = None

        self._last_substream = None

        if required_version < (1, 19):

            # Remote side doesn't support inventory deltas.  Wrap the stream to

            # make sure we don't send any.  If the stream contains inventory

            # deltas we'll interrupt the smart insert_stream request and

            # fallback to VFS.

            stream = self._stop_stream_if_inventory_delta(stream)

            (verb, path, resume_tokens) + lock_args, byte_stream)

        if self._last_substream is not None:

            # The stream included an inventory-delta record, but the remote

            # side isn't new enough to support them.  So we need to send the

            # rest of the stream via VFS.

            self.target_repo.refresh_data()

            return self._resume_stream_with_vfs(response, src_format)

    def _resume_stream_with_vfs(self, response, src_format):

        """Resume sending a stream via VFS, first resending the record and

        substream that couldn't be sent via an insert_stream verb.

        """

        if response[0][0] == 'missing-basis':

            tokens, missing_keys = bencode.bdecode_as_tuple(response[0][1])

            # Ignore missing_keys, we haven't finished inserting yet

        else:

            tokens = []

        def resume_substream():

            # Yield the substream that was interrupted.

            for record in self._last_substream:

                yield record

            self._last_substream = None

        def resume_stream():

            # Finish sending the interrupted substream

            yield ('inventory-deltas', resume_substream())

            # Then simply continue sending the rest of the stream.

            for substream_kind, substream in self._last_stream:

                yield substream_kind, substream

        return self._insert_real(resume_stream(), src_format, tokens)

 

    def _stop_stream_if_inventory_delta(self, stream):

        """Normally this just lets the original stream pass-through unchanged.

 

        However if any 'inventory-deltas' substream occurs it will stop

        streaming, and store the interrupted substream and stream in

        self._last_substream and self._last_stream so that the stream can be

        resumed by _resume_stream_with_vfs.

        """

                    

        stream_iter = iter(stream)

        for substream_kind, substream in stream_iter:

            if substream_kind == 'inventory-deltas':

                self._last_substream = substream

                self._last_stream = stream_iter

                return

            else:

                yield substream_kind, substream

            

    def get_stream_for_missing_keys(self, missing_keys):

        self.from_repository._ensure_real()

        real_repo = self.from_repository._real_repository

        real_source = real_repo._get_source(self.to_format)

        return real_source.get_stream_for_missing_keys(missing_keys)

 

        search_bytes = repo._serialise_search_result(search)

        args = (path, self.to_format.network_name())

        candidate_verbs = [

            ('Repository.get_stream_1.19', (1, 19)),

            ('Repository.get_stream', (1, 13))]

        found_verb = False

        for verb, version in candidate_verbs:

            if medium._is_remote_before(version):

                continue

            try:

                response = repo._call_with_body_bytes_expecting_body(

                    verb, args, search_bytes)

            except errors.UnknownSmartMethod:

                medium._remember_remote_is_before(version)

            else:

                response_tuple, response_handler = response

                found_verb = True

                break

        if not found_verb:

        src_format, stream = smart_repo._byte_stream_to_stream(byte_stream,

            self._record_counter)

        self.from_serialiser = self.from_repository._format._serializer

            revision = self.from_serialiser.read_revision_from_string(

                revision_bytes)

        self._ensure_real()

        return 'Remote: ' + self._custom_format.get_format_description()

    def open(self, a_bzrdir, name=None, ignore_fallbacks=False):

        return a_bzrdir.open_branch(name=name, 

            ignore_fallbacks=ignore_fallbacks)

    def _vfs_initialize(self, a_bzrdir, name):

            result = self._custom_format.initialize(a_bzrdir._real_bzrdir,

                name)

            result = self._custom_format.initialize(a_bzrdir, name)

            result = RemoteBranch(a_bzrdir, a_bzrdir.find_repository(), result,

                                  name=name)

    def initialize(self, a_bzrdir, name=None):

            return self._vfs_initialize(a_bzrdir, name=name)

            return self._vfs_initialize(a_bzrdir, name=name)

        if name is not None:

            # XXX JRV20100304: Support creating colocated branches

            raise errors.NoColocatedBranchSupport(self)

            return self._vfs_initialize(a_bzrdir, name=name)

            format=format, setup_stacking=False, name=name)

class RemoteBranch(branch.Branch, _RpcHelper, lock._RelockDebugMixin):

        _client=None, format=None, setup_stacking=True, name=None):

        :param name: Colocated branch name

        # TODO: deprecate self.base in favor of user_url

        self.base = self.bzrdir.user_url

        self._name = name

        # when we do _ensure_real we may need to pass ignore_fallbacks to the

        # branch.open_branch method.

        self._real_ignore_fallbacks = not setup_stacking

            self._real_branch = self.bzrdir._real_bzrdir.open_branch(

                ignore_fallbacks=self._real_ignore_fallbacks, name=self._name)

            return

        """Lock the branch for read operations.

 

        :return: A bzrlib.lock.LogicalLockResult.

        """

            self._note_lock('r')

        return lock.LogicalLockResult(self.unlock)

            repo_token = self.repository.lock_write().repository_token

        try:

            response = self._call(

                'Branch.lock_write', self._remote_path(), branch_token,

                repo_token or '', **err_context)

        except errors.LockContention, e:

            # The LockContention from the server doesn't have any

            # information about the lock_url. We re-raise LockContention

            # with valid lock_url.

            raise errors.LockContention('(remote lock)',

                self.repository.base.split('.bzr/')[0])

            self._note_lock('w')

            raise errors.ReadOnlyError(self)

        return BranchWriteLockResult(self.unlock, self._lock_token or None)

    @only_raises(errors.LockNotHeld, errors.LockBroken)

    @needs_read_lock

        if isinstance(value, dict):

            if medium._is_remote_before((2, 2)):

                return self._vfs_set_option(value, name, section)

            return self._set_config_option_dict(value, name, section)

        else:

            return self._set_config_option(value, name, section)

 

    def _set_config_option(self, value, name, section):

            medium = self._branch._client._medium

    def _serialize_option_dict(self, option_dict):

        utf8_dict = {}

        for key, value in option_dict.items():

            if isinstance(key, unicode):

                key = key.encode('utf8')

            if isinstance(value, unicode):

                value = value.encode('utf8')