~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

(vila) Revise legal option names to be less drastic. (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005-2012 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
from __future__ import absolute_import
 
18
 
 
19
import bzrlib.bzrdir
17
20
 
18
21
from cStringIO import StringIO
19
 
import sys
20
22
 
21
23
from bzrlib.lazy_import import lazy_import
22
24
lazy_import(globals(), """
23
 
from itertools import chain
 
25
import itertools
24
26
from bzrlib import (
25
 
        bzrdir,
26
 
        cache_utf8,
27
 
        config as _mod_config,
28
 
        debug,
29
 
        errors,
30
 
        lockdir,
31
 
        lockable_files,
32
 
        repository,
33
 
        revision as _mod_revision,
34
 
        rio,
35
 
        symbol_versioning,
36
 
        transport,
37
 
        tsort,
38
 
        ui,
39
 
        urlutils,
40
 
        )
41
 
from bzrlib.config import BranchConfig, TransportConfig
42
 
from bzrlib.repofmt.pack_repo import RepositoryFormatKnitPack5RichRoot
43
 
from bzrlib.tag import (
44
 
    BasicTags,
45
 
    DisabledTags,
 
27
    bzrdir,
 
28
    controldir,
 
29
    cache_utf8,
 
30
    cleanup,
 
31
    config as _mod_config,
 
32
    debug,
 
33
    errors,
 
34
    fetch,
 
35
    graph as _mod_graph,
 
36
    lockdir,
 
37
    lockable_files,
 
38
    remote,
 
39
    repository,
 
40
    revision as _mod_revision,
 
41
    rio,
 
42
    shelf,
 
43
    tag as _mod_tag,
 
44
    transport,
 
45
    ui,
 
46
    urlutils,
 
47
    vf_search,
46
48
    )
 
49
from bzrlib.i18n import gettext, ngettext
47
50
""")
48
51
 
49
 
from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises
50
 
from bzrlib.hooks import HookPoint, Hooks
 
52
# Explicitly import bzrlib.bzrdir so that the BzrProber
 
53
# is guaranteed to be registered.
 
54
import bzrlib.bzrdir
 
55
 
 
56
from bzrlib import (
 
57
    bzrdir,
 
58
    controldir,
 
59
    )
 
60
from bzrlib.decorators import (
 
61
    needs_read_lock,
 
62
    needs_write_lock,
 
63
    only_raises,
 
64
    )
 
65
from bzrlib.hooks import Hooks
51
66
from bzrlib.inter import InterObject
52
 
from bzrlib.lock import _RelockDebugMixin
 
67
from bzrlib.lock import _RelockDebugMixin, LogicalLockResult
53
68
from bzrlib import registry
54
69
from bzrlib.symbol_versioning import (
55
70
    deprecated_in,
58
73
from bzrlib.trace import mutter, mutter_callsite, note, is_quiet
59
74
 
60
75
 
61
 
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
62
 
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
63
 
BZR_BRANCH_FORMAT_6 = "Bazaar Branch Format 6 (bzr 0.15)\n"
64
 
 
65
 
 
66
 
# TODO: Maybe include checks for common corruption of newlines, etc?
67
 
 
68
 
# TODO: Some operations like log might retrieve the same revisions
69
 
# repeatedly to calculate deltas.  We could perhaps have a weakref
70
 
# cache in memory to make this faster.  In general anything can be
71
 
# cached in memory between lock and unlock operations. .. nb thats
72
 
# what the transaction identity map provides
73
 
 
74
 
 
75
 
######################################################################
76
 
# branch objects
77
 
 
78
 
class Branch(object):
 
76
class Branch(controldir.ControlComponent):
79
77
    """Branch holding a history of revisions.
80
78
 
81
 
    base
82
 
        Base directory/url of the branch.
83
 
 
84
 
    hooks: An instance of BranchHooks.
 
79
    :ivar base:
 
80
        Base directory/url of the branch; using control_url and
 
81
        control_transport is more standardized.
 
82
    :ivar hooks: An instance of BranchHooks.
 
83
    :ivar _master_branch_cache: cached result of get_master_branch, see
 
84
        _clear_cached_state.
85
85
    """
86
86
    # this is really an instance variable - FIXME move it there
87
87
    # - RBC 20060112
88
88
    base = None
89
89
 
90
 
    def __init__(self, *ignored, **ignored_too):
 
90
    @property
 
91
    def control_transport(self):
 
92
        return self._transport
 
93
 
 
94
    @property
 
95
    def user_transport(self):
 
96
        return self.bzrdir.user_transport
 
97
 
 
98
    def __init__(self, possible_transports=None):
91
99
        self.tags = self._format.make_tags(self)
92
100
        self._revision_history_cache = None
93
101
        self._revision_id_to_revno_cache = None
94
102
        self._partial_revision_id_to_revno_cache = {}
95
103
        self._partial_revision_history_cache = []
 
104
        self._tags_bytes = None
96
105
        self._last_revision_info_cache = None
 
106
        self._master_branch_cache = None
97
107
        self._merge_sorted_revisions_cache = None
98
 
        self._open_hook()
 
108
        self._open_hook(possible_transports)
99
109
        hooks = Branch.hooks['open']
100
110
        for hook in hooks:
101
111
            hook(self)
102
112
 
103
 
    def _open_hook(self):
 
113
    def _open_hook(self, possible_transports):
104
114
        """Called by init to allow simpler extension of the base class."""
105
115
 
106
 
    def _activate_fallback_location(self, url):
 
116
    def _activate_fallback_location(self, url, possible_transports):
107
117
        """Activate the branch/repository from url as a fallback repository."""
108
 
        repo = self._get_fallback_repository(url)
 
118
        for existing_fallback_repo in self.repository._fallback_repositories:
 
119
            if existing_fallback_repo.user_url == url:
 
120
                # This fallback is already configured.  This probably only
 
121
                # happens because ControlDir.sprout is a horrible mess.  To avoid
 
122
                # confusing _unstack we don't add this a second time.
 
123
                mutter('duplicate activation of fallback %r on %r', url, self)
 
124
                return
 
125
        repo = self._get_fallback_repository(url, possible_transports)
109
126
        if repo.has_same_location(self.repository):
110
 
            raise errors.UnstackableLocationError(self.base, url)
 
127
            raise errors.UnstackableLocationError(self.user_url, url)
111
128
        self.repository.add_fallback_repository(repo)
112
129
 
113
130
    def break_lock(self):
165
182
        For instance, if the branch is at URL/.bzr/branch,
166
183
        Branch.open(URL) -> a Branch instance.
167
184
        """
168
 
        control = bzrdir.BzrDir.open(base, _unsupported,
169
 
                                     possible_transports=possible_transports)
170
 
        return control.open_branch(unsupported=_unsupported)
 
185
        control = controldir.ControlDir.open(base,
 
186
            possible_transports=possible_transports, _unsupported=_unsupported)
 
187
        return control.open_branch(unsupported=_unsupported,
 
188
            possible_transports=possible_transports)
171
189
 
172
190
    @staticmethod
173
 
    def open_from_transport(transport, name=None, _unsupported=False):
 
191
    def open_from_transport(transport, name=None, _unsupported=False,
 
192
            possible_transports=None):
174
193
        """Open the branch rooted at transport"""
175
 
        control = bzrdir.BzrDir.open_from_transport(transport, _unsupported)
176
 
        return control.open_branch(name=name, unsupported=_unsupported)
 
194
        control = controldir.ControlDir.open_from_transport(transport, _unsupported)
 
195
        return control.open_branch(name=name, unsupported=_unsupported,
 
196
            possible_transports=possible_transports)
177
197
 
178
198
    @staticmethod
179
199
    def open_containing(url, possible_transports=None):
187
207
        format, UnknownFormatError or UnsupportedFormatError are raised.
188
208
        If there is one, it is returned, along with the unused portion of url.
189
209
        """
190
 
        control, relpath = bzrdir.BzrDir.open_containing(url,
 
210
        control, relpath = controldir.ControlDir.open_containing(url,
191
211
                                                         possible_transports)
192
 
        return control.open_branch(), relpath
 
212
        branch = control.open_branch(possible_transports=possible_transports)
 
213
        return (branch, relpath)
193
214
 
194
215
    def _push_should_merge_tags(self):
195
216
        """Should _basic_push merge this branch's tags into the target?
200
221
        return self.supports_tags() and self.tags.get_tag_dict()
201
222
 
202
223
    def get_config(self):
203
 
        return BranchConfig(self)
 
224
        """Get a bzrlib.config.BranchConfig for this Branch.
 
225
 
 
226
        This can then be used to get and set configuration options for the
 
227
        branch.
 
228
 
 
229
        :return: A bzrlib.config.BranchConfig.
 
230
        """
 
231
        return _mod_config.BranchConfig(self)
 
232
 
 
233
    def get_config_stack(self):
 
234
        """Get a bzrlib.config.BranchStack for this Branch.
 
235
 
 
236
        This can then be used to get and set configuration options for the
 
237
        branch.
 
238
 
 
239
        :return: A bzrlib.config.BranchStack.
 
240
        """
 
241
        return _mod_config.BranchStack(self)
204
242
 
205
243
    def _get_config(self):
206
244
        """Get the concrete config for just the config in this branch.
214
252
        """
215
253
        raise NotImplementedError(self._get_config)
216
254
 
217
 
    def _get_fallback_repository(self, url):
 
255
    def store_uncommitted(self, creator):
 
256
        """Store uncommitted changes from a ShelfCreator.
 
257
 
 
258
        :param creator: The ShelfCreator containing uncommitted changes, or
 
259
            None to delete any stored changes.
 
260
        :raises: ChangesAlreadyStored if the branch already has changes.
 
261
        """
 
262
        raise NotImplementedError(self.store_uncommitted)
 
263
 
 
264
    def get_unshelver(self, tree):
 
265
        """Return a shelf.Unshelver for this branch and tree.
 
266
 
 
267
        :param tree: The tree to use to construct the Unshelver.
 
268
        :return: an Unshelver or None if no changes are stored.
 
269
        """
 
270
        raise NotImplementedError(self.get_unshelver)
 
271
 
 
272
    def _get_fallback_repository(self, url, possible_transports):
218
273
        """Get the repository we fallback to at url."""
219
274
        url = urlutils.join(self.base, url)
220
 
        a_branch = Branch.open(url,
221
 
            possible_transports=[self.bzrdir.root_transport])
 
275
        a_branch = Branch.open(url, possible_transports=possible_transports)
222
276
        return a_branch.repository
223
277
 
 
278
    @needs_read_lock
224
279
    def _get_tags_bytes(self):
225
280
        """Get the bytes of a serialised tags dict.
226
281
 
233
288
        :return: The bytes of the tags file.
234
289
        :seealso: Branch._set_tags_bytes.
235
290
        """
236
 
        return self._transport.get_bytes('tags')
 
291
        if self._tags_bytes is None:
 
292
            self._tags_bytes = self._transport.get_bytes('tags')
 
293
        return self._tags_bytes
237
294
 
238
295
    def _get_nick(self, local=False, possible_transports=None):
239
296
        config = self.get_config()
241
298
        if not local and not config.has_explicit_nickname():
242
299
            try:
243
300
                master = self.get_master_branch(possible_transports)
 
301
                if master and self.user_url == master.user_url:
 
302
                    raise errors.RecursiveBind(self.user_url)
244
303
                if master is not None:
245
304
                    # return the master branch value
246
305
                    return master.nick
 
306
            except errors.RecursiveBind, e:
 
307
                raise e
247
308
            except errors.BzrError, e:
248
309
                # Silently fall back to local implicit nick if the master is
249
310
                # unavailable
286
347
        new_history.reverse()
287
348
        return new_history
288
349
 
289
 
    def lock_write(self):
 
350
    def lock_write(self, token=None):
 
351
        """Lock the branch for write operations.
 
352
 
 
353
        :param token: A token to permit reacquiring a previously held and
 
354
            preserved lock.
 
355
        :return: A BranchWriteLockResult.
 
356
        """
290
357
        raise NotImplementedError(self.lock_write)
291
358
 
292
359
    def lock_read(self):
 
360
        """Lock the branch for read operations.
 
361
 
 
362
        :return: A bzrlib.lock.LogicalLockResult.
 
363
        """
293
364
        raise NotImplementedError(self.lock_read)
294
365
 
295
366
    def unlock(self):
416
487
            after. If None, the rest of history is included.
417
488
        :param stop_rule: if stop_revision_id is not None, the precise rule
418
489
            to use for termination:
 
490
 
419
491
            * 'exclude' - leave the stop revision out of the result (default)
420
492
            * 'include' - the stop revision is the last item in the result
421
493
            * 'with-merges' - include the stop revision and all of its
422
494
              merged revisions in the result
 
495
            * 'with-merges-without-common-ancestry' - filter out revisions 
 
496
              that are in both ancestries
423
497
        :param direction: either 'reverse' or 'forward':
 
498
 
424
499
            * reverse means return the start_revision_id first, i.e.
425
500
              start at the most recent revision and go backwards in history
426
501
            * forward returns tuples in the opposite order to reverse.
456
531
            stop_revision_id, stop_rule)
457
532
        # Make sure we don't return revisions that are not part of the
458
533
        # start_revision_id ancestry.
459
 
        filtered = self._filter_non_ancestors(filtered)
 
534
        filtered = self._filter_start_non_ancestors(filtered)
460
535
        if direction == 'reverse':
461
536
            return filtered
462
537
        if direction == 'forward':
470
545
        rev_iter = iter(merge_sorted_revisions)
471
546
        if start_revision_id is not None:
472
547
            for node in rev_iter:
473
 
                rev_id = node.key[-1]
 
548
                rev_id = node.key
474
549
                if rev_id != start_revision_id:
475
550
                    continue
476
551
                else:
477
552
                    # The decision to include the start or not
478
553
                    # depends on the stop_rule if a stop is provided
479
554
                    # so pop this node back into the iterator
480
 
                    rev_iter = chain(iter([node]), rev_iter)
 
555
                    rev_iter = itertools.chain(iter([node]), rev_iter)
481
556
                    break
482
557
        if stop_revision_id is None:
483
558
            # Yield everything
484
559
            for node in rev_iter:
485
 
                rev_id = node.key[-1]
 
560
                rev_id = node.key
486
561
                yield (rev_id, node.merge_depth, node.revno,
487
562
                       node.end_of_merge)
488
563
        elif stop_rule == 'exclude':
489
564
            for node in rev_iter:
490
 
                rev_id = node.key[-1]
 
565
                rev_id = node.key
491
566
                if rev_id == stop_revision_id:
492
567
                    return
493
568
                yield (rev_id, node.merge_depth, node.revno,
494
569
                       node.end_of_merge)
495
570
        elif stop_rule == 'include':
496
571
            for node in rev_iter:
497
 
                rev_id = node.key[-1]
 
572
                rev_id = node.key
498
573
                yield (rev_id, node.merge_depth, node.revno,
499
574
                       node.end_of_merge)
500
575
                if rev_id == stop_revision_id:
501
576
                    return
 
577
        elif stop_rule == 'with-merges-without-common-ancestry':
 
578
            # We want to exclude all revisions that are already part of the
 
579
            # stop_revision_id ancestry.
 
580
            graph = self.repository.get_graph()
 
581
            ancestors = graph.find_unique_ancestors(start_revision_id,
 
582
                                                    [stop_revision_id])
 
583
            for node in rev_iter:
 
584
                rev_id = node.key
 
585
                if rev_id not in ancestors:
 
586
                    continue
 
587
                yield (rev_id, node.merge_depth, node.revno,
 
588
                       node.end_of_merge)
502
589
        elif stop_rule == 'with-merges':
503
590
            stop_rev = self.repository.get_revision(stop_revision_id)
504
591
            if stop_rev.parent_ids:
510
597
            reached_stop_revision_id = False
511
598
            revision_id_whitelist = []
512
599
            for node in rev_iter:
513
 
                rev_id = node.key[-1]
 
600
                rev_id = node.key
514
601
                if rev_id == left_parent:
515
602
                    # reached the left parent after the stop_revision
516
603
                    return
527
614
        else:
528
615
            raise ValueError('invalid stop_rule %r' % stop_rule)
529
616
 
530
 
    def _filter_non_ancestors(self, rev_iter):
 
617
    def _filter_start_non_ancestors(self, rev_iter):
531
618
        # If we started from a dotted revno, we want to consider it as a tip
532
619
        # and don't want to yield revisions that are not part of its
533
620
        # ancestry. Given the order guaranteed by the merge sort, we will see
594
681
        :param other: The branch to bind to
595
682
        :type other: Branch
596
683
        """
597
 
        raise errors.UpgradeRequired(self.base)
 
684
        raise errors.UpgradeRequired(self.user_url)
 
685
 
 
686
    def get_append_revisions_only(self):
 
687
        """Whether it is only possible to append revisions to the history.
 
688
        """
 
689
        if not self._format.supports_set_append_revisions_only():
 
690
            return False
 
691
        return self.get_config_stack().get('append_revisions_only')
598
692
 
599
693
    def set_append_revisions_only(self, enabled):
600
694
        if not self._format.supports_set_append_revisions_only():
601
 
            raise errors.UpgradeRequired(self.base)
602
 
        if enabled:
603
 
            value = 'True'
604
 
        else:
605
 
            value = 'False'
606
 
        self.get_config().set_user_option('append_revisions_only', value,
607
 
            warn_masked=True)
 
695
            raise errors.UpgradeRequired(self.user_url)
 
696
        self.get_config_stack().set('append_revisions_only', enabled)
608
697
 
609
698
    def set_reference_info(self, file_id, tree_path, branch_location):
610
699
        """Set the branch location to use for a tree reference."""
615
704
        raise errors.UnsupportedOperation(self.get_reference_info, self)
616
705
 
617
706
    @needs_write_lock
618
 
    def fetch(self, from_branch, last_revision=None, pb=None):
 
707
    def fetch(self, from_branch, last_revision=None, limit=None):
619
708
        """Copy revisions from from_branch into this branch.
620
709
 
621
710
        :param from_branch: Where to copy from.
622
711
        :param last_revision: What revision to stop at (None for at the end
623
712
                              of the branch.
624
 
        :param pb: An optional progress bar to use.
 
713
        :param limit: Optional rough limit of revisions to fetch
625
714
        :return: None
626
715
        """
627
 
        if self.base == from_branch.base:
628
 
            return (0, [])
629
 
        if pb is not None:
630
 
            symbol_versioning.warn(
631
 
                symbol_versioning.deprecated_in((1, 14, 0))
632
 
                % "pb parameter to fetch()")
633
 
        from_branch.lock_read()
634
 
        try:
635
 
            if last_revision is None:
636
 
                last_revision = from_branch.last_revision()
637
 
                last_revision = _mod_revision.ensure_null(last_revision)
638
 
            return self.repository.fetch(from_branch.repository,
639
 
                                         revision_id=last_revision,
640
 
                                         pb=pb)
641
 
        finally:
642
 
            from_branch.unlock()
 
716
        return InterBranch.get(from_branch, self).fetch(last_revision, limit=limit)
643
717
 
644
718
    def get_bound_location(self):
645
719
        """Return the URL of the branch we are bound to.
652
726
    def get_old_bound_location(self):
653
727
        """Return the URL of the branch we used to be bound to
654
728
        """
655
 
        raise errors.UpgradeRequired(self.base)
 
729
        raise errors.UpgradeRequired(self.user_url)
656
730
 
657
 
    def get_commit_builder(self, parents, config=None, timestamp=None,
 
731
    def get_commit_builder(self, parents, config_stack=None, timestamp=None,
658
732
                           timezone=None, committer=None, revprops=None,
659
 
                           revision_id=None):
 
733
                           revision_id=None, lossy=False):
660
734
        """Obtain a CommitBuilder for this branch.
661
735
 
662
736
        :param parents: Revision ids of the parents of the new revision.
666
740
        :param committer: Optional committer to set for commit.
667
741
        :param revprops: Optional dictionary of revision properties.
668
742
        :param revision_id: Optional revision id.
 
743
        :param lossy: Whether to discard data that can not be natively
 
744
            represented, when pushing to a foreign VCS 
669
745
        """
670
746
 
671
 
        if config is None:
672
 
            config = self.get_config()
 
747
        if config_stack is None:
 
748
            config_stack = self.get_config_stack()
673
749
 
674
 
        return self.repository.get_commit_builder(self, parents, config,
675
 
            timestamp, timezone, committer, revprops, revision_id)
 
750
        return self.repository.get_commit_builder(self, parents, config_stack,
 
751
            timestamp, timezone, committer, revprops, revision_id,
 
752
            lossy)
676
753
 
677
754
    def get_master_branch(self, possible_transports=None):
678
755
        """Return the branch we are bound to.
681
758
        """
682
759
        return None
683
760
 
 
761
    @deprecated_method(deprecated_in((2, 5, 0)))
684
762
    def get_revision_delta(self, revno):
685
763
        """Return the delta for one revision.
686
764
 
687
765
        The delta is relative to its mainline predecessor, or the
688
766
        empty tree for revision 1.
689
767
        """
690
 
        rh = self.revision_history()
691
 
        if not (1 <= revno <= len(rh)):
 
768
        try:
 
769
            revid = self.get_rev_id(revno)
 
770
        except errors.NoSuchRevision:
692
771
            raise errors.InvalidRevisionNumber(revno)
693
 
        return self.repository.get_revision_delta(rh[revno-1])
 
772
        return self.repository.get_revision_delta(revid)
694
773
 
695
774
    def get_stacked_on_url(self):
696
775
        """Get the URL this branch is stacked against.
705
784
        """Print `file` to stdout."""
706
785
        raise NotImplementedError(self.print_file)
707
786
 
708
 
    def set_revision_history(self, rev_history):
709
 
        raise NotImplementedError(self.set_revision_history)
 
787
    @needs_write_lock
 
788
    def set_last_revision_info(self, revno, revision_id):
 
789
        """Set the last revision of this branch.
 
790
 
 
791
        The caller is responsible for checking that the revno is correct
 
792
        for this revision id.
 
793
 
 
794
        It may be possible to set the branch last revision to an id not
 
795
        present in the repository.  However, branches can also be
 
796
        configured to check constraints on history, in which case this may not
 
797
        be permitted.
 
798
        """
 
799
        raise NotImplementedError(self.set_last_revision_info)
 
800
 
 
801
    @needs_write_lock
 
802
    def generate_revision_history(self, revision_id, last_rev=None,
 
803
                                  other_branch=None):
 
804
        """See Branch.generate_revision_history"""
 
805
        graph = self.repository.get_graph()
 
806
        (last_revno, last_revid) = self.last_revision_info()
 
807
        known_revision_ids = [
 
808
            (last_revid, last_revno),
 
809
            (_mod_revision.NULL_REVISION, 0),
 
810
            ]
 
811
        if last_rev is not None:
 
812
            if not graph.is_ancestor(last_rev, revision_id):
 
813
                # our previous tip is not merged into stop_revision
 
814
                raise errors.DivergedBranches(self, other_branch)
 
815
        revno = graph.find_distance_to_null(revision_id, known_revision_ids)
 
816
        self.set_last_revision_info(revno, revision_id)
710
817
 
711
818
    @needs_write_lock
712
819
    def set_parent(self, url):
736
843
            stacking.
737
844
        """
738
845
        if not self._format.supports_stacking():
739
 
            raise errors.UnstackableBranchFormat(self._format, self.base)
 
846
            raise errors.UnstackableBranchFormat(self._format, self.user_url)
740
847
        # XXX: Changing from one fallback repository to another does not check
741
848
        # that all the data you need is present in the new fallback.
742
849
        # Possibly it should.
749
856
                return
750
857
            self._unstack()
751
858
        else:
752
 
            self._activate_fallback_location(url)
 
859
            self._activate_fallback_location(url,
 
860
                possible_transports=[self.bzrdir.root_transport])
753
861
        # write this out after the repository is stacked to avoid setting a
754
862
        # stacked config that doesn't work.
755
863
        self._set_config_location('stacked_on_location', url)
756
864
 
757
865
    def _unstack(self):
758
866
        """Change a branch to be unstacked, copying data as needed.
759
 
        
 
867
 
760
868
        Don't call this directly, use set_stacked_on_url(None).
761
869
        """
762
870
        pb = ui.ui_factory.nested_progress_bar()
763
871
        try:
764
 
            pb.update("Unstacking")
 
872
            pb.update(gettext("Unstacking"))
765
873
            # The basic approach here is to fetch the tip of the branch,
766
874
            # including all available ghosts, from the existing stacked
767
875
            # repository into a new repository object without the fallbacks. 
771
879
            old_repository = self.repository
772
880
            if len(old_repository._fallback_repositories) != 1:
773
881
                raise AssertionError("can't cope with fallback repositories "
774
 
                    "of %r" % (self.repository,))
775
 
            # unlock it, including unlocking the fallback
 
882
                    "of %r (fallbacks: %r)" % (old_repository,
 
883
                        old_repository._fallback_repositories))
 
884
            # Open the new repository object.
 
885
            # Repositories don't offer an interface to remove fallback
 
886
            # repositories today; take the conceptually simpler option and just
 
887
            # reopen it.  We reopen it starting from the URL so that we
 
888
            # get a separate connection for RemoteRepositories and can
 
889
            # stream from one of them to the other.  This does mean doing
 
890
            # separate SSH connection setup, but unstacking is not a
 
891
            # common operation so it's tolerable.
 
892
            new_bzrdir = controldir.ControlDir.open(
 
893
                self.bzrdir.root_transport.base)
 
894
            new_repository = new_bzrdir.find_repository()
 
895
            if new_repository._fallback_repositories:
 
896
                raise AssertionError("didn't expect %r to have "
 
897
                    "fallback_repositories"
 
898
                    % (self.repository,))
 
899
            # Replace self.repository with the new repository.
 
900
            # Do our best to transfer the lock state (i.e. lock-tokens and
 
901
            # lock count) of self.repository to the new repository.
 
902
            lock_token = old_repository.lock_write().repository_token
 
903
            self.repository = new_repository
 
904
            if isinstance(self, remote.RemoteBranch):
 
905
                # Remote branches can have a second reference to the old
 
906
                # repository that need to be replaced.
 
907
                if self._real_branch is not None:
 
908
                    self._real_branch.repository = new_repository
 
909
            self.repository.lock_write(token=lock_token)
 
910
            if lock_token is not None:
 
911
                old_repository.leave_lock_in_place()
776
912
            old_repository.unlock()
 
913
            if lock_token is not None:
 
914
                # XXX: self.repository.leave_lock_in_place() before this
 
915
                # function will not be preserved.  Fortunately that doesn't
 
916
                # affect the current default format (2a), and would be a
 
917
                # corner-case anyway.
 
918
                #  - Andrew Bennetts, 2010/06/30
 
919
                self.repository.dont_leave_lock_in_place()
 
920
            old_lock_count = 0
 
921
            while True:
 
922
                try:
 
923
                    old_repository.unlock()
 
924
                except errors.LockNotHeld:
 
925
                    break
 
926
                old_lock_count += 1
 
927
            if old_lock_count == 0:
 
928
                raise AssertionError(
 
929
                    'old_repository should have been locked at least once.')
 
930
            for i in range(old_lock_count-1):
 
931
                self.repository.lock_write()
 
932
            # Fetch from the old repository into the new.
777
933
            old_repository.lock_read()
778
934
            try:
779
 
                # Repositories don't offer an interface to remove fallback
780
 
                # repositories today; take the conceptually simpler option and just
781
 
                # reopen it.  We reopen it starting from the URL so that we
782
 
                # get a separate connection for RemoteRepositories and can
783
 
                # stream from one of them to the other.  This does mean doing
784
 
                # separate SSH connection setup, but unstacking is not a
785
 
                # common operation so it's tolerable.
786
 
                new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
787
 
                new_repository = new_bzrdir.find_repository()
788
 
                self.repository = new_repository
789
 
                if self.repository._fallback_repositories:
790
 
                    raise AssertionError("didn't expect %r to have "
791
 
                        "fallback_repositories"
792
 
                        % (self.repository,))
793
 
                # this is not paired with an unlock because it's just restoring
794
 
                # the previous state; the lock's released when set_stacked_on_url
795
 
                # returns
796
 
                self.repository.lock_write()
797
935
                # XXX: If you unstack a branch while it has a working tree
798
936
                # with a pending merge, the pending-merged revisions will no
799
937
                # longer be present.  You can (probably) revert and remerge.
800
 
                #
801
 
                # XXX: This only fetches up to the tip of the repository; it
802
 
                # doesn't bring across any tags.  That's fairly consistent
803
 
                # with how branch works, but perhaps not ideal.
804
 
                self.repository.fetch(old_repository,
805
 
                    revision_id=self.last_revision(),
806
 
                    find_ghosts=True)
 
938
                try:
 
939
                    tags_to_fetch = set(self.tags.get_reverse_tag_dict())
 
940
                except errors.TagsNotSupported:
 
941
                    tags_to_fetch = set()
 
942
                fetch_spec = vf_search.NotInOtherForRevs(self.repository,
 
943
                    old_repository, required_ids=[self.last_revision()],
 
944
                    if_present_ids=tags_to_fetch, find_ghosts=True).execute()
 
945
                self.repository.fetch(old_repository, fetch_spec=fetch_spec)
807
946
            finally:
808
947
                old_repository.unlock()
809
948
        finally:
814
953
 
815
954
        :seealso: Branch._get_tags_bytes.
816
955
        """
817
 
        return _run_with_write_locked_target(self, self._transport.put_bytes,
818
 
            'tags', bytes)
 
956
        op = cleanup.OperationWithCleanups(self._set_tags_bytes_locked)
 
957
        op.add_cleanup(self.lock_write().unlock)
 
958
        return op.run_simple(bytes)
 
959
 
 
960
    def _set_tags_bytes_locked(self, bytes):
 
961
        self._tags_bytes = bytes
 
962
        return self._transport.put_bytes('tags', bytes)
819
963
 
820
964
    def _cache_revision_history(self, rev_history):
821
965
        """Set the cached revision history to rev_history.
842
986
        This means the next call to revision_history will need to call
843
987
        _gen_revision_history.
844
988
 
845
 
        This API is semi-public; it only for use by subclasses, all other code
846
 
        should consider it to be private.
 
989
        This API is semi-public; it is only for use by subclasses, all other
 
990
        code should consider it to be private.
847
991
        """
848
992
        self._revision_history_cache = None
849
993
        self._revision_id_to_revno_cache = None
850
994
        self._last_revision_info_cache = None
 
995
        self._master_branch_cache = None
851
996
        self._merge_sorted_revisions_cache = None
852
997
        self._partial_revision_history_cache = []
853
998
        self._partial_revision_id_to_revno_cache = {}
 
999
        self._tags_bytes = None
854
1000
 
855
1001
    def _gen_revision_history(self):
856
1002
        """Return sequence of revision hashes on to this branch.
867
1013
        """
868
1014
        raise NotImplementedError(self._gen_revision_history)
869
1015
 
870
 
    @needs_read_lock
871
 
    def revision_history(self):
872
 
        """Return sequence of revision ids on this branch.
873
 
 
874
 
        This method will cache the revision history for as long as it is safe to
875
 
        do so.
876
 
        """
 
1016
    def _revision_history(self):
877
1017
        if 'evil' in debug.debug_flags:
878
1018
            mutter_callsite(3, "revision_history scales with history.")
879
1019
        if self._revision_history_cache is not None:
893
1033
 
894
1034
    def unbind(self):
895
1035
        """Older format branches cannot bind or unbind."""
896
 
        raise errors.UpgradeRequired(self.base)
 
1036
        raise errors.UpgradeRequired(self.user_url)
897
1037
 
898
1038
    def last_revision(self):
899
1039
        """Return last revision id, or NULL_REVISION."""
906
1046
        :return: A tuple (revno, revision_id).
907
1047
        """
908
1048
        if self._last_revision_info_cache is None:
909
 
            self._last_revision_info_cache = self._last_revision_info()
 
1049
            self._last_revision_info_cache = self._read_last_revision_info()
910
1050
        return self._last_revision_info_cache
911
1051
 
912
 
    def _last_revision_info(self):
913
 
        rh = self.revision_history()
914
 
        revno = len(rh)
915
 
        if revno:
916
 
            return (revno, rh[-1])
917
 
        else:
918
 
            return (0, _mod_revision.NULL_REVISION)
919
 
 
920
 
    @deprecated_method(deprecated_in((1, 6, 0)))
921
 
    def missing_revisions(self, other, stop_revision=None):
922
 
        """Return a list of new revisions that would perfectly fit.
923
 
 
924
 
        If self and other have not diverged, return a list of the revisions
925
 
        present in other, but missing from self.
926
 
        """
927
 
        self_history = self.revision_history()
928
 
        self_len = len(self_history)
929
 
        other_history = other.revision_history()
930
 
        other_len = len(other_history)
931
 
        common_index = min(self_len, other_len) -1
932
 
        if common_index >= 0 and \
933
 
            self_history[common_index] != other_history[common_index]:
934
 
            raise errors.DivergedBranches(self, other)
935
 
 
936
 
        if stop_revision is None:
937
 
            stop_revision = other_len
938
 
        else:
939
 
            if stop_revision > other_len:
940
 
                raise errors.NoSuchRevision(self, stop_revision)
941
 
        return other_history[self_len:stop_revision]
942
 
 
943
 
    @needs_write_lock
944
 
    def update_revisions(self, other, stop_revision=None, overwrite=False,
945
 
                         graph=None):
946
 
        """Pull in new perfect-fit revisions.
947
 
 
948
 
        :param other: Another Branch to pull from
949
 
        :param stop_revision: Updated until the given revision
950
 
        :param overwrite: Always set the branch pointer, rather than checking
951
 
            to see if it is a proper descendant.
952
 
        :param graph: A Graph object that can be used to query history
953
 
            information. This can be None.
954
 
        :return: None
955
 
        """
956
 
        return InterBranch.get(other, self).update_revisions(stop_revision,
957
 
            overwrite, graph)
958
 
 
959
 
    def import_last_revision_info(self, source_repo, revno, revid):
 
1052
    def _read_last_revision_info(self):
 
1053
        raise NotImplementedError(self._read_last_revision_info)
 
1054
 
 
1055
    def import_last_revision_info_and_tags(self, source, revno, revid,
 
1056
                                           lossy=False):
960
1057
        """Set the last revision info, importing from another repo if necessary.
961
1058
 
962
1059
        This is used by the bound branch code to upload a revision to
963
1060
        the master branch first before updating the tip of the local branch.
 
1061
        Revisions referenced by source's tags are also transferred.
964
1062
 
965
 
        :param source_repo: Source repository to optionally fetch from
 
1063
        :param source: Source branch to optionally fetch from
966
1064
        :param revno: Revision number of the new tip
967
1065
        :param revid: Revision id of the new tip
 
1066
        :param lossy: Whether to discard metadata that can not be
 
1067
            natively represented
 
1068
        :return: Tuple with the new revision number and revision id
 
1069
            (should only be different from the arguments when lossy=True)
968
1070
        """
969
 
        if not self.repository.has_same_location(source_repo):
970
 
            self.repository.fetch(source_repo, revision_id=revid)
 
1071
        if not self.repository.has_same_location(source.repository):
 
1072
            self.fetch(source, revid)
971
1073
        self.set_last_revision_info(revno, revid)
 
1074
        return (revno, revid)
972
1075
 
973
1076
    def revision_id_to_revno(self, revision_id):
974
1077
        """Given a revision id, return its revno"""
975
1078
        if _mod_revision.is_null(revision_id):
976
1079
            return 0
977
 
        history = self.revision_history()
 
1080
        history = self._revision_history()
978
1081
        try:
979
1082
            return history.index(revision_id) + 1
980
1083
        except ValueError:
995
1098
            self._extend_partial_history(distance_from_last)
996
1099
        return self._partial_revision_history_cache[distance_from_last]
997
1100
 
998
 
    @needs_write_lock
999
1101
    def pull(self, source, overwrite=False, stop_revision=None,
1000
1102
             possible_transports=None, *args, **kwargs):
1001
1103
        """Mirror source into this branch.
1008
1110
            stop_revision=stop_revision,
1009
1111
            possible_transports=possible_transports, *args, **kwargs)
1010
1112
 
1011
 
    def push(self, target, overwrite=False, stop_revision=None, *args,
1012
 
        **kwargs):
 
1113
    def push(self, target, overwrite=False, stop_revision=None, lossy=False,
 
1114
            *args, **kwargs):
1013
1115
        """Mirror this branch into target.
1014
1116
 
1015
1117
        This branch is considered to be 'local', having low latency.
1016
1118
        """
1017
1119
        return InterBranch.get(self, target).push(overwrite, stop_revision,
1018
 
            *args, **kwargs)
1019
 
 
1020
 
    def lossy_push(self, target, stop_revision=None):
1021
 
        """Push deltas into another branch.
1022
 
 
1023
 
        :note: This does not, like push, retain the revision ids from 
1024
 
            the source branch and will, rather than adding bzr-specific 
1025
 
            metadata, push only those semantics of the revision that can be 
1026
 
            natively represented by this branch' VCS.
1027
 
 
1028
 
        :param target: Target branch
1029
 
        :param stop_revision: Revision to push, defaults to last revision.
1030
 
        :return: BranchPushResult with an extra member revidmap: 
1031
 
            A dictionary mapping revision ids from the target branch 
1032
 
            to new revision ids in the target branch, for each 
1033
 
            revision that was pushed.
1034
 
        """
1035
 
        inter = InterBranch.get(self, target)
1036
 
        lossy_push = getattr(inter, "lossy_push", None)
1037
 
        if lossy_push is None:
1038
 
            raise errors.LossyPushToSameVCS(self, target)
1039
 
        return lossy_push(stop_revision)
 
1120
            lossy, *args, **kwargs)
1040
1121
 
1041
1122
    def basis_tree(self):
1042
1123
        """Return `Tree` object for last revision."""
1059
1140
        try:
1060
1141
            return urlutils.join(self.base[:-1], parent)
1061
1142
        except errors.InvalidURLJoin, e:
1062
 
            raise errors.InaccessibleParent(parent, self.base)
 
1143
            raise errors.InaccessibleParent(parent, self.user_url)
1063
1144
 
1064
1145
    def _get_parent_location(self):
1065
1146
        raise NotImplementedError(self._get_parent_location)
1067
1148
    def _set_config_location(self, name, url, config=None,
1068
1149
                             make_relative=False):
1069
1150
        if config is None:
1070
 
            config = self.get_config()
 
1151
            config = self.get_config_stack()
1071
1152
        if url is None:
1072
1153
            url = ''
1073
1154
        elif make_relative:
1074
1155
            url = urlutils.relative_url(self.base, url)
1075
 
        config.set_user_option(name, url, warn_masked=True)
 
1156
        config.set(name, url)
1076
1157
 
1077
1158
    def _get_config_location(self, name, config=None):
1078
1159
        if config is None:
1079
 
            config = self.get_config()
1080
 
        location = config.get_user_option(name)
 
1160
            config = self.get_config_stack()
 
1161
        location = config.get(name)
1081
1162
        if location == '':
1082
1163
            location = None
1083
1164
        return location
1084
1165
 
1085
1166
    def get_child_submit_format(self):
1086
1167
        """Return the preferred format of submissions to this branch."""
1087
 
        return self.get_config().get_user_option("child_submit_format")
 
1168
        return self.get_config_stack().get('child_submit_format')
1088
1169
 
1089
1170
    def get_submit_branch(self):
1090
1171
        """Return the submit location of the branch.
1093
1174
        pattern is that the user can override it by specifying a
1094
1175
        location.
1095
1176
        """
1096
 
        return self.get_config().get_user_option('submit_branch')
 
1177
        return self.get_config_stack().get('submit_branch')
1097
1178
 
1098
1179
    def set_submit_branch(self, location):
1099
1180
        """Return the submit location of the branch.
1102
1183
        pattern is that the user can override it by specifying a
1103
1184
        location.
1104
1185
        """
1105
 
        self.get_config().set_user_option('submit_branch', location,
1106
 
            warn_masked=True)
 
1186
        self.get_config_stack().set('submit_branch', location)
1107
1187
 
1108
1188
    def get_public_branch(self):
1109
1189
        """Return the public location of the branch.
1122
1202
        self._set_config_location('public_branch', location)
1123
1203
 
1124
1204
    def get_push_location(self):
1125
 
        """Return the None or the location to push this branch to."""
1126
 
        push_loc = self.get_config().get_user_option('push_location')
1127
 
        return push_loc
 
1205
        """Return None or the location to push this branch to."""
 
1206
        return self.get_config_stack().get('push_location')
1128
1207
 
1129
1208
    def set_push_location(self, location):
1130
1209
        """Set a new push location for this branch."""
1197
1276
        return result
1198
1277
 
1199
1278
    @needs_read_lock
1200
 
    def sprout(self, to_bzrdir, revision_id=None, repository_policy=None):
 
1279
    def sprout(self, to_bzrdir, revision_id=None, repository_policy=None,
 
1280
            repository=None):
1201
1281
        """Create a new line of development from the branch, into to_bzrdir.
1202
1282
 
1203
1283
        to_bzrdir controls the branch format.
1208
1288
        if (repository_policy is not None and
1209
1289
            repository_policy.requires_stacking()):
1210
1290
            to_bzrdir._format.require_stacking(_skip_repo=True)
1211
 
        result = to_bzrdir.create_branch()
 
1291
        result = to_bzrdir.create_branch(repository=repository)
1212
1292
        result.lock_write()
1213
1293
        try:
1214
1294
            if repository_policy is not None:
1215
1295
                repository_policy.configure_branch(result)
1216
1296
            self.copy_content_into(result, revision_id=revision_id)
1217
 
            result.set_parent(self.bzrdir.root_transport.base)
 
1297
            master_url = self.get_bound_location()
 
1298
            if master_url is None:
 
1299
                result.set_parent(self.bzrdir.root_transport.base)
 
1300
            else:
 
1301
                result.set_parent(master_url)
1218
1302
        finally:
1219
1303
            result.unlock()
1220
1304
        return result
1244
1328
                revno = 1
1245
1329
        destination.set_last_revision_info(revno, revision_id)
1246
1330
 
1247
 
    @needs_read_lock
1248
1331
    def copy_content_into(self, destination, revision_id=None):
1249
1332
        """Copy the content of self into destination.
1250
1333
 
1251
1334
        revision_id: if not None, the revision history in the new branch will
1252
1335
                     be truncated to end with revision_id.
1253
1336
        """
1254
 
        self.update_references(destination)
1255
 
        self._synchronize_history(destination, revision_id)
1256
 
        try:
1257
 
            parent = self.get_parent()
1258
 
        except errors.InaccessibleParent, e:
1259
 
            mutter('parent was not accessible to copy: %s', e)
1260
 
        else:
1261
 
            if parent:
1262
 
                destination.set_parent(parent)
1263
 
        if self._push_should_merge_tags():
1264
 
            self.tags.merge_to(destination.tags)
 
1337
        return InterBranch.get(self, destination).copy_content_into(
 
1338
            revision_id=revision_id)
1265
1339
 
1266
1340
    def update_references(self, target):
1267
1341
        if not getattr(self._format, 'supports_reference_locations', False):
1304
1378
        # TODO: We should probably also check that self.revision_history
1305
1379
        # matches the repository for older branch formats.
1306
1380
        # If looking for the code that cross-checks repository parents against
1307
 
        # the iter_reverse_revision_history output, that is now a repository
 
1381
        # the Graph.iter_lefthand_ancestry output, that is now a repository
1308
1382
        # specific check.
1309
1383
        return result
1310
1384
 
1311
 
    def _get_checkout_format(self):
 
1385
    def _get_checkout_format(self, lightweight=False):
1312
1386
        """Return the most suitable metadir for a checkout of this branch.
1313
1387
        Weaves are used if this branch's repository uses weaves.
1314
1388
        """
1315
 
        if isinstance(self.bzrdir, bzrdir.BzrDirPreSplitOut):
1316
 
            from bzrlib.repofmt import weaverepo
1317
 
            format = bzrdir.BzrDirMetaFormat1()
1318
 
            format.repository_format = weaverepo.RepositoryFormat7()
1319
 
        else:
1320
 
            format = self.repository.bzrdir.checkout_metadir()
1321
 
            format.set_branch_format(self._format)
 
1389
        format = self.repository.bzrdir.checkout_metadir()
 
1390
        format.set_branch_format(self._format)
1322
1391
        return format
1323
1392
 
1324
1393
    def create_clone_on_transport(self, to_transport, revision_id=None,
1325
 
        stacked_on=None, create_prefix=False, use_existing_dir=False):
 
1394
        stacked_on=None, create_prefix=False, use_existing_dir=False,
 
1395
        no_tree=None):
