~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Andrew Bennetts
  • Date: 2010-09-17 04:32:50 UTC
  • mto: This revision was merged to the branch mainline in revision 5433.
  • Revision ID: andrew.bennetts@canonical.com-20100917043250-4f57ifkg0yu321t7
Reuse self._log_file rather than make new StringIO, and wrap pre-existing long line.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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
24
24
from bzrlib import (
25
25
        bzrdir,
26
26
        cache_utf8,
27
 
        cleanup,
28
27
        config as _mod_config,
 
28
        controldir,
29
29
        debug,
30
30
        errors,
31
 
        fetch,
32
 
        graph as _mod_graph,
33
31
        lockdir,
34
32
        lockable_files,
35
33
        remote,
36
34
        repository,
37
35
        revision as _mod_revision,
38
36
        rio,
 
37
        symbol_versioning,
39
38
        transport,
 
39
        tsort,
40
40
        ui,
41
41
        urlutils,
42
42
        )
43
43
from bzrlib.config import BranchConfig, TransportConfig
 
44
from bzrlib.repofmt.pack_repo import RepositoryFormatKnitPack5RichRoot
44
45
from bzrlib.tag import (
45
46
    BasicTags,
46
47
    DisabledTags,
47
48
    )
48
49
""")
49
50
 
50
 
from bzrlib import (
51
 
    controldir,
52
 
    )
53
 
from bzrlib.decorators import (
54
 
    needs_read_lock,
55
 
    needs_write_lock,
56
 
    only_raises,
57
 
    )
58
 
from bzrlib.hooks import Hooks
 
51
from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises
 
52
from bzrlib.hooks import HookPoint, Hooks
59
53
from bzrlib.inter import InterObject
60
54
from bzrlib.lock import _RelockDebugMixin, LogicalLockResult
61
55
from bzrlib import registry
77
71
    :ivar base:
78
72
        Base directory/url of the branch; using control_url and
79
73
        control_transport is more standardized.
80
 
    :ivar hooks: An instance of BranchHooks.
81
 
    :ivar _master_branch_cache: cached result of get_master_branch, see
82
 
        _clear_cached_state.
 
74
 
 
75
    hooks: An instance of BranchHooks.
83
76
    """
84
77
    # this is really an instance variable - FIXME move it there
85
78
    # - RBC 20060112
99
92
        self._revision_id_to_revno_cache = None
100
93
        self._partial_revision_id_to_revno_cache = {}
101
94
        self._partial_revision_history_cache = []
102
 
        self._tags_bytes = None
103
95
        self._last_revision_info_cache = None
104
 
        self._master_branch_cache = None
105
96
        self._merge_sorted_revisions_cache = None
106
97
        self._open_hook()
107
98
        hooks = Branch.hooks['open']
113
104
 
114
105
    def _activate_fallback_location(self, url):
115
106
        """Activate the branch/repository from url as a fallback repository."""
116
 
        for existing_fallback_repo in self.repository._fallback_repositories:
117
 
            if existing_fallback_repo.user_url == url:
118
 
                # This fallback is already configured.  This probably only
119
 
                # happens because BzrDir.sprout is a horrible mess.  To avoid
120
 
                # confusing _unstack we don't add this a second time.
121
 
                mutter('duplicate activation of fallback %r on %r', url, self)
122
 
                return
123
107
        repo = self._get_fallback_repository(url)
124
108
        if repo.has_same_location(self.repository):
125
109
            raise errors.UnstackableLocationError(self.user_url, url)
243
227
            possible_transports=[self.bzrdir.root_transport])
244
228
        return a_branch.repository
245
229
 
246
 
    @needs_read_lock
247
230
    def _get_tags_bytes(self):
248
231
        """Get the bytes of a serialised tags dict.
249
232
 
256
239
        :return: The bytes of the tags file.
257
240
        :seealso: Branch._set_tags_bytes.
258
241
        """
259
 
        if self._tags_bytes is None:
260
 
            self._tags_bytes = self._transport.get_bytes('tags')
261
 
        return self._tags_bytes
 
242
        return self._transport.get_bytes('tags')
262
243
 
263
244
    def _get_nick(self, local=False, possible_transports=None):
264
245
        config = self.get_config()
455
436
            after. If None, the rest of history is included.
456
437
        :param stop_rule: if stop_revision_id is not None, the precise rule
457
438
            to use for termination:
458
 
 
459
439
            * 'exclude' - leave the stop revision out of the result (default)
460
440
            * 'include' - the stop revision is the last item in the result
461
441
            * 'with-merges' - include the stop revision and all of its
463
443
            * 'with-merges-without-common-ancestry' - filter out revisions 
464
444
              that are in both ancestries
465
445
        :param direction: either 'reverse' or 'forward':
466
 
 
467
446
            * reverse means return the start_revision_id first, i.e.
468
447
              start at the most recent revision and go backwards in history
469
448
            * forward returns tuples in the opposite order to reverse.
513
492
        rev_iter = iter(merge_sorted_revisions)
514
493
        if start_revision_id is not None:
515
494
            for node in rev_iter:
516
 
                rev_id = node.key
 
495
                rev_id = node.key[-1]
517
496
                if rev_id != start_revision_id:
518
497
                    continue
519
498
                else:
525
504
        if stop_revision_id is None:
526
505
            # Yield everything
527
506
            for node in rev_iter:
528
 
                rev_id = node.key
 
507
                rev_id = node.key[-1]
529
508
                yield (rev_id, node.merge_depth, node.revno,
530
509
                       node.end_of_merge)
531
510
        elif stop_rule == 'exclude':
532
511
            for node in rev_iter:
533
 
                rev_id = node.key
 
512
                rev_id = node.key[-1]
534
513
                if rev_id == stop_revision_id:
535
514
                    return
536
515
                yield (rev_id, node.merge_depth, node.revno,
537
516
                       node.end_of_merge)
538
517
        elif stop_rule == 'include':
539
518
            for node in rev_iter:
540
 
                rev_id = node.key
 
519
                rev_id = node.key[-1]
541
520
                yield (rev_id, node.merge_depth, node.revno,
542
521
                       node.end_of_merge)
543
522
                if rev_id == stop_revision_id:
549
528
            ancestors = graph.find_unique_ancestors(start_revision_id,
550
529
                                                    [stop_revision_id])
551
530
            for node in rev_iter:
552
 
                rev_id = node.key
 
531
                rev_id = node.key[-1]
553
532
                if rev_id not in ancestors:
554
533
                    continue
555
534
                yield (rev_id, node.merge_depth, node.revno,
565
544
            reached_stop_revision_id = False
566
545
            revision_id_whitelist = []
567
546
            for node in rev_iter:
568
 
                rev_id = node.key
 
547
                rev_id = node.key[-1]
569
548
                if rev_id == left_parent:
570
549
                    # reached the left parent after the stop_revision
571
550
                    return
670
649
        raise errors.UnsupportedOperation(self.get_reference_info, self)
671
650
 
672
651
    @needs_write_lock
673
 
    def fetch(self, from_branch, last_revision=None, limit=None):
 
652
    def fetch(self, from_branch, last_revision=None, pb=None):
674
653
        """Copy revisions from from_branch into this branch.
675
654
 
676
655
        :param from_branch: Where to copy from.
677
656
        :param last_revision: What revision to stop at (None for at the end
678
657
                              of the branch.
679
 
        :param limit: Optional rough limit of revisions to fetch
 
658
        :param pb: An optional progress bar to use.
680
659
        :return: None
681
660
        """
682
 
        return InterBranch.get(from_branch, self).fetch(last_revision, limit=limit)
 
661
        if self.base == from_branch.base:
 
662
            return (0, [])
 
663
        if pb is not None:
 
664
            symbol_versioning.warn(
 
665
                symbol_versioning.deprecated_in((1, 14, 0))
 
666
                % "pb parameter to fetch()")
 
667
        from_branch.lock_read()
 
668
        try:
 
669
            if last_revision is None:
 
670
                last_revision = from_branch.last_revision()
 
671
                last_revision = _mod_revision.ensure_null(last_revision)
 
672
            return self.repository.fetch(from_branch.repository,
 
673
                                         revision_id=last_revision,
 
674
                                         pb=pb)
 
675
        finally:
 
676
            from_branch.unlock()
683
677
 
684
678
    def get_bound_location(self):
685
679
        """Return the URL of the branch we are bound to.
696
690
 
697
691
    def get_commit_builder(self, parents, config=None, timestamp=None,
698
692
                           timezone=None, committer=None, revprops=None,
699
 
                           revision_id=None, lossy=False):
 
693
                           revision_id=None):
700
694
        """Obtain a CommitBuilder for this branch.
701
695
 
702
696
        :param parents: Revision ids of the parents of the new revision.
706
700
        :param committer: Optional committer to set for commit.
707
701
        :param revprops: Optional dictionary of revision properties.
708
702
        :param revision_id: Optional revision id.
709
 
        :param lossy: Whether to discard data that can not be natively
710
 
            represented, when pushing to a foreign VCS 
711
703
        """
712
704
 
713
705
        if config is None:
714
706
            config = self.get_config()
715
707
 
716
708
        return self.repository.get_commit_builder(self, parents, config,
717
 
            timestamp, timezone, committer, revprops, revision_id,
718
 
            lossy)
 
709
            timestamp, timezone, committer, revprops, revision_id)
719
710
 
720
711
    def get_master_branch(self, possible_transports=None):
721
712
        """Return the branch we are bound to.
748
739
        """Print `file` to stdout."""
749
740
        raise NotImplementedError(self.print_file)
750
741
 
751
 
    @deprecated_method(deprecated_in((2, 4, 0)))
752
742
    def set_revision_history(self, rev_history):
753
 
        """See Branch.set_revision_history."""
754
 
        self._set_revision_history(rev_history)
755
 
 
756
 
    @needs_write_lock
757
 
    def _set_revision_history(self, rev_history):
758
 
        if len(rev_history) == 0:
759
 
            revid = _mod_revision.NULL_REVISION
760
 
        else:
761
 
            revid = rev_history[-1]
762
 
        if rev_history != self._lefthand_history(revid):
763
 
            raise errors.NotLefthandHistory(rev_history)
764
 
        self.set_last_revision_info(len(rev_history), revid)
765
 
        self._cache_revision_history(rev_history)
766
 
        for hook in Branch.hooks['set_rh']:
767
 
            hook(self, rev_history)
768
 
 
769
 
    @needs_write_lock
770
 
    def set_last_revision_info(self, revno, revision_id):
771
 
        """Set the last revision of this branch.
772
 
 
773
 
        The caller is responsible for checking that the revno is correct
774
 
        for this revision id.
775
 
 
776
 
        It may be possible to set the branch last revision to an id not
777
 
        present in the repository.  However, branches can also be
778
 
        configured to check constraints on history, in which case this may not
779
 
        be permitted.
780
 
        """
781
 
        raise NotImplementedError(self.set_last_revision_info)
782
 
 
783
 
    @needs_write_lock
784
 
    def generate_revision_history(self, revision_id, last_rev=None,
785
 
                                  other_branch=None):
786
 
        """See Branch.generate_revision_history"""
787
 
        graph = self.repository.get_graph()
788
 
        known_revision_ids = [
789
 
            self.last_revision_info(),
790
 
            (_mod_revision.NULL_REVISION, 0),
791
 
            ]
792
 
        if last_rev is not None:
793
 
            if not graph.is_ancestor(last_rev, revision_id):
794
 
                # our previous tip is not merged into stop_revision
795
 
                raise errors.DivergedBranches(self, other_branch)
796
 
        revno = graph.find_distance_to_null(revision_id, known_revision_ids)
797
 
        self.set_last_revision_info(revno, revision_id)
 
743
        raise NotImplementedError(self.set_revision_history)
798
744
 
799
745
    @needs_write_lock
800
746
    def set_parent(self, url):
844
790
 
845
791
    def _unstack(self):
846
792
        """Change a branch to be unstacked, copying data as needed.
847
 
 
 
793
        
848
794
        Don't call this directly, use set_stacked_on_url(None).
849
795
        """
850
796
        pb = ui.ui_factory.nested_progress_bar()
859
805
            old_repository = self.repository
860
806
            if len(old_repository._fallback_repositories) != 1:
861
807
                raise AssertionError("can't cope with fallback repositories "
862
 
                    "of %r (fallbacks: %r)" % (old_repository,
863
 
                        old_repository._fallback_repositories))
 
808
                    "of %r" % (self.repository,))
