~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: wang
  • Date: 2006-10-29 13:41:32 UTC
  • mto: (2104.4.1 wang_65714)
  • mto: This revision was merged to the branch mainline in revision 2109.
  • Revision ID: wang@ubuntu-20061029134132-3d7f4216f20c4aef
Replace python's difflib by patiencediff because the worst case 
performance is cubic for difflib and people commiting large data 
files are often hurt by this. The worst case performance of patience is 
quadratic. Fix bug 65714.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2008, 2009 Canonical Ltd
 
1
# Copyright (C) 2005, 2006 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
 
18
 
import sys
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
from cStringIO import StringIO
19
19
 
20
20
from bzrlib.lazy_import import lazy_import
21
21
lazy_import(globals(), """
22
 
from itertools import chain
 
22
from copy import deepcopy
 
23
from unittest import TestSuite
 
24
from warnings import warn
 
25
 
 
26
import bzrlib
23
27
from bzrlib import (
24
28
        bzrdir,
25
29
        cache_utf8,
26
 
        config as _mod_config,
27
 
        debug,
28
30
        errors,
29
31
        lockdir,
30
32
        lockable_files,
31
 
        repository,
 
33
        osutils,
32
34
        revision as _mod_revision,
33
 
        symbol_versioning,
34
35
        transport,
35
 
        tsort,
 
36
        tree,
36
37
        ui,
37
38
        urlutils,
38
39
        )
39
 
from bzrlib.config import BranchConfig, TransportConfig
40
 
from bzrlib.repofmt.pack_repo import RepositoryFormatKnitPack5RichRoot
41
 
from bzrlib.tag import (
42
 
    BasicTags,
43
 
    DisabledTags,
44
 
    )
 
40
from bzrlib.config import BranchConfig, TreeConfig
 
41
from bzrlib.lockable_files import LockableFiles, TransportLock
45
42
""")
46
43
 
47
44
from bzrlib.decorators import needs_read_lock, needs_write_lock
48
 
from bzrlib.hooks import HookPoint, Hooks
49
 
from bzrlib.inter import InterObject
50
 
from bzrlib import registry
51
 
from bzrlib.symbol_versioning import (
52
 
    deprecated_in,
53
 
    deprecated_method,
54
 
    )
55
 
from bzrlib.trace import mutter, mutter_callsite, note, is_quiet
 
45
from bzrlib.errors import (BzrError, BzrCheckError, DivergedBranches,
 
46
                           HistoryMissing, InvalidRevisionId,
 
47
                           InvalidRevisionNumber, LockError, NoSuchFile,
 
48
                           NoSuchRevision, NoWorkingTree, NotVersionedError,
 
49
                           NotBranchError, UninitializableFormat,
 
50
                           UnlistableStore, UnlistableBranch,
 
51
                           )
 
52
from bzrlib.symbol_versioning import (deprecated_function,
 
53
                                      deprecated_method,
 
54
                                      DEPRECATED_PARAMETER,
 
55
                                      deprecated_passed,
 
56
                                      zero_eight, zero_nine,
 
57
                                      )
 
58
from bzrlib.trace import mutter, note
56
59
 
57
60
 
58
61
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
59
62
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
60
 
BZR_BRANCH_FORMAT_6 = "Bazaar Branch Format 6 (bzr 0.15)\n"
 
63
BZR_BRANCH_FORMAT_6 = "Bazaar-NG branch, format 6\n"
61
64
 
62
65
 
63
66
# TODO: Maybe include checks for common corruption of newlines, etc?
77
80
 
78
81
    base
79
82
        Base directory/url of the branch.
80
 
 
81
 
    hooks: An instance of BranchHooks.
82
83
    """
83
84
    # this is really an instance variable - FIXME move it there
84
85
    # - RBC 20060112
85
86
    base = None
86
87
 
87
88
    def __init__(self, *ignored, **ignored_too):
88
 
        self.tags = self._format.make_tags(self)
89
 
        self._revision_history_cache = None
90
 
        self._revision_id_to_revno_cache = None
91
 
        self._partial_revision_id_to_revno_cache = {}
92
 
        self._last_revision_info_cache = None
93
 
        self._merge_sorted_revisions_cache = None
94
 
        self._open_hook()
95
 
        hooks = Branch.hooks['open']
96
 
        for hook in hooks:
97
 
            hook(self)
98
 
 
99
 
    def _open_hook(self):
100
 
        """Called by init to allow simpler extension of the base class."""
101
 
 
102
 
    def _activate_fallback_location(self, url):
103
 
        """Activate the branch/repository from url as a fallback repository."""
104
 
        self.repository.add_fallback_repository(
105
 
            self._get_fallback_repository(url))
 
89
        raise NotImplementedError('The Branch class is abstract')
106
90
 
107
91
    def break_lock(self):
108
92
        """Break a lock if one is present from another instance.
118
102
        if master is not None:
119
103
            master.break_lock()
120
104
 
121
 
    def _check_stackable_repo(self):
122
 
        if not self.repository._format.supports_external_lookups:
123
 
            raise errors.UnstackableRepositoryFormat(self.repository._format,
124
 
                self.repository.base)
125
 
 
126
 
    @staticmethod
127
 
    def open(base, _unsupported=False, possible_transports=None):
 
105
    @staticmethod
 
106
    @deprecated_method(zero_eight)
 
107
    def open_downlevel(base):
 
108
        """Open a branch which may be of an old format."""
 
109
        return Branch.open(base, _unsupported=True)
 
110
        
 
111
    @staticmethod
 
112
    def open(base, _unsupported=False):
128
113
        """Open the branch rooted at base.
129
114
 
130
115
        For instance, if the branch is at URL/.bzr/branch,
131
116
        Branch.open(URL) -> a Branch instance.
132
117
        """
133
 
        control = bzrdir.BzrDir.open(base, _unsupported,
134
 
                                     possible_transports=possible_transports)
135
 
        return control.open_branch(_unsupported)
136
 
 
137
 
    @staticmethod
138
 
    def open_from_transport(transport, _unsupported=False):
139
 
        """Open the branch rooted at transport"""
140
 
        control = bzrdir.BzrDir.open_from_transport(transport, _unsupported)
141
 
        return control.open_branch(_unsupported)
142
 
 
143
 
    @staticmethod
144
 
    def open_containing(url, possible_transports=None):
 
118
        control = bzrdir.BzrDir.open(base, _unsupported)
 
119
        return control.open_branch(_unsupported)
 
120
 
 
121
    @staticmethod
 
122
    def open_containing(url):
145
123
        """Open an existing branch which contains url.
146
 
 
 
124
        
147
125
        This probes for a branch at url, and searches upwards from there.
148
126
 
149
127
        Basically we keep looking up until we find the control directory or
150
128
        run into the root.  If there isn't one, raises NotBranchError.
151
 
        If there is one and it is either an unrecognised format or an unsupported
 
129
        If there is one and it is either an unrecognised format or an unsupported 
152
130
        format, UnknownFormatError or UnsupportedFormatError are raised.
153
131
        If there is one, it is returned, along with the unused portion of url.
154
132
        """
155
 
        control, relpath = bzrdir.BzrDir.open_containing(url,
156
 
                                                         possible_transports)
 
133
        control, relpath = bzrdir.BzrDir.open_containing(url)
157
134
        return control.open_branch(), relpath
158
135
 
159
 
    def _push_should_merge_tags(self):
160
 
        """Should _basic_push merge this branch's tags into the target?
161
 
 
162
 
        The default implementation returns False if this branch has no tags,
163
 
        and True the rest of the time.  Subclasses may override this.
164
 
        """
165
 
        return self.supports_tags() and self.tags.get_tag_dict()
 
136
    @staticmethod
 
137
    @deprecated_function(zero_eight)
 
138
    def initialize(base):
 
139
        """Create a new working tree and branch, rooted at 'base' (url)
 
140
 
 
141
        NOTE: This will soon be deprecated in favour of creation
 
142
        through a BzrDir.
 
143
        """
 
144
        return bzrdir.BzrDir.create_standalone_workingtree(base).branch
 
145
 
 
146
    @deprecated_function(zero_eight)
 
147
    def setup_caching(self, cache_root):
 
148
        """Subclasses that care about caching should override this, and set
 
149
        up cached stores located under cache_root.
 
150
        
 
151
        NOTE: This is unused.
 
152
        """
 
153
        pass
166
154
 
167
155
    def get_config(self):
168
156
        return BranchConfig(self)
169
157
 
170
 
    def _get_config(self):
171
 
        """Get the concrete config for just the config in this branch.
172
 
 
173
 
        This is not intended for client use; see Branch.get_config for the
174
 
        public API.
175
 
 
176
 
        Added in 1.14.
177
 
 
178
 
        :return: An object supporting get_option and set_option.
179
 
        """
180
 
        raise NotImplementedError(self._get_config)
181
 
 
182
 
    def _get_fallback_repository(self, url):
183
 
        """Get the repository we fallback to at url."""
184
 
        url = urlutils.join(self.base, url)
185
 
        a_bzrdir = bzrdir.BzrDir.open(url,
186
 
            possible_transports=[self.bzrdir.root_transport])
187
 
        return a_bzrdir.open_branch().repository
188
 
 
189
 
    def _get_tags_bytes(self):
190
 
        """Get the bytes of a serialised tags dict.
191
 
 
192
 
        Note that not all branches support tags, nor do all use the same tags
193
 
        logic: this method is specific to BasicTags. Other tag implementations
194
 
        may use the same method name and behave differently, safely, because
195
 
        of the double-dispatch via
196
 
        format.make_tags->tags_instance->get_tags_dict.
197
 
 
198
 
        :return: The bytes of the tags file.
199
 
        :seealso: Branch._set_tags_bytes.
200
 
        """
201
 
        return self._transport.get_bytes('tags')
202
 
 
203
 
    def _get_nick(self, local=False, possible_transports=None):
204
 
        config = self.get_config()
205
 
        # explicit overrides master, but don't look for master if local is True
206
 
        if not local and not config.has_explicit_nickname():
207
 
            try:
208
 
                master = self.get_master_branch(possible_transports)
209
 
                if master is not None:
210
 
                    # return the master branch value
211
 
                    return master.nick
212
 
            except errors.BzrError, e:
213
 
                # Silently fall back to local implicit nick if the master is
214
 
                # unavailable
215
 
                mutter("Could not connect to bound branch, "
216
 
                    "falling back to local nick.\n " + str(e))
217
 
        return config.get_nickname()
 
158
    def _get_nick(self):
 
159
        return self.get_config().get_nickname()
218
160
 
219
161
    def _set_nick(self, nick):
220
 
        self.get_config().set_user_option('nickname', nick, warn_masked=True)
 
162
        self.get_config().set_user_option('nickname', nick)
221
163
 
222
164
    nick = property(_get_nick, _set_nick)
223
165
 
224
166
    def is_locked(self):
225
167
        raise NotImplementedError(self.is_locked)
226
168
 
227
 
    def _lefthand_history(self, revision_id, last_rev=None,
228
 
                          other_branch=None):
229
 
        if 'evil' in debug.debug_flags:
230
 
            mutter_callsite(4, "_lefthand_history scales with history.")
231
 
        # stop_revision must be a descendant of last_revision
232
 
        graph = self.repository.get_graph()
233
 
        if last_rev is not None:
234
 
            if not graph.is_ancestor(last_rev, revision_id):
235
 
                # our previous tip is not merged into stop_revision
236
 
                raise errors.DivergedBranches(self, other_branch)
237
 
        # make a new revision history from the graph
238
 
        parents_map = graph.get_parent_map([revision_id])
239
 
        if revision_id not in parents_map:
240
 
            raise errors.NoSuchRevision(self, revision_id)
241
 
        current_rev_id = revision_id
242
 
        new_history = []
243
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
244
 
        # Do not include ghosts or graph origin in revision_history
245
 
        while (current_rev_id in parents_map and
246
 
               len(parents_map[current_rev_id]) > 0):
247
 
            check_not_reserved_id(current_rev_id)
248
 
            new_history.append(current_rev_id)
249
 
            current_rev_id = parents_map[current_rev_id][0]
250
 
            parents_map = graph.get_parent_map([current_rev_id])
251
 
        new_history.reverse()
252
 
        return new_history
253
 
 
254
169
    def lock_write(self):
255
170
        raise NotImplementedError(self.lock_write)
256
171
 
267
182
    def get_physical_lock_status(self):
268
183
        raise NotImplementedError(self.get_physical_lock_status)
269
184
 
270
 
    @needs_read_lock
271
 
    def dotted_revno_to_revision_id(self, revno, _cache_reverse=False):
272
 
        """Return the revision_id for a dotted revno.
273
 
 
274
 
        :param revno: a tuple like (1,) or (1,1,2)
275
 
        :param _cache_reverse: a private parameter enabling storage
276
 
           of the reverse mapping in a top level cache. (This should
277
 
           only be done in selective circumstances as we want to
278
 
           avoid having the mapping cached multiple times.)
279
 
        :return: the revision_id
280
 
        :raises errors.NoSuchRevision: if the revno doesn't exist
281
 
        """
282
 
        rev_id = self._do_dotted_revno_to_revision_id(revno)
283
 
        if _cache_reverse:
284
 
            self._partial_revision_id_to_revno_cache[rev_id] = revno
285
 
        return rev_id
286
 
 
287
 
    def _do_dotted_revno_to_revision_id(self, revno):
288
 
        """Worker function for dotted_revno_to_revision_id.
289
 
 
290
 
        Subclasses should override this if they wish to
291
 
        provide a more efficient implementation.
292
 
        """
293
 
        if len(revno) == 1:
294
 
            return self.get_rev_id(revno[0])
295
 
        revision_id_to_revno = self.get_revision_id_to_revno_map()
296
 
        revision_ids = [revision_id for revision_id, this_revno
297
 
                        in revision_id_to_revno.iteritems()
298
 
                        if revno == this_revno]
299
 
        if len(revision_ids) == 1:
300
 
            return revision_ids[0]
301
 
        else:
302
 
            revno_str = '.'.join(map(str, revno))
303
 
            raise errors.NoSuchRevision(self, revno_str)
304
 
 
305
 
    @needs_read_lock
306
 
    def revision_id_to_dotted_revno(self, revision_id):
307
 
        """Given a revision id, return its dotted revno.
308
 
 
309
 
        :return: a tuple like (1,) or (400,1,3).
310
 
        """
311
 
        return self._do_revision_id_to_dotted_revno(revision_id)
312
 
 
313
 
    def _do_revision_id_to_dotted_revno(self, revision_id):
314
 
        """Worker function for revision_id_to_revno."""
315
 
        # Try the caches if they are loaded
316
 
        result = self._partial_revision_id_to_revno_cache.get(revision_id)
317
 
        if result is not None:
318
 
            return result
319
 
        if self._revision_id_to_revno_cache:
320
 
            result = self._revision_id_to_revno_cache.get(revision_id)
321
 
            if result is None:
322
 
                raise errors.NoSuchRevision(self, revision_id)
323
 
        # Try the mainline as it's optimised
324
 
        try:
325
 
            revno = self.revision_id_to_revno(revision_id)
326
 
            return (revno,)
327
 
        except errors.NoSuchRevision:
328
 
            # We need to load and use the full revno map after all
329
 
            result = self.get_revision_id_to_revno_map().get(revision_id)
330
 
            if result is None:
331
 
                raise errors.NoSuchRevision(self, revision_id)
332
 
        return result
333
 
 
334
 
    @needs_read_lock
335
 
    def get_revision_id_to_revno_map(self):
336
 
        """Return the revision_id => dotted revno map.
337
 
 
338
 
        This will be regenerated on demand, but will be cached.
339
 
 
340
 
        :return: A dictionary mapping revision_id => dotted revno.
341
 
            This dictionary should not be modified by the caller.
342
 
        """
343
 
        if self._revision_id_to_revno_cache is not None:
344
 
            mapping = self._revision_id_to_revno_cache
345
 
        else:
346
 
            mapping = self._gen_revno_map()
347
 
            self._cache_revision_id_to_revno(mapping)
348
 
        # TODO: jam 20070417 Since this is being cached, should we be returning
349
 
        #       a copy?
350
 
        # I would rather not, and instead just declare that users should not
351
 
        # modify the return value.
352
 
        return mapping
353
 
 
354
 
    def _gen_revno_map(self):
355
 
        """Create a new mapping from revision ids to dotted revnos.
356
 
 
357
 
        Dotted revnos are generated based on the current tip in the revision
358
 
        history.
359
 
        This is the worker function for get_revision_id_to_revno_map, which
360
 
        just caches the return value.
361
 
 
362
 
        :return: A dictionary mapping revision_id => dotted revno.
363
 
        """
364
 
        revision_id_to_revno = dict((rev_id, revno)
365
 
            for rev_id, depth, revno, end_of_merge
366
 
             in self.iter_merge_sorted_revisions())
367
 
        return revision_id_to_revno
368
 
 
369
 
    @needs_read_lock
370
 
    def iter_merge_sorted_revisions(self, start_revision_id=None,
371
 
            stop_revision_id=None, stop_rule='exclude', direction='reverse'):
372
 
        """Walk the revisions for a branch in merge sorted order.
373
 
 
374
 
        Merge sorted order is the output from a merge-aware,
375
 
        topological sort, i.e. all parents come before their
376
 
        children going forward; the opposite for reverse.
377
 
 
378
 
        :param start_revision_id: the revision_id to begin walking from.
379
 
            If None, the branch tip is used.
380
 
        :param stop_revision_id: the revision_id to terminate the walk
381
 
            after. If None, the rest of history is included.
382
 
        :param stop_rule: if stop_revision_id is not None, the precise rule
383
 
            to use for termination:
384
 
            * 'exclude' - leave the stop revision out of the result (default)
385
 
            * 'include' - the stop revision is the last item in the result
386
 
            * 'with-merges' - include the stop revision and all of its
387
 
              merged revisions in the result
388
 
        :param direction: either 'reverse' or 'forward':
389
 
            * reverse means return the start_revision_id first, i.e.
390
 
              start at the most recent revision and go backwards in history
391
 
            * forward returns tuples in the opposite order to reverse.
392
 
              Note in particular that forward does *not* do any intelligent
393
 
              ordering w.r.t. depth as some clients of this API may like.
394
 
              (If required, that ought to be done at higher layers.)
395
 
 
396
 
        :return: an iterator over (revision_id, depth, revno, end_of_merge)
397
 
            tuples where:
398
 
 
399
 
            * revision_id: the unique id of the revision
400
 
            * depth: How many levels of merging deep this node has been
401
 
              found.
402
 
            * revno_sequence: This field provides a sequence of
403
 
              revision numbers for all revisions. The format is:
404
 
              (REVNO, BRANCHNUM, BRANCHREVNO). BRANCHNUM is the number of the
405
 
              branch that the revno is on. From left to right the REVNO numbers
406
 
              are the sequence numbers within that branch of the revision.
407
 
            * end_of_merge: When True the next node (earlier in history) is
408
 
              part of a different merge.
409
 
        """
