~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Robert Collins
  • Date: 2005-11-28 05:13:41 UTC
  • mfrom: (1185.33.54 merge-recovered)
  • Revision ID: robertc@robertcollins.net-20051128051341-059936f2f29a12c8
Merge from Martin. Adjust check to work with HTTP again.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 Canonical Ltd
 
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
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
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
 
18
import shutil
 
19
import sys
 
20
import os
 
21
import errno
 
22
from warnings import warn
18
23
from cStringIO import StringIO
19
24
 
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
25
 
26
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
 
                                      )
 
27
import bzrlib.inventory as inventory
59
28
from bzrlib.trace import mutter, note
 
29
from bzrlib.osutils import (isdir, quotefn,
 
30
                            rename, splitpath, sha_file, appendpath, 
 
31
                            file_kind, abspath)
 
32
import bzrlib.errors as errors
 
33
from bzrlib.errors import (BzrError, InvalidRevisionNumber, InvalidRevisionId,
 
34
                           NoSuchRevision, HistoryMissing, NotBranchError,
 
35
                           DivergedBranches, LockError, UnlistableStore,
 
36
                           UnlistableBranch, NoSuchFile, NotVersionedError,
 
37
                           NoWorkingTree)
 
38
from bzrlib.textui import show_status
 
39
from bzrlib.revision import (Revision, is_ancestor, get_intervening_revisions,
 
40
                             NULL_REVISION)
 
41
 
 
42
from bzrlib.delta import compare_trees
 
43
from bzrlib.tree import EmptyTree, RevisionTree
 
44
from bzrlib.inventory import Inventory
 
45
from bzrlib.store import copy_all
 
46
from bzrlib.store.text import TextStore
 
47
from bzrlib.store.weave import WeaveStore
 
48
from bzrlib.testament import Testament
 
49
import bzrlib.transactions as transactions
 
50
from bzrlib.transport import Transport, get_transport
 
51
import bzrlib.xml5
 
52
import bzrlib.ui
 
53
from config import TreeConfig
60
54
 
61
55
 
62
56
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
63
57
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
64
58
BZR_BRANCH_FORMAT_6 = "Bazaar-NG branch, format 6\n"
65
 
 
66
 
 
67
 
# TODO: Maybe include checks for common corruption of newlines, etc?
 
59
## TODO: Maybe include checks for common corruption of newlines, etc?
 
60
 
68
61
 
69
62
# TODO: Some operations like log might retrieve the same revisions
70
63
# repeatedly to calculate deltas.  We could perhaps have a weakref
71
64
# 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
 
 
 
65
# cached in memory between lock and unlock operations.
 
66
 
 
67
def find_branch(*ignored, **ignored_too):
 
68
    # XXX: leave this here for about one release, then remove it
 
69
    raise NotImplementedError('find_branch() is not supported anymore, '
 
70
                              'please use one of the new branch constructors')
 
71
 
 
72
 
 
73
def needs_read_lock(unbound):
 
74
    """Decorate unbound to take out and release a read lock."""
 
75
    def decorated(self, *args, **kwargs):
 
76
        self.lock_read()
 
77
        try:
 
78
            return unbound(self, *args, **kwargs)
 
79
        finally:
 
80
            self.unlock()
 
81
    return decorated
 
82
 
 
83
 
 
84
def needs_write_lock(unbound):
 
85
    """Decorate unbound to take out and release a write lock."""
 
86
    def decorated(self, *args, **kwargs):
 
87
        self.lock_write()
 
88
        try:
 
89
            return unbound(self, *args, **kwargs)
 
90
        finally:
 
91
            self.unlock()
 
92
    return decorated
75
93
 
76
94
######################################################################
77
95
# branch objects
82
100
    base
83
101
        Base directory/url of the branch.
84
102
    """
85
 
    # this is really an instance variable - FIXME move it there
86
 
    # - RBC 20060112
87
103
    base = None
88
104
 
89
105
    def __init__(self, *ignored, **ignored_too):
90
106
        raise NotImplementedError('The Branch class is abstract')
91
107
 
92
 
    def break_lock(self):
93
 
        """Break a lock if one is present from another instance.
94
 
 
95
 
        Uses the ui factory to ask for confirmation if the lock may be from
96
 
        an active process.
97
 
 
98
 
        This will probe the repository for its lock as well.
99
 
        """
100
 
        self.control_files.break_lock()
101
 
        self.repository.break_lock()
102
 
        master = self.get_master_branch()
103
 
        if master is not None:
104
 
            master.break_lock()
105
 
 
106
108
    @staticmethod
107
 
    @deprecated_method(zero_eight)
108
109
    def open_downlevel(base):
109
 
        """Open a branch which may be of an old format."""
110
 
        return Branch.open(base, _unsupported=True)
 
110
        """Open a branch which may be of an old format.
 
111
        
 
112
        Only local branches are supported."""
 
113
        return BzrBranch(get_transport(base), relax_version_check=True)
111
114
        
112
115
    @staticmethod
113
 
    def open(base, _unsupported=False):
114
 
        """Open the branch rooted at base.
115
 
 
116
 
        For instance, if the branch is at URL/.bzr/branch,
117
 
        Branch.open(URL) -> a Branch instance.
118
 
        """
119
 
        control = bzrdir.BzrDir.open(base, _unsupported)
120
 
        return control.open_branch(_unsupported)
 
116
    def open(base):
 
117
        """Open an existing branch, rooted at 'base' (url)"""
 
118
        t = get_transport(base)
 
119
        mutter("trying to open %r with transport %r", base, t)
 
120
        return BzrBranch(t)
121
121
 
122
122
    @staticmethod
123
123
    def open_containing(url):
127
127
 
128
128
        Basically we keep looking up until we find the control directory or
129
129
        run into the root.  If there isn't one, raises NotBranchError.
130
 
        If there is one and it is either an unrecognised format or an unsupported 
131
 
        format, UnknownFormatError or UnsupportedFormatError are raised.
132
130
        If there is one, it is returned, along with the unused portion of url.
133
131
        """
134
 
        control, relpath = bzrdir.BzrDir.open_containing(url)
135
 
        return control.open_branch(), relpath
 
132
        t = get_transport(url)
 
133
        while True:
 
134
            try:
 
135
                return BzrBranch(t), t.relpath(url)
 
136
            except NotBranchError:
 
137
                pass
 
138
            new_t = t.clone('..')
 
139
            if new_t.base == t.base:
 
140
                # reached the root, whatever that may be
 
141
                raise NotBranchError(path=url)
 
142
            t = new_t
136
143
 
137
144
    @staticmethod
138
 
    @deprecated_function(zero_eight)
139
145
    def initialize(base):
140
 
        """Create a new working tree and branch, rooted at 'base' (url)
141
 
 
142
 
        NOTE: This will soon be deprecated in favour of creation
143
 
        through a BzrDir.
144
 
        """
145
 
        return bzrdir.BzrDir.create_standalone_workingtree(base).branch
146
 
 
147
 
    @deprecated_function(zero_eight)
 
146
        """Create a new branch, rooted at 'base' (url)"""
 
147
        t = get_transport(base)
 
148
        return BzrBranch(t, init=True)
 
149
 
148
150
    def setup_caching(self, cache_root):
149
151
        """Subclasses that care about caching should override this, and set
150
152
        up cached stores located under cache_root.
151
 
        
152
 
        NOTE: This is unused.
153
153
        """
154
 
        pass
155
 
 
156
 
    def get_config(self):
157
 
        return BranchConfig(self)
 
154
        self.cache_root = cache_root
158
155
 
159
156
    def _get_nick(self):
160
 
        return self.get_config().get_nickname()
 
157
        cfg = self.tree_config()
 
158
        return cfg.get_option(u"nickname", default=self.base.split('/')[-1])
161
159
 
162
160
    def _set_nick(self, nick):
163
 
        self.get_config().set_user_option('nickname', nick)
 
161
        cfg = self.tree_config()
 
162
        cfg.set_option(nick, "nickname")
 
163
        assert cfg.get_option("nickname") == nick
164
164
 
165
165
    nick = property(_get_nick, _set_nick)
166
 
 
167
 
    def is_locked(self):
168
 
        raise NotImplementedError(self.is_locked)
 
166
        
 
167
    def push_stores(self, branch_to):
 
168
        """Copy the content of this branches store to branch_to."""
 
169
        raise NotImplementedError('push_stores is abstract')
 
170
 
 
171
    def get_transaction(self):
 
172
        """Return the current active transaction.
 
173
 
 
174
        If no transaction is active, this returns a passthrough object
 
175
        for which all data is immediately flushed and no caching happens.
 
176
        """
 
177
        raise NotImplementedError('get_transaction is abstract')
169
178
 
170
179
    def lock_write(self):
171
 
        raise NotImplementedError(self.lock_write)
172
 
 
 
180
        raise NotImplementedError('lock_write is abstract')
 
181
        
173
182
    def lock_read(self):
174
 
        raise NotImplementedError(self.lock_read)
 
183
        raise NotImplementedError('lock_read is abstract')
175
184
 
176
185
    def unlock(self):
177
 
        raise NotImplementedError(self.unlock)
178
 
 
179
 
    def peek_lock_mode(self):
180
 
        """Return lock mode for the Branch: 'r', 'w' or None"""
181
 
        raise NotImplementedError(self.peek_lock_mode)
182
 
 
183
 
    def get_physical_lock_status(self):
184
 
        raise NotImplementedError(self.get_physical_lock_status)
 
186
        raise NotImplementedError('unlock is abstract')
185
187
 
186
188
    def abspath(self, name):
187
189
        """Return absolute filename for something in the branch
189
191
        XXX: Robert Collins 20051017 what is this used for? why is it a branch
190
192
        method and not a tree method.
191
193
        """
192
 
        raise NotImplementedError(self.abspath)
193
 
 
194
 
    def bind(self, other):
195
 
        """Bind the local branch the other branch.
196
 
 
197
 
        :param other: The branch to bind to
198
 
        :type other: Branch
199
 
        """
200
 
        raise errors.UpgradeRequired(self.base)
201
 
 
202
 
    @needs_write_lock
203
 
    def fetch(self, from_branch, last_revision=None, pb=None):
204
 
        """Copy revisions from from_branch into this branch.
205
 
 
206
 
        :param from_branch: Where to copy from.
207
 
        :param last_revision: What revision to stop at (None for at the end
208
 
                              of the branch.
209
 
        :param pb: An optional progress bar to use.
210
 
 
211
 
        Returns the copied revision count and the failed revisions in a tuple:
212
 
        (copied, failures).
213
 
        """
214
 
        if self.base == from_branch.base:
215
 
            return (0, [])
216
 
        if pb is None:
217
 
            nested_pb = ui.ui_factory.nested_progress_bar()
218
 
            pb = nested_pb
219
 
        else:
220
 
            nested_pb = None
221
 
 
222
 
        from_branch.lock_read()
223
 
        try:
224
 
            if last_revision is None:
225
 
                pb.update('get source history')
226
 
                from_history = from_branch.revision_history()
227
 
                if from_history:
228
 
                    last_revision = from_history[-1]
229
 
                else:
230
 
                    # no history in the source branch
231
 
                    last_revision = _mod_revision.NULL_REVISION
232
 
            return self.repository.fetch(from_branch.repository,
233
 
                                         revision_id=last_revision,
234
 
                                         pb=nested_pb)
235
 
        finally:
236
 
            if nested_pb is not None:
237
 
                nested_pb.finished()
238
 
            from_branch.unlock()
239
 
 
240
 
    def get_bound_location(self):