1326
1396
        """Create a clone of this branch and its bzrdir.
1327
1397
 
1328
1398
        :param to_transport: The transport to clone onto.
1335
1405
        """
1336
1406
        # XXX: Fix the bzrdir API to allow getting the branch back from the
1337
1407
        # clone call. Or something. 20090224 RBC/spiv.
 
1408
        # XXX: Should this perhaps clone colocated branches as well, 
 
1409
        # rather than just the default branch? 20100319 JRV
1338
1410
        if revision_id is None:
1339
1411
            revision_id = self.last_revision()
1340
1412
        dir_to = self.bzrdir.clone_on_transport(to_transport,
1341
1413
            revision_id=revision_id, stacked_on=stacked_on,
1342
 
            create_prefix=create_prefix, use_existing_dir=use_existing_dir)
 
1414
            create_prefix=create_prefix, use_existing_dir=use_existing_dir,
 
1415
            no_tree=no_tree)
1343
1416
        return dir_to.open_branch()
1344
1417
 
1345
1418
    def create_checkout(self, to_location, revision_id=None,
1350
1423
        :param to_location: The url to produce the checkout at
1351
1424
        :param revision_id: The revision to check out
1352
1425
        :param lightweight: If True, produce a lightweight checkout, otherwise,
1353
 
        produce a bound branch (heavyweight checkout)
 
1426
            produce a bound branch (heavyweight checkout)
1354
1427
        :param accelerator_tree: A tree which can be used for retrieving file
1355
1428
            contents more quickly than the revision tree, i.e. a workingtree.
1356
1429
            The revision tree will be used for cases where accelerator_tree's
1361
1434
        """
1362
1435
        t = transport.get_transport(to_location)
1363
1436
        t.ensure_base()
 
1437
        format = self._get_checkout_format(lightweight=lightweight)
 
1438
        try:
 
1439
            checkout = format.initialize_on_transport(t)
 
1440
        except errors.AlreadyControlDirError:
 
1441
            # It's fine if the control directory already exists,
 
1442
            # as long as there is no existing branch and working tree.
 
1443
            checkout = controldir.ControlDir.open_from_transport(t)
 
1444
            try:
 
1445
                checkout.open_branch()
 
1446
            except errors.NotBranchError:
 
1447
                pass
 
1448
            else:
 
1449
                raise errors.AlreadyControlDirError(t.base)
 
1450
            if checkout.control_transport.base == self.bzrdir.control_transport.base:
 
1451
                # When checking out to the same control directory,
 
1452
                # always create a lightweight checkout
 
1453
                lightweight = True
 
1454
 
1364
1455
        if lightweight:
1365
 
            format = self._get_checkout_format()
1366
 
            checkout = format.initialize_on_transport(t)
1367
 
            from_branch = BranchReferenceFormat().initialize(checkout, 
1368
 
                target_branch=self)
 
1456
            from_branch = checkout.set_branch_reference(target_branch=self)
1369
1457
        else:
1370
 
            format = self._get_checkout_format()
1371
 
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
1372
 
                to_location, force_new_tree=False, format=format)
1373
 
            checkout = checkout_branch.bzrdir
 
1458
            policy = checkout.determine_repository_policy()
 
1459
            repo = policy.acquire_repository()[0]
 
1460
            checkout_branch = checkout.create_branch()
1374
1461
            checkout_branch.bind(self)
1375
1462
            # pull up to the specified revision_id to set the initial
1376
1463
            # branch tip correctly, and seed it with history.
1377
1464
            checkout_branch.pull(self, stop_revision=revision_id)
1378
 
            from_branch=None
 
1465
            from_branch = None
1379
1466
        tree = checkout.create_workingtree(revision_id,
1380
1467
                                           from_branch=from_branch,
1381
1468
                                           accelerator_tree=accelerator_tree,
1402
1489
 
1403
1490
    def reference_parent(self, file_id, path, possible_transports=None):
1404
1491
        """Return the parent branch for a tree-reference file_id
 
1492
 
1405
1493
        :param file_id: The file_id of the tree reference
1406
1494
        :param path: The path of the file_id in the tree
1407
1495
        :return: A branch associated with the file_id
1460
1548
        else:
1461
1549
            raise AssertionError("invalid heads: %r" % (heads,))
1462
1550
 
1463
 
 
1464
 
class BranchFormat(object):
 
1551
    def heads_to_fetch(self):
 
1552
        """Return the heads that must and that should be fetched to copy this
 
1553
        branch into another repo.
 
1554
 
 
1555
        :returns: a 2-tuple of (must_fetch, if_present_fetch).  must_fetch is a
 
1556
            set of heads that must be fetched.  if_present_fetch is a set of
 
1557
            heads that must be fetched if present, but no error is necessary if
 
1558
            they are not present.
 
1559
        """
 
1560
        # For bzr native formats must_fetch is just the tip, and
 
1561
        # if_present_fetch are the tags.
 
1562
        must_fetch = set([self.last_revision()])
 
1563
        if_present_fetch = set()
 
1564
        if self.get_config_stack().get('branch.fetch_tags'):
 
1565
            try:
 
1566
                if_present_fetch = set(self.tags.get_reverse_tag_dict())
 
1567
            except errors.TagsNotSupported:
 
1568
                pass
 
1569
        must_fetch.discard(_mod_revision.NULL_REVISION)
 
1570
        if_present_fetch.discard(_mod_revision.NULL_REVISION)
 
1571
        return must_fetch, if_present_fetch
 
1572
 
 
1573
 
 
1574
class BranchFormat(controldir.ControlComponentFormat):
1465
1575
    """An encapsulation of the initialization and open routines for a format.
1466
1576
 
1467
1577
    Formats provide three things:
1468
1578
     * An initialization routine,
1469
 
     * a format string,
 
1579
     * a format description
1470
1580
     * an open routine.
1471
1581
 
1472
1582
    Formats are placed in an dict by their format string for reference
1473
 
    during branch opening. Its not required that these be instances, they
 
1583
    during branch opening. It's not required that these be instances, they
1474
1584
    can be classes themselves with class methods - it simply depends on
1475
1585
    whether state is needed for a given format or not.
1476
1586
 
1479
1589
    object will be created every time regardless.
1480
1590
    """
1481
1591
 
1482
 
    _default_format = None
1483
 
    """The default format used for new branches."""
1484
 
 
1485
 
    _formats = {}
1486
 
    """The known formats."""
1487
 
 
1488
 
    can_set_append_revisions_only = True
1489
 
 
1490
1592
    def __eq__(self, other):
1491
1593
        return self.__class__ is other.__class__
1492
1594
 
1493
1595
    def __ne__(self, other):
1494
1596
        return not (self == other)
1495
1597
 
1496
 
    @classmethod
1497
 
    def find_format(klass, a_bzrdir, name=None):
1498
 
        """Return the format for the branch object in a_bzrdir."""
1499
 
        try:
1500
 
            transport = a_bzrdir.get_branch_transport(None, name=name)
1501
 
            format_string = transport.get_bytes("format")
1502
 
            return klass._formats[format_string]
1503
 
        except errors.NoSuchFile:
1504
 
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
1505
 
        except KeyError:
1506
 
            raise errors.UnknownFormatError(format=format_string, kind='branch')
1507
 
 
1508
 
    @classmethod
1509
 
    def get_default_format(klass):
1510
 
        """Return the current default format."""
1511
 
        return klass._default_format
1512
 
 
1513
 
    def get_reference(self, a_bzrdir):
1514
 
        """Get the target reference of the branch in a_bzrdir.
 
1598
    def get_reference(self, controldir, name=None):
 
1599
        """Get the target reference of the branch in controldir.
1515
1600
 
1516
1601
        format probing must have been completed before calling
1517
1602
        this method - it is assumed that the format of the branch
1518
 
        in a_bzrdir is correct.
 
1603
        in controldir is correct.
1519
1604
 
1520
 
        :param a_bzrdir: The bzrdir to get the branch data from.
 
1605
        :param controldir: The controldir to get the branch data from.
 
1606
        :param name: Name of the colocated branch to fetch
1521
1607
        :return: None if the branch is not a reference branch.
1522
1608
        """
1523
1609
        return None
1524
1610
 
1525
1611
    @classmethod
1526
 
    def set_reference(self, a_bzrdir, to_branch):
1527
 
        """Set the target reference of the branch in a_bzrdir.
 
1612
    def set_reference(self, controldir, name, to_branch):
 
1613
        """Set the target reference of the branch in controldir.
1528
1614
 
1529
1615
        format probing must have been completed before calling
1530
1616
        this method - it is assumed that the format of the branch
1531
 
        in a_bzrdir is correct.
 
1617
        in controldir is correct.
1532
1618
 
1533
 
        :param a_bzrdir: The bzrdir to set the branch reference for.
 
1619
        :param controldir: The controldir to set the branch reference for.
 
1620
        :param name: Name of colocated branch to set, None for default
1534
1621
        :param to_branch: branch that the checkout is to reference
1535
1622
        """
1536
1623
        raise NotImplementedError(self.set_reference)
1537
1624
 
1538
 
    def get_format_string(self):
1539
 
        """Return the ASCII format string that identifies this format."""
1540
 
        raise NotImplementedError(self.get_format_string)
1541
 
 
1542
1625
    def get_format_description(self):
1543
1626
        """Return the short format description for this format."""
1544
1627
        raise NotImplementedError(self.get_format_description)
1545
1628
 
1546
 
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
1547
 
                           lock_type='metadir', set_format=True):