410
 
        # Note: depth and revno values are in the context of the branch so
411
 
        # we need the full graph to get stable numbers, regardless of the
412
 
        # start_revision_id.
413
 
        if self._merge_sorted_revisions_cache is None:
414
 
            last_revision = self.last_revision()
415
 
            graph = self.repository.get_graph()
416
 
            parent_map = dict(((key, value) for key, value in
417
 
                     graph.iter_ancestry([last_revision]) if value is not None))
418
 
            revision_graph = repository._strip_NULL_ghosts(parent_map)
419
 
            revs = tsort.merge_sort(revision_graph, last_revision, None,
420
 
                generate_revno=True)
421
 
            # Drop the sequence # before caching
422
 
            self._merge_sorted_revisions_cache = [r[1:] for r in revs]
423
 
 
424
 
        filtered = self._filter_merge_sorted_revisions(
425
 
            self._merge_sorted_revisions_cache, start_revision_id,
426
 
            stop_revision_id, stop_rule)
427
 
        if direction == 'reverse':
428
 
            return filtered
429
 
        if direction == 'forward':
430
 
            return reversed(list(filtered))
431
 
        else:
432
 
            raise ValueError('invalid direction %r' % direction)
433
 
 
434
 
    def _filter_merge_sorted_revisions(self, merge_sorted_revisions,
435
 
        start_revision_id, stop_revision_id, stop_rule):
436
 
        """Iterate over an inclusive range of sorted revisions."""
437
 
        rev_iter = iter(merge_sorted_revisions)
438
 
        if start_revision_id is not None:
439
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
440
 
                if rev_id != start_revision_id:
441
 
                    continue
442
 
                else:
443
 
                    # The decision to include the start or not
444
 
                    # depends on the stop_rule if a stop is provided
445
 
                    rev_iter = chain(
446
 
                        iter([(rev_id, depth, revno, end_of_merge)]),
447
 
                        rev_iter)
448
 
                    break
449
 
        if stop_revision_id is None:
450
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
451
 
                yield rev_id, depth, revno, end_of_merge
452
 
        elif stop_rule == 'exclude':
453
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
454
 
                if rev_id == stop_revision_id:
455
 
                    return
456
 
                yield rev_id, depth, revno, end_of_merge
457
 
        elif stop_rule == 'include':
458
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
459
 
                yield rev_id, depth, revno, end_of_merge
460
 
                if rev_id == stop_revision_id:
461
 
                    return
462
 
        elif stop_rule == 'with-merges':
463
 
            stop_rev = self.repository.get_revision(stop_revision_id)
464
 
            if stop_rev.parent_ids:
465
 
                left_parent = stop_rev.parent_ids[0]
466
 
            else:
467
 
                left_parent = _mod_revision.NULL_REVISION
468
 
            for rev_id, depth, revno, end_of_merge in rev_iter:
469
 
                if rev_id == left_parent:
470
 
                    return
471
 
                yield rev_id, depth, revno, end_of_merge
472
 
        else:
473
 
            raise ValueError('invalid stop_rule %r' % stop_rule)
474
 
 
475
 
    def leave_lock_in_place(self):
476
 
        """Tell this branch object not to release the physical lock when this
477
 
        object is unlocked.
478
 
 
479
 
        If lock_write doesn't return a token, then this method is not supported.
480
 
        """
481
 
        self.control_files.leave_in_place()
482
 
 
483
 
    def dont_leave_lock_in_place(self):
484
 
        """Tell this branch object to release the physical lock when this
485
 
        object is unlocked, even if it didn't originally acquire it.
486
 
 
487
 
        If lock_write doesn't return a token, then this method is not supported.
488
 
        """
489
 
        self.control_files.dont_leave_in_place()
 
185
    def abspath(self, name):
 
186
        """Return absolute filename for something in the branch
 
187
        
 
188
        XXX: Robert Collins 20051017 what is this used for? why is it a branch
 
189
        method and not a tree method.
 
190
        """
 
191
        raise NotImplementedError(self.abspath)
490
192
 
491
193
    def bind(self, other):
492
194
        """Bind the local branch the other branch.
504
206
        :param last_revision: What revision to stop at (None for at the end
505
207
                              of the branch.
506
208
        :param pb: An optional progress bar to use.
507
 
        :return: None
 
209
 
 
210
        Returns the copied revision count and the failed revisions in a tuple:
 
211
        (copied, failures).
508
212
        """
509
213
        if self.base == from_branch.base:
510
214
            return (0, [])
511
 
        if pb is not None:
512
 
            symbol_versioning.warn(
513
 
                symbol_versioning.deprecated_in((1, 14, 0))
514
 
                % "pb parameter to fetch()")
 
215
        if pb is None:
 
216
            nested_pb = ui.ui_factory.nested_progress_bar()
 
217
            pb = nested_pb
 
218
        else:
 
219
            nested_pb = None
 
220
 
515
221
        from_branch.lock_read()
516
222
        try:
517
223
            if last_revision is None:
518
 
                last_revision = from_branch.last_revision()
519
 
                last_revision = _mod_revision.ensure_null(last_revision)
 
224
                pb.update('get source history')
 
225
                from_history = from_branch.revision_history()
 
226
                if from_history:
 
227
                    last_revision = from_history[-1]
 
228
                else:
 
229
                    # no history in the source branch
 
230
                    last_revision = _mod_revision.NULL_REVISION
520
231
            return self.repository.fetch(from_branch.repository,
521
232
                                         revision_id=last_revision,
522
 
                                         pb=pb)
 
233
                                         pb=nested_pb)
523
234
        finally:
 
235
            if nested_pb is not None:
 
236
                nested_pb.finished()
524
237
            from_branch.unlock()
525
238
 
526
239
    def get_bound_location(self):
530
243
        branch.
531
244
        """
532
245
        return None
533
 
 
534
 
    def get_old_bound_location(self):
535
 
        """Return the URL of the branch we used to be bound to
536
 
        """
537
 
        raise errors.UpgradeRequired(self.base)
538
 
 
539
 
    def get_commit_builder(self, parents, config=None, timestamp=None,
540
 
                           timezone=None, committer=None, revprops=None,
 
246
    
 
247
    def get_commit_builder(self, parents, config=None, timestamp=None, 
 
248
                           timezone=None, committer=None, revprops=None, 
541
249
                           revision_id=None):
542
250
        """Obtain a CommitBuilder for this branch.
543
 
 
 
251
        
544
252
        :param parents: Revision ids of the parents of the new revision.
545
253
        :param config: Optional configuration to use.
546
254
        :param timestamp: Optional timestamp recorded for commit.
552
260
 
553
261
        if config is None:
554
262
            config = self.get_config()