241
 
        """Return the URL of the branch we are bound to.
242
 
 
243
 
        Older format branches cannot bind, please be sure to use a metadir
244
 
        branch.
245
 
        """
246
 
        return None
247
 
    
248
 
    def get_commit_builder(self, parents, config=None, timestamp=None, 
249
 
                           timezone=None, committer=None, revprops=None, 
250
 
                           revision_id=None):
251
 
        """Obtain a CommitBuilder for this branch.
252
 
        
253
 
        :param parents: Revision ids of the parents of the new revision.
254
 
        :param config: Optional configuration to use.
255
 
        :param timestamp: Optional timestamp recorded for commit.
256
 
        :param timezone: Optional timezone for timestamp.
257
 
        :param committer: Optional committer to set for commit.
258
 
        :param revprops: Optional dictionary of revision properties.
259
 
        :param revision_id: Optional revision id.
260
 
        """
261
 
 
262
 
        if config is None:
263
 
            config = self.get_config()
264
 
        
265
 
        return self.repository.get_commit_builder(self, parents, config, 
266
 
            timestamp, timezone, committer, revprops, revision_id)
267
 
 
268
 
    def get_master_branch(self):
269
 
        """Return the branch we are bound to.
270
 
        
271
 
        :return: Either a Branch, or None
272
 
        """
273
 
        return None
 
194
        raise NotImplementedError('abspath is abstract')
 
195
 
 
196
    def controlfilename(self, file_or_path):
 
197
        """Return location relative to branch."""
 
198
        raise NotImplementedError('controlfilename is abstract')
 
199
 
 
200
    def controlfile(self, file_or_path, mode='r'):
 
201
        """Open a control file for this branch.
 
202
 
 
203
        There are two classes of file in the control directory: text
 
204
        and binary.  binary files are untranslated byte streams.  Text
 
205
        control files are stored with Unix newlines and in UTF-8, even
 
206
        if the platform or locale defaults are different.
 
207
 
 
208
        Controlfiles should almost never be opened in write mode but
 
209
        rather should be atomically copied and replaced using atomicfile.
 
210
        """
 
211
        raise NotImplementedError('controlfile is abstract')
 
212
 
 
213
    def put_controlfile(self, path, f, encode=True):
 
214
        """Write an entry as a controlfile.
 
215
 
 
216
        :param path: The path to put the file, relative to the .bzr control
 
217
                     directory
 
218
        :param f: A file-like or string object whose contents should be copied.
 
219
        :param encode:  If true, encode the contents as utf-8
 
220
        """
 
221
        raise NotImplementedError('put_controlfile is abstract')
 
222
 
 
223
    def put_controlfiles(self, files, encode=True):
 
224
        """Write several entries as controlfiles.
 
225
 
 
226
        :param files: A list of [(path, file)] pairs, where the path is the directory
 
227
                      underneath the bzr control directory
 
228
        :param encode:  If true, encode the contents as utf-8
 
229
        """
 
230
        raise NotImplementedError('put_controlfiles is abstract')
 
231
 
 
232
    def get_root_id(self):
 
233
        """Return the id of this branches root"""
 
234
        raise NotImplementedError('get_root_id is abstract')
 
235
 
 
236
    def set_root_id(self, file_id):
 
237
        raise NotImplementedError('set_root_id is abstract')
 
238
 
 
239
    def print_file(self, file, revno):
 
240
        """Print `file` to stdout."""
 
241
        raise NotImplementedError('print_file is abstract')
 
242
 
 
243
    def append_revision(self, *revision_ids):
 
244
        raise NotImplementedError('append_revision is abstract')
 
245
 
 
246
    def set_revision_history(self, rev_history):
 
247
        raise NotImplementedError('set_revision_history is abstract')
 
248
 
 
249
    def has_revision(self, revision_id):
 
250
        """True if this branch has a copy of the revision.
 
251
 
 
252
        This does not necessarily imply the revision is merge
 
253
        or on the mainline."""
 
254
        raise NotImplementedError('has_revision is abstract')
 
255
 
 
256
    def get_revision_xml(self, revision_id):
 
257
        raise NotImplementedError('get_revision_xml is abstract')
 
258
 
 
259
    def get_revision(self, revision_id):
 
260
        """Return the Revision object for a named revision"""
 
261
        raise NotImplementedError('get_revision is abstract')
274
262
 
275
263
    def get_revision_delta(self, revno):
276
264
        """Return the delta for one revision.
282
270
        rh = self.revision_history()
283
271
        if not (1 <= revno <= len(rh)):
284
272
            raise InvalidRevisionNumber(revno)
285
 
        return self.repository.get_revision_delta(rh[revno-1])
286
 
 
287
 
    def get_root_id(self):
288
 
        """Return the id of this branches root"""
289
 
        raise NotImplementedError(self.get_root_id)
290
 
 
291
 
    def print_file(self, file, revision_id):
292
 
        """Print `file` to stdout."""
293
 
        raise NotImplementedError(self.print_file)
294
 
 
295
 
    def append_revision(self, *revision_ids):
296
 
        raise NotImplementedError(self.append_revision)
297
 
 
298
 
    def set_revision_history(self, rev_history):
299
 
        raise NotImplementedError(self.set_revision_history)
 
273
 
 
274
        # revno is 1-based; list is 0-based
 
275
 
 
276
        new_tree = self.revision_tree(rh[revno-1])
 
277
        if revno == 1:
 
278
            old_tree = EmptyTree()
 
279
        else:
 
280
            old_tree = self.revision_tree(rh[revno-2])
 
281
 
 
282
        return compare_trees(old_tree, new_tree)
 
283
 
 
284
    def get_revision_sha1(self, revision_id):
 
285
        """Hash the stored value of a revision, and return it."""
 
286
        raise NotImplementedError('get_revision_sha1 is abstract')
 
287
 
 
288
    def get_ancestry(self, revision_id):
 
289
        """Return a list of revision-ids integrated by a revision.
 
290
        
 
291
        This currently returns a list, but the ordering is not guaranteed:
 
292
        treat it as a set.
 
293
        """
 
294
        raise NotImplementedError('get_ancestry is abstract')
 
295
 
 
296
    def get_inventory(self, revision_id):
 
297
        """Get Inventory object by hash."""
 
298
        raise NotImplementedError('get_inventory is abstract')
 
299
 
 
300
    def get_inventory_xml(self, revision_id):
 
301
        """Get inventory XML as a file object."""
 
302
        raise NotImplementedError('get_inventory_xml is abstract')
 
303
 
 
304
    def get_inventory_sha1(self, revision_id):
 
305
        """Return the sha1 hash of the inventory entry."""
 
306
        raise NotImplementedError('get_inventory_sha1 is abstract')
 
307
 
 
308
    def get_revision_inventory(self, revision_id):
 
309
        """Return inventory of a past revision."""
 
310
        raise NotImplementedError('get_revision_inventory is abstract')
300
311
 
301
312
    def revision_history(self):
302
313
        """Return sequence of revision hashes on to this branch."""
303
 
        raise NotImplementedError(self.revision_history)
 
314
        raise NotImplementedError('revision_history is abstract')
304
315
 
305
316
    def revno(self):
306
317
        """Return current revision number for this branch.
310
321
        """
311
322
        return len(self.revision_history())
312
323
 
313
 
    def unbind(self):
314
 
        """Older format branches cannot bind or unbind."""
315
 
        raise errors.UpgradeRequired(self.base)
316
 
 
317
324
    def last_revision(self):
318
 
        """Return last revision id, or None"""
 
325
        """Return last patch hash, or None if no history."""
319
326
        ph = self.revision_history()
320
327
        if ph:
321
328
            return ph[-1]
322
329
        else:
323
330
            return None
324
331
 
325
 
    def missing_revisions(self, other, stop_revision=None):
 
332
    def missing_revisions(self, other, stop_revision=None, diverged_ok=False):
326
333
        """Return a list of new revisions that would perfectly fit.
327
334
        
328
335
        If self and other have not diverged, return a list of the revisions
329
336
        present in other, but missing from self.
 
337
 
 
338
        >>> from bzrlib.commit import commit
 
339
        >>> bzrlib.trace.silent = True
 
340
        >>> br1 = ScratchBranch()
 
341
        >>> br2 = ScratchBranch()
 
342
        >>> br1.missing_revisions(br2)
 
343
        []
 
344
        >>> commit(br2, "lala!", rev_id="REVISION-ID-1")
 
345
        >>> br1.missing_revisions(br2)
 
346
        [u'REVISION-ID-1']
 
347
        >>> br2.missing_revisions(br1)
 
348
        []
 
349
        >>> commit(br1, "lala!", rev_id="REVISION-ID-1")
 
350
        >>> br1.missing_revisions(br2)
 
351
        []
 
352
        >>> commit(br2, "lala!", rev_id="REVISION-ID-2A")
 
353
        >>> br1.missing_revisions(br2)
 
354
        [u'REVISION-ID-2A']
 
355
        >>> commit(br1, "lala!", rev_id="REVISION-ID-2B")
 
356
        >>> br1.missing_revisions(br2)
 
357
        Traceback (most recent call last):
 
358
        DivergedBranches: These branches have diverged.
330
359
        """
331
360
        self_history = self.revision_history()
332
361
        self_len = len(self_history)
342
371
        else:
343
372
            assert isinstance(stop_revision, int)
344
373
            if stop_revision > other_len:
345
 
                raise errors.NoSuchRevision(self, stop_revision)
 
374
                raise bzrlib.errors.NoSuchRevision(self, stop_revision)
346
375
        return other_history[self_len:stop_revision]
347
 
 
 
376
    
348
377
    def update_revisions(self, other, stop_revision=None):
349
 
        """Pull in new perfect-fit revisions.
350
 
 
351
 
        :param other: Another Branch to pull from
352
 
        :param stop_revision: Updated until the given revision
353
 
        :return: None
354
 
        """
355
 
        raise NotImplementedError(self.update_revisions)
356
 
 
 
378
        """Pull in new perfect-fit revisions."""
 
379
        raise NotImplementedError('update_revisions is abstract')
 
380
 
 
381
    def pullable_revisions(self, other, stop_revision):
 
382
        raise NotImplementedError('pullable_revisions is abstract')
 
383
        
357
384
    def revision_id_to_revno(self, revision_id):
358
385
        """Given a revision id, return its revno"""
359
386
        if revision_id is None:
370
397
            return None
371
398
        if history is None:
372
399
            history = self.revision_history()
373
 
        if revno <= 0 or revno > len(history):
 
400
        elif revno <= 0 or revno > len(history):
374
401
            raise bzrlib.errors.NoSuchRevision(self, revno)
375
402
        return history[revno - 1]
376
403
 
377
 
    def pull(self, source, overwrite=False, stop_revision=None):
378
 
        raise NotImplementedError(self.pull)
 
404
    def revision_tree(self, revision_id):
 
405
        """Return Tree for a revision on this branch.
 
406
 
 
407
        `revision_id` may be None for the null revision, in which case
 
408
        an `EmptyTree` is returned."""
 
409
        raise NotImplementedError('revision_tree is abstract')
 
410
 
 
411
    def working_tree(self):
 
412
        """Return a `Tree` for the working copy if this is a local branch."""
 
413
        raise NotImplementedError('working_tree is abstract')
 
414
 
 
415
    def pull(self, source, overwrite=False):
 
416
        raise NotImplementedError('pull is abstract')
379
417
 
380
418
    def basis_tree(self):
381
 
        """Return `Tree` object for last revision."""
382
 
        return self.repository.revision_tree(self.last_revision())
 
419
        """Return `Tree` object for last revision.
 