1548
 
        """Initialize a branch in a bzrdir, with specified files
1549
 
 
1550
 
        :param a_bzrdir: The bzrdir to initialize the branch in
1551
 
        :param utf8_files: The files to create as a list of
1552
 
            (filename, content) tuples
1553
 
        :param name: Name of colocated branch to create, if any
1554
 
        :param set_format: If True, set the format with
1555
 
            self.get_format_string.  (BzrBranch4 has its format set
1556
 
            elsewhere)
1557
 
        :return: a branch in this format
1558
 
        """
1559
 
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
1560
 
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1561
 
        lock_map = {
1562
 
            'metadir': ('lock', lockdir.LockDir),
1563
 
            'branch4': ('branch-lock', lockable_files.TransportLock),
1564
 
        }
1565
 
        lock_name, lock_class = lock_map[lock_type]
1566
 
        control_files = lockable_files.LockableFiles(branch_transport,
1567
 
            lock_name, lock_class)
1568
 
        control_files.create_lock()
1569
 
        try:
1570
 
            control_files.lock_write()
1571
 
        except errors.LockContention:
1572
 
            if lock_type != 'branch4':
1573
 
                raise
1574
 
            lock_taken = False
1575
 
        else:
1576
 
            lock_taken = True
1577
 
        if set_format:
1578
 
            utf8_files += [('format', self.get_format_string())]
1579
 
        try:
1580
 
            for (filename, content) in utf8_files:
1581
 
                branch_transport.put_bytes(
1582
 
                    filename, content,
1583
 
                    mode=a_bzrdir._get_file_mode())
1584
 
        finally:
1585
 
            if lock_taken:
1586
 
                control_files.unlock()
1587
 
        return self.open(a_bzrdir, name, _found=True)
1588
 
 
1589
 
    def initialize(self, a_bzrdir, name=None):
1590
 
        """Create a branch of this format in a_bzrdir.
1591
 
        
 
1629
    def _run_post_branch_init_hooks(self, controldir, name, branch):
 
1630
        hooks = Branch.hooks['post_branch_init']
 
1631
        if not hooks:
 
1632
            return
 
1633
        params = BranchInitHookParams(self, controldir, name, branch)
 
1634
        for hook in hooks:
 
1635
            hook(params)
 
1636
 
 
1637
    def initialize(self, controldir, name=None, repository=None,
 
1638
                   append_revisions_only=None):
 
1639
        """Create a branch of this format in controldir.
 
1640
 
1592
1641
        :param name: Name of the colocated branch to create.
1593
1642
        """
1594
1643
        raise NotImplementedError(self.initialize)
1614
1663
        Note that it is normal for branch to be a RemoteBranch when using tags
1615
1664
        on a RemoteBranch.
1616
1665
        """
1617
 
        return DisabledTags(branch)
 
1666
        return _mod_tag.DisabledTags(branch)
1618
1667
 
1619
1668
    def network_name(self):
1620
1669
        """A simple byte string uniquely identifying this format for RPC calls.
1626
1675
        """
1627
1676
        raise NotImplementedError(self.network_name)
1628
1677
 
1629
 
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
1630
 
        """Return the branch object for a_bzrdir
 
1678
    def open(self, controldir, name=None, _found=False, ignore_fallbacks=False,
 
1679
            found_repository=None, possible_transports=None):
 
1680
        """Return the branch object for controldir.
1631
1681
 
1632
 
        :param a_bzrdir: A BzrDir that contains a branch.
 
1682
        :param controldir: A ControlDir that contains a branch.
1633
1683
        :param name: Name of colocated branch to open
