~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Jelmer Vernooij
  • Date: 2011-12-19 13:23:58 UTC
  • mto: This revision was merged to the branch mainline in revision 6386.
  • Revision ID: jelmer@canonical.com-20111219132358-uvs5a6y92gomzacd
Move importing from future until after doc string, otherwise the doc string will disappear.

Show diffs side-by-side

added added

removed removed

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