864
809
            # Open the new repository object.
865
810
            # Repositories don't offer an interface to remove fallback
866
811
            # repositories today; take the conceptually simpler option and just
914
859
                # XXX: If you unstack a branch while it has a working tree
915
860
                # with a pending merge, the pending-merged revisions will no
916
861
                # longer be present.  You can (probably) revert and remerge.
917
 
                try:
918
 
                    tags_to_fetch = set(self.tags.get_reverse_tag_dict())
919
 
                except errors.TagsNotSupported:
920
 
                    tags_to_fetch = set()
921
 
                fetch_spec = _mod_graph.NotInOtherForRevs(self.repository,
922
 
                    old_repository, required_ids=[self.last_revision()],
923
 
                    if_present_ids=tags_to_fetch, find_ghosts=True).execute()
924
 
                self.repository.fetch(old_repository, fetch_spec=fetch_spec)
 
862
                #
 
863
                # XXX: This only fetches up to the tip of the repository; it
 
864
                # doesn't bring across any tags.  That's fairly consistent
 
865
                # with how branch works, but perhaps not ideal.
 
866
                self.repository.fetch(old_repository,
 
867
                    revision_id=self.last_revision(),
 
868
                    find_ghosts=True)
925
869
            finally:
926
870
                old_repository.unlock()
927
871
        finally:
932
876
 
933
877
        :seealso: Branch._get_tags_bytes.
934
878
        """
935
 
        op = cleanup.OperationWithCleanups(self._set_tags_bytes_locked)
936
 
        op.add_cleanup(self.lock_write().unlock)
937
 
        return op.run_simple(bytes)
938
 
 
939
 
    def _set_tags_bytes_locked(self, bytes):
940
 
        self._tags_bytes = bytes
941
 
        return self._transport.put_bytes('tags', bytes)
 
879
        return _run_with_write_locked_target(self, self._transport.put_bytes,
 
880
            'tags', bytes)
942
881
 
943
882
    def _cache_revision_history(self, rev_history):
944
883
        """Set the cached revision history to rev_history.
971
910
        self._revision_history_cache = None
972
911
        self._revision_id_to_revno_cache = None
973
912
        self._last_revision_info_cache = None
974
 
        self._master_branch_cache = None
975
913
        self._merge_sorted_revisions_cache = None
976
914
        self._partial_revision_history_cache = []
977
915
        self._partial_revision_id_to_revno_cache = {}
978
 
        self._tags_bytes = None
979
916
 
980
917
    def _gen_revision_history(self):
981
918
        """Return sequence of revision hashes on to this branch.
1031
968
        :return: A tuple (revno, revision_id).
1032
969
        """
1033
970
        if self._last_revision_info_cache is None:
1034
 
            self._last_revision_info_cache = self._read_last_revision_info()
 
971
            self._last_revision_info_cache = self._last_revision_info()
1035
972
        return self._last_revision_info_cache
1036
973
 
1037
 
    def _read_last_revision_info(self):
1038
 
        raise NotImplementedError(self._read_last_revision_info)
1039
 
 
1040
 
    @deprecated_method(deprecated_in((2, 4, 0)))
 
974
    def _last_revision_info(self):
 
975
        rh = self.revision_history()
 
976
        revno = len(rh)
 
977
        if revno:
 
978
            return (revno, rh[-1])
 
979
        else:
 
980
            return (0, _mod_revision.NULL_REVISION)
 
981
 
 
982
    @deprecated_method(deprecated_in((1, 6, 0)))
 
983
    def missing_revisions(self, other, stop_revision=None):
 
984
        """Return a list of new revisions that would perfectly fit.
 
985
 
 
986
        If self and other have not diverged, return a list of the revisions
 
987
        present in other, but missing from self.
 
988
        """
 
989
        self_history = self.revision_history()
 
990
        self_len = len(self_history)
 
991
        other_history = other.revision_history()
 
992
        other_len = len(other_history)
 
993
        common_index = min(self_len, other_len) -1
 
994
        if common_index >= 0 and \
 
995
            self_history[common_index] != other_history[common_index]:
 
996
            raise errors.DivergedBranches(self, other)
 
997
 
 
998
        if stop_revision is None:
 
999
            stop_revision = other_len
 
1000
        else:
 
1001
            if stop_revision > other_len:
 
1002
                raise errors.NoSuchRevision(self, stop_revision)
 
1003
        return other_history[self_len:stop_revision]
 
1004
 
 
1005
    def update_revisions(self, other, stop_revision=None, overwrite=False,
 
1006
                         graph=None):
 
1007
        """Pull in new perfect-fit revisions.
 
1008
 
 
1009
        :param other: Another Branch to pull from
 
1010
        :param stop_revision: Updated until the given revision
 
1011
        :param overwrite: Always set the branch pointer, rather than checking
 
1012
            to see if it is a proper descendant.
 
1013
        :param graph: A Graph object that can be used to query history
 
1014
            information. This can be None.
 
1015
        :return: None
 
1016
        """
 
1017
        return InterBranch.get(other, self).update_revisions(stop_revision,
 
1018
            overwrite, graph)
 
1019
 
1041
1020
    def import_last_revision_info(self, source_repo, revno, revid):
1042
1021
        """Set the last revision info, importing from another repo if necessary.
1043
1022
 
 
1023
        This is used by the bound branch code to upload a revision to
 
1024
        the master branch first before updating the tip of the local branch.
 
1025
 
1044
1026
        :param source_repo: Source repository to optionally fetch from
1045
1027
        :param revno: Revision number of the new tip
1046
1028
        :param revid: Revision id of the new tip
1049
1031
            self.repository.fetch(source_repo, revision_id=revid)
1050
1032
        self.set_last_revision_info(revno, revid)
1051
1033
 
1052
 
    def import_last_revision_info_and_tags(self, source, revno, revid,
1053
 
                                           lossy=False):
1054
 
        """Set the last revision info, importing from another repo if necessary.
1055
 
 
1056
 
        This is used by the bound branch code to upload a revision to
1057
 
        the master branch first before updating the tip of the local branch.
1058
 
        Revisions referenced by source's tags are also transferred.
1059
 
 
1060
 
        :param source: Source branch to optionally fetch from
1061
 
        :param revno: Revision number of the new tip
1062
 
        :param revid: Revision id of the new tip
1063
 
        :param lossy: Whether to discard metadata that can not be
1064
 
            natively represented
1065
 
        :return: Tuple with the new revision number and revision id
1066
 
            (should only be different from the arguments when lossy=True)
1067
 
        """
1068
 
        if not self.repository.has_same_location(source.repository):
1069
 
            self.fetch(source, revid)
1070
 
        self.set_last_revision_info(revno, revid)
1071
 
        return (revno, revid)
1072
 
 
1073
1034
    def revision_id_to_revno(self, revision_id):
1074
1035
        """Given a revision id, return its revno"""
1075
1036
        if _mod_revision.is_null(revision_id):
1107
1068
            stop_revision=stop_revision,
1108
1069
            possible_transports=possible_transports, *args, **kwargs)
1109
1070
 
1110
 
    def push(self, target, overwrite=False, stop_revision=None, lossy=False,
1111
 
            *args, **kwargs):
 
1071
    def push(self, target, overwrite=False, stop_revision=None, *args,
 
1072
        **kwargs):
1112
1073
        """Mirror this branch into target.
1113
1074
 
1114
1075
        This branch is considered to be 'local', having low latency.
1115
1076
        """
1116
1077
        return InterBranch.get(self, target).push(overwrite, stop_revision,
1117
 
            lossy, *args, **kwargs)
 
1078
            *args, **kwargs)
 
1079
 
 
1080
    def lossy_push(self, target, stop_revision=None):
 
1081
        """Push deltas into another branch.
 
1082
 
 
1083
        :note: This does not, like push, retain the revision ids from 
 
1084
            the source branch and will, rather than adding bzr-specific 
 
1085
            metadata, push only those semantics of the revision that can be 
 
1086
            natively represented by this branch' VCS.
 
1087
 
 
1088
        :param target: Target branch
 
1089
        :param stop_revision: Revision to push, defaults to last revision.
 
1090
        :return: BranchPushResult with an extra member revidmap: 
 
1091
            A dictionary mapping revision ids from the target branch 
 
1092
            to new revision ids in the target branch, for each 
 
1093
            revision that was pushed.
 
1094
        """
 
1095
        inter = InterBranch.get(self, target)
 
1096
        lossy_push = getattr(inter, "lossy_push", None)
 
1097
        if lossy_push is None:
 
1098
            raise errors.LossyPushToSameVCS(self, target)
 
1099
        return lossy_push(stop_revision)
1118
1100
 
1119
1101
    def basis_tree(self):
1120
1102
        """Return `Tree` object for last revision."""
1275
1257
        return result
1276
1258
 
1277
1259
    @needs_read_lock
1278
 
    def sprout(self, to_bzrdir, revision_id=None, repository_policy=None,
1279
 
            repository=None):
 
1260
    def sprout(self, to_bzrdir, revision_id=None, repository_policy=None):
1280
1261
        """Create a new line of development from the branch, into to_bzrdir.
1281
1262
 
1282
1263
        to_bzrdir controls the branch format.
1287
1268
        if (repository_policy is not None and
1288
1269
            repository_policy.requires_stacking()):
1289
1270
            to_bzrdir._format.require_stacking(_skip_repo=True)
1290
 
        result = to_bzrdir.create_branch(repository=repository)
 
1271
        result = to_bzrdir.create_branch()
1291
1272
        result.lock_write()
1292
1273
        try:
1293
1274
            if repository_policy is not None:
1294
1275
                repository_policy.configure_branch(result)
1295
1276
            self.copy_content_into(result, revision_id=revision_id)
1296
 
            master_branch = self.get_master_branch()
1297
 
            if master_branch is None:
1298
 
                result.set_parent(self.bzrdir.root_transport.base)
1299
 
            else:
1300
 
                result.set_parent(master_branch.bzrdir.root_transport.base)
 
1277
            result.set_parent(self.bzrdir.root_transport.base)
1301
1278
        finally:
1302
1279
            result.unlock()
