/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

Merge from bzr.dev.  Breaks a few tests because there are new methods not yet implemented.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
 
 
18
from cStringIO import StringIO
 
19
 
 
20
from bzrlib.lazy_import import lazy_import
 
21
lazy_import(globals(), """
 
22
from copy import deepcopy
 
23
from unittest import TestSuite
 
24
from warnings import warn
 
25
 
 
26
import bzrlib
 
27
from bzrlib import (
 
28
        bzrdir,
 
29
        cache_utf8,
 
30
        config as _mod_config,
 
31
        errors,
 
32
        lockdir,
 
33
        lockable_files,
 
34
        osutils,
 
35
        revision as _mod_revision,
 
36
        transport,
 
37
        tree,
 
38
        ui,
 
39
        urlutils,
 
40
        )
 
41
from bzrlib.config import BranchConfig, TreeConfig
 
42
from bzrlib.lockable_files import LockableFiles, TransportLock
 
43
""")
 
44
 
 
45
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
46
from bzrlib.errors import (BzrError, BzrCheckError, DivergedBranches,
 
47
                           HistoryMissing, InvalidRevisionId,
 
48
                           InvalidRevisionNumber, LockError, NoSuchFile,
 
49
                           NoSuchRevision, NoWorkingTree, NotVersionedError,
 
50
                           NotBranchError, UninitializableFormat,
 
51
                           UnlistableStore, UnlistableBranch,
 
52
                           )
 
53
from bzrlib.symbol_versioning import (deprecated_function,
 
54
                                      deprecated_method,
 
55
                                      DEPRECATED_PARAMETER,
 
56
                                      deprecated_passed,
 
57
                                      zero_eight, zero_nine,
 
58
                                      )
 
59
from bzrlib.trace import mutter, note
 
60
 
 
61
 
 
62
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
 
63
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
 
64
BZR_BRANCH_FORMAT_6 = "Bazaar-NG branch, format 6\n"
 
65
 
 
66
 
 
67
# TODO: Maybe include checks for common corruption of newlines, etc?
 
68
 
 
69
# TODO: Some operations like log might retrieve the same revisions
 
70
# repeatedly to calculate deltas.  We could perhaps have a weakref
 
71
# cache in memory to make this faster.  In general anything can be
 
72
# cached in memory between lock and unlock operations. .. nb thats
 
73
# what the transaction identity map provides
 
74
 
 
75
 
 
76
######################################################################
 
77
# branch objects
 
78
 
 
79
class Branch(object):
 
80
    """Branch holding a history of revisions.
 
81
 
 
82
    base
 
83
        Base directory/url of the branch.
 
84
 
 
85
    hooks: An instance of BranchHooks.
 
86
    """
 
87
    # this is really an instance variable - FIXME move it there
 
88
    # - RBC 20060112
 
89
    base = None
 
90
 
 
91
    def __init__(self, *ignored, **ignored_too):
 
92
        raise NotImplementedError('The Branch class is abstract')
 
93
 
 
94
    def break_lock(self):
 
95
        """Break a lock if one is present from another instance.
 
96
 
 
97
        Uses the ui factory to ask for confirmation if the lock may be from
 
98
        an active process.
 
99
 
 
100
        This will probe the repository for its lock as well.
 
101
        """
 
102
        self.control_files.break_lock()
 
103
        self.repository.break_lock()
 
104
        master = self.get_master_branch()
 
105
        if master is not None:
 
106
            master.break_lock()
 
107
 
 
108
    @staticmethod
 
109
    @deprecated_method(zero_eight)
 
110
    def open_downlevel(base):
 
111
        """Open a branch which may be of an old format."""
 
112
        return Branch.open(base, _unsupported=True)
 
113
        
 
114
    @staticmethod
 
115
    def open(base, _unsupported=False):
 
116
        """Open the branch rooted at base.
 
117
 
 
118
        For instance, if the branch is at URL/.bzr/branch,
 
119
        Branch.open(URL) -> a Branch instance.
 
120
        """
 
121
        control = bzrdir.BzrDir.open(base, _unsupported)
 
122
        return control.open_branch(_unsupported)
 
123
 
 
124
    @staticmethod
 
125
    def open_containing(url):
 
126
        """Open an existing branch which contains url.
 
127
        
 
128
        This probes for a branch at url, and searches upwards from there.
 
129
 
 
130
        Basically we keep looking up until we find the control directory or
 
131
        run into the root.  If there isn't one, raises NotBranchError.
 
132
        If there is one and it is either an unrecognised format or an unsupported 
 
133
        format, UnknownFormatError or UnsupportedFormatError are raised.
 
134
        If there is one, it is returned, along with the unused portion of url.
 
135
        """
 
136
        control, relpath = bzrdir.BzrDir.open_containing(url)
 
137
        return control.open_branch(), relpath
 
138
 
 
139
    @staticmethod
 
140
    @deprecated_function(zero_eight)
 
141
    def initialize(base):
 
142
        """Create a new working tree and branch, rooted at 'base' (url)
 
143
 
 
144
        NOTE: This will soon be deprecated in favour of creation
 
145
        through a BzrDir.
 
146
        """
 
147
        return bzrdir.BzrDir.create_standalone_workingtree(base).branch
 
148
 
 
149
    @deprecated_function(zero_eight)
 
150
    def setup_caching(self, cache_root):
 
151
        """Subclasses that care about caching should override this, and set
 
152
        up cached stores located under cache_root.
 
153
        
 
154
        NOTE: This is unused.
 
155
        """
 
156
        pass
 
157
 
 
158
    def get_config(self):
 
159
        return BranchConfig(self)
 
160
 
 
161
    def _get_nick(self):
 
162
        return self.get_config().get_nickname()
 
163
 
 
164
    def _set_nick(self, nick):
 
165
        self.get_config().set_user_option('nickname', nick)
 
166
 
 
167
    nick = property(_get_nick, _set_nick)
 
168
 
 
169
    def is_locked(self):
 
170
        raise NotImplementedError(self.is_locked)
 
171
 
 
172
    def lock_write(self):
 
173
        raise NotImplementedError(self.lock_write)
 
174
 
 
175
    def lock_read(self):
 
176
        raise NotImplementedError(self.lock_read)
 
177
 
 
178
    def unlock(self):
 
179
        raise NotImplementedError(self.unlock)
 
180
 
 
181
    def peek_lock_mode(self):
 
182
        """Return lock mode for the Branch: 'r', 'w' or None"""
 
183
        raise NotImplementedError(self.peek_lock_mode)
 
184
 
 
185
    def get_physical_lock_status(self):
 
186
        raise NotImplementedError(self.get_physical_lock_status)
 
187
 
 
188
    def leave_lock_in_place(self):
 
189
        """Tell this branch object not to release the physical lock when this
 
190
        object is unlocked.
 
191
        
 
192
        If lock_write doesn't return a token, then this method is not supported.
 
193
        """
 
194
        self.control_files.leave_in_place()
 
195
 
 
196
    def dont_leave_lock_in_place(self):
 
197
        """Tell this branch object to release the physical lock when this
 
198
        object is unlocked, even if it didn't originally acquire it.
 
199
 
 
200
        If lock_write doesn't return a token, then this method is not supported.
 
201
        """
 
202
        self.control_files.dont_leave_in_place()
 
203
 
 
204
    def abspath(self, name):
 
205
        """Return absolute filename for something in the branch
 
206
        
 
207
        XXX: Robert Collins 20051017 what is this used for? why is it a branch
 
208
        method and not a tree method.
 
209
        """
 
210
        raise NotImplementedError(self.abspath)
 
211
 
 
212
    def bind(self, other):
 
213
        """Bind the local branch the other branch.
 
214
 
 
215
        :param other: The branch to bind to
 
216
        :type other: Branch
 
217
        """
 
218
        raise errors.UpgradeRequired(self.base)
 
219
 
 
220
    @needs_write_lock
 
221
    def fetch(self, from_branch, last_revision=None, pb=None):
 
222
        """Copy revisions from from_branch into this branch.
 
223
 
 
224
        :param from_branch: Where to copy from.
 
225
        :param last_revision: What revision to stop at (None for at the end
 
226
                              of the branch.
 
227
        :param pb: An optional progress bar to use.
 
228
 
 
229
        Returns the copied revision count and the failed revisions in a tuple:
 
230
        (copied, failures).
 
231
        """
 
232
        if self.base == from_branch.base:
 
233
            return (0, [])
 
234
        if pb is None:
 
235
            nested_pb = ui.ui_factory.nested_progress_bar()
 
236
            pb = nested_pb
 
237
        else:
 
238
            nested_pb = None
 
239
 
 
240
        from_branch.lock_read()
 
241
        try:
 
242
            if last_revision is None:
 
243
                pb.update('get source history')
 
244
                last_revision = from_branch.last_revision()
 
245
                if last_revision is None:
 
246
                    last_revision = _mod_revision.NULL_REVISION
 
247
            return self.repository.fetch(from_branch.repository,
 
248
                                         revision_id=last_revision,
 
249
                                         pb=nested_pb)
 
250
        finally:
 
251
            if nested_pb is not None:
 
252
                nested_pb.finished()
 
253
            from_branch.unlock()
 
254
 
 
255
    def get_bound_location(self):
 
256
        """Return the URL of the branch we are bound to.
 
257
 
 
258
        Older format branches cannot bind, please be sure to use a metadir
 
259
        branch.
 
260
        """
 
261
        return None
 
262
    
 
263
    def get_old_bound_location(self):
 
264
        """Return the URL of the branch we used to be bound to
 
265
        """
 
266
        raise errors.UpgradeRequired(self.base)
 
267
 
 
268
    def get_commit_builder(self, parents, config=None, timestamp=None, 
 
269
                           timezone=None, committer=None, revprops=None, 
 
270
                           revision_id=None):
 
271
        """Obtain a CommitBuilder for this branch.
 
272
        
 
273
        :param parents: Revision ids of the parents of the new revision.
 
274
        :param config: Optional configuration to use.
 
275
        :param timestamp: Optional timestamp recorded for commit.
 
276
        :param timezone: Optional timezone for timestamp.
 
277
        :param committer: Optional committer to set for commit.
 
278
        :param revprops: Optional dictionary of revision properties.
 
279
        :param revision_id: Optional revision id.
 
280
        """
 
281
 
 
282
        if config is None:
 
283
            config = self.get_config()
 
284
        
 
285
        return self.repository.get_commit_builder(self, parents, config,
 
286
            timestamp, timezone, committer, revprops, revision_id)
 
287
 
 
288
    def get_master_branch(self):
 