420
 
 
421
        If there are no revisions yet, return an `EmptyTree`.
 
422
        """
 
423
        return self.revision_tree(self.last_revision())
383
424
 
384
425
    def rename_one(self, from_rel, to_rel):
385
426
        """Rename one file.
386
427
 
387
428
        This can change the directory or the filename or both.
388
429
        """
389
 
        raise NotImplementedError(self.rename_one)
 
430
        raise NotImplementedError('rename_one is abstract')
390
431
 
391
432
    def move(self, from_paths, to_name):
392
433
        """Rename files.
402
443
        This returns a list of (from_path, to_path) pairs for each
403
444
        entry that is moved.
404
445
        """
405
 
        raise NotImplementedError(self.move)
 
446
        raise NotImplementedError('move is abstract')
406
447
 
407
448
    def get_parent(self):
408
449
        """Return the parent location of the branch.
411
452
        pattern is that the user can override it by specifying a
412
453
        location.
413
454
        """
414
 
        raise NotImplementedError(self.get_parent)
415
 
 
416
 
    def get_submit_branch(self):
417
 
        """Return the submit location of the branch.
418
 
 
419
 
        This is the default location for bundle.  The usual
420
 
        pattern is that the user can override it by specifying a
421
 
        location.
422
 
        """
423
 
        return self.get_config().get_user_option('submit_branch')
424
 
 
425
 
    def set_submit_branch(self, location):
426
 
        """Return the submit location of the branch.
427
 
 
428
 
        This is the default location for bundle.  The usual
429
 
        pattern is that the user can override it by specifying a
430
 
        location.
431
 
        """
432
 
        self.get_config().set_user_option('submit_branch', location)
 
455
        raise NotImplementedError('get_parent is abstract')
433
456
 
434
457
    def get_push_location(self):
435
458
        """Return the None or the location to push this branch to."""
436
 
        raise NotImplementedError(self.get_push_location)
 
459
        raise NotImplementedError('get_push_location is abstract')
437
460
 
438
461
    def set_push_location(self, location):
439
462
        """Set a new push location for this branch."""
440
 
        raise NotImplementedError(self.set_push_location)
 
463
        raise NotImplementedError('set_push_location is abstract')
441
464
 
442
465
    def set_parent(self, url):
443
 
        raise NotImplementedError(self.set_parent)
444
 
 
445
 
    @needs_write_lock
446
 
    def update(self):
447
 
        """Synchronise this branch with the master branch if any. 
448
 
 
449
 
        :return: None or the last_revision pivoted out during the update.
450
 
        """
451
 
        return None
 
466
        raise NotImplementedError('set_parent is abstract')
452
467
 
453
468
    def check_revno(self, revno):
454
469
        """\
465
480
        """
466
481
        if revno < 1 or revno > self.revno():
467
482
            raise InvalidRevisionNumber(revno)
468
 
 
469
 
    @needs_read_lock
470
 
    def clone(self, *args, **kwargs):
471
 
        """Clone this branch into to_bzrdir preserving all semantic values.
472
 
        
473
 
        revision_id: if not None, the revision history in the new branch will
474
 
                     be truncated to end with revision_id.
475
 
        """
476
 
        # for API compatibility, until 0.8 releases we provide the old api:
477
 
        # def clone(self, to_location, revision=None, basis_branch=None, to_branch_format=None):
478
 
        # after 0.8 releases, the *args and **kwargs should be changed:
479
 
        # def clone(self, to_bzrdir, revision_id=None):
480
 
        if (kwargs.get('to_location', None) or
481
 
            kwargs.get('revision', None) or
482
 
            kwargs.get('basis_branch', None) or
483
 
            (len(args) and isinstance(args[0], basestring))):
484
 
            # backwards compatibility api:
485
 
            warn("Branch.clone() has been deprecated for BzrDir.clone() from"
486
 
                 " bzrlib 0.8.", DeprecationWarning, stacklevel=3)
487
 
            # get basis_branch
488
 
            if len(args) > 2:
489
 
                basis_branch = args[2]
490
 
            else:
491
 
                basis_branch = kwargs.get('basis_branch', None)
492
 
            if basis_branch:
493
 
                basis = basis_branch.bzrdir
494
 
            else:
495
 
                basis = None
496
 
            # get revision
497
 
            if len(args) > 1:
498
 
                revision_id = args[1]
499
 
            else:
500
 
                revision_id = kwargs.get('revision', None)
501
 
            # get location
502
 
            if len(args):
503
 
                url = args[0]
504
 
            else:
505
 
                # no default to raise if not provided.
506
 
                url = kwargs.get('to_location')
507
 
            return self.bzrdir.clone(url,
508
 
                                     revision_id=revision_id,
509
 
                                     basis=basis).open_branch()
510
 
        # new cleaner api.
511
 
        # generate args by hand 
512
 
        if len(args) > 1:
513
 
            revision_id = args[1]
514
 
        else:
515
 
            revision_id = kwargs.get('revision_id', None)
516
 
        if len(args):
517
 
            to_bzrdir = args[0]
518
 
        else:
519
 
            # no default to raise if not provided.
520
 
            to_bzrdir = kwargs.get('to_bzrdir')
521
 
        result = self._format.initialize(to_bzrdir)
522
 
        self.copy_content_into(result, revision_id=revision_id)
523
 
        return  result
524
 
 
525
 
    @needs_read_lock
526
 
    def sprout(self, to_bzrdir, revision_id=None):
527
 
        """Create a new line of development from the branch, into to_bzrdir.
528
 
        
529
 
        revision_id: if not None, the revision history in the new branch will
530
 
                     be truncated to end with revision_id.
531
 
        """
532
 
        result = self._format.initialize(to_bzrdir)
533
 
        self.copy_content_into(result, revision_id=revision_id)
534
 
        result.set_parent(self.bzrdir.root_transport.base)
535
 
        return result
536
 
 
537
 
    @needs_read_lock
538
 
    def copy_content_into(self, destination, revision_id=None):
539
 
        """Copy the content of self into destination.
540
 
 
541
 
        revision_id: if not None, the revision history in the new branch will
542
 
                     be truncated to end with revision_id.
543
 
        """
544
 
        new_history = self.revision_history()
545
 
        if revision_id is not None:
546
 
            try:
547
 
                new_history = new_history[:new_history.index(revision_id) + 1]
548
 
            except ValueError:
549
 
                rev = self.repository.get_revision(revision_id)
550
 
                new_history = rev.get_history(self.repository)[1:]
551
 
        destination.set_revision_history(new_history)
552
 
        try:
553
 
            parent = self.get_parent()
554
 
        except errors.InaccessibleParent, e:
555
 
            mutter('parent was not accessible to copy: %s', e)
556
 
        else:
557
 
            if parent:
558
 
                destination.set_parent(parent)
559
 
 
560
 
    @needs_read_lock
561
 
    def check(self):
562
 
        """Check consistency of the branch.
563
 
 
564
 
        In particular this checks that revisions given in the revision-history
565
 
        do actually match up in the revision graph, and that they're all 
566
 
        present in the repository.
567
 
        
568
 
        Callers will typically also want to check the repository.
569
 
 
570
 
        :return: A BranchCheckResult.
571
 
        """
572
 
        mainline_parent_id = None
573
 
        for revision_id in self.revision_history():
574
 
            try:
575
 
                revision = self.repository.get_revision(revision_id)
576
 
            except errors.NoSuchRevision, e:
577
 
                raise errors.BzrCheckError("mainline revision {%s} not in repository"
578
 
                            % revision_id)
579
 
            # In general the first entry on the revision history has no parents.
580
 
            # But it's not illegal for it to have parents listed; this can happen
581
 
            # in imports from Arch when the parents weren't reachable.
582
 
            if mainline_parent_id is not None:
583
 
                if mainline_parent_id not in revision.parent_ids:
584
 
                    raise errors.BzrCheckError("previous revision {%s} not listed among "
585
 
                                        "parents of {%s}"
586
 
                                        % (mainline_parent_id, revision_id))
587
 
            mainline_parent_id = revision_id
588
 
        return BranchCheckResult(self)
589
 
 
590
 
    def _get_checkout_format(self):
591
 
        """Return the most suitable metadir for a checkout of this branch.
592
 
        Weaves are used if this branch's repostory uses weaves.
593
 
        """
594
 
        if isinstance(self.bzrdir, bzrdir.BzrDirPreSplitOut):
595
 
            from bzrlib import repository
596
 
            format = bzrdir.BzrDirMetaFormat1()
597
 
            format.repository_format = repository.RepositoryFormat7()
598
 
        else:
599
 
            format = self.repository.bzrdir.cloning_metadir()
600
 
        return format
601
 
 
602
 
    def create_checkout(self, to_location, revision_id=None, 
603
 
                        lightweight=False):
604
 
        """Create a checkout of a branch.
605
 
        
606
 
        :param to_location: The url to produce the checkout at
607
 
        :param revision_id: The revision to check out
608
 
        :param lightweight: If True, produce a lightweight checkout, otherwise,
609
 
        produce a bound branch (heavyweight checkout)
610
 
        :return: The tree of the created checkout
611
 
        """
612
 
        t = transport.get_transport(to_location)
613
 
        try:
614
 
            t.mkdir('.')
615
 
        except errors.FileExists:
616
 
            pass
617
 
        if lightweight:
618
 
            checkout = bzrdir.BzrDirMetaFormat1().initialize_on_transport(t)
619
 
            BranchReferenceFormat().initialize(checkout, self)
620
 
        else:
621
 
            format = self._get_checkout_format()
622
 
            checkout_branch = bzrdir.BzrDir.create_branch_convenience(
623
 
                to_location, force_new_tree=False, format=format)
624
 
            checkout = checkout_branch.bzrdir
625
 
            checkout_branch.bind(self)
626
 
            # pull up to the specified revision_id to set the initial 
627
 
            # branch tip correctly, and seed it with history.
628
 
            checkout_branch.pull(self, stop_revision=revision_id)
629
 
        return checkout.create_workingtree(revision_id)
630
 
 
631
 
 
632
 
class BranchFormat(object):
633
 
    """An encapsulation of the initialization and open routines for a format.
634
 
 
635
 
    Formats provide three things:
636
 
     * An initialization routine,
637
 
     * a format string,
638
 
     * an open routine.
639
 
 
640
 
    Formats are placed in an dict by their format string for reference 
641
 
    during branch opening. Its not required that these be instances, they
642
 
    can be classes themselves with class methods - it simply depends on 
643
 
    whether state is needed for a given format or not.
644
 
 
645
 
    Once a format is deprecated, just deprecate the initialize and open
646
 
    methods on the format class. Do not deprecate the object, as the 
647
 
    object will be created every time regardless.
648
 
    """
649
 
 
650
 
    _default_format = None
651
 
    """The default format used for new branches."""
652
 
 
653
 
    _formats = {}
654
 
    """The known formats."""
655
 
 
656
 
    @classmethod
657
 
    def find_format(klass, a_bzrdir):
658
 
        """Return the format for the branch object in a_bzrdir."""
659
 
        try:
660
 
            transport = a_bzrdir.get_branch_transport(None)
661
 
            format_string = transport.get("format").read()
662
 
            return klass._formats[format_string]
663
 
        except NoSuchFile:
664
 
            raise NotBranchError(path=transport.base)
665
 
        except KeyError:
666
 
            raise errors.UnknownFormatError(format=format_string)
667
 
 
668
 
    @classmethod
669
 
    def get_default_format(klass):
670
 
        """Return the current default format."""
671
 
        return klass._default_format
672
 
 
673
 
    def get_format_string(self):
674
 
        """Return the ASCII format string that identifies this format."""
675
 
        raise NotImplementedError(self.get_format_string)
676
 
 
677
 
    def get_format_description(self):
678
 
        """Return the short format description for this format."""
679
 
        raise NotImplementedError(self.get_format_string)
680
 
 
681
 
    def initialize(self, a_bzrdir):
682
 
        """Create a branch of this format in a_bzrdir."""
683
 
        raise NotImplementedError(self.initialize)
684
 
 
685
 
    def is_supported(self):
686
 
        """Is this format supported?