1303
1280
        return result
1385
1362
        """Return the most suitable metadir for a checkout of this branch.
1386
1363
        Weaves are used if this branch's repository uses weaves.
1387
1364
        """
1388
 
        format = self.repository.bzrdir.checkout_metadir()
1389
 
        format.set_branch_format(self._format)
 
1365
        if isinstance(self.bzrdir, bzrdir.BzrDirPreSplitOut):
 
1366
            from bzrlib.repofmt import weaverepo
 
1367
            format = bzrdir.BzrDirMetaFormat1()
 
1368
            format.repository_format = weaverepo.RepositoryFormat7()
 
1369
        else:
 
1370
            format = self.repository.bzrdir.checkout_metadir()
 
1371
            format.set_branch_format(self._format)
1390
1372
        return format
1391
1373
 
1392
1374
    def create_clone_on_transport(self, to_transport, revision_id=None,
1393
 
        stacked_on=None, create_prefix=False, use_existing_dir=False,
1394
 
        no_tree=None):
 
1375
        stacked_on=None, create_prefix=False, use_existing_dir=False):
1395
1376
        """Create a clone of this branch and its bzrdir.
1396
1377
 
1397
1378
        :param to_transport: The transport to clone onto.
1410
1391
            revision_id = self.last_revision()
1411
1392
        dir_to = self.bzrdir.clone_on_transport(to_transport,
1412
1393
            revision_id=revision_id, stacked_on=stacked_on,
1413
 
            create_prefix=create_prefix, use_existing_dir=use_existing_dir,
1414
 
            no_tree=no_tree)
 
1394
            create_prefix=create_prefix, use_existing_dir=use_existing_dir)
1415
1395
        return dir_to.open_branch()
1416
1396
 
1417
1397
    def create_checkout(self, to_location, revision_id=None,
1422
1402
        :param to_location: The url to produce the checkout at
1423
1403
        :param revision_id: The revision to check out
1424
1404
        :param lightweight: If True, produce a lightweight checkout, otherwise,
1425
 
            produce a bound branch (heavyweight checkout)
 
1405
        produce a bound branch (heavyweight checkout)
1426
1406
        :param accelerator_tree: A tree which can be used for retrieving file
1427
1407
            contents more quickly than the revision tree, i.e. a workingtree.
1428
1408
            The revision tree will be used for cases where accelerator_tree's
1474
1454
 
1475
1455
    def reference_parent(self, file_id, path, possible_transports=None):
1476
1456
        """Return the parent branch for a tree-reference file_id
1477
 
 
1478
1457
        :param file_id: The file_id of the tree reference
1479
1458
        :param path: The path of the file_id in the tree
1480
1459
        :return: A branch associated with the file_id
1533
1512
        else:
1534
1513
            raise AssertionError("invalid heads: %r" % (heads,))
1535
1514
 
1536
 
    def heads_to_fetch(self):
1537
 
        """Return the heads that must and that should be fetched to copy this
1538
 
        branch into another repo.
1539
 
 
1540
 
        :returns: a 2-tuple of (must_fetch, if_present_fetch).  must_fetch is a
1541
 
            set of heads that must be fetched.  if_present_fetch is a set of
1542
 
            heads that must be fetched if present, but no error is necessary if
1543
 
            they are not present.
1544
 
        """
1545
 
        # For bzr native formats must_fetch is just the tip, and if_present_fetch
1546
 
        # are the tags.
1547
 
        must_fetch = set([self.last_revision()])
1548
 
        try:
1549
 
            if_present_fetch = set(self.tags.get_reverse_tag_dict())
1550
 
        except errors.TagsNotSupported:
1551
 
            if_present_fetch = set()
1552
 
        must_fetch.discard(_mod_revision.NULL_REVISION)
1553
 
        if_present_fetch.discard(_mod_revision.NULL_REVISION)
1554
 
        return must_fetch, if_present_fetch
1555
 
 
1556
 
 
1557
 
class BranchFormat(controldir.ControlComponentFormat):
 
1515
 
 
1516
class BranchFormat(object):
1558
1517
    """An encapsulation of the initialization and open routines for a format.
1559
1518
 
1560
1519
    Formats provide three things:
1563
1522
     * an open routine.
1564
1523
 
1565
1524
    Formats are placed in an dict by their format string for reference
1566
 
    during branch opening. It's not required that these be instances, they
 
1525
    during branch opening. Its not required that these be instances, they
1567
1526
    can be classes themselves with class methods - it simply depends on
1568
1527
    whether state is needed for a given format or not.
1569
1528
 
1572
1531
    object will be created every time regardless.
1573
1532
    """
1574
1533
 
 
1534
    _default_format = None
 
1535
    """The default format used for new branches."""
 
1536
 
 
1537
    _formats = {}
 
1538
    """The known formats."""
 
1539
 
1575
1540
    can_set_append_revisions_only = True
1576
1541
 
1577
1542
    def __eq__(self, other):
1586
1551
        try:
1587
1552
            transport = a_bzrdir.get_branch_transport(None, name=name)
1588
1553
            format_string = transport.get_bytes("format")
1589
 
            return format_registry.get(format_string)
 
1554
            format = klass._formats[format_string]
 
1555
            if isinstance(format, MetaDirBranchFormatFactory):
 
1556
                return format()
 
1557
            return format
1590
1558
        except errors.NoSuchFile:
1591
1559
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
1592
1560
        except KeyError:
1593
1561
            raise errors.UnknownFormatError(format=format_string, kind='branch')
1594
1562
 
1595
1563
    @classmethod
1596
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1597
1564
    def get_default_format(klass):
1598
1565
        """Return the current default format."""
1599
 
        return format_registry.get_default()
 
1566
        return klass._default_format
1600
1567
 
1601
1568
    @classmethod
1602
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1603
1569
    def get_formats(klass):
1604
1570
        """Get all the known formats.
1605
1571
 
1606
1572
        Warning: This triggers a load of all lazy registered formats: do not
1607
1573
        use except when that is desireed.
1608
1574
        """
1609
 
        return format_registry._get_all()
 
1575
        result = []
 
1576
        for fmt in klass._formats.values():
 
1577
            if isinstance(fmt, MetaDirBranchFormatFactory):
 
1578
                fmt = fmt()
 
1579
            result.append(fmt)
 
1580
        return result
1610
1581
 
1611
1582
    def get_reference(self, a_bzrdir, name=None):
1612
1583
        """Get the target reference of the branch in a_bzrdir.
1651
1622
        for hook in hooks:
1652
1623
            hook(params)
1653
1624
 
1654
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
1625
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
 
1626
                           lock_type='metadir', set_format=True):
 
1627
        """Initialize a branch in a bzrdir, with specified files
 
1628
 
 
1629
        :param a_bzrdir: The bzrdir to initialize the branch in
 
1630
        :param utf8_files: The files to create as a list of
 
1631
            (filename, content) tuples
 
1632
        :param name: Name of colocated branch to create, if any
 
1633
        :param set_format: If True, set the format with
 
1634
            self.get_format_string.  (BzrBranch4 has its format set
 
1635
            elsewhere)
 
1636
        :return: a branch in this format
 
1637
        """
 
1638
        mutter('creating branch %r in %s', self, a_bzrdir.user_url)
 
1639
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
 
1640
        lock_map = {
 
1641
            'metadir': ('lock', lockdir.LockDir),
 
1642
            'branch4': ('branch-lock', lockable_files.TransportLock),
 
1643
        }
 
1644
        lock_name, lock_class = lock_map[lock_type]
 
1645
        control_files = lockable_files.LockableFiles(branch_transport,
 
1646
            lock_name, lock_class)
 
1647
        control_files.create_lock()
 
1648
        try:
 
1649
            control_files.lock_write()
 
1650
        except errors.LockContention:
 
1651
            if lock_type != 'branch4':
 
1652
                raise
 
1653
            lock_taken = False
 
1654
        else:
 
1655
            lock_taken = True
 
1656
        if set_format:
 
1657
            utf8_files += [('format', self.get_format_string())]
 
1658
        try:
 
1659
            for (filename, content) in utf8_files:
 
1660
                branch_transport.put_bytes(
 
1661
                    filename, content,
 
1662
                    mode=a_bzrdir._get_file_mode())
 
1663
        finally:
 
1664
            if lock_taken:
 
1665
                control_files.unlock()
 
1666
        branch = self.open(a_bzrdir, name, _found=True)
 
1667
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
 
1668
        return branch
 
1669
 
 
1670
    def initialize(self, a_bzrdir, name=None):
1655
1671
        """Create a branch of this format in a_bzrdir.
1656
1672
        
1657
1673
        :param name: Name of the colocated branch to create.
1691
1707
        """
1692
1708
        raise NotImplementedError(self.network_name)
1693
1709
 
1694
 
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
1695
 
            found_repository=None):
 
1710
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
1696
1711
        """Return the branch object for a_bzrdir
1697
1712
 
1698
1713
        :param a_bzrdir: A BzrDir that contains a branch.
1705
1720
        raise NotImplementedError(self.open)
1706
1721
 
1707
1722
    @classmethod
1708
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1709
1723
    def register_format(klass, format):
1710
1724
        """Register a metadir format.
1711
 
 
 
1725
        
1712
1726
        See MetaDirBranchFormatFactory for the ability to register a format
1713
1727
        without loading the code the format needs until it is actually used.
1714
1728
        """
1715
 
        format_registry.register(format)
 
1729
        klass._formats[format.get_format_string()] = format
 
1730
        # Metadir formats have a network name of their format string, and get
 
1731
        # registered as factories.
 
1732
        if isinstance(format, MetaDirBranchFormatFactory):
 
1733
            network_format_registry.register(format.get_format_string(), format)
 
1734
        else:
 
1735
            network_format_registry.register(format.get_format_string(),
 
1736
                format.__class__)
1716
1737
 
1717
1738
    @classmethod
1718
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1719
1739
    def set_default_format(klass, format):
1720
 
        format_registry.set_default(format)
 
1740
        klass._default_format = format
1721
1741
 
1722
1742
    def supports_set_append_revisions_only(self):
1723
1743
        """True if this format supports set_append_revisions_only."""
1727
1747
        """True if this format records a stacked-on branch."""
1728
1748
        return False
1729
1749
 
1730
 
    def supports_leaving_lock(self):
1731
 
        """True if this format supports leaving locks in place."""
1732
 
        return False # by default
1733
 
 
1734
1750
    @classmethod
1735
 
    @deprecated_method(deprecated_in((2, 4, 0)))
1736
1751
    def unregister_format(klass, format):
1737
 
        format_registry.remove(format)
 
1752
        del klass._formats[format.get_format_string()]
1738
1753
 
1739
1754
    def __str__(self):
1740
1755
        return self.get_format_description().rstrip()
1785
1800
        These are all empty initially, because by default nothing should get
1786
1801
        notified.