289
        """Return the branch we are bound to.
 
290
        
 
291
        :return: Either a Branch, or None
 
292
        """
 
293
        return None
 
294
 
 
295
    def get_revision_delta(self, revno):
 
296
        """Return the delta for one revision.
 
297
 
 
298
        The delta is relative to its mainline predecessor, or the
 
299
        empty tree for revision 1.
 
300
        """
 
301
        assert isinstance(revno, int)
 
302
        rh = self.revision_history()
 
303
        if not (1 <= revno <= len(rh)):
 
304
            raise InvalidRevisionNumber(revno)
 
305
        return self.repository.get_revision_delta(rh[revno-1])
 
306
 
 
307
    def get_root_id(self):
 
308
        """Return the id of this branches root"""
 
309
        raise NotImplementedError(self.get_root_id)
 
310
 
 
311
    def print_file(self, file, revision_id):
 
312
        """Print `file` to stdout."""
 
313
        raise NotImplementedError(self.print_file)
 
314
 
 
315
    def append_revision(self, *revision_ids):
 
316
        raise NotImplementedError(self.append_revision)
 
317
 
 
318
    def set_revision_history(self, rev_history):
 
319
        raise NotImplementedError(self.set_revision_history)
 
320
 
 
321
    def revision_history(self):
 
322
        """Return sequence of revision hashes on to this branch."""
 
323
        raise NotImplementedError(self.revision_history)
 
324
 
 
325
    def revno(self):
 
326
        """Return current revision number for this branch.
 
327
 
 
328
        That is equivalent to the number of revisions committed to
 
329
        this branch.
 
330
        """
 
331
        return len(self.revision_history())
 
332
 
 
333
    def unbind(self):
 
334
        """Older format branches cannot bind or unbind."""
 
335
        raise errors.UpgradeRequired(self.base)
 
336
 
 
337
    def set_append_revisions_only(self, enabled):
 
338
        """Older format branches are never restricted to append-only"""
 
339
        raise errors.UpgradeRequired(self.base)
 
340
 
 
341
    def last_revision(self):
 
342
        """Return last revision id, or None"""
 
343
        ph = self.revision_history()
 
344
        if ph:
 
345
            return ph[-1]
 
346
        else:
 
347
            return None
 
348
 
 
349
    def last_revision_info(self):
 
350
        """Return information about the last revision.
 
351
 
 
352
        :return: A tuple (revno, last_revision_id).
 
353
        """
 
354
        rh = self.revision_history()
 
355
        revno = len(rh)
 
356
        if revno:
 
357
            return (revno, rh[-1])
 
358
        else:
 
359
            return (0, _mod_revision.NULL_REVISION)
 
360
 
 
361
    def missing_revisions(self, other, stop_revision=None):
 
362
        """Return a list of new revisions that would perfectly fit.
 
363
        
 
364
        If self and other have not diverged, return a list of the revisions
 
365
        present in other, but missing from self.
 
366
        """
 
367
        self_history = self.revision_history()
 
368
        self_len = len(self_history)
 
369
        other_history = other.revision_history()
 
370
        other_len = len(other_history)
 
371
        common_index = min(self_len, other_len) -1
 
372
        if common_index >= 0 and \
 
373
            self_history[common_index] != other_history[common_index]:
 
374
            raise DivergedBranches(self, other)
 
375
 
 
376
        if stop_revision is None:
 
377
            stop_revision = other_len
 
378
        else:
 
379
            assert isinstance(stop_revision, int)
 
380
            if stop_revision > other_len:
 
381
                raise errors.NoSuchRevision(self, stop_revision)
 
382
        return other_history[self_len:stop_revision]
 
383
 
 
384
    def update_revisions(self, other, stop_revision=None):
 
385
        """Pull in new perfect-fit revisions.
 
386
 
 
387
        :param other: Another Branch to pull from
 
388
        :param stop_revision: Updated until the given revision
 
389
        :return: None
 
390
        """
 
391
        raise NotImplementedError(self.update_revisions)
 
392
 
 
393
    def revision_id_to_revno(self, revision_id):
 
394
        """Given a revision id, return its revno"""
 
395
        if revision_id is None:
 
396
            return 0
 
397
        revision_id = osutils.safe_revision_id(revision_id)
 
398
        history = self.revision_history()
 
399
        try:
 
400
            return history.index(revision_id) + 1
 
401
        except ValueError:
 
402
            raise bzrlib.errors.NoSuchRevision(self, revision_id)
 
403
 
 
404
    def get_rev_id(self, revno, history=None):
 
405
        """Find the revision id of the specified revno."""
 
406
        if revno == 0:
 
407
            return None
 
408
        if history is None:
 
409
            history = self.revision_history()
 
410
        if revno <= 0 or revno > len(history):
 
411
            raise bzrlib.errors.NoSuchRevision(self, revno)
 
412
        return history[revno - 1]
 
413
 
 
414
    def pull(self, source, overwrite=False, stop_revision=None):
 
415
        """Mirror source into this branch.
 
416
 
 
417
        This branch is considered to be 'local', having low latency.
 
418
        """
 
419
        raise NotImplementedError(self.pull)
 
420
 
 
421
    def push(self, target, overwrite=False, stop_revision=None):
 
422
        """Mirror this branch into target.
 
423
 
 
424
        This branch is considered to be 'local', having low latency.
 
425
        """
 
426
        raise NotImplementedError(self.push)
 
427
 
 
428
    def basis_tree(self):
 
429
        """Return `Tree` object for last revision."""
 
430
        return self.repository.revision_tree(self.last_revision())
 
431
 
 
432
    def rename_one(self, from_rel, to_rel):
 
433
        """Rename one file.
 
434
 
 
435
        This can change the directory or the filename or both.
 
436
        """
 
437
        raise NotImplementedError(self.rename_one)
 
438
 
 
439
    def move(self, from_paths, to_name):
 
440
        """Rename files.
 
441
 
 
442
        to_name must exist as a versioned directory.
 
443
 
 
444
        If to_name exists and is a directory, the files are moved into
 
445
        it, keeping their old names.  If it is a directory, 
 
446
 
 
447
        Note that to_name is only the last component of the new name;
 
448
        this doesn't change the directory.
 
449
 
 
450
        This returns a list of (from_path, to_path) pairs for each
 
451
        entry that is moved.
 
452
        """
 
453
        raise NotImplementedError(self.move)
 
454
 
 
455
    def get_parent(self):
 
456
        """Return the parent location of the branch.
 
457
 
 
458
        This is the default location for push/pull/missing.  The usual
 
459
        pattern is that the user can override it by specifying a
 
460
        location.
 
461
        """
 
462
        raise NotImplementedError(self.get_parent)
 
463
 
 
464
    def get_submit_branch(self):
 
465
        """Return the submit location of the branch.
 
466
 
 
467
        This is the default location for bundle.  The usual
 
468
        pattern is that the user can override it by specifying a
 
469
        location.
 
470
        """
 
471
        return self.get_config().get_user_option('submit_branch')
 
472
 
 
473
    def set_submit_branch(self, location):
 
474
        """Return the submit location of the branch.
 
475
 
 
476
        This is the default location for bundle.  The usual
 
477
        pattern is that the user can override it by specifying a
 
478
        location.
 
479
        """
 
480
        self.get_config().set_user_option('submit_branch', location)
 
481
 
 
482
    def get_push_location(self):
 
483
        """Return the None or the location to push this branch to."""
 
484
        push_loc = self.get_config().get_user_option('push_location')
 
485
        return push_loc
 
486
 
 
487
    def set_push_location(self, location):
 
488
        """Set a new push location for this branch."""
 
489
        raise NotImplementedError(self.set_push_location)
 
490
 
 
491
    def set_parent(self, url):
 
492
        raise NotImplementedError(self.set_parent)
 
493
 
 
494
    @needs_write_lock
 
495
    def update(self):
 
496
        """Synchronise this branch with the master branch if any. 
 
497
 
 
498
        :return: None or the last_revision pivoted out during the update.
 
499
        """
 
500
        return None
 
501
 
 
502
    def check_revno(self, revno):
 
503
        """\
 
504
        Check whether a revno corresponds to any revision.
 
505
        Zero (the NULL revision) is considered valid.
 
506
        """
 
507
        if revno != 0:
 
508
            self.check_real_revno(revno)
 
509
            
 
510
    def check_real_revno(self, revno):
 
511
        """\
 
512
        Check whether a revno corresponds to a real revision.
 
513
        Zero (the NULL revision) is considered invalid
 
514
        """
 
515
        if revno < 1 or revno > self.revno():
 
516
            raise InvalidRevisionNumber(revno)
 
517
 
 
518
    @needs_read_lock
 
519
    def clone(self, *args, **kwargs):
 
520
        """Clone this branch into to_bzrdir preserving all semantic values.
 
521
        
 
522
        revision_id: if not None, the revision history in the new branch will
 
523
                     be truncated to end with revision_id.
 
524
        """
 
525
        # for API compatibility, until 0.8 releases we provide the old api:
 
526
        # def clone(self, to_location, revision=None, basis_branch=None, to_branch_format=None):
 
527
        # after 0.8 releases, the *args and **kwargs should be changed:
 
528
        # def clone(self, to_bzrdir, revision_id=None):
 
529
        if (kwargs.get('to_location', None) or
 
530
            kwargs.get('revision', None) or
 
531
            kwargs.get('basis_branch', None) or
 
532
            (len(args) and isinstance(args[0], basestring))):
 
533
            # backwards compatibility api:
 
534
            warn("Branch.clone() has been deprecated for BzrDir.clone() from"
 
535
                 " bzrlib 0.8.", DeprecationWarning, stacklevel=3)
 
536
            # get basis_branch
 
537
            if len(args) > 2:
 
538
                basis_branch = args[2]
 
539
            else:
 
540
                basis_branch = kwargs.get('basis_branch', None)
 
541
            if basis_branch:
 
542
                basis = basis_branch.bzrdir
 
543
            else:
 
544
                basis = None
 
545
            # get revision
 
546
            if len(args) > 1:
 
547
                revision_id = args[1]
 
548
            else:
 
549
                revision_id = kwargs.get('revision', None)
 
550
            # get location
 
551
            if len(args):
 
552
                url = args[0]
 
553
            else:
 
554
                # no default to raise if not provided.
 
555
                url = kwargs.get('to_location')
 
556
            return self.bzrdir.clone(url,
 
557
                                     revision_id=revision_id,
 
558
                                     basis=basis).open_branch()
 
559
        # new cleaner api.
 
560
        # generate args by hand 
 
561
        if len(args) > 1:
 
562
            revision_id = args[1]
 
563
        else:
 