555
 
 
556
 
        return self.repository.get_commit_builder(self, parents, config,
 
263
        
 
264
        return self.repository.get_commit_builder(self, parents, config, 
557
265
            timestamp, timezone, committer, revprops, revision_id)
558
266
 
559
 
    def get_master_branch(self, possible_transports=None):
 
267
    def get_master_branch(self):
560
268
        """Return the branch we are bound to.
561
 
 
 
269
        
562
270
        :return: Either a Branch, or None
563
271
        """
564
272
        return None
569
277
        The delta is relative to its mainline predecessor, or the
570
278
        empty tree for revision 1.
571
279
        """
 
280
        assert isinstance(revno, int)
572
281
        rh = self.revision_history()
573
282
        if not (1 <= revno <= len(rh)):
574
 
            raise errors.InvalidRevisionNumber(revno)
 
283
            raise InvalidRevisionNumber(revno)
575
284
        return self.repository.get_revision_delta(rh[revno-1])
576
285
 
577
 
    def get_stacked_on_url(self):
578
 
        """Get the URL this branch is stacked against.
579
 
 
580
 
        :raises NotStacked: If the branch is not stacked.
581
 
        :raises UnstackableBranchFormat: If the branch does not support
582
 
            stacking.
583
 
        """
584
 
        raise NotImplementedError(self.get_stacked_on_url)
 
286
    def get_root_id(self):
 
287
        """Return the id of this branches root"""
 
288
        raise NotImplementedError(self.get_root_id)
585
289
 
586
290
    def print_file(self, file, revision_id):
587
291
        """Print `file` to stdout."""
588
292
        raise NotImplementedError(self.print_file)
589
293
 
 
294
    def append_revision(self, *revision_ids):
 
295
        raise NotImplementedError(self.append_revision)
 
296
 
590
297
    def set_revision_history(self, rev_history):
591
298
        raise NotImplementedError(self.set_revision_history)
592
299
 
593
 
    def set_stacked_on_url(self, url):
594
 
        """Set the URL this branch is stacked against.
595
 
 
596
 
        :raises UnstackableBranchFormat: If the branch does not support
597
 
            stacking.
598
 
        :raises UnstackableRepositoryFormat: If the repository does not support
599
 
            stacking.
600
 
        """
601
 
        if not self._format.supports_stacking():
602
 
            raise errors.UnstackableBranchFormat(self._format, self.base)
603
 
        self._check_stackable_repo()
604
 
        if not url:
605
 
            try:
606
 
                old_url = self.get_stacked_on_url()
607
 
            except (errors.NotStacked, errors.UnstackableBranchFormat,
608
 
                errors.UnstackableRepositoryFormat):
609
 
                return
610
 
            url = ''
611
 
            # repositories don't offer an interface to remove fallback
612
 
            # repositories today; take the conceptually simpler option and just
613
 
            # reopen it.
614
 
            self.repository = self.bzrdir.find_repository()
615
 
            # for every revision reference the branch has, ensure it is pulled
616
 
            # in.
617
 
            source_repository = self._get_fallback_repository(old_url)
618
 
            for revision_id in chain([self.last_revision()],
619
 
                self.tags.get_reverse_tag_dict()):
620
 
                self.repository.fetch(source_repository, revision_id,
621
 
                    find_ghosts=True)
622
 
        else:
623
 
            self._activate_fallback_location(url)
624
 
        # write this out after the repository is stacked to avoid setting a
625
 
        # stacked config that doesn't work.
626
 
        self._set_config_location('stacked_on_location', url)
627
 
 
628
 
 
629
 
    def _set_tags_bytes(self, bytes):
630
 
        """Mirror method for _get_tags_bytes.
631
 
 
632
 
        :seealso: Branch._get_tags_bytes.
633
 
        """
634
 
        return _run_with_write_locked_target(self, self._transport.put_bytes,
635
 
            'tags', bytes)
636
 
 
637
 
    def _cache_revision_history(self, rev_history):
638
 
        """Set the cached revision history to rev_history.
639
 
 
640
 
        The revision_history method will use this cache to avoid regenerating
641
 
        the revision history.
642
 
 
643
 
        This API is semi-public; it only for use by subclasses, all other code
644
 
        should consider it to be private.
645
 
        """
646
 
        self._revision_history_cache = rev_history
647
 
 
648
 
    def _cache_revision_id_to_revno(self, revision_id_to_revno):
649
 
        """Set the cached revision_id => revno map to revision_id_to_revno.
650
 
 
651
 
        This API is semi-public; it only for use by subclasses, all other code
652
 
        should consider it to be private.
653
 
        """
654
 
        self._revision_id_to_revno_cache = revision_id_to_revno
655
 
 
656
 
    def _clear_cached_state(self):
657
 
        """Clear any cached data on this branch, e.g. cached revision history.
658
 
 
659
 
        This means the next call to revision_history will need to call
660
 
        _gen_revision_history.
661
 
 
662
 
        This API is semi-public; it only for use by subclasses, all other code
663
 
        should consider it to be private.
664
 
        """
665
 
        self._revision_history_cache = None
666
 
        self._revision_id_to_revno_cache = None
667
 
        self._last_revision_info_cache = None
668
 
        self._merge_sorted_revisions_cache = None
669
 
 
670
 
    def _gen_revision_history(self):
671
 
        """Return sequence of revision hashes on to this branch.
672
 
 
673
 
        Unlike revision_history, this method always regenerates or rereads the
674
 
        revision history, i.e. it does not cache the result, so repeated calls
675
 
        may be expensive.
676
 
 
677
 
        Concrete subclasses should override this instead of revision_history so
678
 
        that subclasses do not need to deal with caching logic.
679
 
 
680
 
        This API is semi-public; it only for use by subclasses, all other code
681
 
        should consider it to be private.
682
 
        """
683
 
        raise NotImplementedError(self._gen_revision_history)
684
 
 
685
 
    @needs_read_lock
686
300
    def revision_history(self):
687
 
        """Return sequence of revision ids on this branch.
688
 
 
689
 
        This method will cache the revision history for as long as it is safe to
690
 
        do so.
691
 
        """
692
 
        if 'evil' in debug.debug_flags:
693
 
            mutter_callsite(3, "revision_history scales with history.")
694
 
        if self._revision_history_cache is not None:
695
 
            history = self._revision_history_cache
696
 
        else:
697
 
            history = self._gen_revision_history()
698
 
            self._cache_revision_history(history)
699
 
        return list(history)
 
301
        """Return sequence of revision hashes on to this branch."""
 
302
        raise NotImplementedError(self.revision_history)
700
303
 
701
304
    def revno(self):
702
305
        """Return current revision number for this branch.
704
307
        That is equivalent to the number of revisions committed to
705
308
        this branch.
706
309
        """
707
 
        return self.last_revision_info()[0]
 
310
        return len(self.revision_history())
708
311
 
709
312
    def unbind(self):
710
313
        """Older format branches cannot bind or unbind."""
711
314
        raise errors.UpgradeRequired(self.base)
712
315
 
713
 
    def set_append_revisions_only(self, enabled):
714
 
        """Older format branches are never restricted to append-only"""
715
 
        raise errors.UpgradeRequired(self.base)
716
 
 
717
316
    def last_revision(self):
718
 
        """Return last revision id, or NULL_REVISION."""
719
 
        return self.last_revision_info()[1]
720
 
 
721
 
    @needs_read_lock
722
 
    def last_revision_info(self):
723
 
        """Return information about the last revision.
724
 
 
725
 
        :return: A tuple (revno, revision_id).
726
 
        """
727
 
        if self._last_revision_info_cache is None:
728
 
            self._last_revision_info_cache = self._last_revision_info()
729
 
        return self._last_revision_info_cache
730
 
 
731
 
    def _last_revision_info(self):
732
 
        rh = self.revision_history()
733
 
        revno = len(rh)
734
 
        if revno:
735
 
            return (revno, rh[-1])
 
317
        """Return last revision id, or None"""
 
318
        ph = self.revision_history()
 
319
        if ph:
 
320
            return ph[-1]
736
321
        else:
737
 
            return (0, _mod_revision.NULL_REVISION)
 
322
            return None
738
323
 
739
 
    @deprecated_method(deprecated_in((1, 6, 0)))
740
324
    def missing_revisions(self, other, stop_revision=None):
741
325
        """Return a list of new revisions that would perfectly fit.
742
 
 
 
326
        
743
327
        If self and other have not diverged, return a list of the revisions
744
328
        present in other, but missing from self.
745
329
        """
750
334
        common_index = min(self_len, other_len) -1
751
335
        if common_index >= 0 and \
752
336
            self_history[common_index] != other_history[common_index]:
753
 
            raise errors.DivergedBranches(self, other)
 
337
            raise DivergedBranches(self, other)
754
338
 
755
339
        if stop_revision is None:
756
340
            stop_revision = other_len
757
341
        else:
 
342
            assert isinstance(stop_revision, int)
758
343
            if stop_revision > other_len:
759
344
                raise errors.NoSuchRevision(self, stop_revision)
760
345
        return other_history[self_len:stop_revision]
761
346
 
762
 
    @needs_write_lock
763
 
    def update_revisions(self, other, stop_revision=None, overwrite=False,
764
 
                         graph=None):
 
347
    def update_revisions(self, other, stop_revision=None):
765
348
        """Pull in new perfect-fit revisions.
766
349
 
767
350
        :param other: Another Branch to pull from
768
351
        :param stop_revision: Updated until the given revision
769
 
        :param overwrite: Always set the branch pointer, rather than checking
770
 
            to see if it is a proper descendant.
771
 
        :param graph: A Graph object that can be used to query history
772
 
            information. This can be None.
773
352
        :return: None
774
353
        """
775
 
        return InterBranch.get(other, self).update_revisions(stop_revision,
776
 
            overwrite, graph)
777
 
 
778
 
    def import_last_revision_info(self, source_repo, revno, revid):
779
 
        """Set the last revision info, importing from another repo if necessary.
780
 
 
781
 
        This is used by the bound branch code to upload a revision to
782
 
        the master branch first before updating the tip of the local branch.
783
 
 
784
 
        :param source_repo: Source repository to optionally fetch from
785
 
        :param revno: Revision number of the new tip
786
 
        :param revid: Revision id of the new tip
787
 
        """
788
 
        if not self.repository.has_same_location(source_repo):
789
 
            self.repository.fetch(source_repo, revision_id=revid)
790
 
        self.set_last_revision_info(revno, revid)
 
354
        raise NotImplementedError(self.update_revisions)
791
355
 
792
356
    def revision_id_to_revno(self, revision_id):
793
357
        """Given a revision id, return its revno"""
794
 
        if _mod_revision.is_null(revision_id):
 
358
        if revision_id is None:
795
359
            return 0
796
360
        history = self.revision_history()
797
361
        try:
798
362
            return history.index(revision_id) + 1
799
363
        except ValueError:
800
 
            raise errors.NoSuchRevision(self, revision_id)
 
364
            raise bzrlib.errors.NoSuchRevision(self, revision_id)
801
365
 
802
366
    def get_rev_id(self, revno, history=None):
803
367
        """Find the revision id of the specified revno."""
804
368
        if revno == 0:
805
 
            return _mod_revision.NULL_REVISION
 
369
            return None
806
370
        if history is None:
807
371
            history = self.revision_history()
808
372
        if revno <= 0 or revno > len(history):
809
 
            raise errors.NoSuchRevision(self, revno)
 
373
            raise bzrlib.errors.NoSuchRevision(self, revno)
810
374
        return history[revno - 1]
811
375
 
812
 
    def pull(self, source, overwrite=False, stop_revision=None,
813
 
             possible_transports=None, _override_hook_target=None):
814
 
        """Mirror source into this branch.
815
 
 
816
 
        This branch is considered to be 'local', having low latency.
817
 
 
818
 
        :returns: PullResult instance
819
 
        """
 
376
    def pull(self, source, overwrite=False, stop_revision=None):
820
377
        raise NotImplementedError(self.pull)
821
378
 
822
 
    def push(self, target, overwrite=False, stop_revision=None):
823
 
        """Mirror this branch into target.
824
 
 
825
 
        This branch is considered to be 'local', having low latency.
826
 
        """
827
 
        raise NotImplementedError(self.push)
828
 
 
829
379
    def basis_tree(self):
830
380
        """Return `Tree` object for last revision."""
831
381
        return self.repository.revision_tree(self.last_revision())
832
382
 
 
383
    def rename_one(self, from_rel, to_rel):
 
384
        """Rename one file.
 
385
 
 
386
        This can change the directory or the filename or both.
 
387
        """
 
388
        raise NotImplementedError(self.rename_one)
 
389
 
 
390
    def move(self, from_paths, to_name):
 
391
        """Rename files.
 
392
 
 
393
        to_name must exist as a versioned directory.
 
394
 
 
395
        If to_name exists and is a directory, the files are moved into
 
396
        it, keeping their old names.  If it is a directory, 
 
397
 
 
398
        Note that to_name is only the last component of the new name;
 
399
        this doesn't change the directory.
 
400
 
 
401
        This returns a list of (from_path, to_path) pairs for each
 
402
        entry that is moved.
 
403
        """
 
404
        raise NotImplementedError(self.move)
 
405
 
833
406
    def get_parent(self):
834
407
        """Return the parent location of the branch.
835
408
 
836
 
        This is the default location for pull/missing.  The usual
 
409
        This is the default location for push/pull/missing.  The usual
837
410
        pattern is that the user can override it by specifying a
838
411
        location.
839
412
        """
840
 
        parent = self._get_parent_location()
841
 
        if parent is None:
842
 
            return parent
843
 
        # This is an old-format absolute path to a local branch
844
 
        # turn it into a url
845
 
        if parent.startswith('/'):
846
 
            parent = urlutils.local_path_to_url(parent.decode('utf8'))
847
 
        try:
848
 
            return urlutils.join(self.base[:-1], parent)
849
 
        except errors.InvalidURLJoin, e:
850
 
            raise errors.InaccessibleParent(parent, self.base)
851
 
 
852
 
    def _get_parent_location(self):
853
 
        raise NotImplementedError(self._get_parent_location)
854
 
 
855
 
    def _set_config_location(self, name, url, config=None,
856
 
                             make_relative=False):
857
 
        if config is None:
858
 
            config = self.get_config()
859
 
        if url is None:
860
 
            url = ''
861
 
        elif make_relative:
862
 
            url = urlutils.relative_url(self.base, url)
863
 
        config.set_user_option(name, url, warn_masked=True)
864
 
 
865
 
    def _get_config_location(self, name, config=None):
866
 
        if config is None:
867
 
            config = self.get_config()
868
 
        location = config.get_user_option(name)
869
 
        if location == '':
870
 
            location = None
871
 
        return location
 
413
        raise NotImplementedError(self.get_parent)
872
414
 
873
415
    def get_submit_branch(self):
874
416
        """Return the submit location of the branch.
886
428
        pattern is that the user can override it by specifying a
887
429
        location.
888
430
        """
889
 
        self.get_config().set_user_option('submit_branch', location,
890
 
            warn_masked=True)
891
 
 
892
 
    def get_public_branch(self):
893
 
        """Return the public location of the branch.
894
 
 
895
 
        This is used by merge directives.
896
 
        """
897
 
        return self._get_config_location('public_branch')
898
 
 
899
 
    def set_public_branch(self, location):
900
 
        """Return the submit location of the branch.
901
 
 
902
 
        This is the default location for bundle.  The usual
903
 
        pattern is that the user can override it by specifying a
904
 
        location.
905
 
        """
906
 
        self._set_config_location('public_branch', location)
 
431
        self.get_config().set_user_option('submit_branch', location)
907
432
 
908
433
    def get_push_location(self):
909
434
        """Return the None or the location to push this branch to."""
910
 
        push_loc = self.get_config().get_user_option('push_location')
911
 
        return push_loc
 
435
        raise NotImplementedError(self.get_push_location)
912
436
 
913
437
    def set_push_location(self, location):
914
438
        """Set a new push location for this branch."""
915
439
        raise NotImplementedError(self.set_push_location)
916
440
 
917
 
    def _run_post_change_branch_tip_hooks(self, old_revno, old_revid):
918
 
        """Run the post_change_branch_tip hooks."""
919
 
        hooks = Branch.hooks['post_change_branch_tip']
920
 
        if not hooks:
921
 
            return
922
 
        new_revno, new_revid = self.last_revision_info()
923
 
        params = ChangeBranchTipParams(
924
 
            self, old_revno, new_revno, old_revid, new_revid)
925
 
        for hook in hooks:
926
 
            hook(params)
927
 
 
928
 
    def _run_pre_change_branch_tip_hooks(self, new_revno, new_revid):
929
 
        """Run the pre_change_branch_tip hooks."""
930
 
        hooks = Branch.hooks['pre_change_branch_tip']
931
 
        if not hooks:
932
 
            return
933
 
        old_revno, old_revid = self.last_revision_info()
934
 
        params = ChangeBranchTipParams(
935
 
            self, old_revno, new_revno, old_revid, new_revid)
936
 
        for hook in hooks:
937
 
            try:
938
 
                hook(params)
939
 
            except errors.TipChangeRejected:
940
 
                raise
941
 
            except Exception:
942
 
                exc_info = sys.exc_info()
943
 
                hook_name = Branch.hooks.get_hook_name(hook)
944
 
                raise errors.HookFailed(
945
 
                    'pre_change_branch_tip', hook_name, exc_info)
946
 
 
947
441
    def set_parent(self, url):
948
442
        raise NotImplementedError(self.set_parent)
949
443
 
950
444
    @needs_write_lock
951
445
    def update(self):
952
 
        """Synchronise this branch with the master branch if any.
 
446
        """Synchronise this branch with the master branch if any. 
953
447
 
954
448
        :return: None or the last_revision pivoted out during the update.
955
449
        """
962
456
        """
963
457
        if revno != 0:
964
458
            self.check_real_revno(revno)
965
 
 
 
459
            
966
460
    def check_real_revno(self, revno):
967
461
        """\
968
462
        Check whether a revno corresponds to a real revision.
969
463
        Zero (the NULL revision) is considered invalid
970
464
        """
971
465
        if revno < 1 or revno > self.revno():
972
 
            raise errors.InvalidRevisionNumber(revno)
 
466
            raise InvalidRevisionNumber(revno)
973
467
 
974
468
    @needs_read_lock
975
 
    def clone(self, to_bzrdir, revision_id=None, repository_policy=None):
 
469
    def clone(self, *args, **kwargs):
976
470
        """Clone this branch into to_bzrdir preserving all semantic values.
977
 
 
978
 
        Most API users will want 'create_clone_on_transport', which creates a
979
 
        new bzrdir and branch on the fly.
980
 
 
 
471
        
981
472
        revision_id: if not None, the revision history in the new branch will
982
473
                     be truncated to end with revision_id.
983
474
        """
984
 
        result = to_bzrdir.create_branch()
985
 
        if repository_policy is not None:
986
 
            repository_policy.configure_branch(result)
 
475
        # for API compatibility, until 0.8 releases we provide the old api:
 
476
        # def clone(self, to_location, revision=None, basis_branch=None, to_branch_format=None):
 
477
        # after 0.8 releases, the *args and **kwargs should be changed:
 
478
        # def clone(self, to_bzrdir, revision_id=None):
 
479
        if (kwargs.get('to_location', None) or
 
480
            kwargs.get('revision', None) or
 
481
            kwargs.get('basis_branch', None) or
 
482
            (len(args) and isinstance(args[0], basestring))):
 
483
            # backwards compatibility api:
 
484
            warn("Branch.clone() has been deprecated for BzrDir.clone() from"
 
485
                 " bzrlib 0.8.", DeprecationWarning, stacklevel=3)
 
486
            # get basis_branch
 
487
            if len(args) > 2:
 
488
                basis_branch = args[2]
 
489
            else:
 
490
                basis_branch = kwargs.get('basis_branch', None)
 
491
            if basis_branch:
 
492
                basis = basis_branch.bzrdir
 
493
            else:
 
494
                basis = None
 
495
            # get revision
 
496
            if len(args) > 1:
 
497
                revision_id = args[1]
 
498
            else:
 
499
                revision_id = kwargs.get('revision', None)
 
500
            # get location
 
501
            if len(args):
 
502
                url = args[0]
 
503
            else:
 
504
                # no default to raise if not provided.
 
505
                url = kwargs.get('to_location')
 
506
            return self.bzrdir.clone(url,
 
507
                                     revision_id=revision_id,
 
508
                                     basis=basis).open_branch()
 
509
        # new cleaner api.
 
510
        # generate args by hand 
 
511
        if len(args) > 1:
 
512
            revision_id = args[1]
 
513
        else:
 
514
            revision_id = kwargs.get('revision_id', None)
 
515
        if len(args):
 
516
            to_bzrdir = args[0]
 
517
        else:
 
518
            # no default to raise if not provided.
 
519
            to_bzrdir = kwargs.get('to_bzrdir')
 
520
        result = self._format.initialize(to_bzrdir)
987
521
        self.copy_content_into(result, revision_id=revision_id)
988
522
        return  result
989
523
 
990
524
    @needs_read_lock
991
 
    def sprout(self, to_bzrdir, revision_id=None, repository_policy=None):
 
525
    def sprout(self, to_bzrdir, revision_id=None):
992
526
        """Create a new line of development from the branch, into to_bzrdir.
993
 
 
994
 
        to_bzrdir controls the branch format.
995
 
 
 
527
        
996
528
        revision_id: if not None, the revision history in the new branch will
997
529
                     be truncated to end with revision_id.
998
530
        """
999
 
        result = to_bzrdir.create_branch()
1000
 
        if repository_policy is not None:
1001
 
            repository_policy.configure_branch(result)
 
531
        result = self._format.initialize(to_bzrdir)
1002
532
        self.copy_content_into(result, revision_id=revision_id)
1003
533
        result.set_parent(self.bzrdir.root_transport.base)
1004
534
        return result
1005
535
 
1006
 
    def _synchronize_history(self, destination, revision_id):
1007
 
        """Synchronize last revision and revision history between branches.
1008
 
 
1009
 
        This version is most efficient when the destination is also a
1010
 
        BzrBranch6, but works for BzrBranch5, as long as the destination's
1011
 
        repository contains all the lefthand ancestors of the intended
1012
 
        last_revision.  If not, set_last_revision_info will fail.
1013
 
 
1014
 
        :param destination: The branch to copy the history into
1015
 
        :param revision_id: The revision-id to truncate history at.  May
1016
 
          be None to copy complete history.
1017
 
        """
1018
 
        source_revno, source_revision_id = self.last_revision_info()
1019
 
        if revision_id is None:
1020
 
            revno, revision_id = source_revno, source_revision_id
1021
 
        elif source_revision_id == revision_id:
1022
 
            # we know the revno without needing to walk all of history
1023
 
            revno = source_revno
1024
 
        else:
1025
 
            # To figure out the revno for a random revision, we need to build
1026
 
            # the revision history, and count its length.
1027
 
            # We don't care about the order, just how long it is.
1028
 
            # Alternatively, we could start at the current location, and count
1029
 
            # backwards. But there is no guarantee that we will find it since
1030
 
            # it may be a merged revision.
1031
 
            revno = len(list(self.repository.iter_reverse_revision_history(
1032
 
                                                                revision_id)))
1033
 
        destination.set_last_revision_info(revno, revision_id)
1034
 
 
1035
536
    @needs_read_lock
1036
537
    def copy_content_into(self, destination, revision_id=None):
1037
538
        """Copy the content of self into destination.
1039
540
        revision_id: if not None, the revision history in the new branch will
1040
541
                     be truncated to end with revision_id.
1041
542
        """
1042
 
        self._synchronize_history(destination, revision_id)
 
543
        new_history = self.revision_history()
 
544
        if revision_id is not None:
 
545
            try:
 
546
                new_history = new_history[:new_history.index(revision_id) + 1]
 
547
            except ValueError:
 
548
                rev = self.repository.get_revision(revision_id)
 
549
                new_history = rev.get_history(self.repository)[1:]
 
550
        destination.set_revision_history(new_history)
1043
551
        try:
1044
552
            parent = self.get_parent()
1045
553
        except errors.InaccessibleParent, e:
1047
555
        else:
1048
556
            if parent:
1049
557
                destination.set_parent(parent)
1050
 
        if self._push_should_merge_tags():
1051
 
            self.tags.merge_to(destination.tags)
1052
558
 
1053
559
    @needs_read_lock
1054
560
    def check(self):
1055
561
        """Check consistency of the branch.
1056
562
 
1057
563
        In particular this checks that revisions given in the revision-history
1058
 
        do actually match up in the revision graph, and that they're all
 
564
        do actually match up in the revision graph, and that they're all 
1059
565
        present in the repository.
1060
 
 
 
566
        
1061
567
        Callers will typically also want to check the repository.
1062
568
 
1063
569
        :return: A BranchCheckResult.
1064
570
        """
1065
571
        mainline_parent_id = None
1066
 
        last_revno, last_revision_id = self.last_revision_info()
1067
 
        real_rev_history = list(self.repository.iter_reverse_revision_history(
1068
 
                                last_revision_id))
1069
 
        real_rev_history.reverse()
1070
 
        if len(real_rev_history) != last_revno:
1071
 
            raise errors.BzrCheckError('revno does not match len(mainline)'
1072
 
                ' %s != %s' % (last_revno, len(real_rev_history)))
1073
 
        # TODO: We should probably also check that real_rev_history actually
1074
 
        #       matches self.revision_history()
1075
 
        for revision_id in real_rev_history:
 
572
        for revision_id in self.revision_history():
1076
573
            try:
1077
574
                revision = self.repository.get_revision(revision_id)
1078
575
            except errors.NoSuchRevision, e:
1091
588
 
1092
589
    def _get_checkout_format(self):
1093
590
        """Return the most suitable metadir for a checkout of this branch.
1094
 
        Weaves are used if this branch's repository uses weaves.
 
591
        Weaves are used if this branch's repostory uses weaves.
1095
592
        """
1096
593
        if isinstance(self.bzrdir, bzrdir.BzrDirPreSplitOut):
1097
 
            from bzrlib.repofmt import weaverepo
 
594
            from bzrlib import repository
1098
595
            format = bzrdir.BzrDirMetaFormat1()
1099
 
            format.repository_format = weaverepo.RepositoryFormat7()
 
596
            format.repository_format = repository.RepositoryFormat7()
1100
597
        else:
1101
 
            format = self.repository.bzrdir.checkout_metadir()
1102
 
            format.set_branch_format(self._format)
 
598
            format = self.repository.bzrdir.cloning_metadir()
1103
599
        return format
1104
600
 
1105
 
    def create_clone_on_transport(self, to_transport, revision_id=None,
1106
 
        stacked_on=None):
1107
 
        """Create a clone of this branch and its bzrdir.
1108
 
 
1109
 
        :param to_transport: The transport to clone onto.
1110
 
        :param revision_id: The revision id to use as tip in the new branch.
1111
 
            If None the tip is obtained from this branch.
1112
 
        :param stacked_on: An optional URL to stack the clone on.
1113
 
        """
1114
 
        # XXX: Fix the bzrdir API to allow getting the branch back from the
1115
 
        # clone call. Or something. 20090224 RBC/spiv.
1116
 
        dir_to = self.bzrdir.clone_on_transport(to_transport,
1117
 
            revision_id=revision_id, stacked_on=stacked_on)
1118
 
        return dir_to.open_branch()
1119
 
 
1120
 
    def create_checkout(self, to_location, revision_id=None,
1121
 
                        lightweight=False, accelerator_tree=None,
1122
 
                        hardlink=False):
 
601
    def create_checkout(self, to_location, revision_id=None, 
 
602
                        lightweight=False):
1123
603
        """Create a checkout of a branch.
1124
 
 
 
604
        
1125
605
        :param to_location: The url to produce the checkout at
1126
606
        :param revision_id: The revision to check out
1127
607
        :param lightweight: If True, produce a lightweight checkout, otherwise,
1128
608
        produce a bound branch (heavyweight checkout)
1129
 
        :param accelerator_tree: A tree which can be used for retrieving file
1130
 
            contents more quickly than the revision tree, i.e. a workingtree.
1131
 
            The revision tree will be used for cases where accelerator_tree's
1132
 
            content is different.
1133
 
        :param hardlink: If true, hard-link files from accelerator_tree,
1134
 
            where possible.
1135
609
        :return: The tree of the created checkout
1136
610
        """
1137
611
        t = transport.get_transport(to_location)
1138
 
        t.ensure_base()
 
612
        try:
 
613
            t.mkdir('.')
 
614
        except errors.FileExists:
 
615
            pass
1139
616
        if lightweight:
1140
 
            format = self._get_checkout_format()
1141
 
            checkout = format.initialize_on_transport(t)
1142
 
            from_branch = BranchReferenceFormat().initialize(checkout, self)
 
617
            checkout = bzrdir.BzrDirMetaFormat1().initialize_on_transport(t)
 
618
            BranchReferenceFormat().initialize(checkout, self)
1143
619
        else:
1144
620
            format = self._get_checkout_format()
1145
621
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
1146
622
                to_location, force_new_tree=False, format=format)
1147
623
            checkout = checkout_branch.bzrdir
1148
624
            checkout_branch.bind(self)
1149
 
            # pull up to the specified revision_id to set the initial
 
625
            # pull up to the specified revision_id to set the initial 
1150
626
            # branch tip correctly, and seed it with history.
1151
627
            checkout_branch.pull(self, stop_revision=revision_id)
1152
 
            from_branch=None
1153
 
        tree = checkout.create_workingtree(revision_id,
1154
 
                                           from_branch=from_branch,
1155
 
                                           accelerator_tree=accelerator_tree,
1156
 
                                           hardlink=hardlink)
1157
 
        basis_tree = tree.basis_tree()
1158
 
        basis_tree.lock_read()
1159
 
        try:
1160
 
            for path, file_id in basis_tree.iter_references():
1161
 
                reference_parent = self.reference_parent(file_id, path)
1162
 
                reference_parent.create_checkout(tree.abspath(path),
1163
 
                    basis_tree.get_reference_revision(file_id, path),
1164
 
                    lightweight)
1165
 
        finally:
1166
 
            basis_tree.unlock()
1167
 
        return tree
1168
 
 
1169
 
    @needs_write_lock
1170
 
    def reconcile(self, thorough=True):
1171
 
        """Make sure the data stored in this branch is consistent."""
1172
 
        from bzrlib.reconcile import BranchReconciler
1173
 
        reconciler = BranchReconciler(self, thorough=thorough)
1174
 
        reconciler.reconcile()
1175
 
        return reconciler
1176
 
 
1177
 
    def reference_parent(self, file_id, path):
1178
 
        """Return the parent branch for a tree-reference file_id
1179
 
        :param file_id: The file_id of the tree reference
1180
 
        :param path: The path of the file_id in the tree
1181
 
        :return: A branch associated with the file_id
1182
 
        """
1183
 
        # FIXME should provide multiple branches, based on config
1184
 
        return Branch.open(self.bzrdir.root_transport.clone(path).base)
1185
 
 
1186
 
    def supports_tags(self):
1187
 
        return self._format.supports_tags()
1188
 
 
1189
 
    def _check_if_descendant_or_diverged(self, revision_a, revision_b, graph,
1190
 
                                         other_branch):
1191
 
        """Ensure that revision_b is a descendant of revision_a.
1192
 
 
1193
 
        This is a helper function for update_revisions.
1194
 
 
1195
 
        :raises: DivergedBranches if revision_b has diverged from revision_a.
1196
 
        :returns: True if revision_b is a descendant of revision_a.
1197
 
        """
1198
 
        relation = self._revision_relations(revision_a, revision_b, graph)
1199
 
        if relation == 'b_descends_from_a':
1200
 
            return True
1201
 
        elif relation == 'diverged':
1202
 
            raise errors.DivergedBranches(self, other_branch)
1203
 
        elif relation == 'a_descends_from_b':
1204
 
            return False
1205
 
        else:
1206
 
            raise AssertionError("invalid relation: %r" % (relation,))
1207
 
 
1208
 
    def _revision_relations(self, revision_a, revision_b, graph):
1209
 
        """Determine the relationship between two revisions.
1210
 
 
1211
 
        :returns: One of: 'a_descends_from_b', 'b_descends_from_a', 'diverged'
1212
 
        """
1213
 
        heads = graph.heads([revision_a, revision_b])
1214
 
        if heads == set([revision_b]):
1215
 
            return 'b_descends_from_a'
1216
 
        elif heads == set([revision_a, revision_b]):
1217
 
            # These branches have diverged
1218
 
            return 'diverged'
1219
 
        elif heads == set([revision_a]):
1220
 
            return 'a_descends_from_b'
1221
 
        else:
1222
 
            raise AssertionError("invalid heads: %r" % (heads,))
 
628
        return checkout.create_workingtree(revision_id)
1223
629
 
1224
630
 
1225
631
class BranchFormat(object):
1230
636
     * a format string,
1231
637
     * an open routine.
1232
638
 
1233
 
    Formats are placed in an dict by their format string for reference
 
639
    Formats are placed in an dict by their format string for reference 
1234
640
    during branch opening. Its not required that these be instances, they
1235
 
    can be classes themselves with class methods - it simply depends on
 
641
    can be classes themselves with class methods - it simply depends on 
1236
642
    whether state is needed for a given format or not.
1237
643
 
1238
644
    Once a format is deprecated, just deprecate the initialize and open
1239
 
    methods on the format class. Do not deprecate the object, as the
 
645
    methods on the format class. Do not deprecate the object, as the 
1240
646
    object will be created every time regardless.
1241
647
    """
1242
648
 
1246
652
    _formats = {}
1247
653
    """The known formats."""
1248
654
 
1249
 
    def __eq__(self, other):
1250
 
        return self.__class__ is other.__class__
1251
 
 
1252
 
    def __ne__(self, other):
1253
 
        return not (self == other)
1254
 
 
1255
655
    @classmethod
1256
656
    def find_format(klass, a_bzrdir):
1257
657
        """Return the format for the branch object in a_bzrdir."""
1259
659
            transport = a_bzrdir.get_branch_transport(None)
1260
660
            format_string = transport.get("format").read()
1261
661
            return klass._formats[format_string]
1262
 
        except errors.NoSuchFile:
1263
 
            raise errors.NotBranchError(path=transport.base)
 
662
        except NoSuchFile:
 
663
            raise NotBranchError(path=transport.base)
1264
664
        except KeyError:
1265
 
            raise errors.UnknownFormatError(format=format_string, kind='branch')
 
665
            raise errors.UnknownFormatError(format=format_string)
1266
666
 
1267
667
    @classmethod
1268
668
    def get_default_format(klass):
1269
669
        """Return the current default format."""
1270
670
        return klass._default_format
1271
671
 
1272
 
    def get_reference(self, a_bzrdir):
1273
 
        """Get the target reference of the branch in a_bzrdir.
1274
 
 
1275
 
        format probing must have been completed before calling
1276
 
        this method - it is assumed that the format of the branch
1277
 
        in a_bzrdir is correct.
1278
 
 
1279
 
        :param a_bzrdir: The bzrdir to get the branch data from.
1280
 
        :return: None if the branch is not a reference branch.
1281
 
        """
1282
 
        return None
1283
 
 
1284
 
    @classmethod
1285
 
    def set_reference(self, a_bzrdir, to_branch):
1286
 
        """Set the target reference of the branch in a_bzrdir.
1287
 
 
1288
 
        format probing must have been completed before calling
1289
 
        this method - it is assumed that the format of the branch
1290
 
        in a_bzrdir is correct.
1291
 
 
1292
 
        :param a_bzrdir: The bzrdir to set the branch reference for.
1293
 
        :param to_branch: branch that the checkout is to reference
1294
 
        """
1295
 
        raise NotImplementedError(self.set_reference)
1296
 
 
1297
672
    def get_format_string(self):
1298
673
        """Return the ASCII format string that identifies this format."""
1299
674
        raise NotImplementedError(self.get_format_string)
1300
675
 
1301
676
    def get_format_description(self):
1302
677
        """Return the short format description for this format."""
1303
 
        raise NotImplementedError(self.get_format_description)
1304
 
 
1305
 
    def _initialize_helper(self, a_bzrdir, utf8_files, lock_type='metadir',
1306
 
                           set_format=True):
1307
 
        """Initialize a branch in a bzrdir, with specified files
1308
 
 
1309
 
        :param a_bzrdir: The bzrdir to initialize the branch in
1310
 
        :param utf8_files: The files to create as a list of
1311
 
            (filename, content) tuples
1312
 
        :param set_format: If True, set the format with
1313
 
            self.get_format_string.  (BzrBranch4 has its format set
1314
 
            elsewhere)
1315
 
        :return: a branch in this format
1316
 
        """
1317
 
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
1318
 
        branch_transport = a_bzrdir.get_branch_transport(self)
1319
 
        lock_map = {
1320
 
            'metadir': ('lock', lockdir.LockDir),
1321
 
            'branch4': ('branch-lock', lockable_files.TransportLock),
1322
 
        }
1323
 
        lock_name, lock_class = lock_map[lock_type]
1324
 
        control_files = lockable_files.LockableFiles(branch_transport,
1325
 
            lock_name, lock_class)
1326
 
        control_files.create_lock()
1327
 
        control_files.lock_write()
1328
 
        if set_format:
1329
 
            utf8_files += [('format', self.get_format_string())]
1330
 
        try:
1331
 
            for (filename, content) in utf8_files:
1332
 
                branch_transport.put_bytes(
1333
 
                    filename, content,
1334
 
                    mode=a_bzrdir._get_file_mode())
1335
 
        finally:
1336
 
            control_files.unlock()
1337
 
        return self.open(a_bzrdir, _found=True)
 
678
        raise NotImplementedError(self.get_format_string)
1338
679
 
1339
680
    def initialize(self, a_bzrdir):
1340
681
        """Create a branch of this format in a_bzrdir."""
1344
685
        """Is this format supported?
1345
686
 
1346
687
        Supported formats can be initialized and opened.
1347
 
        Unsupported formats may not support initialization or committing or
 
688
        Unsupported formats may not support initialization or committing or 
1348
689
        some other features depending on the reason for not being supported.
1349
690
        """
1350
691
        return True
1351
692
 
1352
 
    def make_tags(self, branch):
1353
 
        """Create a tags object for branch.
1354
 
 
1355
 
        This method is on BranchFormat, because BranchFormats are reflected
1356
 
        over the wire via network_name(), whereas full Branch instances require
1357
 
        multiple VFS method calls to operate at all.
1358
 
 
1359
 
        The default implementation returns a disabled-tags instance.
1360
 
 
1361
 
        Note that it is normal for branch to be a RemoteBranch when using tags
1362
 
        on a RemoteBranch.
1363
 
        """
1364
 
        return DisabledTags(branch)
1365
 
 
1366
 
    def network_name(self):
1367
 
        """A simple byte string uniquely identifying this format for RPC calls.
1368
 
 
1369
 
        MetaDir branch formats use their disk format string to identify the
1370
 
        repository over the wire. All in one formats such as bzr < 0.8, and
1371
 
        foreign formats like svn/git and hg should use some marker which is
1372
 
        unique and immutable.
1373
 
        """
1374
 
        raise NotImplementedError(self.network_name)
1375
 
 
1376
 
    def open(self, a_bzrdir, _found=False, ignore_fallbacks=False):
 
693
    def open(self, a_bzrdir, _found=False):
1377
694
        """Return the branch object for a_bzrdir
1378
695
 
1379
 
        :param a_bzrdir: A BzrDir that contains a branch.
1380
 
        :param _found: a private parameter, do not use it. It is used to
1381
 
            indicate if format probing has already be done.
1382
 
        :param ignore_fallbacks: when set, no fallback branches will be opened
1383
 
            (if there are any).  Default is to open fallbacks.
 
696
        _found is a private parameter, do not use it. It is used to indicate
 
697
               if format probing has already be done.
1384
698
        """
1385
699
        raise NotImplementedError(self.open)
1386
700
 
1387
701
    @classmethod
1388
702
    def register_format(klass, format):
1389
 
        """Register a metadir format."""
1390
703
        klass._formats[format.get_format_string()] = format
1391
 
        # Metadir formats have a network name of their format string, and get
1392
 
        # registered as class factories.
1393
 
        network_format_registry.register(format.get_format_string(), format.__class__)
1394
704
 
1395
705
    @classmethod
1396
706
    def set_default_format(klass, format):
1397
707
        klass._default_format = format
1398
708
 
1399
 
    def supports_stacking(self):
1400
 
        """True if this format records a stacked-on branch."""
1401
 
        return False
1402
 
 
1403
709
    @classmethod
1404
710
    def unregister_format(klass, format):
 
711
        assert klass._formats[format.get_format_string()] is format
1405
712
        del klass._formats[format.get_format_string()]
1406
713
 
1407
714
    def __str__(self):
1408
 
        return self.get_format_description().rstrip()
1409
 
 
1410
 
    def supports_tags(self):
1411
 
        """True if this format supports tags stored in the branch"""
1412
 
        return False  # by default
1413
 
 
1414
 
 
1415
 
class BranchHooks(Hooks):
1416
 
    """A dictionary mapping hook name to a list of callables for branch hooks.
1417
 
 
1418
 
    e.g. ['set_rh'] Is the list of items to be called when the
1419
 
    set_revision_history function is invoked.
1420
 
    """
1421
 
 
1422
 
    def __init__(self):
1423
 
        """Create the default hooks.
1424
 
 
1425
 
        These are all empty initially, because by default nothing should get
1426
 
        notified.
1427
 
        """
1428
 
        Hooks.__init__(self)
1429
 
        self.create_hook(HookPoint('set_rh',
1430
 
            "Invoked whenever the revision history has been set via "
1431
 
            "set_revision_history. The api signature is (branch, "
1432
 
            "revision_history), and the branch will be write-locked. "
1433
 
            "The set_rh hook can be expensive for bzr to trigger, a better "
1434
 
            "hook to use is Branch.post_change_branch_tip.", (0, 15), None))
1435
 
        self.create_hook(HookPoint('open',
1436
 
            "Called with the Branch object that has been opened after a "
1437
 
            "branch is opened.", (1, 8), None))
1438
 
        self.create_hook(HookPoint('post_push',
1439
 
            "Called after a push operation completes. post_push is called "
1440
 
            "with a bzrlib.branch.BranchPushResult object and only runs in the "
1441
 
            "bzr client.", (0, 15), None))
1442
 
        self.create_hook(HookPoint('post_pull',
1443
 
            "Called after a pull operation completes. post_pull is called "
1444
 
            "with a bzrlib.branch.PullResult object and only runs in the "
1445
 
            "bzr client.", (0, 15), None))
1446
 
        self.create_hook(HookPoint('pre_commit',
1447
 
            "Called after a commit is calculated but before it is is "
1448
 
            "completed. pre_commit is called with (local, master, old_revno, "
1449
 
            "old_revid, future_revno, future_revid, tree_delta, future_tree"
1450
 
            "). old_revid is NULL_REVISION for the first commit to a branch, "
1451
 
            "tree_delta is a TreeDelta object describing changes from the "
1452
 
            "basis revision. hooks MUST NOT modify this delta. "
1453
 
            " future_tree is an in-memory tree obtained from "
1454
 
            "CommitBuilder.revision_tree() and hooks MUST NOT modify this "
1455
 
            "tree.", (0,91), None))
1456
 
        self.create_hook(HookPoint('post_commit',
1457
 
            "Called in the bzr client after a commit has completed. "
1458
 
            "post_commit is called with (local, master, old_revno, old_revid, "
1459
 
            "new_revno, new_revid). old_revid is NULL_REVISION for the first "
1460
 
            "commit to a branch.", (0, 15), None))
1461
 
        self.create_hook(HookPoint('post_uncommit',
1462
 
            "Called in the bzr client after an uncommit completes. "
1463
 
            "post_uncommit is called with (local, master, old_revno, "
1464
 
            "old_revid, new_revno, new_revid) where local is the local branch "
1465
 
            "or None, master is the target branch, and an empty branch "
1466
 
            "receives new_revno of 0, new_revid of None.", (0, 15), None))
1467
 
        self.create_hook(HookPoint('pre_change_branch_tip',
1468
 
            "Called in bzr client and server before a change to the tip of a "
1469
 
            "branch is made. pre_change_branch_tip is called with a "
1470
 
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1471
 
            "commit, uncommit will all trigger this hook.", (1, 6), None))
1472
 
        self.create_hook(HookPoint('post_change_branch_tip',
1473
 
            "Called in bzr client and server after a change to the tip of a "
1474
 
            "branch is made. post_change_branch_tip is called with a "
1475
 
            "bzrlib.branch.ChangeBranchTipParams. Note that push, pull, "
1476
 
            "commit, uncommit will all trigger this hook.", (1, 4), None))
1477
 
        self.create_hook(HookPoint('transform_fallback_location',
1478
 
            "Called when a stacked branch is activating its fallback "
1479
 
            "locations. transform_fallback_location is called with (branch, "
1480
 
            "url), and should return a new url. Returning the same url "
1481
 
            "allows it to be used as-is, returning a different one can be "
1482
 
            "used to cause the branch to stack on a closer copy of that "
1483
 
            "fallback_location. Note that the branch cannot have history "
1484
 
            "accessing methods called on it during this hook because the "
1485
 
            "fallback locations have not been activated. When there are "
1486
 
            "multiple hooks installed for transform_fallback_location, "
1487
 
            "all are called with the url returned from the previous hook."
1488
 
            "The order is however undefined.", (1, 9), None))
1489
 
 
1490
 
 
1491
 
# install the default hooks into the Branch class.
1492
 
Branch.hooks = BranchHooks()
1493
 
 
1494
 
 
1495
 
class ChangeBranchTipParams(object):
1496
 
    """Object holding parameters passed to *_change_branch_tip hooks.
1497
 
 
1498
 
    There are 5 fields that hooks may wish to access:
1499
 
 
1500
 
    :ivar branch: the branch being changed
1501
 
    :ivar old_revno: revision number before the change
1502
 
    :ivar new_revno: revision number after the change
1503
 
    :ivar old_revid: revision id before the change
1504
 
    :ivar new_revid: revision id after the change
1505
 
 
1506
 
    The revid fields are strings. The revno fields are integers.
1507
 
    """
1508
 
 
1509
 
    def __init__(self, branch, old_revno, new_revno, old_revid, new_revid):
1510
 
        """Create a group of ChangeBranchTip parameters.
1511
 
 
1512
 
        :param branch: The branch being changed.
1513
 
        :param old_revno: Revision number before the change.
1514
 
        :param new_revno: Revision number after the change.
1515
 
        :param old_revid: Tip revision id before the change.
1516
 
        :param new_revid: Tip revision id after the change.
1517
 
        """
1518
 
        self.branch = branch
1519
 
        self.old_revno = old_revno
1520
 
        self.new_revno = new_revno
1521
 
        self.old_revid = old_revid
1522
 
        self.new_revid = new_revid
1523
 
 
1524
 
    def __eq__(self, other):
1525
 
        return self.__dict__ == other.__dict__
1526
 
 
1527
 
    def __repr__(self):
1528
 
        return "<%s of %s from (%s, %s) to (%s, %s)>" % (
1529
 
            self.__class__.__name__, self.branch,
1530
 
            self.old_revno, self.old_revid, self.new_revno, self.new_revid)
 
715
        return self.get_format_string().rstrip()
1531
716
 
1532
717
 
1533
718
class BzrBranchFormat4(BranchFormat):
1544
729
 
1545
730
    def initialize(self, a_bzrdir):
1546
731
        """Create a branch of this format in a_bzrdir."""
 
732
        mutter('creating branch in %s', a_bzrdir.transport.base)
 
733
        branch_transport = a_bzrdir.get_branch_transport(self)
1547
734
        utf8_files = [('revision-history', ''),
1548
735
                      ('branch-name', ''),
1549
736
                      ]
1550
 
        return self._initialize_helper(a_bzrdir, utf8_files,
1551
 
                                       lock_type='branch4', set_format=False)
 
737
        control_files = lockable_files.LockableFiles(branch_transport,
 
738
                             'branch-lock', lockable_files.TransportLock)
 
739
        control_files.create_lock()
 
740
        control_files.lock_write()
 
741
        try:
 
742
            for file, content in utf8_files:
 
743
                control_files.put_utf8(file, content)
 
744
        finally:
 
745
            control_files.unlock()
 
746
        return self.open(a_bzrdir, _found=True)
1552
747
 
1553
748
    def __init__(self):
1554
749
        super(BzrBranchFormat4, self).__init__()
1555
750
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
1556
751
 
1557
 
    def network_name(self):
1558
 
        """The network name for this format is the control dirs disk label."""
1559
 
        return self._matchingbzrdir.get_format_string()
 
752
    def open(self, a_bzrdir, _found=False):
 
753
        """Return the branch object for a_bzrdir
1560
754
 
1561
 
    def open(self, a_bzrdir, _found=False, ignore_fallbacks=False):
1562
 
        """See BranchFormat.open()."""
 
755
        _found is a private parameter, do not use it. It is used to indicate
 
756
               if format probing has already be done.
 
757
        """
1563
758
        if not _found:
1564
759
            # we are being called directly and must probe.
1565
760
            raise NotImplementedError
1572
767
        return "Bazaar-NG branch format 4"
1573
768
 
1574
769
 
1575
 
class BranchFormatMetadir(BranchFormat):
1576
 
    """Common logic for meta-dir based branch formats."""
1577
 
 
1578
 
    def _branch_class(self):
1579
 
        """What class to instantiate on open calls."""
1580
 
        raise NotImplementedError(self._branch_class)
1581
 
 
1582
 
    def network_name(self):
1583
 
        """A simple byte string uniquely identifying this format for RPC calls.
1584
 
 
1585
 
        Metadir branch formats use their format string.
1586
 
        """
1587
 
        return self.get_format_string()
1588
 
 
1589
 
    def open(self, a_bzrdir, _found=False, ignore_fallbacks=False):
1590
 
        """See BranchFormat.open()."""
1591
 
        if not _found:
1592
 
            format = BranchFormat.find_format(a_bzrdir)
1593
 
            if format.__class__ != self.__class__:
1594
 
                raise AssertionError("wrong format %r found for %r" %
1595
 
                    (format, self))
1596
 
        try:
1597
 
            transport = a_bzrdir.get_branch_transport(None)
1598
 
            control_files = lockable_files.LockableFiles(transport, 'lock',
1599
 
                                                         lockdir.LockDir)
1600
 
            return self._branch_class()(_format=self,
1601
 
                              _control_files=control_files,
1602
 
                              a_bzrdir=a_bzrdir,
1603
 
                              _repository=a_bzrdir.find_repository(),
1604
 
                              ignore_fallbacks=ignore_fallbacks)
1605
 
        except errors.NoSuchFile:
1606
 
            raise errors.NotBranchError(path=transport.base)
1607
 
 
1608
 
    def __init__(self):
1609
 
        super(BranchFormatMetadir, self).__init__()
1610
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
1611
 
        self._matchingbzrdir.set_branch_format(self)
1612
 
 
1613
 
    def supports_tags(self):
1614
 
        return True
1615
 
 
1616
 
 
1617
 
class BzrBranchFormat5(BranchFormatMetadir):
 
770
class BzrBranchFormat5(BranchFormat):
1618
771
    """Bzr branch format 5.
1619
772
 
1620
773
    This format has:
1627
780
    This format is new in bzr 0.8.
1628
781
    """
1629
782
 
1630
 
    def _branch_class(self):
1631
 
        return BzrBranch5
1632
 
 
1633
783
    def get_format_string(self):
1634
784
        """See BranchFormat.get_format_string()."""
1635
785
        return "Bazaar-NG branch format 5\n"
1637
787
    def get_format_description(self):
1638
788
        """See BranchFormat.get_format_description()."""
1639
789
        return "Branch format 5"
1640
 
 
 
790
        
1641
791
    def initialize(self, a_bzrdir):
1642
792
        """Create a branch of this format in a_bzrdir."""
 
793
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
 
794
        branch_transport = a_bzrdir.get_branch_transport(self)
1643
795
        utf8_files = [('revision-history', ''),
1644
796
                      ('branch-name', ''),
1645
797
                      ]
1646
 
        return self._initialize_helper(a_bzrdir, utf8_files)
1647
 
 
1648
 
    def supports_tags(self):
1649
 
        return False
1650
 
 
1651
 
 
1652
 
class BzrBranchFormat6(BranchFormatMetadir):
1653
 
    """Branch format with last-revision and tags.
1654
 
 
1655
 
    Unlike previous formats, this has no explicit revision history. Instead,
1656
 
    this just stores the last-revision, and the left-hand history leading
1657
 
    up to there is the history.
1658
 
 
1659
 
    This format was introduced in bzr 0.15
1660
 
    and became the default in 0.91.
1661
 
    """
1662
 
 
1663
 
    def _branch_class(self):
1664
 
        return BzrBranch6
1665
 
 
1666
 
    def get_format_string(self):
1667
 
        """See BranchFormat.get_format_string()."""
1668
 
        return "Bazaar Branch Format 6 (bzr 0.15)\n"
1669
 
 
1670
 
    def get_format_description(self):
1671
 
        """See BranchFormat.get_format_description()."""
1672
 
        return "Branch format 6"
1673
 
 
1674
 
    def initialize(self, a_bzrdir):
1675
 
        """Create a branch of this format in a_bzrdir."""
1676
 
        utf8_files = [('last-revision', '0 null:\n'),
1677
 
                      ('branch.conf', ''),
1678
 
                      ('tags', ''),
1679
 
                      ]
1680
 
        return self._initialize_helper(a_bzrdir, utf8_files)
1681
 
 
1682
 
    def make_tags(self, branch):
1683
 
        """See bzrlib.branch.BranchFormat.make_tags()."""
1684
 
        return BasicTags(branch)
1685
 
 
1686
 
 
1687
 
 
1688
 
class BzrBranchFormat7(BranchFormatMetadir):
1689
 
    """Branch format with last-revision, tags, and a stacked location pointer.
1690
 
 
1691
 
    The stacked location pointer is passed down to the repository and requires
1692
 
    a repository format with supports_external_lookups = True.
1693
 
 
1694
 
    This format was introduced in bzr 1.6.
1695
 
    """
1696
 
 
1697
 
    def _branch_class(self):
1698
 
        return BzrBranch7
1699
 
 
1700
 
    def get_format_string(self):
1701
 
        """See BranchFormat.get_format_string()."""
1702
 
        return "Bazaar Branch Format 7 (needs bzr 1.6)\n"
1703
 
 
1704
 
    def get_format_description(self):
1705
 
        """See BranchFormat.get_format_description()."""
1706
 
        return "Branch format 7"
1707
 
 
1708
 
    def initialize(self, a_bzrdir):
1709
 
        """Create a branch of this format in a_bzrdir."""
1710
 
        utf8_files = [('last-revision', '0 null:\n'),
1711
 
                      ('branch.conf', ''),
1712
 
                      ('tags', ''),
1713
 
                      ]
1714
 
        return self._initialize_helper(a_bzrdir, utf8_files)
 
798
        control_files = lockable_files.LockableFiles(branch_transport, 'lock',
 
799
                                                     lockdir.LockDir)
 
800
        control_files.create_lock()
 
801
        control_files.lock_write()
 
802
        control_files.put_utf8('format', self.get_format_string())
 
803
        try:
 
804
            for file, content in utf8_files:
 
805
                control_files.put_utf8(file, content)
 
806
        finally:
 
807
            control_files.unlock()
 
808
        return self.open(a_bzrdir, _found=True, )
1715
809
 
1716
810
    def __init__(self):
1717
 
        super(BzrBranchFormat7, self).__init__()
1718
 
        self._matchingbzrdir.repository_format = \
1719
 
            RepositoryFormatKnitPack5RichRoot()
1720
 
 
1721
 
    def make_tags(self, branch):
1722
 
        """See bzrlib.branch.BranchFormat.make_tags()."""
1723
 
        return BasicTags(branch)
1724
 
 
1725
 
    def supports_stacking(self):
1726
 
        return True
 
811
        super(BzrBranchFormat5, self).__init__()
 
812
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
 
813
 
 
814
    def open(self, a_bzrdir, _found=False):
 
815
        """Return the branch object for a_bzrdir
 
816
 
 
817
        _found is a private parameter, do not use it. It is used to indicate
 
818
               if format probing has already be done.
 
819
        """
 
820
        if not _found:
 
821
            format = BranchFormat.find_format(a_bzrdir)
 
822
            assert format.__class__ == self.__class__
 
823
        transport = a_bzrdir.get_branch_transport(None)
 
824
        control_files = lockable_files.LockableFiles(transport, 'lock',
 
825
                                                     lockdir.LockDir)
 
826
        return BzrBranch5(_format=self,
 
827
                          _control_files=control_files,
 
828
                          a_bzrdir=a_bzrdir,
 
829
                          _repository=a_bzrdir.find_repository())
 
830
 
 
831
    def __str__(self):
 
832
        return "Bazaar-NG Metadir branch format 5"
1727
833
 
1728
834
 
1729
835
class BranchReferenceFormat(BranchFormat):
1744
850
    def get_format_description(self):
1745
851
        """See BranchFormat.get_format_description()."""
1746
852
        return "Checkout reference format 1"
1747
 
 
1748
 
    def get_reference(self, a_bzrdir):
1749
 
        """See BranchFormat.get_reference()."""
1750
 
        transport = a_bzrdir.get_branch_transport(None)
1751
 
        return transport.get('location').read()
1752
 
 
1753
 
    def set_reference(self, a_bzrdir, to_branch):
1754
 
        """See BranchFormat.set_reference()."""
1755
 
        transport = a_bzrdir.get_branch_transport(None)
1756
 
        location = transport.put_bytes('location', to_branch.base)
1757
 
 
 
853
        
1758
854
    def initialize(self, a_bzrdir, target_branch=None):
1759
855
        """Create a branch of this format in a_bzrdir."""
1760
856
        if target_branch is None:
1766
862
        branch_transport.put_bytes('location',
1767
863
            target_branch.bzrdir.root_transport.base)
1768
864
        branch_transport.put_bytes('format', self.get_format_string())
1769
 
        return self.open(
1770
 
            a_bzrdir, _found=True,
1771
 
            possible_transports=[target_branch.bzrdir.root_transport])
 
865
        return self.open(a_bzrdir, _found=True)
1772
866
 
1773
867
    def __init__(self):
1774
868
        super(BranchReferenceFormat, self).__init__()
1775
869
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
1776
 
        self._matchingbzrdir.set_branch_format(self)
1777
870
 
1778
871
    def _make_reference_clone_function(format, a_branch):
1779
872
        """Create a clone() routine for a branch dynamically."""
1780
 
        def clone(to_bzrdir, revision_id=None,
1781
 
            repository_policy=None):
 
873
        def clone(to_bzrdir, revision_id=None):
1782
874
            """See Branch.clone()."""
1783
875
            return format.initialize(to_bzrdir, a_branch)
1784
876
            # cannot obey revision_id limits when cloning a reference ...
1786
878
            # emit some sort of warning/error to the caller ?!
1787
879
        return clone
1788
880
 
1789
 
    def open(self, a_bzrdir, _found=False, location=None,
1790
 
             possible_transports=None, ignore_fallbacks=False):
 
881
    def open(self, a_bzrdir, _found=False):
1791
882
        """Return the branch that the branch reference in a_bzrdir points at.
1792
883
 
1793
 
        :param a_bzrdir: A BzrDir that contains a branch.
1794
 
        :param _found: a private parameter, do not use it. It is used to
1795
 
            indicate if format probing has already be done.
1796
 
        :param ignore_fallbacks: when set, no fallback branches will be opened
1797
 
            (if there are any).  Default is to open fallbacks.
1798
 
        :param location: The location of the referenced branch.  If
1799
 
            unspecified, this will be determined from the branch reference in
1800
 
            a_bzrdir.
1801
 
        :param possible_transports: An optional reusable transports list.
 
884
        _found is a private parameter, do not use it. It is used to indicate
 
885
               if format probing has already be done.
1802
886
        """
1803
887
        if not _found:
1804
888
            format = BranchFormat.find_format(a_bzrdir)
1805
 
            if format.__class__ != self.__class__:
1806
 
                raise AssertionError("wrong format %r found for %r" %
1807
 
                    (format, self))
1808
 
        if location is None:
1809
 
            location = self.get_reference(a_bzrdir)
1810
 
        real_bzrdir = bzrdir.BzrDir.open(
1811
 
            location, possible_transports=possible_transports)
1812
 
        result = real_bzrdir.open_branch(ignore_fallbacks=ignore_fallbacks)
 
889
            assert format.__class__ == self.__class__
 
890
        transport = a_bzrdir.get_branch_transport(None)
 
891
        real_bzrdir = bzrdir.BzrDir.open(transport.get('location').read())
 
892
        result = real_bzrdir.open_branch()
1813
893
        # this changes the behaviour of result.clone to create a new reference
1814
894
        # rather than a copy of the content of the branch.
1815
895
        # I did not use a proxy object because that needs much more extensive
1822
902
        return result
1823
903
 
1824
904
 
1825
 
network_format_registry = registry.FormatRegistry()
1826
 
"""Registry of formats indexed by their network name.
1827
 
 
1828
 
The network name for a branch format is an identifier that can be used when
1829
 
referring to formats with smart server operations. See
1830
 
BranchFormat.network_name() for more detail.
1831
 
"""
1832
 
 
1833
 
 
1834
905
# formats which have no format string are not discoverable
1835
906
# and not independently creatable, so are not registered.
1836
 
__format5 = BzrBranchFormat5()
1837
 
__format6 = BzrBranchFormat6()
1838
 
__format7 = BzrBranchFormat7()
1839
 
BranchFormat.register_format(__format5)
 
907
__default_format = BzrBranchFormat5()
 
908
BranchFormat.register_format(__default_format)
1840
909
BranchFormat.register_format(BranchReferenceFormat())
1841
 
BranchFormat.register_format(__format6)
1842
 
BranchFormat.register_format(__format7)
1843
 
BranchFormat.set_default_format(__format6)
 
910
BranchFormat.set_default_format(__default_format)
1844
911
_legacy_formats = [BzrBranchFormat4(),
1845
 
    ]
1846
 
network_format_registry.register(
1847
 
    _legacy_formats[0].network_name(), _legacy_formats[0].__class__)
1848
 
 
 
912
                   ]