1787
1802
        """
1788
 
        Hooks.__init__(self, "bzrlib.branch", "Branch.hooks")
1789
 
        self.add_hook('set_rh',
 
1803
        Hooks.__init__(self)
 
1804
        self.create_hook(HookPoint('set_rh',
1790
1805
            "Invoked whenever the revision history has been set via "
1791
1806
            "set_revision_history. The api signature is (branch, "
1792
1807
            "revision_history), and the branch will be write-locked. "
1793
1808
            "The set_rh hook can be expensive for bzr to trigger, a better "
1794
 
            "hook to use is Branch.post_change_branch_tip.", (0, 15))
1795
 
        self.add_hook('open',
 
1809
            "hook to use is Branch.post_change_branch_tip.", (0, 15), None))
 
1810
        self.create_hook(HookPoint('open',
1796
1811
            "Called with the Branch object that has been opened after a "
1797
 
            "branch is opened.", (1, 8))
1798
 
        self.add_hook('post_push',
 
1812
            "branch is opened.", (1, 8), None))
 
1813
        self.create_hook(HookPoint('post_push',
1799
1814
            "Called after a push operation completes. post_push is called "
1800
1815
            "with a bzrlib.branch.BranchPushResult object and only runs in the "
1801
 
            "bzr client.", (0, 15))
1802
 
        self.add_hook('post_pull',
 
1816
            "bzr client.", (0, 15), None))
 
1817
        self.create_hook(HookPoint('post_pull',
1803
1818
            "Called after a pull operation completes. post_pull is called "
1804
1819
            "with a bzrlib.branch.PullResult object and only runs in the "
1805
 
            "bzr client.", (0, 15))
1806
 
        self.add_hook('pre_commit',
1807
 
            "Called after a commit is calculated but before it is "
 
1820
            "bzr client.", (0, 15), None))
 
1821
        self.create_hook(HookPoint('pre_commit',
 
1822
            "Called after a commit is calculated but before it is is "
1808
1823
            "completed. pre_commit is called with (local, master, old_revno, "
1809
1824
            "old_revid, future_revno, future_revid, tree_delta, future_tree"
1810
1825
            "). old_revid is NULL_REVISION for the first commit to a branch, "
1812
1827
            "basis revision. hooks MUST NOT modify this delta. "
1813
1828
            " future_tree is an in-memory tree obtained from "
1814
1829
            "CommitBuilder.revision_tree() and hooks MUST NOT modify this "
1815
 
            "tree.", (0,91))
1816
 
        self.add_hook('post_commit',
 
1830
            "tree.", (0,91), None))
 
1831
        self.create_hook(HookPoint('post_commit',
1817
1832
            "Called in the bzr client after a commit has completed. "
1818
1833
            "post_commit is called with (local, master, old_revno, old_revid, "
1819
1834
            "new_revno, new_revid). old_revid is NULL_REVISION for the first "
1820
 
            "commit to a branch.", (0, 15))
1821
 
        self.add_hook('post_uncommit',
 
1835
            "commit to a branch.", (0, 15), None))
 
1836
        self.create_hook(HookPoint('post_uncommit',
1822
1837
            "Called in the bzr client after an uncommit completes. "
1823
1838
            "post_uncommit is called with (local, master, old_revno, "
1824
1839
            "old_revid, new_revno, new_revid) where local is the local branch "
1825
1840
            "or None, master is the target branch, and an empty branch "
1826
 
            "receives new_revno of 0, new_revid of None.", (0, 15))
1827
 
        self.add_hook('pre_change_branch_tip',
 
1841
            "receives new_revno of 0, new_revid of None.", (0, 15), None))
 
1842
        self.create_hook(HookPoint('pre_change_branch_tip',
1828
1843
            "Called in bzr client and server before a change to the tip of a "
1829
1844
            "branch is made. pre_change_branch_tip is called with a "
1830
1845
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1831
 
            "commit, uncommit will all trigger this hook.", (1, 6))
1832
 
        self.add_hook('post_change_branch_tip',
 
1846
            "commit, uncommit will all trigger this hook.", (1, 6), None))
 
1847
        self.create_hook(HookPoint('post_change_branch_tip',
1833
1848
            "Called in bzr client and server after a change to the tip of a "
1834
1849
            "branch is made. post_change_branch_tip is called with a "
1835
1850
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1836
 
            "commit, uncommit will all trigger this hook.", (1, 4))
1837
 
        self.add_hook('transform_fallback_location',
 
1851
            "commit, uncommit will all trigger this hook.", (1, 4), None))
 
1852
        self.create_hook(HookPoint('transform_fallback_location',
1838
1853
            "Called when a stacked branch is activating its fallback "
1839
1854
            "locations. transform_fallback_location is called with (branch, "
1840
1855
            "url), and should return a new url. Returning the same url "
1845
1860
            "fallback locations have not been activated. When there are "
1846
1861
            "multiple hooks installed for transform_fallback_location, "
1847
1862
            "all are called with the url returned from the previous hook."
1848
 
            "The order is however undefined.", (1, 9))
1849
 
        self.add_hook('automatic_tag_name',
 
1863
            "The order is however undefined.", (1, 9), None))
 
1864
        self.create_hook(HookPoint('automatic_tag_name',
1850
1865
            "Called to determine an automatic tag name for a revision. "
1851
1866
            "automatic_tag_name is called with (branch, revision_id) and "
1852
1867
            "should return a tag name or None if no tag name could be "
1853
1868
            "determined. The first non-None tag name returned will be used.",
1854
 
            (2, 2))
1855
 
        self.add_hook('post_branch_init',
 
1869
            (2, 2), None))
 
1870
        self.create_hook(HookPoint('post_branch_init',
1856
1871
            "Called after new branch initialization completes. "
1857
1872
            "post_branch_init is called with a "
1858
1873
            "bzrlib.branch.BranchInitHookParams. "
1859
1874
            "Note that init, branch and checkout (both heavyweight and "
1860
 
            "lightweight) will all trigger this hook.", (2, 2))
1861
 
        self.add_hook('post_switch',
 
1875
            "lightweight) will all trigger this hook.", (2, 2), None))
 
1876
        self.create_hook(HookPoint('post_switch',
1862
1877
            "Called after a checkout switches branch. "
1863
1878
            "post_switch is called with a "
1864
 
            "bzrlib.branch.SwitchHookParams.", (2, 2))
 
1879
            "bzrlib.branch.SwitchHookParams.", (2, 2), None))
1865
1880
 
1866
1881
 
1867
1882
 
1870
1885
 
1871
1886
 
1872
1887
class ChangeBranchTipParams(object):
1873
 
    """Object holding parameters passed to `*_change_branch_tip` hooks.
 
1888
    """Object holding parameters passed to *_change_branch_tip hooks.
1874
1889
 
1875
1890
    There are 5 fields that hooks may wish to access:
1876
1891
 
1908
1923
 
1909
1924
 
1910
1925
class BranchInitHookParams(object):
1911
 
    """Object holding parameters passed to `*_branch_init` hooks.
 
1926
    """Object holding parameters passed to *_branch_init hooks.
1912
1927
 
1913
1928
    There are 4 fields that hooks may wish to access:
1914
1929
 
1948
1963
 
1949
1964
 
1950
1965
class SwitchHookParams(object):
1951
 
    """Object holding parameters passed to `*_switch` hooks.
 
1966
    """Object holding parameters passed to *_switch hooks.
1952
1967
 
1953
1968
    There are 4 fields that hooks may wish to access:
1954
1969
 
1980
1995
            self.revision_id)
1981
1996
 
1982
1997
 
 
1998
class BzrBranchFormat4(BranchFormat):
 
1999
    """Bzr branch format 4.
 
2000
 
 
2001
    This format has:
 
2002
     - a revision-history file.
 
2003
     - a branch-lock lock file [ to be shared with the bzrdir ]
 
2004
    """
 
2005
 
 
2006
    def get_format_description(self):
 
2007
        """See BranchFormat.get_format_description()."""
 
2008
        return "Branch format 4"
 
2009
 
 
2010
    def initialize(self, a_bzrdir, name=None):
 
2011
        """Create a branch of this format in a_bzrdir."""
 
2012
        utf8_files = [('revision-history', ''),
 
2013
                      ('branch-name', ''),
 
2014
                      ]
 
2015
        return self._initialize_helper(a_bzrdir, utf8_files, name=name,
 
2016
                                       lock_type='branch4', set_format=False)
 
2017
 
 
2018
    def __init__(self):
 
2019
        super(BzrBranchFormat4, self).__init__()
 
2020
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
 
2021
 
 
2022
    def network_name(self):
 
2023
        """The network name for this format is the control dirs disk label."""
 
2024
        return self._matchingbzrdir.get_format_string()
 
2025
 
 
2026
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
 
2027
        """See BranchFormat.open()."""
 
2028
        if not _found:
 
2029
            # we are being called directly and must probe.
 
2030
            raise NotImplementedError
 
2031
        return BzrBranch(_format=self,
 
2032
                         _control_files=a_bzrdir._control_files,
 
2033
                         a_bzrdir=a_bzrdir,
 
2034
                         name=name,
 
2035
                         _repository=a_bzrdir.open_repository())
 
2036
 
 
2037
    def __str__(self):
 
2038
        return "Bazaar-NG branch format 4"
 
2039
 
 
2040
 
1983
2041
class BranchFormatMetadir(BranchFormat):
1984
2042
    """Common logic for meta-dir based branch formats."""
1985
2043
 
1987
2045
        """What class to instantiate on open calls."""
1988
2046
        raise NotImplementedError(self._branch_class)
1989
2047
 
1990
 
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
1991
 
                           repository=None):
1992
 
        """Initialize a branch in a bzrdir, with specified files
1993
 
 
1994
 
        :param a_bzrdir: The bzrdir to initialize the branch in
1995
 
        :param utf8_files: The files to create as a list of
1996
 
            (filename, content) tuples
1997
 
        :param name: Name of colocated branch to create, if any
1998
 
        :return: a branch in this format
1999
 
        """
2000
 
        mutter('creating branch %r in %s', self, a_bzrdir.user_url)
2001
 
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
2002
 
        control_files = lockable_files.LockableFiles(branch_transport,
2003
 
            'lock', lockdir.LockDir)
2004
 
        control_files.create_lock()
2005
 
        control_files.lock_write()
2006
 
        try:
2007
 
            utf8_files += [('format', self.get_format_string())]
2008
 
            for (filename, content) in utf8_files:
2009
 
                branch_transport.put_bytes(
2010
 
                    filename, content,
2011
 
                    mode=a_bzrdir._get_file_mode())
2012
 
        finally:
2013
 
            control_files.unlock()
2014
 
        branch = self.open(a_bzrdir, name, _found=True,
2015
 
                found_repository=repository)
2016
 
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
2017
 
        return branch
2018
 
 
2019
2048
    def network_name(self):
2020
2049
        """A simple byte string uniquely identifying this format for RPC calls.
2021
2050
 
2023
2052
        """
2024
2053
        return self.get_format_string()
2025
2054
 
2026
 
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
2027
 
            found_repository=None):
 
2055
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
2028
2056
        """See BranchFormat.open()."""
2029
2057
        if not _found:
2030
2058
            format = BranchFormat.find_format(a_bzrdir, name=name)
2035
2063
        try:
2036
2064
            control_files = lockable_files.LockableFiles(transport, 'lock',
2037
2065
                                                         lockdir.LockDir)
2038
 
            if found_repository is None:
2039
 
                found_repository = a_bzrdir.find_repository()
2040
2066
            return self._branch_class()(_format=self,
2041
2067
                              _control_files=control_files,
2042
2068
                              name=name,
2043
2069
                              a_bzrdir=a_bzrdir,
2044
 
                              _repository=found_repository,
 
2070
                              _repository=a_bzrdir.find_repository(),
2045
2071
                              ignore_fallbacks=ignore_fallbacks)
2046
2072
        except errors.NoSuchFile:
2047
2073
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
2054
2080
    def supports_tags(self):
2055
2081
        return True
2056
2082
 
2057
 
    def supports_leaving_lock(self):
2058
 
        return True
2059
 
 
2060
2083
 
2061
2084
class BzrBranchFormat5(BranchFormatMetadir):
2062
2085
    """Bzr branch format 5.