687
 
 
688
 
        Supported formats can be initialized and opened.
689
 
        Unsupported formats may not support initialization or committing or 
690
 
        some other features depending on the reason for not being supported.
691
 
        """
692
 
        return True
693
 
 
694
 
    def open(self, a_bzrdir, _found=False):
695
 
        """Return the branch object for a_bzrdir
696
 
 
697
 
        _found is a private parameter, do not use it. It is used to indicate
698
 
               if format probing has already be done.
699
 
        """
700
 
        raise NotImplementedError(self.open)
701
 
 
702
 
    @classmethod
703
 
    def register_format(klass, format):
704
 
        klass._formats[format.get_format_string()] = format
705
 
 
706
 
    @classmethod
707
 
    def set_default_format(klass, format):
708
 
        klass._default_format = format
709
 
 
710
 
    @classmethod
711
 
    def unregister_format(klass, format):
712
 
        assert klass._formats[format.get_format_string()] is format
713
 
        del klass._formats[format.get_format_string()]
714
 
 
715
 
    def __str__(self):
716
 
        return self.get_format_string().rstrip()
717
 
 
718
 
 
719
 
class BzrBranchFormat4(BranchFormat):
720
 
    """Bzr branch format 4.
721
 
 
722
 
    This format has:
723
 
     - a revision-history file.
724
 
     - a branch-lock lock file [ to be shared with the bzrdir ]
725
 
    """
726
 
 
727
 
    def get_format_description(self):
728
 
        """See BranchFormat.get_format_description()."""
729
 
        return "Branch format 4"
730
 
 
731
 
    def initialize(self, a_bzrdir):
732
 
        """Create a branch of this format in a_bzrdir."""
733
 
        mutter('creating branch in %s', a_bzrdir.transport.base)
734
 
        branch_transport = a_bzrdir.get_branch_transport(self)
735
 
        utf8_files = [('revision-history', ''),
736
 
                      ('branch-name', ''),
737
 
                      ]
738
 
        control_files = lockable_files.LockableFiles(branch_transport,
739
 
                             'branch-lock', lockable_files.TransportLock)
740
 
        control_files.create_lock()
741
 
        control_files.lock_write()
742
 
        try:
743
 
            for file, content in utf8_files:
744
 
                control_files.put_utf8(file, content)
745
 
        finally:
746
 
            control_files.unlock()
747
 
        return self.open(a_bzrdir, _found=True)
748
 
 
749
 
    def __init__(self):
750
 
        super(BzrBranchFormat4, self).__init__()
751
 
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
752
 
 
753
 
    def open(self, a_bzrdir, _found=False):
754
 
        """Return the branch object for a_bzrdir
755
 
 
756
 
        _found is a private parameter, do not use it. It is used to indicate
757
 
               if format probing has already be done.
758
 
        """
759
 
        if not _found:
760
 
            # we are being called directly and must probe.
761
 
            raise NotImplementedError
762
 
        return BzrBranch(_format=self,
763
 
                         _control_files=a_bzrdir._control_files,
764
 
                         a_bzrdir=a_bzrdir,
765
 
                         _repository=a_bzrdir.open_repository())
766
 
 
767
 
    def __str__(self):
768
 
        return "Bazaar-NG branch format 4"
769
 
 
770
 
 
771
 
class BzrBranchFormat5(BranchFormat):
772
 
    """Bzr branch format 5.
773
 
 
774
 
    This format has:
775
 
     - a revision-history file.
776
 
     - a format string
777
 
     - a lock dir guarding the branch itself
778
 
     - all of this stored in a branch/ subdirectory
779
 
     - works with shared repositories.
780
 
 
781
 
    This format is new in bzr 0.8.
782
 
    """
783
 
 
784
 
    def get_format_string(self):
785
 
        """See BranchFormat.get_format_string()."""
786
 
        return "Bazaar-NG branch format 5\n"
787
 
 
788
 
    def get_format_description(self):
789
 
        """See BranchFormat.get_format_description()."""
790
 
        return "Branch format 5"
791
 
        
792
 
    def initialize(self, a_bzrdir):
793
 
        """Create a branch of this format in a_bzrdir."""
794
 
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
795
 
        branch_transport = a_bzrdir.get_branch_transport(self)
796
 
        utf8_files = [('revision-history', ''),
797
 
                      ('branch-name', ''),
798
 
                      ]
799
 
        control_files = lockable_files.LockableFiles(branch_transport, 'lock',
800
 
                                                     lockdir.LockDir)
801
 
        control_files.create_lock()
802
 
        control_files.lock_write()
803
 
        control_files.put_utf8('format', self.get_format_string())
804
 
        try:
805
 
            for file, content in utf8_files:
806
 
                control_files.put_utf8(file, content)
807
 
        finally:
808
 
            control_files.unlock()
809
 
        return self.open(a_bzrdir, _found=True, )
810
 
 
811
 
    def __init__(self):
812
 
        super(BzrBranchFormat5, self).__init__()
813
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
814
 
 
815
 
    def open(self, a_bzrdir, _found=False):
816
 
        """Return the branch object for a_bzrdir
817
 
 
818
 
        _found is a private parameter, do not use it. It is used to indicate
819
 
               if format probing has already be done.
820
 
        """
821
 
        if not _found:
822
 
            format = BranchFormat.find_format(a_bzrdir)
823
 
            assert format.__class__ == self.__class__
824
 
        transport = a_bzrdir.get_branch_transport(None)
825
 
        control_files = lockable_files.LockableFiles(transport, 'lock',
826
 
                                                     lockdir.LockDir)
827
 
        return BzrBranch5(_format=self,
828
 
                          _control_files=control_files,
829
 
                          a_bzrdir=a_bzrdir,
830
 
                          _repository=a_bzrdir.find_repository())
831
 
 
832
 
    def __str__(self):
833
 
        return "Bazaar-NG Metadir branch format 5"
834
 
 
835
 
 
836
 
class BranchReferenceFormat(BranchFormat):
837
 
    """Bzr branch reference format.
838
 
 
839
 
    Branch references are used in implementing checkouts, they
840
 
    act as an alias to the real branch which is at some other url.
841
 
 
842
 
    This format has:
843
 
     - A location file
844
 
     - a format string
845
 
    """
846
 
 
847
 
    def get_format_string(self):
848
 
        """See BranchFormat.get_format_string()."""
849
 
        return "Bazaar-NG Branch Reference Format 1\n"
850
 
 
851
 
    def get_format_description(self):
852
 
        """See BranchFormat.get_format_description()."""
853
 
        return "Checkout reference format 1"
854
 
        
855
 
    def initialize(self, a_bzrdir, target_branch=None):
856
 
        """Create a branch of this format in a_bzrdir."""
857
 
        if target_branch is None:
858
 
            # this format does not implement branch itself, thus the implicit
859
 
            # creation contract must see it as uninitializable
860
 
            raise errors.UninitializableFormat(self)
861
 
        mutter('creating branch reference in %s', a_bzrdir.transport.base)
862
 
        branch_transport = a_bzrdir.get_branch_transport(self)
863
 
        branch_transport.put_bytes('location',
864
 
            target_branch.bzrdir.root_transport.base)
865
 
        branch_transport.put_bytes('format', self.get_format_string())
866
 
        return self.open(a_bzrdir, _found=True)
867
 
 
868
 
    def __init__(self):
869
 
        super(BranchReferenceFormat, self).__init__()
870
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
871
 
 
872
 
    def _make_reference_clone_function(format, a_branch):
873
 
        """Create a clone() routine for a branch dynamically."""
874
 
        def clone(to_bzrdir, revision_id=None):
875
 
            """See Branch.clone()."""
876
 
            return format.initialize(to_bzrdir, a_branch)
877
 
            # cannot obey revision_id limits when cloning a reference ...
878
 
            # FIXME RBC 20060210 either nuke revision_id for clone, or
879
 
            # emit some sort of warning/error to the caller ?!
880
 
        return clone
881
 
 
882
 
    def open(self, a_bzrdir, _found=False):
883
 
        """Return the branch that the branch reference in a_bzrdir points at.
884
 
 
885
 
        _found is a private parameter, do not use it. It is used to indicate
886
 
               if format probing has already be done.
887
 
        """
888
 
        if not _found:
889
 
            format = BranchFormat.find_format(a_bzrdir)
890
 
            assert format.__class__ == self.__class__
891
 
        transport = a_bzrdir.get_branch_transport(None)
892
 
        real_bzrdir = bzrdir.BzrDir.open(transport.get('location').read())
893
 
        result = real_bzrdir.open_branch()
894
 
        # this changes the behaviour of result.clone to create a new reference
895
 
        # rather than a copy of the content of the branch.
896
 
        # I did not use a proxy object because that needs much more extensive
897
 
        # testing, and we are only changing one behaviour at the moment.
898
 
        # If we decide to alter more behaviours - i.e. the implicit nickname
899
 
        # then this should be refactored to introduce a tested proxy branch
900
 
        # and a subclass of that for use in overriding clone() and ....
901
 
        # - RBC 20060210
902
 
        result.clone = self._make_reference_clone_function(result)
903
 
        return result
904
 
 
905
 
 
906
 
# formats which have no format string are not discoverable
907
 
# and not independently creatable, so are not registered.
908
 
__default_format = BzrBranchFormat5()
909
 
BranchFormat.register_format(__default_format)
910
 
BranchFormat.register_format(BranchReferenceFormat())
911
 
BranchFormat.set_default_format(__default_format)
912
 
_legacy_formats = [BzrBranchFormat4(),
913
 
                   ]
 
483
        
 
484
    def sign_revision(self, revision_id, gpg_strategy):
 
485
        raise NotImplementedError('sign_revision is abstract')
 
486
 
 
487
    def store_revision_signature(self, gpg_strategy, plaintext, revision_id):
 
488
        raise NotImplementedError('store_revision_signature is abstract')
914
489
 
915
490
class BzrBranch(Branch):
916
491
    """A branch stored in the actual filesystem.
918
493
    Note that it's "local" in the context of the filesystem; it doesn't
919
494
    really matter if it's on an nfs/smb/afs/coda/... share, as long as
920
495
    it's writable, and can be accessed via the normal filesystem API.
 
496
 
 
497
    _lock_mode
 
498
        None, or 'r' or 'w'
 
499
 
 
500
    _lock_count
 
501
        If _lock_mode is true, a positive count of the number of times the
 
502
        lock has been taken.
 
503
 
 
504
    _lock
 
505
        Lock object from bzrlib.lock.