1849
913
 
1850
914
class BzrBranch(Branch):
1851
915
    """A branch stored in the actual filesystem.
1853
917
    Note that it's "local" in the context of the filesystem; it doesn't
1854
918
    really matter if it's on an nfs/smb/afs/coda/... share, as long as
1855
919
    it's writable, and can be accessed via the normal filesystem API.
1856
 
 
1857
 
    :ivar _transport: Transport for file operations on this branch's
1858
 
        control files, typically pointing to the .bzr/branch directory.
1859
 
    :ivar repository: Repository for this branch.
1860
 
    :ivar base: The url of the base directory for this branch; the one
1861
 
        containing the .bzr directory.
1862
920
    """
1863
 
 
1864
 
    def __init__(self, _format=None,
1865
 
                 _control_files=None, a_bzrdir=None, _repository=None,
1866
 
                 ignore_fallbacks=False):
1867
 
        """Create new branch object at a particular location."""
 
921
    
 
922
    def __init__(self, transport=DEPRECATED_PARAMETER, init=DEPRECATED_PARAMETER,
 
923
                 relax_version_check=DEPRECATED_PARAMETER, _format=None,
 
924
                 _control_files=None, a_bzrdir=None, _repository=None):
 
925
        """Create new branch object at a particular location.
 
926
 
 
927
        transport -- A Transport object, defining how to access files.
 
928
        
 
929
        init -- If True, create new control files in a previously
 
930
             unversioned directory.  If False, the branch must already
 
931
             be versioned.
 
932
 
 
933
        relax_version_check -- If true, the usual check for the branch
 
934
            version is not applied.  This is intended only for
 
935
            upgrade/recovery type use; it's not guaranteed that
 
936
            all operations will work on old format branches.
 
937
        """
