~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2007-02-22 14:45:05 UTC
  • mfrom: (2294.1.11 utf8_file_ids)
  • Revision ID: pqm@pqm.ubuntu.com-20070222144505-5f7551602cad9332
(John Arbash Meinel, r=robert) Update apis to expect UTF-8 file ids instead of Unicode

Show diffs side-by-side

added added

removed removed

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