921
506
    """
 
507
    # We actually expect this class to be somewhat short-lived; part of its
 
508
    # purpose is to try to isolate what bits of the branch logic are tied to
 
509
    # filesystem access, so that in a later step, we can extricate them to
 
510
    # a separarte ("storage") class.
 
511
    _lock_mode = None
 
512
    _lock_count = None
 
513
    _lock = None
 
514
    _inventory_weave = None
922
515
    
923
 
    def __init__(self, transport=DEPRECATED_PARAMETER, init=DEPRECATED_PARAMETER,
924
 
                 relax_version_check=DEPRECATED_PARAMETER, _format=None,
925
 
                 _control_files=None, a_bzrdir=None, _repository=None):
 
516
    # Map some sort of prefix into a namespace
 
517
    # stuff like "revno:10", "revid:", etc.
 
518
    # This should match a prefix with a function which accepts
 
519
    REVISION_NAMESPACES = {}
 
520
 
 
521
    def push_stores(self, branch_to):
 
522
        """See Branch.push_stores."""
 
523
        if (self._branch_format != branch_to._branch_format
 
524
            or self._branch_format != 4):
 
525
            from bzrlib.fetch import greedy_fetch
 
526
            mutter("falling back to fetch logic to push between %s(%s) and %s(%s)",
 
527
                   self, self._branch_format, branch_to, branch_to._branch_format)
 
528
            greedy_fetch(to_branch=branch_to, from_branch=self,
 
529
                         revision=self.last_revision())
 
530
            return
 
531
 
 
532
        store_pairs = ((self.text_store,      branch_to.text_store),
 
533
                       (self.inventory_store, branch_to.inventory_store),
 
534
                       (self.revision_store,  branch_to.revision_store))
 
535
        try:
 
536
            for from_store, to_store in store_pairs: 
 
537
                copy_all(from_store, to_store)
 
538
        except UnlistableStore:
 
539
            raise UnlistableBranch(from_store)
 
540
 
 
541
    def __init__(self, transport, init=False,
 
542
                 relax_version_check=False):
926
543
        """Create new branch object at a particular location.
927
544
 
928
545
        transport -- A Transport object, defining how to access files.
935
552
            version is not applied.  This is intended only for
936
553
            upgrade/recovery type use; it's not guaranteed that
937
554
            all operations will work on old format branches.
 
555
 
 
556
        In the test suite, creation of new trees is tested using the
 
557
        `ScratchBranch` class.
938
558
        """
939
 
        if a_bzrdir is None:
940
 
            self.bzrdir = bzrdir.BzrDir.open(transport.base)
941
 
        else:
942
 
            self.bzrdir = a_bzrdir
943
 
        self._transport = self.bzrdir.transport.clone('..')
944
 
        self._base = self._transport.base
945
 
        self._format = _format
946
 
        if _control_files is None:
947
 
            raise ValueError('BzrBranch _control_files is None')
948
 
        self.control_files = _control_files
949
 
        if deprecated_passed(init):
950
 
            warn("BzrBranch.__init__(..., init=XXX): The init parameter is "
951
 
                 "deprecated as of bzr 0.8. Please use Branch.create().",
952
 
                 DeprecationWarning,
953
 
                 stacklevel=2)
954
 
            if init:
955
 
                # this is slower than before deprecation, oh well never mind.
956
 
                # -> its deprecated.
957
 
                self._initialize(transport.base)
958
 
        self._check_format(_format)
959
 
        if deprecated_passed(relax_version_check):
960
 
            warn("BzrBranch.__init__(..., relax_version_check=XXX_: The "
961
 
                 "relax_version_check parameter is deprecated as of bzr 0.8. "
962
 
                 "Please use BzrDir.open_downlevel, or a BzrBranchFormat's "
963
 
                 "open() method.",
964
 
                 DeprecationWarning,
965
 
                 stacklevel=2)
966
 
            if (not relax_version_check
967
 
                and not self._format.is_supported()):
968
 
                raise errors.UnsupportedFormatError(format=fmt)
969
 
        if deprecated_passed(transport):
970
 
            warn("BzrBranch.__init__(transport=XXX...): The transport "
971
 
                 "parameter is deprecated as of bzr 0.8. "
972
 
                 "Please use Branch.open, or bzrdir.open_branch().",
973
 
                 DeprecationWarning,
974
 
                 stacklevel=2)
975
 
        self.repository = _repository
 
559
        assert isinstance(transport, Transport), \
 
560
            "%r is not a Transport" % transport
 
561
        self._transport = transport
 
562
        if init:
 
563
            self._make_control()
 
564
        self._check_format(relax_version_check)
 
565
 
 
566
        def get_store(name, compressed=True, prefixed=False):
 
567
            # FIXME: This approach of assuming stores are all entirely compressed
 
568
            # or entirely uncompressed is tidy, but breaks upgrade from 
 
569
            # some existing branches where there's a mixture; we probably 
 
570
            # still want the option to look for both.
 
571
            relpath = self._rel_controlfilename(name)
 
572
            store = TextStore(self._transport.clone(relpath),
 
573
                              prefixed=prefixed,
 
574
                              compressed=compressed)
 
575
            #if self._transport.should_cache():
 
576
            #    cache_path = os.path.join(self.cache_root, name)
 
577
            #    os.mkdir(cache_path)
 
578
            #    store = bzrlib.store.CachedStore(store, cache_path)
 
579
            return store
 
580
        def get_weave(name, prefixed=False):
 
581
            relpath = self._rel_controlfilename(name)
 
582
            ws = WeaveStore(self._transport.clone(relpath), prefixed=prefixed)
 
583
            if self._transport.should_cache():
 
584
                ws.enable_cache = True
 
585
            return ws
 
586
 
 
587
        if self._branch_format == 4:
 
588
            self.inventory_store = get_store('inventory-store')
 
589
            self.text_store = get_store('text-store')
 
590
            self.revision_store = get_store('revision-store')
 
591
        elif self._branch_format == 5:
 
592
            self.control_weaves = get_weave('')
 
593
            self.weave_store = get_weave('weaves')
 
594
            self.revision_store = get_store('revision-store', compressed=False)
 
595
        elif self._branch_format == 6:
 
596
            self.control_weaves = get_weave('')
 
597
            self.weave_store = get_weave('weaves', prefixed=True)
 
598
            self.revision_store = get_store('revision-store', compressed=False,
 
599
                                            prefixed=True)
 
600
        self.revision_store.register_suffix('sig')
 
601
        self._transaction = None
976
602
 
977
603
    def __str__(self):
978
 
        return '%s(%r)' % (self.__class__.__name__, self.base)
 
604
        return '%s(%r)' % (self.__class__.__name__, self._transport.base)
979
605
 
980
606
    __repr__ = __str__
981
607
 
 
608
    def __del__(self):
 
609
        if self._lock_mode or self._lock:
 
610
            # XXX: This should show something every time, and be suitable for
 
611
            # headless operation and embedding
 
612
            warn("branch %r was not explicitly unlocked" % self)
 
613
            self._lock.unlock()
 
614
 
 
615
        # TODO: It might be best to do this somewhere else,
 
616
        # but it is nice for a Branch object to automatically
 
617
        # cache it's information.
 
618
        # Alternatively, we could have the Transport objects cache requests
 
619
        # See the earlier discussion about how major objects (like Branch)
 
620
        # should never expect their __del__ function to run.
 
621
        if hasattr(self, 'cache_root') and self.cache_root is not None:
 
622
            try:
 
623
                shutil.rmtree(self.cache_root)
 
624
            except:
 
625
                pass
 
626
            self.cache_root = None
 
627
 
982
628
    def _get_base(self):
983
 
        return self._base
 
629
        if self._transport:
 
630
            return self._transport.base
 
631
        return None
984
632
 
985
633
    base = property(_get_base, doc="The URL for the root of this branch.")
986
634
 
987
635
    def _finish_transaction(self):
988
636
        """Exit the current transaction."""
989
 
        return self.control_files._finish_transaction()
 
637
        if self._transaction is None:
 
638
            raise errors.LockError('Branch %s is not in a transaction' %
 
639
                                   self)
 
640
        transaction = self._transaction
 
641
        self._transaction = None
 
642
        transaction.finish()
990
643
 
991
644
    def get_transaction(self):
992
 
        """Return the current active transaction.
993
 
 
994
 
        If no transaction is active, this returns a passthrough object
995
 
        for which all data is immediately flushed and no caching happens.
996
 
        """
997
 
        # this is an explicit function so that we can do tricky stuff
998
 
        # when the storage in rev_storage is elsewhere.
999
 
        # we probably need to hook the two 'lock a location' and 
1000
 
        # 'have a transaction' together more delicately, so that
1001
 
        # we can have two locks (branch and storage) and one transaction
1002
 
        # ... and finishing the transaction unlocks both, but unlocking
1003
 
        # does not. - RBC 20051121
1004
 
        return self.control_files.get_transaction()
1005
 
 
1006
 
    def _set_transaction(self, transaction):
 
645
        """See Branch.get_transaction."""
 
646
        if self._transaction is None:
 
647
            return transactions.PassThroughTransaction()
 
648
        else:
 
649
            return self._transaction
 
650
 
 
651
    def _set_transaction(self, new_transaction):
1007
652
        """Set a new active transaction."""
1008
 
        return self.control_files._set_transaction(transaction)
 
653
        if self._transaction is not None:
 
654
            raise errors.LockError('Branch %s is in a transaction already.' %
 
655
                                   self)
 
656
        self._transaction = new_transaction
 
657
 
 
658
    def lock_write(self):
 
659
        mutter("lock write: %s (%s)", self, self._lock_count)
 
660
        # TODO: Upgrade locking to support using a Transport,
 
661
        # and potentially a remote locking protocol
 
662
        if self._lock_mode:
 
663
            if self._lock_mode != 'w':
 
664
                raise LockError("can't upgrade to a write lock from %r" %
 
665
                                self._lock_mode)
 
666
            self._lock_count += 1
 
667
        else:
 
668
            self._lock = self._transport.lock_write(
 
669
                    self._rel_controlfilename('branch-lock'))
 
670
            self._lock_mode = 'w'
 
671
            self._lock_count = 1
 
672
            self._set_transaction(transactions.PassThroughTransaction())
 
673
 
 
674
    def lock_read(self):
 
675
        mutter("lock read: %s (%s)", self, self._lock_count)
 
676
        if self._lock_mode:
 
677
            assert self._lock_mode in ('r', 'w'), \
 
678
                   "invalid lock mode %r" % self._lock_mode
 
679
            self._lock_count += 1
 
680
        else:
 
681
            self._lock = self._transport.lock_read(
 
682
                    self._rel_controlfilename('branch-lock'))
 
683
            self._lock_mode = 'r'
 
684
            self._lock_count = 1
 
685
            self._set_transaction(transactions.ReadOnlyTransaction())
 
686
            # 5K may be excessive, but hey, its a knob.
 
687
            self.get_transaction().set_cache_size(5000)
 
688
                        
 
689
    def unlock(self):
 
690
        mutter("unlock: %s (%s)", self, self._lock_count)
 
691
        if not self._lock_mode:
 
692
            raise LockError('branch %r is not locked' % (self))
 
693
 
 
694
        if self._lock_count > 1:
 
695
            self._lock_count -= 1
 
696
        else:
 
697
            self._finish_transaction()
 
698
            self._lock.unlock()
 
699
            self._lock = None
 
700
            self._lock_mode = self._lock_count = None
1009
701
 
1010
702
    def abspath(self, name):
1011
703
        """See Branch.abspath."""
1012
 
        return self.control_files._transport.abspath(name)
1013
 
 
1014
 
    def _check_format(self, format):
1015
 
        """Identify the branch format if needed.
1016
 
 
1017
 
        The format is stored as a reference to the format object in
1018
 
        self._format for code that needs to check it later.
1019
 
 
1020
 
        The format parameter is either None or the branch format class
1021
 
        used to open this branch.
1022
 
 
1023
 
        FIXME: DELETE THIS METHOD when pre 0.8 support is removed.
 