1868
938
        if a_bzrdir is None:
1869
 
            raise ValueError('a_bzrdir must be supplied')
 
939
            self.bzrdir = bzrdir.BzrDir.open(transport.base)
1870
940
        else:
1871
941
            self.bzrdir = a_bzrdir
1872
 
        self._base = self.bzrdir.transport.clone('..').base
1873
 
        # XXX: We should be able to just do
1874
 
        #   self.base = self.bzrdir.root_transport.base
1875
 
        # but this does not quite work yet -- mbp 20080522
 
942
        self._transport = self.bzrdir.transport.clone('..')
 
943
        self._base = self._transport.base
1876
944
        self._format = _format
1877
945
        if _control_files is None:
1878
946
            raise ValueError('BzrBranch _control_files is None')
1879
947
        self.control_files = _control_files
1880
 
        self._transport = _control_files._transport
 
948
        if deprecated_passed(init):
 
949
            warn("BzrBranch.__init__(..., init=XXX): The init parameter is "
 
950
                 "deprecated as of bzr 0.8. Please use Branch.create().",
 
951
                 DeprecationWarning,
 
952
                 stacklevel=2)
 
953
            if init:
 
954
                # this is slower than before deprecation, oh well never mind.
 
955
                # -> its deprecated.
 
956
                self._initialize(transport.base)
 
957
        self._check_format(_format)
 
958
        if deprecated_passed(relax_version_check):
 
959
            warn("BzrBranch.__init__(..., relax_version_check=XXX_: The "
 
960
                 "relax_version_check parameter is deprecated as of bzr 0.8. "
 
961
                 "Please use BzrDir.open_downlevel, or a BzrBranchFormat's "
 
962
                 "open() method.",
 
963
                 DeprecationWarning,
 
964
                 stacklevel=2)
 