564
            revision_id = kwargs.get('revision_id', None)
 
565
        if len(args):
 
566
            to_bzrdir = args[0]
 
567
        else:
 
568
            # no default to raise if not provided.
 
569
            to_bzrdir = kwargs.get('to_bzrdir')
 
570
        result = self._format.initialize(to_bzrdir)
 
571
        self.copy_content_into(result, revision_id=revision_id)
 
572
        return  result
 
573
 
 
574
    @needs_read_lock
 
575
    def sprout(self, to_bzrdir, revision_id=None):
 
576
        """Create a new line of development from the branch, into to_bzrdir.
 
577
        
 
578
        revision_id: if not None, the revision history in the new branch will
 
579
                     be truncated to end with revision_id.
 
580
        """
 
581
        result = self._format.initialize(to_bzrdir)
 
582
        self.copy_content_into(result, revision_id=revision_id)
 
583
        result.set_parent(self.bzrdir.root_transport.base)
 
584
        return result
 
585
 
 
586
    def _synchronize_history(self, destination, revision_id):
 
587
        """Synchronize last revision and revision history between branches.
 
588
 
 
589
        This version is most efficient when the destination is also a
 
590
        BzrBranch5, but works for BzrBranch6 as long as the revision
 
591
        history is the true lefthand parent history, and all of the revisions
 
592
        are in the destination's repository.  If not, set_revision_history
 
593
        will fail.
 
594
 
 
595
        :param destination: The branch to copy the history into
 
596
        :param revision_id: The revision-id to truncate history at.  May
 
597
          be None to copy complete history.
 
598
        """
 
599
        new_history = self.revision_history()
 
600
        if revision_id is not None:
 
601
            revision_id = osutils.safe_revision_id(revision_id)
 
602
            try:
 
603
                new_history = new_history[:new_history.index(revision_id) + 1]
 
604
            except ValueError:
 
605
                rev = self.repository.get_revision(revision_id)
 
606
                new_history = rev.get_history(self.repository)[1:]
 
607
        destination.set_revision_history(new_history)
 
608
 
 
609
    @needs_read_lock
 
610
    def copy_content_into(self, destination, revision_id=None):
 
611
        """Copy the content of self into destination.
 
612
 
 
613
        revision_id: if not None, the revision history in the new branch will
 
614
                     be truncated to end with revision_id.
 
615
        """
 
616
        self._synchronize_history(destination, revision_id)
 
617
        try:
 
618
            parent = self.get_parent()
 
619
        except errors.InaccessibleParent, e:
 
620
            mutter('parent was not accessible to copy: %s', e)
 
621
        else:
 
622
            if parent:
 
623
                destination.set_parent(parent)
 
624
 
 
625
    @needs_read_lock
 
626
    def check(self):
 
627
        """Check consistency of the branch.
 
628
 
 
629
        In particular this checks that revisions given in the revision-history
 
630
        do actually match up in the revision graph, and that they're all 
 
631
        present in the repository.
 
632
        
 
633
        Callers will typically also want to check the repository.
 
634
 
 
635
        :return: A BranchCheckResult.
 
636
        """
 
637
        mainline_parent_id = None
 
638
        for revision_id in self.revision_history():
 
639
            try:
 
640
                revision = self.repository.get_revision(revision_id)
 
641
            except errors.NoSuchRevision, e:
 
642
                raise errors.BzrCheckError("mainline revision {%s} not in repository"
 
643
                            % revision_id)
 
644
            # In general the first entry on the revision history has no parents.
 
645
            # But it's not illegal for it to have parents listed; this can happen
 
646
            # in imports from Arch when the parents weren't reachable.
 
647
            if mainline_parent_id is not None:
 
648
                if mainline_parent_id not in revision.parent_ids:
 
649
                    raise errors.BzrCheckError("previous revision {%s} not listed among "
 
650
                                        "parents of {%s}"
 
651
                                        % (mainline_parent_id, revision_id))
 
652
            mainline_parent_id = revision_id
 
653
        return BranchCheckResult(self)
 
654
 
 
655
    def _get_checkout_format(self):
 
656
        """Return the most suitable metadir for a checkout of this branch.
 
657
        Weaves are used if this branch's repostory uses weaves.
 
658
        """
 
659
        if isinstance(self.bzrdir, bzrdir.BzrDirPreSplitOut):
 
660
            from bzrlib.repofmt import weaverepo
 
661
            format = bzrdir.BzrDirMetaFormat1()
 
662
            format.repository_format = weaverepo.RepositoryFormat7()
 
663
        else:
 
664
            format = self.repository.bzrdir.cloning_metadir()
 
665
            format.branch_format = self._format
 
666
        return format
 
667
 
 
668
    def create_checkout(self, to_location, revision_id=None,
 
669
                        lightweight=False):
 
670
        """Create a checkout of a branch.
 
671
        
 
672
        :param to_location: The url to produce the checkout at
 
673
        :param revision_id: The revision to check out
 
674
        :param lightweight: If True, produce a lightweight checkout, otherwise,
 
675
        produce a bound branch (heavyweight checkout)
 
676
        :return: The tree of the created checkout
 
677
        """
 
678
        t = transport.get_transport(to_location)
 
679
        try:
 
680
            t.mkdir('.')
 
681
        except errors.FileExists:
 
682
            pass
 
683
        if lightweight:
 
684
            checkout = bzrdir.BzrDirMetaFormat1().initialize_on_transport(t)
 
685
            BranchReferenceFormat().initialize(checkout, self)
 
686
        else:
 
687
            format = self._get_checkout_format()
 
688
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
 
689
                to_location, force_new_tree=False, format=format)
 
690
            checkout = checkout_branch.bzrdir
 
691
            checkout_branch.bind(self)
 
692
            # pull up to the specified revision_id to set the initial 
 
693
            # branch tip correctly, and seed it with history.
 
694
            checkout_branch.pull(self, stop_revision=revision_id)
 
695
        return checkout.create_workingtree(revision_id)
 
696
 
 
697
 
 
698
class BranchFormat(object):
 
699
    """An encapsulation of the initialization and open routines for a format.
 
700
 
 
701
    Formats provide three things:
 
702
     * An initialization routine,
 
703
     * a format string,
 
704
     * an open routine.
 
705
 
 
706
    Formats are placed in an dict by their format string for reference 
 
707
    during branch opening. Its not required that these be instances, they
 
708
    can be classes themselves with class methods - it simply depends on 
 
709
    whether state is needed for a given format or not.
 
710
 
 
711
    Once a format is deprecated, just deprecate the initialize and open
 
712
    methods on the format class. Do not deprecate the object, as the 
 
713
    object will be created every time regardless.
 
714
    """
 
715
 
 
716
    _default_format = None
 
717
    """The default format used for new branches."""
 
718
 
 
719
    _formats = {}
 
720
    """The known formats."""
 
721
 
 
722
    @classmethod
 
723
    def find_format(klass, a_bzrdir):
 
724
        """Return the format for the branch object in a_bzrdir."""
 
725
        try:
 
726
            transport = a_bzrdir.get_branch_transport(None)
 
727
            format_string = transport.get("format").read()
 
728
            return klass._formats[format_string]
 
729
        except NoSuchFile:
 
730
            raise NotBranchError(path=transport.base)
 
731
        except KeyError:
 
732
            raise errors.UnknownFormatError(format=format_string)
 
733
 
 
734
    @classmethod
 
735
    def get_default_format(klass):
 
736
        """Return the current default format."""
 
737
        return klass._default_format
 
738
 
 
739
    def get_format_string(self):
 
740
        """Return the ASCII format string that identifies this format."""
 
741
        raise NotImplementedError(self.get_format_string)
 
742
 
 
743
    def get_format_description(self):
 
744
        """Return the short format description for this format."""
 
745
        raise NotImplementedError(self.get_format_description)
 
746
 
 
747
    def get_reference(self, a_bzrdir):
 
748
        """Get the target reference of the branch in a_bzrdir.
 
749
 
 
750
        format probing must have been completed before calling
 
751
        this method - it is assumed that the format of the branch
 
752
        in a_bzrdir is correct.
 
753
 
 
754
        :param a_bzrdir: The bzrdir to get the branch data from.
 
755
        :return: None if the branch is not a reference branch.
 
756
        """
 
757
        return None
 
758
 
 
759
    def _initialize_helper(self, a_bzrdir, utf8_files, lock_type='metadir',
 
760
                           set_format=True):
 
761
        """Initialize a branch in a bzrdir, with specified files
 
762
 
 
763
        :param a_bzrdir: The bzrdir to initialize the branch in
 
764
        :param utf8_files: The files to create as a list of
 
765
            (filename, content) tuples
 
766
        :param set_format: If True, set the format with
 
767
            self.get_format_string.  (BzrBranch4 has its format set
 
768
            elsewhere)
 
769
        :return: a branch in this format
 
770
        """
 
771
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
 
772
        branch_transport = a_bzrdir.get_branch_transport(self)
 
773
        lock_map = {
 
774
            'metadir': ('lock', lockdir.LockDir),
 
775
            'branch4': ('branch-lock', lockable_files.TransportLock),
 
776
        }
 
777
        lock_name, lock_class = lock_map[lock_type]
 
778
        control_files = lockable_files.LockableFiles(branch_transport,
 
779
            lock_name, lock_class)
 
780
        control_files.create_lock()
 
781
        control_files.lock_write()
 
782
        if set_format:
 
783
            control_files.put_utf8('format', self.get_format_string())
 
784
        try:
 
785
            for file, content in utf8_files:
 
786
                control_files.put_utf8(file, content)
 
787
        finally:
 
788
            control_files.unlock()
 
789
        return self.open(a_bzrdir, _found=True)
 
790
 
 
791
    def initialize(self, a_bzrdir):
 
792
        """Create a branch of this format in a_bzrdir."""
 
793
        raise NotImplementedError(self.initialize)
 
794
 
 
795
    def is_supported(self):
 
796
        """Is this format supported?
 
797
 
 
798
        Supported formats can be initialized and opened.
 
799
        Unsupported formats may not support initialization or committing or 
 
800
        some other features depending on the reason for not being supported.
 
801
        """
 
802
        return True
 
803
 
 
804
    def open(self, a_bzrdir, _found=False):
 
805
        """Return the branch object for a_bzrdir
 
806
 
 
807
        _found is a private parameter, do not use it. It is used to indicate
 
808
               if format probing has already be done.
 
809
        """
 
810
        raise NotImplementedError(self.open)
 
811
 
 
812
    @classmethod
 
813
    def register_format(klass, format):
 
814
        klass._formats[format.get_format_string()] = format
 