704
        return self._transport.abspath(name)
 
705
 
 
706
    def _rel_controlfilename(self, file_or_path):
 
707
        if not isinstance(file_or_path, basestring):
 
708
            file_or_path = '/'.join(file_or_path)
 
709
        if file_or_path == '':
 
710
            return bzrlib.BZRDIR
 
711
        return bzrlib.transport.urlescape(bzrlib.BZRDIR + '/' + file_or_path)
 
712
 
 
713
    def controlfilename(self, file_or_path):
 
714
        """See Branch.controlfilename."""
 
715
        return self._transport.abspath(self._rel_controlfilename(file_or_path))
 
716
 
 
717
    def controlfile(self, file_or_path, mode='r'):
 
718
        """See Branch.controlfile."""
 
719
        import codecs
 
720
 
 
721
        relpath = self._rel_controlfilename(file_or_path)
 
722
        #TODO: codecs.open() buffers linewise, so it was overloaded with
 
723
        # a much larger buffer, do we need to do the same for getreader/getwriter?
 
724
        if mode == 'rb': 
 
725
            return self._transport.get(relpath)
 
726
        elif mode == 'wb':
 
727
            raise BzrError("Branch.controlfile(mode='wb') is not supported, use put_controlfiles")
 
728
        elif mode == 'r':
 
729
            # XXX: Do we really want errors='replace'?   Perhaps it should be
 
730
            # an error, or at least reported, if there's incorrectly-encoded
 
731
            # data inside a file.
 
732
            # <https://launchpad.net/products/bzr/+bug/3823>
 
733
            return codecs.getreader('utf-8')(self._transport.get(relpath), errors='replace')
 
734
        elif mode == 'w':
 
735
            raise BzrError("Branch.controlfile(mode='w') is not supported, use put_controlfiles")
 
736
        else:
 
737
            raise BzrError("invalid controlfile mode %r" % mode)
 
738
 
 
739
    def put_controlfile(self, path, f, encode=True):
 
740
        """See Branch.put_controlfile."""
 
741
        self.put_controlfiles([(path, f)], encode=encode)
 
742
 
 
743
    def put_controlfiles(self, files, encode=True):
 
744
        """See Branch.put_controlfiles."""
 
745
        import codecs
 
746
        ctrl_files = []
 
747
        for path, f in files:
 
748
            if encode:
 
749
                if isinstance(f, basestring):
 
750
                    f = f.encode('utf-8', 'replace')
 
751
                else:
 
752
                    f = codecs.getwriter('utf-8')(f, errors='replace')
 
753
            path = self._rel_controlfilename(path)
 
754
            ctrl_files.append((path, f))
 
755
        self._transport.put_multi(ctrl_files)
 
756
 
 
757
    def _make_control(self):
 
758
        from bzrlib.inventory import Inventory
 
759
        from bzrlib.weavefile import write_weave_v5
 
760
        from bzrlib.weave import Weave
 
761
        
 
762
        # Create an empty inventory
 
763
        sio = StringIO()
 
764
        # if we want per-tree root ids then this is the place to set
 
765
        # them; they're not needed for now and so ommitted for
 
766
        # simplicity.
 
767
        bzrlib.xml5.serializer_v5.write_inventory(Inventory(), sio)
 
768
        empty_inv = sio.getvalue()
 
769
        sio = StringIO()
 
770
        bzrlib.weavefile.write_weave_v5(Weave(), sio)
 
771
        empty_weave = sio.getvalue()
 
772
 
 
773
        dirs = [[], 'revision-store', 'weaves']
 
774
        files = [('README', 
 
775
            "This is a Bazaar-NG control directory.\n"
 
776
            "Do not change any files in this directory.\n"),
 
777
            ('branch-format', BZR_BRANCH_FORMAT_6),
 
778
            ('revision-history', ''),
 
779
            ('branch-name', ''),
 
780
            ('branch-lock', ''),
 
781
            ('pending-merges', ''),
 
782
            ('inventory', empty_inv),
 
783
            ('inventory.weave', empty_weave),
 
784
            ('ancestry.weave', empty_weave)
 
785
        ]
 
786
        cfn = self._rel_controlfilename
 
787
        self._transport.mkdir_multi([cfn(d) for d in dirs])
 
788
        self.put_controlfiles(files)
 
789
        mutter('created control directory in ' + self._transport.base)
 
790
 
 
791
    def _check_format(self, relax_version_check):
 
792
        """Check this branch format is supported.
 
793
 
 
794
        The format level is stored, as an integer, in
 
795
        self._branch_format for code that needs to check it later.
 
796
 
 
797
        In the future, we might need different in-memory Branch
 
798
        classes to support downlevel branches.  But not yet.
1024
799
        """
1025
 
        if format is None:
1026
 
            format = BranchFormat.find_format(self.bzrdir)
1027
 
        self._format = format
1028
 
        mutter("got branch format %s", self._format)
 
800
        try:
 
801
            fmt = self.controlfile('branch-format', 'r').read()
 
802
        except NoSuchFile:
 
803
            raise NotBranchError(path=self.base)
 
804
        mutter("got branch format %r", fmt)
 
805
        if fmt == BZR_BRANCH_FORMAT_6:
 
806
            self._branch_format = 6
 
807
        elif fmt == BZR_BRANCH_FORMAT_5:
 
808
            self._branch_format = 5
 
809
        elif fmt == BZR_BRANCH_FORMAT_4:
 
810
            self._branch_format = 4
 
811
 
 
812
        if (not relax_version_check
 
813
            and self._branch_format not in (5, 6)):
 
814
            raise errors.UnsupportedFormatError(
 
815
                           'sorry, branch format %r not supported' % fmt,
 
816
                           ['use a different bzr version',
 
817
                            'or remove the .bzr directory'
 
818
                            ' and "bzr init" again'])
1029
819
 
1030
820
    @needs_read_lock
1031
821
    def get_root_id(self):
1032
822
        """See Branch.get_root_id."""
1033
 
        tree = self.repository.revision_tree(self.last_revision())
1034
 
        return tree.inventory.root.file_id
1035
 
 
1036
 
    def is_locked(self):
1037
 
        return self.control_files.is_locked()
1038
 
 
1039
 
    def lock_write(self):
1040
 
        self.repository.lock_write()
1041
 
        try:
1042
 
            self.control_files.lock_write()
1043
 
        except:
1044
 
            self.repository.unlock()
1045
 
            raise
1046
 
 
1047
 
    def lock_read(self):
1048
 
        self.repository.lock_read()
1049
 
        try:
1050
 
            self.control_files.lock_read()
1051
 
        except:
1052
 
            self.repository.unlock()
1053
 
            raise
1054
 
 
1055
 
    def unlock(self):
1056
 
        # TODO: test for failed two phase locks. This is known broken.
1057
 
        try:
1058
 
            self.control_files.unlock()
1059
 
        finally:
1060
 
            self.repository.unlock()
1061
 
        
1062
 
    def peek_lock_mode(self):
1063
 
        if self.control_files._lock_count == 0:
1064
 
            return None
1065
 
        else:
1066
 
            return self.control_files._lock_mode
1067
 
 
1068
 
    def get_physical_lock_status(self):
1069
 
        return self.control_files.get_physical_lock_status()
 
823
        inv = self.get_inventory(self.last_revision())
 
824
        return inv.root.file_id
1070
825
 
1071
826
    @needs_read_lock
1072
 
    def print_file(self, file, revision_id):
 
827
    def print_file(self, file, revno):
1073
828
        """See Branch.print_file."""
1074
 
        return self.repository.print_file(file, revision_id)
 
829
        tree = self.revision_tree(self.get_rev_id(revno))
 
830
        # use inventory as it was in that revision
 
831
        file_id = tree.inventory.path2id(file)
 
832
        if not file_id:
 
833
            raise BzrError("%r is not present in revision %s" % (file, revno))
 
834
        tree.print_file(file_id)
1075
835
 
1076
836
    @needs_write_lock
1077
837
    def append_revision(self, *revision_ids):
1085
845
    @needs_write_lock
1086
846
    def set_revision_history(self, rev_history):
1087
847
        """See Branch.set_revision_history."""
1088
 
        self.control_files.put_utf8(
1089
 
            'revision-history', '\n'.join(rev_history))
1090
 
        transaction = self.get_transaction()
1091
 
        history = transaction.map.find_revision_history()
1092
 
        if history is not None:
1093
 
            # update the revision history in the identity map.
1094
 
            history[:] = list(rev_history)
1095
 
            # this call is disabled because revision_history is 
1096
 
            # not really an object yet, and the transaction is for objects.
1097
 
            # transaction.register_dirty(history)
 
848
        self.put_controlfile('revision-history', '\n'.join(rev_history))
 
849
 
 
850
    def has_revision(self, revision_id):
 
851
        """See Branch.has_revision."""
 
852
        return (revision_id is None
 
853
                or self.revision_store.has_id(revision_id))
 
854
 
 
855
    @needs_read_lock
 
856
    def _get_revision_xml_file(self, revision_id):
 
857
        if not revision_id or not isinstance(revision_id, basestring):
 
858
            raise InvalidRevisionId(revision_id=revision_id, branch=self)
 
859
        try:
 
860
            return self.revision_store.get(revision_id)
 
861
        except (IndexError, KeyError):
 
862
            raise bzrlib.errors.NoSuchRevision(self, revision_id)
 
863
 
 
864
    def get_revision_xml(self, revision_id):
 
865
        """See Branch.get_revision_xml."""
 
866
        return self._get_revision_xml_file(revision_id).read()
 
867
 
 
868
    def get_revision(self, revision_id):
 
869
        """See Branch.get_revision."""
 
870
        xml_file = self._get_revision_xml_file(revision_id)
 
871
 
 
872
        try:
 
873
            r = bzrlib.xml5.serializer_v5.read_revision(xml_file)
 
874
        except SyntaxError, e:
 
875
            raise bzrlib.errors.BzrError('failed to unpack revision_xml',
 
876
                                         [revision_id,
 
877
                                          str(e)])
 
878
            
 
879
        assert r.revision_id == revision_id
 
880
        return r
 
881
 
 
882
    def get_revision_sha1(self, revision_id):
 
883
        """See Branch.get_revision_sha1."""
 
884
        # In the future, revision entries will be signed. At that
 
885
        # point, it is probably best *not* to include the signature
 
886
        # in the revision hash. Because that lets you re-sign
 
887
        # the revision, (add signatures/remove signatures) and still
 
888
        # have all hash pointers stay consistent.
 
889
        # But for now, just hash the contents.
 
890
        return bzrlib.osutils.sha_file(self.get_revision_xml_file(revision_id))
 
891
 
 
892
    def get_ancestry(self, revision_id):
 
893
        """See Branch.get_ancestry."""
 
894
        if revision_id is None:
 
895
            return [None]
 
896
        w = self._get_inventory_weave()
 
897
        return [None] + map(w.idx_to_name,
 
898
                            w.inclusions([w.lookup(revision_id)]))
 
899
 
 
900
    def _get_inventory_weave(self):
 
901
        return self.control_weaves.get_weave('inventory',
 
902
                                             self.get_transaction())
 
903
 
 
904
    def get_inventory(self, revision_id):
 
905
        """See Branch.get_inventory."""
 
906
        xml = self.get_inventory_xml(revision_id)
 
907
        return bzrlib.xml5.serializer_v5.read_inventory_from_string(xml)
 
908
 
 
909
    def get_inventory_xml(self, revision_id):
 
910
        """See Branch.get_inventory_xml."""
 