965
            if (not relax_version_check
 
966
                and not self._format.is_supported()):
 
967
                raise errors.UnsupportedFormatError(format=fmt)
 
968
        if deprecated_passed(transport):
 
969
            warn("BzrBranch.__init__(transport=XXX...): The transport "
 
970
                 "parameter is deprecated as of bzr 0.8. "
 
971
                 "Please use Branch.open, or bzrdir.open_branch().",
 
972
                 DeprecationWarning,
 
973
                 stacklevel=2)
1881
974
        self.repository = _repository
1882
 
        Branch.__init__(self)
1883
975
 
1884
976
    def __str__(self):
1885
977
        return '%s(%r)' % (self.__class__.__name__, self.base)
1887
979
    __repr__ = __str__
1888
980
 
1889
981
    def _get_base(self):
1890
 
        """Returns the directory containing the control directory."""
1891
982
        return self._base
1892
983
 
1893
984
    base = property(_get_base, doc="The URL for the root of this branch.")
1894
985
 
1895
 
    def _get_config(self):
1896
 
        return TransportConfig(self._transport, 'branch.conf')
 
986
    def _finish_transaction(self):
 
987
        """Exit the current transaction."""
 
988
        return self.control_files._finish_transaction()
 
989
 
 
990
    def get_transaction(self):
 
991
        """Return the current active transaction.
 
992
 
 
993
        If no transaction is active, this returns a passthrough object
 
994
        for which all data is immediately flushed and no caching happens.
 
995
        """
 
996
        # this is an explicit function so that we can do tricky stuff
 
997
        # when the storage in rev_storage is elsewhere.
 
998
        # we probably need to hook the two 'lock a location' and 
 
999
        # 'have a transaction' together more delicately, so that
 
1000
        # we can have two locks (branch and storage) and one transaction
 
1001
        # ... and finishing the transaction unlocks both, but unlocking
 
1002
        # does not. - RBC 20051121
 
1003
        return self.control_files.get_transaction()
 
1004
 
 
1005
    def _set_transaction(self, transaction):
 
1006
        """Set a new active transaction."""
 
1007
        return self.control_files._set_transaction(transaction)
 
1008
 
 
1009
    def abspath(self, name):
 
1010
        """See Branch.abspath."""
 
1011
        return self.control_files._transport.abspath(name)
 
1012
 
 
1013
    def _check_format(self, format):
 
1014
        """Identify the branch format if needed.
 
1015
 
 
1016
        The format is stored as a reference to the format object in
 
1017
        self._format for code that needs to check it later.
 
1018
 
 
1019
        The format parameter is either None or the branch format class
 
1020
        used to open this branch.
 
1021
 
 
1022
        FIXME: DELETE THIS METHOD when pre 0.8 support is removed.
 
1023
        """
 
1024
        if format is None:
 
1025
            format = BranchFormat.find_format(self.bzrdir)
 
1026
        self._format = format
 
1027
        mutter("got branch format %s", self._format)
 
1028
 
 
1029
    @needs_read_lock
 
1030
    def get_root_id(self):
 
1031
        """See Branch.get_root_id."""
 
1032
        tree = self.repository.revision_tree(self.last_revision())
 
1033
        return tree.inventory.root.file_id
1897
1034
 
1898
1035
    def is_locked(self):
1899
1036
        return self.control_files.is_locked()
1900
1037
 
1901
 
    def lock_write(self, token=None):
1902
 
        repo_token = self.repository.lock_write()
 
1038
    def lock_write(self):
 
1039
        self.repository.lock_write()
1903
1040
        try:
1904
 
            token = self.control_files.lock_write(token=token)
 
1041
            self.control_files.lock_write()
1905
1042
        except:
1906
1043
            self.repository.unlock()
1907
1044
            raise
1908
 
        return token
1909
1045
 
1910
1046
    def lock_read(self):
1911
1047
        self.repository.lock_read()
1921
1057
            self.control_files.unlock()
1922
1058
        finally:
1923
1059
            self.repository.unlock()
1924
 
        if not self.control_files.is_locked():
1925
 
            # we just released the lock
1926
 
            self._clear_cached_state()
1927
 
 
 
1060
        
1928
1061
    def peek_lock_mode(self):
1929
1062
        if self.control_files._lock_count == 0:
1930
1063
            return None
1939
1072
        """See Branch.print_file."""
1940
1073
        return self.repository.print_file(file, revision_id)
1941
1074
 
1942
 
    def _write_revision_history(self, history):
1943
 
        """Factored out of set_revision_history.
1944
 
 
1945
 
        This performs the actual writing to disk.
1946
 
        It is intended to be called by BzrBranch5.set_revision_history."""
1947
 
        self._transport.put_bytes(
1948
 
            'revision-history', '\n'.join(history),
1949
 
            mode=self.bzrdir._get_file_mode())
 
1075
    @needs_write_lock
 
1076
    def append_revision(self, *revision_ids):
 
1077
        """See Branch.append_revision."""
 
1078
        for revision_id in revision_ids:
 
1079
            mutter("add {%s} to revision-history" % revision_id)
 
1080
        rev_history = self.revision_history()
 
1081
        rev_history.extend(revision_ids)
 
1082
        self.set_revision_history(rev_history)
1950
1083
 
1951
1084
    @needs_write_lock
1952
1085
    def set_revision_history(self, rev_history):
1953
1086
        """See Branch.set_revision_history."""
1954
 
        if 'evil' in debug.debug_flags:
1955
 
            mutter_callsite(3, "set_revision_history scales with history.")
1956
 
        check_not_reserved_id = _mod_revision.check_not_reserved_id
1957
 
        for rev_id in rev_history:
1958
 
            check_not_reserved_id(rev_id)
1959
 
        if Branch.hooks['post_change_branch_tip']:
1960
 
            # Don't calculate the last_revision_info() if there are no hooks
1961
 
            # that will use it.
1962
 
            old_revno, old_revid = self.last_revision_info()
1963
 
        if len(rev_history) == 0:
1964
 
            revid = _mod_revision.NULL_REVISION
1965
 
        else:
1966
 
            revid = rev_history[-1]
1967
 
        self._run_pre_change_branch_tip_hooks(len(rev_history), revid)
1968
 
        self._write_revision_history(rev_history)
1969
 
        self._clear_cached_state()
1970
 
        self._cache_revision_history(rev_history)
1971
 
        for hook in Branch.hooks['set_rh']:
1972
 
            hook(self, rev_history)
1973
 
        if Branch.hooks['post_change_branch_tip']:
1974
 
            self._run_post_change_branch_tip_hooks(old_revno, old_revid)
1975
 
 
1976
 
    def _synchronize_history(self, destination, revision_id):
1977
 
        """Synchronize last revision and revision history between branches.
1978
 
 
1979
 
        This version is most efficient when the destination is also a
1980
 
        BzrBranch5, but works for BzrBranch6 as long as the revision
1981
 
        history is the true lefthand parent history, and all of the revisions
1982
 
        are in the destination's repository.  If not, set_revision_history
1983
 
        will fail.
1984
 
 
1985
 
        :param destination: The branch to copy the history into
1986
 
        :param revision_id: The revision-id to truncate history at.  May
1987
 
          be None to copy complete history.
1988
 
        """
1989
 
        if not isinstance(destination._format, BzrBranchFormat5):
1990
 
            super(BzrBranch, self)._synchronize_history(
1991
 
                destination, revision_id)
1992
 
            return
1993
 
        if revision_id == _mod_revision.NULL_REVISION:
1994
 
            new_history = []
1995
 
        else:
1996
 
            new_history = self.revision_history()
1997
 
        if revision_id is not None and new_history != []:
1998
 
            try:
1999
 
                new_history = new_history[:new_history.index(revision_id) + 1]
2000
 
            except ValueError:
2001
 
                rev = self.repository.get_revision(revision_id)
2002
 
                new_history = rev.get_history(self.repository)[1:]
2003
 
        destination.set_revision_history(new_history)
2004
 
 
2005
 
    @needs_write_lock
2006
 
    def set_last_revision_info(self, revno, revision_id):
2007
 
        """Set the last revision of this branch.
2008
 
 
2009
 
        The caller is responsible for checking that the revno is correct
2010
 
        for this revision id.
2011
 
 
2012
 
        It may be possible to set the branch last revision to an id not
2013
 
        present in the repository.  However, branches can also be
2014
 
        configured to check constraints on history, in which case this may not
2015
 
        be permitted.
2016
 
        """
2017
 
        revision_id = _mod_revision.ensure_null(revision_id)
2018
 
        # this old format stores the full history, but this api doesn't
2019
 
        # provide it, so we must generate, and might as well check it's
2020
 
        # correct
2021
 
        history = self._lefthand_history(revision_id)
2022
 
        if len(history) != revno:
2023
 
            raise AssertionError('%d != %d' % (len(history), revno))
2024
 
        self.set_revision_history(history)
2025
 
 
2026
 
    def _gen_revision_history(self):
2027
 
        history = self._transport.get_bytes('revision-history').split('\n')
2028
 
        if history[-1:] == ['']:
2029
 
            # There shouldn't be a trailing newline, but just in case.
2030
 
            history.pop()
2031
 
        return history
2032
 
 
2033
 
    @needs_write_lock