815
 
 
816
    @classmethod
 
817
    def set_default_format(klass, format):
 
818
        klass._default_format = format
 
819
 
 
820
    @classmethod
 
821
    def unregister_format(klass, format):
 
822
        assert klass._formats[format.get_format_string()] is format
 
823
        del klass._formats[format.get_format_string()]
 
824
 
 
825
    def __str__(self):
 
826
        return self.get_format_string().rstrip()
 
827
 
 
828
 
 
829
class BranchHooks(dict):
 
830
    """A dictionary mapping hook name to a list of callables for branch hooks.
 
831
    
 
832
    e.g. ['set_rh'] Is the list of items to be called when the
 
833
    set_revision_history function is invoked.
 
834
    """
 
835
 
 
836
    def __init__(self):
 
837
        """Create the default hooks.
 
838
 
 
839
        These are all empty initially, because by default nothing should get
 
840
        notified.
 
841
        """
 
842
        dict.__init__(self)
 
843
        # Introduced in 0.15:
 
844
        # invoked whenever the revision history has been set
 
845
        # with set_revision_history. The api signature is
 
846
        # (branch, revision_history), and the branch will
 
847
        # be write-locked.
 
848
        self['set_rh'] = []
 
849
        # invoked after a push operation completes.
 
850
        # the api signature is
 
851
        # (source, local, master, old_revno, old_revid, new_revno, new_revid)
 
852
        # where local is the local branch or None, master is the target 
 
853
        # master branch, and the rest should be self explanatory. The source
 
854
        # is read locked and the target branches write locked. Source will
 
855
        # be the local low-latency branch.
 
856
        self['post_push'] = []
 
857
        # invoked after a pull operation completes.
 
858
        # the api signature is
 
859
        # (source, local, master, old_revno, old_revid, new_revno, new_revid)
 
860
        # where local is the local branch or None, master is the target 
 
861
        # master branch, and the rest should be self explanatory. The source
 
862
        # is read locked and the target branches write locked. The local
 
863
        # branch is the low-latency branch.
 
864
        self['post_pull'] = []
 
865
        # invoked after a commit operation completes.
 
866
        # the api signature is 
 
867
        # (local, master, old_revno, old_revid, new_revno, new_revid)
 
868
        # old_revid is NULL_REVISION for the first commit to a branch.
 
869
        self['post_commit'] = []
 
870
        # invoked after a uncommit operation completes.
 
871
        # the api signature is
 
872
        # (local, master, old_revno, old_revid, new_revno, new_revid) where
 
873
        # local is the local branch or None, master is the target branch,
 
874
        # and an empty branch recieves new_revno of 0, new_revid of None.
 
875
        self['post_uncommit'] = []
 
876
 
 
877
    def install_hook(self, hook_name, a_callable):
 
878
        """Install a_callable in to the hook hook_name.
 
879
 
 
880
        :param hook_name: A hook name. See the __init__ method of BranchHooks
 
881
            for the complete list of hooks.
 
882
        :param a_callable: The callable to be invoked when the hook triggers.
 
883
            The exact signature will depend on the hook - see the __init__ 
 
884
            method of BranchHooks for details on each hook.
 
885
        """
 
886
        try:
 
887
            self[hook_name].append(a_callable)
 
888
        except KeyError:
 
889
            raise errors.UnknownHook('branch', hook_name)
 
890
 
 
891
 
 
892
# install the default hooks into the Branch class.
 
893
Branch.hooks = BranchHooks()
 
894
 
 
895
 
 
896
class BzrBranchFormat4(BranchFormat):
 
897
    """Bzr branch format 4.
 
898
 
 
899
    This format has:
 
900
     - a revision-history file.
 
901
     - a branch-lock lock file [ to be shared with the bzrdir ]
 
902
    """
 
903
 
 
904
    def get_format_description(self):
 
905
        """See BranchFormat.get_format_description()."""
 
906
        return "Branch format 4"
 
907
 
 
908
    def initialize(self, a_bzrdir):
 
909
        """Create a branch of this format in a_bzrdir."""
 
910
        utf8_files = [('revision-history', ''),
 
911
                      ('branch-name', ''),
 
912
                      ]
 
913
        return self._initialize_helper(a_bzrdir, utf8_files,
 
914
                                       lock_type='branch4', set_format=False)
 
915
 
 
916
    def __init__(self):
 
917
        super(BzrBranchFormat4, self).__init__()
 
918
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
 
919
 
 
920
    def open(self, a_bzrdir, _found=False):
 
921
        """Return the branch object for a_bzrdir
 
922
 
 
923
        _found is a private parameter, do not use it. It is used to indicate
 
924
               if format probing has already be done.
 
925
        """
 
926
        if not _found:
 
927
            # we are being called directly and must probe.
 
928
            raise NotImplementedError
 
929
        return BzrBranch(_format=self,
 
930
                         _control_files=a_bzrdir._control_files,
 
931
                         a_bzrdir=a_bzrdir,
 
932
                         _repository=a_bzrdir.open_repository())
 
933
 
 
934
    def __str__(self):
 
935
        return "Bazaar-NG branch format 4"
 
936
 
 
937
 
 
938
class BzrBranchFormat5(BranchFormat):
 
939
    """Bzr branch format 5.
 
940
 
 
941
    This format has:
 
942
     - a revision-history file.
 
943
     - a format string
 
944
     - a lock dir guarding the branch itself
 
945
     - all of this stored in a branch/ subdirectory
 
946
     - works with shared repositories.
 
947
 
 
948
    This format is new in bzr 0.8.
 
949
    """
 
950
 
 
951
    def get_format_string(self):
 
952
        """See BranchFormat.get_format_string()."""
 
953
        return "Bazaar-NG branch format 5\n"
 
954
 
 
955
    def get_format_description(self):
 
956
        """See BranchFormat.get_format_description()."""
 
957
        return "Branch format 5"
 
958
        
 
959
    def initialize(self, a_bzrdir):
 
960
        """Create a branch of this format in a_bzrdir."""
 
961
        utf8_files = [('revision-history', ''),
 
962
                      ('branch-name', ''),
 
963
                      ]
 
964
        return self._initialize_helper(a_bzrdir, utf8_files)
 
965
 
 
966
    def __init__(self):
 
967
        super(BzrBranchFormat5, self).__init__()
 
968
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
 
969
 
 
970
    def open(self, a_bzrdir, _found=False):
 
971
        """Return the branch object for a_bzrdir
 
972
 
 
973
        _found is a private parameter, do not use it. It is used to indicate
 
974
               if format probing has already be done.
 
975
        """
 
976
        if not _found:
 
977
            format = BranchFormat.find_format(a_bzrdir)
 
978
            assert format.__class__ == self.__class__
 
979
        try:
 
980
            transport = a_bzrdir.get_branch_transport(None)
 
981
            control_files = lockable_files.LockableFiles(transport, 'lock',
 
982
                                                         lockdir.LockDir)
 
983
            return BzrBranch5(_format=self,
 
984
                              _control_files=control_files,
 
985
                              a_bzrdir=a_bzrdir,
 
986
                              _repository=a_bzrdir.find_repository())
 
987
        except NoSuchFile:
 
988
            raise NotBranchError(path=transport.base)
 
989
 
 
990
    def __str__(self):
 
991
        return "Bazaar-NG Metadir branch format 5"
 
992
 
 
993
 
 
994
class BzrBranchFormat6(BzrBranchFormat5):
 
995
    """Branch format with last-revision
 
996
 
 
997
    Unlike previous formats, this has no explicit revision history. Instead,
 
998
    this just stores the last-revision, and the left-hand history leading
 
999
    up to there is the history.
 
1000
 
 
1001
    This format was introduced in bzr 0.15
 
1002
    """
 
1003
 
 
1004
    def get_format_string(self):
 
1005
        """See BranchFormat.get_format_string()."""
 
1006
        return "Bazaar-NG branch format 6\n"
 
1007
 
 
1008
    def get_format_description(self):
 
1009
        """See BranchFormat.get_format_description()."""
 
1010
        return "Branch format 6"
 
1011
 
 
1012
    def initialize(self, a_bzrdir):
 
1013
        """Create a branch of this format in a_bzrdir."""
 
1014
        utf8_files = [('last-revision', '0 null:\n'),
 
1015
                      ('branch-name', ''),
 
1016
                      ('branch.conf', '')
 
1017
                      ]
 
1018
        return self._initialize_helper(a_bzrdir, utf8_files)
 
1019
 
 
1020
    def open(self, a_bzrdir, _found=False):
 
1021
        """Return the branch object for a_bzrdir
 
1022
 
 
1023
        _found is a private parameter, do not use it. It is used to indicate
 
1024
               if format probing has already be done.
 
1025
        """
 
1026
        if not _found:
 
1027
            format = BranchFormat.find_format(a_bzrdir)
 
1028
            assert format.__class__ == self.__class__
 
1029
        transport = a_bzrdir.get_branch_transport(None)
 
1030
        control_files = lockable_files.LockableFiles(transport, 'lock',
 
1031
                                                     lockdir.LockDir)
 
1032
        return BzrBranch6(_format=self,
 
1033
                          _control_files=control_files,
 
1034
                          a_bzrdir=a_bzrdir,
 
1035
                          _repository=a_bzrdir.find_repository())
 
1036
 
 
1037
 
 
1038
class BranchReferenceFormat(BranchFormat):
 
1039
    """Bzr branch reference format.
 
1040
 
 
1041
    Branch references are used in implementing checkouts, they
 
1042
    act as an alias to the real branch which is at some other url.
 
1043
 
 
1044
    This format has:
 
1045
     - A location file
 
1046
     - a format string
 
1047
    """
 
1048
 
 
1049
    def get_format_string(self):
 
1050
        """See BranchFormat.get_format_string()."""
 
1051
        return "Bazaar-NG Branch Reference Format 1\n"
 
1052
 
 
1053
    def get_format_description(self):
 
1054
        """See BranchFormat.get_format_description()."""
 
1055
        return "Checkout reference format 1"
 
1056
        
 
1057
    def get_reference(self, a_bzrdir):
 
1058
        """See BranchFormat.get_reference()."""
 
1059
        transport = a_bzrdir.get_branch_transport(None)
 
1060
        return transport.get('location').read()
 
1061
 
 
1062
    def initialize(self, a_bzrdir, target_branch=None):
 
1063
        """Create a branch of this format in a_bzrdir."""
 
1064
        if target_branch is None:
 
1065
            # this format does not implement branch itself, thus the implicit
 
1066
            # creation contract must see it as uninitializable
 
1067
            raise errors.UninitializableFormat(self)
 