2082
2105
        """See BranchFormat.get_format_description()."""
2083
2106
        return "Branch format 5"
2084
2107
 
2085
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2108
    def initialize(self, a_bzrdir, name=None):
2086
2109
        """Create a branch of this format in a_bzrdir."""
2087
2110
        utf8_files = [('revision-history', ''),
2088
2111
                      ('branch-name', ''),
2089
2112
                      ]
2090
 
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
 
2113
        return self._initialize_helper(a_bzrdir, utf8_files, name)
2091
2114
 
2092
2115
    def supports_tags(self):
2093
2116
        return False
2115
2138
        """See BranchFormat.get_format_description()."""
2116
2139
        return "Branch format 6"
2117
2140
 
2118
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2141
    def initialize(self, a_bzrdir, name=None):
2119
2142
        """Create a branch of this format in a_bzrdir."""
2120
2143
        utf8_files = [('last-revision', '0 null:\n'),
2121
2144
                      ('branch.conf', ''),
2122
2145
                      ('tags', ''),
2123
2146
                      ]
2124
 
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
 
2147
        return self._initialize_helper(a_bzrdir, utf8_files, name)
2125
2148
 
2126
2149
    def make_tags(self, branch):
2127
2150
        """See bzrlib.branch.BranchFormat.make_tags()."""
2145
2168
        """See BranchFormat.get_format_description()."""
2146
2169
        return "Branch format 8"
2147
2170
 
2148
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2171
    def initialize(self, a_bzrdir, name=None):
2149
2172
        """Create a branch of this format in a_bzrdir."""
2150
2173
        utf8_files = [('last-revision', '0 null:\n'),
2151
2174
                      ('branch.conf', ''),
2152
2175
                      ('tags', ''),
2153
2176
                      ('references', '')
2154
2177
                      ]
2155
 
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
 
2178
        return self._initialize_helper(a_bzrdir, utf8_files, name)
 
2179
 
 
2180
    def __init__(self):
 
2181
        super(BzrBranchFormat8, self).__init__()
 
2182
        self._matchingbzrdir.repository_format = \
 
2183
            RepositoryFormatKnitPack5RichRoot()
2156
2184
 
2157
2185
    def make_tags(self, branch):
2158
2186
        """See bzrlib.branch.BranchFormat.make_tags()."""
2167
2195
    supports_reference_locations = True
2168
2196
 
2169
2197
 
2170
 
class BzrBranchFormat7(BranchFormatMetadir):
 
2198
class BzrBranchFormat7(BzrBranchFormat8):
2171
2199
    """Branch format with last-revision, tags, and a stacked location pointer.
2172
2200
 
2173
2201
    The stacked location pointer is passed down to the repository and requires
2176
2204
    This format was introduced in bzr 1.6.
2177
2205
    """
2178
2206
 
2179
 
    def initialize(self, a_bzrdir, name=None, repository=None):
 
2207
    def initialize(self, a_bzrdir, name=None):
2180
2208
        """Create a branch of this format in a_bzrdir."""
2181
2209
        utf8_files = [('last-revision', '0 null:\n'),
2182
2210
                      ('branch.conf', ''),
2183
2211
                      ('tags', ''),
2184
2212
                      ]
2185
 
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
 
2213
        return self._initialize_helper(a_bzrdir, utf8_files, name)
2186
2214
 
2187
2215
    def _branch_class(self):
2188
2216
        return BzrBranch7
2198
2226
    def supports_set_append_revisions_only(self):
2199
2227
        return True
2200
2228
 
2201
 
    def supports_stacking(self):
2202
 
        return True
2203
 
 
2204
 
    def make_tags(self, branch):
2205
 
        """See bzrlib.branch.BranchFormat.make_tags()."""
2206
 
        return BasicTags(branch)
2207
 
 
2208
2229
    supports_reference_locations = False
2209
2230
 
2210
2231
 
2237
2258
        transport = a_bzrdir.get_branch_transport(None, name=name)
2238
2259
        location = transport.put_bytes('location', to_branch.base)
2239
2260
 
2240
 
    def initialize(self, a_bzrdir, name=None, target_branch=None,
2241
 
            repository=None):
 
2261
    def initialize(self, a_bzrdir, name=None, target_branch=None):
2242
2262
        """Create a branch of this format in a_bzrdir."""
2243
2263
        if target_branch is None:
2244
2264
            # this format does not implement branch itself, thus the implicit
2272
2292
        return clone
2273
2293
 
2274
2294
    def open(self, a_bzrdir, name=None, _found=False, location=None,
2275
 
             possible_transports=None, ignore_fallbacks=False,
2276
 
             found_repository=None):
 
2295
             possible_transports=None, ignore_fallbacks=False):
2277
2296
        """Return the branch that the branch reference in a_bzrdir points at.
2278
2297
 
2279
2298
        :param a_bzrdir: A BzrDir that contains a branch.
2310
2329
        return result
2311
2330
 
2312
2331
 
2313
 
class BranchFormatRegistry(controldir.ControlComponentFormatRegistry):
2314
 
    """Branch format registry."""
2315
 
 
2316
 
    def __init__(self, other_registry=None):
2317
 
        super(BranchFormatRegistry, self).__init__(other_registry)
2318
 
        self._default_format = None
2319
 
 
2320
 
    def set_default(self, format):
2321
 
        self._default_format = format
2322
 
 
2323
 
    def get_default(self):
2324
 
        return self._default_format
2325
 
 
2326
 
 
2327
2332
network_format_registry = registry.FormatRegistry()
2328
2333
"""Registry of formats indexed by their network name.
2329
2334
 
2332
2337
BranchFormat.network_name() for more detail.
2333
2338
"""
2334
2339
 
2335
 
format_registry = BranchFormatRegistry(network_format_registry)
2336
 
 
2337
2340
 
2338
2341
# formats which have no format string are not discoverable
2339
2342
# and not independently creatable, so are not registered.
2341
2344
__format6 = BzrBranchFormat6()
2342
2345
__format7 = BzrBranchFormat7()
2343
2346
__format8 = BzrBranchFormat8()
2344
 
format_registry.register(__format5)
2345
 
format_registry.register(BranchReferenceFormat())
2346
 
format_registry.register(__format6)
2347
 
format_registry.register(__format7)
2348
 
format_registry.register(__format8)
2349
 
format_registry.set_default(__format7)
 
2347
BranchFormat.register_format(__format5)
 
2348
BranchFormat.register_format(BranchReferenceFormat())
 
2349
BranchFormat.register_format(__format6)
 
2350
BranchFormat.register_format(__format7)
 
2351
BranchFormat.register_format(__format8)
 
2352
BranchFormat.set_default_format(__format7)
 
2353
_legacy_formats = [BzrBranchFormat4(),
 
2354
    ]
 
2355
network_format_registry.register(
 
2356
    _legacy_formats[0].network_name(), _legacy_formats[0].__class__)
2350
2357
 
2351
2358
 
2352
2359
class BranchWriteLockResult(LogicalLockResult):
2500
2507
        """See Branch.print_file."""
2501
2508
        return self.repository.print_file(file, revision_id)
2502
2509
 
 
2510
    def _write_revision_history(self, history):
 
2511
        """Factored out of set_revision_history.
 
2512
 
 
2513
        This performs the actual writing to disk.
 
2514
        It is intended to be called by BzrBranch5.set_revision_history."""
 
2515
        self._transport.put_bytes(
 
2516
            'revision-history', '\n'.join(history),
 
2517
            mode=self.bzrdir._get_file_mode())
 
2518
 
 
2519
    @needs_write_lock
 
2520
    def set_revision_history(self, rev_history):
 
2521
        """See Branch.set_revision_history."""
 
2522
        if 'evil' in debug.debug_flags:
 
2523
            mutter_callsite(3, "set_revision_history scales with history.")
 
2524
        check_not_reserved_id = _mod_revision.check_not_reserved_id
 
2525
        for rev_id in rev_history:
 
2526
            check_not_reserved_id(rev_id)
 
2527
        if Branch.hooks['post_change_branch_tip']:
 
2528
            # Don't calculate the last_revision_info() if there are no hooks
 
2529
            # that will use it.
 
2530
            old_revno, old_revid = self.last_revision_info()
 
2531
        if len(rev_history) == 0:
 
2532
            revid = _mod_revision.NULL_REVISION
 
2533
        else:
 
2534
            revid = rev_history[-1]
 
2535
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
 
2536
        self._write_revision_history(rev_history)
 
2537
        self._clear_cached_state()
 
2538
        self._cache_revision_history(rev_history)
 
2539
        for hook in Branch.hooks['set_rh']:
 
2540
            hook(self, rev_history)
 
2541
        if Branch.hooks['post_change_branch_tip']:
 
2542
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
 
2543
 
 
2544
    def _synchronize_history(self, destination, revision_id):
 
2545
        """Synchronize last revision and revision history between branches.
 
2546
 
 
2547
        This version is most efficient when the destination is also a
 
2548
        BzrBranch5, but works for BzrBranch6 as long as the revision
 
2549
        history is the true lefthand parent history, and all of the revisions
 
2550
        are in the destination's repository.  If not, set_revision_history
 
2551
        will fail.
 
2552
 
 
2553
        :param destination: The branch to copy the history into
 
2554
        :param revision_id: The revision-id to truncate history at.  May
 
2555
          be None to copy complete history.
 
2556
        """
 
2557
        if not isinstance(destination._format, BzrBranchFormat5):
 
2558
            super(BzrBranch, self)._synchronize_history(
 
2559
                destination, revision_id)
 
2560
            return
 
2561
        if revision_id == _mod_revision.NULL_REVISION:
 
2562
            new_history = []
 
2563
        else:
 
2564
            new_history = self.revision_history()
 
2565
        if revision_id is not None and new_history != []:
 
2566
            try:
 
2567
                new_history = new_history[:new_history.index(revision_id) + 1]
 
2568
            except ValueError:
 
2569
                rev = self.repository.get_revision(revision_id)
 
2570
                new_history = rev.get_history(self.repository)[1:]
 
2571
        destination.set_revision_history(new_history)
 
2572
 
2503
2573
    @needs_write_lock
2504
2574
    def set_last_revision_info(self, revno, revision_id):
2505
 
        if not revision_id or not isinstance(revision_id, basestring):
2506
 
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
 
2575
        """Set the last revision of this branch.
 
2576
 
 
2577
        The caller is responsible for checking that the revno is correct
 
2578
        for this revision id.
 
2579
 
 
2580
        It may be possible to set the branch last revision to an id not
 
2581
        present in the repository.  However, branches can also be
 
2582
        configured to check constraints on history, in which case this may not
 
2583
        be permitted.
 