1634
1684
        :param _found: a private parameter, do not use it. It is used to
1635
1685
            indicate if format probing has already be done.
1638
1688
        """
1639
1689
        raise NotImplementedError(self.open)
1640
1690
 
1641
 
    @classmethod
1642
 
    def register_format(klass, format):
1643
 
        """Register a metadir format."""
1644
 
        klass._formats[format.get_format_string()] = format
1645
 
        # Metadir formats have a network name of their format string, and get
1646
 
        # registered as class factories.
1647
 
        network_format_registry.register(format.get_format_string(), format.__class__)
1648
 
 
1649
 
    @classmethod
1650
 
    def set_default_format(klass, format):
1651
 
        klass._default_format = format
1652
 
 
1653
1691
    def supports_set_append_revisions_only(self):
1654
1692
        """True if this format supports set_append_revisions_only."""
1655
1693
        return False
1658
1696
        """True if this format records a stacked-on branch."""
1659
1697
        return False
1660
1698
 
1661
 
    @classmethod
1662
 
    def unregister_format(klass, format):
1663
 
        del klass._formats[format.get_format_string()]
 
1699
    def supports_leaving_lock(self):
 
1700
        """True if this format supports leaving locks in place."""
 
1701
        return False # by default
1664
1702
 
1665
1703
    def __str__(self):
1666
1704
        return self.get_format_description().rstrip()
1669
1707
        """True if this format supports tags stored in the branch"""
1670
1708
        return False  # by default
1671
1709
 
 
1710
    def tags_are_versioned(self):
 
1711
        """Whether the tag container for this branch versions tags."""
 
1712
        return False
 
1713
 
 
1714
    def supports_tags_referencing_ghosts(self):
 
1715
        """True if tags can reference ghost revisions."""
 
1716
        return True
 
1717
 
 
1718
 
 
1719
class MetaDirBranchFormatFactory(registry._LazyObjectGetter):
 
1720
    """A factory for a BranchFormat object, permitting simple lazy registration.
 
1721
    
 
1722
    While none of the built in BranchFormats are lazy registered yet,
 
1723
    bzrlib.tests.test_branch.TestMetaDirBranchFormatFactory demonstrates how to
 
1724
    use it, and the bzr-loom plugin uses it as well (see
 
1725
    bzrlib.plugins.loom.formats).
 
1726
    """
 
1727
 
 
1728
    def __init__(self, format_string, module_name, member_name):
 
1729
        """Create a MetaDirBranchFormatFactory.
 
1730
 
 
1731
        :param format_string: The format string the format has.
 
1732
        :param module_name: Module to load the format class from.
 
1733
        :param member_name: Attribute name within the module for the format class.
 
1734
        """
 
1735
        registry._LazyObjectGetter.__init__(self, module_name, member_name)
 
1736
        self._format_string = format_string
 
1737
 
 
1738
    def get_format_string(self):
 
1739
        """See BranchFormat.get_format_string."""
 
1740
        return self._format_string
 
1741
 
 
1742
    def __call__(self):
 
1743
        """Used for network_format_registry support."""
 
1744
        return self.get_obj()()
 
1745
 
1672
1746
 
1673
1747
class BranchHooks(Hooks):
1674
1748
    """A dictionary mapping hook name to a list of callables for branch hooks.
1675
1749
 
1676
 
    e.g. ['set_rh'] Is the list of items to be called when the
1677
 
    set_revision_history function is invoked.
 
1750
    e.g. ['post_push'] Is the list of items to be called when the
 
1751
    push function is invoked.
1678
1752
    """
1679
1753
 
1680
1754
    def __init__(self):
1683
1757
        These are all empty initially, because by default nothing should get
1684
1758
        notified.
1685
1759
        """
1686
 
        Hooks.__init__(self)
1687
 
        self.create_hook(HookPoint('set_rh',
1688
 
            "Invoked whenever the revision history has been set via "
1689
 
            "set_revision_history. The api signature is (branch, "
1690
 
            "revision_history), and the branch will be write-locked. "
1691
 
            "The set_rh hook can be expensive for bzr to trigger, a better "
1692
 
            "hook to use is Branch.post_change_branch_tip.", (0, 15), None))
1693
 
        self.create_hook(HookPoint('open',
 
1760
        Hooks.__init__(self, "bzrlib.branch", "Branch.hooks")
 
1761
        self.add_hook('open',
1694
1762
            "Called with the Branch object that has been opened after a "
1695
 
            "branch is opened.", (1, 8), None))
1696
 
        self.create_hook(HookPoint('post_push',
 
1763
            "branch is opened.", (1, 8))
 
1764
        self.add_hook('post_push',
1697
1765
            "Called after a push operation completes. post_push is called "
1698
1766
            "with a bzrlib.branch.BranchPushResult object and only runs in the "
1699
 
            "bzr client.", (0, 15), None))
1700
 
        self.create_hook(HookPoint('post_pull',
 
1767
            "bzr client.", (0, 15))
 
1768
        self.add_hook('post_pull',
1701
1769
            "Called after a pull operation completes. post_pull is called "
1702
1770
            "with a bzrlib.branch.PullResult object and only runs in the "
1703
 
            "bzr client.", (0, 15), None))
1704
 
        self.create_hook(HookPoint('pre_commit',
1705
 
            "Called after a commit is calculated but before it is is "
 
1771
            "bzr client.", (0, 15))
 
1772
        self.add_hook('pre_commit',
 
1773
            "Called after a commit is calculated but before it is "
1706
1774
            "completed. pre_commit is called with (local, master, old_revno, "
1707
1775
            "old_revid, future_revno, future_revid, tree_delta, future_tree"
1708
1776
            "). old_revid is NULL_REVISION for the first commit to a branch, "
1710
1778
            "basis revision. hooks MUST NOT modify this delta. "
1711
1779
            " future_tree is an in-memory tree obtained from "
1712
1780
            "CommitBuilder.revision_tree() and hooks MUST NOT modify this "
1713
 
            "tree.", (0,91), None))
1714
 
        self.create_hook(HookPoint('post_commit',
 
1781
            "tree.", (0,91))
 
1782
        self.add_hook('post_commit',
1715
1783
            "Called in the bzr client after a commit has completed. "
1716
1784
            "post_commit is called with (local, master, old_revno, old_revid, "
1717
1785
            "new_revno, new_revid). old_revid is NULL_REVISION for the first "
1718
 
            "commit to a branch.", (0, 15), None))
1719
 
        self.create_hook(HookPoint('post_uncommit',
 
1786
            "commit to a branch.", (0, 15))
 
1787
        self.add_hook('post_uncommit',
1720
1788
            "Called in the bzr client after an uncommit completes. "
1721
1789
            "post_uncommit is called with (local, master, old_revno, "
1722
1790
            "old_revid, new_revno, new_revid) where local is the local branch "
1723
1791
            "or None, master is the target branch, and an empty branch "
1724
 
            "receives new_revno of 0, new_revid of None.", (0, 15), None))
1725
 
        self.create_hook(HookPoint('pre_change_branch_tip',
 
1792
            "receives new_revno of 0, new_revid of None.", (0, 15))
 
1793
        self.add_hook('pre_change_branch_tip',
1726
1794
            "Called in bzr client and server before a change to the tip of a "
1727
1795
            "branch is made. pre_change_branch_tip is called with a "
1728
1796
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1729
 
            "commit, uncommit will all trigger this hook.", (1, 6), None))
1730
 
        self.create_hook(HookPoint('post_change_branch_tip',
 
1797
            "commit, uncommit will all trigger this hook.", (1, 6))
 
1798
        self.add_hook('post_change_branch_tip',
1731
1799
            "Called in bzr client and server after a change to the tip of a "
1732
1800
            "branch is made. post_change_branch_tip is called with a "
1733
1801
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1734
 
            "commit, uncommit will all trigger this hook.", (1, 4), None))
1735
 
        self.create_hook(HookPoint('transform_fallback_location',
 
1802
            "commit, uncommit will all trigger this hook.", (1, 4))
 
1803
        self.add_hook('transform_fallback_location',
1736
1804
            "Called when a stacked branch is activating its fallback "
1737
1805
            "locations. transform_fallback_location is called with (branch, "
1738
1806
            "url), and should return a new url. Returning the same url "
1743
1811
            "fallback locations have not been activated. When there are "
1744
1812
            "multiple hooks installed for transform_fallback_location, "
1745
1813
            "all are called with the url returned from the previous hook."
1746
 
            "The order is however undefined.", (1, 9), None))
1747
 
        self.create_hook(HookPoint('automatic_tag_name',
1748
 
            "Called to determine an automatic tag name for a revision."
 
1814
            "The order is however undefined.", (1, 9))
 
1815
        self.add_hook('automatic_tag_name',
 
1816
            "Called to determine an automatic tag name for a revision. "
1749
1817
            "automatic_tag_name is called with (branch, revision_id) and "
1750
1818
            "should return a tag name or None if no tag name could be "
1751
1819
            "determined. The first non-None tag name returned will be used.",
1752
 
            (2, 2), None))
 
1820
            (2, 2))
 
1821
        self.add_hook('post_branch_init',
 
1822
            "Called after new branch initialization completes. "
 
1823
            "post_branch_init is called with a "
 
1824
            "bzrlib.branch.BranchInitHookParams. "
 
1825
            "Note that init, branch and checkout (both heavyweight and "
 
1826
            "lightweight) will all trigger this hook.", (2, 2))
 
1827
        self.add_hook('post_switch',
 
1828
            "Called after a checkout switches branch. "
 
1829
            "post_switch is called with a "
 
1830
            "bzrlib.branch.SwitchHookParams.", (2, 2))
1753
1831
 
1754
1832
 
1755
1833
 
1758
1836
 
1759
1837
 
1760
1838
class ChangeBranchTipParams(object):
1761
 
    """Object holding parameters passed to *_change_branch_tip hooks.
 
1839
    """Object holding parameters passed to `*_change_branch_tip` hooks.
1762
1840
 
1763
1841
    There are 5 fields that hooks may wish to access:
1764
1842
 
1795
1873
            self.old_revno, self.old_revid, self.new_revno, self.new_revid)
1796
1874
 
1797
1875
 
1798
 
class BzrBranchFormat4(BranchFormat):
1799
 
    """Bzr branch format 4.
1800
 
 
1801
 
    This format has:
1802
 
     - a revision-history file.
1803
 
     - a branch-lock lock file [ to be shared with the bzrdir ]
1804
 
    """
1805
 
 
1806
 
    def get_format_description(self):
1807
 
        """See BranchFormat.get_format_description()."""
1808
 
        return "Branch format 4"
1809
 
 
1810
 
    def initialize(self, a_bzrdir, name=None):
1811
 
        """Create a branch of this format in a_bzrdir."""
1812
 
        utf8_files = [('revision-history', ''),
1813
 
                      ('branch-name', ''),
1814
 
                      ]
1815
 
        return self._initialize_helper(a_bzrdir, utf8_files, name=name,
1816
 
                                       lock_type='branch4', set_format=False)
 
1876
class BranchInitHookParams(object):
 
1877
    """Object holding parameters passed to `*_branch_init` hooks.
 
1878
 
 
1879
    There are 4 fields that hooks may wish to access:
 
1880
 
 
1881
    :ivar format: the branch format
 
1882
    :ivar bzrdir: the ControlDir where the branch will be/has been initialized
 
1883
    :ivar name: name of colocated branch, if any (or None)
 
1884
    :ivar branch: the branch created
 
1885
 
 
1886
    Note that for lightweight checkouts, the bzrdir and format fields refer to
 
1887
    the checkout, hence they are different from the corresponding fields in
 
1888
    branch, which refer to the original branch.
 
1889
    """
 
1890
 
 
1891
    def __init__(self, format, controldir, name, branch):
 
1892
        """Create a group of BranchInitHook parameters.
 
1893
 
 
1894
        :param format: the branch format
 
1895
        :param controldir: the ControlDir where the branch will be/has been
 
1896
            initialized
 
1897
        :param name: name of colocated branch, if any (or None)
 
1898
        :param branch: the branch created
 
1899
 
 
1900
        Note that for lightweight checkouts, the bzrdir and format fields refer
 
1901
        to the checkout, hence they are different from the corresponding fields
 
1902
        in branch, which refer to the original branch.
 
1903
        """
 
1904
        self.format = format
 
1905
        self.bzrdir = controldir
 
1906
        self.name = name
 
1907
        self.branch = branch
 
1908
 
 
1909
    def __eq__(self, other):
 
1910
        return self.__dict__ == other.__dict__
 
1911
 
 
1912
    def __repr__(self):
 
1913
        return "<%s of %s>" % (self.__class__.__name__, self.branch)
 
1914
 
 
1915
 
 
1916
class SwitchHookParams(object):
 
1917
    """Object holding parameters passed to `*_switch` hooks.
 
1918
 
 
1919
    There are 4 fields that hooks may wish to access:
 
1920
 
 
1921
    :ivar control_dir: ControlDir of the checkout to change
 
1922
    :ivar to_branch: branch that the checkout is to reference
 
1923
    :ivar force: skip the check for local commits in a heavy checkout
 
1924
    :ivar revision_id: revision ID to switch to (or None)
 
1925
    """
 
1926
 
 
1927
    def __init__(self, control_dir, to_branch, force, revision_id):
 
1928
        """Create a group of SwitchHook parameters.
 
1929
 
 
1930
        :param control_dir: ControlDir of the checkout to change
 
1931
        :param to_branch: branch that the checkout is to reference
 
1932
        :param force: skip the check for local commits in a heavy checkout
 
1933
        :param revision_id: revision ID to switch to (or None)
 
1934
        """
 
1935
        self.control_dir = control_dir
 
1936
        self.to_branch = to_branch
 
1937
        self.force = force
 
1938
        self.revision_id = revision_id
 
1939
 
 
1940
    def __eq__(self, other):
 
1941
        return self.__dict__ == other.__dict__
 
1942
 
 
1943
    def __repr__(self):
 
1944
        return "<%s for %s to (%s, %s)>" % (self.__class__.__name__,
 
1945
            self.control_dir, self.to_branch,
 
1946
            self.revision_id)
 
1947
 
 
1948
 
 
1949
class BranchFormatMetadir(bzrdir.BzrFormat, BranchFormat):
 
1950
    """Base class for branch formats that live in meta directories.
 
1951
    """
1817
1952
 
1818
1953
    def __init__(self):
1819
 
        super(BzrBranchFormat4, self).__init__()
1820
 
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
1821
 
 
1822
 
    def network_name(self):
1823
 
        """The network name for this format is the control dirs disk label."""
1824
 
        return self._matchingbzrdir.get_format_string()
1825
 
 
1826
 
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
1827
 
        """See BranchFormat.open()."""
1828
 
        if not _found:
1829
 
            # we are being called directly and must probe.
1830
 
            raise NotImplementedError
1831
 
        return BzrBranch(_format=self,
1832
 
                         _control_files=a_bzrdir._control_files,
1833
 
                         a_bzrdir=a_bzrdir,
1834
 
                         name=name,
1835
 
                         _repository=a_bzrdir.open_repository())
1836
 
 
1837
 
    def __str__(self):
1838
 
        return "Bazaar-NG branch format 4"
1839
 
 
1840
 
 
1841
 
class BranchFormatMetadir(BranchFormat):
1842
 
    """Common logic for meta-dir based branch formats."""
 
1954
        BranchFormat.__init__(self)
 
1955
        bzrdir.BzrFormat.__init__(self)
 
1956
 
 
1957
    @classmethod
 
1958
    def find_format(klass, controldir, name=None):
 
1959
        """Return the format for the branch object in controldir."""
 
1960
        try:
 
1961
            transport = controldir.get_branch_transport(None, name=name)
 
1962
        except errors.NoSuchFile:
 
1963
            raise errors.NotBranchError(path=name, bzrdir=controldir)
 
1964
        try:
 
1965
            format_string = transport.get_bytes("format")
 
1966
        except errors.NoSuchFile:
 
1967
            raise errors.NotBranchError(path=transport.base, bzrdir=controldir)
 
1968
        return klass._find_format(format_registry, 'branch', format_string)
1843
1969
 
1844
1970
    def _branch_class(self):
1845
1971
        """What class to instantiate on open calls."""
1846
1972
        raise NotImplementedError(self._branch_class)
1847
1973
 
1848
 
    def network_name(self):
1849
 
        """A simple byte string uniquely identifying this format for RPC calls.
1850
 
 
1851
 
        Metadir branch formats use their format string.
 
1974
    def _get_initial_config(self, append_revisions_only=None):
 
1975
        if append_revisions_only:
 
1976
            return "append_revisions_only = True\n"
 
1977
        else:
 
1978
            # Avoid writing anything if append_revisions_only is disabled,
 
1979
            # as that is the default.
 
1980
            return ""
 
1981
 
 
1982
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
 
1983
                           repository=None):
 
1984
        """Initialize a branch in a control dir, with specified files
 
1985
 
 
1986
        :param a_bzrdir: The bzrdir to initialize the branch in
 
1987
        :param utf8_files: The files to create as a list of
 
1988
            (filename, content) tuples
 
1989
        :param name: Name of colocated branch to create, if any
 
1990
        :return: a branch in this format
1852
1991
        """
1853
 
        return self.get_format_string()
 
1992
        if name is None:
 
1993
            name = a_bzrdir._get_selected_branch()
 
1994
        mutter('creating branch %r in %s', self, a_bzrdir.user_url)
 
1995
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
 
1996
        control_files = lockable_files.LockableFiles(branch_transport,
 
1997
            'lock', lockdir.LockDir)
 
1998
        control_files.create_lock()
 
1999
        control_files.lock_write()
 
2000
        try:
 
2001
            utf8_files += [('format', self.as_string())]
 
2002
            for (filename, content) in utf8_files:
 
2003
                branch_transport.put_bytes(
 
2004
                    filename, content,
 
2005
                    mode=a_bzrdir._get_file_mode())
 
2006
        finally:
 
2007
            control_files.unlock()
 
2008
        branch = self.open(a_bzrdir, name, _found=True,
 
2009
                found_repository=repository)
 
2010
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
 
2011
        return branch
1854
2012
 
1855
 
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
 
2013
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False,
 
2014
            found_repository=None, possible_transports=None):
1856
2015
        """See BranchFormat.open()."""
 
2016
        if name is None:
 
2017
            name = a_bzrdir._get_selected_branch()
1857
2018
        if not _found:
1858
 
            format = BranchFormat.find_format(a_bzrdir, name=name)
 
2019
            format = BranchFormatMetadir.find_format(a_bzrdir, name=name)
1859
2020
            if format.__class__ != self.__class__:
1860
2021
                raise AssertionError("wrong format %r found for %r" %
1861
2022
                    (format, self))
 
2023
        transport = a_bzrdir.get_branch_transport(None, name=name)
1862
2024
        try:
1863
 
            transport = a_bzrdir.get_branch_transport(None, name=name)
1864
2025
            control_files = lockable_files.LockableFiles(transport, 'lock',
1865
2026
                                                         lockdir.LockDir)
 
2027
            if found_repository is None:
 
2028
                found_repository = a_bzrdir.find_repository()
1866
2029
            return self._branch_class()(_format=self,
1867
2030
                              _control_files=control_files,
1868
2031
                              name=name,
1869
2032
                              a_bzrdir=a_bzrdir,
1870
 
                              _repository=a_bzrdir.find_repository(),
1871
 
                              ignore_fallbacks=ignore_fallbacks)
 
2033
                              _repository=found_repository,
 
2034
                              ignore_fallbacks=ignore_fallbacks,
 
2035
                              possible_transports=possible_transports)
1872
2036
        except errors.NoSuchFile:
1873
2037
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
1874
2038
 
1875
 
    def __init__(self):
1876
 
        super(BranchFormatMetadir, self).__init__()
1877
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
1878
 
        self._matchingbzrdir.set_branch_format(self)
1879
 
 
1880
 
    def supports_tags(self):
1881
 
        return True
1882
 
 
1883
 
 
1884
 
class BzrBranchFormat5(BranchFormatMetadir):
1885
 
    """Bzr branch format 5.