1068
        mutter('creating branch reference in %s', a_bzrdir.transport.base)
 
1069
        branch_transport = a_bzrdir.get_branch_transport(self)
 
1070
        branch_transport.put_bytes('location',
 
1071
            target_branch.bzrdir.root_transport.base)
 
1072
        branch_transport.put_bytes('format', self.get_format_string())
 
1073
        return self.open(a_bzrdir, _found=True)
 
1074
 
 
1075
    def __init__(self):
 
1076
        super(BranchReferenceFormat, self).__init__()
 
1077
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
 
1078
 
 
1079
    def _make_reference_clone_function(format, a_branch):
 
1080
        """Create a clone() routine for a branch dynamically."""
 
1081
        def clone(to_bzrdir, revision_id=None):
 
1082
            """See Branch.clone()."""
 
1083
            return format.initialize(to_bzrdir, a_branch)
 
1084
            # cannot obey revision_id limits when cloning a reference ...
 
1085
            # FIXME RBC 20060210 either nuke revision_id for clone, or
 
1086
            # emit some sort of warning/error to the caller ?!
 
1087
        return clone
 
1088
 
 
1089
    def open(self, a_bzrdir, _found=False, location=None):
 
1090
        """Return the branch that the branch reference in a_bzrdir points at.
 
1091
 
 
1092
        _found is a private parameter, do not use it. It is used to indicate
 
1093
               if format probing has already be done.
 
1094
        """
 
1095
        if not _found:
 
1096
            format = BranchFormat.find_format(a_bzrdir)
 
1097
            assert format.__class__ == self.__class__
 
1098
        if location is None:
 
1099
            location = self.get_reference(a_bzrdir)
 
1100
        real_bzrdir = bzrdir.BzrDir.open(location)
 
1101
        result = real_bzrdir.open_branch()
 
1102
        # this changes the behaviour of result.clone to create a new reference
 
1103
        # rather than a copy of the content of the branch.
 
1104
        # I did not use a proxy object because that needs much more extensive
 
1105
        # testing, and we are only changing one behaviour at the moment.
 
1106
        # If we decide to alter more behaviours - i.e. the implicit nickname
 
1107
        # then this should be refactored to introduce a tested proxy branch
 
1108
        # and a subclass of that for use in overriding clone() and ....
 
1109
        # - RBC 20060210
 
1110
        result.clone = self._make_reference_clone_function(result)
 
1111
        return result
 
1112
 
 
1113
 
 
1114
# formats which have no format string are not discoverable
 
1115
# and not independently creatable, so are not registered.
 
1116
__default_format = BzrBranchFormat5()
 
1117
BranchFormat.register_format(__default_format)
 
1118
BranchFormat.register_format(BranchReferenceFormat())
 
1119
BranchFormat.register_format(BzrBranchFormat6())
 
1120
BranchFormat.set_default_format(__default_format)
 
1121
_legacy_formats = [BzrBranchFormat4(),
 
1122
                   ]
 
1123
 
 
1124
class BzrBranch(Branch):
 
1125
    """A branch stored in the actual filesystem.
 
1126
 
 
1127
    Note that it's "local" in the context of the filesystem; it doesn't
 
1128
    really matter if it's on an nfs/smb/afs/coda/... share, as long as
 
1129
    it's writable, and can be accessed via the normal filesystem API.
 
1130
    """
 
1131
    
 
1132
    def __init__(self, transport=DEPRECATED_PARAMETER, init=DEPRECATED_PARAMETER,
 
1133
                 relax_version_check=DEPRECATED_PARAMETER, _format=None,
 
1134
                 _control_files=None, a_bzrdir=None, _repository=None):
 
1135
        """Create new branch object at a particular location.
 
1136
 
 
1137
        transport -- A Transport object, defining how to access files.
 
1138
        
 
1139
        init -- If True, create new control files in a previously
 
1140
             unversioned directory.  If False, the branch must already
 
1141
             be versioned.
 
1142
 
 
1143
        relax_version_check -- If true, the usual check for the branch
 
1144
            version is not applied.  This is intended only for
 
1145
            upgrade/recovery type use; it's not guaranteed that
 
1146
            all operations will work on old format branches.
 
1147
        """
 
1148
        if a_bzrdir is None:
 
1149
            self.bzrdir = bzrdir.BzrDir.open(transport.base)
 
1150
        else:
 
1151
            self.bzrdir = a_bzrdir
 
1152
        self._transport = self.bzrdir.transport.clone('..')
 
1153
        self._base = self._transport.base
 
1154
        self._format = _format
 
1155
        if _control_files is None:
 
1156
            raise ValueError('BzrBranch _control_files is None')
 
1157
        self.control_files = _control_files
 
1158
        if deprecated_passed(init):
 
1159
            warn("BzrBranch.__init__(..., init=XXX): The init parameter is "
 
1160
                 "deprecated as of bzr 0.8. Please use Branch.create().",
 
1161
                 DeprecationWarning,
 
1162
                 stacklevel=2)
 
1163
            if init:
 
1164
                # this is slower than before deprecation, oh well never mind.
 
1165
                # -> its deprecated.
 
1166
                self._initialize(transport.base)
 
1167
        self._check_format(_format)
 
1168
        if deprecated_passed(relax_version_check):
 
1169
            warn("BzrBranch.__init__(..., relax_version_check=XXX_: The "
 
1170
                 "relax_version_check parameter is deprecated as of bzr 0.8. "
 
1171
                 "Please use BzrDir.open_downlevel, or a BzrBranchFormat's "
 
1172
                 "open() method.",
 
1173
                 DeprecationWarning,
 
1174
                 stacklevel=2)
 
1175
            if (not relax_version_check
 
1176
                and not self._format.is_supported()):
 
1177
                raise errors.UnsupportedFormatError(format=fmt)
 
1178
        if deprecated_passed(transport):
 
1179
            warn("BzrBranch.__init__(transport=XXX...): The transport "
 
1180
                 "parameter is deprecated as of bzr 0.8. "
 
1181
                 "Please use Branch.open, or bzrdir.open_branch().",
 
1182
                 DeprecationWarning,
 
1183
                 stacklevel=2)
 
1184
        self.repository = _repository
 
1185
 
 
1186
    def __str__(self):
 
1187
        return '%s(%r)' % (self.__class__.__name__, self.base)
 
1188
 
 
1189
    __repr__ = __str__
 
1190
 
 
1191
    def _get_base(self):
 
1192
        return self._base
 
1193
 
 
1194
    base = property(_get_base, doc="The URL for the root of this branch.")
 
1195
 
 
1196
    def _finish_transaction(self):
 
1197
        """Exit the current transaction."""
 
1198
        return self.control_files._finish_transaction()
 
1199
 
 
1200
    def get_transaction(self):
 
1201
        """Return the current active transaction.
 
1202
 
 
1203
        If no transaction is active, this returns a passthrough object
 
1204
        for which all data is immediately flushed and no caching happens.
 
1205
        """
 
1206
        # this is an explicit function so that we can do tricky stuff
 
1207
        # when the storage in rev_storage is elsewhere.
 
1208
        # we probably need to hook the two 'lock a location' and 
 
1209
        # 'have a transaction' together more delicately, so that
 
1210
        # we can have two locks (branch and storage) and one transaction
 
1211
        # ... and finishing the transaction unlocks both, but unlocking
 
1212
        # does not. - RBC 20051121
 
1213
        return self.control_files.get_transaction()
 
1214
 
 
1215
    def _set_transaction(self, transaction):
 
1216
        """Set a new active transaction."""
 
1217
        return self.control_files._set_transaction(transaction)
 
1218
 
 
1219
    def abspath(self, name):
 
1220
        """See Branch.abspath."""
 
1221
        return self.control_files._transport.abspath(name)
 
1222
 
 
1223
    def _check_format(self, format):
 
1224
        """Identify the branch format if needed.
 
1225
 
 
1226
        The format is stored as a reference to the format object in
 
1227
        self._format for code that needs to check it later.
 
1228
 
 
1229
        The format parameter is either None or the branch format class
 
1230
        used to open this branch.
 
1231
 
 
1232
        FIXME: DELETE THIS METHOD when pre 0.8 support is removed.
 
1233
        """
 
1234
        if format is None:
 
1235
            format = BranchFormat.find_format(self.bzrdir)
 
1236
        self._format = format
 
1237
        mutter("got branch format %s", self._format)
 
1238
 
 
1239
    @needs_read_lock
 
1240
    def get_root_id(self):
 
1241
        """See Branch.get_root_id."""
 
1242
        tree = self.repository.revision_tree(self.last_revision())
 
1243
        return tree.inventory.root.file_id
 
1244
 
 
1245
    def is_locked(self):
 
1246
        return self.control_files.is_locked()
 
1247
 
 
1248
    def lock_write(self, tokens=None):
 
1249
        if tokens is not None:
 
1250
            branch_token, repo_token = tokens
 
1251
        else:
 
1252
            branch_token = repo_token = None
 
1253
        repo_token = self.repository.lock_write(token=repo_token)
 
1254
        try:
 
1255
            branch_token = self.control_files.lock_write(token=branch_token)
 
1256
        except:
 
1257
            self.repository.unlock()
 
1258
            raise
 
1259
        else:
 
1260
            tokens = (branch_token, repo_token)
 
1261
            assert tokens == (None, None) or None not in tokens, (
 
1262
                'Both branch and repository locks must return tokens, or else '
 
1263
                'neither must return tokens.  Got %r.' % (tokens,))
 
1264
            if tokens == (None, None):
 
1265
                return None
 
1266
            else:
 
1267
                return tokens
 
1268
 
 
1269
    def lock_read(self):
 
1270
        self.repository.lock_read()
 
1271
        try:
 
1272
            self.control_files.lock_read()
 
1273
        except:
 
1274
            self.repository.unlock()
 
1275
            raise
 
1276
 
 
1277
    def unlock(self):
 
1278
        # TODO: test for failed two phase locks. This is known broken.
 
1279
        try:
 
1280
            self.control_files.unlock()
 
1281
        finally:
 
1282
            self.repository.unlock()
 
1283
        
 
1284
    def peek_lock_mode(self):
 
1285
        if self.control_files._lock_count == 0:
 
1286
            return None
 
1287
        else:
 
1288
            return self.control_files._lock_mode
 
1289
 
 
1290
    def get_physical_lock_status(self):
 
1291
        return self.control_files.get_physical_lock_status()
 
1292
 
 
1293
    @needs_read_lock
 
1294
    def print_file(self, file, revision_id):
 
1295
        """See Branch.print_file."""
 
1296
        return self.repository.print_file(file, revision_id)
 