2584
        """
2507
2585
        revision_id = _mod_revision.ensure_null(revision_id)
2508
 
        old_revno, old_revid = self.last_revision_info()
2509
 
        if self._get_append_revisions_only():
2510
 
            self._check_history_violation(revision_id)
2511
 
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
2512
 
        self._write_last_revision_info(revno, revision_id)
2513
 
        self._clear_cached_state()
2514
 
        self._last_revision_info_cache = revno, revision_id
2515
 
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
 
2586
        # this old format stores the full history, but this api doesn't
 
2587
        # provide it, so we must generate, and might as well check it's
 
2588
        # correct
 
2589
        history = self._lefthand_history(revision_id)
 
2590
        if len(history) != revno:
 
2591
            raise AssertionError('%d != %d' % (len(history), revno))
 
2592
        self.set_revision_history(history)
 
2593
 
 
2594
    def _gen_revision_history(self):
 
2595
        history = self._transport.get_bytes('revision-history').split('\n')
 
2596
        if history[-1:] == ['']:
 
2597
            # There shouldn't be a trailing newline, but just in case.
 
2598
            history.pop()
 
2599
        return history
 
2600
 
 
2601
    @needs_write_lock
 
2602
    def generate_revision_history(self, revision_id, last_rev=None,
 
2603
        other_branch=None):
 
2604
        """Create a new revision history that will finish with revision_id.
 
2605
 
 
2606
        :param revision_id: the new tip to use.
 
2607
        :param last_rev: The previous last_revision. If not None, then this
 
2608
            must be a ancestory of revision_id, or DivergedBranches is raised.
 
2609
        :param other_branch: The other branch that DivergedBranches should
 
2610
            raise with respect to.
 
2611
        """
 
2612
        self.set_revision_history(self._lefthand_history(revision_id,
 
2613
            last_rev, other_branch))
2516
2614
 
2517
2615
    def basis_tree(self):
2518
2616
        """See Branch.basis_tree."""
2527
2625
                pass
2528
2626
        return None
2529
2627
 
 
2628
    def _basic_push(self, target, overwrite, stop_revision):
 
2629
        """Basic implementation of push without bound branches or hooks.
 
2630
 
 
2631
        Must be called with source read locked and target write locked.
 
2632
        """
 
2633
        result = BranchPushResult()
 
2634
        result.source_branch = self
 
2635
        result.target_branch = target
 
2636
        result.old_revno, result.old_revid = target.last_revision_info()
 
2637
        self.update_references(target)
 
2638
        if result.old_revid != self.last_revision():
 
2639
            # We assume that during 'push' this repository is closer than
 
2640
            # the target.
 
2641
            graph = self.repository.get_graph(target.repository)
 
2642
            target.update_revisions(self, stop_revision,
 
2643
                overwrite=overwrite, graph=graph)
 
2644
        if self._push_should_merge_tags():
 
2645
            result.tag_conflicts = self.tags.merge_to(target.tags,
 
2646
                overwrite)
 
2647
        result.new_revno, result.new_revid = target.last_revision_info()
 
2648
        return result
 
2649
 
2530
2650
    def get_stacked_on_url(self):
2531
2651
        raise errors.UnstackableBranchFormat(self._format, self.user_url)
2532
2652
 
2543
2663
            self._transport.put_bytes('parent', url + '\n',
2544
2664
                mode=self.bzrdir._get_file_mode())
2545
2665
 
 
2666
 
 
2667
class BzrBranch5(BzrBranch):
 
2668
    """A format 5 branch. This supports new features over plain branches.
 
2669
 
 
2670
    It has support for a master_branch which is the data for bound branches.
 
2671
    """
 
2672
 
 
2673
    def get_bound_location(self):
 
2674
        try:
 
2675
            return self._transport.get_bytes('bound')[:-1]
 
2676
        except errors.NoSuchFile:
 
2677
            return None
 
2678
 
 
2679
    @needs_read_lock
 
2680
    def get_master_branch(self, possible_transports=None):
 
2681
        """Return the branch we are bound to.
 
2682
 
 
2683
        :return: Either a Branch, or None
 
2684
 
 
2685
        This could memoise the branch, but if thats done
 
2686
        it must be revalidated on each new lock.
 
2687
        So for now we just don't memoise it.
 
2688
        # RBC 20060304 review this decision.
 
2689
        """
 
2690
        bound_loc = self.get_bound_location()
 
2691
        if not bound_loc:
 
2692
            return None
 
2693
        try:
 
2694
            return Branch.open(bound_loc,
 
2695
                               possible_transports=possible_transports)
 
2696
        except (errors.NotBranchError, errors.ConnectionError), e:
 
2697
            raise errors.BoundBranchConnectionFailure(
 
2698
                    self, bound_loc, e)
 
2699
 
2546
2700
    @needs_write_lock
2547
 
    def unbind(self):
2548
 
        """If bound, unbind"""
2549
 
        return self.set_bound_location(None)
 
2701
    def set_bound_location(self, location):
 
2702
        """Set the target where this branch is bound to.
 
2703
 
 
2704
        :param location: URL to the target branch
 
2705
        """
 
2706
        if location:
 
2707
            self._transport.put_bytes('bound', location+'\n',
 
2708
                mode=self.bzrdir._get_file_mode())
 
2709
        else:
 
2710
            try:
 
2711
                self._transport.delete('bound')
 
2712
            except errors.NoSuchFile:
 
2713
                return False
 
2714
            return True
2550
2715
 
2551
2716
    @needs_write_lock
2552
2717
    def bind(self, other):
2574
2739
        # history around
2575
2740
        self.set_bound_location(other.base)
2576
2741
 
2577
 
    def get_bound_location(self):
2578
 
        try:
2579
 
            return self._transport.get_bytes('bound')[:-1]
2580
 
        except errors.NoSuchFile:
2581
 
            return None
2582
 
 
2583
 
    @needs_read_lock
2584
 
    def get_master_branch(self, possible_transports=None):
2585
 
        """Return the branch we are bound to.
2586
 
 
2587
 
        :return: Either a Branch, or None
2588
 
        """
2589
 
        if self._master_branch_cache is None:
2590
 
            self._master_branch_cache = self._get_master_branch(
2591
 
                possible_transports)
2592
 
        return self._master_branch_cache
2593
 
 
2594
 
    def _get_master_branch(self, possible_transports):
2595
 
        bound_loc = self.get_bound_location()
2596
 
        if not bound_loc:
2597
 
            return None
2598
 
        try:
2599
 
            return Branch.open(bound_loc,
2600
 
                               possible_transports=possible_transports)
2601
 
        except (errors.NotBranchError, errors.ConnectionError), e:
2602
 
            raise errors.BoundBranchConnectionFailure(
2603
 
                    self, bound_loc, e)
2604
 
 
2605
2742
    @needs_write_lock
2606
 
    def set_bound_location(self, location):
2607
 
        """Set the target where this branch is bound to.
2608
 
 
2609
 
        :param location: URL to the target branch
2610
 
        """
2611
 
        self._master_branch_cache = None
2612
 
        if location:
2613
 
            self._transport.put_bytes('bound', location+'\n',
2614
 
                mode=self.bzrdir._get_file_mode())
2615
 
        else:
2616
 
            try:
2617
 
                self._transport.delete('bound')
2618
 
            except errors.NoSuchFile:
2619
 
                return False
2620
 
            return True
 
2743
    def unbind(self):
 
2744
        """If bound, unbind"""
 
2745
        return self.set_bound_location(None)
2621
2746
 
2622
2747
    @needs_write_lock
2623
2748
    def update(self, possible_transports=None):
2636
2761
            return old_tip
2637
2762
        return None
2638
2763
 
2639
 
    def _read_last_revision_info(self):
2640
 
        revision_string = self._transport.get_bytes('last-revision')
2641
 
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
2642
 
        revision_id = cache_utf8.get_cached_utf8(revision_id)
2643
 
        revno = int(revno)
2644
 
        return revno, revision_id
2645
 
 
2646
 
    def _write_last_revision_info(self, revno, revision_id):
2647
 
        """Simply write out the revision id, with no checks.
2648
 
 
2649
 
        Use set_last_revision_info to perform this safely.
2650
 
 
2651
 
        Does not update the revision_history cache.
2652
 
        """
2653
 
        revision_id = _mod_revision.ensure_null(revision_id)
2654
 
        out_string = '%d %s\n' % (revno, revision_id)
2655
 
        self._transport.put_bytes('last-revision', out_string,
2656
 
            mode=self.bzrdir._get_file_mode())
2657
 
 
2658
 
 
2659
 
class FullHistoryBzrBranch(BzrBranch):
2660
 
    """Bzr branch which contains the full revision history."""
2661
 
 
2662
 
    @needs_write_lock
2663
 
    def set_last_revision_info(self, revno, revision_id):
2664
 
        if not revision_id or not isinstance(revision_id, basestring):
2665
 
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
2666
 
        revision_id = _mod_revision.ensure_null(revision_id)
2667
 
        # this old format stores the full history, but this api doesn't
2668
 
        # provide it, so we must generate, and might as well check it's
2669
 
        # correct
2670
 
        history = self._lefthand_history(revision_id)
2671
 
        if len(history) != revno:
2672
 
            raise AssertionError('%d != %d' % (len(history), revno))
2673
 
        self._set_revision_history(history)
2674
 
 
2675
 
    def _read_last_revision_info(self):
2676
 
        rh = self.revision_history()
2677
 
        revno = len(rh)
2678
 
        if revno:
2679
 
            return (revno, rh[-1])
2680
 
        else:
2681
 
            return (0, _mod_revision.NULL_REVISION)
2682
 
 
2683
 
    @deprecated_method(deprecated_in((2, 4, 0)))
2684
 
    @needs_write_lock
2685
 
    def set_revision_history(self, rev_history):
2686
 
        """See Branch.set_revision_history."""
2687
 
        self._set_revision_history(rev_history)
2688
 
 
2689
 
    def _set_revision_history(self, rev_history):
2690
 
        if 'evil' in debug.debug_flags:
2691
 
            mutter_callsite(3, "set_revision_history scales with history.")
2692
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
2693
 
        for rev_id in rev_history:
2694
 
            check_not_reserved_id(rev_id)
2695
 
        if Branch.hooks['post_change_branch_tip']:
2696
 
            # Don't calculate the last_revision_info() if there are no hooks
2697
 
            # that will use it.
2698
 
            old_revno, old_revid = self.last_revision_info()
2699
 
        if len(rev_history) == 0:
2700
 
            revid = _mod_revision.NULL_REVISION
2701
 
        else:
2702
 
            revid = rev_history[-1]
2703
 
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
2704
 
        self._write_revision_history(rev_history)
2705
 
        self._clear_cached_state()
2706
 
        self._cache_revision_history(rev_history)
2707
 
        for hook in Branch.hooks['set_rh']:
2708
 
            hook(self, rev_history)
2709
 
        if Branch.hooks['post_change_branch_tip']:
2710
 
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2711
 
 
2712
 
    def _write_revision_history(self, history):
2713
 
        """Factored out of set_revision_history.
2714
 
 
2715
 
        This performs the actual writing to disk.