1886
 
 
1887
 
    This format has:
1888
 
     - a revision-history file.
1889
 
     - a format string
1890
 
     - a lock dir guarding the branch itself
1891
 
     - all of this stored in a branch/ subdirectory
1892
 
     - works with shared repositories.
1893
 
 
1894
 
    This format is new in bzr 0.8.
1895
 
    """
1896
 
 
1897
 
    def _branch_class(self):
1898
 
        return BzrBranch5
1899
 
 
1900
 
    def get_format_string(self):
1901
 
        """See BranchFormat.get_format_string()."""
1902
 
        return "Bazaar-NG branch format 5\n"
1903
 
 
1904
 
    def get_format_description(self):
1905
 
        """See BranchFormat.get_format_description()."""
1906
 
        return "Branch format 5"
1907
 
 
1908
 
    def initialize(self, a_bzrdir, name=None):
1909
 
        """Create a branch of this format in a_bzrdir."""
1910
 
        utf8_files = [('revision-history', ''),
1911
 
                      ('branch-name', ''),
1912
 
                      ]
1913
 
        return self._initialize_helper(a_bzrdir, utf8_files, name)
1914
 
 
1915
 
    def supports_tags(self):
1916
 
        return False
 
2039
    @property
 
2040
    def _matchingbzrdir(self):
 
2041
        ret = bzrdir.BzrDirMetaFormat1()
 
2042
        ret.set_branch_format(self)
 
2043
        return ret
 
2044
 
 
2045
    def supports_tags(self):
 
2046
        return True
 
2047
 
 
2048
    def supports_leaving_lock(self):
 
2049
        return True
 
2050
 
 
2051
    def check_support_status(self, allow_unsupported, recommend_upgrade=True,
 
2052
            basedir=None):
 
2053
        BranchFormat.check_support_status(self,
 
2054
            allow_unsupported=allow_unsupported, recommend_upgrade=recommend_upgrade,
 
2055
            basedir=basedir)
 
2056
        bzrdir.BzrFormat.check_support_status(self, allow_unsupported=allow_unsupported,
 
2057
            recommend_upgrade=recommend_upgrade, basedir=basedir)
1917
2058
 
1918
2059
 
1919
2060
class BzrBranchFormat6(BranchFormatMetadir):
1930
2071
    def _branch_class(self):
1931
2072
        return BzrBranch6
1932
2073
 
1933
 
    def get_format_string(self):
 
2074
    @classmethod
 
2075
    def get_format_string(cls):
1934
2076
        """See BranchFormat.get_format_string()."""
1935
2077
        return "Bazaar Branch Format 6 (bzr 0.15)\n"
1936
2078
 
1938
2080
        """See BranchFormat.get_format_description()."""
1939
2081
        return "Branch format 6"
1940
2082
 
1941
 
    def initialize(self, a_bzrdir, name=None):
 
2083
    def initialize(self, a_bzrdir, name=None, repository=None,
 
2084
                   append_revisions_only=None):
1942
2085
        """Create a branch of this format in a_bzrdir."""
1943
2086
        utf8_files = [('last-revision', '0 null:\n'),
1944
 
                      ('branch.conf', ''),
 
2087
                      ('branch.conf',
 
2088
                          self._get_initial_config(append_revisions_only)),
1945
2089
                      ('tags', ''),
1946
2090
                      ]
1947
 
        return self._initialize_helper(a_bzrdir, utf8_files, name)
 
2091
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
1948
2092
 
1949
2093
    def make_tags(self, branch):
1950
2094
        """See bzrlib.branch.BranchFormat.make_tags()."""
1951
 
        return BasicTags(branch)
 
2095
        return _mod_tag.BasicTags(branch)
1952
2096
 
1953
2097
    def supports_set_append_revisions_only(self):
1954
2098
        return True
1960
2104
    def _branch_class(self):
1961
2105
        return BzrBranch8
1962
2106
 
1963
 
    def get_format_string(self):
 
2107
    @classmethod
 
2108
    def get_format_string(cls):
1964
2109
        """See BranchFormat.get_format_string()."""
1965
2110
        return "Bazaar Branch Format 8 (needs bzr 1.15)\n"
1966
2111
 
1968
2113
        """See BranchFormat.get_format_description()."""
1969
2114
        return "Branch format 8"
1970
2115
 
1971
 
    def initialize(self, a_bzrdir, name=None):
 
2116
    def initialize(self, a_bzrdir, name=None, repository=None,
 
2117
                   append_revisions_only=None):
1972
2118
        """Create a branch of this format in a_bzrdir."""
1973
2119
        utf8_files = [('last-revision', '0 null:\n'),
1974
 
                      ('branch.conf', ''),
 
2120
                      ('branch.conf',
 
2121
                          self._get_initial_config(append_revisions_only)),
1975
2122
                      ('tags', ''),
1976
2123
                      ('references', '')
1977
2124
                      ]
1978
 
        return self._initialize_helper(a_bzrdir, utf8_files, name)
1979
 
 
1980
 
    def __init__(self):
1981
 
        super(BzrBranchFormat8, self).__init__()
1982
 
        self._matchingbzrdir.repository_format = \
1983
 
            RepositoryFormatKnitPack5RichRoot()
 
2125
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
1984
2126
 
1985
2127
    def make_tags(self, branch):
1986
2128
        """See bzrlib.branch.BranchFormat.make_tags()."""
1987
 
        return BasicTags(branch)
 
2129
        return _mod_tag.BasicTags(branch)
1988
2130
 
1989
2131
    def supports_set_append_revisions_only(self):
1990
2132
        return True
1995
2137
    supports_reference_locations = True
1996
2138
 
1997
2139
 
1998
 
class BzrBranchFormat7(BzrBranchFormat8):
 
2140
class BzrBranchFormat7(BranchFormatMetadir):
1999
2141
    """Branch format with last-revision, tags, and a stacked location pointer.
2000
2142
 
2001
2143
    The stacked location pointer is passed down to the repository and requires
2004
2146
    This format was introduced in bzr 1.6.
2005
2147
    """
2006
2148
 
2007
 
    def initialize(self, a_bzrdir, name=None):
 
2149
    def initialize(self, a_bzrdir, name=None, repository=None,
 
2150
                   append_revisions_only=None):
2008
2151
        """Create a branch of this format in a_bzrdir."""
2009
2152
        utf8_files = [('last-revision', '0 null:\n'),
2010
 
                      ('branch.conf', ''),
 
2153
                      ('branch.conf',
 
2154
                          self._get_initial_config(append_revisions_only)),
2011
2155
                      ('tags', ''),
2012
2156
                      ]
2013
 
        return self._initialize_helper(a_bzrdir, utf8_files, name)
 
2157
        return self._initialize_helper(a_bzrdir, utf8_files, name, repository)
2014
2158
 
2015
2159
    def _branch_class(self):
2016
2160
        return BzrBranch7
2017
2161
 
2018
 
    def get_format_string(self):
 
2162
    @classmethod
 
2163
    def get_format_string(cls):
2019
2164
        """See BranchFormat.get_format_string()."""
2020
2165
        return "Bazaar Branch Format 7 (needs bzr 1.6)\n"
2021
2166
 
2026
2171
    def supports_set_append_revisions_only(self):
2027
2172
        return True
2028
2173
 
 
2174
    def supports_stacking(self):
 
2175
        return True
 
2176
 
 
2177
    def make_tags(self, branch):
 
2178
        """See bzrlib.branch.BranchFormat.make_tags()."""
 
2179
        return _mod_tag.BasicTags(branch)
 
2180
 
2029
2181
    supports_reference_locations = False
2030
2182
 
2031
2183
 
2032
 
class BranchReferenceFormat(BranchFormat):
 
2184
class BranchReferenceFormat(BranchFormatMetadir):
2033
2185
    """Bzr branch reference format.
2034
2186
 
2035
2187
    Branch references are used in implementing checkouts, they
2040
2192
     - a format string
2041
2193
    """
2042
2194
 
2043
 
    def get_format_string(self):
 
2195
    @classmethod
 
2196
    def get_format_string(cls):
2044
2197
        """See BranchFormat.get_format_string()."""
2045
2198
        return "Bazaar-NG Branch Reference Format 1\n"
2046
2199
 
2048
2201
        """See BranchFormat.get_format_description()."""
2049
2202
        return "Checkout reference format 1"
2050
2203
 
2051
 
    def get_reference(self, a_bzrdir):
 
2204
    def get_reference(self, a_bzrdir, name=None):
2052
2205
        """See BranchFormat.get_reference()."""
2053
 
        transport = a_bzrdir.get_branch_transport(None)
 
2206
        transport = a_bzrdir.get_branch_transport(None, name=name)
2054
2207
        return transport.get_bytes('location')
2055
2208
 
2056
 
    def set_reference(self, a_bzrdir, to_branch):
 
2209
    def set_reference(self, a_bzrdir, name, to_branch):
2057
2210
        """See BranchFormat.set_reference()."""
2058
 
        transport = a_bzrdir.get_branch_transport(None)
 
2211
        transport = a_bzrdir.get_branch_transport(None, name=name)
2059
2212
        location = transport.put_bytes('location', to_branch.base)
2060
2213
 
2061
 
    def initialize(self, a_bzrdir, name=None, target_branch=None):
 
2214
    def initialize(self, a_bzrdir, name=None, target_branch=None,
 
2215
            repository=None, append_revisions_only=None):
2062
2216
        """Create a branch of this format in a_bzrdir."""
2063
2217
        if target_branch is None:
2064
2218
            # this format does not implement branch itself, thus the implicit
2065
2219
            # creation contract must see it as uninitializable
2066
2220
            raise errors.UninitializableFormat(self)
2067
 
        mutter('creating branch reference in %s', a_bzrdir.transport.base)
 
2221
        mutter('creating branch reference in %s', a_bzrdir.user_url)
 
2222
        if a_bzrdir._format.fixed_components:
 
2223
            raise errors.IncompatibleFormat(self, a_bzrdir._format)
 
2224
        if name is None:
 
2225
            name = a_bzrdir._get_selected_branch()
2068
2226
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
2069
2227
        branch_transport.put_bytes('location',
2070
 
            target_branch.bzrdir.root_transport.base)
2071
 
        branch_transport.put_bytes('format', self.get_format_string())
2072
 
        return self.open(
2073
 
            a_bzrdir, name, _found=True,
 
2228
            target_branch.user_url)
 
2229
        branch_transport.put_bytes('format', self.as_string())
 
2230
        branch = self.open(a_bzrdir, name, _found=True,
2074
2231
            possible_transports=[target_branch.bzrdir.root_transport])
2075
 
 
2076
 
    def __init__(self):
2077
 
        super(BranchReferenceFormat, self).__init__()
2078
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
2079
 
        self._matchingbzrdir.set_branch_format(self)
 
2232
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
 
2233
        return branch
2080
2234
 
2081
2235
    def _make_reference_clone_function(format, a_branch):
2082
2236
        """Create a clone() routine for a branch dynamically."""
2090
2244
        return clone
2091
2245
 
2092
2246
    def open(self, a_bzrdir, name=None, _found=False, location=None,
2093
 
             possible_transports=None, ignore_fallbacks=False):
 
2247
             possible_transports=None, ignore_fallbacks=False,
 
2248
             found_repository=None):
2094
2249
        """Return the branch that the branch reference in a_bzrdir points at.
2095
2250
 
2096
2251
        :param a_bzrdir: A BzrDir that contains a branch.
2104
2259
            a_bzrdir.
2105
2260
        :param possible_transports: An optional reusable transports list.
2106
2261
        """
 
2262
        if name is None:
 
2263
            name = a_bzrdir._get_selected_branch()
2107
2264
        if not _found:
2108
 
            format = BranchFormat.find_format(a_bzrdir, name=name)
 
2265
            format = BranchFormatMetadir.find_format(a_bzrdir, name=name)
2109
2266
            if format.__class__ != self.__class__:
2110
2267
                raise AssertionError("wrong format %r found for %r" %
2111
2268
                    (format, self))
2112
2269
        if location is None:
2113
 
            location = self.get_reference(a_bzrdir)
2114
 
        real_bzrdir = bzrdir.BzrDir.open(
 
2270
            location = self.get_reference(a_bzrdir, name)
 
2271
        real_bzrdir = controldir.ControlDir.open(
2115
2272
            location, possible_transports=possible_transports)
2116
 
        result = real_bzrdir.open_branch(name=name, 
2117
 
            ignore_fallbacks=ignore_fallbacks)
 
2273
        result = real_bzrdir.open_branch(ignore_fallbacks=ignore_fallbacks,
 
2274
            possible_transports=possible_transports)
2118
2275
        # this changes the behaviour of result.clone to create a new reference
2119
2276
        # rather than a copy of the content of the branch.
2120
2277
        # I did not use a proxy object because that needs much more extensive
2127
2284
        return result
2128
2285
 
2129
2286
 
 
2287
class BranchFormatRegistry(controldir.ControlComponentFormatRegistry):
 
2288
    """Branch format registry."""
 
2289
 
 
2290
    def __init__(self, other_registry=None):
 
2291
        super(BranchFormatRegistry, self).__init__(other_registry)
 
2292
        self._default_format = None
 
2293
 
 
2294
    def set_default(self, format):
 
2295
        self._default_format = format
 
2296
 
 
2297
    def get_default(self):
 
2298
        return self._default_format
 
2299
 
 
2300
 
2130
2301
network_format_registry = registry.FormatRegistry()
2131
2302
"""Registry of formats indexed by their network name.
2132
2303
 
2135
2306
BranchFormat.network_name() for more detail.
2136
2307
"""
2137
2308
 
 
2309
format_registry = BranchFormatRegistry(network_format_registry)
 
2310
 
2138
2311
 
2139
2312
# formats which have no format string are not discoverable
2140
2313
# and not independently creatable, so are not registered.
2141
 
__format5 = BzrBranchFormat5()
2142
2314
__format6 = BzrBranchFormat6()
2143
2315
__format7 = BzrBranchFormat7()
2144
2316
__format8 = BzrBranchFormat8()
2145
 
BranchFormat.register_format(__format5)
2146
 
BranchFormat.register_format(BranchReferenceFormat())
2147
 
BranchFormat.register_format(__format6)
2148
 
BranchFormat.register_format(__format7)
2149
 
BranchFormat.register_format(__format8)
2150
 
BranchFormat.set_default_format(__format7)
2151
 
_legacy_formats = [BzrBranchFormat4(),
2152
 
    ]
2153
 
network_format_registry.register(
2154
 
    _legacy_formats[0].network_name(), _legacy_formats[0].__class__)
 
2317
format_registry.register_lazy(
 
2318
    "Bazaar-NG branch format 5\n", "bzrlib.branchfmt.fullhistory", "BzrBranchFormat5")
 
2319
format_registry.register(BranchReferenceFormat())
 
2320
format_registry.register(__format6)
 
2321
format_registry.register(__format7)
 
2322
format_registry.register(__format8)
 
2323
format_registry.set_default(__format7)
 
2324
 
 
2325
 
 
2326
class BranchWriteLockResult(LogicalLockResult):
 
2327
    """The result of write locking a branch.
 
2328
 
 
2329
    :ivar branch_token: The token obtained from the underlying branch lock, or
 
2330
        None.
 
2331
    :ivar unlock: A callable which will unlock the lock.
 
2332
    """
 
2333
 
 
2334
    def __init__(self, unlock, branch_token):
 
2335
        LogicalLockResult.__init__(self, unlock)
 
2336
        self.branch_token = branch_token
 
2337
 
 
2338
    def __repr__(self):
 
2339
        return "BranchWriteLockResult(%s, %s)" % (self.branch_token,
 
2340
            self.unlock)
2155
2341
 
2156
2342
 
2157
2343
class BzrBranch(Branch, _RelockDebugMixin):
2172
2358
 
2173
2359
    def __init__(self, _format=None,
2174
2360
                 _control_files=None, a_bzrdir=None, name=None,
2175
 
                 _repository=None, ignore_fallbacks=False):
 
2361
                 _repository=None, ignore_fallbacks=False,
 
2362
                 possible_transports=None):
2176
2363
        """Create new branch object at a particular location."""
2177
2364
        if a_bzrdir is None:
2178
2365
            raise ValueError('a_bzrdir must be supplied')
2179
 
        else:
2180
 
            self.bzrdir = a_bzrdir
2181
 
        self._base = self.bzrdir.transport.clone('..').base
 
2366
        if name is None:
 
2367
            raise ValueError('name must be supplied')
 
2368
        self.bzrdir = a_bzrdir
 
2369
        self._user_transport = self.bzrdir.transport.clone('..')
 
2370
        if name != "":
 
2371
            self._user_transport.set_segment_parameter(
 
2372
                "branch", urlutils.escape(name))
 
2373
        self._base = self._user_transport.base
2182
2374
        self.name = name
2183
 
        # XXX: We should be able to just do
2184
 
        #   self.base = self.bzrdir.root_transport.base
2185
 
        # but this does not quite work yet -- mbp 20080522
2186
2375
        self._format = _format
2187
2376
        if _control_files is None:
2188
2377
            raise ValueError('BzrBranch _control_files is None')
2189
2378
        self.control_files = _control_files
2190
2379
        self._transport = _control_files._transport
2191
2380
        self.repository = _repository
2192
 
        Branch.__init__(self)
 
2381
        self.conf_store = None
 
2382
        Branch.__init__(self, possible_transports)
2193
2383
 
2194
2384
    def __str__(self):
2195
 
        if self.name is None:
2196
 
            return '%s(%r)' % (self.__class__.__name__, self.base)
2197
 
        else:
2198
 
            return '%s(%r,%r)' % (self.__class__.__name__, self.base, self.name)
 
2385
        return '%s(%s)' % (self.__class__.__name__, self.user_url)
2199
2386
 
2200
2387
    __repr__ = __str__
2201
2388
 
2205
2392
 
2206
2393
    base = property(_get_base, doc="The URL for the root of this branch.")
2207
2394
 
 
2395
    @property
 
2396
    def user_transport(self):
 
2397
        return self._user_transport
 
2398
 
2208
2399
    def _get_config(self):
2209
 
        return TransportConfig(self._transport, 'branch.conf')
 
2400
        return _mod_config.TransportConfig(self._transport, 'branch.conf')
 
2401
 
 
2402
    def _get_config_store(self):
 
2403
        if self.conf_store is None:
 
2404
            self.conf_store =  _mod_config.BranchStore(self)
 
2405
        return self.conf_store
 
2406
 
 
2407
    def _uncommitted_branch(self):
 
2408
        """Return the branch that may contain uncommitted changes."""
 
2409
        master = self.get_master_branch()
 
2410
        if master is not None:
 
2411
            return master
 
2412
        else:
 
2413
            return self
 
2414
 
 
2415
    def store_uncommitted(self, creator):
 
2416
        """Store uncommitted changes from a ShelfCreator.
 
