~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Patch Queue Manager
  • Date: 2013-10-07 17:04:34 UTC
  • mfrom: (6588.1.1 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20131007170434-mb0ahksmrzsnhi1i
(vila) Stricter checks on configuration option names (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005-2012 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
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
from __future__ import absolute_import
 
18
 
 
19
import bzrlib.bzrdir
17
20
 
18
21
from cStringIO import StringIO
19
 
import sys
20
22
 
21
23
from bzrlib.lazy_import import lazy_import
22
24
lazy_import(globals(), """
23
 
from itertools import chain
 
25
import itertools
24
26
from bzrlib import (
25
 
        bzrdir,
26
 
        cache_utf8,
27
 
        config as _mod_config,
28
 
        controldir,
29
 
        debug,
30
 
        errors,
31
 
        fetch,
32
 
        graph as _mod_graph,
33
 
        lockdir,
34
 
        lockable_files,
35
 
        remote,
36
 
        repository,
37
 
        revision as _mod_revision,
38
 
        rio,
39
 
        symbol_versioning,
40
 
        transport,
41
 
        tsort,
42
 
        ui,
43
 
        urlutils,
44
 
        )
45
 
from bzrlib.config import BranchConfig, TransportConfig
46
 
from bzrlib.repofmt.pack_repo import RepositoryFormatKnitPack5RichRoot
47
 
from bzrlib.tag import (
48
 
    BasicTags,
49
 
    DisabledTags,
 
27
    bzrdir,
 
28
    controldir,
 
29
    cache_utf8,
 
30
    cleanup,
 
31
    config as _mod_config,
 
32
    debug,
 
33
    errors,
 
34
    fetch,
 
35
    graph as _mod_graph,
 
36
    lockdir,
 
37
    lockable_files,
 
38
    remote,
 
39
    repository,
 
40
    revision as _mod_revision,
 
41
    rio,
 
42
    shelf,
 
43
    tag as _mod_tag,
 
44
    transport,
 
45
    ui,
 
46
    urlutils,
 
47
    vf_search,
50
48
    )
 
49
from bzrlib.i18n import gettext, ngettext
51
50
""")
52
51
 
53
 
from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises
54
 
from bzrlib.hooks import HookPoint, Hooks
 
52
# Explicitly import bzrlib.bzrdir so that the BzrProber
 
53
# is guaranteed to be registered.
 
54
import bzrlib.bzrdir
 
55
 
 
56
from bzrlib import (
 
57
    bzrdir,
 
58
    controldir,
 
59
    )
 
60
from bzrlib.decorators import (
 
61
    needs_read_lock,
 
62
    needs_write_lock,
 
63
    only_raises,
 
64
    )
 
65
from bzrlib.hooks import Hooks
55
66
from bzrlib.inter import InterObject
56
67
from bzrlib.lock import _RelockDebugMixin, LogicalLockResult
57
68
from bzrlib import registry
62
73
from bzrlib.trace import mutter, mutter_callsite, note, is_quiet
63
74
 
64
75
 
65
 
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
66
 
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
67
 
BZR_BRANCH_FORMAT_6 = "Bazaar Branch Format 6 (bzr 0.15)\n"
68
 
 
69
 
 
70
76
class Branch(controldir.ControlComponent):
71
77
    """Branch holding a history of revisions.
72
78
 
73
79
    :ivar base:
74
80
        Base directory/url of the branch; using control_url and
75
81
        control_transport is more standardized.
76
 
 
77
 
    hooks: An instance of BranchHooks.
 
82
    :ivar hooks: An instance of BranchHooks.
 
83
    :ivar _master_branch_cache: cached result of get_master_branch, see
 
84
        _clear_cached_state.
78
85
    """
79
86
    # this is really an instance variable - FIXME move it there
80
87
    # - RBC 20060112
88
95
    def user_transport(self):
89
96
        return self.bzrdir.user_transport
90
97
 
91
 
    def __init__(self, *ignored, **ignored_too):
 
98
    def __init__(self, possible_transports=None):
92
99
        self.tags = self._format.make_tags(self)
93
100
        self._revision_history_cache = None
94
101
        self._revision_id_to_revno_cache = None
96
103
        self._partial_revision_history_cache = []
97
104
        self._tags_bytes = None
98
105
        self._last_revision_info_cache = None
 
106
        self._master_branch_cache = None
99
107
        self._merge_sorted_revisions_cache = None
100
 
        self._open_hook()
 
108
        self._open_hook(possible_transports)
101
109
        hooks = Branch.hooks['open']
102
110
        for hook in hooks:
103
111
            hook(self)
104
112
 
105
 
    def _open_hook(self):
 
113
    def _open_hook(self, possible_transports):
106
114
        """Called by init to allow simpler extension of the base class."""
107
115
 
108
 
    def _activate_fallback_location(self, url):
 
116
    def _activate_fallback_location(self, url, possible_transports):
109
117
        """Activate the branch/repository from url as a fallback repository."""
110
118
        for existing_fallback_repo in self.repository._fallback_repositories:
111
119
            if existing_fallback_repo.user_url == url:
112
120
                # This fallback is already configured.  This probably only
113
 
                # happens because BzrDir.sprout is a horrible mess.  To avoid
 
121
                # happens because ControlDir.sprout is a horrible mess.  To avoid
114
122
                # confusing _unstack we don't add this a second time.
115
123
                mutter('duplicate activation of fallback %r on %r', url, self)
116
124
                return
117
 
        repo = self._get_fallback_repository(url)
 
125
        repo = self._get_fallback_repository(url, possible_transports)
118
126
        if repo.has_same_location(self.repository):
119
127
            raise errors.UnstackableLocationError(self.user_url, url)
120
128
        self.repository.add_fallback_repository(repo)
174
182
        For instance, if the branch is at URL/.bzr/branch,
175
183
        Branch.open(URL) -> a Branch instance.
176
184
        """
177
 
        control = bzrdir.BzrDir.open(base, _unsupported,
178
 
                                     possible_transports=possible_transports)
179
 
        return control.open_branch(unsupported=_unsupported)
 
185
        control = controldir.ControlDir.open(base,
 
186
            possible_transports=possible_transports, _unsupported=_unsupported)
 
187
        return control.open_branch(unsupported=_unsupported,
 
188
            possible_transports=possible_transports)
180
189
 
181
190
    @staticmethod
182
 
    def open_from_transport(transport, name=None, _unsupported=False):
 
191
    def open_from_transport(transport, name=None, _unsupported=False,
 
192
            possible_transports=None):
183
193
        """Open the branch rooted at transport"""
184
 
        control = bzrdir.BzrDir.open_from_transport(transport, _unsupported)
185
 
        return control.open_branch(name=name, unsupported=_unsupported)
 
194
        control = controldir.ControlDir.open_from_transport(transport, _unsupported)
 
195
        return control.open_branch(name=name, unsupported=_unsupported,
 
196
            possible_transports=possible_transports)
186
197
 
187
198
    @staticmethod
188
199
    def open_containing(url, possible_transports=None):
196
207
        format, UnknownFormatError or UnsupportedFormatError are raised.
197
208
        If there is one, it is returned, along with the unused portion of url.
198
209
        """
199
 
        control, relpath = bzrdir.BzrDir.open_containing(url,
 
210
        control, relpath = controldir.ControlDir.open_containing(url,
200
211
                                                         possible_transports)
201
 
        return control.open_branch(), relpath
 
212
        branch = control.open_branch(possible_transports=possible_transports)
 
213
        return (branch, relpath)
202
214
 
203
215
    def _push_should_merge_tags(self):
204
216
        """Should _basic_push merge this branch's tags into the target?
216
228
 
217
229
        :return: A bzrlib.config.BranchConfig.
218
230
        """
219
 
        return BranchConfig(self)
 
231
        return _mod_config.BranchConfig(self)
 
232
 
 
233
    def get_config_stack(self):
 
234
        """Get a bzrlib.config.BranchStack for this Branch.
 
235
 
 
236
        This can then be used to get and set configuration options for the
 
237
        branch.
 
238
 
 
239
        :return: A bzrlib.config.BranchStack.
 
240
        """
 
241
        return _mod_config.BranchStack(self)
220
242
 
221
243
    def _get_config(self):
222
244
        """Get the concrete config for just the config in this branch.
230
252
        """
231
253
        raise NotImplementedError(self._get_config)
232
254
 
233
 
    def _get_fallback_repository(self, url):
 
255
    def store_uncommitted(self, creator):
 
256
        """Store uncommitted changes from a ShelfCreator.
 
257
 
 
258
        :param creator: The ShelfCreator containing uncommitted changes, or
 
259
            None to delete any stored changes.
 
260
        :raises: ChangesAlreadyStored if the branch already has changes.
 
261
        """
 
262
        raise NotImplementedError(self.store_uncommitted)
 
263
 
 
264
    def get_unshelver(self, tree):
 
265
        """Return a shelf.Unshelver for this branch and tree.
 
266
 
 
267
        :param tree: The tree to use to construct the Unshelver.
 
268
        :return: an Unshelver or None if no changes are stored.
 
269
        """
 
270
        raise NotImplementedError(self.get_unshelver)
 
271
 
 
272
    def _get_fallback_repository(self, url, possible_transports):
234
273
        """Get the repository we fallback to at url."""
235
274
        url = urlutils.join(self.base, url)
236
 
        a_branch = Branch.open(url,
237
 
            possible_transports=[self.bzrdir.root_transport])
 
275
        a_branch = Branch.open(url, possible_transports=possible_transports)
238
276
        return a_branch.repository
239
277
 
240
278
    @needs_read_lock
449
487
            after. If None, the rest of history is included.
450
488
        :param stop_rule: if stop_revision_id is not None, the precise rule
451
489
            to use for termination:
 
490
 
452
491
            * 'exclude' - leave the stop revision out of the result (default)
453
492
            * 'include' - the stop revision is the last item in the result
454
493
            * 'with-merges' - include the stop revision and all of its
456
495
            * 'with-merges-without-common-ancestry' - filter out revisions 
457
496
              that are in both ancestries
458
497
        :param direction: either 'reverse' or 'forward':
 
498
 
459
499
            * reverse means return the start_revision_id first, i.e.
460
500
              start at the most recent revision and go backwards in history
461
501
            * forward returns tuples in the opposite order to reverse.
505
545
        rev_iter = iter(merge_sorted_revisions)
506
546
        if start_revision_id is not None:
507
547
            for node in rev_iter:
508
 
                rev_id = node.key[-1]
 
548
                rev_id = node.key
509
549
                if rev_id != start_revision_id:
510
550
                    continue
511
551
                else:
512
552
                    # The decision to include the start or not
513
553
                    # depends on the stop_rule if a stop is provided
514
554
                    # so pop this node back into the iterator
515
 
                    rev_iter = chain(iter([node]), rev_iter)
 
555
                    rev_iter = itertools.chain(iter([node]), rev_iter)
516
556
                    break
517
557
        if stop_revision_id is None:
518
558
            # Yield everything
519
559
            for node in rev_iter:
520
 
                rev_id = node.key[-1]
 
560
                rev_id = node.key
521
561
                yield (rev_id, node.merge_depth, node.revno,
522
562
                       node.end_of_merge)
523
563
        elif stop_rule == 'exclude':
524
564
            for node in rev_iter:
525
 
                rev_id = node.key[-1]
 
565
                rev_id = node.key
526
566
                if rev_id == stop_revision_id:
527
567
                    return
528
568
                yield (rev_id, node.merge_depth, node.revno,
529
569
                       node.end_of_merge)
530
570
        elif stop_rule == 'include':
531
571
            for node in rev_iter:
532
 
                rev_id = node.key[-1]
 
572
                rev_id = node.key
533
573
                yield (rev_id, node.merge_depth, node.revno,
534
574
                       node.end_of_merge)
535
575
                if rev_id == stop_revision_id:
541
581
            ancestors = graph.find_unique_ancestors(start_revision_id,
542
582
                                                    [stop_revision_id])
543
583
            for node in rev_iter:
544
 
                rev_id = node.key[-1]
 
584
                rev_id = node.key
545
585
                if rev_id not in ancestors:
546
586
                    continue
547
587
                yield (rev_id, node.merge_depth, node.revno,
557
597
            reached_stop_revision_id = False
558
598
            revision_id_whitelist = []
559
599
            for node in rev_iter:
560
 
                rev_id = node.key[-1]
 
600
                rev_id = node.key
561
601
                if rev_id == left_parent:
562
602
                    # reached the left parent after the stop_revision
563
603
                    return
643
683
        """
644
684
        raise errors.UpgradeRequired(self.user_url)
645
685
 
 
686
    def get_append_revisions_only(self):
 
687
        """Whether it is only possible to append revisions to the history.
 
688
        """
 
689
        if not self._format.supports_set_append_revisions_only():
 
690
            return False
 
691
        return self.get_config_stack().get('append_revisions_only')
 
692
 
646
693
    def set_append_revisions_only(self, enabled):
647
694
        if not self._format.supports_set_append_revisions_only():
648
695
            raise errors.UpgradeRequired(self.user_url)
649
 
        if enabled:
650
 
            value = 'True'
651
 
        else:
652
 
            value = 'False'
653
 
        self.get_config().set_user_option('append_revisions_only', value,
654
 
            warn_masked=True)
 
696
        self.get_config_stack().set('append_revisions_only', enabled)
655
697
 
656
698
    def set_reference_info(self, file_id, tree_path, branch_location):
657
699
        """Set the branch location to use for a tree reference."""
662
704
        raise errors.UnsupportedOperation(self.get_reference_info, self)
663
705
 
664
706
    @needs_write_lock
665
 
    def fetch(self, from_branch, last_revision=None, fetch_spec=None):
 
707
    def fetch(self, from_branch, last_revision=None, limit=None):
666
708
        """Copy revisions from from_branch into this branch.
667
709
 
668
710
        :param from_branch: Where to copy from.
669
711
        :param last_revision: What revision to stop at (None for at the end
670
712
                              of the branch.
671
 
        :param fetch_spec: If specified, a SearchResult or
672
 
            PendingAncestryResult that describes which revisions to copy.  This
673
 
            allows copying multiple heads at once.  Mutually exclusive with
674
 
            last_revision.
 
713
        :param limit: Optional rough limit of revisions to fetch
675
714
        :return: None
676
715
        """
677
 
        if fetch_spec is not None and last_revision is not None:
678
 
            raise AssertionError(
679
 
                "fetch_spec and last_revision are mutually exclusive.")
680
 
        if self.base == from_branch.base:
681
 
            return (0, [])
682
 
        from_branch.lock_read()
683
 
        try:
684
 
            if last_revision is None and fetch_spec is None:
685
 
                last_revision = from_branch.last_revision()
686
 
                last_revision = _mod_revision.ensure_null(last_revision)
687
 
            return self.repository.fetch(from_branch.repository,
688
 
                                         revision_id=last_revision,
689
 
                                         fetch_spec=fetch_spec)
690
 
        finally:
691
 
            from_branch.unlock()
 
716
        return InterBranch.get(from_branch, self).fetch(last_revision, limit=limit)
692
717
 
693
718
    def get_bound_location(self):
694
719
        """Return the URL of the branch we are bound to.
703
728
        """
704
729
        raise errors.UpgradeRequired(self.user_url)
705
730
 
706
 
    def get_commit_builder(self, parents, config=None, timestamp=None,
 
731
    def get_commit_builder(self, parents, config_stack=None, timestamp=None,
707
732
                           timezone=None, committer=None, revprops=None,
708
 
                           revision_id=None):
 
733
                           revision_id=None, lossy=False):
709
734
        """Obtain a CommitBuilder for this branch.
710
735
 
711
736
        :param parents: Revision ids of the parents of the new revision.
715
740
        :param committer: Optional committer to set for commit.
716
741
        :param revprops: Optional dictionary of revision properties.
717
742
        :param revision_id: Optional revision id.
 
743
        :param lossy: Whether to discard data that can not be natively
 
744
            represented, when pushing to a foreign VCS 
718
745
        """
719
746
 
720
 
        if config is None:
721
 
            config = self.get_config()
 
747
        if config_stack is None:
 
748
            config_stack = self.get_config_stack()
722
749
 
723
 
        return self.repository.get_commit_builder(self, parents, config,
724
 
            timestamp, timezone, committer, revprops, revision_id)
 
750
        return self.repository.get_commit_builder(self, parents, config_stack,
 
751
            timestamp, timezone, committer, revprops, revision_id,
 
752
            lossy)
725
753
 
726
754
    def get_master_branch(self, possible_transports=None):
727
755
        """Return the branch we are bound to.
730
758
        """
731
759
        return None
732
760
 
 
761
    @deprecated_method(deprecated_in((2, 5, 0)))
733
762
    def get_revision_delta(self, revno):
734
763
        """Return the delta for one revision.
735
764
 
736
765
        The delta is relative to its mainline predecessor, or the
737
766
        empty tree for revision 1.
738
767
        """
739
 
        rh = self.revision_history()
740
 
        if not (1 <= revno <= len(rh)):
 
768
        try:
 
769
            revid = self.get_rev_id(revno)
 
770
        except errors.NoSuchRevision:
741
771
            raise errors.InvalidRevisionNumber(revno)
742
 
        return self.repository.get_revision_delta(rh[revno-1])
 
772
        return self.repository.get_revision_delta(revid)
743
773
 
744
774
    def get_stacked_on_url(self):
745
775
        """Get the URL this branch is stacked against.
754
784
        """Print `file` to stdout."""
755
785
        raise NotImplementedError(self.print_file)
756
786
 
757
 
    def set_revision_history(self, rev_history):
758
 
        raise NotImplementedError(self.set_revision_history)
 
787
    @needs_write_lock
 
788
    def set_last_revision_info(self, revno, revision_id):
 
789
        """Set the last revision of this branch.
 
790
 
 
791
        The caller is responsible for checking that the revno is correct
 
792
        for this revision id.
 
793
 
 
794
        It may be possible to set the branch last revision to an id not
 
795
        present in the repository.  However, branches can also be
 
796
        configured to check constraints on history, in which case this may not
 
797
        be permitted.
 
798
        """
 
799
        raise NotImplementedError(self.set_last_revision_info)
 
800
 
 
801
    @needs_write_lock
 
802
    def generate_revision_history(self, revision_id, last_rev=None,
 
803
                                  other_branch=None):
 
804
        """See Branch.generate_revision_history"""
 
805
        graph = self.repository.get_graph()
 
806
        (last_revno, last_revid) = self.last_revision_info()
 
807
        known_revision_ids = [
 
808
            (last_revid, last_revno),
 
809
            (_mod_revision.NULL_REVISION, 0),
 
810
            ]
 
811
        if last_rev is not None:
 
812
            if not graph.is_ancestor(last_rev, revision_id):
 
813
                # our previous tip is not merged into stop_revision
 
814
                raise errors.DivergedBranches(self, other_branch)
 
815
        revno = graph.find_distance_to_null(revision_id, known_revision_ids)
 
816
        self.set_last_revision_info(revno, revision_id)
759
817
 
760
818
    @needs_write_lock
761
819
    def set_parent(self, url):
798
856
                return
799
857
            self._unstack()
800
858
        else:
801
 
            self._activate_fallback_location(url)
 
859
            self._activate_fallback_location(url,
 
860
                possible_transports=[self.bzrdir.root_transport])
802
861
        # write this out after the repository is stacked to avoid setting a
803
862
        # stacked config that doesn't work.
804
863
        self._set_config_location('stacked_on_location', url)
805
864
 
806
865
    def _unstack(self):
807
866
        """Change a branch to be unstacked, copying data as needed.
808
 
        
 
867
 
809
868
        Don't call this directly, use set_stacked_on_url(None).
810
869
        """
811
870
        pb = ui.ui_factory.nested_progress_bar()
812
871
        try:
813
 
            pb.update("Unstacking")
 
872
            pb.update(gettext("Unstacking"))
814
873
            # The basic approach here is to fetch the tip of the branch,
815
874
            # including all available ghosts, from the existing stacked
816
875
            # repository into a new repository object without the fallbacks. 
830
889
            # stream from one of them to the other.  This does mean doing
831
890
            # separate SSH connection setup, but unstacking is not a
832
891
            # common operation so it's tolerable.
833
 
            new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
 
892
            new_bzrdir = controldir.ControlDir.open(
 
893
                self.bzrdir.root_transport.base)
834
894
            new_repository = new_bzrdir.find_repository()
835
895
            if new_repository._fallback_repositories:
836
896
                raise AssertionError("didn't expect %r to have "
879
939
                    tags_to_fetch = set(self.tags.get_reverse_tag_dict())
880
940
                except errors.TagsNotSupported:
881
941
                    tags_to_fetch = set()
882
 
                fetch_spec = _mod_graph.NotInOtherForRevs(self.repository,
 
942
                fetch_spec = vf_search.NotInOtherForRevs(self.repository,
883
943
                    old_repository, required_ids=[self.last_revision()],
884
944
                    if_present_ids=tags_to_fetch, find_ghosts=True).execute()
885
945
                self.repository.fetch(old_repository, fetch_spec=fetch_spec)
893
953
 
894
954
        :seealso: Branch._get_tags_bytes.
895
955
        """
896
 
        return _run_with_write_locked_target(self, self._set_tags_bytes_locked,
897
 
                bytes)
 
956
        op = cleanup.OperationWithCleanups(self._set_tags_bytes_locked)
 
957
        op.add_cleanup(self.lock_write().unlock)
 
958
        return op.run_simple(bytes)
898
959
 
899
960
    def _set_tags_bytes_locked(self, bytes):
900
961
        self._tags_bytes = bytes
925
986
        This means the next call to revision_history will need to call
926
987
        _gen_revision_history.
927
988
 
928
 
        This API is semi-public; it only for use by subclasses, all other code
929
 
        should consider it to be private.
 
989
        This API is semi-public; it is only for use by subclasses, all other
 
990
        code should consider it to be private.
930
991
        """
931
992
        self._revision_history_cache = None
932
993
        self._revision_id_to_revno_cache = None
933
994
        self._last_revision_info_cache = None
 
995
        self._master_branch_cache = None
934
996
        self._merge_sorted_revisions_cache = None
935
997
        self._partial_revision_history_cache = []
936
998
        self._partial_revision_id_to_revno_cache = {}
951
1013
        """
952
1014
        raise NotImplementedError(self._gen_revision_history)
953
1015
 
954
 
    @needs_read_lock
955
 
    def revision_history(self):
956
 
        """Return sequence of revision ids on this branch.
957
 
 
958
 
        This method will cache the revision history for as long as it is safe to
959
 
        do so.
960
 
        """
 
1016
    def _revision_history(self):
961
1017
        if 'evil' in debug.debug_flags:
962
1018
            mutter_callsite(3, "revision_history scales with history.")
963
1019
        if self._revision_history_cache is not None:
990
1046
        :return: A tuple (revno, revision_id).
991
1047
        """
992
1048
        if self._last_revision_info_cache is None:
993
 
            self._last_revision_info_cache = self._last_revision_info()
 
1049
            self._last_revision_info_cache = self._read_last_revision_info()
994
1050
        return self._last_revision_info_cache
995
1051
 
996
 
    def _last_revision_info(self):
997
 
        rh = self.revision_history()
998
 
        revno = len(rh)
999
 
        if revno:
1000
 
            return (revno, rh[-1])
1001
 
        else:
1002
 
            return (0, _mod_revision.NULL_REVISION)
1003
 
 
1004
 
    def update_revisions(self, other, stop_revision=None, overwrite=False,
1005
 
                         graph=None, fetch_tags=True):
1006
 
        """Pull in new perfect-fit revisions.
1007
 
 
1008
 
        :param other: Another Branch to pull from
1009
 
        :param stop_revision: Updated until the given revision
1010
 
        :param overwrite: Always set the branch pointer, rather than checking
1011
 
            to see if it is a proper descendant.
1012
 
        :param graph: A Graph object that can be used to query history
1013
 
            information. This can be None.
1014
 
        :param fetch_tags: Flag that specifies if tags from other should be
1015
 
            fetched too.
1016
 
        :return: None
1017
 
        """
1018
 
        return InterBranch.get(other, self).update_revisions(stop_revision,
1019
 
            overwrite, graph, fetch_tags=fetch_tags)
1020
 
 
1021
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1022
 
    def import_last_revision_info(self, source_repo, revno, revid):
1023
 
        """Set the last revision info, importing from another repo if necessary.
1024
 
 
1025
 
        :param source_repo: Source repository to optionally fetch from
1026
 
        :param revno: Revision number of the new tip
1027
 
        :param revid: Revision id of the new tip
1028
 
        """
1029
 
        if not self.repository.has_same_location(source_repo):
1030
 
            self.repository.fetch(source_repo, revision_id=revid)
1031
 
        self.set_last_revision_info(revno, revid)
1032
 
 
1033
 
    def import_last_revision_info_and_tags(self, source, revno, revid):
 
1052
    def _read_last_revision_info(self):
 
1053
        raise NotImplementedError(self._read_last_revision_info)
 
1054
 
 
1055
    def import_last_revision_info_and_tags(self, source, revno, revid,
 
1056
                                           lossy=False):
1034
1057
        """Set the last revision info, importing from another repo if necessary.
1035
1058
 
1036
1059
        This is used by the bound branch code to upload a revision to
1040
1063
        :param source: Source branch to optionally fetch from
1041
1064
        :param revno: Revision number of the new tip
1042
1065
        :param revid: Revision id of the new tip
 
1066
        :param lossy: Whether to discard metadata that can not be
 
1067
            natively represented
 
1068
        :return: Tuple with the new revision number and revision id
 
1069
            (should only be different from the arguments when lossy=True)
1043
1070
        """
1044
1071
        if not self.repository.has_same_location(source.repository):
1045
 
            try:
1046
 
                tags_to_fetch = set(source.tags.get_reverse_tag_dict())
1047
 
            except errors.TagsNotSupported:
1048
 
                tags_to_fetch = set()
1049
 
            fetch_spec = _mod_graph.NotInOtherForRevs(self.repository,
1050
 
                source.repository, [revid],
1051
 
                if_present_ids=tags_to_fetch).execute()
1052
 
            self.repository.fetch(source.repository, fetch_spec=fetch_spec)
 
1072
            self.fetch(source, revid)
1053
1073
        self.set_last_revision_info(revno, revid)
 
1074
        return (revno, revid)
1054
1075
 
1055
1076
    def revision_id_to_revno(self, revision_id):
1056
1077
        """Given a revision id, return its revno"""
1057
1078
        if _mod_revision.is_null(revision_id):
1058
1079
            return 0
1059
 
        history = self.revision_history()
 
1080
        history = self._revision_history()
1060
1081
        try:
1061
1082
            return history.index(revision_id) + 1
1062
1083
        except ValueError:
1089
1110
            stop_revision=stop_revision,
1090
1111
            possible_transports=possible_transports, *args, **kwargs)
1091
1112
 
1092
 
    def push(self, target, overwrite=False, stop_revision=None, *args,
1093
 
        **kwargs):
 
1113
    def push(self, target, overwrite=False, stop_revision=None, lossy=False,
 
1114
            *args, **kwargs):
1094
1115
        """Mirror this branch into target.
1095
1116
 
1096
1117
        This branch is considered to be 'local', having low latency.
1097
1118
        """
1098
1119
        return InterBranch.get(self, target).push(overwrite, stop_revision,
1099
 
            *args, **kwargs)
1100
 
 
1101
 
    def lossy_push(self, target, stop_revision=None):
1102
 
        """Push deltas into another branch.
1103
 
 
1104
 
        :note: This does not, like push, retain the revision ids from 
1105
 
            the source branch and will, rather than adding bzr-specific 
1106
 
            metadata, push only those semantics of the revision that can be 
1107
 
            natively represented by this branch' VCS.
1108
 
 
1109
 
        :param target: Target branch
1110
 
        :param stop_revision: Revision to push, defaults to last revision.
1111
 
        :return: BranchPushResult with an extra member revidmap: 
1112
 
            A dictionary mapping revision ids from the target branch 
1113
 
            to new revision ids in the target branch, for each 
1114
 
            revision that was pushed.
1115
 
        """
1116
 
        inter = InterBranch.get(self, target)
1117
 
        lossy_push = getattr(inter, "lossy_push", None)
1118
 
        if lossy_push is None:
1119
 
            raise errors.LossyPushToSameVCS(self, target)
1120
 
        return lossy_push(stop_revision)
 
1120
            lossy, *args, **kwargs)
1121
1121
 
1122
1122
    def basis_tree(self):
1123
1123
        """Return `Tree` object for last revision."""
1148
1148
    def _set_config_location(self, name, url, config=None,
1149
1149
                             make_relative=False):
1150
1150
        if config is None:
1151
 
            config = self.get_config()
 
1151
            config = self.get_config_stack()
1152
1152
        if url is None:
1153
1153
            url = ''
1154
1154
        elif make_relative:
1155
1155
            url = urlutils.relative_url(self.base, url)
1156
 
        config.set_user_option(name, url, warn_masked=True)
 
1156
        config.set(name, url)
1157
1157
 
1158
1158
    def _get_config_location(self, name, config=None):
1159
1159
        if config is None:
1160
 
            config = self.get_config()
1161
 
        location = config.get_user_option(name)
 
1160
            config = self.get_config_stack()
 
1161
        location = config.get(name)
1162
1162
        if location == '':
1163
1163
            location = None
1164
1164
        return location
1165
1165
 
1166
1166
    def get_child_submit_format(self):
1167
1167
        """Return the preferred format of submissions to this branch."""
1168
 
        return self.get_config().get_user_option("child_submit_format")
 
1168
        return self.get_config_stack().get('child_submit_format')
1169
1169
 
1170
1170
    def get_submit_branch(self):
1171
1171
        """Return the submit location of the branch.
1174
1174
        pattern is that the user can override it by specifying a
1175
1175
        location.
1176
1176
        """
1177
 
        return self.get_config().get_user_option('submit_branch')
 
1177
        return self.get_config_stack().get('submit_branch')
1178
1178
 
1179
1179
    def set_submit_branch(self, location):
1180
1180
        """Return the submit location of the branch.
1183
1183
        pattern is that the user can override it by specifying a
1184
1184
        location.
1185
1185
        """
1186
 
        self.get_config().set_user_option('submit_branch', location,
1187
 
            warn_masked=True)
 
1186
        self.get_config_stack().set('submit_branch', location)
1188
1187
 
1189
1188
    def get_public_branch(self):
1190
1189
        """Return the public location of the branch.
1203
1202
        self._set_config_location('public_branch', location)
1204
1203
 
1205
1204
    def get_push_location(self):
1206
 
        """Return the None or the location to push this branch to."""
1207
 
        push_loc = self.get_config().get_user_option('push_location')
1208
 
        return push_loc
 
1205
        """Return None or the location to push this branch to."""
 
1206
        return self.get_config_stack().get('push_location')
1209
1207
 
1210
1208
    def set_push_location(self, location):
1211
1209
        """Set a new push location for this branch."""
1296
1294
            if repository_policy is not None:
1297
1295
                repository_policy.configure_branch(result)
1298
1296
            self.copy_content_into(result, revision_id=revision_id)
1299
 
            result.set_parent(self.bzrdir.root_transport.base)
 
1297
            master_url = self.get_bound_location()
 
1298
            if master_url is None:
 
1299
                result.set_parent(self.bzrdir.root_transport.base)
 
1300
            else:
 
1301
                result.set_parent(master_url)
1300
1302
        finally:
1301
1303
            result.unlock()
1302
1304
        return result
1376
1378
        # TODO: We should probably also check that self.revision_history
1377
1379
        # matches the repository for older branch formats.
1378
1380
        # If looking for the code that cross-checks repository parents against
1379
 
        # the iter_reverse_revision_history output, that is now a repository
 
1381
        # the Graph.iter_lefthand_ancestry output, that is now a repository
1380
1382
        # specific check.
1381
1383
        return result
1382
1384
 
1383
 
    def _get_checkout_format(self):
 
1385
    def _get_checkout_format(self, lightweight=False):
1384
1386
        """Return the most suitable metadir for a checkout of this branch.
1385
1387
        Weaves are used if this branch's repository uses weaves.
1386
1388
        """
1421
1423
        :param to_location: The url to produce the checkout at
1422
1424
        :param revision_id: The revision to check out
1423
1425
        :param lightweight: If True, produce a lightweight checkout, otherwise,
1424
 
        produce a bound branch (heavyweight checkout)
 
1426
            produce a bound branch (heavyweight checkout)
1425
1427
        :param accelerator_tree: A tree which can be used for retrieving file
1426
1428
            contents more quickly than the revision tree, i.e. a workingtree.
1427
1429
            The revision tree will be used for cases where accelerator_tree's
1432
1434
        """
1433
1435
        t = transport.get_transport(to_location)
1434
1436
        t.ensure_base()
 
1437
        format = self._get_checkout_format(lightweight=lightweight)
 
1438
        try:
 
1439
            checkout = format.initialize_on_transport(t)
 
1440
        except errors.AlreadyControlDirError:
 
1441
            # It's fine if the control directory already exists,
 
1442
            # as long as there is no existing branch and working tree.
 
1443
            checkout = controldir.ControlDir.open_from_transport(t)
 
1444
            try:
 
1445
                checkout.open_branch()
 
1446
            except errors.NotBranchError:
 
1447
                pass
 
1448
            else:
 
1449
                raise errors.AlreadyControlDirError(t.base)
 
1450
            if checkout.control_transport.base == self.bzrdir.control_transport.base:
 
1451
                # When checking out to the same control directory,
 
1452
                # always create a lightweight checkout
 
1453
                lightweight = True
 
1454
 
1435
1455
        if lightweight:
1436
 
            format = self._get_checkout_format()
1437
 
            checkout = format.initialize_on_transport(t)
1438
 
            from_branch = BranchReferenceFormat().initialize(checkout, 
1439
 
                target_branch=self)
 
1456
            from_branch = checkout.set_branch_reference(target_branch=self)
1440
1457
        else:
1441
 
            format = self._get_checkout_format()
1442
 
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
1443
 
                to_location, force_new_tree=False, format=format)
1444
 
            checkout = checkout_branch.bzrdir
 
1458
            policy = checkout.determine_repository_policy()
 
1459
            repo = policy.acquire_repository()[0]
 
1460
            checkout_branch = checkout.create_branch()
1445
1461
            checkout_branch.bind(self)
1446
1462
            # pull up to the specified revision_id to set the initial
1447
1463
            # branch tip correctly, and seed it with history.
1448
1464
            checkout_branch.pull(self, stop_revision=revision_id)
1449
 
            from_branch=None
 
1465
            from_branch = None
1450
1466
        tree = checkout.create_workingtree(revision_id,
1451
1467
                                           from_branch=from_branch,
1452
1468
                                           accelerator_tree=accelerator_tree,
1473
1489
 
1474
1490
    def reference_parent(self, file_id, path, possible_transports=None):
1475
1491
        """Return the parent branch for a tree-reference file_id
 
1492
 
1476
1493
        :param file_id: The file_id of the tree reference
1477
1494
        :param path: The path of the file_id in the tree
1478
1495
        :return: A branch associated with the file_id
1540
1557
            heads that must be fetched if present, but no error is necessary if
1541
1558
            they are not present.
1542
1559
        """
1543
 
        # For bzr native formats must_fetch is just the tip, and if_present_fetch
1544
 
        # are the tags.
 
1560
        # For bzr native formats must_fetch is just the tip, and
 
1561
        # if_present_fetch are the tags.
1545
1562
        must_fetch = set([self.last_revision()])
1546
 
        try:
1547
 
            if_present_fetch = set(self.tags.get_reverse_tag_dict())
1548
 
        except errors.TagsNotSupported:
1549
 
            if_present_fetch = set()
 
1563
        if_present_fetch = set()
 
1564
        if self.get_config_stack().get('branch.fetch_tags'):
 
1565
            try:
 
1566
                if_present_fetch = set(self.tags.get_reverse_tag_dict())
 
1567
            except errors.TagsNotSupported:
 
1568
                pass
1550
1569
        must_fetch.discard(_mod_revision.NULL_REVISION)
1551
1570
        if_present_fetch.discard(_mod_revision.NULL_REVISION)
1552
1571
        return must_fetch, if_present_fetch
1557
1576
 
1558
1577
    Formats provide three things:
1559
1578
     * An initialization routine,
1560
 
     * a format string,
 
1579
     * a format description
1561
1580
     * an open routine.
1562
1581
 
1563
1582
    Formats are placed in an dict by their format string for reference
1570
1589
    object will be created every time regardless.
1571
1590
    """
1572
1591
 
1573
 
    can_set_append_revisions_only = True
1574
 
 
1575
1592
    def __eq__(self, other):
1576
1593
        return self.__class__ is other.__class__
1577
1594
 
1578
1595
    def __ne__(self, other):
1579
1596
        return not (self == other)
1580
1597
 
1581
 
    @classmethod
1582
 
    def find_format(klass, a_bzrdir, name=None):
1583
 
        """Return the format for the branch object in a_bzrdir."""
1584
 
        try:
1585
 
            transport = a_bzrdir.get_branch_transport(None, name=name)
1586
 
            format_string = transport.get_bytes("format")
1587
 
            return format_registry.get(format_string)
1588
 
        except errors.NoSuchFile:
1589
 
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
1590
 
        except KeyError:
1591
 
            raise errors.UnknownFormatError(format=format_string, kind='branch')
1592
 
 
1593
 
    @classmethod
1594
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1595
 
    def get_default_format(klass):
1596
 
        """Return the current default format."""
1597
 
        return format_registry.get_default()
1598
 
 
1599
 
    @classmethod
1600
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1601
 
    def get_formats(klass):
1602
 
        """Get all the known formats.
1603
 
 
1604
 
        Warning: This triggers a load of all lazy registered formats: do not
1605
 
        use except when that is desireed.
1606
 
        """
1607
 
        return format_registry._get_all()
1608
 
 
1609
 
    def get_reference(self, a_bzrdir, name=None):
1610
 
        """Get the target reference of the branch in a_bzrdir.
 
1598
    def get_reference(self, controldir, name=None):
 
1599
        """Get the target reference of the branch in controldir.
1611
1600
 
1612
1601
        format probing must have been completed before calling
1613
1602
        this method - it is assumed that the format of the branch
1614
 
        in a_bzrdir is correct.
 
1603
        in controldir is correct.
1615
1604
 
1616
 
        :param a_bzrdir: The bzrdir to get the branch data from.
 
1605
        :param controldir: The controldir to get the branch data from.
1617
1606
        :param name: Name of the colocated branch to fetch
1618
1607
        :return: None if the branch is not a reference branch.
1619
1608
        """
1620
1609
        return None
1621
1610
 
1622
1611
    @classmethod
1623
 
    def set_reference(self, a_bzrdir, name, to_branch):
1624
 
        """Set the target reference of the branch in a_bzrdir.
 
1612
    def set_reference(self, controldir, name, to_branch):
 
1613
        """Set the target reference of the branch in controldir.
1625
1614
 
1626
1615
        format probing must have been completed before calling
1627
1616
        this method - it is assumed that the format of the branch
1628
 
        in a_bzrdir is correct.
 
1617
        in controldir is correct.
1629
1618
 
1630
 
        :param a_bzrdir: The bzrdir to set the branch reference for.
 
1619
        :param controldir: The controldir to set the branch reference for.
1631
1620
        :param name: Name of colocated branch to set, None for default
1632
1621
        :param to_branch: branch that the checkout is to reference
1633
1622
        """
1634
1623
        raise NotImplementedError(self.set_reference)
1635
1624
 
1636
 
    def get_format_string(self):
1637
 
        """Return the ASCII format string that identifies this format."""
1638
 
        raise NotImplementedError(self.get_format_string)
1639
 
 
1640
1625
    def get_format_description(self):
1641
1626
        """Return the short format description for this format."""
1642
1627
        raise NotImplementedError(self.get_format_description)
1643
1628
 
1644
 
    def _run_post_branch_init_hooks(self, a_bzrdir, name, branch):
 
1629
    def _run_post_branch_init_hooks(self, controldir, name, branch):
1645
1630
        hooks = Branch.hooks['post_branch_init']
1646
1631
        if not hooks:
1647
1632
            return
1648
 
        params = BranchInitHookParams(self, a_bzrdir, name, branch)
 
1633
        params = BranchInitHookParams(self, controldir, name, branch)
1649
1634
        for hook in hooks:
1650
1635
            hook(params)
1651
1636
 
1652
 
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
1653
 
                           repository=None, lock_type='metadir',
1654
 
                           set_format=True):
1655
 
        """Initialize a branch in a bzrdir, with specified files
1656
 
 
1657
 
        :param a_bzrdir: The bzrdir to initialize the branch in
1658
 
        :param utf8_files: The files to create as a list of
1659
 
            (filename, content) tuples
1660
 
        :param name: Name of colocated branch to create, if any
1661
 
        :param set_format: If True, set the format with
1662
 
            self.get_format_string.  (BzrBranch4 has its format set
1663
 
            elsewhere)
1664
 
        :return: a branch in this format
1665
 
        """
1666
 
        mutter('creating branch %r in %s', self, a_bzrdir.user_url)
1667
 
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1668
 
        lock_map = {
1669
 
            'metadir': ('lock', lockdir.LockDir),
1670
 
            'branch4': ('branch-lock', lockable_files.TransportLock),
1671
 
        }
1672
 
        lock_name, lock_class = lock_map[lock_type]
1673
 
        control_files = lockable_files.LockableFiles(branch_transport,
1674
 
            lock_name, lock_class)
1675
 
        control_files.create_lock()
1676
 
        try:
1677
 
            control_files.lock_write()
1678
 
        except errors.LockContention:
1679
 
            if lock_type != 'branch4':
1680
 
                raise
1681
 
            lock_taken = False
1682
 
        else:
1683
 
            lock_taken = True
1684
 
        if set_format:
1685
 
            utf8_files += [('format', self.get_format_string())]
1686
 
        try:
1687
 
            for (filename, content) in utf8_files:
1688
 
                branch_transport.put_bytes(
1689
 
                    filename, content,
1690
 
                    mode=a_bzrdir._get_file_mode())
1691
 
        finally:
1692
 
            if lock_taken:
1693
 
                control_files.unlock()
1694
 
        branch = self.open(a_bzrdir, name, _found=True,
1695
 
                found_repository=repository)
1696
 
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
1697
 
        return branch
1698
 
 
1699
 
    def initialize(self, a_bzrdir, name=None, repository=None):
1700
 
        """Create a branch of this format in a_bzrdir.
1701
 
        
 
1637
    def initialize(self, controldir, name=None, repository=None,
 
1638
                   append_revisions_only=None):
 
1639
        """Create a branch of this format in controldir.
 
1640
 
1702
1641
        :param name: Name of the colocated branch to create.
1703
1642
        """
1704
1643
        raise NotImplementedError(self.initialize)
1724
1663
        Note that it is normal for branch to be a RemoteBranch when using tags
1725
1664
        on a RemoteBranch.
1726
1665
        """
1727
 
        return DisabledTags(branch)
 
1666
        return _mod_tag.DisabledTags(branch)
1728
1667
 
1729
1668
    def network_name(self):
1730
1669
        """A simple byte string uniquely identifying this format for RPC calls.
1736
1675
        """
1737
1676
        raise NotImplementedError(self.network_name)
1738
1677
 
1739
 
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
1740
 
            found_repository=None):
1741
 
        """Return the branch object for a_bzrdir
 
1678
    def open(self, controldir, name=None, _found=False, ignore_fallbacks=False,
 
1679
            found_repository=None, possible_transports=None):
 
1680
        """Return the branch object for controldir.
1742
1681
 
1743
 
        :param a_bzrdir: A BzrDir that contains a branch.
 
1682
        :param controldir: A ControlDir that contains a branch.
1744
1683
        :param name: Name of colocated branch to open
1745
1684
        :param _found: a private parameter, do not use it. It is used to
1746
1685
            indicate if format probing has already be done.
1749
1688
        """
1750
1689
        raise NotImplementedError(self.open)
1751
1690
 
1752
 
    @classmethod
1753
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1754
 
    def register_format(klass, format):
1755
 
        """Register a metadir format.
1756
 
 
1757
 
        See MetaDirBranchFormatFactory for the ability to register a format
1758
 
        without loading the code the format needs until it is actually used.
1759
 
        """
1760
 
        format_registry.register(format)
1761
 
 
1762
 
    @classmethod
1763
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1764
 
    def set_default_format(klass, format):
1765
 
        format_registry.set_default(format)
1766
 
 
1767
1691
    def supports_set_append_revisions_only(self):
1768
1692
        """True if this format supports set_append_revisions_only."""
1769
1693
        return False
1776
1700
        """True if this format supports leaving locks in place."""
1777
1701
        return False # by default
1778
1702
 
1779
 
    @classmethod
1780
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1781
 
    def unregister_format(klass, format):
1782
 
        format_registry.remove(format)
1783
 
 
1784
1703
    def __str__(self):
1785
1704
        return self.get_format_description().rstrip()
1786
1705
 
1788
1707
        """True if this format supports tags stored in the branch"""
1789
1708
        return False  # by default
1790
1709
 
 
1710
    def tags_are_versioned(self):
 
1711
        """Whether the tag container for this branch versions tags."""
 
1712
        return False
 
1713
 
 
1714
    def supports_tags_referencing_ghosts(self):
 
1715
        """True if tags can reference ghost revisions."""
 
1716
        return True
 
1717
 
1791
1718
 
1792
1719
class MetaDirBranchFormatFactory(registry._LazyObjectGetter):
1793
1720
    """A factory for a BranchFormat object, permitting simple lazy registration.
1807
1734
        """
1808
1735
        registry._LazyObjectGetter.__init__(self, module_name, member_name)
1809
1736
        self._format_string = format_string
1810
 
        
 
1737
 
1811
1738
    def get_format_string(self):
1812
1739
        """See BranchFormat.get_format_string."""
1813
1740
        return self._format_string
1820
1747
class BranchHooks(Hooks):
1821
1748
    """A dictionary mapping hook name to a list of callables for branch hooks.
1822
1749
 
1823
 
    e.g. ['set_rh'] Is the list of items to be called when the
1824
 
    set_revision_history function is invoked.
 
1750
    e.g. ['post_push'] Is the list of items to be called when the
 
1751
    push function is invoked.
1825
1752
    """
1826
1753
 
1827
1754
    def __init__(self):
1830
1757
        These are all empty initially, because by default nothing should get
1831
1758
        notified.
1832
1759
        """
1833
 
        Hooks.__init__(self)
1834
 
        self.create_hook(HookPoint('set_rh',
1835
 
            "Invoked whenever the revision history has been set via "
1836
 
            "set_revision_history. The api signature is (branch, "
1837
 
            "revision_history), and the branch will be write-locked. "
1838
 
            "The set_rh hook can be expensive for bzr to trigger, a better "
1839
 
            "hook to use is Branch.post_change_branch_tip.", (0, 15), None))
1840
 
        self.create_hook(HookPoint('open',
 
1760
        Hooks.__init__(self, "bzrlib.branch", "Branch.hooks")
 
1761
        self.add_hook('open',
1841
1762
            "Called with the Branch object that has been opened after a "
1842
 
            "branch is opened.", (1, 8), None))
1843
 
        self.create_hook(HookPoint('post_push',
 
1763
            "branch is opened.", (1, 8))
 
1764
        self.add_hook('post_push',
1844
1765
            "Called after a push operation completes. post_push is called "
1845
1766
            "with a bzrlib.branch.BranchPushResult object and only runs in the "
1846
 
            "bzr client.", (0, 15), None))
1847
 
        self.create_hook(HookPoint('post_pull',
 
1767
            "bzr client.", (0, 15))
 
1768
        self.add_hook('post_pull',
1848
1769
            "Called after a pull operation completes. post_pull is called "
1849
1770
            "with a bzrlib.branch.PullResult object and only runs in the "
1850
 
            "bzr client.", (0, 15), None))
1851
 
        self.create_hook(HookPoint('pre_commit',
 
1771
            "bzr client.", (0, 15))
 
1772
        self.add_hook('pre_commit',
1852
1773
            "Called after a commit is calculated but before it is "
1853
1774
            "completed. pre_commit is called with (local, master, old_revno, "
1854
1775
            "old_revid, future_revno, future_revid, tree_delta, future_tree"
1857
1778
            "basis revision. hooks MUST NOT modify this delta. "
1858
1779
            " future_tree is an in-memory tree obtained from "
1859
1780
            "CommitBuilder.revision_tree() and hooks MUST NOT modify this "
1860
 
            "tree.", (0,91), None))
1861
 
        self.create_hook(HookPoint('post_commit',
 
1781
            "tree.", (0,91))
 
1782
        self.add_hook('post_commit',
1862
1783
            "Called in the bzr client after a commit has completed. "
1863
1784
            "post_commit is called with (local, master, old_revno, old_revid, "
1864
1785
            "new_revno, new_revid). old_revid is NULL_REVISION for the first "
1865
 
            "commit to a branch.", (0, 15), None))
1866
 
        self.create_hook(HookPoint('post_uncommit',
 
1786
            "commit to a branch.", (0, 15))
 
1787
        self.add_hook('post_uncommit',
1867
1788
            "Called in the bzr client after an uncommit completes. "
1868
1789
            "post_uncommit is called with (local, master, old_revno, "
1869
1790
            "old_revid, new_revno, new_revid) where local is the local branch "
1870
1791
            "or None, master is the target branch, and an empty branch "
1871
 
            "receives new_revno of 0, new_revid of None.", (0, 15), None))
1872
 
        self.create_hook(HookPoint('pre_change_branch_tip',
 
1792
            "receives new_revno of 0, new_revid of None.", (0, 15))
 
1793
        self.add_hook('pre_change_branch_tip',
1873
1794
            "Called in bzr client and server before a change to the tip of a "
1874
1795
            "branch is made. pre_change_branch_tip is called with a "
1875
1796
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1876
 
            "commit, uncommit will all trigger this hook.", (1, 6), None))
1877
 
        self.create_hook(HookPoint('post_change_branch_tip',
 
1797
            "commit, uncommit will all trigger this hook.", (1, 6))
 
1798
        self.add_hook('post_change_branch_tip',
1878
1799
            "Called in bzr client and server after a change to the tip of a "
1879
1800
            "branch is made. post_change_branch_tip is called with a "
1880
1801
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1881
 
            "commit, uncommit will all trigger this hook.", (1, 4), None))
1882
 
        self.create_hook(HookPoint('transform_fallback_location',
 
1802
            "commit, uncommit will all trigger this hook.", (1, 4))
 
1803
        self.add_hook('transform_fallback_location',
1883
1804
            "Called when a stacked branch is activating its fallback "
1884
1805
            "locations. transform_fallback_location is called with (branch, "
1885
1806
            "url), and should return a new url. Returning the same url "
1890
1811
            "fallback locations have not been activated. When there are "
1891
1812
            "multiple hooks installed for transform_fallback_location, "
1892
1813
            "all are called with the url returned from the previous hook."
1893
 
            "The order is however undefined.", (1, 9), None))
1894
 
        self.create_hook(HookPoint('automatic_tag_name',
 
1814
            "The order is however undefined.", (1, 9))
 
1815
        self.add_hook('automatic_tag_name',
1895
1816
            "Called to determine an automatic tag name for a revision. "
1896
1817
            "automatic_tag_name is called with (branch, revision_id) and "
1897
1818
            "should return a tag name or None if no tag name could be "
1898
1819
            "determined. The first non-None tag name returned will be used.",
1899
 
            (2, 2), None))
1900
 
        self.create_hook(HookPoint('post_branch_init',
 
1820
            (2, 2))
 
1821
        self.add_hook('post_branch_init',
1901
1822
            "Called after new branch initialization completes. "
1902
1823
            "post_branch_init is called with a "
1903
1824
            "bzrlib.branch.BranchInitHookParams. "
1904
1825
            "Note that init, branch and checkout (both heavyweight and "
1905
 
            "lightweight) will all trigger this hook.", (2, 2), None))
1906
 
        self.create_hook(HookPoint('post_switch',
 
1826
            "lightweight) will all trigger this hook.", (2, 2))
 
1827
        self.add_hook('post_switch',
1907
1828
            "Called after a checkout switches branch. "
1908
1829
            "post_switch is called with a "
1909
 
            "bzrlib.branch.SwitchHookParams.", (2, 2), None))
 
1830
            "bzrlib.branch.SwitchHookParams.", (2, 2))
1910
1831
 
1911
1832
 
1912
1833
 
1915
1836
 
1916
1837
 
1917
1838
class ChangeBranchTipParams(object):
1918
 
    """Object holding parameters passed to *_change_branch_tip hooks.
 
1839
    """Object holding parameters passed to `*_change_branch_tip` hooks.
1919
1840
 
1920
1841
    There are 5 fields that hooks may wish to access:
1921
1842
 
1953
1874
 
1954
1875
 
1955
1876
class BranchInitHookParams(object):
1956
 
    """Object holding parameters passed to *_branch_init hooks.
 
1877
    """Object holding parameters passed to `*_branch_init` hooks.
1957
1878
 
1958
1879
    There are 4 fields that hooks may wish to access:
1959
1880
 
1960
1881
    :ivar format: the branch format
1961
 
    :ivar bzrdir: the BzrDir where the branch will be/has been initialized
 
1882
    :ivar bzrdir: the ControlDir where the branch will be/has been initialized
1962
1883
    :ivar name: name of colocated branch, if any (or None)
1963
1884
    :ivar branch: the branch created
1964
1885
 
1967
1888
    branch, which refer to the original branch.
1968
1889
    """
1969
1890
 
1970
 
    def __init__(self, format, a_bzrdir, name, branch):
 
1891
    def __init__(self, format, controldir, name, branch):
1971
1892
        """Create a group of BranchInitHook parameters.
1972
1893
 
1973
1894
        :param format: the branch format
1974
 
        :param a_bzrdir: the BzrDir where the branch will be/has been
 
1895
        :param controldir: the ControlDir where the branch will be/has been
1975
1896
            initialized
1976
1897
        :param name: name of colocated branch, if any (or None)
1977
1898
        :param branch: the branch created
1981
1902
        in branch, which refer to the original branch.
1982
1903
        """
1983
1904
        self.format = format
1984
 
        self.bzrdir = a_bzrdir
 
1905
        self.bzrdir = controldir
1985
1906
        self.name = name
1986
1907
        self.branch = branch
1987
1908
 
1993
1914
 
1994
1915
 
1995
1916
class SwitchHookParams(object):
1996
 
    """Object holding parameters passed to *_switch hooks.
 
1917
    """Object holding parameters passed to `*_switch` hooks.
1997
1918
 
1998
1919
    There are 4 fields that hooks may wish to access:
1999
1920
 
2000
 
    :ivar control_dir: BzrDir of the checkout to change
 
1921
    :ivar control_dir: ControlDir of the checkout to change
2001
1922
    :ivar to_branch: branch that the checkout is to reference
2002
1923
    :ivar force: skip the check for local commits in a heavy checkout
2003
1924
    :ivar revision_id: revision ID to switch to (or None)
2006
1927
    def __init__(self, control_dir, to_branch, force, revision_id):
2007
1928
        """Create a group of SwitchHook parameters.
2008
1929
 
2009
 
        :param control_dir: BzrDir of the checkout to change
 
1930
        :param control_dir: ControlDir of the checkout to change
2010
1931
        :param to_branch: branch that the checkout is to reference
2011
1932
        :param force: skip the check for local commits in a heavy checkout
2012
1933
        :param revision_id: revision ID to switch to (or None)
2025
1946
            self.revision_id)
2026
1947
 
2027
1948
 
2028
 
class BzrBranchFormat4(BranchFormat):
2029
 
    """Bzr branch format 4.
2030
 
 
2031
 
    This format has:
2032
 
     - a revision-history file.
2033
 
     - a branch-lock lock file [ to be shared with the bzrdir ]
 
1949
class BranchFormatMetadir(bzrdir.BzrFormat, BranchFormat):
 
1950
    """Base class for branch formats that live in meta directories.
2034
1951
    """
2035
1952
 
2036
 
    def get_format_description(self):
2037
 
        """See BranchFormat.get_format_description()."""
2038
 
        return "Branch format 4"
2039
 
 
2040
 
    def initialize(self, a_bzrdir, name=None, repository=None):
2041
 
        """Create a branch of this format in a_bzrdir."""
2042
 
        if repository is not None:
2043
 
            raise NotImplementedError(
2044
 
                "initialize(repository=<not None>) on %r" % (self,))
2045
 
        utf8_files = [('revision-history', ''),
2046
 
                      ('branch-name', ''),
2047
 
                      ]
2048
 
        return self._initialize_helper(a_bzrdir, utf8_files, name=name,
2049
 
                                       lock_type='branch4', set_format=False)
2050
 
 
2051
1953
    def __init__(self):
2052
 
        super(BzrBranchFormat4, self).__init__()
2053
 
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
2054
 
 
2055
 
    def network_name(self):
2056
 
        """The network name for this format is the control dirs disk label."""
2057
 
        return self._matchingbzrdir.get_format_string()
2058
 
 
2059
 
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
2060
 
            found_repository=None):
2061
 
        """See BranchFormat.open()."""
2062
 
        if not _found:
2063
 
            # we are being called directly and must probe.
2064
 
            raise NotImplementedError
2065
 
        if found_repository is None:
2066
 
            found_repository = a_bzrdir.open_repository()
2067
 
        return BzrBranchPreSplitOut(_format=self,
2068
 
                         _control_files=a_bzrdir._control_files,
2069
 
                         a_bzrdir=a_bzrdir,
2070
 
                         name=name,
2071
 
                         _repository=found_repository)
2072
 
 
2073
 
    def __str__(self):
2074
 
        return "Bazaar-NG branch format 4"
2075
 
 
2076
 
    def supports_leaving_lock(self):
2077
 
        return False
2078
 
 
2079
 
 
2080
 
class BranchFormatMetadir(BranchFormat):
2081
 
    """Common logic for meta-dir based branch formats."""
 
1954
        BranchFormat.__init__(self)
 
1955
        bzrdir.BzrFormat.__init__(self)
 
1956
 
 
1957
    @classmethod
 
1958
    def find_format(klass, controldir, name=None):
 
1959
        """Return the format for the branch object in controldir."""
 
1960
        try:
 
1961
            transport = controldir.get_branch_transport(None, name=name)
 
1962
        except errors.NoSuchFile:
 
1963
            raise errors.NotBranchError(path=name, bzrdir=controldir)
 
1964
        try:
 
1965
            format_string = transport.get_bytes("format")
 
1966
        except errors.NoSuchFile:
 
1967
            raise errors.NotBranchError(path=transport.base, bzrdir=controldir)
 
1968
        return klass._find_format(format_registry, 'branch', format_string)
2082
1969
 
2083
1970
    def _branch_class(self):
2084
1971
        """What class to instantiate on open calls."""
2085
1972
        raise NotImplementedError(self._branch_class)
2086
1973
 
2087
 
    def network_name(self):
2088
 
        """A simple byte string uniquely identifying this format for RPC calls.
2089
 
 
2090
 
        Metadir branch formats use their format string.
 
1974
    def _get_initial_config(self, append_revisions_only=None):
 
1975
        if append_revisions_only:
 
1976
            return "append_revisions_only = True\n"
 
1977
        else:
 
1978
            # Avoid writing anything if append_revisions_only is disabled,
 
1979
            # as that is the default.
 
1980
            return ""
 
1981
 
 
1982
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
 
1983
                           repository=None):
 
1984
        """Initialize a branch in a control dir, with specified files
 
1985
 
 
1986
        :param a_bzrdir: The bzrdir to initialize the branch in
 
1987
        :param utf8_files: The files to create as a list of
 
1988
            (filename, content) tuples
 
1989
        :param name: Name of colocated branch to create, if any
 
1990
        :return: a branch in this format
2091
1991
        """
2092
 
        return self.get_format_string()
 
1992
        if name is None:
 
1993
            name = a_bzrdir._get_selected_branch()
 
1994
        mutter('creating branch %r in %s', self, a_bzrdir.user_url)
 
1995
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
 
1996
        control_files = lockable_files.LockableFiles(branch_transport,
 
1997
            'lock', lockdir.LockDir)
 
1998
        control_files.create_lock()
 
1999
        control_files.lock_write()
 
2000
        try:
 
2001
            utf8_files += [('format', self.as_string())]
 
2002
            for (filename, content) in utf8_files:
 
2003
                branch_transport.put_bytes(
 
2004
                    filename, content,
 
2005
                    mode=a_bzrdir._get_file_mode())
 
2006
        finally:
 
2007
            control_files.unlock()
 
2008
        branch = self.open(a_bzrdir, name, _found=True,
 
2009
                found_repository=repository)
 
2010
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
 
2011
        return branch
2093
2012
 
2094
2013
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
2095
 
            found_repository=None):
 
2014
            found_repository=None, possible_transports=None):
2096
2015
        """See BranchFormat.open()."""
 
2016
        if name is None:
 
2017
            name = a_bzrdir._get_selected_branch()
2097
2018
        if not _found:
2098
 
            format = BranchFormat.find_format(a_bzrdir, name=name)
 
2019
            format = BranchFormatMetadir.find_format(a_bzrdir, name=name)
2099
2020
            if format.__class__ != self.__class__:
2100
2021
                raise AssertionError("wrong format %r found for %r" %
2101
2022
                    (format, self))
2110
2031
                              name=name,
2111
2032
                              a_bzrdir=a_bzrdir,
2112
2033
                              _repository=found_repository,
2113
 
                              ignore_fallbacks=ignore_fallbacks)
 
2034
                              ignore_fallbacks=ignore_fallbacks,
 
2035
                              possible_transports=possible_transports)
2114
2036
        except errors.NoSuchFile:
2115
2037
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
2116
2038
 
2117
 
    def __init__(self):
2118
 
        super(BranchFormatMetadir, self).__init__()
2119
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
2120
 
        self._matchingbzrdir.set_branch_format(self)
 
2039
    @property
 
2040
    def _matchingbzrdir(self):
 
2041
        ret = bzrdir.BzrDirMetaFormat1()
 
2042
        ret.set_branch_format(self)
 
2043
        return ret
2121
2044
 
2122
2045
    def supports_tags(self):
2123
2046
        return True
2125
2048
    def supports_leaving_lock(self):
2126
2049
        return True
2127
2050
 
2128
 
 
2129
 
class BzrBranchFormat5(BranchFormatMetadir):
2130
 
    """Bzr branch format 5.
2131
 
 
2132
 
    This format has:
2133
 
     - a revision-history file.
2134
 
     - a format string
2135
 
     - a lock dir guarding the branch itself
2136
 
     - all of this stored in a branch/ subdirectory
2137
 
     - works with shared repositories.
2138
 
 
2139
 
    This format is new in bzr 0.8.
2140
 
    """
2141
 
 
2142
 
    def _branch_class(self):
2143
 
        return BzrBranch5
2144
 
 
2145
 
    def get_format_string(self):
2146
 
        """See BranchFormat.get_format_string()."""
2147
 
        return "Bazaar-NG branch format 5\n"
2148
 
 
2149
 
    def get_format_description(self):
2150
 
        """See BranchFormat.get_format_description()."""
2151
 
        return "Branch format 5"
2152
 
 
2153
 
    def initialize(self, a_bzrdir, name=None, repository=None):
2154
 
        """Create a branch of this format in a_bzrdir."""
2155
 
        utf8_files = [('revision-history', ''),
2156
 
                      ('branch-name', ''),
2157
 
                      ]
2158
 
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
2159
 
 
2160
 
    def supports_tags(self):
2161
 
        return False
 
2051
    def check_support_status(self, allow_unsupported, recommend_upgrade=True,
 
2052
            basedir=None):
 
2053
        BranchFormat.check_support_status(self,
 
2054
            allow_unsupported=allow_unsupported, recommend_upgrade=recommend_upgrade,
 
2055
            basedir=basedir)
 
2056
        bzrdir.BzrFormat.check_support_status(self, allow_unsupported=allow_unsupported,
 
2057
            recommend_upgrade=recommend_upgrade, basedir=basedir)
2162
2058
 
2163
2059
 
2164
2060
class BzrBranchFormat6(BranchFormatMetadir):
2175
2071
    def _branch_class(self):
2176
2072
        return BzrBranch6
2177
2073
 
2178
 
    def get_format_string(self):
 
2074
    @classmethod
 
2075
    def get_format_string(cls):
2179
2076
        """See BranchFormat.get_format_string()."""
2180
2077
        return "Bazaar Branch Format 6 (bzr 0.15)\n"
2181
2078
 
2183
2080
        """See BranchFormat.get_format_description()."""
2184
2081
        return "Branch format 6"
2185
2082
 
2186
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2083
    def initialize(self, a_bzrdir, name=None, repository=None,
 
2084
                   append_revisions_only=None):
2187
2085
        """Create a branch of this format in a_bzrdir."""
2188
2086
        utf8_files = [('last-revision', '0 null:\n'),
2189
 
                      ('branch.conf', ''),
 
2087
                      ('branch.conf',
 
2088
                          self._get_initial_config(append_revisions_only)),
2190
2089
                      ('tags', ''),
2191
2090
                      ]
2192
2091
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
2193
2092
 
2194
2093
    def make_tags(self, branch):
2195
2094
        """See bzrlib.branch.BranchFormat.make_tags()."""
2196
 
        return BasicTags(branch)
 
2095
        return _mod_tag.BasicTags(branch)
2197
2096
 
2198
2097
    def supports_set_append_revisions_only(self):
2199
2098
        return True
2205
2104
    def _branch_class(self):
2206
2105
        return BzrBranch8
2207
2106
 
2208
 
    def get_format_string(self):
 
2107
    @classmethod
 
2108
    def get_format_string(cls):
2209
2109
        """See BranchFormat.get_format_string()."""
2210
2110
        return "Bazaar Branch Format 8 (needs bzr 1.15)\n"
2211
2111
 
2213
2113
        """See BranchFormat.get_format_description()."""
2214
2114
        return "Branch format 8"
2215
2115
 
2216
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2116
    def initialize(self, a_bzrdir, name=None, repository=None,
 
2117
                   append_revisions_only=None):
2217
2118
        """Create a branch of this format in a_bzrdir."""
2218
2119
        utf8_files = [('last-revision', '0 null:\n'),
2219
 
                      ('branch.conf', ''),
 
2120
                      ('branch.conf',
 
2121
                          self._get_initial_config(append_revisions_only)),
2220
2122
                      ('tags', ''),
2221
2123
                      ('references', '')
2222
2124
                      ]
2223
2125
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
2224
2126
 
2225
 
    def __init__(self):
2226
 
        super(BzrBranchFormat8, self).__init__()
2227
 
        self._matchingbzrdir.repository_format = \
2228
 
            RepositoryFormatKnitPack5RichRoot()
2229
 
 
2230
2127
    def make_tags(self, branch):
2231
2128
        """See bzrlib.branch.BranchFormat.make_tags()."""
2232
 
        return BasicTags(branch)
 
2129
        return _mod_tag.BasicTags(branch)
2233
2130
 
2234
2131
    def supports_set_append_revisions_only(self):
2235
2132
        return True
2240
2137
    supports_reference_locations = True
2241
2138
 
2242
2139
 
2243
 
class BzrBranchFormat7(BzrBranchFormat8):
 
2140
class BzrBranchFormat7(BranchFormatMetadir):
2244
2141
    """Branch format with last-revision, tags, and a stacked location pointer.
2245
2142
 
2246
2143
    The stacked location pointer is passed down to the repository and requires
2249
2146
    This format was introduced in bzr 1.6.
2250
2147
    """
2251
2148
 
2252
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2149
    def initialize(self, a_bzrdir, name=None, repository=None,
 
2150
                   append_revisions_only=None):
2253
2151
        """Create a branch of this format in a_bzrdir."""
2254
2152
        utf8_files = [('last-revision', '0 null:\n'),
2255
 
                      ('branch.conf', ''),
 
2153
                      ('branch.conf',
 
2154
                          self._get_initial_config(append_revisions_only)),
2256
2155
                      ('tags', ''),
2257
2156
                      ]
2258
2157
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
2260
2159
    def _branch_class(self):
2261
2160
        return BzrBranch7
2262
2161
 
2263
 
    def get_format_string(self):
 
2162
    @classmethod
 
2163
    def get_format_string(cls):
2264
2164
        """See BranchFormat.get_format_string()."""
2265
2165
        return "Bazaar Branch Format 7 (needs bzr 1.6)\n"
2266
2166
 
2271
2171
    def supports_set_append_revisions_only(self):
2272
2172
        return True
2273
2173
 
 
2174
    def supports_stacking(self):
 
2175
        return True
 
2176
 
 
2177
    def make_tags(self, branch):
 
2178
        """See bzrlib.branch.BranchFormat.make_tags()."""
 
2179
        return _mod_tag.BasicTags(branch)
 
2180
 
2274
2181
    supports_reference_locations = False
2275
2182
 
2276
2183
 
2277
 
class BranchReferenceFormat(BranchFormat):
 
2184
class BranchReferenceFormat(BranchFormatMetadir):
2278
2185
    """Bzr branch reference format.
2279
2186
 
2280
2187
    Branch references are used in implementing checkouts, they
2285
2192
     - a format string
2286
2193
    """
2287
2194
 
2288
 
    def get_format_string(self):
 
2195
    @classmethod
 
2196
    def get_format_string(cls):
2289
2197
        """See BranchFormat.get_format_string()."""
2290
2198
        return "Bazaar-NG Branch Reference Format 1\n"
2291
2199
 
2304
2212
        location = transport.put_bytes('location', to_branch.base)
2305
2213
 
2306
2214
    def initialize(self, a_bzrdir, name=None, target_branch=None,
2307
 
            repository=None):
 
2215
            repository=None, append_revisions_only=None):
2308
2216
        """Create a branch of this format in a_bzrdir."""
2309
2217
        if target_branch is None:
2310
2218
            # this format does not implement branch itself, thus the implicit
2311
2219
            # creation contract must see it as uninitializable
2312
2220
            raise errors.UninitializableFormat(self)
2313
2221
        mutter('creating branch reference in %s', a_bzrdir.user_url)
 
2222
        if a_bzrdir._format.fixed_components:
 
2223
            raise errors.IncompatibleFormat(self, a_bzrdir._format)
 
2224
        if name is None:
 
2225
            name = a_bzrdir._get_selected_branch()
2314
2226
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
2315
2227
        branch_transport.put_bytes('location',
2316
 
            target_branch.bzrdir.user_url)
2317
 
        branch_transport.put_bytes('format', self.get_format_string())
2318
 
        branch = self.open(
2319
 
            a_bzrdir, name, _found=True,
 
2228
            target_branch.user_url)
 
2229
        branch_transport.put_bytes('format', self.as_string())
 
2230
        branch = self.open(a_bzrdir, name, _found=True,
2320
2231
            possible_transports=[target_branch.bzrdir.root_transport])
2321
2232
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
2322
2233
        return branch
2323
2234
 
2324
 
    def __init__(self):
2325
 
        super(BranchReferenceFormat, self).__init__()
2326
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
2327
 
        self._matchingbzrdir.set_branch_format(self)
2328
 
 
2329
2235
    def _make_reference_clone_function(format, a_branch):
2330
2236
        """Create a clone() routine for a branch dynamically."""
2331
2237
        def clone(to_bzrdir, revision_id=None,
2353
2259
            a_bzrdir.
2354
2260
        :param possible_transports: An optional reusable transports list.
2355
2261
        """
 
2262
        if name is None:
 
2263
            name = a_bzrdir._get_selected_branch()
2356
2264
        if not _found:
2357
 
            format = BranchFormat.find_format(a_bzrdir, name=name)
 
2265
            format = BranchFormatMetadir.find_format(a_bzrdir, name=name)
2358
2266
            if format.__class__ != self.__class__:
2359
2267
                raise AssertionError("wrong format %r found for %r" %
2360
2268
                    (format, self))
2361
2269
        if location is None:
2362
2270
            location = self.get_reference(a_bzrdir, name)
2363
 
        real_bzrdir = bzrdir.BzrDir.open(
 
2271
        real_bzrdir = controldir.ControlDir.open(
2364
2272
            location, possible_transports=possible_transports)
2365
 
        result = real_bzrdir.open_branch(name=name, 
2366
 
            ignore_fallbacks=ignore_fallbacks)
 
2273
        result = real_bzrdir.open_branch(ignore_fallbacks=ignore_fallbacks,
 
2274
            possible_transports=possible_transports)
2367
2275
        # this changes the behaviour of result.clone to create a new reference
2368
2276
        # rather than a copy of the content of the branch.
2369
2277
        # I did not use a proxy object because that needs much more extensive
2403
2311
 
2404
2312
# formats which have no format string are not discoverable
2405
2313
# and not independently creatable, so are not registered.
2406
 
__format4 = BzrBranchFormat4()
2407
 
__format5 = BzrBranchFormat5()
2408
2314
__format6 = BzrBranchFormat6()
2409
2315
__format7 = BzrBranchFormat7()
2410
2316
__format8 = BzrBranchFormat8()
2411
 
format_registry.register(__format5)
 
2317
format_registry.register_lazy(
 
2318
    "Bazaar-NG branch format 5\n", "bzrlib.branchfmt.fullhistory", "BzrBranchFormat5")
2412
2319
format_registry.register(BranchReferenceFormat())
2413
2320
format_registry.register(__format6)
2414
2321
format_registry.register(__format7)
2415
2322
format_registry.register(__format8)
2416
2323
format_registry.set_default(__format7)
2417
 
format_registry.register_extra(__format4)
2418
 
network_format_registry.register(__format4.network_name(), __format4)
2419
2324
 
2420
2325
 
2421
2326
class BranchWriteLockResult(LogicalLockResult):
2453
2358
 
2454
2359
    def __init__(self, _format=None,
2455
2360
                 _control_files=None, a_bzrdir=None, name=None,
2456
 
                 _repository=None, ignore_fallbacks=False):
 
2361
                 _repository=None, ignore_fallbacks=False,
 
2362
                 possible_transports=None):
2457
2363
        """Create new branch object at a particular location."""
2458
2364
        if a_bzrdir is None:
2459
2365
            raise ValueError('a_bzrdir must be supplied')
2460
 
        else:
2461
 
            self.bzrdir = a_bzrdir
2462
 
        self._base = self.bzrdir.transport.clone('..').base
 
2366
        if name is None:
 
2367
            raise ValueError('name must be supplied')
 
2368
        self.bzrdir = a_bzrdir
 
2369
        self._user_transport = self.bzrdir.transport.clone('..')
 
2370
        if name != "":
 
2371
            self._user_transport.set_segment_parameter(
 
2372
                "branch", urlutils.escape(name))
 
2373
        self._base = self._user_transport.base
2463
2374
        self.name = name
2464
 
        # XXX: We should be able to just do
2465
 
        #   self.base = self.bzrdir.root_transport.base
2466
 
        # but this does not quite work yet -- mbp 20080522
2467
2375
        self._format = _format
2468
2376
        if _control_files is None:
2469
2377
            raise ValueError('BzrBranch _control_files is None')
2470
2378
        self.control_files = _control_files
2471
2379
        self._transport = _control_files._transport
2472
2380
        self.repository = _repository
2473
 
        Branch.__init__(self)
 
2381
        self.conf_store = None
 
2382
        Branch.__init__(self, possible_transports)
2474
2383
 
2475
2384
    def __str__(self):
2476
 
        if self.name is None:
2477
 
            return '%s(%s)' % (self.__class__.__name__, self.user_url)
2478
 
        else:
2479
 
            return '%s(%s,%s)' % (self.__class__.__name__, self.user_url,
2480
 
                self.name)
 
2385
        return '%s(%s)' % (self.__class__.__name__, self.user_url)
2481
2386
 
2482
2387
    __repr__ = __str__
2483
2388
 
2487
2392
 
2488
2393
    base = property(_get_base, doc="The URL for the root of this branch.")
2489
2394
 
 
2395
    @property
 
2396
    def user_transport(self):
 
2397
        return self._user_transport
 
2398
 
2490
2399
    def _get_config(self):
2491
 
        return TransportConfig(self._transport, 'branch.conf')
 
2400
        return _mod_config.TransportConfig(self._transport, 'branch.conf')
 
2401
 
 
2402
    def _get_config_store(self):
 
2403
        if self.conf_store is None:
 
2404
            self.conf_store =  _mod_config.BranchStore(self)
 
2405
        return self.conf_store
 
2406
 
 
2407
    def _uncommitted_branch(self):
 
2408
        """Return the branch that may contain uncommitted changes."""
 
2409
        master = self.get_master_branch()
 
2410
        if master is not None:
 
2411
            return master
 
2412
        else:
 
2413
            return self
 
2414
 
 
2415
    def store_uncommitted(self, creator):
 
2416
        """Store uncommitted changes from a ShelfCreator.
 
2417
 
 
2418
        :param creator: The ShelfCreator containing uncommitted changes, or
 
2419
            None to delete any stored changes.
 
2420
        :raises: ChangesAlreadyStored if the branch already has changes.
 
2421
        """
 
2422
        branch = self._uncommitted_branch()
 
2423
        if creator is None:
 
2424
            branch._transport.delete('stored-transform')
 
2425
            return
 
2426
        if branch._transport.has('stored-transform'):
 
2427
            raise errors.ChangesAlreadyStored
 
2428
        transform = StringIO()
 
2429
        creator.write_shelf(transform)
 
2430
        transform.seek(0)
 
2431
        branch._transport.put_file('stored-transform', transform)
 
2432
 
 
2433
    def get_unshelver(self, tree):
 
2434
        """Return a shelf.Unshelver for this branch and tree.
 
2435
 
 
2436
        :param tree: The tree to use to construct the Unshelver.
 
2437
        :return: an Unshelver or None if no changes are stored.
 
2438
        """
 
2439
        branch = self._uncommitted_branch()
 
2440
        try:
 
2441
            transform = branch._transport.get('stored-transform')
 
2442
        except errors.NoSuchFile:
 
2443
            return None
 
2444
        return shelf.Unshelver.from_tree_and_shelf(tree, transform)
2492
2445
 
2493
2446
    def is_locked(self):
2494
2447
        return self.control_files.is_locked()
2502
2455
        """
2503
2456
        if not self.is_locked():
2504
2457
            self._note_lock('w')
2505
 
        # All-in-one needs to always unlock/lock.
2506
 
        repo_control = getattr(self.repository, 'control_files', None)
2507
 
        if self.control_files == repo_control or not self.is_locked():
2508
2458
            self.repository._warn_if_deprecated(self)
2509
2459
            self.repository.lock_write()
2510
2460
            took_lock = True
2525
2475
        """
2526
2476
        if not self.is_locked():
2527
2477
            self._note_lock('r')
2528
 
        # All-in-one needs to always unlock/lock.
2529
 
        repo_control = getattr(self.repository, 'control_files', None)
2530
 
        if self.control_files == repo_control or not self.is_locked():
2531
2478
            self.repository._warn_if_deprecated(self)
2532
2479
            self.repository.lock_read()
2533
2480
            took_lock = True
2543
2490
 
2544
2491
    @only_raises(errors.LockNotHeld, errors.LockBroken)
2545
2492
    def unlock(self):
 
2493
        if self.control_files._lock_count == 1 and self.conf_store is not None:
 
2494
            self.conf_store.save_changes()
2546
2495
        try:
2547
2496
            self.control_files.unlock()
2548
2497
        finally:
2549
 
            # All-in-one needs to always unlock/lock.
2550
 
            repo_control = getattr(self.repository, 'control_files', None)
2551
 
            if (self.control_files == repo_control or
2552
 
                not self.control_files.is_locked()):
2553
 
                self.repository.unlock()
2554
2498
            if not self.control_files.is_locked():
 
2499
                self.repository.unlock()
2555
2500
                # we just released the lock
2556
2501
                self._clear_cached_state()
2557
2502
 
2569
2514
        """See Branch.print_file."""
2570
2515
        return self.repository.print_file(file, revision_id)
2571
2516
 
2572
 
    def _write_revision_history(self, history):
2573
 
        """Factored out of set_revision_history.
2574
 
 
2575
 
        This performs the actual writing to disk.
2576
 
        It is intended to be called by BzrBranch5.set_revision_history."""
2577
 
        self._transport.put_bytes(
2578
 
            'revision-history', '\n'.join(history),
2579
 
            mode=self.bzrdir._get_file_mode())
2580
 
 
2581
 
    @needs_write_lock
2582
 
    def set_revision_history(self, rev_history):
2583
 
        """See Branch.set_revision_history."""
2584
 
        if 'evil' in debug.debug_flags:
2585
 
            mutter_callsite(3, "set_revision_history scales with history.")
2586
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
2587
 
        for rev_id in rev_history:
2588
 
            check_not_reserved_id(rev_id)
2589
 
        if Branch.hooks['post_change_branch_tip']:
2590
 
            # Don't calculate the last_revision_info() if there are no hooks
2591
 
            # that will use it.
2592
 
            old_revno, old_revid = self.last_revision_info()
2593
 
        if len(rev_history) == 0:
2594
 
            revid = _mod_revision.NULL_REVISION
2595
 
        else:
2596
 
            revid = rev_history[-1]
2597
 
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
2598
 
        self._write_revision_history(rev_history)
2599
 
        self._clear_cached_state()
2600
 
        self._cache_revision_history(rev_history)
2601
 
        for hook in Branch.hooks['set_rh']:
2602
 
            hook(self, rev_history)
2603
 
        if Branch.hooks['post_change_branch_tip']:
2604
 
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2605
 
 
2606
 
    def _synchronize_history(self, destination, revision_id):
2607
 
        """Synchronize last revision and revision history between branches.
2608
 
 
2609
 
        This version is most efficient when the destination is also a
2610
 
        BzrBranch5, but works for BzrBranch6 as long as the revision
2611
 
        history is the true lefthand parent history, and all of the revisions
2612
 
        are in the destination's repository.  If not, set_revision_history
2613
 
        will fail.
2614
 
 
2615
 
        :param destination: The branch to copy the history into
2616
 
        :param revision_id: The revision-id to truncate history at.  May
2617
 
          be None to copy complete history.
2618
 
        """
2619
 
        if not isinstance(destination._format, BzrBranchFormat5):
2620
 
            super(BzrBranch, self)._synchronize_history(
2621
 
                destination, revision_id)
2622
 
            return
2623
 
        if revision_id == _mod_revision.NULL_REVISION:
2624
 
            new_history = []
2625
 
        else:
2626
 
            new_history = self.revision_history()
2627
 
        if revision_id is not None and new_history != []:
2628
 
            try:
2629
 
                new_history = new_history[:new_history.index(revision_id) + 1]
2630
 
            except ValueError:
2631
 
                rev = self.repository.get_revision(revision_id)
2632
 
                new_history = rev.get_history(self.repository)[1:]
2633
 
        destination.set_revision_history(new_history)
2634
 
 
2635
2517
    @needs_write_lock
2636
2518
    def set_last_revision_info(self, revno, revision_id):
2637
 
        """Set the last revision of this branch.
2638
 
 
2639
 
        The caller is responsible for checking that the revno is correct
2640
 
        for this revision id.
2641
 
 
2642
 
        It may be possible to set the branch last revision to an id not
2643
 
        present in the repository.  However, branches can also be
2644
 
        configured to check constraints on history, in which case this may not
2645
 
        be permitted.
2646
 
        """
 
2519
        if not revision_id or not isinstance(revision_id, basestring):
 
2520
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
2647
2521
        revision_id = _mod_revision.ensure_null(revision_id)
2648
 
        # this old format stores the full history, but this api doesn't
2649
 
        # provide it, so we must generate, and might as well check it's
2650
 
        # correct
2651
 
        history = self._lefthand_history(revision_id)
2652
 
        if len(history) != revno:
2653
 
            raise AssertionError('%d != %d' % (len(history), revno))
2654
 
        self.set_revision_history(history)
2655
 
 
2656
 
    def _gen_revision_history(self):
2657
 
        history = self._transport.get_bytes('revision-history').split('\n')
2658
 
        if history[-1:] == ['']:
2659
 
            # There shouldn't be a trailing newline, but just in case.
2660
 
            history.pop()
2661
 
        return history
2662
 
 
2663
 
    @needs_write_lock
2664
 
    def generate_revision_history(self, revision_id, last_rev=None,
2665
 
        other_branch=None):
2666
 
        """Create a new revision history that will finish with revision_id.
2667
 
 
2668
 
        :param revision_id: the new tip to use.
2669
 
        :param last_rev: The previous last_revision. If not None, then this
2670
 
            must be a ancestory of revision_id, or DivergedBranches is raised.
2671
 
        :param other_branch: The other branch that DivergedBranches should
2672
 
            raise with respect to.
2673
 
        """
2674
 
        self.set_revision_history(self._lefthand_history(revision_id,
2675
 
            last_rev, other_branch))
 
2522
        old_revno, old_revid = self.last_revision_info()
 
2523
        if self.get_append_revisions_only():
 
2524
            self._check_history_violation(revision_id)
 
2525
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
 
2526
        self._write_last_revision_info(revno, revision_id)
 
2527
        self._clear_cached_state()
 
2528
        self._last_revision_info_cache = revno, revision_id
 
2529
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2676
2530
 
2677
2531
    def basis_tree(self):
2678
2532
        """See Branch.basis_tree."""
2687
2541
                pass
2688
2542
        return None
2689
2543
 
2690
 
    def _basic_push(self, target, overwrite, stop_revision):
2691
 
        """Basic implementation of push without bound branches or hooks.
2692
 
 
2693
 
        Must be called with source read locked and target write locked.
2694
 
        """
2695
 
        result = BranchPushResult()
2696
 
        result.source_branch = self
2697
 
        result.target_branch = target
2698
 
        result.old_revno, result.old_revid = target.last_revision_info()
2699
 
        self.update_references(target)
2700
 
        if result.old_revid != stop_revision:
2701
 
            # We assume that during 'push' this repository is closer than
2702
 
            # the target.
2703
 
            graph = self.repository.get_graph(target.repository)
2704
 
            target.update_revisions(self, stop_revision,
2705
 
                overwrite=overwrite, graph=graph)
2706
 
        if self._push_should_merge_tags():
2707
 
            result.tag_conflicts = self.tags.merge_to(target.tags,
2708
 
                overwrite)
2709
 
        result.new_revno, result.new_revid = target.last_revision_info()
2710
 
        return result
2711
 
 
2712
2544
    def get_stacked_on_url(self):
2713
2545
        raise errors.UnstackableBranchFormat(self._format, self.user_url)
2714
2546
 
2725
2557
            self._transport.put_bytes('parent', url + '\n',
2726
2558
                mode=self.bzrdir._get_file_mode())
2727
2559
 
2728
 
 
2729
 
class BzrBranchPreSplitOut(BzrBranch):
2730
 
 
2731
 
    def _get_checkout_format(self):
2732
 
        """Return the most suitable metadir for a checkout of this branch.
2733
 
        Weaves are used if this branch's repository uses weaves.
2734
 
        """
2735
 
        from bzrlib.repofmt.weaverepo import RepositoryFormat7
2736
 
        from bzrlib.bzrdir import BzrDirMetaFormat1
2737
 
        format = BzrDirMetaFormat1()
2738
 
        format.repository_format = RepositoryFormat7()
2739
 
        return format
2740
 
 
2741
 
 
2742
 
class BzrBranch5(BzrBranch):
2743
 
    """A format 5 branch. This supports new features over plain branches.
2744
 
 
2745
 
    It has support for a master_branch which is the data for bound branches.
2746
 
    """
2747
 
 
2748
 
    def get_bound_location(self):
2749
 
        try:
2750
 
            return self._transport.get_bytes('bound')[:-1]
2751
 
        except errors.NoSuchFile:
2752
 
            return None
2753
 
 
2754
 
    @needs_read_lock
2755
 
    def get_master_branch(self, possible_transports=None):
2756
 
        """Return the branch we are bound to.
2757
 
 
2758
 
        :return: Either a Branch, or None
2759
 
 
2760
 
        This could memoise the branch, but if thats done
2761
 
        it must be revalidated on each new lock.
2762
 
        So for now we just don't memoise it.
2763
 
        # RBC 20060304 review this decision.
2764
 
        """
2765
 
        bound_loc = self.get_bound_location()
2766
 
        if not bound_loc:
2767
 
            return None
2768
 
        try:
2769
 
            return Branch.open(bound_loc,
2770
 
                               possible_transports=possible_transports)
2771
 
        except (errors.NotBranchError, errors.ConnectionError), e:
2772
 
            raise errors.BoundBranchConnectionFailure(
2773
 
                    self, bound_loc, e)
2774
 
 
2775
2560
    @needs_write_lock
2776
 
    def set_bound_location(self, location):
2777
 
        """Set the target where this branch is bound to.
2778
 
 
2779
 
        :param location: URL to the target branch
2780
 
        """
2781
 
        if location:
2782
 
            self._transport.put_bytes('bound', location+'\n',
2783
 
                mode=self.bzrdir._get_file_mode())
2784
 
        else:
2785
 
            try:
2786
 
                self._transport.delete('bound')
2787
 
            except errors.NoSuchFile:
2788
 
                return False
2789
 
            return True
 
2561
    def unbind(self):
 
2562
        """If bound, unbind"""
 
2563
        return self.set_bound_location(None)
2790
2564
 
2791
2565
    @needs_write_lock
2792
2566
    def bind(self, other):
2814
2588
        # history around
2815
2589
        self.set_bound_location(other.base)
2816
2590
 
 
2591
    def get_bound_location(self):
 
2592
        try:
 
2593
            return self._transport.get_bytes('bound')[:-1]
 
2594
        except errors.NoSuchFile:
 
2595
            return None
 
2596
 
 
2597
    @needs_read_lock
 
2598
    def get_master_branch(self, possible_transports=None):
 
2599
        """Return the branch we are bound to.
 
2600
 
 
2601
        :return: Either a Branch, or None
 
2602
        """
 
2603
        if self._master_branch_cache is None:
 
2604
            self._master_branch_cache = self._get_master_branch(
 
2605
                possible_transports)
 
2606
        return self._master_branch_cache
 
2607
 
 
2608
    def _get_master_branch(self, possible_transports):
 
2609
        bound_loc = self.get_bound_location()
 
2610
        if not bound_loc:
 
2611
            return None
 
2612
        try:
 
2613
            return Branch.open(bound_loc,
 
2614
                               possible_transports=possible_transports)
 
2615
        except (errors.NotBranchError, errors.ConnectionError), e:
 
2616
            raise errors.BoundBranchConnectionFailure(
 
2617
                    self, bound_loc, e)
 
2618
 
2817
2619
    @needs_write_lock
2818
 
    def unbind(self):
2819
 
        """If bound, unbind"""
2820
 
        return self.set_bound_location(None)
 
2620
    def set_bound_location(self, location):
 
2621
        """Set the target where this branch is bound to.
 
2622
 
 
2623
        :param location: URL to the target branch
 
2624
        """
 
2625
        self._master_branch_cache = None
 
2626
        if location:
 
2627
            self._transport.put_bytes('bound', location+'\n',
 
2628
                mode=self.bzrdir._get_file_mode())
 
2629
        else:
 
2630
            try:
 
2631
                self._transport.delete('bound')
 
2632
            except errors.NoSuchFile:
 
2633
                return False
 
2634
            return True
2821
2635
 
2822
2636
    @needs_write_lock
2823
2637
    def update(self, possible_transports=None):
2836
2650
            return old_tip
2837
2651
        return None
2838
2652
 
2839
 
 
2840
 
class BzrBranch8(BzrBranch5):
 
2653
    def _read_last_revision_info(self):
 
2654
        revision_string = self._transport.get_bytes('last-revision')
 
2655
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
 
2656
        revision_id = cache_utf8.get_cached_utf8(revision_id)
 
2657
        revno = int(revno)
 
2658
        return revno, revision_id
 
2659
 
 
2660
    def _write_last_revision_info(self, revno, revision_id):
 
2661
        """Simply write out the revision id, with no checks.
 
2662
 
 
2663
        Use set_last_revision_info to perform this safely.
 
2664
 
 
2665
        Does not update the revision_history cache.
 
2666
        """
 
2667
        revision_id = _mod_revision.ensure_null(revision_id)
 
2668
        out_string = '%d %s\n' % (revno, revision_id)
 
2669
        self._transport.put_bytes('last-revision', out_string,
 
2670
            mode=self.bzrdir._get_file_mode())
 
2671
 
 
2672
    @needs_write_lock
 
2673
    def update_feature_flags(self, updated_flags):
 
2674
        """Update the feature flags for this branch.
 
2675
 
 
2676
        :param updated_flags: Dictionary mapping feature names to necessities
 
2677
            A necessity can be None to indicate the feature should be removed
 
2678
        """
 
2679
        self._format._update_feature_flags(updated_flags)
 
2680
        self.control_transport.put_bytes('format', self._format.as_string())
 
2681
 
 
2682
 
 
2683
class BzrBranch8(BzrBranch):
2841
2684
    """A branch that stores tree-reference locations."""
2842
2685
 
2843
 
    def _open_hook(self):
 
2686
    def _open_hook(self, possible_transports=None):
2844
2687
        if self._ignore_fallbacks:
2845
2688
            return
 
2689
        if possible_transports is None:
 
2690
            possible_transports = [self.bzrdir.root_transport]
2846
2691
        try:
2847
2692
            url = self.get_stacked_on_url()
2848
2693
        except (errors.UnstackableRepositoryFormat, errors.NotStacked,
2856
2701
                    raise AssertionError(
2857
2702
                        "'transform_fallback_location' hook %s returned "
2858
2703
                        "None, not a URL." % hook_name)
2859
 
            self._activate_fallback_location(url)
 
2704
            self._activate_fallback_location(url,
 
2705
                possible_transports=possible_transports)
2860
2706
 
2861
2707
    def __init__(self, *args, **kwargs):
2862
2708
        self._ignore_fallbacks = kwargs.get('ignore_fallbacks', False)
2869
2715
        self._last_revision_info_cache = None
2870
2716
        self._reference_info = None
2871
2717
 
2872
 
    def _last_revision_info(self):
2873
 
        revision_string = self._transport.get_bytes('last-revision')
2874
 
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
2875
 
        revision_id = cache_utf8.get_cached_utf8(revision_id)
2876
 
        revno = int(revno)
2877
 
        return revno, revision_id
2878
 
 
2879
 
    def _write_last_revision_info(self, revno, revision_id):
2880
 
        """Simply write out the revision id, with no checks.
2881
 
 
2882
 
        Use set_last_revision_info to perform this safely.
2883
 
 
2884
 
        Does not update the revision_history cache.
2885
 
        Intended to be called by set_last_revision_info and
2886
 
        _write_revision_history.
2887
 
        """
2888
 
        revision_id = _mod_revision.ensure_null(revision_id)
2889
 
        out_string = '%d %s\n' % (revno, revision_id)
2890
 
        self._transport.put_bytes('last-revision', out_string,
2891
 
            mode=self.bzrdir._get_file_mode())
2892
 
 
2893
 
    @needs_write_lock
2894
 
    def set_last_revision_info(self, revno, revision_id):
2895
 
        revision_id = _mod_revision.ensure_null(revision_id)
2896
 
        old_revno, old_revid = self.last_revision_info()
2897
 
        if self._get_append_revisions_only():
2898
 
            self._check_history_violation(revision_id)
2899
 
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
2900
 
        self._write_last_revision_info(revno, revision_id)
2901
 
        self._clear_cached_state()
2902
 
        self._last_revision_info_cache = revno, revision_id
2903
 
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2904
 
 
2905
 
    def _synchronize_history(self, destination, revision_id):
2906
 
        """Synchronize last revision and revision history between branches.
2907
 
 
2908
 
        :see: Branch._synchronize_history
2909
 
        """
2910
 
        # XXX: The base Branch has a fast implementation of this method based
2911
 
        # on set_last_revision_info, but BzrBranch/BzrBranch5 have a slower one
2912
 
        # that uses set_revision_history.  This class inherits from BzrBranch5,
2913
 
        # but wants the fast implementation, so it calls
2914
 
        # Branch._synchronize_history directly.
2915
 
        Branch._synchronize_history(self, destination, revision_id)
2916
 
 
2917
2718
    def _check_history_violation(self, revision_id):
2918
 
        last_revision = _mod_revision.ensure_null(self.last_revision())
 
2719
        current_revid = self.last_revision()
 
2720
        last_revision = _mod_revision.ensure_null(current_revid)
2919
2721
        if _mod_revision.is_null(last_revision):
2920
2722
            return
2921
 
        if last_revision not in self._lefthand_history(revision_id):
2922
 
            raise errors.AppendRevisionsOnlyViolation(self.user_url)
 
2723
        graph = self.repository.get_graph()
 
2724
        for lh_ancestor in graph.iter_lefthand_ancestry(revision_id):
 
2725
            if lh_ancestor == current_revid:
 
2726
                return
 
2727
        raise errors.AppendRevisionsOnlyViolation(self.user_url)
2923
2728
 
2924
2729
    def _gen_revision_history(self):
2925
2730
        """Generate the revision history from last revision
2928
2733
        self._extend_partial_history(stop_index=last_revno-1)
2929
2734
        return list(reversed(self._partial_revision_history_cache))
2930
2735
 
2931
 
    def _write_revision_history(self, history):
2932
 
        """Factored out of set_revision_history.
2933
 
 
2934
 
        This performs the actual writing to disk, with format-specific checks.
2935
 
        It is intended to be called by BzrBranch5.set_revision_history.
2936
 
        """
2937
 
        if len(history) == 0:
2938
 
            last_revision = 'null:'
2939
 
        else:
2940
 
            if history != self._lefthand_history(history[-1]):
2941
 
                raise errors.NotLefthandHistory(history)
2942
 
            last_revision = history[-1]
2943
 
        if self._get_append_revisions_only():
2944
 
            self._check_history_violation(last_revision)
2945
 
        self._write_last_revision_info(len(history), last_revision)
2946
 
 
2947
2736
    @needs_write_lock
2948
2737
    def _set_parent_location(self, url):
2949
2738
        """Set the parent branch"""
3035
2824
 
3036
2825
    def set_bound_location(self, location):
3037
2826
        """See Branch.set_push_location."""
 
2827
        self._master_branch_cache = None
3038
2828
        result = None
3039
 
        config = self.get_config()
 
2829
        conf = self.get_config_stack()
3040
2830
        if location is None:
3041
 
            if config.get_user_option('bound') != 'True':
 
2831
            if not conf.get('bound'):
3042
2832
                return False
3043
2833
            else:
3044
 
                config.set_user_option('bound', 'False', warn_masked=True)
 
2834
                conf.set('bound', 'False')
3045
2835
                return True
3046
2836
        else:
3047
2837
            self._set_config_location('bound_location', location,
3048
 
                                      config=config)
3049
 
            config.set_user_option('bound', 'True', warn_masked=True)
 
2838
                                      config=conf)
 
2839
            conf.set('bound', 'True')
3050
2840
        return True
3051
2841
 
3052
2842
    def _get_bound_location(self, bound):
3053
2843
        """Return the bound location in the config file.
3054
2844
 
3055
2845
        Return None if the bound parameter does not match"""
3056
 
        config = self.get_config()
3057
 
        config_bound = (config.get_user_option('bound') == 'True')
3058
 
        if config_bound != bound:
 
2846
        conf = self.get_config_stack()
 
2847
        if conf.get('bound') != bound:
3059
2848
            return None
3060
 
        return self._get_config_location('bound_location', config=config)
 
2849
        return self._get_config_location('bound_location', config=conf)
3061
2850
 
3062
2851
    def get_bound_location(self):
3063
 
        """See Branch.set_push_location."""
 
2852
        """See Branch.get_bound_location."""
3064
2853
        return self._get_bound_location(True)
3065
2854
 
3066
2855
    def get_old_bound_location(self):
3071
2860
        # you can always ask for the URL; but you might not be able to use it
3072
2861
        # if the repo can't support stacking.
3073
2862
        ## self._check_stackable_repo()
3074
 
        stacked_url = self._get_config_location('stacked_on_location')
 
2863
        # stacked_on_location is only ever defined in branch.conf, so don't
 
2864
        # waste effort reading the whole stack of config files.
 
2865
        conf = _mod_config.BranchOnlyStack(self)
 
2866
        stacked_url = self._get_config_location('stacked_on_location',
 
2867
                                                config=conf)
3075
2868
        if stacked_url is None:
3076
2869
            raise errors.NotStacked(self)
3077
 
        return stacked_url
3078
 
 
3079
 
    def _get_append_revisions_only(self):
3080
 
        return self.get_config(
3081
 
            ).get_user_option_as_bool('append_revisions_only')
3082
 
 
3083
 
    @needs_write_lock
3084
 
    def generate_revision_history(self, revision_id, last_rev=None,
3085
 
                                  other_branch=None):
3086
 
        """See BzrBranch5.generate_revision_history"""
3087
 
        history = self._lefthand_history(revision_id, last_rev, other_branch)
3088
 
        revno = len(history)
3089
 
        self.set_last_revision_info(revno, revision_id)
 
2870
        return stacked_url.encode('utf-8')
3090
2871
 
3091
2872
    @needs_read_lock
3092
2873
    def get_rev_id(self, revno, history=None):
3122
2903
            except errors.RevisionNotPresent, e:
3123
2904
                raise errors.GhostRevisionsHaveNoRevno(revision_id, e.revision_id)
3124
2905
            index = len(self._partial_revision_history_cache) - 1
 
2906
            if index < 0:
 
2907
                raise errors.NoSuchRevision(self, revision_id)
3125
2908
            if self._partial_revision_history_cache[index] != revision_id:
3126
2909
                raise errors.NoSuchRevision(self, revision_id)
3127
2910
        return self.revno() - index
3179
2962
    :ivar local_branch: target branch if there is a Master, else None
3180
2963
    :ivar target_branch: Target/destination branch object. (write locked)
3181
2964
    :ivar tag_conflicts: A list of tag conflicts, see BasicTags.merge_to
 
2965
    :ivar tag_updates: A dict with new tags, see BasicTags.merge_to
3182
2966
    """
3183
2967
 
3184
 
    @deprecated_method(deprecated_in((2, 3, 0)))
3185
 
    def __int__(self):
3186
 
        """Return the relative change in revno.
3187
 
 
3188
 
        :deprecated: Use `new_revno` and `old_revno` instead.
3189
 
        """
3190
 
        return self.new_revno - self.old_revno
3191
 
 
3192
2968
    def report(self, to_file):
 
2969
        tag_conflicts = getattr(self, "tag_conflicts", None)
 
2970
        tag_updates = getattr(self, "tag_updates", None)
3193
2971
        if not is_quiet():
3194
 
            if self.old_revid == self.new_revid:
3195
 
                to_file.write('No revisions to pull.\n')
3196
 
            else:
 
2972
            if self.old_revid != self.new_revid:
3197
2973
                to_file.write('Now on revision %d.\n' % self.new_revno)
 
2974
            if tag_updates:
 
2975
                to_file.write('%d tag(s) updated.\n' % len(tag_updates))
 
2976
            if self.old_revid == self.new_revid and not tag_updates:
 
2977
                if not tag_conflicts:
 
2978
                    to_file.write('No revisions or tags to pull.\n')
 
2979
                else:
 
2980
                    to_file.write('No revisions to pull.\n')
3198
2981
        self._show_tag_conficts(to_file)
3199
2982
 
3200
2983
 
3217
3000
        target, otherwise it will be None.
3218
3001
    """
3219
3002
 
3220
 
    @deprecated_method(deprecated_in((2, 3, 0)))
3221
 
    def __int__(self):
3222
 
        """Return the relative change in revno.
3223
 
 
3224
 
        :deprecated: Use `new_revno` and `old_revno` instead.
3225
 
        """
3226
 
        return self.new_revno - self.old_revno
3227
 
 
3228
3003
    def report(self, to_file):
3229
 
        """Write a human-readable description of the result."""
3230
 
        if self.old_revid == self.new_revid:
3231
 
            note('No new revisions to push.')
3232
 
        else:
3233
 
            note('Pushed up to revision %d.' % self.new_revno)
 
3004
        # TODO: This function gets passed a to_file, but then
 
3005
        # ignores it and calls note() instead. This is also
 
3006
        # inconsistent with PullResult(), which writes to stdout.
 
3007
        # -- JRV20110901, bug #838853
 
3008
        tag_conflicts = getattr(self, "tag_conflicts", None)
 
3009
        tag_updates = getattr(self, "tag_updates", None)
 
3010
        if not is_quiet():
 
3011
            if self.old_revid != self.new_revid:
 
3012
                note(gettext('Pushed up to revision %d.') % self.new_revno)
 
3013
            if tag_updates:
 
3014
                note(ngettext('%d tag updated.', '%d tags updated.', len(tag_updates)) % len(tag_updates))
 
3015
            if self.old_revid == self.new_revid and not tag_updates:
 
3016
                if not tag_conflicts:
 
3017
                    note(gettext('No new revisions or tags to push.'))
 
3018
                else:
 
3019
                    note(gettext('No new revisions to push.'))
3234
3020
        self._show_tag_conficts(to_file)
3235
3021
 
3236
3022
 
3250
3036
        :param verbose: Requests more detailed display of what was checked,
3251
3037
            if any.
3252
3038
        """
3253
 
        note('checked branch %s format %s', self.branch.user_url,
3254
 
            self.branch._format)
 
3039
        note(gettext('checked branch {0} format {1}').format(
 
3040
                                self.branch.user_url, self.branch._format))
3255
3041
        for error in self.errors:
3256
 
            note('found error:%s', error)
 
3042
            note(gettext('found error:%s'), error)
3257
3043
 
3258
3044
 
3259
3045
class Converter5to6(object):
3266
3052
 
3267
3053
        # Copy source data into target
3268
3054
        new_branch._write_last_revision_info(*branch.last_revision_info())
3269
 
        new_branch.set_parent(branch.get_parent())
3270
 
        new_branch.set_bound_location(branch.get_bound_location())
3271
 
        new_branch.set_push_location(branch.get_push_location())
 
3055
        new_branch.lock_write()
 
3056
        try:
 
3057
            new_branch.set_parent(branch.get_parent())
 
3058
            new_branch.set_bound_location(branch.get_bound_location())
 
3059
            new_branch.set_push_location(branch.get_push_location())
 
3060
        finally:
 
3061
            new_branch.unlock()
3272
3062
 
3273
3063
        # New branch has no tags by default
3274
3064
        new_branch.tags._set_tag_dict({})
3275
3065
 
3276
3066
        # Copying done; now update target format
3277
3067
        new_branch._transport.put_bytes('format',
3278
 
            format.get_format_string(),
 
3068
            format.as_string(),
3279
3069
            mode=new_branch.bzrdir._get_file_mode())
3280
3070
 
3281
3071
        # Clean up old files
3282
3072
        new_branch._transport.delete('revision-history')
 
3073
        branch.lock_write()
3283
3074
        try:
3284
 
            branch.set_parent(None)
3285
 
        except errors.NoSuchFile:
3286
 
            pass
3287
 
        branch.set_bound_location(None)
 
3075
            try:
 
3076
                branch.set_parent(None)
 
3077
            except errors.NoSuchFile:
 
3078
                pass
 
3079
            branch.set_bound_location(None)
 
3080
        finally:
 
3081
            branch.unlock()
3288
3082
 
3289
3083
 
3290
3084
class Converter6to7(object):
3294
3088
        format = BzrBranchFormat7()
3295
3089
        branch._set_config_location('stacked_on_location', '')
3296
3090
        # update target format
3297
 
        branch._transport.put_bytes('format', format.get_format_string())
 
3091
        branch._transport.put_bytes('format', format.as_string())
3298
3092
 
3299
3093
 
3300
3094
class Converter7to8(object):
3301
 
    """Perform an in-place upgrade of format 6 to format 7"""
 
3095
    """Perform an in-place upgrade of format 7 to format 8"""
3302
3096
 
3303
3097
    def convert(self, branch):
3304
3098
        format = BzrBranchFormat8()
3305
3099
        branch._transport.put_bytes('references', '')
3306
3100
        # update target format
3307
 
        branch._transport.put_bytes('format', format.get_format_string())
3308
 
 
3309
 
 
3310
 
def _run_with_write_locked_target(target, callable, *args, **kwargs):
3311
 
    """Run ``callable(*args, **kwargs)``, write-locking target for the
3312
 
    duration.
3313
 
 
3314
 
    _run_with_write_locked_target will attempt to release the lock it acquires.
3315
 
 
3316
 
    If an exception is raised by callable, then that exception *will* be
3317
 
    propagated, even if the unlock attempt raises its own error.  Thus
3318
 
    _run_with_write_locked_target should be preferred to simply doing::
3319
 
 
3320
 
        target.lock_write()
3321
 
        try:
3322
 
            return callable(*args, **kwargs)
3323
 
        finally:
3324
 
            target.unlock()
3325
 
 
3326
 
    """
3327
 
    # This is very similar to bzrlib.decorators.needs_write_lock.  Perhaps they
3328
 
    # should share code?
3329
 
    target.lock_write()
3330
 
    try:
3331
 
        result = callable(*args, **kwargs)
3332
 
    except:
3333
 
        exc_info = sys.exc_info()
3334
 
        try:
3335
 
            target.unlock()
3336
 
        finally:
3337
 
            raise exc_info[0], exc_info[1], exc_info[2]
3338
 
    else:
3339
 
        target.unlock()
3340
 
        return result
 
3101
        branch._transport.put_bytes('format', format.as_string())
3341
3102
 
3342
3103
 
3343
3104
class InterBranch(InterObject):
3373
3134
        raise NotImplementedError(self.pull)
3374
3135
 
3375
3136
    @needs_write_lock
3376
 
    def update_revisions(self, stop_revision=None, overwrite=False,
3377
 
                         graph=None, fetch_tags=True):
3378
 
        """Pull in new perfect-fit revisions.
3379
 
 
3380
 
        :param stop_revision: Updated until the given revision
3381
 
        :param overwrite: Always set the branch pointer, rather than checking
3382
 
            to see if it is a proper descendant.
3383
 
        :param graph: A Graph object that can be used to query history
3384
 
            information. This can be None.
3385
 
        :param fetch_tags: Flag that specifies if tags from source should be
3386
 
            fetched too.
3387
 
        :return: None
3388
 
        """
3389
 
        raise NotImplementedError(self.update_revisions)
3390
 
 
3391
 
    @needs_write_lock
3392
 
    def push(self, overwrite=False, stop_revision=None,
 
3137
    def push(self, overwrite=False, stop_revision=None, lossy=False,
3393
3138
             _override_hook_source_branch=None):
3394
3139
        """Mirror the source branch into the target branch.
3395
3140
 
3406
3151
        """
3407
3152
        raise NotImplementedError(self.copy_content_into)
3408
3153
 
 
3154
    @needs_write_lock
 
3155
    def fetch(self, stop_revision=None, limit=None):
 
3156
        """Fetch revisions.
 
3157
 
 
3158
        :param stop_revision: Last revision to fetch
 
3159
        :param limit: Optional rough limit of revisions to fetch
 
3160
        """
 
3161
        raise NotImplementedError(self.fetch)
 
3162
 
 
3163
 
 
3164
def _fix_overwrite_type(overwrite):
 
3165
    if isinstance(overwrite, bool):
 
3166
        if overwrite:
 
3167
            return ["history", "tags"]
 
3168
        else:
 
3169
            return []
 
3170
    return overwrite
 
3171
 
3409
3172
 
3410
3173
class GenericInterBranch(InterBranch):
3411
3174
    """InterBranch implementation that uses public Branch functions."""
3446
3209
            self.source.tags.merge_to(self.target.tags)
3447
3210
 
3448
3211
    @needs_write_lock
3449
 
    def update_revisions(self, stop_revision=None, overwrite=False,
3450
 
        graph=None, fetch_tags=True):
3451
 
        """See InterBranch.update_revisions()."""
 
3212
    def fetch(self, stop_revision=None, limit=None):
 
3213
        if self.target.base == self.source.base:
 
3214
            return (0, [])
 
3215
        self.source.lock_read()
 
3216
        try:
 
3217
            fetch_spec_factory = fetch.FetchSpecFactory()
 
3218
            fetch_spec_factory.source_branch = self.source
 
3219
            fetch_spec_factory.source_branch_stop_revision_id = stop_revision
 
3220
            fetch_spec_factory.source_repo = self.source.repository
 
3221
            fetch_spec_factory.target_repo = self.target.repository
 
3222
            fetch_spec_factory.target_repo_kind = fetch.TargetRepoKinds.PREEXISTING
 
3223
            fetch_spec_factory.limit = limit
 
3224
            fetch_spec = fetch_spec_factory.make_fetch_spec()
 
3225
            return self.target.repository.fetch(self.source.repository,
 
3226
                fetch_spec=fetch_spec)
 
3227
        finally:
 
3228
            self.source.unlock()
 
3229
 
 
3230
    @needs_write_lock
 
3231
    def _update_revisions(self, stop_revision=None, overwrite=False,
 
3232
            graph=None):
3452
3233
        other_revno, other_last_revision = self.source.last_revision_info()
3453
3234
        stop_revno = None # unknown
3454
3235
        if stop_revision is None:
3465
3246
        # case of having something to pull, and so that the check for
3466
3247
        # already merged can operate on the just fetched graph, which will
3467
3248
        # be cached in memory.
3468
 
        if fetch_tags:
3469
 
            fetch_spec_factory = fetch.FetchSpecFactory()
3470
 
            fetch_spec_factory.source_branch = self.source
3471
 
            fetch_spec_factory.source_branch_stop_revision_id = stop_revision
3472
 
            fetch_spec_factory.source_repo = self.source.repository
3473
 
            fetch_spec_factory.target_repo = self.target.repository
3474
 
            fetch_spec_factory.target_repo_kind = fetch.TargetRepoKinds.PREEXISTING
3475
 
            fetch_spec = fetch_spec_factory.make_fetch_spec()
3476
 
        else:
3477
 
            fetch_spec = _mod_graph.NotInOtherForRevs(self.target.repository,
3478
 
                self.source.repository, revision_ids=[stop_revision]).execute()
3479
 
        self.target.fetch(self.source, fetch_spec=fetch_spec)
 
3249
        self.fetch(stop_revision=stop_revision)
3480
3250
        # Check to see if one is an ancestor of the other
3481
3251
        if not overwrite:
3482
3252
            if graph is None:
3510
3280
        if local and not bound_location:
3511
3281
            raise errors.LocalRequiresBoundBranch()
3512
3282
        master_branch = None
3513
 
        source_is_master = (self.source.user_url == bound_location)
 
3283
        source_is_master = False
 
3284
        if bound_location:
 
3285
            # bound_location comes from a config file, some care has to be
 
3286
            # taken to relate it to source.user_url
 
3287
            normalized = urlutils.normalize_url(bound_location)
 
3288
            try:
 
3289
                relpath = self.source.user_transport.relpath(normalized)
 
3290
                source_is_master = (relpath == '')
 
3291
            except (errors.PathNotChild, errors.InvalidURL):
 
3292
                source_is_master = False
3514
3293
        if not local and bound_location and not source_is_master:
3515
3294
            # not pulling from master, so we need to update master.
3516
3295
            master_branch = self.target.get_master_branch(possible_transports)
3529
3308
            if master_branch:
3530
3309
                master_branch.unlock()
3531
3310
 
3532
 
    def push(self, overwrite=False, stop_revision=None,
 
3311
    def push(self, overwrite=False, stop_revision=None, lossy=False,
3533
3312
             _override_hook_source_branch=None):
3534
3313
        """See InterBranch.push.
3535
3314
 
3536
3315
        This is the basic concrete implementation of push()
3537
3316
 
3538
 
        :param _override_hook_source_branch: If specified, run
3539
 
        the hooks passing this Branch as the source, rather than self.
3540
 
        This is for use of RemoteBranch, where push is delegated to the
3541
 
        underlying vfs-based Branch.
 
3317
        :param _override_hook_source_branch: If specified, run the hooks
 
3318
            passing this Branch as the source, rather than self.  This is for
 
3319
            use of RemoteBranch, where push is delegated to the underlying
 
3320
            vfs-based Branch.
3542
3321
        """
 
3322
        if lossy:
 
3323
            raise errors.LossyPushToSameVCS(self.source, self.target)
3543
3324
        # TODO: Public option to disable running hooks - should be trivial but
3544
3325
        # needs tests.
3545
 
        self.source.lock_read()
3546
 
        try:
3547
 
            return _run_with_write_locked_target(
3548
 
                self.target, self._push_with_bound_branches, overwrite,
3549
 
                stop_revision,
3550
 
                _override_hook_source_branch=_override_hook_source_branch)
3551
 
        finally:
3552
 
            self.source.unlock()
3553
 
 
3554
 
    def _push_with_bound_branches(self, overwrite, stop_revision,
 
3326
 
 
3327
        op = cleanup.OperationWithCleanups(self._push_with_bound_branches)
 
3328
        op.add_cleanup(self.source.lock_read().unlock)
 
3329
        op.add_cleanup(self.target.lock_write().unlock)
 
3330
        return op.run(overwrite, stop_revision,
 
3331
            _override_hook_source_branch=_override_hook_source_branch)
 
3332
 
 
3333
    def _basic_push(self, overwrite, stop_revision):
 
3334
        """Basic implementation of push without bound branches or hooks.
 
3335
 
 
3336
        Must be called with source read locked and target write locked.
 
3337
        """
 
3338
        result = BranchPushResult()
 
3339
        result.source_branch = self.source
 
3340
        result.target_branch = self.target
 
3341
        result.old_revno, result.old_revid = self.target.last_revision_info()
 
3342
        self.source.update_references(self.target)
 
3343
        overwrite = _fix_overwrite_type(overwrite)
 
3344
        if result.old_revid != stop_revision:
 
3345
            # We assume that during 'push' this repository is closer than
 
3346
            # the target.
 
3347
            graph = self.source.repository.get_graph(self.target.repository)
 
3348
            self._update_revisions(stop_revision,
 
3349
                overwrite=("history" in overwrite),
 
3350
                graph=graph)
 
3351
        if self.source._push_should_merge_tags():
 
3352
            result.tag_updates, result.tag_conflicts = (
 
3353
                self.source.tags.merge_to(
 
3354
                self.target.tags, "tags" in overwrite))
 
3355
        result.new_revno, result.new_revid = self.target.last_revision_info()
 
3356
        return result
 
3357
 
 
3358
    def _push_with_bound_branches(self, operation, overwrite, stop_revision,
3555
3359
            _override_hook_source_branch=None):
3556
3360
        """Push from source into target, and into target's master if any.
3557
3361
        """
3569
3373
            # be bound to itself? -- mbp 20070507
3570
3374
            master_branch = self.target.get_master_branch()
3571
3375
            master_branch.lock_write()
3572
 
            try:
3573
 
                # push into the master from the source branch.
3574
 
                self.source._basic_push(master_branch, overwrite, stop_revision)
3575
 
                # and push into the target branch from the source. Note that we
3576
 
                # push from the source branch again, because it's considered the
3577
 
                # highest bandwidth repository.
3578
 
                result = self.source._basic_push(self.target, overwrite,
3579
 
                    stop_revision)
3580
 
                result.master_branch = master_branch
3581
 
                result.local_branch = self.target
3582
 
                _run_hooks()
3583
 
                return result
3584
 
            finally:
3585
 
                master_branch.unlock()
 
3376
            operation.add_cleanup(master_branch.unlock)
 
3377
            # push into the master from the source branch.
 
3378
            master_inter = InterBranch.get(self.source, master_branch)
 
3379
            master_inter._basic_push(overwrite, stop_revision)
 
3380
            # and push into the target branch from the source. Note that
 
3381
            # we push from the source branch again, because it's considered
 
3382
            # the highest bandwidth repository.
 
3383
            result = self._basic_push(overwrite, stop_revision)
 
3384
            result.master_branch = master_branch
 
3385
            result.local_branch = self.target
3586
3386
        else:
 
3387
            master_branch = None
3587
3388
            # no master branch
3588
 
            result = self.source._basic_push(self.target, overwrite,
3589
 
                stop_revision)
 
3389
            result = self._basic_push(overwrite, stop_revision)
3590
3390
            # TODO: Why set master_branch and local_branch if there's no
3591
3391
            # binding?  Maybe cleaner to just leave them unset? -- mbp
3592
3392
            # 20070504
3593
3393
            result.master_branch = self.target
3594
3394
            result.local_branch = None
3595
 
            _run_hooks()
3596
 
            return result
 
3395
        _run_hooks()
 
3396
        return result
3597
3397
 
3598
3398
    def _pull(self, overwrite=False, stop_revision=None,
3599
3399
             possible_transports=None, _hook_master=None, run_hooks=True,
3635
3435
            # -- JRV20090506
3636
3436
            result.old_revno, result.old_revid = \
3637
3437
                self.target.last_revision_info()
3638
 
            self.target.update_revisions(self.source, stop_revision,
3639
 
                overwrite=overwrite, graph=graph)
 
3438
            overwrite = _fix_overwrite_type(overwrite)
 
3439
            self._update_revisions(stop_revision,
 
3440
                overwrite=("history" in overwrite),
 
3441
                graph=graph)
3640
3442
            # TODO: The old revid should be specified when merging tags, 
3641
3443
            # so a tags implementation that versions tags can only 
3642
3444
            # pull in the most recent changes. -- JRV20090506
3643
 
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3644
 
                overwrite, ignore_master=not merge_tags_to_master)
 
3445
            result.tag_updates, result.tag_conflicts = (
 
3446
                self.source.tags.merge_to(self.target.tags,
 
3447
                    "tags" in overwrite,
 
3448
                    ignore_master=not merge_tags_to_master))
3645
3449
            result.new_revno, result.new_revid = self.target.last_revision_info()
3646
3450
            if _hook_master:
3647
3451
                result.master_branch = _hook_master