911
        try:
 
912
            assert isinstance(revision_id, basestring), type(revision_id)
 
913
            iw = self._get_inventory_weave()
 
914
            return iw.get_text(iw.lookup(revision_id))
 
915
        except IndexError:
 
916
            raise bzrlib.errors.HistoryMissing(self, 'inventory', revision_id)
 
917
 
 
918
    def get_inventory_sha1(self, revision_id):
 
919
        """See Branch.get_inventory_sha1."""
 
920
        return self.get_revision(revision_id).inventory_sha1
 
921
 
 
922
    def get_revision_inventory(self, revision_id):
 
923
        """See Branch.get_revision_inventory."""
 
924
        # TODO: Unify this with get_inventory()
 
925
        # bzr 0.0.6 and later imposes the constraint that the inventory_id
 
926
        # must be the same as its revision, so this is trivial.
 
927
        if revision_id == None:
 
928
            # This does not make sense: if there is no revision,
 
929
            # then it is the current tree inventory surely ?!
 
930
            # and thus get_root_id() is something that looks at the last
 
931
            # commit on the branch, and the get_root_id is an inventory check.
 
932
            raise NotImplementedError
 
933
            # return Inventory(self.get_root_id())
1098
934
        else:
1099
 
            transaction.map.add_revision_history(rev_history)
1100
 
            # this call is disabled because revision_history is 
1101
 
            # not really an object yet, and the transaction is for objects.
1102
 
            # transaction.register_clean(history)
 
935
            return self.get_inventory(revision_id)
1103
936
 
1104
937
    @needs_read_lock
1105
938
    def revision_history(self):
1107
940
        transaction = self.get_transaction()
1108
941
        history = transaction.map.find_revision_history()
1109
942
        if history is not None:
1110
 
            # mutter("cache hit for revision-history in %s", self)
 
943
            mutter("cache hit for revision-history in %s", self)
1111
944
            return list(history)
1112
 
        decode_utf8 = cache_utf8.decode
1113
 
        history = [decode_utf8(l.rstrip('\r\n')) for l in
1114
 
                self.control_files.get('revision-history').readlines()]
 
945
        history = [l.rstrip('\r\n') for l in
 
946
                self.controlfile('revision-history', 'r').readlines()]
1115
947
        transaction.map.add_revision_history(history)
1116
948
        # this call is disabled because revision_history is 
1117
949
        # not really an object yet, and the transaction is for objects.
1118
950
        # transaction.register_clean(history, precious=True)
1119
951
        return list(history)
1120
952
 
1121
 
    @needs_write_lock
1122
 
    def generate_revision_history(self, revision_id, last_rev=None, 
1123
 
        other_branch=None):
1124
 
        """Create a new revision history that will finish with revision_id.
1125
 
        
1126
 
        :param revision_id: the new tip to use.
1127
 
        :param last_rev: The previous last_revision. If not None, then this
1128
 
            must be a ancestory of revision_id, or DivergedBranches is raised.
1129
 
        :param other_branch: The other branch that DivergedBranches should
1130
 
            raise with respect to.
1131
 
        """
1132
 
        # stop_revision must be a descendant of last_revision
1133
 
        stop_graph = self.repository.get_revision_graph(revision_id)
1134
 
        if last_rev is not None and last_rev not in stop_graph:
1135
 
            # our previous tip is not merged into stop_revision
1136
 
            raise errors.DivergedBranches(self, other_branch)
1137
 
        # make a new revision history from the graph
1138
 
        current_rev_id = revision_id
1139
 
        new_history = []
1140
 
        while current_rev_id not in (None, _mod_revision.NULL_REVISION):
1141
 
            new_history.append(current_rev_id)
1142
 
            current_rev_id_parents = stop_graph[current_rev_id]
1143
 
            try:
1144
 
                current_rev_id = current_rev_id_parents[0]
1145
 
            except IndexError:
1146
 
                current_rev_id = None
1147
 
        new_history.reverse()
1148
 
        self.set_revision_history(new_history)
1149
 
 
1150
 
    @needs_write_lock
1151
953
    def update_revisions(self, other, stop_revision=None):
1152
954
        """See Branch.update_revisions."""
1153
 
        other.lock_read()
 
955
        from bzrlib.fetch import greedy_fetch
 
956
        if stop_revision is None:
 
957
            stop_revision = other.last_revision()
 
958
        ### Should this be checking is_ancestor instead of revision_history?
 
959
        if (stop_revision is not None and 
 
960
            stop_revision in self.revision_history()):
 
961
            return
 
962
        greedy_fetch(to_branch=self, from_branch=other,
 
963
                     revision=stop_revision)
 
964
        pullable_revs = self.pullable_revisions(other, stop_revision)
 
965
        if len(pullable_revs) > 0:
 
966
            self.append_revision(*pullable_revs)
 
967
 
 
968
    def pullable_revisions(self, other, stop_revision):
 
969
        """See Branch.pullable_revisions."""
 
970
        other_revno = other.revision_id_to_revno(stop_revision)
1154
971
        try:
1155
 
            if stop_revision is None:
1156
 
                stop_revision = other.last_revision()
1157
 
                if stop_revision is None:
1158
 
                    # if there are no commits, we're done.
1159
 
                    return
1160
 
            # whats the current last revision, before we fetch [and change it
1161
 
            # possibly]
1162
 
            last_rev = self.last_revision()
1163
 
            # we fetch here regardless of whether we need to so that we pickup
1164
 
            # filled in ghosts.
1165
 
            self.fetch(other, stop_revision)
1166
 
            my_ancestry = self.repository.get_ancestry(last_rev)
1167
 
            if stop_revision in my_ancestry:
1168
 
                # last_revision is a descendant of stop_revision
1169
 
                return
1170
 
            self.generate_revision_history(stop_revision, last_rev=last_rev,
1171
 
                other_branch=other)
1172
 
        finally:
1173
 
            other.unlock()
1174
 
 
1175
 
    def basis_tree(self):
1176
 
        """See Branch.basis_tree."""
1177
 
        return self.repository.revision_tree(self.last_revision())
1178
 
 
1179
 
    @deprecated_method(zero_eight)
 
972
            return self.missing_revisions(other, other_revno)
 
973
        except DivergedBranches, e:
 
974
            try:
 
975
                pullable_revs = get_intervening_revisions(self.last_revision(),
 
976
                                                          stop_revision, self)
 
977
                assert self.last_revision() not in pullable_revs
 
978
                return pullable_revs
 
979
            except bzrlib.errors.NotAncestor:
 
980
                if is_ancestor(self.last_revision(), stop_revision, self):
 
981
                    return []
 
982
                else:
 
983
                    raise e
 
984
        
 
985
    def revision_tree(self, revision_id):
 
986
        """See Branch.revision_tree."""
 
987
        # TODO: refactor this to use an existing revision object
 
988
        # so we don't need to read it in twice.
 
989
        if revision_id == None or revision_id == NULL_REVISION:
 
990
            return EmptyTree()
 
991
        else:
 
992
            inv = self.get_revision_inventory(revision_id)
 
993
            return RevisionTree(self.weave_store, inv, revision_id)
 
994
 
1180
995
    def working_tree(self):
1181
 
        """Create a Working tree object for this branch."""
1182
 
 
1183
 
        from bzrlib.transport.local import LocalTransport
1184
 
        if (self.base.find('://') != -1 or 
1185
 
            not isinstance(self._transport, LocalTransport)):
 
996
        """See Branch.working_tree."""
 
997
        from bzrlib.workingtree import WorkingTree
 
998
        if self._transport.base.find('://') != -1:
1186
999
            raise NoWorkingTree(self.base)
1187
 
        return self.bzrdir.open_workingtree()
 
1000
        return WorkingTree(self.base, branch=self)
1188
1001
 
1189
1002
    @needs_write_lock
1190
 
    def pull(self, source, overwrite=False, stop_revision=None):
 
1003
    def pull(self, source, overwrite=False):
1191
1004
        """See Branch.pull."""
1192
1005
        source.lock_read()
1193
1006
        try:
1194
1007
            old_count = len(self.revision_history())
1195
1008
            try:
1196
 
                self.update_revisions(source, stop_revision)
 
1009
                self.update_revisions(source)
1197
1010
            except DivergedBranches:
1198
1011
                if not overwrite:
1199
1012
                    raise
1200
 
            if overwrite:
1201
1013
                self.set_revision_history(source.revision_history())
1202
1014
            new_count = len(self.revision_history())
1203
1015
            return new_count - old_count
1206
1018
 
1207
1019
    def get_parent(self):
1208
1020
        """See Branch.get_parent."""
1209
 
 
 
1021
        import errno
1210
1022
        _locs = ['parent', 'pull', 'x-pull']
1211
 
        assert self.base[-1] == '/'
1212
1023
        for l in _locs:
1213
1024
            try:
1214
 
                parent = self.control_files.get(l).read().strip('\n')
1215
 
            except NoSuchFile:
1216
 
                continue
1217
 
            # This is an old-format absolute path to a local branch
1218
 
            # turn it into a url
1219
 
            if parent.startswith('/'):
1220
 
                parent = urlutils.local_path_to_url(parent.decode('utf8'))
1221
 
            try:
1222
 
                return urlutils.join(self.base[:-1], parent)
1223
 
            except errors.InvalidURLJoin, e:
1224
 
                raise errors.InaccessibleParent(parent, self.base)
 
1025
                return self.controlfile(l, 'r').read().strip('\n')
 
1026
            except IOError, e:
 
1027
                if e.errno != errno.ENOENT:
 
1028
                    raise
1225
1029
        return None
1226
1030
 
1227
1031
    def get_push_location(self):
1228
1032
        """See Branch.get_push_location."""
1229
 
        push_loc = self.get_config().get_user_option('push_location')
 
1033
        config = bzrlib.config.BranchConfig(self)
 
1034
        push_loc = config.get_user_option('push_location')
1230
1035
        return push_loc
1231
1036
 
1232
1037
    def set_push_location(self, location):
1233
1038
        """See Branch.set_push_location."""
1234
 
        self.get_config().set_user_option(
1235
 
            'push_location', location,
1236
 
            store=_mod_config.STORE_LOCATION_NORECURSE)
 
1039
        config = bzrlib.config.LocationConfig(self.base)
 
1040
        config.set_user_option('push_location', location)
1237
1041
 
1238
1042
    @needs_write_lock
1239
1043
    def set_parent(self, url):
1240
1044
        """See Branch.set_parent."""
1241
1045
        # TODO: Maybe delete old location files?
1242
 
        # URLs should never be unicode, even on the local fs,
1243
 
        # FIXUP this and get_parent in a future branch format bump:
1244
 
        # read and rewrite the file, and have the new format code read
1245
 
        # using .get not .get_utf8. RBC 20060125
1246
 
        if url is None:
1247
 
            self.control_files._transport.delete('parent')
1248
 
        else:
1249
 
            if isinstance(url, unicode):
1250
 
                try: 
1251
 
                    url = url.encode('ascii')
1252
 
                except UnicodeEncodeError:
1253
 
                    raise bzrlib.errors.InvalidURL(url,
1254
 
                        "Urls must be 7-bit ascii, "
1255
 
                        "use bzrlib.urlutils.escape")
1256
 
                    
1257
 
            url = urlutils.relative_url(self.base, url)
1258
 
            self.control_files.put('parent', StringIO(url + '\n'))
 
1046
        from bzrlib.atomicfile import AtomicFile
 
1047
        f = AtomicFile(self.controlfilename('parent'))
 
1048
        try:
 
1049
            f.write(url + '\n')
 