1297
 
 
1298
    @needs_write_lock
 
1299
    def append_revision(self, *revision_ids):
 
1300
        """See Branch.append_revision."""
 
1301
        revision_ids = [osutils.safe_revision_id(r) for r in revision_ids]
 
1302
        for revision_id in revision_ids:
 
1303
            _mod_revision.check_not_reserved_id(revision_id)
 
1304
            mutter("add {%s} to revision-history" % revision_id)
 
1305
        rev_history = self.revision_history()
 
1306
        rev_history.extend(revision_ids)
 
1307
        self.set_revision_history(rev_history)
 
1308
 
 
1309
    def _write_revision_history(self, history):
 
1310
        """Factored out of set_revision_history.
 
1311
 
 
1312
        This performs the actual writing to disk.
 
1313
        It is intended to be called by BzrBranch5.set_revision_history."""
 
1314
        self.control_files.put_bytes(
 
1315
            'revision-history', '\n'.join(history))
 
1316
 
 
1317
    @needs_write_lock
 
1318
    def set_revision_history(self, rev_history):
 
1319
        """See Branch.set_revision_history."""
 
1320
        rev_history = [osutils.safe_revision_id(r) for r in rev_history]
 
1321
        self._write_revision_history(rev_history)
 
1322
        transaction = self.get_transaction()
 
1323
        history = transaction.map.find_revision_history()
 
1324
        if history is not None:
 
1325
            # update the revision history in the identity map.
 
1326
            history[:] = list(rev_history)
 
1327
            # this call is disabled because revision_history is 
 
1328
            # not really an object yet, and the transaction is for objects.
 
1329
            # transaction.register_dirty(history)
 
1330
        else:
 
1331
            transaction.map.add_revision_history(rev_history)
 
1332
            # this call is disabled because revision_history is 
 
1333
            # not really an object yet, and the transaction is for objects.
 
1334
            # transaction.register_clean(history)
 
1335
        for hook in Branch.hooks['set_rh']:
 
1336
            hook(self, rev_history)
 
1337
 
 
1338
    @needs_write_lock
 
1339
    def set_last_revision_info(self, revno, revision_id):
 
1340
        revision_id = osutils.safe_revision_id(revision_id)
 
1341
        history = self._lefthand_history(revision_id)
 
1342
        assert len(history) == revno, '%d != %d' % (len(history), revno)
 
1343
        self.set_revision_history(history)
 
1344
 
 
1345
    def _gen_revision_history(self):
 
1346
        get_cached_utf8 = cache_utf8.get_cached_utf8
 
1347
        history = [get_cached_utf8(l.rstrip('\r\n')) for l in
 
1348
                self.control_files.get('revision-history').readlines()]
 
1349
        return history
 
1350
 
 
1351
    @needs_read_lock
 
1352
    def revision_history(self):
 
1353
        """See Branch.revision_history."""
 
1354
        transaction = self.get_transaction()
 
1355
        history = transaction.map.find_revision_history()
 
1356
        if history is not None:
 
1357
            # mutter("cache hit for revision-history in %s", self)
 
1358
            return list(history)
 
1359
        history = self._gen_revision_history()
 
1360
        transaction.map.add_revision_history(history)
 
1361
        # this call is disabled because revision_history is 
 
1362
        # not really an object yet, and the transaction is for objects.
 
1363
        # transaction.register_clean(history, precious=True)
 
1364
        return list(history)
 
1365
 
 
1366
    def _lefthand_history(self, revision_id, last_rev=None,
 
1367
                          other_branch=None):
 
1368
        # stop_revision must be a descendant of last_revision
 
1369
        stop_graph = self.repository.get_revision_graph(revision_id)
 
1370
        if last_rev is not None and last_rev not in stop_graph:
 
1371
            # our previous tip is not merged into stop_revision
 
1372
            raise errors.DivergedBranches(self, other_branch)
 
1373
        # make a new revision history from the graph
 
1374
        current_rev_id = revision_id
 
1375
        new_history = []
 
1376
        while current_rev_id not in (None, _mod_revision.NULL_REVISION):
 
1377
            new_history.append(current_rev_id)
 
1378
            current_rev_id_parents = stop_graph[current_rev_id]
 
1379
            try:
 
1380
                current_rev_id = current_rev_id_parents[0]
 
1381
            except IndexError:
 
1382
                current_rev_id = None
 
1383
        new_history.reverse()
 
1384
        return new_history
 
1385
 
 
1386
    @needs_write_lock
 
1387
    def generate_revision_history(self, revision_id, last_rev=None,
 
1388
        other_branch=None):
 
1389
        """Create a new revision history that will finish with revision_id.
 
1390
 
 
1391
        :param revision_id: the new tip to use.
 
1392
        :param last_rev: The previous last_revision. If not None, then this
 
1393
            must be a ancestory of revision_id, or DivergedBranches is raised.
 
1394
        :param other_branch: The other branch that DivergedBranches should
 
1395
            raise with respect to.
 
1396
        """
 
1397
        revision_id = osutils.safe_revision_id(revision_id)
 
1398
        self.set_revision_history(self._lefthand_history(revision_id,
 
1399
            last_rev, other_branch))
 
1400
 
 
1401
    @needs_write_lock
 
1402
    def update_revisions(self, other, stop_revision=None):
 
1403
        """See Branch.update_revisions."""
 
1404
        other.lock_read()
 
1405
        try:
 
1406
            if stop_revision is None:
 
1407
                stop_revision = other.last_revision()
 
1408
                if stop_revision is None:
 
1409
                    # if there are no commits, we're done.
 
1410
                    return
 
1411
            else:
 
1412
                stop_revision = osutils.safe_revision_id(stop_revision)
 
1413
            # whats the current last revision, before we fetch [and change it
 
1414
            # possibly]
 
1415
            last_rev = self.last_revision()
 
1416
            # we fetch here regardless of whether we need to so that we pickup
 
1417
            # filled in ghosts.
 
1418
            self.fetch(other, stop_revision)
 
1419
            my_ancestry = self.repository.get_ancestry(last_rev)
 
1420
            if stop_revision in my_ancestry:
 
1421
                # last_revision is a descendant of stop_revision
 
1422
                return
 
1423
            self.generate_revision_history(stop_revision, last_rev=last_rev,
 
1424
                other_branch=other)
 
1425
        finally:
 
1426
            other.unlock()
 
1427
 
 
1428
    def basis_tree(self):
 
1429
        """See Branch.basis_tree."""
 
1430
        return self.repository.revision_tree(self.last_revision())
 
1431
 
 
1432
    @deprecated_method(zero_eight)
 
1433
    def working_tree(self):
 
1434
        """Create a Working tree object for this branch."""
 
1435
 
 
1436
        from bzrlib.transport.local import LocalTransport
 
1437
        if (self.base.find('://') != -1 or 
 
1438
            not isinstance(self._transport, LocalTransport)):
 
1439
            raise NoWorkingTree(self.base)
 
1440
        return self.bzrdir.open_workingtree()
 
1441
 
 
1442
    @needs_write_lock
 
1443
    def pull(self, source, overwrite=False, stop_revision=None,
 
1444
        _hook_master=None, _run_hooks=True):
 
1445
        """See Branch.pull.
 
1446
 
 
1447
        :param _hook_master: Private parameter - set the branch to 
 
1448
            be supplied as the master to push hooks.
 
1449
        :param _run_hooks: Private parameter - allow disabling of
 
1450
            hooks, used when pushing to a master branch.
 
1451
        """
 
1452
        source.lock_read()
 
1453
        try:
 
1454
            old_count, old_tip = self.last_revision_info()
 
1455
            try:
 
1456
                self.update_revisions(source, stop_revision)
 
1457
            except DivergedBranches:
 
1458
                if not overwrite:
 
1459
                    raise
 
1460
            if overwrite:
 
1461
                self.set_revision_history(source.revision_history())
 
1462
            new_count, new_tip = self.last_revision_info()
 
1463
            if _run_hooks:
 
1464
                if _hook_master:
 
1465
                    _hook_local = self
 
1466
                else:
 
1467
                    _hook_master = self
 
1468
                    _hook_local = None
 
1469
                for hook in Branch.hooks['post_pull']:
 
1470
                    hook(source, _hook_local, _hook_master, old_count, old_tip,
 
1471
                        new_count, new_tip)
 
1472
            return new_count - old_count
 
1473
        finally:
 
1474
            source.unlock()
 
1475
 
 
1476
    def _get_parent_location(self):
 
1477
        _locs = ['parent', 'pull', 'x-pull']
 
1478
        for l in _locs:
 
1479
            try:
 
1480
                return self.control_files.get(l).read().strip('\n')
 
1481
            except NoSuchFile:
 
1482
                pass
 
1483
        return None
 
1484
 
 
1485
    @needs_read_lock
 
1486
    def push(self, target, overwrite=False, stop_revision=None,
 
1487
        _hook_master=None, _run_hooks=True):
 
1488
        """See Branch.push.
 
1489
        
 
1490
        :param _hook_master: Private parameter - set the branch to 
 
1491
            be supplied as the master to push hooks.
 
1492
        :param _run_hooks: Private parameter - allow disabling of
 
1493
            hooks, used when pushing to a master branch.
 
1494
        """
 
1495
        target.lock_write()
 
1496
        try:
 
1497
            old_count, old_tip = target.last_revision_info()
 
1498
            try:
 
1499
                target.update_revisions(self, stop_revision)
 
1500
            except DivergedBranches:
 
1501
                if not overwrite:
 
1502
                    raise
 
1503
            if overwrite:
 
1504
                target.set_revision_history(self.revision_history())
 
1505
            new_count, new_tip = target.last_revision_info()
 
1506
            if _run_hooks:
 
1507
                if _hook_master:
 
1508
                    _hook_local = target
 
1509
                else:
 
1510
                    _hook_master = target
 
1511
                    _hook_local = None
 
1512
                for hook in Branch.hooks['post_push']:
 
1513
                    hook(self, _hook_local, _hook_master, old_count, old_tip,
 
1514
                        new_count, new_tip)
 
1515
            return new_count - old_count
 
1516
        finally:
 
1517
            target.unlock()
 
1518
 
 
1519
    def get_parent(self):
 
1520
        """See Branch.get_parent."""
 
1521
 
 
1522
        assert self.base[-1] == '/'
 
1523
        parent = self._get_parent_location()
 
1524
        if parent is None:
 
1525
            return parent
 
1526
        # This is an old-format absolute path to a local branch
 
1527
        # turn it into a url
 
1528
        if parent.startswith('/'):
 
1529
            parent = urlutils.local_path_to_url(parent.decode('utf8'))
 