2417
 
 
2418
        :param creator: The ShelfCreator containing uncommitted changes, or
 
2419
            None to delete any stored changes.
 
2420
        :raises: ChangesAlreadyStored if the branch already has changes.
 
2421
        """
 
2422
        branch = self._uncommitted_branch()
 
2423
        if creator is None:
 
2424
            branch._transport.delete('stored-transform')
 
2425
            return
 
2426
        if branch._transport.has('stored-transform'):
 
2427
            raise errors.ChangesAlreadyStored
 
2428
        transform = StringIO()
 
2429
        creator.write_shelf(transform)
 
2430
        transform.seek(0)
 
2431
        branch._transport.put_file('stored-transform', transform)
 
2432
 
 
2433
    def get_unshelver(self, tree):
 
2434
        """Return a shelf.Unshelver for this branch and tree.
 
2435
 
 
2436
        :param tree: The tree to use to construct the Unshelver.
 
2437
        :return: an Unshelver or None if no changes are stored.
 
2438
        """
 
2439
        branch = self._uncommitted_branch()
 
2440
        try:
 
2441
            transform = branch._transport.get('stored-transform')
 
2442
        except errors.NoSuchFile:
 
2443
            return None
 
2444
        return shelf.Unshelver.from_tree_and_shelf(tree, transform)
2210
2445
 
2211
2446
    def is_locked(self):
2212
2447
        return self.control_files.is_locked()
2213
2448
 
2214
2449
    def lock_write(self, token=None):
 
2450
        """Lock the branch for write operations.
 
2451
 
 
2452
        :param token: A token to permit reacquiring a previously held and
 
2453
            preserved lock.
 
2454
        :return: A BranchWriteLockResult.
 
2455
        """
2215
2456
        if not self.is_locked():
2216
2457
            self._note_lock('w')
2217
 
        # All-in-one needs to always unlock/lock.
2218
 
        repo_control = getattr(self.repository, 'control_files', None)
2219
 
        if self.control_files == repo_control or not self.is_locked():
2220
2458
            self.repository._warn_if_deprecated(self)
2221
2459
            self.repository.lock_write()
2222
2460
            took_lock = True
2223
2461
        else:
2224
2462
            took_lock = False
2225
2463
        try:
2226
 
            return self.control_files.lock_write(token=token)
 
2464
            return BranchWriteLockResult(self.unlock,
 
2465
                self.control_files.lock_write(token=token))
2227
2466
        except:
2228
2467
            if took_lock:
2229
2468
                self.repository.unlock()
2230
2469
            raise
2231
2470
 
2232
2471
    def lock_read(self):
 
2472
        """Lock the branch for read operations.
 
2473
 
 
2474
        :return: A bzrlib.lock.LogicalLockResult.
 
2475
        """
2233
2476
        if not self.is_locked():
2234
2477
            self._note_lock('r')
2235
 
        # All-in-one needs to always unlock/lock.
2236
 
        repo_control = getattr(self.repository, 'control_files', None)
2237
 
        if self.control_files == repo_control or not self.is_locked():
2238
2478
            self.repository._warn_if_deprecated(self)
2239
2479
            self.repository.lock_read()
2240
2480
            took_lock = True
2242
2482
            took_lock = False
2243
2483
        try:
2244
2484
            self.control_files.lock_read()
 
2485
            return LogicalLockResult(self.unlock)
2245
2486
        except:
2246
2487
            if took_lock:
2247
2488
                self.repository.unlock()
2249
2490
 
2250
2491
    @only_raises(errors.LockNotHeld, errors.LockBroken)
2251
2492
    def unlock(self):
 
2493
        if self.control_files._lock_count == 1 and self.conf_store is not None:
 
2494
            self.conf_store.save_changes()
2252
2495
        try:
2253
2496
            self.control_files.unlock()
2254
2497
        finally:
2255
 
            # All-in-one needs to always unlock/lock.
2256
 
            repo_control = getattr(self.repository, 'control_files', None)
2257
 
            if (self.control_files == repo_control or
2258
 
                not self.control_files.is_locked()):
2259
 
                self.repository.unlock()
2260
2498
            if not self.control_files.is_locked():
 
2499
                self.repository.unlock()
2261
2500
                # we just released the lock
2262
2501
                self._clear_cached_state()
2263
2502
 
2275
2514
        """See Branch.print_file."""
2276
2515
        return self.repository.print_file(file, revision_id)
2277
2516
 
2278
 
    def _write_revision_history(self, history):
2279
 
        """Factored out of set_revision_history.
2280
 
 
2281
 
        This performs the actual writing to disk.
2282
 
        It is intended to be called by BzrBranch5.set_revision_history."""
2283
 
        self._transport.put_bytes(
2284
 
            'revision-history', '\n'.join(history),
2285
 
            mode=self.bzrdir._get_file_mode())
2286
 
 
2287
 
    @needs_write_lock
2288
 
    def set_revision_history(self, rev_history):
2289
 
        """See Branch.set_revision_history."""
2290
 
        if 'evil' in debug.debug_flags:
2291
 
            mutter_callsite(3, "set_revision_history scales with history.")
2292
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
2293
 
        for rev_id in rev_history:
2294
 
            check_not_reserved_id(rev_id)
2295
 
        if Branch.hooks['post_change_branch_tip']:
2296
 
            # Don't calculate the last_revision_info() if there are no hooks
2297
 
            # that will use it.
2298
 
            old_revno, old_revid = self.last_revision_info()
2299
 
        if len(rev_history) == 0:
2300
 
            revid = _mod_revision.NULL_REVISION
2301
 
        else:
2302
 
            revid = rev_history[-1]
2303
 
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
2304
 
        self._write_revision_history(rev_history)
2305
 
        self._clear_cached_state()
2306
 
        self._cache_revision_history(rev_history)
2307
 
        for hook in Branch.hooks['set_rh']:
2308
 
            hook(self, rev_history)
2309
 
        if Branch.hooks['post_change_branch_tip']:
2310
 
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2311
 
 
2312
 
    def _synchronize_history(self, destination, revision_id):
2313
 
        """Synchronize last revision and revision history between branches.
2314
 
 
2315
 
        This version is most efficient when the destination is also a
2316
 
        BzrBranch5, but works for BzrBranch6 as long as the revision
2317
 
        history is the true lefthand parent history, and all of the revisions
2318
 
        are in the destination's repository.  If not, set_revision_history
2319
 
        will fail.
2320
 
 
2321
 
        :param destination: The branch to copy the history into
2322
 
        :param revision_id: The revision-id to truncate history at.  May
2323
 
          be None to copy complete history.
2324
 
        """
2325
 
        if not isinstance(destination._format, BzrBranchFormat5):
2326
 
            super(BzrBranch, self)._synchronize_history(
2327
 
                destination, revision_id)
2328
 
            return
2329
 
        if revision_id == _mod_revision.NULL_REVISION:
2330
 
            new_history = []
2331
 
        else:
2332
 
            new_history = self.revision_history()
2333
 
        if revision_id is not None and new_history != []:
2334
 
            try:
2335
 
                new_history = new_history[:new_history.index(revision_id) + 1]
2336
 
            except ValueError:
2337
 
                rev = self.repository.get_revision(revision_id)
2338
 
                new_history = rev.get_history(self.repository)[1:]
2339
 
        destination.set_revision_history(new_history)
2340
 
 
2341
2517
    @needs_write_lock
2342
2518
    def set_last_revision_info(self, revno, revision_id):
2343
 
        """Set the last revision of this branch.
2344
 
 
2345
 
        The caller is responsible for checking that the revno is correct
2346
 
        for this revision id.
2347
 
 
2348
 
        It may be possible to set the branch last revision to an id not
2349
 
        present in the repository.  However, branches can also be
2350
 
        configured to check constraints on history, in which case this may not
2351
 
        be permitted.
2352
 
        """
 
2519
        if not revision_id or not isinstance(revision_id, basestring):
 
2520
            raise errors.InvalidRevisionId(revision_id=revision_id, branch=self)
2353
2521
        revision_id = _mod_revision.ensure_null(revision_id)
2354
 
        # this old format stores the full history, but this api doesn't
2355
 
        # provide it, so we must generate, and might as well check it's
2356
 
        # correct
2357
 
        history = self._lefthand_history(revision_id)
2358
 
        if len(history) != revno:
2359
 
            raise AssertionError('%d != %d' % (len(history), revno))
2360
 
        self.set_revision_history(history)
2361
 
 
2362
 
    def _gen_revision_history(self):
2363
 
        history = self._transport.get_bytes('revision-history').split('\n')
2364
 
        if history[-1:] == ['']:
2365
 
            # There shouldn't be a trailing newline, but just in case.
2366
 
            history.pop()
2367
 
        return history
2368
 
 
2369
 
    @needs_write_lock
2370
 
    def generate_revision_history(self, revision_id, last_rev=None,
2371
 
        other_branch=None):
2372
 
        """Create a new revision history that will finish with revision_id.
2373
 
 
2374
 
        :param revision_id: the new tip to use.
2375
 
        :param last_rev: The previous last_revision. If not None, then this
2376
 
            must be a ancestory of revision_id, or DivergedBranches is raised.
2377
 
        :param other_branch: The other branch that DivergedBranches should
2378
 
            raise with respect to.
2379
 
        """
2380
 
        self.set_revision_history(self._lefthand_history(revision_id,
2381
 
            last_rev, other_branch))
 
2522
        old_revno, old_revid = self.last_revision_info()
 
2523
        if self.get_append_revisions_only():
 
2524
            self._check_history_violation(revision_id)
 
2525
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
 
2526
        self._write_last_revision_info(revno, revision_id)
 
2527
        self._clear_cached_state()
 
2528
        self._last_revision_info_cache = revno, revision_id
 
2529
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2382
2530
 
2383
2531
    def basis_tree(self):
2384
2532
        """See Branch.basis_tree."""
2393
2541
                pass
2394
2542
        return None
2395
2543
 
2396
 
    def _basic_push(self, target, overwrite, stop_revision):
2397
 
        """Basic implementation of push without bound branches or hooks.
2398
 
 
2399
 
        Must be called with source read locked and target write locked.
2400
 
        """
2401
 
        result = BranchPushResult()
2402
 
        result.source_branch = self
2403
 
        result.target_branch = target
2404
 
        result.old_revno, result.old_revid = target.last_revision_info()
2405
 
        self.update_references(target)
2406
 
        if result.old_revid != self.last_revision():
2407
 
            # We assume that during 'push' this repository is closer than
2408
 
            # the target.
2409
 
            graph = self.repository.get_graph(target.repository)
2410
 
            target.update_revisions(self, stop_revision,
2411
 
                overwrite=overwrite, graph=graph)
2412
 
        if self._push_should_merge_tags():
2413
 
            result.tag_conflicts = self.tags.merge_to(target.tags,
2414
 
                overwrite)
2415
 
        result.new_revno, result.new_revid = target.last_revision_info()
2416
 
        return result
2417
 
 
2418
2544
    def get_stacked_on_url(self):
2419
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
 
2545
        raise errors.UnstackableBranchFormat(self._format, self.user_url)
2420
2546
 
2421
2547
    def set_push_location(self, location):
2422
2548
        """See Branch.set_push_location."""
2431
2557
            self._transport.put_bytes('parent', url + '\n',
2432
2558
                mode=self.bzrdir._get_file_mode())
2433
2559
 
2434
 
 
2435
 
class BzrBranch5(BzrBranch):
2436
 
    """A format 5 branch. This supports new features over plain branches.
2437
 
 
2438
 
    It has support for a master_branch which is the data for bound branches.
2439
 
    """
2440
 
 
2441
 
    def get_bound_location(self):
2442
 
        try:
2443
 
            return self._transport.get_bytes('bound')[:-1]
2444
 
        except errors.NoSuchFile:
2445
 
            return None
2446
 
 
2447
 
    @needs_read_lock
2448
 
    def get_master_branch(self, possible_transports=None):
2449
 
        """Return the branch we are bound to.
2450
 
 
2451
 
        :return: Either a Branch, or None
2452
 
 
2453
 
        This could memoise the branch, but if thats done
2454
 
        it must be revalidated on each new lock.
2455
 
        So for now we just don't memoise it.
2456
 
        # RBC 20060304 review this decision.
2457
 
        """
2458
 
        bound_loc = self.get_bound_location()
2459
 
        if not bound_loc:
2460
 
            return None
2461
 
        try:
2462
 
            return Branch.open(bound_loc,
2463
 
                               possible_transports=possible_transports)
2464
 
        except (errors.NotBranchError, errors.ConnectionError), e:
2465
 
            raise errors.BoundBranchConnectionFailure(
2466
 
                    self, bound_loc, e)
2467
 
 
2468
2560
    @needs_write_lock
2469
 
    def set_bound_location(self, location):
2470
 
        """Set the target where this branch is bound to.
2471
 
 
2472
 
        :param location: URL to the target branch
2473
 
        """
2474
 
        if location:
2475
 
            self._transport.put_bytes('bound', location+'\n',
2476
 
                mode=self.bzrdir._get_file_mode())
2477
 
        else:
2478
 
            try:
2479
 
                self._transport.delete('bound')
2480
 
            except errors.NoSuchFile:
2481
 
                return False
2482
 
            return True
 
2561
    def unbind(self):
 
2562
        """If bound, unbind"""
 
2563
        return self.set_bound_location(None)
2483
2564
 
2484
2565
    @needs_write_lock
2485
2566
    def bind(self, other):
2507
2588
        # history around
2508
2589
        self.set_bound_location(other.base)
2509
2590
 
 
2591
    def get_bound_location(self):
 
2592
        try:
 
2593
            return self._transport.get_bytes('bound')[:-1]
 
2594
        except errors.NoSuchFile:
 
2595
            return None
 
2596
 
 
2597
    @needs_read_lock
 
2598
    def get_master_branch(self, possible_transports=None):
 
2599
        """Return the branch we are bound to.
 
2600
 
 
2601
        :return: Either a Branch, or None
 
2602
        """
 
2603
        if self._master_branch_cache is None:
 
2604
            self._master_branch_cache = self._get_master_branch(
 
2605
                possible_transports)
 
2606
        return self._master_branch_cache
 
2607
 
 
2608
    def _get_master_branch(self, possible_transports):
 
2609
        bound_loc = self.get_bound_location()
 
2610
        if not bound_loc:
 
2611
            return None
 
2612
        try:
 
2613
            return Branch.open(bound_loc,
 
2614
                               possible_transports=possible_transports)
 
2615
        except (errors.NotBranchError, errors.ConnectionError), e:
 
2616
            raise errors.BoundBranchConnectionFailure(
 
2617
                    self, bound_loc, e)
 
2618
 
2510
2619
    @needs_write_lock
2511
 
    def unbind(self):
2512
 
        """If bound, unbind"""
2513
 
        return self.set_bound_location(None)
 
2620
    def set_bound_location(self, location):
 
2621
        """Set the target where this branch is bound to.
 
2622
 
 
2623
        :param location: URL to the target branch
 
2624
        """
 
2625
        self._master_branch_cache = None
 
2626
        if location:
 
2627
            self._transport.put_bytes('bound', location+'\n',
 
2628
                mode=self.bzrdir._get_file_mode())
 
2629
        else:
 
2630
            try:
 
2631
                self._transport.delete('bound')
 
2632
            except errors.NoSuchFile:
 
2633
                return False
 
2634
            return True
2514
2635
 
2515
2636
    @needs_write_lock
2516
2637
    def update(self, possible_transports=None):
2529
2650
            return old_tip
2530
2651
        return None
2531
2652
 
2532
 
 
2533
 
class BzrBranch8(BzrBranch5):
 
2653
    def _read_last_revision_info(self):
 
2654
        revision_string = self._transport.get_bytes('last-revision')
 
2655
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
 
2656
        revision_id = cache_utf8.get_cached_utf8(revision_id)
 
2657
        revno = int(revno)
 
2658
        return revno, revision_id
 
2659
 
 
2660
    def _write_last_revision_info(self, revno, revision_id):
 
2661
        """Simply write out the revision id, with no checks.
 
2662
 
 
2663
        Use set_last_revision_info to perform this safely.
 
2664
 
 
2665
        Does not update the revision_history cache.
 
2666
        """
 
2667
        revision_id = _mod_revision.ensure_null(revision_id)
 
2668
        out_string = '%d %s\n' % (revno, revision_id)
 
2669
        self._transport.put_bytes('last-revision', out_string,
 
2670
            mode=self.bzrdir._get_file_mode())
 
2671
 
 
2672
    @needs_write_lock
 
2673
    def update_feature_flags(self, updated_flags):
 
2674
        """Update the feature flags for this branch.
 
2675
 
 
2676
        :param updated_flags: Dictionary mapping feature names to necessities
 
2677
            A necessity can be None to indicate the feature should be removed
 
2678
        """
 
2679
        self._format._update_feature_flags(updated_flags)
 
2680
        self.control_transport.put_bytes('format', self._format.as_string())
 
2681
 
 
2682
 
 
2683
class BzrBranch8(BzrBranch):
2534
2684
    """A branch that stores tree-reference locations."""
2535
2685
 
2536
 
    def _open_hook(self):
 
2686
    def _open_hook(self, possible_transports=None):
2537
2687
        if self._ignore_fallbacks:
2538
2688
            return
 
2689
        if possible_transports is None:
 
2690
            possible_transports = [self.bzrdir.root_transport]
2539
2691
        try:
2540
2692
            url = self.get_stacked_on_url()
2541
2693
        except (errors.UnstackableRepositoryFormat, errors.NotStacked,
2549
2701
                    raise AssertionError(
2550
2702
                        "'transform_fallback_location' hook %s returned "
2551
2703
                        "None, not a URL." % hook_name)
2552
 
            self._activate_fallback_location(url)
 
2704
            self._activate_fallback_location(url,
 
2705
                possible_transports=possible_transports)
2553
2706
 
2554
2707
    def __init__(self, *args, **kwargs):
2555
2708
        self._ignore_fallbacks = kwargs.get('ignore_fallbacks', False)
2562
2715
        self._last_revision_info_cache = None
2563
2716
        self._reference_info = None
2564
2717
 
2565
 
    def _last_revision_info(self):
2566
 
        revision_string = self._transport.get_bytes('last-revision')
2567
 
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
2568
 
        revision_id = cache_utf8.get_cached_utf8(revision_id)
2569
 
        revno = int(revno)
2570
 
        return revno, revision_id
2571
 
 
2572
 
    def _write_last_revision_info(self, revno, revision_id):
2573
 
        """Simply write out the revision id, with no checks.
2574
 
 
2575
 
        Use set_last_revision_info to perform this safely.
2576
 
 
2577
 
        Does not update the revision_history cache.
2578
 
        Intended to be called by set_last_revision_info and
2579
 
        _write_revision_history.