2716
 
        It is intended to be called by set_revision_history."""
2717
 
        self._transport.put_bytes(
2718
 
            'revision-history', '\n'.join(history),
2719
 
            mode=self.bzrdir._get_file_mode())
2720
 
 
2721
 
    def _gen_revision_history(self):
2722
 
        history = self._transport.get_bytes('revision-history').split('\n')
2723
 
        if history[-1:] == ['']:
2724
 
            # There shouldn't be a trailing newline, but just in case.
2725
 
            history.pop()
2726
 
        return history
2727
 
 
2728
 
    def _synchronize_history(self, destination, revision_id):
2729
 
        if not isinstance(destination, FullHistoryBzrBranch):
2730
 
            super(BzrBranch, self)._synchronize_history(
2731
 
                destination, revision_id)
2732
 
            return
2733
 
        if revision_id == _mod_revision.NULL_REVISION:
2734
 
            new_history = []
2735
 
        else:
2736
 
            new_history = self.revision_history()
2737
 
        if revision_id is not None and new_history != []:
2738
 
            try:
2739
 
                new_history = new_history[:new_history.index(revision_id) + 1]
2740
 
            except ValueError:
2741
 
                rev = self.repository.get_revision(revision_id)
2742
 
                new_history = rev.get_history(self.repository)[1:]
2743
 
        destination._set_revision_history(new_history)
2744
 
 
2745
 
    @needs_write_lock
2746
 
    def generate_revision_history(self, revision_id, last_rev=None,
2747
 
        other_branch=None):
2748
 
        """Create a new revision history that will finish with revision_id.
2749
 
 
2750
 
        :param revision_id: the new tip to use.
2751
 
        :param last_rev: The previous last_revision. If not None, then this
2752
 
            must be a ancestory of revision_id, or DivergedBranches is raised.
2753
 
        :param other_branch: The other branch that DivergedBranches should
2754
 
            raise with respect to.
2755
 
        """
2756
 
        self._set_revision_history(self._lefthand_history(revision_id,
2757
 
            last_rev, other_branch))
2758
 
 
2759
 
 
2760
 
class BzrBranch5(FullHistoryBzrBranch):
2761
 
    """A format 5 branch. This supports new features over plain branches.
2762
 
 
2763
 
    It has support for a master_branch which is the data for bound branches.
2764
 
    """
2765
 
 
2766
 
 
2767
 
class BzrBranch8(BzrBranch):
 
2764
 
 
2765
class BzrBranch8(BzrBranch5):
2768
2766
    """A branch that stores tree-reference locations."""
2769
2767
 
2770
2768
    def _open_hook(self):
2796
2794
        self._last_revision_info_cache = None
2797
2795
        self._reference_info = None
2798
2796
 
 
2797
    def _last_revision_info(self):
 
2798
        revision_string = self._transport.get_bytes('last-revision')
 
2799
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
 
2800
        revision_id = cache_utf8.get_cached_utf8(revision_id)
 
2801
        revno = int(revno)
 
2802
        return revno, revision_id
 
2803
 
 
2804
    def _write_last_revision_info(self, revno, revision_id):
 
2805
        """Simply write out the revision id, with no checks.
 
2806
 
 
2807
        Use set_last_revision_info to perform this safely.
 
2808
 
 
2809
        Does not update the revision_history cache.
 
2810
        Intended to be called by set_last_revision_info and
 
2811
        _write_revision_history.
 
2812
        """
 
2813
        revision_id = _mod_revision.ensure_null(revision_id)
 
2814
        out_string = '%d %s\n' % (revno, revision_id)
 
2815
        self._transport.put_bytes('last-revision', out_string,
 
2816
            mode=self.bzrdir._get_file_mode())
 
2817
 
 
2818
    @needs_write_lock
 
2819
    def set_last_revision_info(self, revno, revision_id):
 
2820
        revision_id = _mod_revision.ensure_null(revision_id)
 
2821
        old_revno, old_revid = self.last_revision_info()
 
2822
        if self._get_append_revisions_only():
 
2823
            self._check_history_violation(revision_id)
 
2824
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
 
2825
        self._write_last_revision_info(revno, revision_id)
 
2826
        self._clear_cached_state()
 
2827
        self._last_revision_info_cache = revno, revision_id
 
2828
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
 
2829
 
 
2830
    def _synchronize_history(self, destination, revision_id):
 
2831
        """Synchronize last revision and revision history between branches.
 
2832
 
 
2833
        :see: Branch._synchronize_history
 
2834
        """
 
2835
        # XXX: The base Branch has a fast implementation of this method based
 
2836
        # on set_last_revision_info, but BzrBranch/BzrBranch5 have a slower one
 
2837
        # that uses set_revision_history.  This class inherits from BzrBranch5,
 
2838
        # but wants the fast implementation, so it calls
 
2839
        # Branch._synchronize_history directly.
 
2840
        Branch._synchronize_history(self, destination, revision_id)
 
2841
 
2799
2842
    def _check_history_violation(self, revision_id):
2800
 
        current_revid = self.last_revision()
2801
 
        last_revision = _mod_revision.ensure_null(current_revid)
 
2843
        last_revision = _mod_revision.ensure_null(self.last_revision())
2802
2844
        if _mod_revision.is_null(last_revision):
2803
2845
            return
2804
 
        graph = self.repository.get_graph()
2805
 
        for lh_ancestor in graph.iter_lefthand_ancestry(revision_id):
2806
 
            if lh_ancestor == current_revid:
2807
 
                return
2808
 
        raise errors.AppendRevisionsOnlyViolation(self.user_url)
 
2846
        if last_revision not in self._lefthand_history(revision_id):
 
2847
            raise errors.AppendRevisionsOnlyViolation(self.user_url)
2809
2848
 
2810
2849
    def _gen_revision_history(self):
2811
2850
        """Generate the revision history from last revision
2814
2853
        self._extend_partial_history(stop_index=last_revno-1)
2815
2854
        return list(reversed(self._partial_revision_history_cache))
2816
2855
 
 
2856
    def _write_revision_history(self, history):
 
2857
        """Factored out of set_revision_history.
 
2858
 
 
2859
        This performs the actual writing to disk, with format-specific checks.
 
2860
        It is intended to be called by BzrBranch5.set_revision_history.
 
2861
        """
 
2862
        if len(history) == 0:
 
2863
            last_revision = 'null:'
 
2864
        else:
 
2865
            if history != self._lefthand_history(history[-1]):
 
2866
                raise errors.NotLefthandHistory(history)
 
2867
            last_revision = history[-1]
 
2868
        if self._get_append_revisions_only():
 
2869
            self._check_history_violation(last_revision)
 
2870
        self._write_last_revision_info(len(history), last_revision)
 
2871
 
2817
2872
    @needs_write_lock
2818
2873
    def _set_parent_location(self, url):
2819
2874
        """Set the parent branch"""
2905
2960
 
2906
2961
    def set_bound_location(self, location):
2907
2962
        """See Branch.set_push_location."""
2908
 
        self._master_branch_cache = None
2909
2963
        result = None
2910
2964
        config = self.get_config()
2911
2965
        if location is None:
2942
2996
        # you can always ask for the URL; but you might not be able to use it
2943
2997
        # if the repo can't support stacking.
2944
2998
        ## self._check_stackable_repo()
2945
 
        # stacked_on_location is only ever defined in branch.conf, so don't
2946
 
        # waste effort reading the whole stack of config files.
2947
 
        config = self.get_config()._get_branch_data_config()
2948
 
        stacked_url = self._get_config_location('stacked_on_location',
2949
 
            config=config)
 
2999
        stacked_url = self._get_config_location('stacked_on_location')
2950
3000
        if stacked_url is None:
2951
3001
            raise errors.NotStacked(self)
2952
3002
        return stacked_url
2955
3005
        return self.get_config(
2956
3006
            ).get_user_option_as_bool('append_revisions_only')
2957
3007
 
 
3008
    @needs_write_lock
 
3009
    def generate_revision_history(self, revision_id, last_rev=None,
 
3010
                                  other_branch=None):
 
3011
        """See BzrBranch5.generate_revision_history"""
 
3012
        history = self._lefthand_history(revision_id, last_rev, other_branch)
 
3013
        revno = len(history)
 
3014
        self.set_last_revision_info(revno, revision_id)
 
3015
 
2958
3016
    @needs_read_lock
2959
3017
    def get_rev_id(self, revno, history=None):
2960
3018
        """Find the revision id of the specified revno."""
2984
3042
        try:
2985
3043
            index = self._partial_revision_history_cache.index(revision_id)
2986
3044
        except ValueError:
2987
 
            try:
2988
 
                self._extend_partial_history(stop_revision=revision_id)
2989
 
            except errors.RevisionNotPresent, e:
2990
 
                raise errors.GhostRevisionsHaveNoRevno(revision_id, e.revision_id)
 
3045
            self._extend_partial_history(stop_revision=revision_id)
2991
3046
            index = len(self._partial_revision_history_cache) - 1
2992
3047
            if self._partial_revision_history_cache[index] != revision_id:
2993
3048
                raise errors.NoSuchRevision(self, revision_id)
3174
3229
        branch._transport.put_bytes('format', format.get_format_string())
3175
3230
 
3176
3231
 
 
3232
def _run_with_write_locked_target(target, callable, *args, **kwargs):
 
3233
    """Run ``callable(*args, **kwargs)``, write-locking target for the
 
3234
    duration.
 
3235
 
 
3236
    _run_with_write_locked_target will attempt to release the lock it acquires.
 
3237
 
 
3238
    If an exception is raised by callable, then that exception *will* be
 
3239
    propagated, even if the unlock attempt raises its own error.  Thus
 
3240
    _run_with_write_locked_target should be preferred to simply doing::
 
3241
 
 
3242
        target.lock_write()
 
3243
        try:
 
3244
            return callable(*args, **kwargs)
 
3245
        finally:
 
3246
            target.unlock()
 
3247
 
 
3248
    """
 
3249
    # This is very similar to bzrlib.decorators.needs_write_lock.  Perhaps they
 
3250
    # should share code?
 
3251
    target.lock_write()
 
3252
    try:
 
3253
        result = callable(*args, **kwargs)
 
3254
    except:
 
3255
        exc_info = sys.exc_info()
 
3256
        try:
 
3257
            target.unlock()
 
3258
        finally:
 
3259
            raise exc_info[0], exc_info[1], exc_info[2]
 
3260
    else:
 
3261
        target.unlock()
 
3262
        return result
 
3263
 
 
3264
 
3177
3265
class InterBranch(InterObject):
3178
3266
    """This class represents operations taking place between two branches.
3179
3267
 
3207
3295
        raise NotImplementedError(self.pull)
3208
3296
 
3209
3297
    @needs_write_lock
3210
 
    def push(self, overwrite=False, stop_revision=None, lossy=False,
 
3298
    def update_revisions(self, stop_revision=None, overwrite=False,
 
3299
                         graph=None):
 
3300
        """Pull in new perfect-fit revisions.
 
3301
 
 
3302
        :param stop_revision: Updated until the given revision
 
3303
        :param overwrite: Always set the branch pointer, rather than checking
 
3304
            to see if it is a proper descendant.
 
3305
        :param graph: A Graph object that can be used to query history
 
3306
            information. This can be None.
 
3307
        :return: None
 
3308
        """
 
3309
        raise NotImplementedError(self.update_revisions)
 
3310
 
 
3311
    @needs_write_lock
 
3312
    def push(self, overwrite=False, stop_revision=None,
3211
3313
             _override_hook_source_branch=None):
3212
3314
        """Mirror the source branch into the target branch.