1530
        try:
 
1531
            return urlutils.join(self.base[:-1], parent)
 
1532
        except errors.InvalidURLJoin, e:
 
1533
            raise errors.InaccessibleParent(parent, self.base)
 
1534
 
 
1535
    def set_push_location(self, location):
 
1536
        """See Branch.set_push_location."""
 
1537
        self.get_config().set_user_option(
 
1538
            'push_location', location,
 
1539
            store=_mod_config.STORE_LOCATION_NORECURSE)
 
1540
 
 
1541
    @needs_write_lock
 
1542
    def set_parent(self, url):
 
1543
        """See Branch.set_parent."""
 
1544
        # TODO: Maybe delete old location files?
 
1545
        # URLs should never be unicode, even on the local fs,
 
1546
        # FIXUP this and get_parent in a future branch format bump:
 
1547
        # read and rewrite the file, and have the new format code read
 
1548
        # using .get not .get_utf8. RBC 20060125
 
1549
        if url is not None:
 
1550
            if isinstance(url, unicode):
 
1551
                try: 
 
1552
                    url = url.encode('ascii')
 
1553
                except UnicodeEncodeError:
 
1554
                    raise bzrlib.errors.InvalidURL(url,
 
1555
                        "Urls must be 7-bit ascii, "
 
1556
                        "use bzrlib.urlutils.escape")
 
1557
            url = urlutils.relative_url(self.base, url)
 
1558
        self._set_parent_location(url)
 
1559
 
 
1560
    def _set_parent_location(self, url):
 
1561
        if url is None:
 
1562
            self.control_files._transport.delete('parent')
 
1563
        else:
 
1564
            assert isinstance(url, str)
 
1565
            self.control_files.put_bytes('parent', url + '\n')
 
1566
 
 
1567
    @deprecated_function(zero_nine)
 
1568
    def tree_config(self):
 
1569
        """DEPRECATED; call get_config instead.  
 
1570
        TreeConfig has become part of BranchConfig."""
 
1571
        return TreeConfig(self)
 
1572
 
 
1573
 
 
1574
class BzrBranch5(BzrBranch):
 
1575
    """A format 5 branch. This supports new features over plan branches.
 
1576
 
 
1577
    It has support for a master_branch which is the data for bound branches.
 
1578
    """
 
1579
 
 
1580
    def __init__(self,
 
1581
                 _format,
 
1582
                 _control_files,
 
1583
                 a_bzrdir,
 
1584
                 _repository):
 
1585
        super(BzrBranch5, self).__init__(_format=_format,
 
1586
                                         _control_files=_control_files,
 
1587
                                         a_bzrdir=a_bzrdir,
 
1588
                                         _repository=_repository)
 
1589
        
 
1590
    @needs_write_lock
 
1591
    def pull(self, source, overwrite=False, stop_revision=None,
 
1592
        _run_hooks=True):
 
1593
        """Extends branch.pull to be bound branch aware.
 
1594
        
 
1595
        :param _run_hooks: Private parameter used to force hook running
 
1596
            off during bound branch double-pushing.
 
1597
        """
 
1598
        bound_location = self.get_bound_location()
 
1599
        master_branch = None
 
1600
        if bound_location and source.base != bound_location:
 
1601
            # not pulling from master, so we need to update master.
 
1602
            master_branch = self.get_master_branch()
 
1603
            master_branch.lock_write()
 
1604
        try:
 
1605
            if master_branch:
 
1606
                # pull from source into master.
 
1607
                master_branch.pull(source, overwrite, stop_revision,
 
1608
                    _run_hooks=False)
 
1609
            return super(BzrBranch5, self).pull(source, overwrite,
 
1610
                stop_revision, _hook_master=master_branch,
 
1611
                _run_hooks=_run_hooks)
 
1612
        finally:
 
1613
            if master_branch:
 
1614
                master_branch.unlock()
 
1615
 
 
1616
    @needs_read_lock
 
1617
    def push(self, target, overwrite=False, stop_revision=None):
 
1618
        """Updates branch.push to be bound branch aware."""
 
1619
        bound_location = target.get_bound_location()
 
1620
        master_branch = None
 
1621
        if bound_location and target.base != bound_location:
 
1622
            # not pushing to master, so we need to update master.
 
1623
            master_branch = target.get_master_branch()
 
1624
            master_branch.lock_write()
 
1625
        try:
 
1626
            if master_branch:
 
1627
                # push into the master from this branch.
 
1628
                super(BzrBranch5, self).push(master_branch, overwrite,
 
1629
                    stop_revision, _run_hooks=False)
 
1630
            # and push into the target branch from this. Note that we push from
 
1631
            # this branch again, because its considered the highest bandwidth
 
1632
            # repository.
 
1633
            return super(BzrBranch5, self).push(target, overwrite,
 
1634
                stop_revision, _hook_master=master_branch)
 
1635
        finally:
 
1636
            if master_branch:
 
1637
                master_branch.unlock()
 
1638
 
 
1639
    def get_bound_location(self):
 
1640
        try:
 
1641
            return self.control_files.get_utf8('bound').read()[:-1]
 
1642
        except errors.NoSuchFile:
 
1643
            return None
 
1644
 
 
1645
    @needs_read_lock
 
1646
    def get_master_branch(self):
 
1647
        """Return the branch we are bound to.
 
1648
        
 
1649
        :return: Either a Branch, or None
 
1650
 
 
1651
        This could memoise the branch, but if thats done
 
1652
        it must be revalidated on each new lock.
 
1653
        So for now we just don't memoise it.
 
1654
        # RBC 20060304 review this decision.
 
1655
        """
 
1656
        bound_loc = self.get_bound_location()
 
1657
        if not bound_loc:
 
1658
            return None
 
1659
        try:
 
1660
            return Branch.open(bound_loc)
 
1661
        except (errors.NotBranchError, errors.ConnectionError), e:
 
1662
            raise errors.BoundBranchConnectionFailure(
 
1663
                    self, bound_loc, e)
 
1664
 
 
1665
    @needs_write_lock
 
1666
    def set_bound_location(self, location):
 
1667
        """Set the target where this branch is bound to.
 
1668
 
 
1669
        :param location: URL to the target branch
 
1670
        """
 
1671
        if location:
 
1672
            self.control_files.put_utf8('bound', location+'\n')
 
1673
        else:
 
1674
            try:
 
1675
                self.control_files._transport.delete('bound')
 
1676
            except NoSuchFile:
 
1677
                return False
 
1678
            return True
 
1679
 
 
1680
    @needs_write_lock
 
1681
    def bind(self, other):
 
1682
        """Bind this branch to the branch other.
 
1683
 
 
1684
        This does not push or pull data between the branches, though it does
 
1685
        check for divergence to raise an error when the branches are not
 
1686
        either the same, or one a prefix of the other. That behaviour may not
 
1687
        be useful, so that check may be removed in future.
 
1688
        
 
1689
        :param other: The branch to bind to
 
1690
        :type other: Branch
 
1691
        """
 
1692
        # TODO: jam 20051230 Consider checking if the target is bound
 
1693
        #       It is debatable whether you should be able to bind to
 
1694
        #       a branch which is itself bound.
 
1695
        #       Committing is obviously forbidden,
 
1696
        #       but binding itself may not be.
 
1697
        #       Since we *have* to check at commit time, we don't
 
1698
        #       *need* to check here
 
1699
 
 
1700
        # we want to raise diverged if:
 
1701
        # last_rev is not in the other_last_rev history, AND
 
1702
        # other_last_rev is not in our history, and do it without pulling
 
1703
        # history around
 
1704
        last_rev = self.last_revision()
 
1705
        if last_rev is not None:
 
1706
            other.lock_read()
 
1707
            try:
 
1708
                other_last_rev = other.last_revision()
 
1709
                if other_last_rev is not None:
 
1710
                    # neither branch is new, we have to do some work to
 
1711
                    # ascertain diversion.
 
1712
                    remote_graph = other.repository.get_revision_graph(
 
1713
                        other_last_rev)
 
1714
                    local_graph = self.repository.get_revision_graph(last_rev)
 
1715
                    if (last_rev not in remote_graph and
 
1716
                        other_last_rev not in local_graph):
 
1717
                        raise errors.DivergedBranches(self, other)
 
1718
            finally:
 
1719
                other.unlock()
 
1720
        self.set_bound_location(other.base)
 
1721
 
 
1722
    @needs_write_lock
 
1723
    def unbind(self):
 
1724
        """If bound, unbind"""
 
1725
        return self.set_bound_location(None)
 
1726
 
 
1727
    @needs_write_lock
 
1728
    def update(self):
 
1729
        """Synchronise this branch with the master branch if any. 
 
1730
 
 
1731
        :return: None or the last_revision that was pivoted out during the
 
1732
                 update.
 
1733
        """
 
1734
        master = self.get_master_branch()
 
1735
        if master is not None:
 
1736
            old_tip = self.last_revision()
 
1737
            self.pull(master, overwrite=True)
 
1738
            if old_tip in self.repository.get_ancestry(self.last_revision()):
 
1739
                return None
 
1740
            return old_tip
 
1741
        return None
 
1742
 
 
1743
 
 
1744
class BzrBranch6(BzrBranch5):
 
1745
 
 
1746
    @needs_read_lock
 
1747
    def last_revision_info(self):
 
1748
        revision_string = self.control_files.get('last-revision').read()
 
1749
        revno, revision_id = revision_string.rstrip('\n').split(' ', 1)
 
1750
        revision_id = cache_utf8.get_cached_utf8(revision_id)
 
1751
        revno = int(revno)
 
1752
        return revno, revision_id
 
1753
 
 
1754
    def last_revision(self):
 
1755
        """Return last revision id, or None"""
 
1756
        revision_id = self.last_revision_info()[1]
 
1757
        if revision_id == _mod_revision.NULL_REVISION:
 
1758
            revision_id = None
 
1759
        return revision_id
 
1760
 
 
1761
    def _write_last_revision_info(self, revno, revision_id):
 
1762
        """Simply write out the revision id, with no checks.
 
1763
 
 
1764
        Use set_last_revision_info to perform this safely.
 
1765
 
 
1766
        Does not update the revision_history cache.
 
1767
        Intended to be called by set_last_revision_info and
 
1768
        _write_revision_history.
 
1769
        """
 
1770
        if revision_id is None:
 
1771
            revision_id = 'null:'
 
1772
        out_string = '%d %s\n' % (revno, revision_id)
 
1773
        self.control_files.put_bytes('last-revision', out_string)
 
1774
 
 
1775
    @needs_write_lock
 