2034
 
    def generate_revision_history(self, revision_id, last_rev=None,
 
1087
        self.control_files.put_utf8(
 
1088
            'revision-history', '\n'.join(rev_history))
 
1089
        transaction = self.get_transaction()
 
1090
        history = transaction.map.find_revision_history()
 
1091
        if history is not None:
 
1092
            # update the revision history in the identity map.
 
1093
            history[:] = list(rev_history)
 
1094
            # this call is disabled because revision_history is 
 
1095
            # not really an object yet, and the transaction is for objects.
 
1096
            # transaction.register_dirty(history)
 
1097
        else:
 
1098
            transaction.map.add_revision_history(rev_history)
 
1099
            # this call is disabled because revision_history is 
 
1100
            # not really an object yet, and the transaction is for objects.
 
1101
            # transaction.register_clean(history)
 
1102
 
 
1103
    @needs_read_lock
 
1104
    def revision_history(self):
 
1105
        """See Branch.revision_history."""
 
1106
        transaction = self.get_transaction()
 
1107
        history = transaction.map.find_revision_history()
 
1108
        if history is not None:
 
1109
            # mutter("cache hit for revision-history in %s", self)
 
1110
            return list(history)
 
1111
        decode_utf8 = cache_utf8.decode
 
1112
        history = [decode_utf8(l.rstrip('\r\n')) for l in
 
1113
                self.control_files.get('revision-history').readlines()]
 
1114
        transaction.map.add_revision_history(history)
 
1115
        # this call is disabled because revision_history is 
 
1116
        # not really an object yet, and the transaction is for objects.
 
1117
        # transaction.register_clean(history, precious=True)
 
1118
        return list(history)
 
1119
 
 
1120
    @needs_write_lock
 
1121
    def generate_revision_history(self, revision_id, last_rev=None, 
2035
1122
        other_branch=None):
2036
1123
        """Create a new revision history that will finish with revision_id.
2037
 
 
 
1124
        
2038
1125
        :param revision_id: the new tip to use.
2039
1126
        :param last_rev: The previous last_revision. If not None, then this
2040
1127
            must be a ancestory of revision_id, or DivergedBranches is raised.
2041
1128
        :param other_branch: The other branch that DivergedBranches should
2042
1129
            raise with respect to.
2043
1130
        """
2044
 
        self.set_revision_history(self._lefthand_history(revision_id,
2045
 
            last_rev, other_branch))
 
1131
        # stop_revision must be a descendant of last_revision
 
1132
        stop_graph = self.repository.get_revision_graph(revision_id)
 
1133
        if last_rev is not None and last_rev not in stop_graph:
 
1134
            # our previous tip is not merged into stop_revision
 
1135
            raise errors.DivergedBranches(self, other_branch)
 
1136
        # make a new revision history from the graph
 
1137
        current_rev_id = revision_id
 
1138
        new_history = []
 
1139
        while current_rev_id not in (None, _mod_revision.NULL_REVISION):
 
1140
            new_history.append(current_rev_id)
 
1141
            current_rev_id_parents = stop_graph[current_rev_id]
 
1142
            try:
 
1143
                current_rev_id = current_rev_id_parents[0]
 
1144
            except IndexError:
 
1145
                current_rev_id = None
 
1146
        new_history.reverse()
 
1147
        self.set_revision_history(new_history)
 
1148
 
 
1149
    @needs_write_lock
 
1150
    def update_revisions(self, other, stop_revision=None):
 
1151
        """See Branch.update_revisions."""
 
1152
        other.lock_read()
 
1153
        try:
 
1154
            if stop_revision is None:
 
1155
                stop_revision = other.last_revision()
 
1156
                if stop_revision is None:
 
1157
                    # if there are no commits, we're done.
 
1158
                    return
 
1159
            # whats the current last revision, before we fetch [and change it
 
1160
            # possibly]
 
1161
            last_rev = self.last_revision()
 
1162
            # we fetch here regardless of whether we need to so that we pickup
 
1163
            # filled in ghosts.
 
1164
            self.fetch(other, stop_revision)
 
1165
            my_ancestry = self.repository.get_ancestry(last_rev)
 
1166
            if stop_revision in my_ancestry:
 
1167
                # last_revision is a descendant of stop_revision
 
1168
                return
 
1169
            self.generate_revision_history(stop_revision, last_rev=last_rev,
 
1170
                other_branch=other)
 
1171
        finally:
 
1172
            other.unlock()
2046
1173
 
2047
1174
    def basis_tree(self):
2048
1175
        """See Branch.basis_tree."""
2049
1176
        return self.repository.revision_tree(self.last_revision())
2050
1177
 
 
1178
    @deprecated_method(zero_eight)
 
1179
    def working_tree(self):
 
1180
        """Create a Working tree object for this branch."""
 
1181
 
 
1182
        from bzrlib.transport.local import LocalTransport
 
1183
        if (self.base.find('://') != -1 or 
 
1184
            not isinstance(self._transport, LocalTransport)):
 
1185
            raise NoWorkingTree(self.base)
 
1186
        return self.bzrdir.open_workingtree()
 
1187
 
2051
1188
    @needs_write_lock
2052
 
    def pull(self, source, overwrite=False, stop_revision=None,
2053
 
             _hook_master=None, run_hooks=True, possible_transports=None,
2054
 
             _override_hook_target=None):
2055
 
        """See Branch.pull.
2056
 
 
2057
 
        :param _hook_master: Private parameter - set the branch to
2058
 
            be supplied as the master to pull hooks.
2059
 
        :param run_hooks: Private parameter - if false, this branch
2060
 
            is being called because it's the master of the primary branch,
2061
 
            so it should not run its hooks.
2062
 
        :param _override_hook_target: Private parameter - set the branch to be
2063
 
            supplied as the target_branch to pull hooks.
2064
 
        """
2065
 
        result = PullResult()
2066
 
        result.source_branch = source
2067
 
        if _override_hook_target is None:
2068
 
            result.target_branch = self
2069
 
        else:
2070
 
            result.target_branch = _override_hook_target
 
1189
    def pull(self, source, overwrite=False, stop_revision=None):
 
1190
        """See Branch.pull."""
2071
1191
        source.lock_read()
2072
1192
        try:
2073
 
            # We assume that during 'pull' the local repository is closer than
2074
 
            # the remote one.
2075
 
            graph = self.repository.get_graph(source.repository)
2076
 
            result.old_revno, result.old_revid = self.last_revision_info()
2077
 
            self.update_revisions(source, stop_revision, overwrite=overwrite,
2078
 
                                  graph=graph)
2079
 
            result.tag_conflicts = source.tags.merge_to(self.tags, overwrite)
2080
 
            result.new_revno, result.new_revid = self.last_revision_info()
2081
 
            if _hook_master:
2082
 
                result.master_branch = _hook_master
2083
 
                result.local_branch = result.target_branch
2084
 
            else:
2085
 
                result.master_branch = result.target_branch
2086
 
                result.local_branch = None
2087
 
            if run_hooks:
2088
 
                for hook in Branch.hooks['post_pull']:
2089
 
                    hook(result)
 
1193
            old_count = len(self.revision_history())
 
1194
            try:
 
1195
                self.update_revisions(source, stop_revision)
 
1196
            except DivergedBranches:
 
1197
                if not overwrite:
 
1198
                    raise
 
1199
            if overwrite:
 
1200
                self.set_revision_history(source.revision_history())
 
1201
            new_count = len(self.revision_history())
 
1202
            return new_count - old_count
2090
1203
        finally:
2091
1204
            source.unlock()
2092
 
        return result
2093
 
 
2094
 
    def _get_parent_location(self):
 
1205
 
 
1206
    def get_parent(self):
 
1207
        """See Branch.get_parent."""
 
1208
 
2095
1209
        _locs = ['parent', 'pull', 'x-pull']
 
1210
        assert self.base[-1] == '/'
2096
1211
        for l in _locs:
2097
1212
            try:
2098
 
                return self._transport.get_bytes(l).strip('\n')
2099
 
            except errors.NoSuchFile:
2100
 
                pass
 
1213
                parent = self.control_files.get(l).read().strip('\n')
 
1214
            except NoSuchFile:
 
1215
                continue
 
1216
            # This is an old-format absolute path to a local branch
 
1217
            # turn it into a url
 
1218
            if parent.startswith('/'):
 
1219
                parent = urlutils.local_path_to_url(parent.decode('utf8'))
 
1220
            try:
 
1221
                return urlutils.join(self.base[:-1], parent)
 
1222
            except errors.InvalidURLJoin, e:
 
1223
                raise errors.InaccessibleParent(parent, self.base)
2101
1224
        return None
2102
1225
 
2103
 
    @needs_read_lock
2104
 
    def push(self, target, overwrite=False, stop_revision=None,
2105
 
             _override_hook_source_branch=None):
2106
 
        """See Branch.push.
2107
 
 
2108
 
        This is the basic concrete implementation of push()
2109
 
 
2110
 
        :param _override_hook_source_branch: If specified, run
2111
 
        the hooks passing this Branch as the source, rather than self.
2112
 
        This is for use of RemoteBranch, where push is delegated to the
2113
 
        underlying vfs-based Branch.
2114
 
        """
2115
 
        # TODO: Public option to disable running hooks - should be trivial but
2116
 
        # needs tests.
2117
 
        return _run_with_write_locked_target(
2118
 
            target, self._push_with_bound_branches, target, overwrite,
2119
 
            stop_revision,
2120
 
            _override_hook_source_branch=_override_hook_source_branch)
2121
 
 
2122
 
    def _push_with_bound_branches(self, target, overwrite,
2123
 
            stop_revision,
2124
 
            _override_hook_source_branch=None):
2125
 
        """Push from self into target, and into target's master if any.
2126
 
 
2127
 
        This is on the base BzrBranch class even though it doesn't support
2128
 
        bound branches because the *target* might be bound.
2129
 
        """
2130
 
        def _run_hooks():
2131
 
            if _override_hook_source_branch:
2132
 
                result.source_branch = _override_hook_source_branch
2133
 
            for hook in Branch.hooks['post_push']:
2134
 
                hook(result)
2135
 
 
2136
 
        bound_location = target.get_bound_location()
2137
 
        if bound_location and target.base != bound_location:
2138
 
            # there is a master branch.
2139
 
            #
2140
 
            # XXX: Why the second check?  Is it even supported for a branch to
2141
 
            # be bound to itself? -- mbp 20070507
2142
 
            master_branch = target.get_master_branch()
2143
 
            master_branch.lock_write()
2144
 
            try:
2145
 
                # push into the master from this branch.
2146
 
                self._basic_push(master_branch, overwrite, stop_revision)
2147
 
                # and push into the target branch from this. Note that we push from
2148
 
                # this branch again, because its considered the highest bandwidth
2149
 
                # repository.
2150
 
                result = self._basic_push(target, overwrite, stop_revision)
2151
 
                result.master_branch = master_branch
2152
 
                result.local_branch = target
2153
 
                _run_hooks()
2154
 
                return result
2155
 
            finally:
2156
 
                master_branch.unlock()
2157
 
        else:
2158
 
            # no master branch
2159
 
            result = self._basic_push(target, overwrite, stop_revision)
2160
 
            # TODO: Why set master_branch and local_branch if there's no
2161
 
            # binding?  Maybe cleaner to just leave them unset? -- mbp
2162
 
            # 20070504
2163
 
            result.master_branch = target
2164
 
            result.local_branch = None
2165
 
            _run_hooks()
2166
 
            return result
2167
 
 
2168
 
    def _basic_push(self, target, overwrite, stop_revision):
2169
 
        """Basic implementation of push without bound branches or hooks.
2170
 
 
2171
 
        Must be called with self read locked and target write locked.
2172
 
        """
2173
 
        result = BranchPushResult()
2174
 
        result.source_branch = self
2175
 
        result.target_branch = target
2176
 
        result.old_revno, result.old_revid = target.last_revision_info()
2177
 
        if result.old_revid != self.last_revision():
2178
 
            # We assume that during 'push' this repository is closer than
2179
 
            # the target.
2180
 
            graph = self.repository.get_graph(target.repository)
2181
 
            target.update_revisions(self, stop_revision, overwrite=overwrite,
2182
 
                                    graph=graph)
2183
 
        if self._push_should_merge_tags():
2184
 
            result.tag_conflicts = self.tags.merge_to(target.tags, overwrite)
2185
 
        result.new_revno, result.new_revid = target.last_revision_info()
2186
 
        return result
2187
 
 
2188
 
    def get_stacked_on_url(self):
2189
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
 
1226
    def get_push_location(self):
 
1227
        """See Branch.get_push_location."""
 
1228
        push_loc = self.get_config().get_user_option('push_location')
 
1229
        return push_loc
2190
1230
 
2191
1231
    def set_push_location(self, location):
2192
1232
        """See Branch.set_push_location."""
2193
 
        self.get_config().set_user_option(
2194
 
            'push_location', location,
2195
 
            store=_mod_config.STORE_LOCATION_NORECURSE)
 
1233
        self.get_config().set_user_option('push_location', location, 
 
1234
                                          local=True)
2196
1235
 
2197
1236
    @needs_write_lock
2198
1237
    def set_parent(self, url):
2200
1239
        # TODO: Maybe delete old location files?
2201
1240
        # URLs should never be unicode, even on the local fs,
2202
1241
        # FIXUP this and get_parent in a future branch format bump:
2203
 
        # read and rewrite the file. RBC 20060125
2204
 
        if url is not None:
 
1242
        # read and rewrite the file, and have the new format code read
 
1243
        # using .get not .get_utf8. RBC 20060125
 
1244
        if url is None:
 
1245
            self.control_files._transport.delete('parent')
 
1246
        else:
2205
1247
            if isinstance(url, unicode):
2206
 
                try:
 
1248
                try: 
2207
1249
                    url = url.encode('ascii')
2208
1250
                except UnicodeEncodeError:
2209
 
                    raise errors.InvalidURL(url,
 
1251
                    raise bzrlib.errors.InvalidURL(url,
2210
1252
                        "Urls must be 7-bit ascii, "
2211
1253
                        "use bzrlib.urlutils.escape")
 
1254
                    
2212
1255
            url = urlutils.relative_url(self.base, url)
2213
 
        self._set_parent_location(url)
 
1256
            self.control_files.put('parent', StringIO(url + '\n'))
2214
1257
 
2215
 
    def _set_parent_location(self, url):
2216
 
        if url is None:
2217
 
            self._transport.delete('parent')
2218
 
        else:
2219
 
            self._transport.put_bytes('parent', url + '\n',
2220
 
                mode=self.bzrdir._get_file_mode())
 
1258
    @deprecated_function(zero_nine)
 
1259
    def tree_config(self):
 
1260
        """DEPRECATED; call get_config instead.  
 
1261
        TreeConfig has become part of BranchConfig."""
 
1262
        return TreeConfig(self)
2221
1263
 
2222
1264
 
2223
1265
class BzrBranch5(BzrBranch):
2224
 
    """A format 5 branch. This supports new features over plain branches.
 
1266
    """A format 5 branch. This supports new features over plan branches.
2225
1267
 
2226
1268
    It has support for a master_branch which is the data for bound branches.
2227
1269
    """
2228
1270
 
 
1271
    def __init__(self,
 
1272
                 _format,
 
1273
                 _control_files,
 
1274
                 a_bzrdir,
 
1275
                 _repository):
 
1276
        super(BzrBranch5, self).__init__(_format=_format,
 
1277
                                         _control_files=_control_files,
 
1278
                                         a_bzrdir=a_bzrdir,
 
1279
                                         _repository=_repository)
 
1280
        
2229
1281
    @needs_write_lock
2230
 
    def pull(self, source, overwrite=False, stop_revision=None,
2231
 
             run_hooks=True, possible_transports=None,
2232
 
             _override_hook_target=None):
2233
 
        """Pull from source into self, updating my master if any.
2234
 
 
2235
 
        :param run_hooks: Private parameter - if false, this branch
2236
 
            is being called because it's the master of the primary branch,
2237
 
            so it should not run its hooks.
2238
 
        """
 
1282
    def pull(self, source, overwrite=False, stop_revision=None):
 
1283
        """Updates branch.pull to be bound branch aware."""
2239
1284
        bound_location = self.get_bound_location()
2240
 
        master_branch = None
2241
 
        if bound_location and source.base != bound_location:
 
1285
        if source.base != bound_location:
2242
1286
            # not pulling from master, so we need to update master.
2243
 
            master_branch = self.get_master_branch(possible_transports)
2244
 
            master_branch.lock_write()
2245
 
        try:
2246
 
            if master_branch:
2247
 
                # pull from source into master.
2248
 
                master_branch.pull(source, overwrite, stop_revision,
2249
 
                    run_hooks=False)
2250
 
            return super(BzrBranch5, self).pull(source, overwrite,
2251
 
                stop_revision, _hook_master=master_branch,
2252
 
                run_hooks=run_hooks,
2253
 
                _override_hook_target=_override_hook_target)
2254
 
        finally:
2255
 
            if master_branch:
2256
 
                master_branch.unlock()
 
1287
            master_branch = self.get_master_branch()
 
1288
            if master_branch:
 
1289
                master_branch.pull(source)
 
1290
                source = master_branch
 
1291
        return super(BzrBranch5, self).pull(source, overwrite, stop_revision)
2257
1292
 
2258
1293
    def get_bound_location(self):
2259
1294
        try:
2260
 
            return self._transport.get_bytes('bound')[:-1]
 
1295
            return self.control_files.get_utf8('bound').read()[:-1]
2261
1296
        except errors.NoSuchFile:
2262
1297
            return None
2263
1298
 
2264
1299
    @needs_read_lock
2265
 
    def get_master_branch(self, possible_transports=None):
 
1300
    def get_master_branch(self):
2266
1301
        """Return the branch we are bound to.
2267
 
 
 
1302
        
2268
1303
        :return: Either a Branch, or None
2269
1304
 
2270
1305
        This could memoise the branch, but if thats done
2276
1311
        if not bound_loc:
2277
1312
            return None
2278
1313
        try:
2279
 
            return Branch.open(bound_loc,
2280
 
                               possible_transports=possible_transports)
 
1314
            return Branch.open(bound_loc)
2281
1315
        except (errors.NotBranchError, errors.ConnectionError), e:
2282
1316
            raise errors.BoundBranchConnectionFailure(
2283
1317
                    self, bound_loc, e)
2289
1323
        :param location: URL to the target branch
2290
1324
        """
2291
1325
        if location:
2292
 
            self._transport.put_bytes('bound', location+'\n',
2293
 
                mode=self.bzrdir._get_file_mode())
 
1326
            self.control_files.put_utf8('bound', location+'\n')
2294
1327
        else:
2295
1328
            try:
2296
 
                self._transport.delete('bound')
2297
 
            except errors.NoSuchFile:
 
1329
                self.control_files._transport.delete('bound')
 
1330
            except NoSuchFile:
2298
1331
                return False
2299
1332
            return True
2300
1333
 
2306
1339
        check for divergence to raise an error when the branches are not
2307
1340
        either the same, or one a prefix of the other. That behaviour may not
2308
1341
        be useful, so that check may be removed in future.
2309
 
 
 
1342
        
2310
1343
        :param other: The branch to bind to
2311
1344
        :type other: Branch
2312
1345
        """
2322
1355
        # last_rev is not in the other_last_rev history, AND
2323
1356
        # other_last_rev is not in our history, and do it without pulling
2324
1357
        # history around
 
1358
        last_rev = self.last_revision()
 
1359
        if last_rev is not None:
 
1360
            other.lock_read()
 
1361
            try:
 
1362
                other_last_rev = other.last_revision()
 
1363
                if other_last_rev is not None:
 
1364
                    # neither branch is new, we have to do some work to
 
1365
                    # ascertain diversion.
 
1366
                    remote_graph = other.repository.get_revision_graph(
 
1367
                        other_last_rev)
 
1368
                    local_graph = self.repository.get_revision_graph(last_rev)
 
1369
                    if (last_rev not in remote_graph and
 
1370
                        other_last_rev not in local_graph):
 
1371
                        raise errors.DivergedBranches(self, other)
 
1372
            finally:
 
1373
                other.unlock()
2325
1374
        self.set_bound_location(other.base)
2326
1375
 
2327
1376
    @needs_write_lock
2330
1379
        return self.set_bound_location(None)
2331
1380
 
2332
1381
    @needs_write_lock
2333
 
    def update(self, possible_transports=None):
2334
 
        """Synchronise this branch with the master branch if any.
 
1382
    def update(self):
 
1383
        """Synchronise this branch with the master branch if any. 
2335
1384
 
2336
1385
        :return: None or the last_revision that was pivoted out during the
2337
1386
                 update.
2338
1387
        """
2339
 
        master = self.get_master_branch(possible_transports)
 
1388
        master = self.get_master_branch()
2340
1389
        if master is not None:
2341
 
            old_tip = _mod_revision.ensure_null(self.last_revision())
 
1390
            old_tip = self.last_revision()
2342
1391
            self.pull(master, overwrite=True)
2343
 
            if self.repository.get_graph().is_ancestor(old_tip,
2344
 
                _mod_revision.ensure_null(self.last_revision())):
 
1392
            if old_tip in self.repository.get_ancestry(self.last_revision()):
2345
1393
                return None
2346
1394
            return old_tip
2347
1395
        return None
2348
1396
 
2349
1397
 
2350
 
class BzrBranch7(BzrBranch5):
2351
 
    """A branch with support for a fallback repository."""
2352
 
 
2353
 
    def _open_hook(self):
2354
 
        if self._ignore_fallbacks:
2355
 
            return
2356
 
        try:
2357
 
            url = self.get_stacked_on_url()
2358
 
        except (errors.UnstackableRepositoryFormat, errors.NotStacked,
2359
 
            errors.UnstackableBranchFormat):
2360
 
            pass
2361
 
        else:
2362
 
            for hook in Branch.hooks['transform_fallback_location']:
2363
 
                url = hook(self, url)
2364
 
                if url is None:
2365
 
                    hook_name = Branch.hooks.get_hook_name(hook)
2366
 
                    raise AssertionError(
2367
 
                        "'transform_fallback_location' hook %s returned "
2368
 
                        "None, not a URL." % hook_name)
2369
 
            self._activate_fallback_location(url)
2370
 
 
2371
 
    def __init__(self, *args, **kwargs):
2372
 
        self._ignore_fallbacks = kwargs.get('ignore_fallbacks', False)
2373
 
        super(BzrBranch7, self).__init__(*args, **kwargs)
2374
 
        self._last_revision_info_cache = None
2375
 
        self._partial_revision_history_cache = []
2376
 
 
2377
 
    def _clear_cached_state(self):
2378
 
        super(BzrBranch7, self)._clear_cached_state()
2379
 
        self._last_revision_info_cache = None
2380
 
        self._partial_revision_history_cache = []
2381
 
 
2382
 
    def _last_revision_info(self):
2383
 
        revision_string = self._transport.get_bytes('last-revision')
2384
 
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
2385
 
        revision_id = cache_utf8.get_cached_utf8(revision_id)
2386
 
        revno = int(revno)
2387
 
        return revno, revision_id
2388
 
 
2389
 
    def _write_last_revision_info(self, revno, revision_id):
2390
 
        """Simply write out the revision id, with no checks.
2391
 
 
2392
 
        Use set_last_revision_info to perform this safely.
2393
 
 
2394
 
        Does not update the revision_history cache.
2395
 
        Intended to be called by set_last_revision_info and
2396
 
        _write_revision_history.
2397
 
        """
2398
 
        revision_id = _mod_revision.ensure_null(revision_id)
2399
 
        out_string = '%d %s\n' % (revno, revision_id)
2400
 
        self._transport.put_bytes('last-revision', out_string,
2401
 
            mode=self.bzrdir._get_file_mode())
2402
 
 
2403
 
    @needs_write_lock
2404
 
    def set_last_revision_info(self, revno, revision_id):
2405
 
        revision_id = _mod_revision.ensure_null(revision_id)
2406
 
        old_revno, old_revid = self.last_revision_info()
2407
 
        if self._get_append_revisions_only():
2408
 
            self._check_history_violation(revision_id)
2409
 
        self._run_pre_change_branch_tip_hooks(revno, revision_id)
2410
 
        self._write_last_revision_info(revno, revision_id)
2411
 
        self._clear_cached_state()
2412
 
        self._last_revision_info_cache = revno, revision_id
2413
 
        self._run_post_change_branch_tip_hooks(old_revno, old_revid)
2414
 
 
2415
 
    def _synchronize_history(self, destination, revision_id):
2416
 
        """Synchronize last revision and revision history between branches.
2417
 
 
2418
 
        :see: Branch._synchronize_history
2419
 
        """
2420
 
        # XXX: The base Branch has a fast implementation of this method based
2421
 
        # on set_last_revision_info, but BzrBranch/BzrBranch5 have a slower one
2422
 
        # that uses set_revision_history.  This class inherits from BzrBranch5,
2423
 
        # but wants the fast implementation, so it calls
2424
 
        # Branch._synchronize_history directly.
2425
 
        Branch._synchronize_history(self, destination, revision_id)
2426
 
 
2427
 
    def _check_history_violation(self, revision_id):
2428
 
        last_revision = _mod_revision.ensure_null(self.last_revision())
2429
 
        if _mod_revision.is_null(last_revision):
2430
 
            return
2431
 
        if last_revision not in self._lefthand_history(revision_id):
2432
 
            raise errors.AppendRevisionsOnlyViolation(self.base)
2433
 
 
2434
 
    def _gen_revision_history(self):
2435
 
        """Generate the revision history from last revision
2436
 
        """
2437
 
        last_revno, last_revision = self.last_revision_info()
2438
 
        self._extend_partial_history(stop_index=last_revno-1)
2439
 
        return list(reversed(self._partial_revision_history_cache))
2440
 
 
2441
 
    def _extend_partial_history(self, stop_index=None, stop_revision=None):
2442
 
        """Extend the partial history to include a given index
2443
 
 
2444
 
        If a stop_index is supplied, stop when that index has been reached.
2445
 
        If a stop_revision is supplied, stop when that revision is
2446
 
        encountered.  Otherwise, stop when the beginning of history is
2447
 
        reached.
2448
 
 
2449
 
        :param stop_index: The index which should be present.  When it is
2450
 
            present, history extension will stop.
2451
 
        :param revision_id: The revision id which should be present.  When
2452
 
            it is encountered, history extension will stop.
2453
 
        """
2454
 
        repo = self.repository
2455
 
        if len(self._partial_revision_history_cache) == 0:
2456
 
            iterator = repo.iter_reverse_revision_history(self.last_revision())
2457
 
        else:
2458
 
            start_revision = self._partial_revision_history_cache[-1]
2459
 
            iterator = repo.iter_reverse_revision_history(start_revision)
2460
 
            #skip the last revision in the list
2461
 
            next_revision = iterator.next()
2462
 
        for revision_id in iterator:
2463
 
            self._partial_revision_history_cache.append(revision_id)
2464
 
            if (stop_index is not None and
2465
 
                len(self._partial_revision_history_cache) > stop_index):
2466
 
                break
2467
 
            if revision_id == stop_revision:
2468
 
                break
2469
 
 
2470
 
    def _write_revision_history(self, history):
2471
 
        """Factored out of set_revision_history.
2472
 
 
2473
 
        This performs the actual writing to disk, with format-specific checks.
2474
 
        It is intended to be called by BzrBranch5.set_revision_history.
2475
 
        """
2476
 
        if len(history) == 0:
2477
 
            last_revision = 'null:'
2478
 
        else:
2479
 
            if history != self._lefthand_history(history[-1]):
2480
 
                raise errors.NotLefthandHistory(history)
2481
 
            last_revision = history[-1]
2482
 
        if self._get_append_revisions_only():
2483
 
            self._check_history_violation(last_revision)
2484
 
        self._write_last_revision_info(len(history), last_revision)
2485
 
 
2486
 
    @needs_write_lock
2487
 
    def _set_parent_location(self, url):
2488
 
        """Set the parent branch"""
2489
 
        self._set_config_location('parent_location', url, make_relative=True)
2490
 
 
2491
 
    @needs_read_lock
2492
 
    def _get_parent_location(self):
2493
 
        """Set the parent branch"""
2494
 
        return self._get_config_location('parent_location')
2495
 
 
2496
 
    def set_push_location(self, location):
2497
 
        """See Branch.set_push_location."""
2498
 
        self._set_config_location('push_location', location)
2499
 
 
2500
 
    def set_bound_location(self, location):
2501
 
        """See Branch.set_push_location."""
2502
 
        result = None
2503
 
        config = self.get_config()
2504
 
        if location is None:
2505
 
            if config.get_user_option('bound') != 'True':
2506
 
                return False
2507
 
            else:
2508
 
                config.set_user_option('bound', 'False', warn_masked=True)
2509
 
                return True
2510
 
        else:
2511
 
            self._set_config_location('bound_location', location,
2512
 
                                      config=config)
2513
 
            config.set_user_option('bound', 'True', warn_masked=True)
2514
 
        return True
2515
 
 
2516
 
    def _get_bound_location(self, bound):
2517
 
        """Return the bound location in the config file.
2518
 
 
2519
 
        Return None if the bound parameter does not match"""
2520
 
        config = self.get_config()
2521
 
        config_bound = (config.get_user_option('bound') == 'True')
2522
 
        if config_bound != bound:
2523
 
            return None
2524
 
        return self._get_config_location('bound_location', config=config)
2525
 
 
2526
 
    def get_bound_location(self):
2527
 
        """See Branch.set_push_location."""
2528
 
        return self._get_bound_location(True)
2529
 
 
2530
 
    def get_old_bound_location(self):
2531
 
        """See Branch.get_old_bound_location"""
2532
 
        return self._get_bound_location(False)
2533
 
 
2534
 
    def get_stacked_on_url(self):
2535
 
        # you can always ask for the URL; but you might not be able to use it
2536
 
        # if the repo can't support stacking.
2537
 
        ## self._check_stackable_repo()
2538
 
        stacked_url = self._get_config_location('stacked_on_location')
2539
 
        if stacked_url is None:
2540
 
            raise errors.NotStacked(self)
2541
 
        return stacked_url
2542
 
 
2543
 
    def set_append_revisions_only(self, enabled):
2544
 
        if enabled:
2545
 
            value = 'True'
2546
 
        else:
2547
 
            value = 'False'
2548
 
        self.get_config().set_user_option('append_revisions_only', value,
2549
 
            warn_masked=True)
2550
 
 
2551
 
    def _get_append_revisions_only(self):
2552
 
        value = self.get_config().get_user_option('append_revisions_only')
2553
 
        return value == 'True'
2554
 
 
2555
 
    @needs_write_lock
2556
 
    def generate_revision_history(self, revision_id, last_rev=None,
2557
 
                                  other_branch=None):
2558
 
        """See BzrBranch5.generate_revision_history"""
2559
 
        history = self._lefthand_history(revision_id, last_rev, other_branch)
2560
 
        revno = len(history)
2561
 
        self.set_last_revision_info(revno, revision_id)
2562
 
 
2563
 
    @needs_read_lock
2564
 
    def get_rev_id(self, revno, history=None):
2565
 
        """Find the revision id of the specified revno."""
2566
 
        if revno == 0:
2567
 
            return _mod_revision.NULL_REVISION
2568
 
 
2569
 
        last_revno, last_revision_id = self.last_revision_info()
2570
 
        if revno <= 0 or revno > last_revno:
2571
 
            raise errors.NoSuchRevision(self, revno)
2572
 
 
2573
 
        if history is not None:
2574
 
            return history[revno - 1]
2575
 
 
2576
 
        index = last_revno - revno
2577
 
        if len(self._partial_revision_history_cache) <= index:
2578
 
            self._extend_partial_history(stop_index=index)
2579
 
        if len(self._partial_revision_history_cache) > index:
2580
 
            return self._partial_revision_history_cache[index]
2581
 
        else:
2582
 
            raise errors.NoSuchRevision(self, revno)
2583
 
 
2584
 
    @needs_read_lock
2585
 
    def revision_id_to_revno(self, revision_id):
2586
 
        """Given a revision id, return its revno"""
2587
 
        if _mod_revision.is_null(revision_id):
2588
 
            return 0
2589
 
        try:
2590
 
            index = self._partial_revision_history_cache.index(revision_id)
2591
 
        except ValueError:
2592
 
            self._extend_partial_history(stop_revision=revision_id)
2593
 
            index = len(self._partial_revision_history_cache) - 1
2594
 
            if self._partial_revision_history_cache[index] != revision_id:
2595
 
                raise errors.NoSuchRevision(self, revision_id)
2596
 
        return self.revno() - index
2597
 
 
2598
 
 
2599
 
class BzrBranch6(BzrBranch7):
2600
 
    """See BzrBranchFormat6 for the capabilities of this branch.
2601
 
 
2602
 
    This subclass of BzrBranch7 disables the new features BzrBranch7 added,
2603
 
    i.e. stacking.
2604
 
    """
2605
 
 
2606
 
    def get_stacked_on_url(self):
2607
 
        raise errors.UnstackableBranchFormat(self._format, self.base)
2608
 
 
2609
 
 
2610
 
######################################################################
2611
 
# results of operations
2612
 
 
2613
 
 
2614
 
class _Result(object):
2615
 
 
2616
 
    def _show_tag_conficts(self, to_file):
2617
 
        if not getattr(self, 'tag_conflicts', None):
2618
 
            return
2619
 
        to_file.write('Conflicting tags:\n')
2620
 
        for name, value1, value2 in self.tag_conflicts:
2621
 
            to_file.write('    %s\n' % (name, ))
2622
 
 
2623
 
 
2624
 
class PullResult(_Result):
2625
 
    """Result of a Branch.pull operation.
2626
 
 
2627
 
    :ivar old_revno: Revision number before pull.
2628
 
    :ivar new_revno: Revision number after pull.
2629
 
    :ivar old_revid: Tip revision id before pull.
2630
 
    :ivar new_revid: Tip revision id after pull.
2631
 
    :ivar source_branch: Source (local) branch object. (read locked)
2632
 
    :ivar master_branch: Master branch of the target, or the target if no
2633
 
        Master
2634
 
    :ivar local_branch: target branch if there is a Master, else None
2635
 
    :ivar target_branch: Target/destination branch object. (write locked)
2636
 
    :ivar tag_conflicts: A list of tag conflicts, see BasicTags.merge_to
2637
 
    """
2638
 
 
2639
 
    def __int__(self):
2640
 
        # DEPRECATED: pull used to return the change in revno
2641
 
        return self.new_revno - self.old_revno
2642
 
 
2643
 
    def report(self, to_file):
2644
 
        if not is_quiet():
2645
 
            if self.old_revid == self.new_revid:
2646
 
                to_file.write('No revisions to pull.\n')
2647
 
            else:
2648
 
                to_file.write('Now on revision %d.\n' % self.new_revno)
2649
 
        self._show_tag_conficts(to_file)
2650
 
 
2651
 
 
2652
 
class BranchPushResult(_Result):
2653
 
    """Result of a Branch.push operation.
2654
 
 
2655
 
    :ivar old_revno: Revision number (eg 10) of the target before push.
2656
 
    :ivar new_revno: Revision number (eg 12) of the target after push.
2657
 
    :ivar old_revid: Tip revision id (eg joe@foo.com-1234234-aoeua34) of target
2658
 
        before the push.
2659
 
    :ivar new_revid: Tip revision id (eg joe@foo.com-5676566-boa234a) of target
2660
 
        after the push.
2661
 
    :ivar source_branch: Source branch object that the push was from. This is
2662
 
        read locked, and generally is a local (and thus low latency) branch.
2663
 
    :ivar master_branch: If target is a bound branch, the master branch of
2664
 
        target, or target itself. Always write locked.
2665
 
    :ivar target_branch: The direct Branch where data is being sent (write
2666
 
        locked).
2667
 
    :ivar local_branch: If the target is a bound branch this will be the
2668
 
        target, otherwise it will be None.
2669
 
    """
2670
 
 
2671
 
    def __int__(self):
2672
 
        # DEPRECATED: push used to return the change in revno
2673
 
        return self.new_revno - self.old_revno
2674
 
 
2675
 
    def report(self, to_file):
2676
 
        """Write a human-readable description of the result."""
2677
 
        if self.old_revid == self.new_revid:
2678
 
            note('No new revisions to push.')
2679
 
        else:
2680
 
            note('Pushed up to revision %d.' % self.new_revno)
2681
 
        self._show_tag_conficts(to_file)
 
1398
class BranchTestProviderAdapter(object):
 
1399
    """A tool to generate a suite testing multiple branch formats at once.
 
1400
 
 
1401
    This is done by copying the test once for each transport and injecting
 
1402
    the transport_server, transport_readonly_server, and branch_format
 
1403
    classes into each copy. Each copy is also given a new id() to make it
 
1404
    easy to identify.
 
1405
    """
 
1406
 
 
1407
    def __init__(self, transport_server, transport_readonly_server, formats):
 
1408
        self._transport_server = transport_server
 
1409
        self._transport_readonly_server = transport_readonly_server
 
1410
        self._formats = formats
 
1411
    
 
1412
    def adapt(self, test):
 
1413
        result = TestSuite()
 
1414
        for branch_format, bzrdir_format in self._formats:
 
1415
            new_test = deepcopy(test)
 
1416
            new_test.transport_server = self._transport_server
 
1417
            new_test.transport_readonly_server = self._transport_readonly_server
 
1418
            new_test.bzrdir_format = bzrdir_format
 
1419
            new_test.branch_format = branch_format
 
1420
            def make_new_test_id():
 
1421
                new_id = "%s(%s)" % (new_test.id(), branch_format.__class__.__name__)
 
1422
                return lambda: new_id
 
1423
            new_test.id = make_new_test_id()
 
1424
            result.addTest(new_test)
 
1425
        return result
2682
1426
 
2683
1427
 
2684
1428
class BranchCheckResult(object):
2692
1436
 
2693
1437
    def report_results(self, verbose):
2694
1438
        """Report the check results via trace.note.
2695
 
 
 
1439
        
2696
1440
        :param verbose: Requests more detailed display of what was checked,
2697
1441
            if any.
2698
1442
        """
2701
1445
             self.branch._format)
2702
1446
 
2703
1447
 
2704
 
class Converter5to6(object):
2705
 
    """Perform an in-place upgrade of format 5 to format 6"""
2706
 
 
2707
 
    def convert(self, branch):
2708
 
        # Data for 5 and 6 can peacefully coexist.
2709
 
        format = BzrBranchFormat6()
2710
 
        new_branch = format.open(branch.bzrdir, _found=True)
2711
 
 
2712
 
        # Copy source data into target
2713
 
        new_branch._write_last_revision_info(*branch.last_revision_info())
2714
 
        new_branch.set_parent(branch.get_parent())
2715
 
        new_branch.set_bound_location(branch.get_bound_location())
2716
 
        new_branch.set_push_location(branch.get_push_location())
2717
 
 
2718
 
        # New branch has no tags by default
2719
 
        new_branch.tags._set_tag_dict({})
2720
 
 
2721
 
        # Copying done; now update target format
2722
 
        new_branch._transport.put_bytes('format',
2723
 
            format.get_format_string(),
2724
 
            mode=new_branch.bzrdir._get_file_mode())
2725
 
 
2726
 
        # Clean up old files
2727
 
        new_branch._transport.delete('revision-history')
2728
 
        try:
2729
 
            branch.set_parent(None)
2730
 
        except errors.NoSuchFile:
2731
 
            pass
2732
 
        branch.set_bound_location(None)
2733
 
 
2734
 
 
2735
 
class Converter6to7(object):
2736
 
    """Perform an in-place upgrade of format 6 to format 7"""
2737
 
 
2738
 
    def convert(self, branch):
2739
 
        format = BzrBranchFormat7()
2740
 
        branch._set_config_location('stacked_on_location', '')
2741
 
        # update target format
2742
 
        branch._transport.put_bytes('format', format.get_format_string())
2743
 
 
2744
 
 
2745
 
 
2746
 
def _run_with_write_locked_target(target, callable, *args, **kwargs):
2747
 
    """Run ``callable(*args, **kwargs)``, write-locking target for the
2748
 
    duration.
2749
 
 
2750
 
    _run_with_write_locked_target will attempt to release the lock it acquires.
2751
 
 
2752
 
    If an exception is raised by callable, then that exception *will* be
2753
 
    propagated, even if the unlock attempt raises its own error.  Thus
2754
 
    _run_with_write_locked_target should be preferred to simply doing::
2755
 
 
2756
 
        target.lock_write()
2757
 
        try:
2758
 
            return callable(*args, **kwargs)
2759
 
        finally:
2760
 
            target.unlock()
2761
 
 
2762
 
    """
2763
 
    # This is very similar to bzrlib.decorators.needs_write_lock.  Perhaps they
2764
 
    # should share code?
2765
 
    target.lock_write()
2766
 
    try:
2767
 
        result = callable(*args, **kwargs)
2768
 
    except:
2769
 
        exc_info = sys.exc_info()
2770
 
        try:
2771
 
            target.unlock()
2772
 
        finally:
2773
 
            raise exc_info[0], exc_info[1], exc_info[2]
2774
 
    else:
2775
 
        target.unlock()
2776
 
        return result
2777
 
 
2778
 
 
2779
 
class InterBranch(InterObject):
2780
 
    """This class represents operations taking place between two branches.
2781
 
 
2782
 
    Its instances have methods like pull() and push() and contain
2783
 
    references to the source and target repositories these operations
2784
 
    can be carried out on.
2785
 
    """
2786
 
 
2787
 
    _optimisers = []
2788
 
    """The available optimised InterBranch types."""
2789
 
 
2790
 
    @staticmethod
2791
 
    def _get_branch_formats_to_test():
2792
 
        """Return a tuple with the Branch formats to use when testing."""
2793
 
        raise NotImplementedError(self._get_branch_formats_to_test)
2794
 
 
2795
 
    def update_revisions(self, stop_revision=None, overwrite=False,
2796
 
                         graph=None):
2797
 
        """Pull in new perfect-fit revisions.
2798
 
 
2799
 
        :param stop_revision: Updated until the given revision
2800
 
        :param overwrite: Always set the branch pointer, rather than checking
2801
 
            to see if it is a proper descendant.
2802
 
        :param graph: A Graph object that can be used to query history
2803
 
            information. This can be None.
2804
 
        :return: None
2805
 
        """
2806
 
        raise NotImplementedError(self.update_revisions)
2807
 
 
2808
 
 
2809
 
class GenericInterBranch(InterBranch):
2810
 
    """InterBranch implementation that uses public Branch functions.
2811
 
    """
2812
 
 
2813
 
    @staticmethod
2814
 
    def _get_branch_formats_to_test():
2815
 
        return BranchFormat._default_format, BranchFormat._default_format
2816
 
 
2817
 
    def update_revisions(self, stop_revision=None, overwrite=False,
2818
 
        graph=None):
2819
 
        """See InterBranch.update_revisions()."""
2820
 
        self.source.lock_read()
2821
 
        try:
2822
 
            other_revno, other_last_revision = self.source.last_revision_info()
2823
 
            stop_revno = None # unknown
2824
 
            if stop_revision is None:
2825
 
                stop_revision = other_last_revision
2826
 
                if _mod_revision.is_null(stop_revision):
2827
 
                    # if there are no commits, we're done.
2828
 
                    return
2829
 
                stop_revno = other_revno
2830
 
 
2831
 
            # what's the current last revision, before we fetch [and change it
2832
 
            # possibly]
2833
 
            last_rev = _mod_revision.ensure_null(self.target.last_revision())
2834
 
            # we fetch here so that we don't process data twice in the common
2835
 
            # case of having something to pull, and so that the check for
2836
 
            # already merged can operate on the just fetched graph, which will
2837
 
            # be cached in memory.
2838
 
            self.target.fetch(self.source, stop_revision)
2839
 
            # Check to see if one is an ancestor of the other
2840
 
            if not overwrite:
2841
 
                if graph is None:
2842
 
                    graph = self.target.repository.get_graph()
2843
 
                if self.target._check_if_descendant_or_diverged(
2844
 
                        stop_revision, last_rev, graph, self.source):
2845
 
                    # stop_revision is a descendant of last_rev, but we aren't
2846
 
                    # overwriting, so we're done.
2847
 
                    return
2848
 
            if stop_revno is None:
2849
 
                if graph is None:
2850
 
                    graph = self.target.repository.get_graph()
2851
 
                this_revno, this_last_revision = \
2852
 
                        self.target.last_revision_info()
2853
 
                stop_revno = graph.find_distance_to_null(stop_revision,
2854
 
                                [(other_last_revision, other_revno),
2855
 
                                 (this_last_revision, this_revno)])
2856
 
            self.target.set_last_revision_info(stop_revno, stop_revision)
2857
 
        finally:
2858
 
            self.source.unlock()
2859
 
 
2860
 
    @classmethod
2861
 
    def is_compatible(self, source, target):
2862
 
        # GenericBranch uses the public API, so always compatible
2863
 
        return True
2864
 
 
2865
 
 
2866
 
InterBranch.register_optimiser(GenericInterBranch)
 
1448
######################################################################
 
1449
# predicates
 
1450
 
 
1451
 
 
1452
@deprecated_function(zero_eight)
 
1453
def is_control_file(*args, **kwargs):
 
1454
    """See bzrlib.workingtree.is_control_file."""
 
1455
    return bzrlib.workingtree.is_control_file(*args, **kwargs)