1050
            f.commit()
 
1051
        finally:
 
1052
            f.close()
1259
1053
 
1260
 
    @deprecated_function(zero_nine)
1261
1054
    def tree_config(self):
1262
 
        """DEPRECATED; call get_config instead.  
1263
 
        TreeConfig has become part of BranchConfig."""
1264
1055
        return TreeConfig(self)
1265
1056
 
1266
 
 
1267
 
class BzrBranch5(BzrBranch):
1268
 
    """A format 5 branch. This supports new features over plan branches.
1269
 
 
1270
 
    It has support for a master_branch which is the data for bound branches.
 
1057
    def sign_revision(self, revision_id, gpg_strategy):
 
1058
        """See Branch.sign_revision."""
 
1059
        plaintext = Testament.from_revision(self, revision_id).as_short_text()
 
1060
        self.store_revision_signature(gpg_strategy, plaintext, revision_id)
 
1061
 
 
1062
    @needs_write_lock
 
1063
    def store_revision_signature(self, gpg_strategy, plaintext, revision_id):
 
1064
        """See Branch.store_revision_signature."""
 
1065
        self.revision_store.add(StringIO(gpg_strategy.sign(plaintext)), 
 
1066
                                revision_id, "sig")
 
1067
 
 
1068
 
 
1069
class ScratchBranch(BzrBranch):
 
1070
    """Special test class: a branch that cleans up after itself.
 
1071
 
 
1072
    >>> b = ScratchBranch()
 
1073
    >>> isdir(b.base)
 
1074
    True
 
1075
    >>> bd = b.base
 
1076
    >>> b._transport.__del__()
 
1077
    >>> isdir(bd)
 
1078
    False
1271
1079
    """
1272
1080
 
1273
 
    def __init__(self,
1274
 
                 _format,
1275
 
                 _control_files,
1276
 
                 a_bzrdir,
1277
 
                 _repository):
1278
 
        super(BzrBranch5, self).__init__(_format=_format,
1279
 
                                         _control_files=_control_files,
1280
 
                                         a_bzrdir=a_bzrdir,
1281
 
                                         _repository=_repository)
1282
 
        
1283
 
    @needs_write_lock
1284
 
    def pull(self, source, overwrite=False, stop_revision=None):
1285
 
        """Updates branch.pull to be bound branch aware."""
1286
 
        bound_location = self.get_bound_location()
1287
 
        if source.base != bound_location:
1288
 
            # not pulling from master, so we need to update master.
1289
 
            master_branch = self.get_master_branch()
1290
 
            if master_branch:
1291
 
                master_branch.pull(source)
1292
 
                source = master_branch
1293
 
        return super(BzrBranch5, self).pull(source, overwrite, stop_revision)
1294
 
 
1295
 
    def get_bound_location(self):
1296
 
        try:
1297
 
            return self.control_files.get_utf8('bound').read()[:-1]
1298
 
        except errors.NoSuchFile:
1299
 
            return None
1300
 
 
1301
 
    @needs_read_lock
1302
 
    def get_master_branch(self):
1303
 
        """Return the branch we are bound to.
1304
 
        
1305
 
        :return: Either a Branch, or None
1306
 
 
1307
 
        This could memoise the branch, but if thats done
1308
 
        it must be revalidated on each new lock.
1309
 
        So for now we just don't memoise it.
1310
 
        # RBC 20060304 review this decision.
1311
 
        """
1312
 
        bound_loc = self.get_bound_location()
1313
 
        if not bound_loc:
1314
 
            return None
1315
 
        try:
1316
 
            return Branch.open(bound_loc)
1317
 
        except (errors.NotBranchError, errors.ConnectionError), e:
1318
 
            raise errors.BoundBranchConnectionFailure(
1319
 
                    self, bound_loc, e)
1320
 
 
1321
 
    @needs_write_lock
1322
 
    def set_bound_location(self, location):
1323
 
        """Set the target where this branch is bound to.
1324
 
 
1325
 
        :param location: URL to the target branch
1326
 
        """
1327
 
        if location:
1328
 
            self.control_files.put_utf8('bound', location+'\n')
 
1081
    def __init__(self, files=[], dirs=[], transport=None):
 
1082
        """Make a test branch.
 
1083
 
 
1084
        This creates a temporary directory and runs init-tree in it.
 
1085
 
 
1086
        If any files are listed, they are created in the working copy.
 
1087
        """
 
1088
        if transport is None:
 
1089
            transport = bzrlib.transport.local.ScratchTransport()
 
1090
            super(ScratchBranch, self).__init__(transport, init=True)
1329
1091
        else:
1330
 
            try:
1331
 
                self.control_files._transport.delete('bound')
1332
 
            except NoSuchFile:
1333
 
                return False
 
1092
            super(ScratchBranch, self).__init__(transport)
 
1093
 
 
1094
        for d in dirs:
 
1095
            self._transport.mkdir(d)
 
1096
            
 
1097
        for f in files:
 
1098
            self._transport.put(f, 'content of %s' % f)
 
1099
 
 
1100
 
 
1101
    def clone(self):
 
1102
        """
 
1103
        >>> orig = ScratchBranch(files=["file1", "file2"])
 
1104
        >>> clone = orig.clone()
 
1105
        >>> if os.name != 'nt':
 
1106
        ...   os.path.samefile(orig.base, clone.base)
 
1107
        ... else:
 
1108
        ...   orig.base == clone.base
 
1109
        ...
 
1110
        False
 
1111
        >>> os.path.isfile(os.path.join(clone.base, "file1"))
 
1112
        True
 
1113
        """
 
1114
        from shutil import copytree
 
1115
        from tempfile import mkdtemp
 
1116
        base = mkdtemp()
 
1117
        os.rmdir(base)
 
1118
        copytree(self.base, base, symlinks=True)
 
1119
        return ScratchBranch(
 
1120
            transport=bzrlib.transport.local.ScratchTransport(base))
 
1121
    
 
1122
 
 
1123
######################################################################
 
1124
# predicates
 
1125
 
 
1126
 
 
1127
def is_control_file(filename):
 
1128
    ## FIXME: better check
 
1129
    filename = os.path.normpath(filename)
 
1130
    while filename != '':
 
1131
        head, tail = os.path.split(filename)
 
1132
        ## mutter('check %r for control file' % ((head, tail), ))
 
1133
        if tail == bzrlib.BZRDIR:
1334
1134
            return True
1335
 
 
1336
 
    @needs_write_lock
1337
 
    def bind(self, other):
1338
 
        """Bind this branch to the branch other.
1339
 
 
1340
 
        This does not push or pull data between the branches, though it does
1341
 
        check for divergence to raise an error when the branches are not
1342
 
        either the same, or one a prefix of the other. That behaviour may not
1343
 
        be useful, so that check may be removed in future.
1344
 
        
1345
 
        :param other: The branch to bind to
1346
 
        :type other: Branch
1347
 
        """
1348
 
        # TODO: jam 20051230 Consider checking if the target is bound
1349
 
        #       It is debatable whether you should be able to bind to
1350
 
        #       a branch which is itself bound.
1351
 
        #       Committing is obviously forbidden,
1352
 
        #       but binding itself may not be.
1353
 
        #       Since we *have* to check at commit time, we don't
1354
 
        #       *need* to check here
1355
 
 
1356
 
        # we want to raise diverged if:
1357
 
        # last_rev is not in the other_last_rev history, AND
1358
 
        # other_last_rev is not in our history, and do it without pulling
1359
 
        # history around
1360
 
        last_rev = self.last_revision()
1361
 
        if last_rev is not None:
1362
 
            other.lock_read()
1363
 
            try:
1364
 
                other_last_rev = other.last_revision()
1365
 
                if other_last_rev is not None:
1366
 
                    # neither branch is new, we have to do some work to
1367
 
                    # ascertain diversion.
1368
 
                    remote_graph = other.repository.get_revision_graph(
1369
 
                        other_last_rev)
1370
 
                    local_graph = self.repository.get_revision_graph(last_rev)
1371
 
                    if (last_rev not in remote_graph and
1372
 
                        other_last_rev not in local_graph):
1373
 
                        raise errors.DivergedBranches(self, other)
1374
 
            finally:
1375
 
                other.unlock()
1376
 
        self.set_bound_location(other.base)
1377
 
 
1378
 
    @needs_write_lock
1379
 
    def unbind(self):
1380
 
        """If bound, unbind"""
1381
 
        return self.set_bound_location(None)
1382
 
 
1383
 
    @needs_write_lock
1384
 
    def update(self):
1385
 
        """Synchronise this branch with the master branch if any. 
1386
 
 
1387
 
        :return: None or the last_revision that was pivoted out during the
1388
 
                 update.
1389
 
        """
1390
 
        master = self.get_master_branch()
1391
 
        if master is not None:
1392
 
            old_tip = self.last_revision()
1393
 
            self.pull(master, overwrite=True)
1394
 
            if old_tip in self.repository.get_ancestry(self.last_revision()):
1395
 
                return None
1396
 
            return old_tip
1397
 
        return None
1398
 
 
1399
 
 
1400
 
class BranchTestProviderAdapter(object):
1401
 
    """A tool to generate a suite testing multiple branch formats at once.
1402
 
 
1403
 
    This is done by copying the test once for each transport and injecting
1404
 
    the transport_server, transport_readonly_server, and branch_format
1405
 
    classes into each copy. Each copy is also given a new id() to make it
1406
 
    easy to identify.
1407
 
    """
1408
 
 
1409
 
    def __init__(self, transport_server, transport_readonly_server, formats):
1410
 
        self._transport_server = transport_server
1411
 
        self._transport_readonly_server = transport_readonly_server
1412
 
        self._formats = formats
1413
 
    
1414
 
    def adapt(self, test):
1415
 
        result = TestSuite()
1416
 
        for branch_format, bzrdir_format in self._formats:
1417
 
            new_test = deepcopy(test)
1418
 
            new_test.transport_server = self._transport_server
1419
 
            new_test.transport_readonly_server = self._transport_readonly_server
1420
 
            new_test.bzrdir_format = bzrdir_format
1421
 
            new_test.branch_format = branch_format
1422
 
            def make_new_test_id():
1423
 
                new_id = "%s(%s)" % (new_test.id(), branch_format.__class__.__name__)
1424
 
                return lambda: new_id
1425
 
            new_test.id = make_new_test_id()
1426
 
            result.addTest(new_test)
1427
 
        return result
1428
 
 
1429
 
 
1430
 
class BranchCheckResult(object):
1431
 
    """Results of checking branch consistency.
1432
 
 
1433
 
    :see: Branch.check
1434
 
    """
1435
 
 
1436
 
    def __init__(self, branch):
1437
 
        self.branch = branch
1438
 
 
1439
 
    def report_results(self, verbose):
1440
 
        """Report the check results via trace.note.
1441
 
        
1442
 
        :param verbose: Requests more detailed display of what was checked,
1443
 
            if any.
1444
 
        """
1445
 
        note('checked branch %s format %s',
1446
 
             self.branch.base,
1447
 
             self.branch._format)
1448
 
 
1449
 
 
1450
 
######################################################################
1451
 
# predicates
1452
 
 
1453
 
 
1454
 
@deprecated_function(zero_eight)
1455
 
def is_control_file(*args, **kwargs):
1456
 
    """See bzrlib.workingtree.is_control_file."""
1457
 
    from bzrlib import workingtree
1458
 
    return workingtree.is_control_file(*args, **kwargs)
 
1135
        if filename == head:
 
1136
            break
 
1137
        filename = head
 
1138
    return False