3213
3315
 
3224
3326
        """
3225
3327
        raise NotImplementedError(self.copy_content_into)
3226
3328
 
3227
 
    @needs_write_lock
3228
 
    def fetch(self, stop_revision=None, limit=None):
3229
 
        """Fetch revisions.
3230
 
 
3231
 
        :param stop_revision: Last revision to fetch
3232
 
        :param limit: Optional rough limit of revisions to fetch
3233
 
        """
3234
 
        raise NotImplementedError(self.fetch)
3235
 
 
3236
3329
 
3237
3330
class GenericInterBranch(InterBranch):
3238
3331
    """InterBranch implementation that uses public Branch functions."""
3244
3337
 
3245
3338
    @classmethod
3246
3339
    def _get_branch_formats_to_test(klass):
3247
 
        return [(format_registry.get_default(), format_registry.get_default())]
 
3340
        return [(BranchFormat._default_format, BranchFormat._default_format)]
3248
3341
 
3249
3342
    @classmethod
3250
3343
    def unwrap_format(klass, format):
3251
3344
        if isinstance(format, remote.RemoteBranchFormat):
3252
3345
            format._ensure_real()
3253
3346
            return format._custom_format
3254
 
        return format
 
3347
        return format                                                                                                  
3255
3348
 
3256
3349
    @needs_write_lock
3257
3350
    def copy_content_into(self, revision_id=None):
3273
3366
            self.source.tags.merge_to(self.target.tags)
3274
3367
 
3275
3368
    @needs_write_lock
3276
 
    def fetch(self, stop_revision=None, limit=None):
3277
 
        if self.target.base == self.source.base:
3278
 
            return (0, [])
3279
 
        self.source.lock_read()
3280
 
        try:
3281
 
            fetch_spec_factory = fetch.FetchSpecFactory()
3282
 
            fetch_spec_factory.source_branch = self.source
3283
 
            fetch_spec_factory.source_branch_stop_revision_id = stop_revision
3284
 
            fetch_spec_factory.source_repo = self.source.repository
3285
 
            fetch_spec_factory.target_repo = self.target.repository
3286
 
            fetch_spec_factory.target_repo_kind = fetch.TargetRepoKinds.PREEXISTING
3287
 
            fetch_spec_factory.limit = limit
3288
 
            fetch_spec = fetch_spec_factory.make_fetch_spec()
3289
 
            return self.target.repository.fetch(self.source.repository,
3290
 
                fetch_spec=fetch_spec)
3291
 
        finally:
3292
 
            self.source.unlock()
3293
 
 
3294
 
    @needs_write_lock
3295
 
    def _update_revisions(self, stop_revision=None, overwrite=False,
3296
 
            graph=None):
 
3369
    def update_revisions(self, stop_revision=None, overwrite=False,
 
3370
        graph=None):
 
3371
        """See InterBranch.update_revisions()."""
3297
3372
        other_revno, other_last_revision = self.source.last_revision_info()
3298
3373
        stop_revno = None # unknown
3299
3374
        if stop_revision is None:
3310
3385
        # case of having something to pull, and so that the check for
3311
3386
        # already merged can operate on the just fetched graph, which will
3312
3387
        # be cached in memory.
3313
 
        self.fetch(stop_revision=stop_revision)
 
3388
        self.target.fetch(self.source, stop_revision)
3314
3389
        # Check to see if one is an ancestor of the other
3315
3390
        if not overwrite:
3316
3391
            if graph is None:
3344
3419
        if local and not bound_location:
3345
3420
            raise errors.LocalRequiresBoundBranch()
3346
3421
        master_branch = None
3347
 
        source_is_master = (self.source.user_url == bound_location)
3348
 
        if not local and bound_location and not source_is_master:
 
3422
        if not local and bound_location and self.source.user_url != bound_location:
3349
3423
            # not pulling from master, so we need to update master.
3350
3424
            master_branch = self.target.get_master_branch(possible_transports)
3351
3425
            master_branch.lock_write()
3357
3431
            return self._pull(overwrite,
3358
3432
                stop_revision, _hook_master=master_branch,
3359
3433
                run_hooks=run_hooks,
3360
 
                _override_hook_target=_override_hook_target,
3361
 
                merge_tags_to_master=not source_is_master)
 
3434
                _override_hook_target=_override_hook_target)
3362
3435
        finally:
3363
3436
            if master_branch:
3364
3437
                master_branch.unlock()
3365
3438
 
3366
 
    def push(self, overwrite=False, stop_revision=None, lossy=False,
 
3439
    def push(self, overwrite=False, stop_revision=None,
3367
3440
             _override_hook_source_branch=None):
3368
3441
        """See InterBranch.push.
3369
3442
 
3370
3443
        This is the basic concrete implementation of push()
3371
3444
 
3372
 
        :param _override_hook_source_branch: If specified, run the hooks
3373
 
            passing this Branch as the source, rather than self.  This is for
3374
 
            use of RemoteBranch, where push is delegated to the underlying
3375
 
            vfs-based Branch.
 
3445
        :param _override_hook_source_branch: If specified, run
 
3446
        the hooks passing this Branch as the source, rather than self.
 
3447
        This is for use of RemoteBranch, where push is delegated to the
 
3448
        underlying vfs-based Branch.
3376
3449
        """
3377
 
        if lossy:
3378
 
            raise errors.LossyPushToSameVCS(self.source, self.target)
3379
3450
        # TODO: Public option to disable running hooks - should be trivial but
3380
3451
        # needs tests.
3381
 
 
3382
 
        op = cleanup.OperationWithCleanups(self._push_with_bound_branches)
3383
 
        op.add_cleanup(self.source.lock_read().unlock)
3384
 
        op.add_cleanup(self.target.lock_write().unlock)
3385
 
        return op.run(overwrite, stop_revision,
3386
 
            _override_hook_source_branch=_override_hook_source_branch)
3387
 
 
3388
 
    def _basic_push(self, overwrite, stop_revision):
3389
 
        """Basic implementation of push without bound branches or hooks.
3390
 
 
3391
 
        Must be called with source read locked and target write locked.
3392
 
        """
3393
 
        result = BranchPushResult()
3394
 
        result.source_branch = self.source
3395
 
        result.target_branch = self.target
3396
 
        result.old_revno, result.old_revid = self.target.last_revision_info()
3397
 
        self.source.update_references(self.target)
3398
 
        if result.old_revid != stop_revision:
3399
 
            # We assume that during 'push' this repository is closer than
3400
 
            # the target.
3401
 
            graph = self.source.repository.get_graph(self.target.repository)
3402
 
            self._update_revisions(stop_revision, overwrite=overwrite,
3403
 
                    graph=graph)
3404
 
        if self.source._push_should_merge_tags():
3405
 
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3406
 
                overwrite)
3407
 
        result.new_revno, result.new_revid = self.target.last_revision_info()
3408
 
        return result
3409
 
 
3410
 
    def _push_with_bound_branches(self, operation, overwrite, stop_revision,
 
3452
        self.source.lock_read()
 
3453
        try:
 
3454
            return _run_with_write_locked_target(
 
3455
                self.target, self._push_with_bound_branches, overwrite,
 
3456
                stop_revision,
 
3457
                _override_hook_source_branch=_override_hook_source_branch)
 
3458
        finally:
 
3459
            self.source.unlock()
 
3460
 
 
3461
    def _push_with_bound_branches(self, overwrite, stop_revision,
3411
3462
            _override_hook_source_branch=None):
3412
3463
        """Push from source into target, and into target's master if any.
3413
3464
        """
3425
3476
            # be bound to itself? -- mbp 20070507
3426
3477
            master_branch = self.target.get_master_branch()
3427
3478
            master_branch.lock_write()
3428
 
            operation.add_cleanup(master_branch.unlock)
3429
 
            # push into the master from the source branch.
3430
 
            master_inter = InterBranch.get(self.source, master_branch)
3431
 
            master_inter._basic_push(overwrite, stop_revision)
3432
 
            # and push into the target branch from the source. Note that
3433
 
            # we push from the source branch again, because it's considered
3434
 
            # the highest bandwidth repository.
3435
 
            result = self._basic_push(overwrite, stop_revision)
3436
 
            result.master_branch = master_branch
3437
 
            result.local_branch = self.target
 
3479
            try:
 
3480
                # push into the master from the source branch.
 
3481
                self.source._basic_push(master_branch, overwrite, stop_revision)
 
3482
                # and push into the target branch from the source. Note that we
 
3483
                # push from the source branch again, because its considered the
 
3484
                # highest bandwidth repository.
 
3485
                result = self.source._basic_push(self.target, overwrite,
 
3486
                    stop_revision)
 
3487
                result.master_branch = master_branch
 
3488
                result.local_branch = self.target
 
3489
                _run_hooks()
 
3490
                return result
 
3491
            finally:
 
3492
                master_branch.unlock()
3438
3493
        else:
3439
 
            master_branch = None
3440
3494
            # no master branch
3441
 
            result = self._basic_push(overwrite, stop_revision)
 
3495
            result = self.source._basic_push(self.target, overwrite,
 
3496
                stop_revision)
3442
3497
            # TODO: Why set master_branch and local_branch if there's no
3443
3498
            # binding?  Maybe cleaner to just leave them unset? -- mbp
3444
3499
            # 20070504
3445
3500
            result.master_branch = self.target
3446
3501
            result.local_branch = None
3447
 
        _run_hooks()
3448
 
        return result
 
3502
            _run_hooks()
 
3503
            return result
3449
3504
 
3450
3505
    def _pull(self, overwrite=False, stop_revision=None,
3451
3506
             possible_transports=None, _hook_master=None, run_hooks=True,
3452
 
             _override_hook_target=None, local=False,
3453
 
             merge_tags_to_master=True):
 
3507
             _override_hook_target=None, local=False):
3454
3508
        """See Branch.pull.
3455
3509
 
3456
3510
        This function is the core worker, used by GenericInterBranch.pull to
3461
3515
        :param run_hooks: Private parameter - if false, this branch
3462
3516
            is being called because it's the master of the primary branch,
3463
3517
            so it should not run its hooks.
3464
 
            is being called because it's the master of the primary branch,
3465
 
            so it should not run its hooks.
3466
3518
        :param _override_hook_target: Private parameter - set the branch to be
3467
3519
            supplied as the target_branch to pull hooks.
3468
3520
        :param local: Only update the local branch, and not the bound branch.
3487
3539
            # -- JRV20090506
3488
3540
            result.old_revno, result.old_revid = \
3489
3541
                self.target.last_revision_info()
3490
 
            self._update_revisions(stop_revision, overwrite=overwrite,
3491
 
                graph=graph)
 
3542
            self.target.update_revisions(self.source, stop_revision,
 
3543
                overwrite=overwrite, graph=graph)
3492
3544
            # TODO: The old revid should be specified when merging tags, 
3493
3545
            # so a tags implementation that versions tags can only 
3494
3546
            # pull in the most recent changes. -- JRV20090506
3495
3547
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3496
 
                overwrite, ignore_master=not merge_tags_to_master)
 
3548
                overwrite)
3497
3549
            result.new_revno, result.new_revid = self.target.last_revision_info()
3498
3550
            if _hook_master:
3499
3551
                result.master_branch = _hook_master