2580
 
        """
2581
 
        revision_id = _mod_revision.ensure_null(revision_id)
2582
 
        out_string = '%d %s\n' % (revno, revision_id)
2583
 
        self._transport.put_bytes('last-revision', out_string,
2584
 
            mode=self.bzrdir._get_file_mode())
2585
 
 
2586
 
    @needs_write_lock
2587
 
    def set_last_revision_info(self, revno, revision_id):
2588
 
        revision_id = _mod_revision.ensure_null(revision_id)
2589
 
        old_revno, old_revid = self.last_revision_info()
2590
 
        if self._get_append_revisions_only():
2591
 
            self._check_history_violation(revision_id)
2592
 
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
2593
 
        self._write_last_revision_info(revno, revision_id)
2594
 
        self._clear_cached_state()
2595
 
        self._last_revision_info_cache = revno, revision_id
2596
 
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2597
 
 
2598
 
    def _synchronize_history(self, destination, revision_id):
2599
 
        """Synchronize last revision and revision history between branches.
2600
 
 
2601
 
        :see: Branch._synchronize_history
2602
 
        """
2603
 
        # XXX: The base Branch has a fast implementation of this method based
2604
 
        # on set_last_revision_info, but BzrBranch/BzrBranch5 have a slower one
2605
 
        # that uses set_revision_history.  This class inherits from BzrBranch5,
2606
 
        # but wants the fast implementation, so it calls
2607
 
        # Branch._synchronize_history directly.
2608
 
        Branch._synchronize_history(self, destination, revision_id)
2609
 
 
2610
2718
    def _check_history_violation(self, revision_id):
2611
 
        last_revision = _mod_revision.ensure_null(self.last_revision())
 
2719
        current_revid = self.last_revision()
 
2720
        last_revision = _mod_revision.ensure_null(current_revid)
2612
2721
        if _mod_revision.is_null(last_revision):
2613
2722
            return
2614
 
        if last_revision not in self._lefthand_history(revision_id):
2615
 
            raise errors.AppendRevisionsOnlyViolation(self.base)
 
2723
        graph = self.repository.get_graph()
 
2724
        for lh_ancestor in graph.iter_lefthand_ancestry(revision_id):
 
2725
            if lh_ancestor == current_revid:
 
2726
                return
 
2727
        raise errors.AppendRevisionsOnlyViolation(self.user_url)
2616
2728
 
2617
2729
    def _gen_revision_history(self):
2618
2730
        """Generate the revision history from last revision
2621
2733
        self._extend_partial_history(stop_index=last_revno-1)
2622
2734
        return list(reversed(self._partial_revision_history_cache))
2623
2735
 
2624
 
    def _write_revision_history(self, history):
2625
 
        """Factored out of set_revision_history.
2626
 
 
2627
 
        This performs the actual writing to disk, with format-specific checks.
2628
 
        It is intended to be called by BzrBranch5.set_revision_history.
2629
 
        """
2630
 
        if len(history) == 0:
2631
 
            last_revision = 'null:'
2632
 
        else:
2633
 
            if history != self._lefthand_history(history[-1]):
2634
 
                raise errors.NotLefthandHistory(history)
2635
 
            last_revision = history[-1]
2636
 
        if self._get_append_revisions_only():
2637
 
            self._check_history_violation(last_revision)
2638
 
        self._write_last_revision_info(len(history), last_revision)
2639
 
 
2640
2736
    @needs_write_lock
2641
2737
    def _set_parent_location(self, url):
2642
2738
        """Set the parent branch"""
2718
2814
        if branch_location is None:
2719
2815
            return Branch.reference_parent(self, file_id, path,
2720
2816
                                           possible_transports)
2721
 
        branch_location = urlutils.join(self.base, branch_location)
 
2817
        branch_location = urlutils.join(self.user_url, branch_location)
2722
2818
        return Branch.open(branch_location,
2723
2819
                           possible_transports=possible_transports)
2724
2820
 
2728
2824
 
2729
2825
    def set_bound_location(self, location):
2730
2826
        """See Branch.set_push_location."""
 
2827
        self._master_branch_cache = None
2731
2828
        result = None
2732
 
        config = self.get_config()
 
2829
        conf = self.get_config_stack()
2733
2830
        if location is None:
2734
 
            if config.get_user_option('bound') != 'True':
 
2831
            if not conf.get('bound'):
2735
2832
                return False
2736
2833
            else:
2737
 
                config.set_user_option('bound', 'False', warn_masked=True)
 
2834
                conf.set('bound', 'False')
2738
2835
                return True
2739
2836
        else:
2740
2837
            self._set_config_location('bound_location', location,
2741
 
                                      config=config)
2742
 
            config.set_user_option('bound', 'True', warn_masked=True)
 
2838
                                      config=conf)
 
2839
            conf.set('bound', 'True')
2743
2840
        return True
2744
2841
 
2745
2842
    def _get_bound_location(self, bound):
2746
2843
        """Return the bound location in the config file.
2747
2844
 
2748
2845
        Return None if the bound parameter does not match"""
2749
 
        config = self.get_config()
2750
 
        config_bound = (config.get_user_option('bound') == 'True')
2751
 
        if config_bound != bound:
 
2846
        conf = self.get_config_stack()
 
2847
        if conf.get('bound') != bound:
2752
2848
            return None
2753
 
        return self._get_config_location('bound_location', config=config)
 
2849
        return self._get_config_location('bound_location', config=conf)
2754
2850
 
2755
2851
    def get_bound_location(self):
2756
 
        """See Branch.set_push_location."""
 
2852
        """See Branch.get_bound_location."""
2757
2853
        return self._get_bound_location(True)
2758
2854
 
2759
2855
    def get_old_bound_location(self):
2764
2860
        # you can always ask for the URL; but you might not be able to use it
2765
2861
        # if the repo can't support stacking.
2766
2862
        ## self._check_stackable_repo()
2767
 
        stacked_url = self._get_config_location('stacked_on_location')
 
2863
        # stacked_on_location is only ever defined in branch.conf, so don't
 
2864
        # waste effort reading the whole stack of config files.
 
2865
        conf = _mod_config.BranchOnlyStack(self)
 
2866
        stacked_url = self._get_config_location('stacked_on_location',
 
2867
                                                config=conf)
2768
2868
        if stacked_url is None:
2769
2869
            raise errors.NotStacked(self)
2770
 
        return stacked_url
2771
 
 
2772
 
    def _get_append_revisions_only(self):
2773
 
        value = self.get_config().get_user_option('append_revisions_only')
2774
 
        return value == 'True'
2775
 
 
2776
 
    @needs_write_lock
2777
 
    def generate_revision_history(self, revision_id, last_rev=None,
2778
 
                                  other_branch=None):
2779
 
        """See BzrBranch5.generate_revision_history"""
2780
 
        history = self._lefthand_history(revision_id, last_rev, other_branch)
2781
 
        revno = len(history)
2782
 
        self.set_last_revision_info(revno, revision_id)
 
2870
        return stacked_url.encode('utf-8')
2783
2871
 
2784
2872
    @needs_read_lock
2785
2873
    def get_rev_id(self, revno, history=None):
2810
2898
        try:
2811
2899
            index = self._partial_revision_history_cache.index(revision_id)
2812
2900
        except ValueError:
2813
 
            self._extend_partial_history(stop_revision=revision_id)
 
2901
            try:
 
2902
                self._extend_partial_history(stop_revision=revision_id)
 
2903
            except errors.RevisionNotPresent, e:
 
2904
                raise errors.GhostRevisionsHaveNoRevno(revision_id, e.revision_id)
2814
2905
            index = len(self._partial_revision_history_cache) - 1
 
2906
            if index < 0:
 
2907
                raise errors.NoSuchRevision(self, revision_id)
2815
2908
            if self._partial_revision_history_cache[index] != revision_id:
2816
2909
                raise errors.NoSuchRevision(self, revision_id)
2817
2910
        return self.revno() - index
2839
2932
    """
2840
2933
 
2841
2934
    def get_stacked_on_url(self):
2842
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
 
2935
        raise errors.UnstackableBranchFormat(self._format, self.user_url)
2843
2936
 
2844
2937
 
2845
2938
######################################################################
2869
2962
    :ivar local_branch: target branch if there is a Master, else None
2870
2963
    :ivar target_branch: Target/destination branch object. (write locked)
2871
2964
    :ivar tag_conflicts: A list of tag conflicts, see BasicTags.merge_to
 
2965
    :ivar tag_updates: A dict with new tags, see BasicTags.merge_to
2872
2966
    """
2873
2967
 
2874
 
    def __int__(self):
2875
 
        # DEPRECATED: pull used to return the change in revno
2876
 
        return self.new_revno - self.old_revno
2877
 
 
2878
2968
    def report(self, to_file):
 
2969
        tag_conflicts = getattr(self, "tag_conflicts", None)
 
2970
        tag_updates = getattr(self, "tag_updates", None)
2879
2971
        if not is_quiet():
2880
 
            if self.old_revid == self.new_revid:
2881
 
                to_file.write('No revisions to pull.\n')
2882
 
            else:
 
2972
            if self.old_revid != self.new_revid:
2883
2973
                to_file.write('Now on revision %d.\n' % self.new_revno)
 
2974
            if tag_updates:
 
2975
                to_file.write('%d tag(s) updated.\n' % len(tag_updates))
 
2976
            if self.old_revid == self.new_revid and not tag_updates:
 
2977
                if not tag_conflicts:
 
2978
                    to_file.write('No revisions or tags to pull.\n')
 
2979
                else:
 
2980
                    to_file.write('No revisions to pull.\n')
2884
2981
        self._show_tag_conficts(to_file)
2885
2982
 
2886
2983
 
2903
3000
        target, otherwise it will be None.
2904
3001
    """
2905
3002
 
2906
 
    def __int__(self):
2907
 
        # DEPRECATED: push used to return the change in revno
2908
 
        return self.new_revno - self.old_revno
2909
 
 
2910
3003
    def report(self, to_file):
2911
 
        """Write a human-readable description of the result."""
2912
 
        if self.old_revid == self.new_revid:
2913
 
            note('No new revisions to push.')
2914
 
        else:
2915
 
            note('Pushed up to revision %d.' % self.new_revno)
 
3004
        # TODO: This function gets passed a to_file, but then
 
3005
        # ignores it and calls note() instead. This is also
 
3006
        # inconsistent with PullResult(), which writes to stdout.
 
3007
        # -- JRV20110901, bug #838853
 
3008
        tag_conflicts = getattr(self, "tag_conflicts", None)
 
3009
        tag_updates = getattr(self, "tag_updates", None)
 
3010
        if not is_quiet():
 
3011
            if self.old_revid != self.new_revid:
 
3012
                note(gettext('Pushed up to revision %d.') % self.new_revno)
 
3013
            if tag_updates:
 
3014
                note(ngettext('%d tag updated.', '%d tags updated.', len(tag_updates)) % len(tag_updates))
 
3015
            if self.old_revid == self.new_revid and not tag_updates:
 
3016
                if not tag_conflicts:
 
3017
                    note(gettext('No new revisions or tags to push.'))
 
3018
                else:
 
3019
                    note(gettext('No new revisions to push.'))
2916
3020
        self._show_tag_conficts(to_file)
2917
3021
 
2918
3022
 
2932
3036
        :param verbose: Requests more detailed display of what was checked,
2933
3037
            if any.
2934
3038
        """
2935
 
        note('checked branch %s format %s', self.branch.base,
2936
 
            self.branch._format)
 
3039
        note(gettext('checked branch {0} format {1}').format(
 
3040
                                self.branch.user_url, self.branch._format))
2937
3041
        for error in self.errors:
2938
 
            note('found error:%s', error)
 
3042
            note(gettext('found error:%s'), error)
2939
3043
 
2940
3044
 
2941
3045
class Converter5to6(object):
2948
3052
 
2949
3053
        # Copy source data into target
2950
3054
        new_branch._write_last_revision_info(*branch.last_revision_info())
2951
 
        new_branch.set_parent(branch.get_parent())
2952
 
        new_branch.set_bound_location(branch.get_bound_location())
2953
 
        new_branch.set_push_location(branch.get_push_location())
 
3055
        new_branch.lock_write()
 
3056
        try:
 
3057
            new_branch.set_parent(branch.get_parent())
 
3058
            new_branch.set_bound_location(branch.get_bound_location())
 
3059
            new_branch.set_push_location(branch.get_push_location())
 
3060
        finally:
 
3061
            new_branch.unlock()
2954
3062
 
2955
3063
        # New branch has no tags by default
2956
3064
        new_branch.tags._set_tag_dict({})
2957
3065
 
2958
3066
        # Copying done; now update target format
2959
3067
        new_branch._transport.put_bytes('format',
2960
 
            format.get_format_string(),
 
3068
            format.as_string(),
2961
3069
            mode=new_branch.bzrdir._get_file_mode())
2962
3070
 
2963
3071
        # Clean up old files
2964
3072
        new_branch._transport.delete('revision-history')
 
3073
        branch.lock_write()
2965
3074
        try:
2966
 
            branch.set_parent(None)
2967
 
        except errors.NoSuchFile:
2968
 
            pass
2969
 
        branch.set_bound_location(None)
 
3075
            try:
 
3076
                branch.set_parent(None)
 
3077
            except errors.NoSuchFile:
 
3078
                pass
 
3079
            branch.set_bound_location(None)
 
3080
        finally:
 
3081
            branch.unlock()
2970
3082
 
2971
3083
 
2972
3084
class Converter6to7(object):
2976
3088
        format = BzrBranchFormat7()
2977
3089
        branch._set_config_location('stacked_on_location', '')
2978
3090
        # update target format
2979
 
        branch._transport.put_bytes('format', format.get_format_string())
 
3091
        branch._transport.put_bytes('format', format.as_string())
2980
3092
 
2981
3093
 
2982
3094
class Converter7to8(object):
2983
 
    """Perform an in-place upgrade of format 6 to format 7"""
 
3095
    """Perform an in-place upgrade of format 7 to format 8"""
2984
3096
 
2985
3097
    def convert(self, branch):
2986
3098
        format = BzrBranchFormat8()
2987
3099
        branch._transport.put_bytes('references', '')
2988
3100
        # update target format
2989
 
        branch._transport.put_bytes('format', format.get_format_string())
2990
 
 
2991
 
 
2992
 
def _run_with_write_locked_target(target, callable, *args, **kwargs):
2993
 
    """Run ``callable(*args, **kwargs)``, write-locking target for the
2994
 
    duration.
2995
 
 
2996
 
    _run_with_write_locked_target will attempt to release the lock it acquires.
2997
 
 
2998
 
    If an exception is raised by callable, then that exception *will* be
2999
 
    propagated, even if the unlock attempt raises its own error.  Thus
3000
 
    _run_with_write_locked_target should be preferred to simply doing::
3001
 
 
3002
 
        target.lock_write()
3003
 
        try:
3004
 
            return callable(*args, **kwargs)
3005
 
        finally:
3006
 
            target.unlock()
3007
 
 
3008
 
    """
3009
 
    # This is very similar to bzrlib.decorators.needs_write_lock.  Perhaps they
3010
 
    # should share code?
3011
 
    target.lock_write()
3012
 
    try:
3013
 
        result = callable(*args, **kwargs)
3014
 
    except:
3015
 
        exc_info = sys.exc_info()
3016
 
        try:
3017
 
            target.unlock()
3018
 
        finally:
3019
 
            raise exc_info[0], exc_info[1], exc_info[2]
3020
 
    else:
3021
 
        target.unlock()
3022
 
        return result
 
3101
        branch._transport.put_bytes('format', format.as_string())
3023
3102
 
3024
3103
 
3025
3104
class InterBranch(InterObject):
3033
3112
    _optimisers = []
3034
3113
    """The available optimised InterBranch types."""
3035
3114
 
3036
 
    @staticmethod
3037
 
    def _get_branch_formats_to_test():
3038
 
        """Return a tuple with the Branch formats to use when testing."""
3039
 
        raise NotImplementedError(InterBranch._get_branch_formats_to_test)
 
3115
    @classmethod
 
3116
    def _get_branch_formats_to_test(klass):
 
3117
        """Return an iterable of format tuples for testing.
 
3118
        
 
3119
        :return: An iterable of (from_format, to_format) to use when testing
 
3120
            this InterBranch class. Each InterBranch class should define this
 
3121
            method itself.
 
3122
        """
 
3123
        raise NotImplementedError(klass._get_branch_formats_to_test)
3040
3124
 
 
3125
    @needs_write_lock
3041
3126
    def pull(self, overwrite=False, stop_revision=None,
3042
3127
             possible_transports=None, local=False):
3043
3128
        """Mirror source into target branch.
3048
3133
        """
3049
3134
        raise NotImplementedError(self.pull)
3050
3135
 
3051
 
    def update_revisions(self, stop_revision=None, overwrite=False,
3052
 
                         graph=None):
3053
 
        """Pull in new perfect-fit revisions.
3054
 
 
3055
 
        :param stop_revision: Updated until the given revision
3056
 
        :param overwrite: Always set the branch pointer, rather than checking
3057
 
            to see if it is a proper descendant.
3058
 
        :param graph: A Graph object that can be used to query history
3059
 
            information. This can be None.
3060
 
        :return: None
3061
 
        """
3062
 
        raise NotImplementedError(self.update_revisions)