1776
    def set_last_revision_info(self, revno, revision_id):
 
1777
        revision_id = osutils.safe_revision_id(revision_id)
 
1778
        if self._get_append_revisions_only():
 
1779
            self._check_history_violation(revision_id)
 
1780
        self._write_last_revision_info(revno, revision_id)
 
1781
        transaction = self.get_transaction()
 
1782
        cached_history = transaction.map.find_revision_history()
 
1783
        if cached_history is not None:
 
1784
            transaction.map.remove_object(cached_history)
 
1785
 
 
1786
    def _check_history_violation(self, revision_id):
 
1787
        last_revision = self.last_revision()
 
1788
        if last_revision is None:
 
1789
            return
 
1790
        if last_revision not in self._lefthand_history(revision_id):
 
1791
            raise errors.AppendRevisionsOnlyViolation(self.base)
 
1792
 
 
1793
    def _gen_revision_history(self):
 
1794
        """Generate the revision history from last revision
 
1795
        """
 
1796
        history = list(self.repository.iter_reverse_revision_history(
 
1797
            self.last_revision()))
 
1798
        history.reverse()
 
1799
        return history
 
1800
 
 
1801
    def _write_revision_history(self, history):
 
1802
        """Factored out of set_revision_history.
 
1803
 
 
1804
        This performs the actual writing to disk, with format-specific checks.
 
1805
        It is intended to be called by BzrBranch5.set_revision_history.
 
1806
        """
 
1807
        if len(history) == 0:
 
1808
            last_revision = 'null:'
 
1809
        else:
 
1810
            if history != self._lefthand_history(history[-1]):
 
1811
                raise errors.NotLefthandHistory(history)
 
1812
            last_revision = history[-1]
 
1813
        if self._get_append_revisions_only():
 
1814
            self._check_history_violation(last_revision)
 
1815
        self._write_last_revision_info(len(history), last_revision)
 
1816
 
 
1817
    @needs_write_lock
 
1818
    def append_revision(self, *revision_ids):
 
1819
        revision_ids = [osutils.safe_revision_id(r) for r in revision_ids]
 
1820
        if len(revision_ids) == 0:
 
1821
            return
 
1822
        prev_revno, prev_revision = self.last_revision_info()
 
1823
        for revision in self.repository.get_revisions(revision_ids):
 
1824
            if prev_revision == _mod_revision.NULL_REVISION:
 
1825
                if revision.parent_ids != []:
 
1826
                    raise errors.NotLeftParentDescendant(self, prev_revision,
 
1827
                                                         revision.revision_id)
 
1828
            else:
 
1829
                if revision.parent_ids[0] != prev_revision:
 
1830
                    raise errors.NotLeftParentDescendant(self, prev_revision,
 
1831
                                                         revision.revision_id)
 
1832
            prev_revision = revision.revision_id
 
1833
        self.set_last_revision_info(prev_revno + len(revision_ids),
 
1834
                                    revision_ids[-1])
 
1835
 
 
1836
    def _set_config_location(self, name, url, config=None,
 
1837
                             make_relative=False):
 
1838
        if config is None:
 
1839
            config = self.get_config()
 
1840
        if url is None:
 
1841
            url = ''
 
1842
        elif make_relative:
 
1843
            url = urlutils.relative_url(self.base, url)
 
1844
        config.set_user_option(name, url)
 
1845
 
 
1846
 
 
1847
    def _get_config_location(self, name, config=None):
 
1848
        if config is None:
 
1849
            config = self.get_config()
 
1850
        location = config.get_user_option(name)
 
1851
        if location == '':
 
1852
            location = None
 
1853
        return location
 
1854
 
 
1855
    @needs_write_lock
 
1856
    def _set_parent_location(self, url):
 
1857
        """Set the parent branch"""
 
1858
        self._set_config_location('parent_location', url, make_relative=True)
 
1859
 
 
1860
    @needs_read_lock
 
1861
    def _get_parent_location(self):
 
1862
        """Set the parent branch"""
 
1863
        return self._get_config_location('parent_location')
 
1864
 
 
1865
    def set_push_location(self, location):
 
1866
        """See Branch.set_push_location."""
 
1867
        self._set_config_location('push_location', location)
 
1868
 
 
1869
    def set_bound_location(self, location):
 
1870
        """See Branch.set_push_location."""
 
1871
        result = None
 
1872
        config = self.get_config()
 
1873
        if location is None:
 
1874
            if config.get_user_option('bound') != 'True':
 
1875
                return False
 
1876
            else:
 
1877
                config.set_user_option('bound', 'False')
 
1878
                return True
 
1879
        else:
 
1880
            self._set_config_location('bound_location', location,
 
1881
                                      config=config)
 
1882
            config.set_user_option('bound', 'True')
 
1883
        return True
 
1884
 
 
1885
    def _get_bound_location(self, bound):
 
1886
        """Return the bound location in the config file.
 
1887
 
 
1888
        Return None if the bound parameter does not match"""
 
1889
        config = self.get_config()
 
1890
        config_bound = (config.get_user_option('bound') == 'True')
 
1891
        if config_bound != bound:
 
1892
            return None
 
1893
        return self._get_config_location('bound_location', config=config)
 
1894
 
 
1895
    def get_bound_location(self):
 
1896
        """See Branch.set_push_location."""
 
1897
        return self._get_bound_location(True)
 
1898
 
 
1899
    def get_old_bound_location(self):
 
1900
        """See Branch.get_old_bound_location"""
 
1901
        return self._get_bound_location(False)
 
1902
 
 
1903
    def set_append_revisions_only(self, enabled):
 
1904
        if enabled:
 
1905
            value = 'True'
 
1906
        else:
 
1907
            value = 'False'
 
1908
        self.get_config().set_user_option('append_revisions_only', value)
 
1909
 
 
1910
    def _get_append_revisions_only(self):
 
1911
        value = self.get_config().get_user_option('append_revisions_only')
 
1912
        return value == 'True'
 
1913
 
 
1914
    def _synchronize_history(self, destination, revision_id):
 
1915
        """Synchronize last revision and revision history between branches.
 
1916
 
 
1917
        This version is most efficient when the destination is also a
 
1918
        BzrBranch6, but works for BzrBranch5, as long as the destination's
 
1919
        repository contains all the lefthand ancestors of the intended
 
1920
        last_revision.  If not, set_last_revision_info will fail.
 
1921
 
 
1922
        :param destination: The branch to copy the history into
 
1923
        :param revision_id: The revision-id to truncate history at.  May
 
1924
          be None to copy complete history.
 
1925
        """
 
1926
        if revision_id is None:
 
1927
            revno, revision_id = self.last_revision_info()
 
1928
        else:
 
1929
            revno = self.revision_id_to_revno(revision_id)
 
1930
        destination.set_last_revision_info(revno, revision_id)
 
1931
 
 
1932
 
 
1933
class BranchTestProviderAdapter(object):
 
1934
    """A tool to generate a suite testing multiple branch formats at once.
 
1935
 
 
1936
    This is done by copying the test once for each transport and injecting
 
1937
    the transport_server, transport_readonly_server, and branch_format
 
1938
    classes into each copy. Each copy is also given a new id() to make it
 
1939
    easy to identify.
 
1940
    """
 
1941
 
 
1942
    def __init__(self, transport_server, transport_readonly_server, formats,
 
1943
        vfs_transport_factory=None):
 
1944
        self._transport_server = transport_server
 
1945
        self._transport_readonly_server = transport_readonly_server
 
1946
        self._formats = formats
 
1947
        self._vfs_transport_factory = vfs_transport_factory
 
1948
    
 
1949
    def adapt(self, test):
 
1950
        result = TestSuite()
 
1951
        for branch_format, bzrdir_format in self._formats:
 
1952
            new_test = deepcopy(test)
 
1953
            new_test.transport_server = self._transport_server
 
1954
            new_test.transport_readonly_server = self._transport_readonly_server
 
1955
            if self._vfs_transport_factory:
 
1956
                new_test.vfs_transport_factory = self._vfs_transport_factory
 
1957
            new_test.bzrdir_format = bzrdir_format
 
1958
            new_test.branch_format = branch_format
 
1959
            def make_new_test_id():
 
1960
                new_id = "%s(%s)" % (new_test.id(), branch_format.__class__.__name__)
 
1961
                return lambda: new_id
 
1962
            new_test.id = make_new_test_id()
 
1963
            result.addTest(new_test)
 
1964
        return result
 
1965
 
 
1966
 
 
1967
class BranchCheckResult(object):
 
1968
    """Results of checking branch consistency.
 
1969
 
 
1970
    :see: Branch.check
 
1971
    """
 
1972
 
 
1973
    def __init__(self, branch):
 
1974
        self.branch = branch
 
1975
 
 
1976
    def report_results(self, verbose):
 
1977
        """Report the check results via trace.note.
 
1978
        
 
1979
        :param verbose: Requests more detailed display of what was checked,
 
1980
            if any.
 
1981
        """
 
1982
        note('checked branch %s format %s',
 
1983
             self.branch.base,
 
1984
             self.branch._format)
 
1985
 
 
1986
 
 
1987
######################################################################
 
1988
# predicates
 
1989
 
 
1990
 
 
1991
@deprecated_function(zero_eight)
 
1992
def is_control_file(*args, **kwargs):
 
1993
    """See bzrlib.workingtree.is_control_file."""
 
1994
    from bzrlib import workingtree
 
1995
    return workingtree.is_control_file(*args, **kwargs)
 
1996
 
 
1997
 
 
1998
class Converter5to6(object):
 
1999
    """Perform an in-place upgrade of format 5 to format 6"""
 
2000
 
 
2001
    def convert(self, branch):
 
2002
        # Data for 5 and 6 can peacefully coexist.
 
2003
        format = BzrBranchFormat6()
 
2004
        new_branch = format.open(branch.bzrdir, _found=True)
 
2005
 
 
2006
        # Copy source data into target
 
2007
        new_branch.set_last_revision_info(*branch.last_revision_info())
 
2008
        new_branch.set_parent(branch.get_parent())
 
2009
        new_branch.set_bound_location(branch.get_bound_location())
 
2010
        new_branch.set_push_location(branch.get_push_location())
 
2011
 
 
2012
        # Copying done; now update target format
 
2013
        new_branch.control_files.put_utf8('format',
 
2014
            format.get_format_string())
 
2015
 
 
2016
        # Clean up old files
 
2017
        new_branch.control_files._transport.delete('revision-history')
 
2018
        try:
 
2019
            branch.set_parent(None)
 
2020
        except NoSuchFile:
 
2021
            pass
 
2022
        branch.set_bound_location(None)