~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-29 22:03:03 UTC
  • mfrom: (5416.2.6 jam-integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100929220303-cr95h8iwtggco721
(mbp) Add 'break-lock --force'

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2008 Canonical Ltd
 
1
# Copyright (C) 2005-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
 
18
from cStringIO import StringIO
 
19
import sys
17
20
 
18
21
from bzrlib.lazy_import import lazy_import
19
22
lazy_import(globals(), """
 
23
from itertools import chain
20
24
from bzrlib import (
21
25
        bzrdir,
22
26
        cache_utf8,
23
27
        config as _mod_config,
 
28
        controldir,
24
29
        debug,
25
30
        errors,
26
31
        lockdir,
27
32
        lockable_files,
 
33
        remote,
28
34
        repository,
29
35
        revision as _mod_revision,
 
36
        rio,
 
37
        symbol_versioning,
30
38
        transport,
31
39
        tsort,
32
40
        ui,
33
41
        urlutils,
34
42
        )
35
 
from bzrlib.config import BranchConfig
 
43
from bzrlib.config import BranchConfig, TransportConfig
 
44
from bzrlib.repofmt.pack_repo import RepositoryFormatKnitPack5RichRoot
36
45
from bzrlib.tag import (
37
46
    BasicTags,
38
47
    DisabledTags,
39
48
    )
40
49
""")
41
50
 
42
 
from bzrlib.decorators import needs_read_lock, needs_write_lock
43
 
from bzrlib.hooks import Hooks
 
51
from bzrlib.decorators import needs_read_lock, needs_write_lock, only_raises
 
52
from bzrlib.hooks import HookPoint, Hooks
 
53
from bzrlib.inter import InterObject
 
54
from bzrlib.lock import _RelockDebugMixin, LogicalLockResult
 
55
from bzrlib import registry
44
56
from bzrlib.symbol_versioning import (
45
57
    deprecated_in,
46
58
    deprecated_method,
53
65
BZR_BRANCH_FORMAT_6 = "Bazaar Branch Format 6 (bzr 0.15)\n"
54
66
 
55
67
 
56
 
# TODO: Maybe include checks for common corruption of newlines, etc?
57
 
 
58
 
# TODO: Some operations like log might retrieve the same revisions
59
 
# repeatedly to calculate deltas.  We could perhaps have a weakref
60
 
# cache in memory to make this faster.  In general anything can be
61
 
# cached in memory between lock and unlock operations. .. nb thats
62
 
# what the transaction identity map provides
63
 
 
64
 
 
65
 
######################################################################
66
 
# branch objects
67
 
 
68
 
class Branch(object):
 
68
class Branch(controldir.ControlComponent):
69
69
    """Branch holding a history of revisions.
70
70
 
71
 
    base
72
 
        Base directory/url of the branch.
 
71
    :ivar base:
 
72
        Base directory/url of the branch; using control_url and
 
73
        control_transport is more standardized.
73
74
 
74
75
    hooks: An instance of BranchHooks.
75
76
    """
77
78
    # - RBC 20060112
78
79
    base = None
79
80
 
80
 
    # override this to set the strategy for storing tags
81
 
    def _make_tags(self):
82
 
        return DisabledTags(self)
 
81
    @property
 
82
    def control_transport(self):
 
83
        return self._transport
 
84
 
 
85
    @property
 
86
    def user_transport(self):
 
87
        return self.bzrdir.user_transport
83
88
 
84
89
    def __init__(self, *ignored, **ignored_too):
85
 
        self.tags = self._make_tags()
 
90
        self.tags = self._format.make_tags(self)
86
91
        self._revision_history_cache = None
87
92
        self._revision_id_to_revno_cache = None
 
93
        self._partial_revision_id_to_revno_cache = {}
 
94
        self._partial_revision_history_cache = []
88
95
        self._last_revision_info_cache = None
 
96
        self._merge_sorted_revisions_cache = None
 
97
        self._open_hook()
 
98
        hooks = Branch.hooks['open']
 
99
        for hook in hooks:
 
100
            hook(self)
 
101
 
 
102
    def _open_hook(self):
 
103
        """Called by init to allow simpler extension of the base class."""
 
104
 
 
105
    def _activate_fallback_location(self, url):
 
106
        """Activate the branch/repository from url as a fallback repository."""
 
107
        repo = self._get_fallback_repository(url)
 
108
        if repo.has_same_location(self.repository):
 
109
            raise errors.UnstackableLocationError(self.user_url, url)
 
110
        self.repository.add_fallback_repository(repo)
89
111
 
90
112
    def break_lock(self):
91
113
        """Break a lock if one is present from another instance.
101
123
        if master is not None:
102
124
            master.break_lock()
103
125
 
 
126
    def _check_stackable_repo(self):
 
127
        if not self.repository._format.supports_external_lookups:
 
128
            raise errors.UnstackableRepositoryFormat(self.repository._format,
 
129
                self.repository.base)
 
130
 
 
131
    def _extend_partial_history(self, stop_index=None, stop_revision=None):
 
132
        """Extend the partial history to include a given index
 
133
 
 
134
        If a stop_index is supplied, stop when that index has been reached.
 
135
        If a stop_revision is supplied, stop when that revision is
 
136
        encountered.  Otherwise, stop when the beginning of history is
 
137
        reached.
 
138
 
 
139
        :param stop_index: The index which should be present.  When it is
 
140
            present, history extension will stop.
 
141
        :param stop_revision: The revision id which should be present.  When
 
142
            it is encountered, history extension will stop.
 
143
        """
 
144
        if len(self._partial_revision_history_cache) == 0:
 
145
            self._partial_revision_history_cache = [self.last_revision()]
 
146
        repository._iter_for_revno(
 
147
            self.repository, self._partial_revision_history_cache,
 
148
            stop_index=stop_index, stop_revision=stop_revision)
 
149
        if self._partial_revision_history_cache[-1] == _mod_revision.NULL_REVISION:
 
150
            self._partial_revision_history_cache.pop()
 
151
 
 
152
    def _get_check_refs(self):
 
153
        """Get the references needed for check().
 
154
 
 
155
        See bzrlib.check.
 
156
        """
 
157
        revid = self.last_revision()
 
158
        return [('revision-existence', revid), ('lefthand-distance', revid)]
 
159
 
104
160
    @staticmethod
105
161
    def open(base, _unsupported=False, possible_transports=None):
106
162
        """Open the branch rooted at base.
110
166
        """
111
167
        control = bzrdir.BzrDir.open(base, _unsupported,
112
168
                                     possible_transports=possible_transports)
113
 
        return control.open_branch(_unsupported)
 
169
        return control.open_branch(unsupported=_unsupported)
114
170
 
115
171
    @staticmethod
116
 
    def open_from_transport(transport, _unsupported=False):
 
172
    def open_from_transport(transport, name=None, _unsupported=False):
117
173
        """Open the branch rooted at transport"""
118
174
        control = bzrdir.BzrDir.open_from_transport(transport, _unsupported)
119
 
        return control.open_branch(_unsupported)
 
175
        return control.open_branch(name=name, unsupported=_unsupported)
120
176
 
121
177
    @staticmethod
122
178
    def open_containing(url, possible_transports=None):
123
179
        """Open an existing branch which contains url.
124
 
        
 
180
 
125
181
        This probes for a branch at url, and searches upwards from there.
126
182
 
127
183
        Basically we keep looking up until we find the control directory or
128
184
        run into the root.  If there isn't one, raises NotBranchError.
129
 
        If there is one and it is either an unrecognised format or an unsupported 
 
185
        If there is one and it is either an unrecognised format or an unsupported
130
186
        format, UnknownFormatError or UnsupportedFormatError are raised.
131
187
        If there is one, it is returned, along with the unused portion of url.
132
188
        """
134
190
                                                         possible_transports)
135
191
        return control.open_branch(), relpath
136
192
 
 
193
    def _push_should_merge_tags(self):
 
194
        """Should _basic_push merge this branch's tags into the target?
 
195
 
 
196
        The default implementation returns False if this branch has no tags,
 
197
        and True the rest of the time.  Subclasses may override this.
 
198
        """
 
199
        return self.supports_tags() and self.tags.get_tag_dict()
 
200
 
137
201
    def get_config(self):
 
202
        """Get a bzrlib.config.BranchConfig for this Branch.
 
203
 
 
204
        This can then be used to get and set configuration options for the
 
205
        branch.
 
206
 
 
207
        :return: A bzrlib.config.BranchConfig.
 
208
        """
138
209
        return BranchConfig(self)
139
210
 
140
 
    def _get_nick(self):
141
 
        return self.get_config().get_nickname()
 
211
    def _get_config(self):
 
212
        """Get the concrete config for just the config in this branch.
 
213
 
 
214
        This is not intended for client use; see Branch.get_config for the
 
215
        public API.
 
216
 
 
217
        Added in 1.14.
 
218
 
 
219
        :return: An object supporting get_option and set_option.
 
220
        """
 
221
        raise NotImplementedError(self._get_config)
 
222
 
 
223
    def _get_fallback_repository(self, url):
 
224
        """Get the repository we fallback to at url."""
 
225
        url = urlutils.join(self.base, url)
 
226
        a_branch = Branch.open(url,
 
227
            possible_transports=[self.bzrdir.root_transport])
 
228
        return a_branch.repository
 
229
 
 
230
    def _get_tags_bytes(self):
 
231
        """Get the bytes of a serialised tags dict.
 
232
 
 
233
        Note that not all branches support tags, nor do all use the same tags
 
234
        logic: this method is specific to BasicTags. Other tag implementations
 
235
        may use the same method name and behave differently, safely, because
 
236
        of the double-dispatch via
 
237
        format.make_tags->tags_instance->get_tags_dict.
 
238
 
 
239
        :return: The bytes of the tags file.
 
240
        :seealso: Branch._set_tags_bytes.
 
241
        """
 
242
        return self._transport.get_bytes('tags')
 
243
 
 
244
    def _get_nick(self, local=False, possible_transports=None):
 
245
        config = self.get_config()
 
246
        # explicit overrides master, but don't look for master if local is True
 
247
        if not local and not config.has_explicit_nickname():
 
248
            try:
 
249
                master = self.get_master_branch(possible_transports)
 
250
                if master and self.user_url == master.user_url:
 
251
                    raise errors.RecursiveBind(self.user_url)
 
252
                if master is not None:
 
253
                    # return the master branch value
 
254
                    return master.nick
 
255
            except errors.RecursiveBind, e:
 
256
                raise e
 
257
            except errors.BzrError, e:
 
258
                # Silently fall back to local implicit nick if the master is
 
259
                # unavailable
 
260
                mutter("Could not connect to bound branch, "
 
261
                    "falling back to local nick.\n " + str(e))
 
262
        return config.get_nickname()
142
263
 
143
264
    def _set_nick(self, nick):
144
265
        self.get_config().set_user_option('nickname', nick, warn_masked=True)
148
269
    def is_locked(self):
149
270
        raise NotImplementedError(self.is_locked)
150
271
 
151
 
    def lock_write(self):
 
272
    def _lefthand_history(self, revision_id, last_rev=None,
 
273
                          other_branch=None):
 
274
        if 'evil' in debug.debug_flags:
 
275
            mutter_callsite(4, "_lefthand_history scales with history.")
 
276
        # stop_revision must be a descendant of last_revision
 
277
        graph = self.repository.get_graph()
 
278
        if last_rev is not None:
 
279
            if not graph.is_ancestor(last_rev, revision_id):
 
280
                # our previous tip is not merged into stop_revision
 
281
                raise errors.DivergedBranches(self, other_branch)
 
282
        # make a new revision history from the graph
 
283
        parents_map = graph.get_parent_map([revision_id])
 
284
        if revision_id not in parents_map:
 
285
            raise errors.NoSuchRevision(self, revision_id)
 
286
        current_rev_id = revision_id
 
287
        new_history = []
 
288
        check_not_reserved_id = _mod_revision.check_not_reserved_id
 
289
        # Do not include ghosts or graph origin in revision_history
 
290
        while (current_rev_id in parents_map and
 
291
               len(parents_map[current_rev_id]) > 0):
 
292
            check_not_reserved_id(current_rev_id)
 
293
            new_history.append(current_rev_id)
 
294
            current_rev_id = parents_map[current_rev_id][0]
 
295
            parents_map = graph.get_parent_map([current_rev_id])
 
296
        new_history.reverse()
 
297
        return new_history
 
298
 
 
299
    def lock_write(self, token=None):
 
300
        """Lock the branch for write operations.
 
301
 
 
302
        :param token: A token to permit reacquiring a previously held and
 
303
            preserved lock.
 
304
        :return: A BranchWriteLockResult.
 
305
        """
152
306
        raise NotImplementedError(self.lock_write)
153
307
 
154
308
    def lock_read(self):
 
309
        """Lock the branch for read operations.
 
310
 
 
311
        :return: A bzrlib.lock.LogicalLockResult.
 
312
        """
155
313
        raise NotImplementedError(self.lock_read)
156
314
 
157
315
    def unlock(self):
165
323
        raise NotImplementedError(self.get_physical_lock_status)
166
324
 
167
325
    @needs_read_lock
 
326
    def dotted_revno_to_revision_id(self, revno, _cache_reverse=False):
 
327
        """Return the revision_id for a dotted revno.
 
328
 
 
329
        :param revno: a tuple like (1,) or (1,1,2)
 
330
        :param _cache_reverse: a private parameter enabling storage
 
331
           of the reverse mapping in a top level cache. (This should
 
332
           only be done in selective circumstances as we want to
 
333
           avoid having the mapping cached multiple times.)
 
334
        :return: the revision_id
 
335
        :raises errors.NoSuchRevision: if the revno doesn't exist
 
336
        """
 
337
        rev_id = self._do_dotted_revno_to_revision_id(revno)
 
338
        if _cache_reverse:
 
339
            self._partial_revision_id_to_revno_cache[rev_id] = revno
 
340
        return rev_id
 
341
 
 
342
    def _do_dotted_revno_to_revision_id(self, revno):
 
343
        """Worker function for dotted_revno_to_revision_id.
 
344
 
 
345
        Subclasses should override this if they wish to
 
346
        provide a more efficient implementation.
 
347
        """
 
348
        if len(revno) == 1:
 
349
            return self.get_rev_id(revno[0])
 
350
        revision_id_to_revno = self.get_revision_id_to_revno_map()
 
351
        revision_ids = [revision_id for revision_id, this_revno
 
352
                        in revision_id_to_revno.iteritems()
 
353
                        if revno == this_revno]
 
354
        if len(revision_ids) == 1:
 
355
            return revision_ids[0]
 
356
        else:
 
357
            revno_str = '.'.join(map(str, revno))
 
358
            raise errors.NoSuchRevision(self, revno_str)
 
359
 
 
360
    @needs_read_lock
 
361
    def revision_id_to_dotted_revno(self, revision_id):
 
362
        """Given a revision id, return its dotted revno.
 
363
 
 
364
        :return: a tuple like (1,) or (400,1,3).
 
365
        """
 
366
        return self._do_revision_id_to_dotted_revno(revision_id)
 
367
 
 
368
    def _do_revision_id_to_dotted_revno(self, revision_id):
 
369
        """Worker function for revision_id_to_revno."""
 
370
        # Try the caches if they are loaded
 
371
        result = self._partial_revision_id_to_revno_cache.get(revision_id)
 
372
        if result is not None:
 
373
            return result
 
374
        if self._revision_id_to_revno_cache:
 
375
            result = self._revision_id_to_revno_cache.get(revision_id)
 
376
            if result is None:
 
377
                raise errors.NoSuchRevision(self, revision_id)
 
378
        # Try the mainline as it's optimised
 
379
        try:
 
380
            revno = self.revision_id_to_revno(revision_id)
 
381
            return (revno,)
 
382
        except errors.NoSuchRevision:
 
383
            # We need to load and use the full revno map after all
 
384
            result = self.get_revision_id_to_revno_map().get(revision_id)
 
385
            if result is None:
 
386
                raise errors.NoSuchRevision(self, revision_id)
 
387
        return result
 
388
 
 
389
    @needs_read_lock
168
390
    def get_revision_id_to_revno_map(self):
169
391
        """Return the revision_id => dotted revno map.
170
392
 
194
416
 
195
417
        :return: A dictionary mapping revision_id => dotted revno.
196
418
        """
197
 
        last_revision = self.last_revision()
198
 
        revision_graph = repository._old_get_graph(self.repository,
199
 
            last_revision)
200
 
        merge_sorted_revisions = tsort.merge_sort(
201
 
            revision_graph,
202
 
            last_revision,
203
 
            None,
204
 
            generate_revno=True)
205
419
        revision_id_to_revno = dict((rev_id, revno)
206
 
                                    for seq_num, rev_id, depth, revno, end_of_merge
207
 
                                     in merge_sorted_revisions)
 
420
            for rev_id, depth, revno, end_of_merge
 
421
             in self.iter_merge_sorted_revisions())
208
422
        return revision_id_to_revno
209
423
 
 
424
    @needs_read_lock
 
425
    def iter_merge_sorted_revisions(self, start_revision_id=None,
 
426
            stop_revision_id=None, stop_rule='exclude', direction='reverse'):
 
427
        """Walk the revisions for a branch in merge sorted order.
 
428
 
 
429
        Merge sorted order is the output from a merge-aware,
 
430
        topological sort, i.e. all parents come before their
 
431
        children going forward; the opposite for reverse.
 
432
 
 
433
        :param start_revision_id: the revision_id to begin walking from.
 
434
            If None, the branch tip is used.
 
435
        :param stop_revision_id: the revision_id to terminate the walk
 
436
            after. If None, the rest of history is included.
 
437
        :param stop_rule: if stop_revision_id is not None, the precise rule
 
438
            to use for termination:
 
439
            * 'exclude' - leave the stop revision out of the result (default)
 
440
            * 'include' - the stop revision is the last item in the result
 
441
            * 'with-merges' - include the stop revision and all of its
 
442
              merged revisions in the result
 
443
            * 'with-merges-without-common-ancestry' - filter out revisions 
 
444
              that are in both ancestries
 
445
        :param direction: either 'reverse' or 'forward':
 
446
            * reverse means return the start_revision_id first, i.e.
 
447
              start at the most recent revision and go backwards in history
 
448
            * forward returns tuples in the opposite order to reverse.
 
449
              Note in particular that forward does *not* do any intelligent
 
450
              ordering w.r.t. depth as some clients of this API may like.
 
451
              (If required, that ought to be done at higher layers.)
 
452
 
 
453
        :return: an iterator over (revision_id, depth, revno, end_of_merge)
 
454
            tuples where:
 
455
 
 
456
            * revision_id: the unique id of the revision
 
457
            * depth: How many levels of merging deep this node has been
 
458
              found.
 
459
            * revno_sequence: This field provides a sequence of
 
460
              revision numbers for all revisions. The format is:
 
461
              (REVNO, BRANCHNUM, BRANCHREVNO). BRANCHNUM is the number of the
 
462
              branch that the revno is on. From left to right the REVNO numbers
 
463
              are the sequence numbers within that branch of the revision.
 
464
            * end_of_merge: When True the next node (earlier in history) is
 
465
              part of a different merge.
 
466
        """
 
467
        # Note: depth and revno values are in the context of the branch so
 
468
        # we need the full graph to get stable numbers, regardless of the
 
469
        # start_revision_id.
 
470
        if self._merge_sorted_revisions_cache is None:
 
471
            last_revision = self.last_revision()
 
472
            known_graph = self.repository.get_known_graph_ancestry(
 
473
                [last_revision])
 
474
            self._merge_sorted_revisions_cache = known_graph.merge_sort(
 
475
                last_revision)
 
476
        filtered = self._filter_merge_sorted_revisions(
 
477
            self._merge_sorted_revisions_cache, start_revision_id,
 
478
            stop_revision_id, stop_rule)
 
479
        # Make sure we don't return revisions that are not part of the
 
480
        # start_revision_id ancestry.
 
481
        filtered = self._filter_start_non_ancestors(filtered)
 
482
        if direction == 'reverse':
 
483
            return filtered
 
484
        if direction == 'forward':
 
485
            return reversed(list(filtered))
 
486
        else:
 
487
            raise ValueError('invalid direction %r' % direction)
 
488
 
 
489
    def _filter_merge_sorted_revisions(self, merge_sorted_revisions,
 
490
        start_revision_id, stop_revision_id, stop_rule):
 
491
        """Iterate over an inclusive range of sorted revisions."""
 
492
        rev_iter = iter(merge_sorted_revisions)
 
493
        if start_revision_id is not None:
 
494
            for node in rev_iter:
 
495
                rev_id = node.key[-1]
 
496
                if rev_id != start_revision_id:
 
497
                    continue
 
498
                else:
 
499
                    # The decision to include the start or not
 
500
                    # depends on the stop_rule if a stop is provided
 
501
                    # so pop this node back into the iterator
 
502
                    rev_iter = chain(iter([node]), rev_iter)
 
503
                    break
 
504
        if stop_revision_id is None:
 
505
            # Yield everything
 
506
            for node in rev_iter:
 
507
                rev_id = node.key[-1]
 
508
                yield (rev_id, node.merge_depth, node.revno,
 
509
                       node.end_of_merge)
 
510
        elif stop_rule == 'exclude':
 
511
            for node in rev_iter:
 
512
                rev_id = node.key[-1]
 
513
                if rev_id == stop_revision_id:
 
514
                    return
 
515
                yield (rev_id, node.merge_depth, node.revno,
 
516
                       node.end_of_merge)
 
517
        elif stop_rule == 'include':
 
518
            for node in rev_iter:
 
519
                rev_id = node.key[-1]
 
520
                yield (rev_id, node.merge_depth, node.revno,
 
521
                       node.end_of_merge)
 
522
                if rev_id == stop_revision_id:
 
523
                    return
 
524
        elif stop_rule == 'with-merges-without-common-ancestry':
 
525
            # We want to exclude all revisions that are already part of the
 
526
            # stop_revision_id ancestry.
 
527
            graph = self.repository.get_graph()
 
528
            ancestors = graph.find_unique_ancestors(start_revision_id,
 
529
                                                    [stop_revision_id])
 
530
            for node in rev_iter:
 
531
                rev_id = node.key[-1]
 
532
                if rev_id not in ancestors:
 
533
                    continue
 
534
                yield (rev_id, node.merge_depth, node.revno,
 
535
                       node.end_of_merge)
 
536
        elif stop_rule == 'with-merges':
 
537
            stop_rev = self.repository.get_revision(stop_revision_id)
 
538
            if stop_rev.parent_ids:
 
539
                left_parent = stop_rev.parent_ids[0]
 
540
            else:
 
541
                left_parent = _mod_revision.NULL_REVISION
 
542
            # left_parent is the actual revision we want to stop logging at,
 
543
            # since we want to show the merged revisions after the stop_rev too
 
544
            reached_stop_revision_id = False
 
545
            revision_id_whitelist = []
 
546
            for node in rev_iter:
 
547
                rev_id = node.key[-1]
 
548
                if rev_id == left_parent:
 
549
                    # reached the left parent after the stop_revision
 
550
                    return
 
551
                if (not reached_stop_revision_id or
 
552
                        rev_id in revision_id_whitelist):
 
553
                    yield (rev_id, node.merge_depth, node.revno,
 
554
                       node.end_of_merge)
 
555
                    if reached_stop_revision_id or rev_id == stop_revision_id:
 
556
                        # only do the merged revs of rev_id from now on
 
557
                        rev = self.repository.get_revision(rev_id)
 
558
                        if rev.parent_ids:
 
559
                            reached_stop_revision_id = True
 
560
                            revision_id_whitelist.extend(rev.parent_ids)
 
561
        else:
 
562
            raise ValueError('invalid stop_rule %r' % stop_rule)
 
563
 
 
564
    def _filter_start_non_ancestors(self, rev_iter):
 
565
        # If we started from a dotted revno, we want to consider it as a tip
 
566
        # and don't want to yield revisions that are not part of its
 
567
        # ancestry. Given the order guaranteed by the merge sort, we will see
 
568
        # uninteresting descendants of the first parent of our tip before the
 
569
        # tip itself.
 
570
        first = rev_iter.next()
 
571
        (rev_id, merge_depth, revno, end_of_merge) = first
 
572
        yield first
 
573
        if not merge_depth:
 
574
            # We start at a mainline revision so by definition, all others
 
575
            # revisions in rev_iter are ancestors
 
576
            for node in rev_iter:
 
577
                yield node
 
578
 
 
579
        clean = False
 
580
        whitelist = set()
 
581
        pmap = self.repository.get_parent_map([rev_id])
 
582
        parents = pmap.get(rev_id, [])
 
583
        if parents:
 
584
            whitelist.update(parents)
 
585
        else:
 
586
            # If there is no parents, there is nothing of interest left
 
587
 
 
588
            # FIXME: It's hard to test this scenario here as this code is never
 
589
            # called in that case. -- vila 20100322
 
590
            return
 
591
 
 
592
        for (rev_id, merge_depth, revno, end_of_merge) in rev_iter:
 
593
            if not clean:
 
594
                if rev_id in whitelist:
 
595
                    pmap = self.repository.get_parent_map([rev_id])
 
596
                    parents = pmap.get(rev_id, [])
 
597
                    whitelist.remove(rev_id)
 
598
                    whitelist.update(parents)
 
599
                    if merge_depth == 0:
 
600
                        # We've reached the mainline, there is nothing left to
 
601
                        # filter
 
602
                        clean = True
 
603
                else:
 
604
                    # A revision that is not part of the ancestry of our
 
605
                    # starting revision.
 
606
                    continue
 
607
            yield (rev_id, merge_depth, revno, end_of_merge)
 
608
 
210
609
    def leave_lock_in_place(self):
211
610
        """Tell this branch object not to release the physical lock when this
212
611
        object is unlocked.
213
 
        
 
612
 
214
613
        If lock_write doesn't return a token, then this method is not supported.
215
614
        """
216
615
        self.control_files.leave_in_place()
223
622
        """
224
623
        self.control_files.dont_leave_in_place()
225
624
 
226
 
    @deprecated_method(deprecated_in((0, 16, 0)))
227
 
    def abspath(self, name):
228
 
        """Return absolute filename for something in the branch
229
 
        
230
 
        XXX: Robert Collins 20051017 what is this used for? why is it a branch
231
 
        method and not a tree method.
232
 
        """
233
 
        raise NotImplementedError(self.abspath)
234
 
 
235
625
    def bind(self, other):
236
626
        """Bind the local branch the other branch.
237
627
 
238
628
        :param other: The branch to bind to
239
629
        :type other: Branch
240
630
        """
241
 
        raise errors.UpgradeRequired(self.base)
 
631
        raise errors.UpgradeRequired(self.user_url)
 
632
 
 
633
    def set_append_revisions_only(self, enabled):
 
634
        if not self._format.supports_set_append_revisions_only():
 
635
            raise errors.UpgradeRequired(self.user_url)
 
636
        if enabled:
 
637
            value = 'True'
 
638
        else:
 
639
            value = 'False'
 
640
        self.get_config().set_user_option('append_revisions_only', value,
 
641
            warn_masked=True)
 
642
 
 
643
    def set_reference_info(self, file_id, tree_path, branch_location):
 
644
        """Set the branch location to use for a tree reference."""
 
645
        raise errors.UnsupportedOperation(self.set_reference_info, self)
 
646
 
 
647
    def get_reference_info(self, file_id):
 
648
        """Get the tree_path and branch_location for a tree reference."""
 
649
        raise errors.UnsupportedOperation(self.get_reference_info, self)
242
650
 
243
651
    @needs_write_lock
244
652
    def fetch(self, from_branch, last_revision=None, pb=None):
248
656
        :param last_revision: What revision to stop at (None for at the end
249
657
                              of the branch.
250
658
        :param pb: An optional progress bar to use.
251
 
 
252
 
        Returns the copied revision count and the failed revisions in a tuple:
253
 
        (copied, failures).
 
659
        :return: None
254
660
        """
255
661
        if self.base == from_branch.base:
256
662
            return (0, [])
257
 
        if pb is None:
258
 
            nested_pb = ui.ui_factory.nested_progress_bar()
259
 
            pb = nested_pb
260
 
        else:
261
 
            nested_pb = None
262
 
 
 
663
        if pb is not None:
 
664
            symbol_versioning.warn(
 
665
                symbol_versioning.deprecated_in((1, 14, 0))
 
666
                % "pb parameter to fetch()")
263
667
        from_branch.lock_read()
264
668
        try:
265
669
            if last_revision is None:
266
 
                pb.update('get source history')
267
670
                last_revision = from_branch.last_revision()
268
671
                last_revision = _mod_revision.ensure_null(last_revision)
269
672
            return self.repository.fetch(from_branch.repository,
270
673
                                         revision_id=last_revision,
271
 
                                         pb=nested_pb)
 
674
                                         pb=pb)
272
675
        finally:
273
 
            if nested_pb is not None:
274
 
                nested_pb.finished()
275
676
            from_branch.unlock()
276
677
 
277
678
    def get_bound_location(self):
281
682
        branch.
282
683
        """
283
684
        return None
284
 
    
 
685
 
285
686
    def get_old_bound_location(self):
286
687
        """Return the URL of the branch we used to be bound to
287
688
        """
288
 
        raise errors.UpgradeRequired(self.base)
 
689
        raise errors.UpgradeRequired(self.user_url)
289
690
 
290
 
    def get_commit_builder(self, parents, config=None, timestamp=None, 
291
 
                           timezone=None, committer=None, revprops=None, 
 
691
    def get_commit_builder(self, parents, config=None, timestamp=None,
 
692
                           timezone=None, committer=None, revprops=None,
292
693
                           revision_id=None):
293
694
        """Obtain a CommitBuilder for this branch.
294
 
        
 
695
 
295
696
        :param parents: Revision ids of the parents of the new revision.
296
697
        :param config: Optional configuration to use.
297
698
        :param timestamp: Optional timestamp recorded for commit.
303
704
 
304
705
        if config is None:
305
706
            config = self.get_config()
306
 
        
 
707
 
307
708
        return self.repository.get_commit_builder(self, parents, config,
308
709
            timestamp, timezone, committer, revprops, revision_id)
309
710
 
310
711
    def get_master_branch(self, possible_transports=None):
311
712
        """Return the branch we are bound to.
312
 
        
 
713
 
313
714
        :return: Either a Branch, or None
314
715
        """
315
716
        return None
325
726
            raise errors.InvalidRevisionNumber(revno)
326
727
        return self.repository.get_revision_delta(rh[revno-1])
327
728
 
 
729
    def get_stacked_on_url(self):
 
730
        """Get the URL this branch is stacked against.
 
731
 
 
732
        :raises NotStacked: If the branch is not stacked.
 
733
        :raises UnstackableBranchFormat: If the branch does not support
 
734
            stacking.
 
735
        """
 
736
        raise NotImplementedError(self.get_stacked_on_url)
 
737
 
328
738
    def print_file(self, file, revision_id):
329
739
        """Print `file` to stdout."""
330
740
        raise NotImplementedError(self.print_file)
332
742
    def set_revision_history(self, rev_history):
333
743
        raise NotImplementedError(self.set_revision_history)
334
744
 
 
745
    @needs_write_lock
 
746
    def set_parent(self, url):
 
747
        """See Branch.set_parent."""
 
748
        # TODO: Maybe delete old location files?
 
749
        # URLs should never be unicode, even on the local fs,
 
750
        # FIXUP this and get_parent in a future branch format bump:
 
751
        # read and rewrite the file. RBC 20060125
 
752
        if url is not None:
 
753
            if isinstance(url, unicode):
 
754
                try:
 
755
                    url = url.encode('ascii')
 
756
                except UnicodeEncodeError:
 
757
                    raise errors.InvalidURL(url,
 
758
                        "Urls must be 7-bit ascii, "
 
759
                        "use bzrlib.urlutils.escape")
 
760
            url = urlutils.relative_url(self.base, url)
 
761
        self._set_parent_location(url)
 
762
 
 
763
    @needs_write_lock
 
764
    def set_stacked_on_url(self, url):
 
765
        """Set the URL this branch is stacked against.
 
766
 
 
767
        :raises UnstackableBranchFormat: If the branch does not support
 
768
            stacking.
 
769
        :raises UnstackableRepositoryFormat: If the repository does not support
 
770
            stacking.
 
771
        """
 
772
        if not self._format.supports_stacking():
 
773
            raise errors.UnstackableBranchFormat(self._format, self.user_url)
 
774
        # XXX: Changing from one fallback repository to another does not check
 
775
        # that all the data you need is present in the new fallback.
 
776
        # Possibly it should.
 
777
        self._check_stackable_repo()
 
778
        if not url:
 
779
            try:
 
780
                old_url = self.get_stacked_on_url()
 
781
            except (errors.NotStacked, errors.UnstackableBranchFormat,
 
782
                errors.UnstackableRepositoryFormat):
 
783
                return
 
784
            self._unstack()
 
785
        else:
 
786
            self._activate_fallback_location(url)
 
787
        # write this out after the repository is stacked to avoid setting a
 
788
        # stacked config that doesn't work.
 
789
        self._set_config_location('stacked_on_location', url)
 
790
 
 
791
    def _unstack(self):
 
792
        """Change a branch to be unstacked, copying data as needed.
 
793
        
 
794
        Don't call this directly, use set_stacked_on_url(None).
 
795
        """
 
796
        pb = ui.ui_factory.nested_progress_bar()
 
797
        try:
 
798
            pb.update("Unstacking")
 
799
            # The basic approach here is to fetch the tip of the branch,
 
800
            # including all available ghosts, from the existing stacked
 
801
            # repository into a new repository object without the fallbacks. 
 
802
            #
 
803
            # XXX: See <https://launchpad.net/bugs/397286> - this may not be
 
804
            # correct for CHKMap repostiories
 
805
            old_repository = self.repository
 
806
            if len(old_repository._fallback_repositories) != 1:
 
807
                raise AssertionError("can't cope with fallback repositories "
 
808
                    "of %r" % (self.repository,))
 
809
            # Open the new repository object.
 
810
            # Repositories don't offer an interface to remove fallback
 
811
            # repositories today; take the conceptually simpler option and just
 
812
            # reopen it.  We reopen it starting from the URL so that we
 
813
            # get a separate connection for RemoteRepositories and can
 
814
            # stream from one of them to the other.  This does mean doing
 
815
            # separate SSH connection setup, but unstacking is not a
 
816
            # common operation so it's tolerable.
 
817
            new_bzrdir = bzrdir.BzrDir.open(self.bzrdir.root_transport.base)
 
818
            new_repository = new_bzrdir.find_repository()
 
819
            if new_repository._fallback_repositories:
 
820
                raise AssertionError("didn't expect %r to have "
 
821
                    "fallback_repositories"
 
822
                    % (self.repository,))
 
823
            # Replace self.repository with the new repository.
 
824
            # Do our best to transfer the lock state (i.e. lock-tokens and
 
825
            # lock count) of self.repository to the new repository.
 
826
            lock_token = old_repository.lock_write().repository_token
 
827
            self.repository = new_repository
 
828
            if isinstance(self, remote.RemoteBranch):
 
829
                # Remote branches can have a second reference to the old
 
830
                # repository that need to be replaced.
 
831
                if self._real_branch is not None:
 
832
                    self._real_branch.repository = new_repository
 
833
            self.repository.lock_write(token=lock_token)
 
834
            if lock_token is not None:
 
835
                old_repository.leave_lock_in_place()
 
836
            old_repository.unlock()
 
837
            if lock_token is not None:
 
838
                # XXX: self.repository.leave_lock_in_place() before this
 
839
                # function will not be preserved.  Fortunately that doesn't
 
840
                # affect the current default format (2a), and would be a
 
841
                # corner-case anyway.
 
842
                #  - Andrew Bennetts, 2010/06/30
 
843
                self.repository.dont_leave_lock_in_place()
 
844
            old_lock_count = 0
 
845
            while True:
 
846
                try:
 
847
                    old_repository.unlock()
 
848
                except errors.LockNotHeld:
 
849
                    break
 
850
                old_lock_count += 1
 
851
            if old_lock_count == 0:
 
852
                raise AssertionError(
 
853
                    'old_repository should have been locked at least once.')
 
854
            for i in range(old_lock_count-1):
 
855
                self.repository.lock_write()
 
856
            # Fetch from the old repository into the new.
 
857
            old_repository.lock_read()
 
858
            try:
 
859
                # XXX: If you unstack a branch while it has a working tree
 
860
                # with a pending merge, the pending-merged revisions will no
 
861
                # longer be present.  You can (probably) revert and remerge.
 
862
                #
 
863
                # XXX: This only fetches up to the tip of the repository; it
 
864
                # doesn't bring across any tags.  That's fairly consistent
 
865
                # with how branch works, but perhaps not ideal.
 
866
                self.repository.fetch(old_repository,
 
867
                    revision_id=self.last_revision(),
 
868
                    find_ghosts=True)
 
869
            finally:
 
870
                old_repository.unlock()
 
871
        finally:
 
872
            pb.finished()
 
873
 
 
874
    def _set_tags_bytes(self, bytes):
 
875
        """Mirror method for _get_tags_bytes.
 
876
 
 
877
        :seealso: Branch._get_tags_bytes.
 
878
        """
 
879
        return _run_with_write_locked_target(self, self._transport.put_bytes,
 
880
            'tags', bytes)
 
881
 
335
882
    def _cache_revision_history(self, rev_history):
336
883
        """Set the cached revision history to rev_history.
337
884
 
363
910
        self._revision_history_cache = None
364
911
        self._revision_id_to_revno_cache = None
365
912
        self._last_revision_info_cache = None
 
913
        self._merge_sorted_revisions_cache = None
 
914
        self._partial_revision_history_cache = []
 
915
        self._partial_revision_id_to_revno_cache = {}
366
916
 
367
917
    def _gen_revision_history(self):
368
918
        """Return sequence of revision hashes on to this branch.
369
 
        
 
919
 
370
920
        Unlike revision_history, this method always regenerates or rereads the
371
921
        revision history, i.e. it does not cache the result, so repeated calls
372
922
        may be expensive.
373
923
 
374
924
        Concrete subclasses should override this instead of revision_history so
375
925
        that subclasses do not need to deal with caching logic.
376
 
        
 
926
 
377
927
        This API is semi-public; it only for use by subclasses, all other code
378
928
        should consider it to be private.
379
929
        """
382
932
    @needs_read_lock
383
933
    def revision_history(self):
384
934
        """Return sequence of revision ids on this branch.
385
 
        
 
935
 
386
936
        This method will cache the revision history for as long as it is safe to
387
937
        do so.
388
938
        """
405
955
 
406
956
    def unbind(self):
407
957
        """Older format branches cannot bind or unbind."""
408
 
        raise errors.UpgradeRequired(self.base)
409
 
 
410
 
    def set_append_revisions_only(self, enabled):
411
 
        """Older format branches are never restricted to append-only"""
412
 
        raise errors.UpgradeRequired(self.base)
 
958
        raise errors.UpgradeRequired(self.user_url)
413
959
 
414
960
    def last_revision(self):
415
961
        """Return last revision id, or NULL_REVISION."""
436
982
    @deprecated_method(deprecated_in((1, 6, 0)))
437
983
    def missing_revisions(self, other, stop_revision=None):
438
984
        """Return a list of new revisions that would perfectly fit.
439
 
        
 
985
 
440
986
        If self and other have not diverged, return a list of the revisions
441
987
        present in other, but missing from self.
442
988
        """
456
1002
                raise errors.NoSuchRevision(self, stop_revision)
457
1003
        return other_history[self_len:stop_revision]
458
1004
 
459
 
    @needs_write_lock
460
1005
    def update_revisions(self, other, stop_revision=None, overwrite=False,
461
1006
                         graph=None):
462
1007
        """Pull in new perfect-fit revisions.
469
1014
            information. This can be None.
470
1015
        :return: None
471
1016
        """
472
 
        other.lock_read()
473
 
        try:
474
 
            other_revno, other_last_revision = other.last_revision_info()
475
 
            stop_revno = None # unknown
476
 
            if stop_revision is None:
477
 
                stop_revision = other_last_revision
478
 
                if _mod_revision.is_null(stop_revision):
479
 
                    # if there are no commits, we're done.
480
 
                    return
481
 
                stop_revno = other_revno
482
 
 
483
 
            # what's the current last revision, before we fetch [and change it
484
 
            # possibly]
485
 
            last_rev = _mod_revision.ensure_null(self.last_revision())
486
 
            # we fetch here so that we don't process data twice in the common
487
 
            # case of having something to pull, and so that the check for 
488
 
            # already merged can operate on the just fetched graph, which will
489
 
            # be cached in memory.
490
 
            self.fetch(other, stop_revision)
491
 
            # Check to see if one is an ancestor of the other
492
 
            if not overwrite:
493
 
                if graph is None:
494
 
                    graph = self.repository.get_graph()
495
 
                if self._check_if_descendant_or_diverged(
496
 
                        stop_revision, last_rev, graph, other):
497
 
                    # stop_revision is a descendant of last_rev, but we aren't
498
 
                    # overwriting, so we're done.
499
 
                    return
500
 
            if stop_revno is None:
501
 
                if graph is None:
502
 
                    graph = self.repository.get_graph()
503
 
                this_revno, this_last_revision = self.last_revision_info()
504
 
                stop_revno = graph.find_distance_to_null(stop_revision,
505
 
                                [(other_last_revision, other_revno),
506
 
                                 (this_last_revision, this_revno)])
507
 
            self.set_last_revision_info(stop_revno, stop_revision)
508
 
        finally:
509
 
            other.unlock()
510
 
 
511
 
 
 
1017
        return InterBranch.get(other, self).update_revisions(stop_revision,
 
1018
            overwrite, graph)
 
1019
 
 
1020
    def import_last_revision_info(self, source_repo, revno, revid):
 
1021
        """Set the last revision info, importing from another repo if necessary.
 
1022
 
 
1023
        This is used by the bound branch code to upload a revision to
 
1024
        the master branch first before updating the tip of the local branch.
 
1025
 
 
1026
        :param source_repo: Source repository to optionally fetch from
 
1027
        :param revno: Revision number of the new tip
 
1028
        :param revid: Revision id of the new tip
 
1029
        """
 
1030
        if not self.repository.has_same_location(source_repo):
 
1031
            self.repository.fetch(source_repo, revision_id=revid)
 
1032
        self.set_last_revision_info(revno, revid)
512
1033
 
513
1034
    def revision_id_to_revno(self, revision_id):
514
1035
        """Given a revision id, return its revno"""
520
1041
        except ValueError:
521
1042
            raise errors.NoSuchRevision(self, revision_id)
522
1043
 
 
1044
    @needs_read_lock
523
1045
    def get_rev_id(self, revno, history=None):
524
1046
        """Find the revision id of the specified revno."""
525
1047
        if revno == 0:
526
1048
            return _mod_revision.NULL_REVISION
527
 
        if history is None:
528
 
            history = self.revision_history()
529
 
        if revno <= 0 or revno > len(history):
 
1049
        last_revno, last_revid = self.last_revision_info()
 
1050
        if revno == last_revno:
 
1051
            return last_revid
 
1052
        if revno <= 0 or revno > last_revno:
530
1053
            raise errors.NoSuchRevision(self, revno)
531
 
        return history[revno - 1]
 
1054
        distance_from_last = last_revno - revno
 
1055
        if len(self._partial_revision_history_cache) <= distance_from_last:
 
1056
            self._extend_partial_history(distance_from_last)
 
1057
        return self._partial_revision_history_cache[distance_from_last]
532
1058
 
533
1059
    def pull(self, source, overwrite=False, stop_revision=None,
534
 
             possible_transports=None, _override_hook_target=None):
 
1060
             possible_transports=None, *args, **kwargs):
535
1061
        """Mirror source into this branch.
536
1062
 
537
1063
        This branch is considered to be 'local', having low latency.
538
1064
 
539
1065
        :returns: PullResult instance
540
1066
        """
541
 
        raise NotImplementedError(self.pull)
 
1067
        return InterBranch.get(source, self).pull(overwrite=overwrite,
 
1068
            stop_revision=stop_revision,
 
1069
            possible_transports=possible_transports, *args, **kwargs)
542
1070
 
543
 
    def push(self, target, overwrite=False, stop_revision=None):
 
1071
    def push(self, target, overwrite=False, stop_revision=None, *args,
 
1072
        **kwargs):
544
1073
        """Mirror this branch into target.
545
1074
 
546
1075
        This branch is considered to be 'local', having low latency.
547
1076
        """
548
 
        raise NotImplementedError(self.push)
 
1077
        return InterBranch.get(self, target).push(overwrite, stop_revision,
 
1078
            *args, **kwargs)
 
1079
 
 
1080
    def lossy_push(self, target, stop_revision=None):
 
1081
        """Push deltas into another branch.
 
1082
 
 
1083
        :note: This does not, like push, retain the revision ids from 
 
1084
            the source branch and will, rather than adding bzr-specific 
 
1085
            metadata, push only those semantics of the revision that can be 
 
1086
            natively represented by this branch' VCS.
 
1087
 
 
1088
        :param target: Target branch
 
1089
        :param stop_revision: Revision to push, defaults to last revision.
 
1090
        :return: BranchPushResult with an extra member revidmap: 
 
1091
            A dictionary mapping revision ids from the target branch 
 
1092
            to new revision ids in the target branch, for each 
 
1093
            revision that was pushed.
 
1094
        """
 
1095
        inter = InterBranch.get(self, target)
 
1096
        lossy_push = getattr(inter, "lossy_push", None)
 
1097
        if lossy_push is None:
 
1098
            raise errors.LossyPushToSameVCS(self, target)
 
1099
        return lossy_push(stop_revision)
549
1100
 
550
1101
    def basis_tree(self):
551
1102
        """Return `Tree` object for last revision."""
552
1103
        return self.repository.revision_tree(self.last_revision())
553
1104
 
554
 
    def rename_one(self, from_rel, to_rel):
555
 
        """Rename one file.
556
 
 
557
 
        This can change the directory or the filename or both.
558
 
        """
559
 
        raise NotImplementedError(self.rename_one)
560
 
 
561
 
    def move(self, from_paths, to_name):
562
 
        """Rename files.
563
 
 
564
 
        to_name must exist as a versioned directory.
565
 
 
566
 
        If to_name exists and is a directory, the files are moved into
567
 
        it, keeping their old names.  If it is a directory, 
568
 
 
569
 
        Note that to_name is only the last component of the new name;
570
 
        this doesn't change the directory.
571
 
 
572
 
        This returns a list of (from_path, to_path) pairs for each
573
 
        entry that is moved.
574
 
        """
575
 
        raise NotImplementedError(self.move)
576
 
 
577
1105
    def get_parent(self):
578
1106
        """Return the parent location of the branch.
579
1107
 
580
 
        This is the default location for push/pull/missing.  The usual
 
1108
        This is the default location for pull/missing.  The usual
581
1109
        pattern is that the user can override it by specifying a
582
1110
        location.
583
1111
        """
584
 
        raise NotImplementedError(self.get_parent)
 
1112
        parent = self._get_parent_location()
 
1113
        if parent is None:
 
1114
            return parent
 
1115
        # This is an old-format absolute path to a local branch
 
1116
        # turn it into a url
 
1117
        if parent.startswith('/'):
 
1118
            parent = urlutils.local_path_to_url(parent.decode('utf8'))
 
1119
        try:
 
1120
            return urlutils.join(self.base[:-1], parent)
 
1121
        except errors.InvalidURLJoin, e:
 
1122
            raise errors.InaccessibleParent(parent, self.user_url)
 
1123
 
 
1124
    def _get_parent_location(self):
 
1125
        raise NotImplementedError(self._get_parent_location)
585
1126
 
586
1127
    def _set_config_location(self, name, url, config=None,
587
1128
                             make_relative=False):
601
1142
            location = None
602
1143
        return location
603
1144
 
 
1145
    def get_child_submit_format(self):
 
1146
        """Return the preferred format of submissions to this branch."""
 
1147
        return self.get_config().get_user_option("child_submit_format")
 
1148
 
604
1149
    def get_submit_branch(self):
605
1150
        """Return the submit location of the branch.
606
1151
 
623
1168
    def get_public_branch(self):
624
1169
        """Return the public location of the branch.
625
1170
 
626
 
        This is is used by merge directives.
 
1171
        This is used by merge directives.
627
1172
        """
628
1173
        return self._get_config_location('public_branch')
629
1174
 
645
1190
        """Set a new push location for this branch."""
646
1191
        raise NotImplementedError(self.set_push_location)
647
1192
 
648
 
    def set_parent(self, url):
649
 
        raise NotImplementedError(self.set_parent)
 
1193
    def _run_post_change_branch_tip_hooks(self, old_revno, old_revid):
 
1194
        """Run the post_change_branch_tip hooks."""
 
1195
        hooks = Branch.hooks['post_change_branch_tip']
 
1196
        if not hooks:
 
1197
            return
 
1198
        new_revno, new_revid = self.last_revision_info()
 
1199
        params = ChangeBranchTipParams(
 
1200
            self, old_revno, new_revno, old_revid, new_revid)
 
1201
        for hook in hooks:
 
1202
            hook(params)
 
1203
 
 
1204
    def _run_pre_change_branch_tip_hooks(self, new_revno, new_revid):
 
1205
        """Run the pre_change_branch_tip hooks."""
 
1206
        hooks = Branch.hooks['pre_change_branch_tip']
 
1207
        if not hooks:
 
1208
            return
 
1209
        old_revno, old_revid = self.last_revision_info()
 
1210
        params = ChangeBranchTipParams(
 
1211
            self, old_revno, new_revno, old_revid, new_revid)
 
1212
        for hook in hooks:
 
1213
            hook(params)
650
1214
 
651
1215
    @needs_write_lock
652
1216
    def update(self):
653
 
        """Synchronise this branch with the master branch if any. 
 
1217
        """Synchronise this branch with the master branch if any.
654
1218
 
655
1219
        :return: None or the last_revision pivoted out during the update.
656
1220
        """
663
1227
        """
664
1228
        if revno != 0:
665
1229
            self.check_real_revno(revno)
666
 
            
 
1230
 
667
1231
    def check_real_revno(self, revno):
668
1232
        """\
669
1233
        Check whether a revno corresponds to a real revision.
673
1237
            raise errors.InvalidRevisionNumber(revno)
674
1238
 
675
1239
    @needs_read_lock
676
 
    def clone(self, to_bzrdir, revision_id=None):
 
1240
    def clone(self, to_bzrdir, revision_id=None, repository_policy=None):
677
1241
        """Clone this branch into to_bzrdir preserving all semantic values.
678
 
        
 
1242
 
 
1243
        Most API users will want 'create_clone_on_transport', which creates a
 
1244
        new bzrdir and branch on the fly.
 
1245
 
679
1246
        revision_id: if not None, the revision history in the new branch will
680
1247
                     be truncated to end with revision_id.
681
1248
        """
682
 
        result = self._format.initialize(to_bzrdir)
683
 
        self.copy_content_into(result, revision_id=revision_id)
684
 
        return  result
 
1249
        result = to_bzrdir.create_branch()
 
1250
        result.lock_write()
 
1251
        try:
 
1252
            if repository_policy is not None:
 
1253
                repository_policy.configure_branch(result)
 
1254
            self.copy_content_into(result, revision_id=revision_id)
 
1255
        finally:
 
1256
            result.unlock()
 
1257
        return result
685
1258
 
686
1259
    @needs_read_lock
687
 
    def sprout(self, to_bzrdir, revision_id=None):
 
1260
    def sprout(self, to_bzrdir, revision_id=None, repository_policy=None):
688
1261
        """Create a new line of development from the branch, into to_bzrdir.
689
 
        
 
1262
 
 
1263
        to_bzrdir controls the branch format.
 
1264
 
690
1265
        revision_id: if not None, the revision history in the new branch will
691
1266
                     be truncated to end with revision_id.
692
1267
        """
693
 
        result = self._format.initialize(to_bzrdir)
694
 
        self.copy_content_into(result, revision_id=revision_id)
695
 
        result.set_parent(self.bzrdir.root_transport.base)
 
1268
        if (repository_policy is not None and
 
1269
            repository_policy.requires_stacking()):
 
1270
            to_bzrdir._format.require_stacking(_skip_repo=True)
 
1271
        result = to_bzrdir.create_branch()
 
1272
        result.lock_write()
 
1273
        try:
 
1274
            if repository_policy is not None:
 
1275
                repository_policy.configure_branch(result)
 
1276
            self.copy_content_into(result, revision_id=revision_id)
 
1277
            result.set_parent(self.bzrdir.root_transport.base)
 
1278
        finally:
 
1279
            result.unlock()
696
1280
        return result
697
1281
 
698
1282
    def _synchronize_history(self, destination, revision_id):
699
1283
        """Synchronize last revision and revision history between branches.
700
1284
 
701
1285
        This version is most efficient when the destination is also a
702
 
        BzrBranch5, but works for BzrBranch6 as long as the revision
703
 
        history is the true lefthand parent history, and all of the revisions
704
 
        are in the destination's repository.  If not, set_revision_history
705
 
        will fail.
 
1286
        BzrBranch6, but works for BzrBranch5, as long as the destination's
 
1287
        repository contains all the lefthand ancestors of the intended
 
1288
        last_revision.  If not, set_last_revision_info will fail.
706
1289
 
707
1290
        :param destination: The branch to copy the history into
708
1291
        :param revision_id: The revision-id to truncate history at.  May
709
1292
          be None to copy complete history.
710
1293
        """
711
 
        if revision_id == _mod_revision.NULL_REVISION:
712
 
            new_history = []
713
 
        new_history = self.revision_history()
714
 
        if revision_id is not None and new_history != []:
 
1294
        source_revno, source_revision_id = self.last_revision_info()
 
1295
        if revision_id is None:
 
1296
            revno, revision_id = source_revno, source_revision_id
 
1297
        else:
 
1298
            graph = self.repository.get_graph()
715
1299
            try:
716
 
                new_history = new_history[:new_history.index(revision_id) + 1]
717
 
            except ValueError:
718
 
                rev = self.repository.get_revision(revision_id)
719
 
                new_history = rev.get_history(self.repository)[1:]
720
 
        destination.set_revision_history(new_history)
 
1300
                revno = graph.find_distance_to_null(revision_id, 
 
1301
                    [(source_revision_id, source_revno)])
 
1302
            except errors.GhostRevisionsHaveNoRevno:
 
1303
                # Default to 1, if we can't find anything else
 
1304
                revno = 1
 
1305
        destination.set_last_revision_info(revno, revision_id)
721
1306
 
722
 
    @needs_read_lock
723
1307
    def copy_content_into(self, destination, revision_id=None):
724
1308
        """Copy the content of self into destination.
725
1309
 
726
1310
        revision_id: if not None, the revision history in the new branch will
727
1311
                     be truncated to end with revision_id.
728
1312
        """
729
 
        self._synchronize_history(destination, revision_id)
730
 
        try:
731
 
            parent = self.get_parent()
732
 
        except errors.InaccessibleParent, e:
733
 
            mutter('parent was not accessible to copy: %s', e)
734
 
        else:
735
 
            if parent:
736
 
                destination.set_parent(parent)
737
 
        self.tags.merge_to(destination.tags)
 
1313
        return InterBranch.get(self, destination).copy_content_into(
 
1314
            revision_id=revision_id)
 
1315
 
 
1316
    def update_references(self, target):
 
1317
        if not getattr(self._format, 'supports_reference_locations', False):
 
1318
            return
 
1319
        reference_dict = self._get_all_reference_info()
 
1320
        if len(reference_dict) == 0:
 
1321
            return
 
1322
        old_base = self.base
 
1323
        new_base = target.base
 
1324
        target_reference_dict = target._get_all_reference_info()
 
1325
        for file_id, (tree_path, branch_location) in (
 
1326
            reference_dict.items()):
 
1327
            branch_location = urlutils.rebase_url(branch_location,
 
1328
                                                  old_base, new_base)
 
1329
            target_reference_dict.setdefault(
 
1330
                file_id, (tree_path, branch_location))
 
1331
        target._set_all_reference_info(target_reference_dict)
738
1332
 
739
1333
    @needs_read_lock
740
 
    def check(self):
 
1334
    def check(self, refs):
741
1335
        """Check consistency of the branch.
742
1336
 
743
1337
        In particular this checks that revisions given in the revision-history
744
 
        do actually match up in the revision graph, and that they're all 
 
1338
        do actually match up in the revision graph, and that they're all
745
1339
        present in the repository.
746
 
        
 
1340
 
747
1341
        Callers will typically also want to check the repository.
748
1342
 
 
1343
        :param refs: Calculated refs for this branch as specified by
 
1344
            branch._get_check_refs()
749
1345
        :return: A BranchCheckResult.
750
1346
        """
751
 
        mainline_parent_id = None
 
1347
        result = BranchCheckResult(self)
752
1348
        last_revno, last_revision_id = self.last_revision_info()
753
 
        real_rev_history = list(self.repository.iter_reverse_revision_history(
754
 
                                last_revision_id))
755
 
        real_rev_history.reverse()
756
 
        if len(real_rev_history) != last_revno:
757
 
            raise errors.BzrCheckError('revno does not match len(mainline)'
758
 
                ' %s != %s' % (last_revno, len(real_rev_history)))
759
 
        # TODO: We should probably also check that real_rev_history actually
760
 
        #       matches self.revision_history()
761
 
        for revision_id in real_rev_history:
762
 
            try:
763
 
                revision = self.repository.get_revision(revision_id)
764
 
            except errors.NoSuchRevision, e:
765
 
                raise errors.BzrCheckError("mainline revision {%s} not in repository"
766
 
                            % revision_id)
767
 
            # In general the first entry on the revision history has no parents.
768
 
            # But it's not illegal for it to have parents listed; this can happen
769
 
            # in imports from Arch when the parents weren't reachable.
770
 
            if mainline_parent_id is not None:
771
 
                if mainline_parent_id not in revision.parent_ids:
772
 
                    raise errors.BzrCheckError("previous revision {%s} not listed among "
773
 
                                        "parents of {%s}"
774
 
                                        % (mainline_parent_id, revision_id))
775
 
            mainline_parent_id = revision_id
776
 
        return BranchCheckResult(self)
 
1349
        actual_revno = refs[('lefthand-distance', last_revision_id)]
 
1350
        if actual_revno != last_revno:
 
1351
            result.errors.append(errors.BzrCheckError(
 
1352
                'revno does not match len(mainline) %s != %s' % (
 
1353
                last_revno, actual_revno)))
 
1354
        # TODO: We should probably also check that self.revision_history
 
1355
        # matches the repository for older branch formats.
 
1356
        # If looking for the code that cross-checks repository parents against
 
1357
        # the iter_reverse_revision_history output, that is now a repository
 
1358
        # specific check.
 
1359
        return result
777
1360
 
778
1361
    def _get_checkout_format(self):
779
1362
        """Return the most suitable metadir for a checkout of this branch.
788
1371
            format.set_branch_format(self._format)
789
1372
        return format
790
1373
 
 
1374
    def create_clone_on_transport(self, to_transport, revision_id=None,
 
1375
        stacked_on=None, create_prefix=False, use_existing_dir=False):
 
1376
        """Create a clone of this branch and its bzrdir.
 
1377
 
 
1378
        :param to_transport: The transport to clone onto.
 
1379
        :param revision_id: The revision id to use as tip in the new branch.
 
1380
            If None the tip is obtained from this branch.
 
1381
        :param stacked_on: An optional URL to stack the clone on.
 
1382
        :param create_prefix: Create any missing directories leading up to
 
1383
            to_transport.
 
1384
        :param use_existing_dir: Use an existing directory if one exists.
 
1385
        """
 
1386
        # XXX: Fix the bzrdir API to allow getting the branch back from the
 
1387
        # clone call. Or something. 20090224 RBC/spiv.
 
1388
        # XXX: Should this perhaps clone colocated branches as well, 
 
1389
        # rather than just the default branch? 20100319 JRV
 
1390
        if revision_id is None:
 
1391
            revision_id = self.last_revision()
 
1392
        dir_to = self.bzrdir.clone_on_transport(to_transport,
 
1393
            revision_id=revision_id, stacked_on=stacked_on,
 
1394
            create_prefix=create_prefix, use_existing_dir=use_existing_dir)
 
1395
        return dir_to.open_branch()
 
1396
 
791
1397
    def create_checkout(self, to_location, revision_id=None,
792
1398
                        lightweight=False, accelerator_tree=None,
793
1399
                        hardlink=False):
794
1400
        """Create a checkout of a branch.
795
 
        
 
1401
 
796
1402
        :param to_location: The url to produce the checkout at
797
1403
        :param revision_id: The revision to check out
798
1404
        :param lightweight: If True, produce a lightweight checkout, otherwise,
810
1416
        if lightweight:
811
1417
            format = self._get_checkout_format()
812
1418
            checkout = format.initialize_on_transport(t)
813
 
            from_branch = BranchReferenceFormat().initialize(checkout, self)
 
1419
            from_branch = BranchReferenceFormat().initialize(checkout, 
 
1420
                target_branch=self)
814
1421
        else:
815
1422
            format = self._get_checkout_format()
816
1423
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
817
1424
                to_location, force_new_tree=False, format=format)
818
1425
            checkout = checkout_branch.bzrdir
819
1426
            checkout_branch.bind(self)
820
 
            # pull up to the specified revision_id to set the initial 
 
1427
            # pull up to the specified revision_id to set the initial
821
1428
            # branch tip correctly, and seed it with history.
822
1429
            checkout_branch.pull(self, stop_revision=revision_id)
823
1430
            from_branch=None
845
1452
        reconciler.reconcile()
846
1453
        return reconciler
847
1454
 
848
 
    def reference_parent(self, file_id, path):
 
1455
    def reference_parent(self, file_id, path, possible_transports=None):
849
1456
        """Return the parent branch for a tree-reference file_id
850
1457
        :param file_id: The file_id of the tree reference
851
1458
        :param path: The path of the file_id in the tree
852
1459
        :return: A branch associated with the file_id
853
1460
        """
854
1461
        # FIXME should provide multiple branches, based on config
855
 
        return Branch.open(self.bzrdir.root_transport.clone(path).base)
 
1462
        return Branch.open(self.bzrdir.root_transport.clone(path).base,
 
1463
                           possible_transports=possible_transports)
856
1464
 
857
1465
    def supports_tags(self):
858
1466
        return self._format.supports_tags()
859
1467
 
 
1468
    def automatic_tag_name(self, revision_id):
 
1469
        """Try to automatically find the tag name for a revision.
 
1470
 
 
1471
        :param revision_id: Revision id of the revision.
 
1472
        :return: A tag name or None if no tag name could be determined.
 
1473
        """
 
1474
        for hook in Branch.hooks['automatic_tag_name']:
 
1475
            ret = hook(self, revision_id)
 
1476
            if ret is not None:
 
1477
                return ret
 
1478
        return None
 
1479
 
860
1480
    def _check_if_descendant_or_diverged(self, revision_a, revision_b, graph,
861
1481
                                         other_branch):
862
1482
        """Ensure that revision_b is a descendant of revision_a.
863
1483
 
864
1484
        This is a helper function for update_revisions.
865
 
        
 
1485
 
866
1486
        :raises: DivergedBranches if revision_b has diverged from revision_a.
867
1487
        :returns: True if revision_b is a descendant of revision_a.
868
1488
        """
874
1494
        elif relation == 'a_descends_from_b':
875
1495
            return False
876
1496
        else:
877
 
            raise AssertionError("invalid heads: %r" % heads)
 
1497
            raise AssertionError("invalid relation: %r" % (relation,))
878
1498
 
879
1499
    def _revision_relations(self, revision_a, revision_b, graph):
880
1500
        """Determine the relationship between two revisions.
881
 
        
 
1501
 
882
1502
        :returns: One of: 'a_descends_from_b', 'b_descends_from_a', 'diverged'
883
1503
        """
884
1504
        heads = graph.heads([revision_a, revision_b])
890
1510
        elif heads == set([revision_a]):
891
1511
            return 'a_descends_from_b'
892
1512
        else:
893
 
            raise AssertionError("invalid heads: %r" % heads)
 
1513
            raise AssertionError("invalid heads: %r" % (heads,))
894
1514
 
895
1515
 
896
1516
class BranchFormat(object):
901
1521
     * a format string,
902
1522
     * an open routine.
903
1523
 
904
 
    Formats are placed in an dict by their format string for reference 
905
 
    during branch opening. Its not required that these be instances, they
906
 
    can be classes themselves with class methods - it simply depends on 
 
1524
    Formats are placed in an dict by their format string for reference
 
1525
    during branch opening. It's not required that these be instances, they
 
1526
    can be classes themselves with class methods - it simply depends on
907
1527
    whether state is needed for a given format or not.
908
1528
 
909
1529
    Once a format is deprecated, just deprecate the initialize and open
910
 
    methods on the format class. Do not deprecate the object, as the 
 
1530
    methods on the format class. Do not deprecate the object, as the
911
1531
    object will be created every time regardless.
912
1532
    """
913
1533
 
917
1537
    _formats = {}
918
1538
    """The known formats."""
919
1539
 
 
1540
    can_set_append_revisions_only = True
 
1541
 
920
1542
    def __eq__(self, other):
921
1543
        return self.__class__ is other.__class__
922
1544
 
924
1546
        return not (self == other)
925
1547
 
926
1548
    @classmethod
927
 
    def find_format(klass, a_bzrdir):
 
1549
    def find_format(klass, a_bzrdir, name=None):
928
1550
        """Return the format for the branch object in a_bzrdir."""
929
1551
        try:
930
 
            transport = a_bzrdir.get_branch_transport(None)
931
 
            format_string = transport.get("format").read()
932
 
            return klass._formats[format_string]
 
1552
            transport = a_bzrdir.get_branch_transport(None, name=name)
 
1553
            format_string = transport.get_bytes("format")
 
1554
            format = klass._formats[format_string]
 
1555
            if isinstance(format, MetaDirBranchFormatFactory):
 
1556
                return format()
 
1557
            return format
933
1558
        except errors.NoSuchFile:
934
 
            raise errors.NotBranchError(path=transport.base)
 
1559
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
935
1560
        except KeyError:
936
1561
            raise errors.UnknownFormatError(format=format_string, kind='branch')
937
1562
 
940
1565
        """Return the current default format."""
941
1566
        return klass._default_format
942
1567
 
943
 
    def get_reference(self, a_bzrdir):
 
1568
    @classmethod
 
1569
    def get_formats(klass):
 
1570
        """Get all the known formats.
 
1571
 
 
1572
        Warning: This triggers a load of all lazy registered formats: do not
 
1573
        use except when that is desireed.
 
1574
        """
 
1575
        result = []
 
1576
        for fmt in klass._formats.values():
 
1577
            if isinstance(fmt, MetaDirBranchFormatFactory):
 
1578
                fmt = fmt()
 
1579
            result.append(fmt)
 
1580
        return result
 
1581
 
 
1582
    def get_reference(self, a_bzrdir, name=None):
944
1583
        """Get the target reference of the branch in a_bzrdir.
945
1584
 
946
1585
        format probing must have been completed before calling
948
1587
        in a_bzrdir is correct.
949
1588
 
950
1589
        :param a_bzrdir: The bzrdir to get the branch data from.
 
1590
        :param name: Name of the colocated branch to fetch
951
1591
        :return: None if the branch is not a reference branch.
952
1592
        """
953
1593
        return None
954
1594
 
955
1595
    @classmethod
956
 
    def set_reference(self, a_bzrdir, to_branch):
 
1596
    def set_reference(self, a_bzrdir, name, to_branch):
957
1597
        """Set the target reference of the branch in a_bzrdir.
958
1598
 
959
1599
        format probing must have been completed before calling
961
1601
        in a_bzrdir is correct.
962
1602
 
963
1603
        :param a_bzrdir: The bzrdir to set the branch reference for.
 
1604
        :param name: Name of colocated branch to set, None for default
964
1605
        :param to_branch: branch that the checkout is to reference
965
1606
        """
966
1607
        raise NotImplementedError(self.set_reference)
973
1614
        """Return the short format description for this format."""
974
1615
        raise NotImplementedError(self.get_format_description)
975
1616
 
976
 
    def _initialize_helper(self, a_bzrdir, utf8_files, lock_type='metadir',
977
 
                           set_format=True):
 
1617
    def _run_post_branch_init_hooks(self, a_bzrdir, name, branch):
 
1618
        hooks = Branch.hooks['post_branch_init']
 
1619
        if not hooks:
 
1620
            return
 
1621
        params = BranchInitHookParams(self, a_bzrdir, name, branch)
 
1622
        for hook in hooks:
 
1623
            hook(params)
 
1624
 
 
1625
    def _initialize_helper(self, a_bzrdir, utf8_files, name=None,
 
1626
                           lock_type='metadir', set_format=True):
978
1627
        """Initialize a branch in a bzrdir, with specified files
979
1628
 
980
1629
        :param a_bzrdir: The bzrdir to initialize the branch in
981
1630
        :param utf8_files: The files to create as a list of
982
1631
            (filename, content) tuples
 
1632
        :param name: Name of colocated branch to create, if any
983
1633
        :param set_format: If True, set the format with
984
1634
            self.get_format_string.  (BzrBranch4 has its format set
985
1635
            elsewhere)
986
1636
        :return: a branch in this format
987
1637
        """
988
 
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
989
 
        branch_transport = a_bzrdir.get_branch_transport(self)
 
1638
        mutter('creating branch %r in %s', self, a_bzrdir.user_url)
 
1639
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
990
1640
        lock_map = {
991
1641
            'metadir': ('lock', lockdir.LockDir),
992
1642
            'branch4': ('branch-lock', lockable_files.TransportLock),
995
1645
        control_files = lockable_files.LockableFiles(branch_transport,
996
1646
            lock_name, lock_class)
997
1647
        control_files.create_lock()
998
 
        control_files.lock_write()
 
1648
        try:
 
1649
            control_files.lock_write()
 
1650
        except errors.LockContention:
 
1651
            if lock_type != 'branch4':
 
1652
                raise
 
1653
            lock_taken = False
 
1654
        else:
 
1655
            lock_taken = True
999
1656
        if set_format:
1000
1657
            utf8_files += [('format', self.get_format_string())]
1001
1658
        try:
1004
1661
                    filename, content,
1005
1662
                    mode=a_bzrdir._get_file_mode())
1006
1663
        finally:
1007
 
            control_files.unlock()
1008
 
        return self.open(a_bzrdir, _found=True)
 
1664
            if lock_taken:
 
1665
                control_files.unlock()
 
1666
        branch = self.open(a_bzrdir, name, _found=True)
 
1667
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
 
1668
        return branch
1009
1669
 
1010
 
    def initialize(self, a_bzrdir):
1011
 
        """Create a branch of this format in a_bzrdir."""
 
1670
    def initialize(self, a_bzrdir, name=None):
 
1671
        """Create a branch of this format in a_bzrdir.
 
1672
        
 
1673
        :param name: Name of the colocated branch to create.
 
1674
        """
1012
1675
        raise NotImplementedError(self.initialize)
1013
1676
 
1014
1677
    def is_supported(self):
1015
1678
        """Is this format supported?
1016
1679
 
1017
1680
        Supported formats can be initialized and opened.
1018
 
        Unsupported formats may not support initialization or committing or 
 
1681
        Unsupported formats may not support initialization or committing or
1019
1682
        some other features depending on the reason for not being supported.
1020
1683
        """
1021
1684
        return True
1022
1685
 
1023
 
    def open(self, a_bzrdir, _found=False):
 
1686
    def make_tags(self, branch):
 
1687
        """Create a tags object for branch.
 
1688
 
 
1689
        This method is on BranchFormat, because BranchFormats are reflected
 
1690
        over the wire via network_name(), whereas full Branch instances require
 
1691
        multiple VFS method calls to operate at all.
 
1692
 
 
1693
        The default implementation returns a disabled-tags instance.
 
1694
 
 
1695
        Note that it is normal for branch to be a RemoteBranch when using tags
 
1696
        on a RemoteBranch.
 
1697
        """
 
1698
        return DisabledTags(branch)
 
1699
 
 
1700
    def network_name(self):
 
1701
        """A simple byte string uniquely identifying this format for RPC calls.
 
1702
 
 
1703
        MetaDir branch formats use their disk format string to identify the
 
1704
        repository over the wire. All in one formats such as bzr < 0.8, and
 
1705
        foreign formats like svn/git and hg should use some marker which is
 
1706
        unique and immutable.
 
1707
        """
 
1708
        raise NotImplementedError(self.network_name)
 
1709
 
 
1710
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
1024
1711
        """Return the branch object for a_bzrdir
1025
1712
 
1026
 
        _found is a private parameter, do not use it. It is used to indicate
1027
 
               if format probing has already be done.
 
1713
        :param a_bzrdir: A BzrDir that contains a branch.
 
1714
        :param name: Name of colocated branch to open
 
1715
        :param _found: a private parameter, do not use it. It is used to
 
1716
            indicate if format probing has already be done.
 
1717
        :param ignore_fallbacks: when set, no fallback branches will be opened
 
1718
            (if there are any).  Default is to open fallbacks.
1028
1719
        """
1029
1720
        raise NotImplementedError(self.open)
1030
1721
 
1031
1722
    @classmethod
1032
1723
    def register_format(klass, format):
 
1724
        """Register a metadir format.
 
1725
        
 
1726
        See MetaDirBranchFormatFactory for the ability to register a format
 
1727
        without loading the code the format needs until it is actually used.
 
1728
        """
1033
1729
        klass._formats[format.get_format_string()] = format
 
1730
        # Metadir formats have a network name of their format string, and get
 
1731
        # registered as factories.
 
1732
        if isinstance(format, MetaDirBranchFormatFactory):
 
1733
            network_format_registry.register(format.get_format_string(), format)
 
1734
        else:
 
1735
            network_format_registry.register(format.get_format_string(),
 
1736
                format.__class__)
1034
1737
 
1035
1738
    @classmethod
1036
1739
    def set_default_format(klass, format):
1037
1740
        klass._default_format = format
1038
1741
 
 
1742
    def supports_set_append_revisions_only(self):
 
1743
        """True if this format supports set_append_revisions_only."""
 
1744
        return False
 
1745
 
 
1746
    def supports_stacking(self):
 
1747
        """True if this format records a stacked-on branch."""
 
1748
        return False
 
1749
 
1039
1750
    @classmethod
1040
1751
    def unregister_format(klass, format):
1041
1752
        del klass._formats[format.get_format_string()]
1042
1753
 
1043
1754
    def __str__(self):
1044
 
        return self.get_format_string().rstrip()
 
1755
        return self.get_format_description().rstrip()
1045
1756
 
1046
1757
    def supports_tags(self):
1047
1758
        """True if this format supports tags stored in the branch"""
1048
1759
        return False  # by default
1049
1760
 
1050
1761
 
 
1762
class MetaDirBranchFormatFactory(registry._LazyObjectGetter):
 
1763
    """A factory for a BranchFormat object, permitting simple lazy registration.
 
1764
    
 
1765
    While none of the built in BranchFormats are lazy registered yet,
 
1766
    bzrlib.tests.test_branch.TestMetaDirBranchFormatFactory demonstrates how to
 
1767
    use it, and the bzr-loom plugin uses it as well (see
 
1768
    bzrlib.plugins.loom.formats).
 
1769
    """
 
1770
 
 
1771
    def __init__(self, format_string, module_name, member_name):
 
1772
        """Create a MetaDirBranchFormatFactory.
 
1773
 
 
1774
        :param format_string: The format string the format has.
 
1775
        :param module_name: Module to load the format class from.
 
1776
        :param member_name: Attribute name within the module for the format class.
 
1777
        """
 
1778
        registry._LazyObjectGetter.__init__(self, module_name, member_name)
 
1779
        self._format_string = format_string
 
1780
        
 
1781
    def get_format_string(self):
 
1782
        """See BranchFormat.get_format_string."""
 
1783
        return self._format_string
 
1784
 
 
1785
    def __call__(self):
 
1786
        """Used for network_format_registry support."""
 
1787
        return self.get_obj()()
 
1788
 
 
1789
 
1051
1790
class BranchHooks(Hooks):
1052
1791
    """A dictionary mapping hook name to a list of callables for branch hooks.
1053
 
    
 
1792
 
1054
1793
    e.g. ['set_rh'] Is the list of items to be called when the
1055
1794
    set_revision_history function is invoked.
1056
1795
    """
1062
1801
        notified.
1063
1802
        """
1064
1803
        Hooks.__init__(self)
1065
 
        # Introduced in 0.15:
1066
 
        # invoked whenever the revision history has been set
1067
 
        # with set_revision_history. The api signature is
1068
 
        # (branch, revision_history), and the branch will
1069
 
        # be write-locked.
1070
 
        self['set_rh'] = []
1071
 
        # invoked after a push operation completes.
1072
 
        # the api signature is
1073
 
        # (push_result)
1074
 
        # containing the members
1075
 
        # (source, local, master, old_revno, old_revid, new_revno, new_revid)
1076
 
        # where local is the local target branch or None, master is the target 
1077
 
        # master branch, and the rest should be self explanatory. The source
1078
 
        # is read locked and the target branches write locked. Source will
1079
 
        # be the local low-latency branch.
1080
 
        self['post_push'] = []
1081
 
        # invoked after a pull operation completes.
1082
 
        # the api signature is
1083
 
        # (pull_result)
1084
 
        # containing the members
1085
 
        # (source, local, master, old_revno, old_revid, new_revno, new_revid)
1086
 
        # where local is the local branch or None, master is the target 
1087
 
        # master branch, and the rest should be self explanatory. The source
1088
 
        # is read locked and the target branches write locked. The local
1089
 
        # branch is the low-latency branch.
1090
 
        self['post_pull'] = []
1091
 
        # invoked before a commit operation takes place.
1092
 
        # the api signature is
1093
 
        # (local, master, old_revno, old_revid, future_revno, future_revid,
1094
 
        #  tree_delta, future_tree).
1095
 
        # old_revid is NULL_REVISION for the first commit to a branch
1096
 
        # tree_delta is a TreeDelta object describing changes from the basis
1097
 
        # revision, hooks MUST NOT modify this delta
1098
 
        # future_tree is an in-memory tree obtained from
1099
 
        # CommitBuilder.revision_tree() and hooks MUST NOT modify this tree
1100
 
        self['pre_commit'] = []
1101
 
        # invoked after a commit operation completes.
1102
 
        # the api signature is 
1103
 
        # (local, master, old_revno, old_revid, new_revno, new_revid)
1104
 
        # old_revid is NULL_REVISION for the first commit to a branch.
1105
 
        self['post_commit'] = []
1106
 
        # invoked after a uncommit operation completes.
1107
 
        # the api signature is
1108
 
        # (local, master, old_revno, old_revid, new_revno, new_revid) where
1109
 
        # local is the local branch or None, master is the target branch,
1110
 
        # and an empty branch recieves new_revno of 0, new_revid of None.
1111
 
        self['post_uncommit'] = []
1112
 
        # Introduced in 1.6
1113
 
        # Invoked before the tip of a branch changes.
1114
 
        # the api signature is
1115
 
        # (params) where params is a ChangeBranchTipParams with the members
1116
 
        # (branch, old_revno, new_revno, old_revid, new_revid)
1117
 
        self['pre_change_branch_tip'] = []
1118
 
        # Introduced in 1.4
1119
 
        # Invoked after the tip of a branch changes.
1120
 
        # the api signature is
1121
 
        # (params) where params is a ChangeBranchTipParams with the members
1122
 
        # (branch, old_revno, new_revno, old_revid, new_revid)
1123
 
        self['post_change_branch_tip'] = []
 
1804
        self.create_hook(HookPoint('set_rh',
 
1805
            "Invoked whenever the revision history has been set via "
 
1806
            "set_revision_history. The api signature is (branch, "
 
1807
            "revision_history), and the branch will be write-locked. "
 
1808
            "The set_rh hook can be expensive for bzr to trigger, a better "
 
1809
            "hook to use is Branch.post_change_branch_tip.", (0, 15), None))
 
1810
        self.create_hook(HookPoint('open',
 
1811
            "Called with the Branch object that has been opened after a "
 
1812
            "branch is opened.", (1, 8), None))
 
1813
        self.create_hook(HookPoint('post_push',
 
1814
            "Called after a push operation completes. post_push is called "
 
1815
            "with a bzrlib.branch.BranchPushResult object and only runs in the "
 
1816
            "bzr client.", (0, 15), None))
 
1817
        self.create_hook(HookPoint('post_pull',
 
1818
            "Called after a pull operation completes. post_pull is called "
 
1819
            "with a bzrlib.branch.PullResult object and only runs in the "
 
1820
            "bzr client.", (0, 15), None))
 
1821
        self.create_hook(HookPoint('pre_commit',
 
1822
            "Called after a commit is calculated but before it is "
 
1823
            "completed. pre_commit is called with (local, master, old_revno, "
 
1824
            "old_revid, future_revno, future_revid, tree_delta, future_tree"
 
1825
            "). old_revid is NULL_REVISION for the first commit to a branch, "
 
1826
            "tree_delta is a TreeDelta object describing changes from the "
 
1827
            "basis revision. hooks MUST NOT modify this delta. "
 
1828
            " future_tree is an in-memory tree obtained from "
 
1829
            "CommitBuilder.revision_tree() and hooks MUST NOT modify this "
 
1830
            "tree.", (0,91), None))
 
1831
        self.create_hook(HookPoint('post_commit',
 
1832
            "Called in the bzr client after a commit has completed. "
 
1833
            "post_commit is called with (local, master, old_revno, old_revid, "
 
1834
            "new_revno, new_revid). old_revid is NULL_REVISION for the first "
 
1835
            "commit to a branch.", (0, 15), None))
 
1836
        self.create_hook(HookPoint('post_uncommit',
 
1837
            "Called in the bzr client after an uncommit completes. "
 
1838
            "post_uncommit is called with (local, master, old_revno, "
 
1839
            "old_revid, new_revno, new_revid) where local is the local branch "
 
1840
            "or None, master is the target branch, and an empty branch "
 
1841
            "receives new_revno of 0, new_revid of None.", (0, 15), None))
 
1842
        self.create_hook(HookPoint('pre_change_branch_tip',
 
1843
            "Called in bzr client and server before a change to the tip of a "
 
1844
            "branch is made. pre_change_branch_tip is called with a "
 
1845
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
 
1846
            "commit, uncommit will all trigger this hook.", (1, 6), None))
 
1847
        self.create_hook(HookPoint('post_change_branch_tip',
 
1848
            "Called in bzr client and server after a change to the tip of a "
 
1849
            "branch is made. post_change_branch_tip is called with a "
 
1850
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
 
1851
            "commit, uncommit will all trigger this hook.", (1, 4), None))
 
1852
        self.create_hook(HookPoint('transform_fallback_location',
 
1853
            "Called when a stacked branch is activating its fallback "
 
1854
            "locations. transform_fallback_location is called with (branch, "
 
1855
            "url), and should return a new url. Returning the same url "
 
1856
            "allows it to be used as-is, returning a different one can be "
 
1857
            "used to cause the branch to stack on a closer copy of that "
 
1858
            "fallback_location. Note that the branch cannot have history "
 
1859
            "accessing methods called on it during this hook because the "
 
1860
            "fallback locations have not been activated. When there are "
 
1861
            "multiple hooks installed for transform_fallback_location, "
 
1862
            "all are called with the url returned from the previous hook."
 
1863
            "The order is however undefined.", (1, 9), None))
 
1864
        self.create_hook(HookPoint('automatic_tag_name',
 
1865
            "Called to determine an automatic tag name for a revision. "
 
1866
            "automatic_tag_name is called with (branch, revision_id) and "
 
1867
            "should return a tag name or None if no tag name could be "
 
1868
            "determined. The first non-None tag name returned will be used.",
 
1869
            (2, 2), None))
 
1870
        self.create_hook(HookPoint('post_branch_init',
 
1871
            "Called after new branch initialization completes. "
 
1872
            "post_branch_init is called with a "
 
1873
            "bzrlib.branch.BranchInitHookParams. "
 
1874
            "Note that init, branch and checkout (both heavyweight and "
 
1875
            "lightweight) will all trigger this hook.", (2, 2), None))
 
1876
        self.create_hook(HookPoint('post_switch',
 
1877
            "Called after a checkout switches branch. "
 
1878
            "post_switch is called with a "
 
1879
            "bzrlib.branch.SwitchHookParams.", (2, 2), None))
 
1880
 
1124
1881
 
1125
1882
 
1126
1883
# install the default hooks into the Branch class.
1158
1915
 
1159
1916
    def __eq__(self, other):
1160
1917
        return self.__dict__ == other.__dict__
1161
 
    
 
1918
 
1162
1919
    def __repr__(self):
1163
1920
        return "<%s of %s from (%s, %s) to (%s, %s)>" % (
1164
 
            self.__class__.__name__, self.branch, 
 
1921
            self.__class__.__name__, self.branch,
1165
1922
            self.old_revno, self.old_revid, self.new_revno, self.new_revid)
1166
1923
 
1167
1924
 
 
1925
class BranchInitHookParams(object):
 
1926
    """Object holding parameters passed to *_branch_init hooks.
 
1927
 
 
1928
    There are 4 fields that hooks may wish to access:
 
1929
 
 
1930
    :ivar format: the branch format
 
1931
    :ivar bzrdir: the BzrDir where the branch will be/has been initialized
 
1932
    :ivar name: name of colocated branch, if any (or None)
 
1933
    :ivar branch: the branch created
 
1934
 
 
1935
    Note that for lightweight checkouts, the bzrdir and format fields refer to
 
1936
    the checkout, hence they are different from the corresponding fields in
 
1937
    branch, which refer to the original branch.
 
1938
    """
 
1939
 
 
1940
    def __init__(self, format, a_bzrdir, name, branch):
 
1941
        """Create a group of BranchInitHook parameters.
 
1942
 
 
1943
        :param format: the branch format
 
1944
        :param a_bzrdir: the BzrDir where the branch will be/has been
 
1945
            initialized
 
1946
        :param name: name of colocated branch, if any (or None)
 
1947
        :param branch: the branch created
 
1948
 
 
1949
        Note that for lightweight checkouts, the bzrdir and format fields refer
 
1950
        to the checkout, hence they are different from the corresponding fields
 
1951
        in branch, which refer to the original branch.
 
1952
        """
 
1953
        self.format = format
 
1954
        self.bzrdir = a_bzrdir
 
1955
        self.name = name
 
1956
        self.branch = branch
 
1957
 
 
1958
    def __eq__(self, other):
 
1959
        return self.__dict__ == other.__dict__
 
1960
 
 
1961
    def __repr__(self):
 
1962
        return "<%s of %s>" % (self.__class__.__name__, self.branch)
 
1963
 
 
1964
 
 
1965
class SwitchHookParams(object):
 
1966
    """Object holding parameters passed to *_switch hooks.
 
1967
 
 
1968
    There are 4 fields that hooks may wish to access:
 
1969
 
 
1970
    :ivar control_dir: BzrDir of the checkout to change
 
1971
    :ivar to_branch: branch that the checkout is to reference
 
1972
    :ivar force: skip the check for local commits in a heavy checkout
 
1973
    :ivar revision_id: revision ID to switch to (or None)
 
1974
    """
 
1975
 
 
1976
    def __init__(self, control_dir, to_branch, force, revision_id):
 
1977
        """Create a group of SwitchHook parameters.
 
1978
 
 
1979
        :param control_dir: BzrDir of the checkout to change
 
1980
        :param to_branch: branch that the checkout is to reference
 
1981
        :param force: skip the check for local commits in a heavy checkout
 
1982
        :param revision_id: revision ID to switch to (or None)
 
1983
        """
 
1984
        self.control_dir = control_dir
 
1985
        self.to_branch = to_branch
 
1986
        self.force = force
 
1987
        self.revision_id = revision_id
 
1988
 
 
1989
    def __eq__(self, other):
 
1990
        return self.__dict__ == other.__dict__
 
1991
 
 
1992
    def __repr__(self):
 
1993
        return "<%s for %s to (%s, %s)>" % (self.__class__.__name__,
 
1994
            self.control_dir, self.to_branch,
 
1995
            self.revision_id)
 
1996
 
 
1997
 
1168
1998
class BzrBranchFormat4(BranchFormat):
1169
1999
    """Bzr branch format 4.
1170
2000
 
1177
2007
        """See BranchFormat.get_format_description()."""
1178
2008
        return "Branch format 4"
1179
2009
 
1180
 
    def initialize(self, a_bzrdir):
 
2010
    def initialize(self, a_bzrdir, name=None):
1181
2011
        """Create a branch of this format in a_bzrdir."""
1182
2012
        utf8_files = [('revision-history', ''),
1183
2013
                      ('branch-name', ''),
1184
2014
                      ]
1185
 
        return self._initialize_helper(a_bzrdir, utf8_files,
 
2015
        return self._initialize_helper(a_bzrdir, utf8_files, name=name,
1186
2016
                                       lock_type='branch4', set_format=False)
1187
2017
 
1188
2018
    def __init__(self):
1189
2019
        super(BzrBranchFormat4, self).__init__()
1190
2020
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
1191
2021
 
1192
 
    def open(self, a_bzrdir, _found=False):
1193
 
        """Return the branch object for a_bzrdir
 
2022
    def network_name(self):
 
2023
        """The network name for this format is the control dirs disk label."""
 
2024
        return self._matchingbzrdir.get_format_string()
1194
2025
 
1195
 
        _found is a private parameter, do not use it. It is used to indicate
1196
 
               if format probing has already be done.
1197
 
        """
 
2026
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
 
2027
        """See BranchFormat.open()."""
1198
2028
        if not _found:
1199
2029
            # we are being called directly and must probe.
1200
2030
            raise NotImplementedError
1201
2031
        return BzrBranch(_format=self,
1202
2032
                         _control_files=a_bzrdir._control_files,
1203
2033
                         a_bzrdir=a_bzrdir,
 
2034
                         name=name,
1204
2035
                         _repository=a_bzrdir.open_repository())
1205
2036
 
1206
2037
    def __str__(self):
1207
2038
        return "Bazaar-NG branch format 4"
1208
2039
 
1209
2040
 
1210
 
class BzrBranchFormat5(BranchFormat):
 
2041
class BranchFormatMetadir(BranchFormat):
 
2042
    """Common logic for meta-dir based branch formats."""
 
2043
 
 
2044
    def _branch_class(self):
 
2045
        """What class to instantiate on open calls."""
 
2046
        raise NotImplementedError(self._branch_class)
 
2047
 
 
2048
    def network_name(self):
 
2049
        """A simple byte string uniquely identifying this format for RPC calls.
 
2050
 
 
2051
        Metadir branch formats use their format string.
 
2052
        """
 
2053
        return self.get_format_string()
 
2054
 
 
2055
    def open(self, a_bzrdir, name=None, _found=False, ignore_fallbacks=False):
 
2056
        """See BranchFormat.open()."""
 
2057
        if not _found:
 
2058
            format = BranchFormat.find_format(a_bzrdir, name=name)
 
2059
            if format.__class__ != self.__class__:
 
2060
                raise AssertionError("wrong format %r found for %r" %
 
2061
                    (format, self))
 
2062
        transport = a_bzrdir.get_branch_transport(None, name=name)
 
2063
        try:
 
2064
            control_files = lockable_files.LockableFiles(transport, 'lock',
 
2065
                                                         lockdir.LockDir)
 
2066
            return self._branch_class()(_format=self,
 
2067
                              _control_files=control_files,
 
2068
                              name=name,
 
2069
                              a_bzrdir=a_bzrdir,
 
2070
                              _repository=a_bzrdir.find_repository(),
 
2071
                              ignore_fallbacks=ignore_fallbacks)
 
2072
        except errors.NoSuchFile:
 
2073
            raise errors.NotBranchError(path=transport.base, bzrdir=a_bzrdir)
 
2074
 
 
2075
    def __init__(self):
 
2076
        super(BranchFormatMetadir, self).__init__()
 
2077
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
 
2078
        self._matchingbzrdir.set_branch_format(self)
 
2079
 
 
2080
    def supports_tags(self):
 
2081
        return True
 
2082
 
 
2083
 
 
2084
class BzrBranchFormat5(BranchFormatMetadir):
1211
2085
    """Bzr branch format 5.
1212
2086
 
1213
2087
    This format has:
1220
2094
    This format is new in bzr 0.8.
1221
2095
    """
1222
2096
 
 
2097
    def _branch_class(self):
 
2098
        return BzrBranch5
 
2099
 
1223
2100
    def get_format_string(self):
1224
2101
        """See BranchFormat.get_format_string()."""
1225
2102
        return "Bazaar-NG branch format 5\n"
1227
2104
    def get_format_description(self):
1228
2105
        """See BranchFormat.get_format_description()."""
1229
2106
        return "Branch format 5"
1230
 
        
1231
 
    def initialize(self, a_bzrdir):
 
2107
 
 
2108
    def initialize(self, a_bzrdir, name=None):
1232
2109
        """Create a branch of this format in a_bzrdir."""
1233
2110
        utf8_files = [('revision-history', ''),
1234
2111
                      ('branch-name', ''),
1235
2112
                      ]
1236
 
        return self._initialize_helper(a_bzrdir, utf8_files)
1237
 
 
1238
 
    def __init__(self):
1239
 
        super(BzrBranchFormat5, self).__init__()
1240
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
1241
 
 
1242
 
    def open(self, a_bzrdir, _found=False):
1243
 
        """Return the branch object for a_bzrdir
1244
 
 
1245
 
        _found is a private parameter, do not use it. It is used to indicate
1246
 
               if format probing has already be done.
1247
 
        """
1248
 
        if not _found:
1249
 
            format = BranchFormat.find_format(a_bzrdir)
1250
 
            if format.__class__ != self.__class__:
1251
 
                raise AssertionError("wrong format %r found for %r" %
1252
 
                    (format, self))
1253
 
        try:
1254
 
            transport = a_bzrdir.get_branch_transport(None)
1255
 
            control_files = lockable_files.LockableFiles(transport, 'lock',
1256
 
                                                         lockdir.LockDir)
1257
 
            return BzrBranch5(_format=self,
1258
 
                              _control_files=control_files,
1259
 
                              a_bzrdir=a_bzrdir,
1260
 
                              _repository=a_bzrdir.find_repository())
1261
 
        except errors.NoSuchFile:
1262
 
            raise errors.NotBranchError(path=transport.base)
1263
 
 
1264
 
 
1265
 
class BzrBranchFormat6(BzrBranchFormat5):
 
2113
        return self._initialize_helper(a_bzrdir, utf8_files, name)
 
2114
 
 
2115
    def supports_tags(self):
 
2116
        return False
 
2117
 
 
2118
 
 
2119
class BzrBranchFormat6(BranchFormatMetadir):
1266
2120
    """Branch format with last-revision and tags.
1267
2121
 
1268
2122
    Unlike previous formats, this has no explicit revision history. Instead,
1273
2127
    and became the default in 0.91.
1274
2128
    """
1275
2129
 
 
2130
    def _branch_class(self):
 
2131
        return BzrBranch6
 
2132
 
1276
2133
    def get_format_string(self):
1277
2134
        """See BranchFormat.get_format_string()."""
1278
2135
        return "Bazaar Branch Format 6 (bzr 0.15)\n"
1281
2138
        """See BranchFormat.get_format_description()."""
1282
2139
        return "Branch format 6"
1283
2140
 
1284
 
    def initialize(self, a_bzrdir):
1285
 
        """Create a branch of this format in a_bzrdir."""
1286
 
        utf8_files = [('last-revision', '0 null:\n'),
1287
 
                      ('branch.conf', ''),
1288
 
                      ('tags', ''),
1289
 
                      ]
1290
 
        return self._initialize_helper(a_bzrdir, utf8_files)
1291
 
 
1292
 
    def open(self, a_bzrdir, _found=False):
1293
 
        """Return the branch object for a_bzrdir
1294
 
 
1295
 
        _found is a private parameter, do not use it. It is used to indicate
1296
 
               if format probing has already be done.
1297
 
        """
1298
 
        if not _found:
1299
 
            format = BranchFormat.find_format(a_bzrdir)
1300
 
            if format.__class__ != self.__class__:
1301
 
                raise AssertionError("wrong format %r found for %r" %
1302
 
                    (format, self))
1303
 
        transport = a_bzrdir.get_branch_transport(None)
1304
 
        control_files = lockable_files.LockableFiles(transport, 'lock',
1305
 
                                                     lockdir.LockDir)
1306
 
        return BzrBranch6(_format=self,
1307
 
                          _control_files=control_files,
1308
 
                          a_bzrdir=a_bzrdir,
1309
 
                          _repository=a_bzrdir.find_repository())
1310
 
 
1311
 
    def supports_tags(self):
1312
 
        return True
 
2141
    def initialize(self, a_bzrdir, name=None):
 
2142
        """Create a branch of this format in a_bzrdir."""
 
2143
        utf8_files = [('last-revision', '0 null:\n'),
 
2144
                      ('branch.conf', ''),
 
2145
                      ('tags', ''),
 
2146
                      ]
 
2147
        return self._initialize_helper(a_bzrdir, utf8_files, name)
 
2148
 
 
2149
    def make_tags(self, branch):
 
2150
        """See bzrlib.branch.BranchFormat.make_tags()."""
 
2151
        return BasicTags(branch)
 
2152
 
 
2153
    def supports_set_append_revisions_only(self):
 
2154
        return True
 
2155
 
 
2156
 
 
2157
class BzrBranchFormat8(BranchFormatMetadir):
 
2158
    """Metadir format supporting storing locations of subtree branches."""
 
2159
 
 
2160
    def _branch_class(self):
 
2161
        return BzrBranch8
 
2162
 
 
2163
    def get_format_string(self):
 
2164
        """See BranchFormat.get_format_string()."""
 
2165
        return "Bazaar Branch Format 8 (needs bzr 1.15)\n"
 
2166
 
 
2167
    def get_format_description(self):
 
2168
        """See BranchFormat.get_format_description()."""
 
2169
        return "Branch format 8"
 
2170
 
 
2171
    def initialize(self, a_bzrdir, name=None):
 
2172
        """Create a branch of this format in a_bzrdir."""
 
2173
        utf8_files = [('last-revision', '0 null:\n'),
 
2174
                      ('branch.conf', ''),
 
2175
                      ('tags', ''),
 
2176
                      ('references', '')
 
2177
                      ]
 
2178
        return self._initialize_helper(a_bzrdir, utf8_files, name)
 
2179
 
 
2180
    def __init__(self):
 
2181
        super(BzrBranchFormat8, self).__init__()
 
2182
        self._matchingbzrdir.repository_format = \
 
2183
            RepositoryFormatKnitPack5RichRoot()
 
2184
 
 
2185
    def make_tags(self, branch):
 
2186
        """See bzrlib.branch.BranchFormat.make_tags()."""
 
2187
        return BasicTags(branch)
 
2188
 
 
2189
    def supports_set_append_revisions_only(self):
 
2190
        return True
 
2191
 
 
2192
    def supports_stacking(self):
 
2193
        return True
 
2194
 
 
2195
    supports_reference_locations = True
 
2196
 
 
2197
 
 
2198
class BzrBranchFormat7(BzrBranchFormat8):
 
2199
    """Branch format with last-revision, tags, and a stacked location pointer.
 
2200
 
 
2201
    The stacked location pointer is passed down to the repository and requires
 
2202
    a repository format with supports_external_lookups = True.
 
2203
 
 
2204
    This format was introduced in bzr 1.6.
 
2205
    """
 
2206
 
 
2207
    def initialize(self, a_bzrdir, name=None):
 
2208
        """Create a branch of this format in a_bzrdir."""
 
2209
        utf8_files = [('last-revision', '0 null:\n'),
 
2210
                      ('branch.conf', ''),
 
2211
                      ('tags', ''),
 
2212
                      ]
 
2213
        return self._initialize_helper(a_bzrdir, utf8_files, name)
 
2214
 
 
2215
    def _branch_class(self):
 
2216
        return BzrBranch7
 
2217
 
 
2218
    def get_format_string(self):
 
2219
        """See BranchFormat.get_format_string()."""
 
2220
        return "Bazaar Branch Format 7 (needs bzr 1.6)\n"
 
2221
 
 
2222
    def get_format_description(self):
 
2223
        """See BranchFormat.get_format_description()."""
 
2224
        return "Branch format 7"
 
2225
 
 
2226
    def supports_set_append_revisions_only(self):
 
2227
        return True
 
2228
 
 
2229
    supports_reference_locations = False
1313
2230
 
1314
2231
 
1315
2232
class BranchReferenceFormat(BranchFormat):
1331
2248
        """See BranchFormat.get_format_description()."""
1332
2249
        return "Checkout reference format 1"
1333
2250
 
1334
 
    def get_reference(self, a_bzrdir):
 
2251
    def get_reference(self, a_bzrdir, name=None):
1335
2252
        """See BranchFormat.get_reference()."""
1336
 
        transport = a_bzrdir.get_branch_transport(None)
1337
 
        return transport.get('location').read()
 
2253
        transport = a_bzrdir.get_branch_transport(None, name=name)
 
2254
        return transport.get_bytes('location')
1338
2255
 
1339
 
    def set_reference(self, a_bzrdir, to_branch):
 
2256
    def set_reference(self, a_bzrdir, name, to_branch):
1340
2257
        """See BranchFormat.set_reference()."""
1341
 
        transport = a_bzrdir.get_branch_transport(None)
 
2258
        transport = a_bzrdir.get_branch_transport(None, name=name)
1342
2259
        location = transport.put_bytes('location', to_branch.base)
1343
2260
 
1344
 
    def initialize(self, a_bzrdir, target_branch=None):
 
2261
    def initialize(self, a_bzrdir, name=None, target_branch=None):
1345
2262
        """Create a branch of this format in a_bzrdir."""
1346
2263
        if target_branch is None:
1347
2264
            # this format does not implement branch itself, thus the implicit
1348
2265
            # creation contract must see it as uninitializable
1349
2266
            raise errors.UninitializableFormat(self)
1350
 
        mutter('creating branch reference in %s', a_bzrdir.transport.base)
1351
 
        branch_transport = a_bzrdir.get_branch_transport(self)
 
2267
        mutter('creating branch reference in %s', a_bzrdir.user_url)
 
2268
        branch_transport = a_bzrdir.get_branch_transport(self, name=name)
1352
2269
        branch_transport.put_bytes('location',
1353
 
            target_branch.bzrdir.root_transport.base)
 
2270
            target_branch.bzrdir.user_url)
1354
2271
        branch_transport.put_bytes('format', self.get_format_string())
1355
 
        return self.open(
1356
 
            a_bzrdir, _found=True,
 
2272
        branch = self.open(
 
2273
            a_bzrdir, name, _found=True,
1357
2274
            possible_transports=[target_branch.bzrdir.root_transport])
 
2275
        self._run_post_branch_init_hooks(a_bzrdir, name, branch)
 
2276
        return branch
1358
2277
 
1359
2278
    def __init__(self):
1360
2279
        super(BranchReferenceFormat, self).__init__()
1361
2280
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
 
2281
        self._matchingbzrdir.set_branch_format(self)
1362
2282
 
1363
2283
    def _make_reference_clone_function(format, a_branch):
1364
2284
        """Create a clone() routine for a branch dynamically."""
1365
 
        def clone(to_bzrdir, revision_id=None):
 
2285
        def clone(to_bzrdir, revision_id=None,
 
2286
            repository_policy=None):
1366
2287
            """See Branch.clone()."""
1367
 
            return format.initialize(to_bzrdir, a_branch)
 
2288
            return format.initialize(to_bzrdir, target_branch=a_branch)
1368
2289
            # cannot obey revision_id limits when cloning a reference ...
1369
2290
            # FIXME RBC 20060210 either nuke revision_id for clone, or
1370
2291
            # emit some sort of warning/error to the caller ?!
1371
2292
        return clone
1372
2293
 
1373
 
    def open(self, a_bzrdir, _found=False, location=None,
1374
 
             possible_transports=None):
 
2294
    def open(self, a_bzrdir, name=None, _found=False, location=None,
 
2295
             possible_transports=None, ignore_fallbacks=False):
1375
2296
        """Return the branch that the branch reference in a_bzrdir points at.
1376
2297
 
1377
 
        _found is a private parameter, do not use it. It is used to indicate
1378
 
               if format probing has already be done.
 
2298
        :param a_bzrdir: A BzrDir that contains a branch.
 
2299
        :param name: Name of colocated branch to open, if any
 
2300
        :param _found: a private parameter, do not use it. It is used to
 
2301
            indicate if format probing has already be done.
 
2302
        :param ignore_fallbacks: when set, no fallback branches will be opened
 
2303
            (if there are any).  Default is to open fallbacks.
 
2304
        :param location: The location of the referenced branch.  If
 
2305
            unspecified, this will be determined from the branch reference in
 
2306
            a_bzrdir.
 
2307
        :param possible_transports: An optional reusable transports list.
1379
2308
        """
1380
2309
        if not _found:
1381
 
            format = BranchFormat.find_format(a_bzrdir)
 
2310
            format = BranchFormat.find_format(a_bzrdir, name=name)
1382
2311
            if format.__class__ != self.__class__:
1383
2312
                raise AssertionError("wrong format %r found for %r" %
1384
2313
                    (format, self))
1385
2314
        if location is None:
1386
 
            location = self.get_reference(a_bzrdir)
 
2315
            location = self.get_reference(a_bzrdir, name)
1387
2316
        real_bzrdir = bzrdir.BzrDir.open(
1388
2317
            location, possible_transports=possible_transports)
1389
 
        result = real_bzrdir.open_branch()
 
2318
        result = real_bzrdir.open_branch(name=name, 
 
2319
            ignore_fallbacks=ignore_fallbacks)
1390
2320
        # this changes the behaviour of result.clone to create a new reference
1391
2321
        # rather than a copy of the content of the branch.
1392
2322
        # I did not use a proxy object because that needs much more extensive
1399
2329
        return result
1400
2330
 
1401
2331
 
 
2332
network_format_registry = registry.FormatRegistry()
 
2333
"""Registry of formats indexed by their network name.
 
2334
 
 
2335
The network name for a branch format is an identifier that can be used when
 
2336
referring to formats with smart server operations. See
 
2337
BranchFormat.network_name() for more detail.
 
2338
"""
 
2339
 
 
2340
 
1402
2341
# formats which have no format string are not discoverable
1403
2342
# and not independently creatable, so are not registered.
1404
2343
__format5 = BzrBranchFormat5()
1405
2344
__format6 = BzrBranchFormat6()
 
2345
__format7 = BzrBranchFormat7()
 
2346
__format8 = BzrBranchFormat8()
1406
2347
BranchFormat.register_format(__format5)
1407
2348
BranchFormat.register_format(BranchReferenceFormat())
1408
2349
BranchFormat.register_format(__format6)
1409
 
BranchFormat.set_default_format(__format6)
 
2350
BranchFormat.register_format(__format7)
 
2351
BranchFormat.register_format(__format8)
 
2352
BranchFormat.set_default_format(__format7)
1410
2353
_legacy_formats = [BzrBranchFormat4(),
1411
 
                   ]
1412
 
 
1413
 
class BzrBranch(Branch):
 
2354
    ]
 
2355
network_format_registry.register(
 
2356
    _legacy_formats[0].network_name(), _legacy_formats[0].__class__)
 
2357
 
 
2358
 
 
2359
class BranchWriteLockResult(LogicalLockResult):
 
2360
    """The result of write locking a branch.
 
2361
 
 
2362
    :ivar branch_token: The token obtained from the underlying branch lock, or
 
2363
        None.
 
2364
    :ivar unlock: A callable which will unlock the lock.
 
2365
    """
 
2366
 
 
2367
    def __init__(self, unlock, branch_token):
 
2368
        LogicalLockResult.__init__(self, unlock)
 
2369
        self.branch_token = branch_token
 
2370
 
 
2371
    def __repr__(self):
 
2372
        return "BranchWriteLockResult(%s, %s)" % (self.branch_token,
 
2373
            self.unlock)
 
2374
 
 
2375
 
 
2376
class BzrBranch(Branch, _RelockDebugMixin):
1414
2377
    """A branch stored in the actual filesystem.
1415
2378
 
1416
2379
    Note that it's "local" in the context of the filesystem; it doesn't
1417
2380
    really matter if it's on an nfs/smb/afs/coda/... share, as long as
1418
2381
    it's writable, and can be accessed via the normal filesystem API.
1419
2382
 
1420
 
    :ivar _transport: Transport for file operations on this branch's 
 
2383
    :ivar _transport: Transport for file operations on this branch's
1421
2384
        control files, typically pointing to the .bzr/branch directory.
1422
2385
    :ivar repository: Repository for this branch.
1423
 
    :ivar base: The url of the base directory for this branch; the one 
 
2386
    :ivar base: The url of the base directory for this branch; the one
1424
2387
        containing the .bzr directory.
 
2388
    :ivar name: Optional colocated branch name as it exists in the control
 
2389
        directory.
1425
2390
    """
1426
 
    
 
2391
 
1427
2392
    def __init__(self, _format=None,
1428
 
                 _control_files=None, a_bzrdir=None, _repository=None):
 
2393
                 _control_files=None, a_bzrdir=None, name=None,
 
2394
                 _repository=None, ignore_fallbacks=False):
1429
2395
        """Create new branch object at a particular location."""
1430
 
        Branch.__init__(self)
1431
2396
        if a_bzrdir is None:
1432
2397
            raise ValueError('a_bzrdir must be supplied')
1433
2398
        else:
1434
2399
            self.bzrdir = a_bzrdir
1435
2400
        self._base = self.bzrdir.transport.clone('..').base
 
2401
        self.name = name
1436
2402
        # XXX: We should be able to just do
1437
2403
        #   self.base = self.bzrdir.root_transport.base
1438
2404
        # but this does not quite work yet -- mbp 20080522
1442
2408
        self.control_files = _control_files
1443
2409
        self._transport = _control_files._transport
1444
2410
        self.repository = _repository
 
2411
        Branch.__init__(self)
1445
2412
 
1446
2413
    def __str__(self):
1447
 
        return '%s(%r)' % (self.__class__.__name__, self.base)
 
2414
        if self.name is None:
 
2415
            return '%s(%s)' % (self.__class__.__name__, self.user_url)
 
2416
        else:
 
2417
            return '%s(%s,%s)' % (self.__class__.__name__, self.user_url,
 
2418
                self.name)
1448
2419
 
1449
2420
    __repr__ = __str__
1450
2421
 
1454
2425
 
1455
2426
    base = property(_get_base, doc="The URL for the root of this branch.")
1456
2427
 
1457
 
    @deprecated_method(deprecated_in((0, 16, 0)))
1458
 
    def abspath(self, name):
1459
 
        """See Branch.abspath."""
1460
 
        return self._transport.abspath(name)
 
2428
    def _get_config(self):
 
2429
        return TransportConfig(self._transport, 'branch.conf')
1461
2430
 
1462
2431
    def is_locked(self):
1463
2432
        return self.control_files.is_locked()
1464
2433
 
1465
2434
    def lock_write(self, token=None):
1466
 
        repo_token = self.repository.lock_write()
 
2435
        """Lock the branch for write operations.
 
2436
 
 
2437
        :param token: A token to permit reacquiring a previously held and
 
2438
            preserved lock.
 
2439
        :return: A BranchWriteLockResult.
 
2440
        """
 
2441
        if not self.is_locked():
 
2442
            self._note_lock('w')
 
2443
        # All-in-one needs to always unlock/lock.
 
2444
        repo_control = getattr(self.repository, 'control_files', None)
 
2445
        if self.control_files == repo_control or not self.is_locked():
 
2446
            self.repository._warn_if_deprecated(self)
 
2447
            self.repository.lock_write()
 
2448
            took_lock = True
 
2449
        else:
 
2450
            took_lock = False
1467
2451
        try:
1468
 
            token = self.control_files.lock_write(token=token)
 
2452
            return BranchWriteLockResult(self.unlock,
 
2453
                self.control_files.lock_write(token=token))
1469
2454
        except:
1470
 
            self.repository.unlock()
 
2455
            if took_lock:
 
2456
                self.repository.unlock()
1471
2457
            raise
1472
 
        return token
1473
2458
 
1474
2459
    def lock_read(self):
1475
 
        self.repository.lock_read()
 
2460
        """Lock the branch for read operations.
 
2461
 
 
2462
        :return: A bzrlib.lock.LogicalLockResult.
 
2463
        """
 
2464
        if not self.is_locked():
 
2465
            self._note_lock('r')
 
2466
        # All-in-one needs to always unlock/lock.
 
2467
        repo_control = getattr(self.repository, 'control_files', None)
 
2468
        if self.control_files == repo_control or not self.is_locked():
 
2469
            self.repository._warn_if_deprecated(self)
 
2470
            self.repository.lock_read()
 
2471
            took_lock = True
 
2472
        else:
 
2473
            took_lock = False
1476
2474
        try:
1477
2475
            self.control_files.lock_read()
 
2476
            return LogicalLockResult(self.unlock)
1478
2477
        except:
1479
 
            self.repository.unlock()
 
2478
            if took_lock:
 
2479
                self.repository.unlock()
1480
2480
            raise
1481
2481
 
 
2482
    @only_raises(errors.LockNotHeld, errors.LockBroken)
1482
2483
    def unlock(self):
1483
 
        # TODO: test for failed two phase locks. This is known broken.
1484
2484
        try:
1485
2485
            self.control_files.unlock()
1486
2486
        finally:
1487
 
            self.repository.unlock()
1488
 
        if not self.control_files.is_locked():
1489
 
            # we just released the lock
1490
 
            self._clear_cached_state()
1491
 
        
 
2487
            # All-in-one needs to always unlock/lock.
 
2488
            repo_control = getattr(self.repository, 'control_files', None)
 
2489
            if (self.control_files == repo_control or
 
2490
                not self.control_files.is_locked()):
 
2491
                self.repository.unlock()
 
2492
            if not self.control_files.is_locked():
 
2493
                # we just released the lock
 
2494
                self._clear_cached_state()
 
2495
 
1492
2496
    def peek_lock_mode(self):
1493
2497
        if self.control_files._lock_count == 0:
1494
2498
            return None
1520
2524
        check_not_reserved_id = _mod_revision.check_not_reserved_id
1521
2525
        for rev_id in rev_history:
1522
2526
            check_not_reserved_id(rev_id)
 
2527
        if Branch.hooks['post_change_branch_tip']:
 
2528
            # Don't calculate the last_revision_info() if there are no hooks
 
2529
            # that will use it.
 
2530
            old_revno, old_revid = self.last_revision_info()
 
2531
        if len(rev_history) == 0:
 
2532
            revid = _mod_revision.NULL_REVISION
 
2533
        else:
 
2534
            revid = rev_history[-1]
 
2535
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
1523
2536
        self._write_revision_history(rev_history)
1524
2537
        self._clear_cached_state()
1525
2538
        self._cache_revision_history(rev_history)
1526
2539
        for hook in Branch.hooks['set_rh']:
1527
2540
            hook(self, rev_history)
1528
 
 
1529
 
    def _run_pre_change_branch_tip_hooks(self, new_revno, new_revid):
1530
 
        """Run the pre_change_branch_tip hooks."""
1531
 
        hooks = Branch.hooks['pre_change_branch_tip']
1532
 
        if not hooks:
1533
 
            return
1534
 
        old_revno, old_revid = self.last_revision_info()
1535
 
        params = ChangeBranchTipParams(
1536
 
            self, old_revno, new_revno, old_revid, new_revid)
1537
 
        for hook in hooks:
1538
 
            hook(params)
1539
 
 
1540
 
    def _run_post_change_branch_tip_hooks(self, old_revno, old_revid):
1541
 
        """Run the post_change_branch_tip hooks."""
1542
 
        hooks = Branch.hooks['post_change_branch_tip']
1543
 
        if not hooks:
1544
 
            return
1545
 
        new_revno, new_revid = self.last_revision_info()
1546
 
        params = ChangeBranchTipParams(
1547
 
            self, old_revno, new_revno, old_revid, new_revid)
1548
 
        for hook in hooks:
1549
 
            hook(params)
1550
 
 
 
2541
        if Branch.hooks['post_change_branch_tip']:
 
2542
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
 
2543
 
 
2544
    def _synchronize_history(self, destination, revision_id):
 
2545
        """Synchronize last revision and revision history between branches.
 
2546
 
 
2547
        This version is most efficient when the destination is also a
 
2548
        BzrBranch5, but works for BzrBranch6 as long as the revision
 
2549
        history is the true lefthand parent history, and all of the revisions
 
2550
        are in the destination's repository.  If not, set_revision_history
 
2551
        will fail.
 
2552
 
 
2553
        :param destination: The branch to copy the history into
 
2554
        :param revision_id: The revision-id to truncate history at.  May
 
2555
          be None to copy complete history.
 
2556
        """
 
2557
        if not isinstance(destination._format, BzrBranchFormat5):
 
2558
            super(BzrBranch, self)._synchronize_history(
 
2559
                destination, revision_id)
 
2560
            return
 
2561
        if revision_id == _mod_revision.NULL_REVISION:
 
2562
            new_history = []
 
2563
        else:
 
2564
            new_history = self.revision_history()
 
2565
        if revision_id is not None and new_history != []:
 
2566
            try:
 
2567
                new_history = new_history[:new_history.index(revision_id) + 1]
 
2568
            except ValueError:
 
2569
                rev = self.repository.get_revision(revision_id)
 
2570
                new_history = rev.get_history(self.repository)[1:]
 
2571
        destination.set_revision_history(new_history)
 
2572
 
1551
2573
    @needs_write_lock
1552
2574
    def set_last_revision_info(self, revno, revision_id):
1553
2575
        """Set the last revision of this branch.
1556
2578
        for this revision id.
1557
2579
 
1558
2580
        It may be possible to set the branch last revision to an id not
1559
 
        present in the repository.  However, branches can also be 
 
2581
        present in the repository.  However, branches can also be
1560
2582
        configured to check constraints on history, in which case this may not
1561
2583
        be permitted.
1562
2584
        """
1563
2585
        revision_id = _mod_revision.ensure_null(revision_id)
1564
 
        old_revno, old_revid = self.last_revision_info()
1565
2586
        # this old format stores the full history, but this api doesn't
1566
2587
        # provide it, so we must generate, and might as well check it's
1567
2588
        # correct
1568
2589
        history = self._lefthand_history(revision_id)
1569
2590
        if len(history) != revno:
1570
2591
            raise AssertionError('%d != %d' % (len(history), revno))
1571
 
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
1572
2592
        self.set_revision_history(history)
1573
 
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
1574
2593
 
1575
2594
    def _gen_revision_history(self):
1576
2595
        history = self._transport.get_bytes('revision-history').split('\n')
1579
2598
            history.pop()
1580
2599
        return history
1581
2600
 
1582
 
    def _lefthand_history(self, revision_id, last_rev=None,
1583
 
                          other_branch=None):
1584
 
        if 'evil' in debug.debug_flags:
1585
 
            mutter_callsite(4, "_lefthand_history scales with history.")
1586
 
        # stop_revision must be a descendant of last_revision
1587
 
        graph = self.repository.get_graph()
1588
 
        if last_rev is not None:
1589
 
            if not graph.is_ancestor(last_rev, revision_id):
1590
 
                # our previous tip is not merged into stop_revision
1591
 
                raise errors.DivergedBranches(self, other_branch)
1592
 
        # make a new revision history from the graph
1593
 
        parents_map = graph.get_parent_map([revision_id])
1594
 
        if revision_id not in parents_map:
1595
 
            raise errors.NoSuchRevision(self, revision_id)
1596
 
        current_rev_id = revision_id
1597
 
        new_history = []
1598
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
1599
 
        # Do not include ghosts or graph origin in revision_history
1600
 
        while (current_rev_id in parents_map and
1601
 
               len(parents_map[current_rev_id]) > 0):
1602
 
            check_not_reserved_id(current_rev_id)
1603
 
            new_history.append(current_rev_id)
1604
 
            current_rev_id = parents_map[current_rev_id][0]
1605
 
            parents_map = graph.get_parent_map([current_rev_id])
1606
 
        new_history.reverse()
1607
 
        return new_history
1608
 
 
1609
2601
    @needs_write_lock
1610
2602
    def generate_revision_history(self, revision_id, last_rev=None,
1611
2603
        other_branch=None):
1624
2616
        """See Branch.basis_tree."""
1625
2617
        return self.repository.revision_tree(self.last_revision())
1626
2618
 
1627
 
    @needs_write_lock
1628
 
    def pull(self, source, overwrite=False, stop_revision=None,
1629
 
             _hook_master=None, run_hooks=True, possible_transports=None,
1630
 
             _override_hook_target=None):
1631
 
        """See Branch.pull.
1632
 
 
1633
 
        :param _hook_master: Private parameter - set the branch to 
1634
 
            be supplied as the master to pull hooks.
1635
 
        :param run_hooks: Private parameter - if false, this branch
1636
 
            is being called because it's the master of the primary branch,
1637
 
            so it should not run its hooks.
1638
 
        :param _override_hook_target: Private parameter - set the branch to be
1639
 
            supplied as the target_branch to pull hooks.
1640
 
        """
1641
 
        result = PullResult()
1642
 
        result.source_branch = source
1643
 
        if _override_hook_target is None:
1644
 
            result.target_branch = self
1645
 
        else:
1646
 
            result.target_branch = _override_hook_target
1647
 
        source.lock_read()
1648
 
        try:
1649
 
            # We assume that during 'pull' the local repository is closer than
1650
 
            # the remote one.
1651
 
            graph = self.repository.get_graph(source.repository)
1652
 
            result.old_revno, result.old_revid = self.last_revision_info()
1653
 
            self.update_revisions(source, stop_revision, overwrite=overwrite,
1654
 
                                  graph=graph)
1655
 
            result.tag_conflicts = source.tags.merge_to(self.tags, overwrite)
1656
 
            result.new_revno, result.new_revid = self.last_revision_info()
1657
 
            if _hook_master:
1658
 
                result.master_branch = _hook_master
1659
 
                result.local_branch = result.target_branch
1660
 
            else:
1661
 
                result.master_branch = result.target_branch
1662
 
                result.local_branch = None
1663
 
            if run_hooks:
1664
 
                for hook in Branch.hooks['post_pull']:
1665
 
                    hook(result)
1666
 
        finally:
1667
 
            source.unlock()
1668
 
        return result
1669
 
 
1670
2619
    def _get_parent_location(self):
1671
2620
        _locs = ['parent', 'pull', 'x-pull']
1672
2621
        for l in _locs:
1676
2625
                pass
1677
2626
        return None
1678
2627
 
1679
 
    @needs_read_lock
1680
 
    def push(self, target, overwrite=False, stop_revision=None,
1681
 
             _override_hook_source_branch=None):
1682
 
        """See Branch.push.
1683
 
 
1684
 
        This is the basic concrete implementation of push()
1685
 
 
1686
 
        :param _override_hook_source_branch: If specified, run
1687
 
        the hooks passing this Branch as the source, rather than self.  
1688
 
        This is for use of RemoteBranch, where push is delegated to the
1689
 
        underlying vfs-based Branch. 
1690
 
        """
1691
 
        # TODO: Public option to disable running hooks - should be trivial but
1692
 
        # needs tests.
1693
 
        target.lock_write()
1694
 
        try:
1695
 
            result = self._push_with_bound_branches(target, overwrite,
1696
 
                    stop_revision,
1697
 
                    _override_hook_source_branch=_override_hook_source_branch)
1698
 
            return result
1699
 
        finally:
1700
 
            target.unlock()
1701
 
 
1702
 
    def _push_with_bound_branches(self, target, overwrite,
1703
 
            stop_revision,
1704
 
            _override_hook_source_branch=None):
1705
 
        """Push from self into target, and into target's master if any.
1706
 
        
1707
 
        This is on the base BzrBranch class even though it doesn't support 
1708
 
        bound branches because the *target* might be bound.
1709
 
        """
1710
 
        def _run_hooks():
1711
 
            if _override_hook_source_branch:
1712
 
                result.source_branch = _override_hook_source_branch
1713
 
            for hook in Branch.hooks['post_push']:
1714
 
                hook(result)
1715
 
 
1716
 
        bound_location = target.get_bound_location()
1717
 
        if bound_location and target.base != bound_location:
1718
 
            # there is a master branch.
1719
 
            #
1720
 
            # XXX: Why the second check?  Is it even supported for a branch to
1721
 
            # be bound to itself? -- mbp 20070507
1722
 
            master_branch = target.get_master_branch()
1723
 
            master_branch.lock_write()
1724
 
            try:
1725
 
                # push into the master from this branch.
1726
 
                self._basic_push(master_branch, overwrite, stop_revision)
1727
 
                # and push into the target branch from this. Note that we push from
1728
 
                # this branch again, because its considered the highest bandwidth
1729
 
                # repository.
1730
 
                result = self._basic_push(target, overwrite, stop_revision)
1731
 
                result.master_branch = master_branch
1732
 
                result.local_branch = target
1733
 
                _run_hooks()
1734
 
                return result
1735
 
            finally:
1736
 
                master_branch.unlock()
1737
 
        else:
1738
 
            # no master branch
1739
 
            result = self._basic_push(target, overwrite, stop_revision)
1740
 
            # TODO: Why set master_branch and local_branch if there's no
1741
 
            # binding?  Maybe cleaner to just leave them unset? -- mbp
1742
 
            # 20070504
1743
 
            result.master_branch = target
1744
 
            result.local_branch = None
1745
 
            _run_hooks()
1746
 
            return result
1747
 
 
1748
2628
    def _basic_push(self, target, overwrite, stop_revision):
1749
2629
        """Basic implementation of push without bound branches or hooks.
1750
2630
 
1751
 
        Must be called with self read locked and target write locked.
 
2631
        Must be called with source read locked and target write locked.
1752
2632
        """
1753
 
        result = PushResult()
 
2633
        result = BranchPushResult()
1754
2634
        result.source_branch = self
1755
2635
        result.target_branch = target
1756
2636
        result.old_revno, result.old_revid = target.last_revision_info()
1757
 
 
1758
 
        # We assume that during 'push' this repository is closer than
1759
 
        # the target.
1760
 
        graph = self.repository.get_graph(target.repository)
1761
 
        target.update_revisions(self, stop_revision, overwrite=overwrite,
1762
 
                                graph=graph)
1763
 
        result.tag_conflicts = self.tags.merge_to(target.tags, overwrite)
 
2637
        self.update_references(target)
 
2638
        if result.old_revid != self.last_revision():
 
2639
            # We assume that during 'push' this repository is closer than
 
2640
            # the target.
 
2641
            graph = self.repository.get_graph(target.repository)
 
2642
            target.update_revisions(self, stop_revision,
 
2643
                overwrite=overwrite, graph=graph)
 
2644
        if self._push_should_merge_tags():
 
2645
            result.tag_conflicts = self.tags.merge_to(target.tags,
 
2646
                overwrite)
1764
2647
        result.new_revno, result.new_revid = target.last_revision_info()
1765
2648
        return result
1766
2649
 
1767
 
    def get_parent(self):
1768
 
        """See Branch.get_parent."""
1769
 
        parent = self._get_parent_location()
1770
 
        if parent is None:
1771
 
            return parent
1772
 
        # This is an old-format absolute path to a local branch
1773
 
        # turn it into a url
1774
 
        if parent.startswith('/'):
1775
 
            parent = urlutils.local_path_to_url(parent.decode('utf8'))
1776
 
        try:
1777
 
            return urlutils.join(self.base[:-1], parent)
1778
 
        except errors.InvalidURLJoin, e:
1779
 
            raise errors.InaccessibleParent(parent, self.base)
 
2650
    def get_stacked_on_url(self):
 
2651
        raise errors.UnstackableBranchFormat(self._format, self.user_url)
1780
2652
 
1781
2653
    def set_push_location(self, location):
1782
2654
        """See Branch.set_push_location."""
1784
2656
            'push_location', location,
1785
2657
            store=_mod_config.STORE_LOCATION_NORECURSE)
1786
2658
 
1787
 
    @needs_write_lock
1788
 
    def set_parent(self, url):
1789
 
        """See Branch.set_parent."""
1790
 
        # TODO: Maybe delete old location files?
1791
 
        # URLs should never be unicode, even on the local fs,
1792
 
        # FIXUP this and get_parent in a future branch format bump:
1793
 
        # read and rewrite the file. RBC 20060125
1794
 
        if url is not None:
1795
 
            if isinstance(url, unicode):
1796
 
                try:
1797
 
                    url = url.encode('ascii')
1798
 
                except UnicodeEncodeError:
1799
 
                    raise errors.InvalidURL(url,
1800
 
                        "Urls must be 7-bit ascii, "
1801
 
                        "use bzrlib.urlutils.escape")
1802
 
            url = urlutils.relative_url(self.base, url)
1803
 
        self._set_parent_location(url)
1804
 
 
1805
2659
    def _set_parent_location(self, url):
1806
2660
        if url is None:
1807
2661
            self._transport.delete('parent')
1816
2670
    It has support for a master_branch which is the data for bound branches.
1817
2671
    """
1818
2672
 
1819
 
    def __init__(self,
1820
 
                 _format,
1821
 
                 _control_files,
1822
 
                 a_bzrdir,
1823
 
                 _repository):
1824
 
        super(BzrBranch5, self).__init__(_format=_format,
1825
 
                                         _control_files=_control_files,
1826
 
                                         a_bzrdir=a_bzrdir,
1827
 
                                         _repository=_repository)
1828
 
        
1829
 
    @needs_write_lock
1830
 
    def pull(self, source, overwrite=False, stop_revision=None,
1831
 
             run_hooks=True, possible_transports=None,
1832
 
             _override_hook_target=None):
1833
 
        """Pull from source into self, updating my master if any.
1834
 
        
1835
 
        :param run_hooks: Private parameter - if false, this branch
1836
 
            is being called because it's the master of the primary branch,
1837
 
            so it should not run its hooks.
1838
 
        """
1839
 
        bound_location = self.get_bound_location()
1840
 
        master_branch = None
1841
 
        if bound_location and source.base != bound_location:
1842
 
            # not pulling from master, so we need to update master.
1843
 
            master_branch = self.get_master_branch(possible_transports)
1844
 
            master_branch.lock_write()
1845
 
        try:
1846
 
            if master_branch:
1847
 
                # pull from source into master.
1848
 
                master_branch.pull(source, overwrite, stop_revision,
1849
 
                    run_hooks=False)
1850
 
            return super(BzrBranch5, self).pull(source, overwrite,
1851
 
                stop_revision, _hook_master=master_branch,
1852
 
                run_hooks=run_hooks,
1853
 
                _override_hook_target=_override_hook_target)
1854
 
        finally:
1855
 
            if master_branch:
1856
 
                master_branch.unlock()
1857
 
 
1858
2673
    def get_bound_location(self):
1859
2674
        try:
1860
2675
            return self._transport.get_bytes('bound')[:-1]
1864
2679
    @needs_read_lock
1865
2680
    def get_master_branch(self, possible_transports=None):
1866
2681
        """Return the branch we are bound to.
1867
 
        
 
2682
 
1868
2683
        :return: Either a Branch, or None
1869
2684
 
1870
2685
        This could memoise the branch, but if thats done
1906
2721
        check for divergence to raise an error when the branches are not
1907
2722
        either the same, or one a prefix of the other. That behaviour may not
1908
2723
        be useful, so that check may be removed in future.
1909
 
        
 
2724
 
1910
2725
        :param other: The branch to bind to
1911
2726
        :type other: Branch
1912
2727
        """
1931
2746
 
1932
2747
    @needs_write_lock
1933
2748
    def update(self, possible_transports=None):
1934
 
        """Synchronise this branch with the master branch if any. 
 
2749
        """Synchronise this branch with the master branch if any.
1935
2750
 
1936
2751
        :return: None or the last_revision that was pivoted out during the
1937
2752
                 update.
1947
2762
        return None
1948
2763
 
1949
2764
 
1950
 
class BzrBranch6(BzrBranch5):
 
2765
class BzrBranch8(BzrBranch5):
 
2766
    """A branch that stores tree-reference locations."""
 
2767
 
 
2768
    def _open_hook(self):
 
2769
        if self._ignore_fallbacks:
 
2770
            return
 
2771
        try:
 
2772
            url = self.get_stacked_on_url()
 
2773
        except (errors.UnstackableRepositoryFormat, errors.NotStacked,
 
2774
            errors.UnstackableBranchFormat):
 
2775
            pass
 
2776
        else:
 
2777
            for hook in Branch.hooks['transform_fallback_location']:
 
2778
                url = hook(self, url)
 
2779
                if url is None:
 
2780
                    hook_name = Branch.hooks.get_hook_name(hook)
 
2781
                    raise AssertionError(
 
2782
                        "'transform_fallback_location' hook %s returned "
 
2783
                        "None, not a URL." % hook_name)
 
2784
            self._activate_fallback_location(url)
1951
2785
 
1952
2786
    def __init__(self, *args, **kwargs):
1953
 
        super(BzrBranch6, self).__init__(*args, **kwargs)
1954
 
        self._partial_revision_history_cache = []
 
2787
        self._ignore_fallbacks = kwargs.get('ignore_fallbacks', False)
 
2788
        super(BzrBranch8, self).__init__(*args, **kwargs)
 
2789
        self._last_revision_info_cache = None
 
2790
        self._reference_info = None
1955
2791
 
1956
2792
    def _clear_cached_state(self):
1957
 
        super(BzrBranch6, self)._clear_cached_state()
1958
 
        self._partial_revision_history_cache = []
 
2793
        super(BzrBranch8, self)._clear_cached_state()
 
2794
        self._last_revision_info_cache = None
 
2795
        self._reference_info = None
1959
2796
 
1960
2797
    def _last_revision_info(self):
1961
2798
        revision_string = self._transport.get_bytes('last-revision')
1990
2827
        self._last_revision_info_cache = revno, revision_id
1991
2828
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
1992
2829
 
 
2830
    def _synchronize_history(self, destination, revision_id):
 
2831
        """Synchronize last revision and revision history between branches.
 
2832
 
 
2833
        :see: Branch._synchronize_history
 
2834
        """
 
2835
        # XXX: The base Branch has a fast implementation of this method based
 
2836
        # on set_last_revision_info, but BzrBranch/BzrBranch5 have a slower one
 
2837
        # that uses set_revision_history.  This class inherits from BzrBranch5,
 
2838
        # but wants the fast implementation, so it calls
 
2839
        # Branch._synchronize_history directly.
 
2840
        Branch._synchronize_history(self, destination, revision_id)
 
2841
 
1993
2842
    def _check_history_violation(self, revision_id):
1994
2843
        last_revision = _mod_revision.ensure_null(self.last_revision())
1995
2844
        if _mod_revision.is_null(last_revision):
1996
2845
            return
1997
2846
        if last_revision not in self._lefthand_history(revision_id):
1998
 
            raise errors.AppendRevisionsOnlyViolation(self.base)
 
2847
            raise errors.AppendRevisionsOnlyViolation(self.user_url)
1999
2848
 
2000
2849
    def _gen_revision_history(self):
2001
2850
        """Generate the revision history from last revision
2004
2853
        self._extend_partial_history(stop_index=last_revno-1)
2005
2854
        return list(reversed(self._partial_revision_history_cache))
2006
2855
 
2007
 
    def _extend_partial_history(self, stop_index=None, stop_revision=None):
2008
 
        """Extend the partial history to include a given index
2009
 
 
2010
 
        If a stop_index is supplied, stop when that index has been reached.
2011
 
        If a stop_revision is supplied, stop when that revision is
2012
 
        encountered.  Otherwise, stop when the beginning of history is
2013
 
        reached.
2014
 
 
2015
 
        :param stop_index: The index which should be present.  When it is
2016
 
            present, history extension will stop.
2017
 
        :param revision_id: The revision id which should be present.  When
2018
 
            it is encountered, history extension will stop.
2019
 
        """
2020
 
        repo = self.repository
2021
 
        if len(self._partial_revision_history_cache) == 0:
2022
 
            iterator = repo.iter_reverse_revision_history(self.last_revision())
2023
 
        else:
2024
 
            start_revision = self._partial_revision_history_cache[-1]
2025
 
            iterator = repo.iter_reverse_revision_history(start_revision)
2026
 
            #skip the last revision in the list
2027
 
            next_revision = iterator.next()
2028
 
        for revision_id in iterator:
2029
 
            self._partial_revision_history_cache.append(revision_id)
2030
 
            if (stop_index is not None and
2031
 
                len(self._partial_revision_history_cache) > stop_index):
2032
 
                break
2033
 
            if revision_id == stop_revision:
2034
 
                break
2035
 
 
2036
2856
    def _write_revision_history(self, history):
2037
2857
        """Factored out of set_revision_history.
2038
2858
 
2059
2879
        """Set the parent branch"""
2060
2880
        return self._get_config_location('parent_location')
2061
2881
 
 
2882
    @needs_write_lock
 
2883
    def _set_all_reference_info(self, info_dict):
 
2884
        """Replace all reference info stored in a branch.
 
2885
 
 
2886
        :param info_dict: A dict of {file_id: (tree_path, branch_location)}
 
2887
        """
 
2888
        s = StringIO()
 
2889
        writer = rio.RioWriter(s)
 
2890
        for key, (tree_path, branch_location) in info_dict.iteritems():
 
2891
            stanza = rio.Stanza(file_id=key, tree_path=tree_path,
 
2892
                                branch_location=branch_location)
 
2893
            writer.write_stanza(stanza)
 
2894
        self._transport.put_bytes('references', s.getvalue())
 
2895
        self._reference_info = info_dict
 
2896
 
 
2897
    @needs_read_lock
 
2898
    def _get_all_reference_info(self):
 
2899
        """Return all the reference info stored in a branch.
 
2900
 
 
2901
        :return: A dict of {file_id: (tree_path, branch_location)}
 
2902
        """
 
2903
        if self._reference_info is not None:
 
2904
            return self._reference_info
 
2905
        rio_file = self._transport.get('references')
 
2906
        try:
 
2907
            stanzas = rio.read_stanzas(rio_file)
 
2908
            info_dict = dict((s['file_id'], (s['tree_path'],
 
2909
                             s['branch_location'])) for s in stanzas)
 
2910
        finally:
 
2911
            rio_file.close()
 
2912
        self._reference_info = info_dict
 
2913
        return info_dict
 
2914
 
 
2915
    def set_reference_info(self, file_id, tree_path, branch_location):
 
2916
        """Set the branch location to use for a tree reference.
 
2917
 
 
2918
        :param file_id: The file-id of the tree reference.
 
2919
        :param tree_path: The path of the tree reference in the tree.
 
2920
        :param branch_location: The location of the branch to retrieve tree
 
2921
            references from.
 
2922
        """
 
2923
        info_dict = self._get_all_reference_info()
 
2924
        info_dict[file_id] = (tree_path, branch_location)
 
2925
        if None in (tree_path, branch_location):
 
2926
            if tree_path is not None:
 
2927
                raise ValueError('tree_path must be None when branch_location'
 
2928
                                 ' is None.')
 
2929
            if branch_location is not None:
 
2930
                raise ValueError('branch_location must be None when tree_path'
 
2931
                                 ' is None.')
 
2932
            del info_dict[file_id]
 
2933
        self._set_all_reference_info(info_dict)
 
2934
 
 
2935
    def get_reference_info(self, file_id):
 
2936
        """Get the tree_path and branch_location for a tree reference.
 
2937
 
 
2938
        :return: a tuple of (tree_path, branch_location)
 
2939
        """
 
2940
        return self._get_all_reference_info().get(file_id, (None, None))
 
2941
 
 
2942
    def reference_parent(self, file_id, path, possible_transports=None):
 
2943
        """Return the parent branch for a tree-reference file_id.
 
2944
 
 
2945
        :param file_id: The file_id of the tree reference
 
2946
        :param path: The path of the file_id in the tree
 
2947
        :return: A branch associated with the file_id
 
2948
        """
 
2949
        branch_location = self.get_reference_info(file_id)[1]
 
2950
        if branch_location is None:
 
2951
            return Branch.reference_parent(self, file_id, path,
 
2952
                                           possible_transports)
 
2953
        branch_location = urlutils.join(self.user_url, branch_location)
 
2954
        return Branch.open(branch_location,
 
2955
                           possible_transports=possible_transports)
 
2956
 
2062
2957
    def set_push_location(self, location):
2063
2958
        """See Branch.set_push_location."""
2064
2959
        self._set_config_location('push_location', location)
2097
2992
        """See Branch.get_old_bound_location"""
2098
2993
        return self._get_bound_location(False)
2099
2994
 
2100
 
    def set_append_revisions_only(self, enabled):
2101
 
        if enabled:
2102
 
            value = 'True'
2103
 
        else:
2104
 
            value = 'False'
2105
 
        self.get_config().set_user_option('append_revisions_only', value,
2106
 
            warn_masked=True)
 
2995
    def get_stacked_on_url(self):
 
2996
        # you can always ask for the URL; but you might not be able to use it
 
2997
        # if the repo can't support stacking.
 
2998
        ## self._check_stackable_repo()
 
2999
        stacked_url = self._get_config_location('stacked_on_location')
 
3000
        if stacked_url is None:
 
3001
            raise errors.NotStacked(self)
 
3002
        return stacked_url
2107
3003
 
2108
3004
    def _get_append_revisions_only(self):
2109
 
        value = self.get_config().get_user_option('append_revisions_only')
2110
 
        return value == 'True'
2111
 
 
2112
 
    def _synchronize_history(self, destination, revision_id):
2113
 
        """Synchronize last revision and revision history between branches.
2114
 
 
2115
 
        This version is most efficient when the destination is also a
2116
 
        BzrBranch6, but works for BzrBranch5, as long as the destination's
2117
 
        repository contains all the lefthand ancestors of the intended
2118
 
        last_revision.  If not, set_last_revision_info will fail.
2119
 
 
2120
 
        :param destination: The branch to copy the history into
2121
 
        :param revision_id: The revision-id to truncate history at.  May
2122
 
          be None to copy complete history.
2123
 
        """
2124
 
        source_revno, source_revision_id = self.last_revision_info()
2125
 
        if revision_id is None:
2126
 
            revno, revision_id = source_revno, source_revision_id
2127
 
        elif source_revision_id == revision_id:
2128
 
            # we know the revno without needing to walk all of history
2129
 
            revno = source_revno
2130
 
        else:
2131
 
            # To figure out the revno for a random revision, we need to build
2132
 
            # the revision history, and count its length.
2133
 
            # We don't care about the order, just how long it is.
2134
 
            # Alternatively, we could start at the current location, and count
2135
 
            # backwards. But there is no guarantee that we will find it since
2136
 
            # it may be a merged revision.
2137
 
            revno = len(list(self.repository.iter_reverse_revision_history(
2138
 
                                                                revision_id)))
2139
 
        destination.set_last_revision_info(revno, revision_id)
2140
 
 
2141
 
    def _make_tags(self):
2142
 
        return BasicTags(self)
 
3005
        return self.get_config(
 
3006
            ).get_user_option_as_bool('append_revisions_only')
2143
3007
 
2144
3008
    @needs_write_lock
2145
3009
    def generate_revision_history(self, revision_id, last_rev=None,
2185
3049
        return self.revno() - index
2186
3050
 
2187
3051
 
 
3052
class BzrBranch7(BzrBranch8):
 
3053
    """A branch with support for a fallback repository."""
 
3054
 
 
3055
    def set_reference_info(self, file_id, tree_path, branch_location):
 
3056
        Branch.set_reference_info(self, file_id, tree_path, branch_location)
 
3057
 
 
3058
    def get_reference_info(self, file_id):
 
3059
        Branch.get_reference_info(self, file_id)
 
3060
 
 
3061
    def reference_parent(self, file_id, path, possible_transports=None):
 
3062
        return Branch.reference_parent(self, file_id, path,
 
3063
                                       possible_transports)
 
3064
 
 
3065
 
 
3066
class BzrBranch6(BzrBranch7):
 
3067
    """See BzrBranchFormat6 for the capabilities of this branch.
 
3068
 
 
3069
    This subclass of BzrBranch7 disables the new features BzrBranch7 added,
 
3070
    i.e. stacking.
 
3071
    """
 
3072
 
 
3073
    def get_stacked_on_url(self):
 
3074
        raise errors.UnstackableBranchFormat(self._format, self.user_url)
 
3075
 
 
3076
 
2188
3077
######################################################################
2189
3078
# results of operations
2190
3079
 
2206
3095
    :ivar new_revno: Revision number after pull.
2207
3096
    :ivar old_revid: Tip revision id before pull.
2208
3097
    :ivar new_revid: Tip revision id after pull.
2209
 
    :ivar source_branch: Source (local) branch object.
 
3098
    :ivar source_branch: Source (local) branch object. (read locked)
2210
3099
    :ivar master_branch: Master branch of the target, or the target if no
2211
3100
        Master
2212
3101
    :ivar local_branch: target branch if there is a Master, else None
2213
 
    :ivar target_branch: Target/destination branch object.
 
3102
    :ivar target_branch: Target/destination branch object. (write locked)
2214
3103
    :ivar tag_conflicts: A list of tag conflicts, see BasicTags.merge_to
2215
3104
    """
2216
3105
 
 
3106
    @deprecated_method(deprecated_in((2, 3, 0)))
2217
3107
    def __int__(self):
2218
 
        # DEPRECATED: pull used to return the change in revno
 
3108
        """Return the relative change in revno.
 
3109
 
 
3110
        :deprecated: Use `new_revno` and `old_revno` instead.
 
3111
        """
2219
3112
        return self.new_revno - self.old_revno
2220
3113
 
2221
3114
    def report(self, to_file):
2227
3120
        self._show_tag_conficts(to_file)
2228
3121
 
2229
3122
 
2230
 
class PushResult(_Result):
 
3123
class BranchPushResult(_Result):
2231
3124
    """Result of a Branch.push operation.
2232
3125
 
2233
 
    :ivar old_revno: Revision number before push.
2234
 
    :ivar new_revno: Revision number after push.
2235
 
    :ivar old_revid: Tip revision id before push.
2236
 
    :ivar new_revid: Tip revision id after push.
2237
 
    :ivar source_branch: Source branch object.
2238
 
    :ivar master_branch: Master branch of the target, or None.
2239
 
    :ivar target_branch: Target/destination branch object.
 
3126
    :ivar old_revno: Revision number (eg 10) of the target before push.
 
3127
    :ivar new_revno: Revision number (eg 12) of the target after push.
 
3128
    :ivar old_revid: Tip revision id (eg joe@foo.com-1234234-aoeua34) of target
 
3129
        before the push.
 
3130
    :ivar new_revid: Tip revision id (eg joe@foo.com-5676566-boa234a) of target
 
3131
        after the push.
 
3132
    :ivar source_branch: Source branch object that the push was from. This is
 
3133
        read locked, and generally is a local (and thus low latency) branch.
 
3134
    :ivar master_branch: If target is a bound branch, the master branch of
 
3135
        target, or target itself. Always write locked.
 
3136
    :ivar target_branch: The direct Branch where data is being sent (write
 
3137
        locked).
 
3138
    :ivar local_branch: If the target is a bound branch this will be the
 
3139
        target, otherwise it will be None.
2240
3140
    """
2241
3141
 
 
3142
    @deprecated_method(deprecated_in((2, 3, 0)))
2242
3143
    def __int__(self):
2243
 
        # DEPRECATED: push used to return the change in revno
 
3144
        """Return the relative change in revno.
 
3145
 
 
3146
        :deprecated: Use `new_revno` and `old_revno` instead.
 
3147
        """
2244
3148
        return self.new_revno - self.old_revno
2245
3149
 
2246
3150
    def report(self, to_file):
2247
3151
        """Write a human-readable description of the result."""
2248
3152
        if self.old_revid == self.new_revid:
2249
 
            to_file.write('No new revisions to push.\n')
 
3153
            note('No new revisions to push.')
2250
3154
        else:
2251
 
            to_file.write('Pushed up to revision %d.\n' % self.new_revno)
 
3155
            note('Pushed up to revision %d.' % self.new_revno)
2252
3156
        self._show_tag_conficts(to_file)
2253
3157
 
2254
3158
 
2260
3164
 
2261
3165
    def __init__(self, branch):
2262
3166
        self.branch = branch
 
3167
        self.errors = []
2263
3168
 
2264
3169
    def report_results(self, verbose):
2265
3170
        """Report the check results via trace.note.
2266
 
        
 
3171
 
2267
3172
        :param verbose: Requests more detailed display of what was checked,
2268
3173
            if any.
2269
3174
        """
2270
 
        note('checked branch %s format %s',
2271
 
             self.branch.base,
2272
 
             self.branch._format)
 
3175
        note('checked branch %s format %s', self.branch.user_url,
 
3176
            self.branch._format)
 
3177
        for error in self.errors:
 
3178
            note('found error:%s', error)
2273
3179
 
2274
3180
 
2275
3181
class Converter5to6(object):
2301
3207
        except errors.NoSuchFile:
2302
3208
            pass
2303
3209
        branch.set_bound_location(None)
 
3210
 
 
3211
 
 
3212
class Converter6to7(object):
 
3213
    """Perform an in-place upgrade of format 6 to format 7"""
 
3214
 
 
3215
    def convert(self, branch):
 
3216
        format = BzrBranchFormat7()
 
3217
        branch._set_config_location('stacked_on_location', '')
 
3218
        # update target format
 
3219
        branch._transport.put_bytes('format', format.get_format_string())
 
3220
 
 
3221
 
 
3222
class Converter7to8(object):
 
3223
    """Perform an in-place upgrade of format 6 to format 7"""
 
3224
 
 
3225
    def convert(self, branch):
 
3226
        format = BzrBranchFormat8()
 
3227
        branch._transport.put_bytes('references', '')
 
3228
        # update target format
 
3229
        branch._transport.put_bytes('format', format.get_format_string())
 
3230
 
 
3231
 
 
3232
def _run_with_write_locked_target(target, callable, *args, **kwargs):
 
3233
    """Run ``callable(*args, **kwargs)``, write-locking target for the
 
3234
    duration.
 
3235
 
 
3236
    _run_with_write_locked_target will attempt to release the lock it acquires.
 
3237
 
 
3238
    If an exception is raised by callable, then that exception *will* be
 
3239
    propagated, even if the unlock attempt raises its own error.  Thus
 
3240
    _run_with_write_locked_target should be preferred to simply doing::
 
3241
 
 
3242
        target.lock_write()
 
3243
        try:
 
3244
            return callable(*args, **kwargs)
 
3245
        finally:
 
3246
            target.unlock()
 
3247
 
 
3248
    """
 
3249
    # This is very similar to bzrlib.decorators.needs_write_lock.  Perhaps they
 
3250
    # should share code?
 
3251
    target.lock_write()
 
3252
    try:
 
3253
        result = callable(*args, **kwargs)
 
3254
    except:
 
3255
        exc_info = sys.exc_info()
 
3256
        try:
 
3257
            target.unlock()
 
3258
        finally:
 
3259
            raise exc_info[0], exc_info[1], exc_info[2]
 
3260
    else:
 
3261
        target.unlock()
 
3262
        return result
 
3263
 
 
3264
 
 
3265
class InterBranch(InterObject):
 
3266
    """This class represents operations taking place between two branches.
 
3267
 
 
3268
    Its instances have methods like pull() and push() and contain
 
3269
    references to the source and target repositories these operations
 
3270
    can be carried out on.
 
3271
    """
 
3272
 
 
3273
    _optimisers = []
 
3274
    """The available optimised InterBranch types."""
 
3275
 
 
3276
    @classmethod
 
3277
    def _get_branch_formats_to_test(klass):
 
3278
        """Return an iterable of format tuples for testing.
 
3279
        
 
3280
        :return: An iterable of (from_format, to_format) to use when testing
 
3281
            this InterBranch class. Each InterBranch class should define this
 
3282
            method itself.
 
3283
        """
 
3284
        raise NotImplementedError(klass._get_branch_formats_to_test)
 
3285
 
 
3286
    @needs_write_lock
 
3287
    def pull(self, overwrite=False, stop_revision=None,
 
3288
             possible_transports=None, local=False):
 
3289
        """Mirror source into target branch.
 
3290
 
 
3291
        The target branch is considered to be 'local', having low latency.
 
3292
 
 
3293
        :returns: PullResult instance
 
3294
        """
 
3295
        raise NotImplementedError(self.pull)
 
3296
 
 
3297
    @needs_write_lock
 
3298
    def update_revisions(self, stop_revision=None, overwrite=False,
 
3299
                         graph=None):
 
3300
        """Pull in new perfect-fit revisions.
 
3301
 
 
3302
        :param stop_revision: Updated until the given revision
 
3303
        :param overwrite: Always set the branch pointer, rather than checking
 
3304
            to see if it is a proper descendant.
 
3305
        :param graph: A Graph object that can be used to query history
 
3306
            information. This can be None.
 
3307
        :return: None
 
3308
        """
 
3309
        raise NotImplementedError(self.update_revisions)
 
3310
 
 
3311
    @needs_write_lock
 
3312
    def push(self, overwrite=False, stop_revision=None,
 
3313
             _override_hook_source_branch=None):
 
3314
        """Mirror the source branch into the target branch.
 
3315
 
 
3316
        The source branch is considered to be 'local', having low latency.
 
3317
        """
 
3318
        raise NotImplementedError(self.push)
 
3319
 
 
3320
    @needs_write_lock
 
3321
    def copy_content_into(self, revision_id=None):
 
3322
        """Copy the content of source into target
 
3323
 
 
3324
        revision_id: if not None, the revision history in the new branch will
 
3325
                     be truncated to end with revision_id.
 
3326
        """
 
3327
        raise NotImplementedError(self.copy_content_into)
 
3328
 
 
3329
 
 
3330
class GenericInterBranch(InterBranch):
 
3331
    """InterBranch implementation that uses public Branch functions."""
 
3332
 
 
3333
    @classmethod
 
3334
    def is_compatible(klass, source, target):
 
3335
        # GenericBranch uses the public API, so always compatible
 
3336
        return True
 
3337
 
 
3338
    @classmethod
 
3339
    def _get_branch_formats_to_test(klass):
 
3340
        return [(BranchFormat._default_format, BranchFormat._default_format)]
 
3341
 
 
3342
    @classmethod
 
3343
    def unwrap_format(klass, format):
 
3344
        if isinstance(format, remote.RemoteBranchFormat):
 
3345
            format._ensure_real()
 
3346
            return format._custom_format
 
3347
        return format                                                                                                  
 
3348
 
 
3349
    @needs_write_lock
 
3350
    def copy_content_into(self, revision_id=None):
 
3351
        """Copy the content of source into target
 
3352
 
 
3353
        revision_id: if not None, the revision history in the new branch will
 
3354
                     be truncated to end with revision_id.
 
3355
        """
 
3356
        self.source.update_references(self.target)
 
3357
        self.source._synchronize_history(self.target, revision_id)
 
3358
        try:
 
3359
            parent = self.source.get_parent()
 
3360
        except errors.InaccessibleParent, e:
 
3361
            mutter('parent was not accessible to copy: %s', e)
 
3362
        else:
 
3363
            if parent:
 
3364
                self.target.set_parent(parent)
 
3365
        if self.source._push_should_merge_tags():
 
3366
            self.source.tags.merge_to(self.target.tags)
 
3367
 
 
3368
    @needs_write_lock
 
3369
    def update_revisions(self, stop_revision=None, overwrite=False,
 
3370
        graph=None):
 
3371
        """See InterBranch.update_revisions()."""
 
3372
        other_revno, other_last_revision = self.source.last_revision_info()
 
3373
        stop_revno = None # unknown
 
3374
        if stop_revision is None:
 
3375
            stop_revision = other_last_revision
 
3376
            if _mod_revision.is_null(stop_revision):
 
3377
                # if there are no commits, we're done.
 
3378
                return
 
3379
            stop_revno = other_revno
 
3380
 
 
3381
        # what's the current last revision, before we fetch [and change it
 
3382
        # possibly]
 
3383
        last_rev = _mod_revision.ensure_null(self.target.last_revision())
 
3384
        # we fetch here so that we don't process data twice in the common
 
3385
        # case of having something to pull, and so that the check for
 
3386
        # already merged can operate on the just fetched graph, which will
 
3387
        # be cached in memory.
 
3388
        self.target.fetch(self.source, stop_revision)
 
3389
        # Check to see if one is an ancestor of the other
 
3390
        if not overwrite:
 
3391
            if graph is None:
 
3392
                graph = self.target.repository.get_graph()
 
3393
            if self.target._check_if_descendant_or_diverged(
 
3394
                    stop_revision, last_rev, graph, self.source):
 
3395
                # stop_revision is a descendant of last_rev, but we aren't
 
3396
                # overwriting, so we're done.
 
3397
                return
 
3398
        if stop_revno is None:
 
3399
            if graph is None:
 
3400
                graph = self.target.repository.get_graph()
 
3401
            this_revno, this_last_revision = \
 
3402
                    self.target.last_revision_info()
 
3403
            stop_revno = graph.find_distance_to_null(stop_revision,
 
3404
                            [(other_last_revision, other_revno),
 
3405
                             (this_last_revision, this_revno)])
 
3406
        self.target.set_last_revision_info(stop_revno, stop_revision)
 
3407
 
 
3408
    @needs_write_lock
 
3409
    def pull(self, overwrite=False, stop_revision=None,
 
3410
             possible_transports=None, run_hooks=True,
 
3411
             _override_hook_target=None, local=False):
 
3412
        """Pull from source into self, updating my master if any.
 
3413
 
 
3414
        :param run_hooks: Private parameter - if false, this branch
 
3415
            is being called because it's the master of the primary branch,
 
3416
            so it should not run its hooks.
 
3417
        """
 
3418
        bound_location = self.target.get_bound_location()
 
3419
        if local and not bound_location:
 
3420
            raise errors.LocalRequiresBoundBranch()
 
3421
        master_branch = None
 
3422
        if not local and bound_location and self.source.user_url != bound_location:
 
3423
            # not pulling from master, so we need to update master.
 
3424
            master_branch = self.target.get_master_branch(possible_transports)
 
3425
            master_branch.lock_write()
 
3426
        try:
 
3427
            if master_branch:
 
3428
                # pull from source into master.
 
3429
                master_branch.pull(self.source, overwrite, stop_revision,
 
3430
                    run_hooks=False)
 
3431
            return self._pull(overwrite,
 
3432
                stop_revision, _hook_master=master_branch,
 
3433
                run_hooks=run_hooks,
 
3434
                _override_hook_target=_override_hook_target)
 
3435
        finally:
 
3436
            if master_branch:
 
3437
                master_branch.unlock()
 
3438
 
 
3439
    def push(self, overwrite=False, stop_revision=None,
 
3440
             _override_hook_source_branch=None):
 
3441
        """See InterBranch.push.
 
3442
 
 
3443
        This is the basic concrete implementation of push()
 
3444
 
 
3445
        :param _override_hook_source_branch: If specified, run
 
3446
        the hooks passing this Branch as the source, rather than self.
 
3447
        This is for use of RemoteBranch, where push is delegated to the
 
3448
        underlying vfs-based Branch.
 
3449
        """
 
3450
        # TODO: Public option to disable running hooks - should be trivial but
 
3451
        # needs tests.
 
3452
        self.source.lock_read()
 
3453
        try:
 
3454
            return _run_with_write_locked_target(
 
3455
                self.target, self._push_with_bound_branches, overwrite,
 
3456
                stop_revision,
 
3457
                _override_hook_source_branch=_override_hook_source_branch)
 
3458
        finally:
 
3459
            self.source.unlock()
 
3460
 
 
3461
    def _push_with_bound_branches(self, overwrite, stop_revision,
 
3462
            _override_hook_source_branch=None):
 
3463
        """Push from source into target, and into target's master if any.
 
3464
        """
 
3465
        def _run_hooks():
 
3466
            if _override_hook_source_branch:
 
3467
                result.source_branch = _override_hook_source_branch
 
3468
            for hook in Branch.hooks['post_push']:
 
3469
                hook(result)
 
3470
 
 
3471
        bound_location = self.target.get_bound_location()
 
3472
        if bound_location and self.target.base != bound_location:
 
3473
            # there is a master branch.
 
3474
            #
 
3475
            # XXX: Why the second check?  Is it even supported for a branch to
 
3476
            # be bound to itself? -- mbp 20070507
 
3477
            master_branch = self.target.get_master_branch()
 
3478
            master_branch.lock_write()
 
3479
            try:
 
3480
                # push into the master from the source branch.
 
3481
                self.source._basic_push(master_branch, overwrite, stop_revision)
 
3482
                # and push into the target branch from the source. Note that we
 
3483
                # push from the source branch again, because it's considered the
 
3484
                # highest bandwidth repository.
 
3485
                result = self.source._basic_push(self.target, overwrite,
 
3486
                    stop_revision)
 
3487
                result.master_branch = master_branch
 
3488
                result.local_branch = self.target
 
3489
                _run_hooks()
 
3490
                return result
 
3491
            finally:
 
3492
                master_branch.unlock()
 
3493
        else:
 
3494
            # no master branch
 
3495
            result = self.source._basic_push(self.target, overwrite,
 
3496
                stop_revision)
 
3497
            # TODO: Why set master_branch and local_branch if there's no
 
3498
            # binding?  Maybe cleaner to just leave them unset? -- mbp
 
3499
            # 20070504
 
3500
            result.master_branch = self.target
 
3501
            result.local_branch = None
 
3502
            _run_hooks()
 
3503
            return result
 
3504
 
 
3505
    def _pull(self, overwrite=False, stop_revision=None,
 
3506
             possible_transports=None, _hook_master=None, run_hooks=True,
 
3507
             _override_hook_target=None, local=False):
 
3508
        """See Branch.pull.
 
3509
 
 
3510
        This function is the core worker, used by GenericInterBranch.pull to
 
3511
        avoid duplication when pulling source->master and source->local.
 
3512
 
 
3513
        :param _hook_master: Private parameter - set the branch to
 
3514
            be supplied as the master to pull hooks.
 
3515
        :param run_hooks: Private parameter - if false, this branch
 
3516
            is being called because it's the master of the primary branch,
 
3517
            so it should not run its hooks.
 
3518
        :param _override_hook_target: Private parameter - set the branch to be
 
3519
            supplied as the target_branch to pull hooks.
 
3520
        :param local: Only update the local branch, and not the bound branch.
 
3521
        """
 
3522
        # This type of branch can't be bound.
 
3523
        if local:
 
3524
            raise errors.LocalRequiresBoundBranch()
 
3525
        result = PullResult()
 
3526
        result.source_branch = self.source
 
3527
        if _override_hook_target is None:
 
3528
            result.target_branch = self.target
 
3529
        else:
 
3530
            result.target_branch = _override_hook_target
 
3531
        self.source.lock_read()
 
3532
        try:
 
3533
            # We assume that during 'pull' the target repository is closer than
 
3534
            # the source one.
 
3535
            self.source.update_references(self.target)
 
3536
            graph = self.target.repository.get_graph(self.source.repository)
 
3537
            # TODO: Branch formats should have a flag that indicates 
 
3538
            # that revno's are expensive, and pull() should honor that flag.
 
3539
            # -- JRV20090506
 
3540
            result.old_revno, result.old_revid = \
 
3541
                self.target.last_revision_info()
 
3542
            self.target.update_revisions(self.source, stop_revision,
 
3543
                overwrite=overwrite, graph=graph)
 
3544
            # TODO: The old revid should be specified when merging tags, 
 
3545
            # so a tags implementation that versions tags can only 
 
3546
            # pull in the most recent changes. -- JRV20090506
 
3547
            result.tag_conflicts = self.source.tags.merge_to(self.target.tags,
 
3548
                overwrite)
 
3549
            result.new_revno, result.new_revid = self.target.last_revision_info()
 
3550
            if _hook_master:
 
3551
                result.master_branch = _hook_master
 
3552
                result.local_branch = result.target_branch
 
3553
            else:
 
3554
                result.master_branch = result.target_branch
 
3555
                result.local_branch = None
 
3556
            if run_hooks:
 
3557
                for hook in Branch.hooks['post_pull']:
 
3558
                    hook(result)
 
3559
        finally:
 
3560
            self.source.unlock()
 
3561
        return result
 
3562
 
 
3563
 
 
3564
InterBranch.register_optimiser(GenericInterBranch)