3063
 
 
3064
 
    def push(self, overwrite=False, stop_revision=None,
 
3136
    @needs_write_lock
 
3137
    def push(self, overwrite=False, stop_revision=None, lossy=False,
3065
3138
             _override_hook_source_branch=None):
3066
3139
        """Mirror the source branch into the target branch.
3067
3140
 
3069
3142
        """
3070
3143
        raise NotImplementedError(self.push)
3071
3144
 
 
3145
    @needs_write_lock
 
3146
    def copy_content_into(self, revision_id=None):
 
3147
        """Copy the content of source into target
 
3148
 
 
3149
        revision_id: if not None, the revision history in the new branch will
 
3150
                     be truncated to end with revision_id.
 
3151
        """
 
3152
        raise NotImplementedError(self.copy_content_into)
 
3153
 
 
3154
    @needs_write_lock
 
3155
    def fetch(self, stop_revision=None, limit=None):
 
3156
        """Fetch revisions.
 
3157
 
 
3158
        :param stop_revision: Last revision to fetch
 
3159
        :param limit: Optional rough limit of revisions to fetch
 
3160
        """
 
3161
        raise NotImplementedError(self.fetch)
 
3162
 
 
3163
 
 
3164
def _fix_overwrite_type(overwrite):
 
3165
    if isinstance(overwrite, bool):
 
3166
        if overwrite:
 
3167
            return ["history", "tags"]
 
3168
        else:
 
3169
            return []
 
3170
    return overwrite
 
3171
 
3072
3172
 
3073
3173
class GenericInterBranch(InterBranch):
3074
 
    """InterBranch implementation that uses public Branch functions.
3075
 
    """
3076
 
 
3077
 
    @staticmethod
3078
 
    def _get_branch_formats_to_test():
3079
 
        return BranchFormat._default_format, BranchFormat._default_format
3080
 
 
3081
 
    def update_revisions(self, stop_revision=None, overwrite=False,
3082
 
        graph=None):
3083
 
        """See InterBranch.update_revisions()."""
 
3174
    """InterBranch implementation that uses public Branch functions."""
 
3175
 
 
3176
    @classmethod
 
3177
    def is_compatible(klass, source, target):
 
3178
        # GenericBranch uses the public API, so always compatible
 
3179
        return True
 
3180
 
 
3181
    @classmethod
 
3182
    def _get_branch_formats_to_test(klass):
 
3183
        return [(format_registry.get_default(), format_registry.get_default())]
 
3184
 
 
3185
    @classmethod
 
3186
    def unwrap_format(klass, format):
 
3187
        if isinstance(format, remote.RemoteBranchFormat):
 
3188
            format._ensure_real()
 
3189
            return format._custom_format
 
3190
        return format
 
3191
 
 
3192
    @needs_write_lock
 
3193
    def copy_content_into(self, revision_id=None):
 
3194
        """Copy the content of source into target
 
3195
 
 
3196
        revision_id: if not None, the revision history in the new branch will
 
3197
                     be truncated to end with revision_id.
 
3198
        """
 
3199
        self.source.update_references(self.target)
 
3200
        self.source._synchronize_history(self.target, revision_id)
 
3201
        try:
 
3202
            parent = self.source.get_parent()
 
3203
        except errors.InaccessibleParent, e:
 
3204
            mutter('parent was not accessible to copy: %s', e)
 
3205
        else:
 
3206
            if parent:
 
3207
                self.target.set_parent(parent)
 
3208
        if self.source._push_should_merge_tags():
 
3209
            self.source.tags.merge_to(self.target.tags)
 
3210
 
 
3211
    @needs_write_lock
 
3212
    def fetch(self, stop_revision=None, limit=None):
 
3213
        if self.target.base == self.source.base:
 
3214
            return (0, [])
3084
3215
        self.source.lock_read()
3085
3216
        try:
3086
 
            other_revno, other_last_revision = self.source.last_revision_info()
3087
 
            stop_revno = None # unknown
3088
 
            if stop_revision is None:
3089
 
                stop_revision = other_last_revision
3090
 
                if _mod_revision.is_null(stop_revision):
3091
 
                    # if there are no commits, we're done.
3092
 
                    return
3093
 
                stop_revno = other_revno
3094
 
 
3095
 
            # what's the current last revision, before we fetch [and change it
3096
 
            # possibly]
3097
 
            last_rev = _mod_revision.ensure_null(self.target.last_revision())
3098
 
            # we fetch here so that we don't process data twice in the common
3099
 
            # case of having something to pull, and so that the check for
3100
 
            # already merged can operate on the just fetched graph, which will
3101
 
            # be cached in memory.
3102
 
            self.target.fetch(self.source, stop_revision)
3103
 
            # Check to see if one is an ancestor of the other
3104
 
            if not overwrite:
3105
 
                if graph is None:
3106
 
                    graph = self.target.repository.get_graph()
3107
 
                if self.target._check_if_descendant_or_diverged(
3108
 
                        stop_revision, last_rev, graph, self.source):
3109
 
                    # stop_revision is a descendant of last_rev, but we aren't
3110
 
                    # overwriting, so we're done.
3111
 
                    return
3112
 
            if stop_revno is None:
3113
 
                if graph is None:
3114
 
                    graph = self.target.repository.get_graph()
3115
 
                this_revno, this_last_revision = \
3116
 
                        self.target.last_revision_info()
3117
 
                stop_revno = graph.find_distance_to_null(stop_revision,
3118
 
                                [(other_last_revision, other_revno),
3119
 
                                 (this_last_revision, this_revno)])
3120
 
            self.target.set_last_revision_info(stop_revno, stop_revision)
 
3217
            fetch_spec_factory = fetch.FetchSpecFactory()
 
3218
            fetch_spec_factory.source_branch = self.source
 
3219
            fetch_spec_factory.source_branch_stop_revision_id = stop_revision
 
3220
            fetch_spec_factory.source_repo = self.source.repository
 
3221
            fetch_spec_factory.target_repo = self.target.repository
 
3222
            fetch_spec_factory.target_repo_kind = fetch.TargetRepoKinds.PREEXISTING
 
3223
            fetch_spec_factory.limit = limit
 
3224
            fetch_spec = fetch_spec_factory.make_fetch_spec()
 
3225
            return self.target.repository.fetch(self.source.repository,
 
3226
                fetch_spec=fetch_spec)
3121
3227
        finally:
3122
3228
            self.source.unlock()
3123
3229
 
 
3230
    @needs_write_lock
 
3231
    def _update_revisions(self, stop_revision=None, overwrite=False,
 
3232
            graph=None):
 
3233
        other_revno, other_last_revision = self.source.last_revision_info()
 
3234
        stop_revno = None # unknown
 
3235
        if stop_revision is None:
 
3236
            stop_revision = other_last_revision
 
3237
            if _mod_revision.is_null(stop_revision):
 
3238
                # if there are no commits, we're done.
 
3239
                return
 
3240
            stop_revno = other_revno
 
3241
 
 
3242
        # what's the current last revision, before we fetch [and change it
 
3243
        # possibly]
 
3244
        last_rev = _mod_revision.ensure_null(self.target.last_revision())
 
3245
        # we fetch here so that we don't process data twice in the common
 
3246
        # case of having something to pull, and so that the check for
 
3247
        # already merged can operate on the just fetched graph, which will
 
3248
        # be cached in memory.
 
3249
        self.fetch(stop_revision=stop_revision)
 
3250
        # Check to see if one is an ancestor of the other
 
3251
        if not overwrite:
 
3252
            if graph is None:
 
3253
                graph = self.target.repository.get_graph()
 
3254
            if self.target._check_if_descendant_or_diverged(
 
3255
                    stop_revision, last_rev, graph, self.source):
 
3256
                # stop_revision is a descendant of last_rev, but we aren't
 
3257
                # overwriting, so we're done.
 
3258
                return
 
3259
        if stop_revno is None:
 
3260
            if graph is None:
 
3261
                graph = self.target.repository.get_graph()
 
3262
            this_revno, this_last_revision = \
 
3263
                    self.target.last_revision_info()
 
3264
            stop_revno = graph.find_distance_to_null(stop_revision,
 
3265
                            [(other_last_revision, other_revno),
 
3266
                             (this_last_revision, this_revno)])
 
3267
        self.target.set_last_revision_info(stop_revno, stop_revision)
 
3268
 
 
3269
    @needs_write_lock
3124
3270
    def pull(self, overwrite=False, stop_revision=None,
 
3271
             possible_transports=None, run_hooks=True,
 
3272
             _override_hook_target=None, local=False):
 
3273
        """Pull from source into self, updating my master if any.
 
3274
 
 
3275
        :param run_hooks: Private parameter - if false, this branch
 
3276
            is being called because it's the master of the primary branch,
 
3277
            so it should not run its hooks.
 
3278
        """
 
3279
        bound_location = self.target.get_bound_location()
 
3280
        if local and not bound_location:
 
3281
            raise errors.LocalRequiresBoundBranch()
 
3282
        master_branch = None
 
3283
        source_is_master = False
 
3284
        if bound_location:
 
3285
            # bound_location comes from a config file, some care has to be
 
3286
            # taken to relate it to source.user_url
 
3287
            normalized = urlutils.normalize_url(bound_location)
 
3288
            try:
 
3289
                relpath = self.source.user_transport.relpath(normalized)
 
3290
                source_is_master = (relpath == '')
 
3291
            except (errors.PathNotChild, errors.InvalidURL):
 
3292
                source_is_master = False
 
3293
        if not local and bound_location and not source_is_master:
 
3294
            # not pulling from master, so we need to update master.
 
3295
            master_branch = self.target.get_master_branch(possible_transports)
 
3296
            master_branch.lock_write()
 
3297
        try:
 
3298
            if master_branch:
 
3299
                # pull from source into master.
 
3300
                master_branch.pull(self.source, overwrite, stop_revision,
 
3301
                    run_hooks=False)
 
3302
            return self._pull(overwrite,
 
3303
                stop_revision, _hook_master=master_branch,
 
3304
                run_hooks=run_hooks,
 
3305
                _override_hook_target=_override_hook_target,
 
3306
                merge_tags_to_master=not source_is_master)
 
3307
        finally:
 
3308
            if master_branch:
 
3309
                master_branch.unlock()
 
3310
 
 
3311
    def push(self, overwrite=False, stop_revision=None, lossy=False,
 
3312
             _override_hook_source_branch=None):
 
3313
        """See InterBranch.push.
 
3314
 
 
3315
        This is the basic concrete implementation of push()
 
3316
 
 
3317
        :param _override_hook_source_branch: If specified, run the hooks
 
3318
            passing this Branch as the source, rather than self.  This is for
 
3319
            use of RemoteBranch, where push is delegated to the underlying
 
3320
            vfs-based Branch.
 
3321
        """
 
3322
        if lossy:
 
3323
            raise errors.LossyPushToSameVCS(self.source, self.target)
 
3324
        # TODO: Public option to disable running hooks - should be trivial but
 
3325
        # needs tests.
 
3326
 
 
3327
        op = cleanup.OperationWithCleanups(self._push_with_bound_branches)
 
3328
        op.add_cleanup(self.source.lock_read().unlock)
 
3329
        op.add_cleanup(self.target.lock_write().unlock)
 
3330
        return op.run(overwrite, stop_revision,
 
3331
            _override_hook_source_branch=_override_hook_source_branch)
 
3332
 
 
3333
    def _basic_push(self, overwrite, stop_revision):
 
3334
        """Basic implementation of push without bound branches or hooks.
 
3335
 
 
3336
        Must be called with source read locked and target write locked.
 
3337
        """
 
3338
        result = BranchPushResult()
 
3339
        result.source_branch = self.source
 
3340
        result.target_branch = self.target
 
3341
        result.old_revno, result.old_revid = self.target.last_revision_info()
 
3342
        self.source.update_references(self.target)
 
3343
        overwrite = _fix_overwrite_type(overwrite)
 
3344
        if result.old_revid != stop_revision:
 
3345
            # We assume that during 'push' this repository is closer than
 
3346
            # the target.
 
3347
            graph = self.source.repository.get_graph(self.target.repository)
 
3348
            self._update_revisions(stop_revision,
 
3349
                overwrite=("history" in overwrite),
 
3350
                graph=graph)
 
3351
        if self.source._push_should_merge_tags():
 
3352
            result.tag_updates, result.tag_conflicts = (
 
3353
                self.source.tags.merge_to(
 
3354
                self.target.tags, "tags" in overwrite))
 
3355
        result.new_revno, result.new_revid = self.target.last_revision_info()
 
3356
        return result
 
3357
 
 
3358
    def _push_with_bound_branches(self, operation, overwrite, stop_revision,
 
3359
            _override_hook_source_branch=None):
 
3360
        """Push from source into target, and into target's master if any.
 
3361
        """
 
3362
        def _run_hooks():
 
3363
            if _override_hook_source_branch:
 
3364
                result.source_branch = _override_hook_source_branch
 
3365
            for hook in Branch.hooks['post_push']:
 
3366
                hook(result)
 
3367
 
 
3368
        bound_location = self.target.get_bound_location()
 
3369
        if bound_location and self.target.base != bound_location:
 
3370
            # there is a master branch.
 
3371
            #
 
3372
            # XXX: Why the second check?  Is it even supported for a branch to
 
3373
            # be bound to itself? -- mbp 20070507
 
3374
            master_branch = self.target.get_master_branch()
 
3375
            master_branch.lock_write()
 
3376
            operation.add_cleanup(master_branch.unlock)
 
3377
            # push into the master from the source branch.
 
3378
            master_inter = InterBranch.get(self.source, master_branch)
 
3379
            master_inter._basic_push(overwrite, stop_revision)
 
3380
            # and push into the target branch from the source. Note that
 
3381
            # we push from the source branch again, because it's considered
 
3382
            # the highest bandwidth repository.
 
3383
            result = self._basic_push(overwrite, stop_revision)
 
3384
            result.master_branch = master_branch
 
3385
            result.local_branch = self.target
 
3386
        else:
 
3387
            master_branch = None
 
3388
            # no master branch
 
3389
            result = self._basic_push(overwrite, stop_revision)
 
3390
            # TODO: Why set master_branch and local_branch if there's no
 
3391
            # binding?  Maybe cleaner to just leave them unset? -- mbp
 
3392
            # 20070504
 
3393
            result.master_branch = self.target
 
3394
            result.local_branch = None
 
3395
        _run_hooks()
 
3396
        return result
 
3397
 
 
3398
    def _pull(self, overwrite=False, stop_revision=None,
3125
3399
             possible_transports=None, _hook_master=None, run_hooks=True,
3126
 
             _override_hook_target=None, local=False):
 
3400
             _override_hook_target=None, local=False,
 
3401
             merge_tags_to_master=True):
3127
3402
        """See Branch.pull.
3128
3403
 
 
3404
        This function is the core worker, used by GenericInterBranch.pull to
 
3405
        avoid duplication when pulling source->master and source->local.
 
3406
 
3129
3407
        :param _hook_master: Private parameter - set the branch to
3130
3408
            be supplied as the master to pull hooks.
3131
3409
        :param run_hooks: Private parameter - if false, this branch
3132
3410
            is being called because it's the master of the primary branch,
3133
3411
            so it should not run its hooks.
 
3412
            is being called because it's the master of the primary branch,
 
3413
            so it should not run its hooks.
3134
3414
        :param _override_hook_target: Private parameter - set the branch to be
3135
3415
            supplied as the target_branch to pull hooks.
3136
3416
        :param local: Only update the local branch, and not the bound branch.
3155
3435
            # -- JRV20090506
3156
3436
            result.old_revno, result.old_revid = \
3157
3437
                self.target.last_revision_info()
3158
 
            self.target.update_revisions(self.source, stop_revision,
3159
 
                overwrite=overwrite, graph=graph)
 
3438
            overwrite = _fix_overwrite_type(overwrite)
 
3439
            self._update_revisions(stop_revision,
 
3440
                overwrite=("history" in overwrite),
 
3441
                graph=graph)
3160
3442
            # TODO: The old revid should be specified when merging tags, 
3161
3443
            # so a tags implementation that versions tags can only 
3162
3444
            # pull in the most recent changes. -- JRV20090506
3163
 
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
3164
 
                overwrite)
 
3445
            result.tag_updates, result.tag_conflicts = (
 
3446
                self.source.tags.merge_to(self.target.tags,
 
3447
                    "tags" in overwrite,
 
3448
                    ignore_master=not merge_tags_to_master))
3165
3449
            result.new_revno, result.new_revid = self.target.last_revision_info()
3166
3450
            if _hook_master:
3167
3451
                result.master_branch = _hook_master
3176
3460
            self.source.unlock()
3177
3461
        return result
3178
3462
 
3179
 
    def push(self, overwrite=False, stop_revision=None,
3180
 
             _override_hook_source_branch=None):
3181
 
        """See InterBranch.push.
3182
 
 
3183
 
        This is the basic concrete implementation of push()
3184
 
 
3185
 
        :param _override_hook_source_branch: If specified, run
3186
 
        the hooks passing this Branch as the source, rather than self.
3187
 
        This is for use of RemoteBranch, where push is delegated to the
3188
 
        underlying vfs-based Branch.
3189
 
        """
3190
 
        # TODO: Public option to disable running hooks - should be trivial but
3191
 
        # needs tests.
3192
 
        self.source.lock_read()
3193
 
        try:
3194
 
            return _run_with_write_locked_target(
3195
 
                self.target, self._push_with_bound_branches, overwrite,
3196
 
                stop_revision,
3197
 
                _override_hook_source_branch=_override_hook_source_branch)
3198
 
        finally:
3199
 
            self.source.unlock()
3200
 
 
3201
 
    def _push_with_bound_branches(self, overwrite, stop_revision,
3202
 
            _override_hook_source_branch=None):
3203
 
        """Push from source into target, and into target's master if any.
3204
 
        """
3205
 
        def _run_hooks():
3206
 
            if _override_hook_source_branch:
3207
 
                result.source_branch = _override_hook_source_branch
3208
 
            for hook in Branch.hooks['post_push']:
3209
 
                hook(result)
3210
 
 
3211
 
        bound_location = self.target.get_bound_location()
3212
 
        if bound_location and self.target.base != bound_location:
3213
 
            # there is a master branch.
3214
 
            #
3215
 
            # XXX: Why the second check?  Is it even supported for a branch to
3216
 
            # be bound to itself? -- mbp 20070507
3217
 
            master_branch = self.target.get_master_branch()
3218
 
            master_branch.lock_write()
3219
 
            try:
3220
 
                # push into the master from the source branch.
3221
 
                self.source._basic_push(master_branch, overwrite, stop_revision)
3222
 
                # and push into the target branch from the source. Note that we
3223
 
                # push from the source branch again, because its considered the
3224
 
                # highest bandwidth repository.
3225
 
                result = self.source._basic_push(self.target, overwrite,
3226
 
                    stop_revision)
3227
 
                result.master_branch = master_branch
3228
 
                result.local_branch = self.target
3229
 
                _run_hooks()
3230
 
                return result
3231
 
            finally:
3232
 
                master_branch.unlock()
3233
 
        else:
3234
 
            # no master branch
3235
 
            result = self.source._basic_push(self.target, overwrite,
3236
 
                stop_revision)
3237
 
            # TODO: Why set master_branch and local_branch if there's no
3238
 
            # binding?  Maybe cleaner to just leave them unset? -- mbp
3239
 
            # 20070504
3240
 
            result.master_branch = self.target
3241
 
            result.local_branch = None
3242
 
            _run_hooks()
3243
 
            return result
3244
 
 
3245
 
    @classmethod
3246
 
    def is_compatible(self, source, target):
3247
 
        # GenericBranch uses the public API, so always compatible
3248
 
        return True
3249
 
 
3250
 
 
3251
 
class InterToBranch5(GenericInterBranch):
3252
 
 
3253
 
    @staticmethod
3254
 
    def _get_branch_formats_to_test():
3255
 
        return BranchFormat._default_format, BzrBranchFormat5()
3256
 
 
3257
 
    def pull(self, overwrite=False, stop_revision=None,
3258
 
             possible_transports=None, run_hooks=True,
3259
 
             _override_hook_target=None, local=False):
3260
 
        """Pull from source into self, updating my master if any.
3261
 
 
3262
 
        :param run_hooks: Private parameter - if false, this branch
3263
 
            is being called because it's the master of the primary branch,
3264
 
            so it should not run its hooks.
3265
 
        """
3266
 
        bound_location = self.target.get_bound_location()
3267
 
        if local and not bound_location:
3268
 
            raise errors.LocalRequiresBoundBranch()
3269
 
        master_branch = None
3270
 
        if not local and bound_location and self.source.base != bound_location:
3271
 
            # not pulling from master, so we need to update master.
3272
 
            master_branch = self.target.get_master_branch(possible_transports)
3273
 
            master_branch.lock_write()
3274
 
        try:
3275
 
            if master_branch:
3276
 
                # pull from source into master.
3277
 
                master_branch.pull(self.source, overwrite, stop_revision,
3278
 
                    run_hooks=False)
3279
 
            return super(InterToBranch5, self).pull(overwrite,
3280
 
                stop_revision, _hook_master=master_branch,
3281
 
                run_hooks=run_hooks,
3282
 
                _override_hook_target=_override_hook_target)
3283
 
        finally:
3284
 
            if master_branch:
3285
 
                master_branch.unlock()
3286
 
 
3287
3463
 
3288
3464
InterBranch.register_optimiser(GenericInterBranch)
3289
 
InterBranch.register_optimiser(InterToBranch5)