~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/branch.py

  • Committer: Martin Pool
  • Date: 2005-05-03 08:00:27 UTC
  • Revision ID: mbp@sourcefrog.net-20050503080027-908edb5b39982198
doc

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 Canonical Ltd
 
2
 
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
18
 
from copy import deepcopy
19
 
from cStringIO import StringIO
20
 
import errno
21
 
import os
22
 
import shutil
23
 
import sys
24
 
from unittest import TestSuite
25
 
from warnings import warn
 
18
from sets import Set
 
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
26
23
 
27
24
import bzrlib
28
 
import bzrlib.bzrdir as bzrdir
29
 
from bzrlib.config import TreeConfig
30
 
from bzrlib.decorators import needs_read_lock, needs_write_lock
31
 
from bzrlib.delta import compare_trees
32
 
import bzrlib.errors as errors
33
 
from bzrlib.errors import (BzrError, InvalidRevisionNumber, InvalidRevisionId,
34
 
                           NoSuchRevision, HistoryMissing, NotBranchError,
35
 
                           DivergedBranches, LockError,
36
 
                           UninitializableFormat,
37
 
                           UnlistableStore,
38
 
                           UnlistableBranch, NoSuchFile, NotVersionedError,
39
 
                           NoWorkingTree)
40
 
import bzrlib.inventory as inventory
41
 
from bzrlib.inventory import Inventory
42
 
from bzrlib.lockable_files import LockableFiles, TransportLock
43
 
from bzrlib.lockdir import LockDir
44
 
from bzrlib.osutils import (isdir, quotefn,
45
 
                            rename, splitpath, sha_file,
46
 
                            file_kind, abspath, normpath, pathjoin,
47
 
                            safe_unicode,
48
 
                            rmtree,
49
 
                            )
50
 
from bzrlib.textui import show_status
51
 
from bzrlib.trace import mutter, note
52
 
from bzrlib.tree import EmptyTree, RevisionTree
53
 
from bzrlib.repository import Repository
54
 
from bzrlib.revision import (
55
 
                             is_ancestor,
56
 
                             NULL_REVISION,
57
 
                             Revision,
58
 
                             )
59
 
from bzrlib.store import copy_all
60
 
from bzrlib.symbol_versioning import *
61
 
import bzrlib.transactions as transactions
62
 
from bzrlib.transport import Transport, get_transport
63
 
from bzrlib.tree import EmptyTree, RevisionTree
64
 
import bzrlib.ui
65
 
import bzrlib.xml5
66
 
 
67
 
 
68
 
BZR_BRANCH_FORMAT_4 = "Bazaar-NG branch, format 0.0.4\n"
69
 
BZR_BRANCH_FORMAT_5 = "Bazaar-NG branch, format 5\n"
70
 
BZR_BRANCH_FORMAT_6 = "Bazaar-NG branch, format 6\n"
71
 
 
72
 
 
73
 
# TODO: Maybe include checks for common corruption of newlines, etc?
74
 
 
75
 
# TODO: Some operations like log might retrieve the same revisions
76
 
# repeatedly to calculate deltas.  We could perhaps have a weakref
77
 
# cache in memory to make this faster.  In general anything can be
78
 
# cached in memory between lock and unlock operations. .. nb thats
79
 
# what the transaction identity map provides
 
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
    
80
68
 
81
69
 
82
70
######################################################################
83
71
# branch objects
84
72
 
85
 
class Branch(object):
 
73
class Branch:
86
74
    """Branch holding a history of revisions.
87
75
 
88
 
    base
89
 
        Base directory/url of the branch.
 
76
    TODO: Perhaps use different stores for different classes of object,
 
77
           so that we can keep track of how much space each one uses,
 
78
           or garbage-collect them.
 
79
 
 
80
    TODO: Add a RemoteBranch subclass.  For the basic case of read-only
 
81
           HTTP access this should be very easy by, 
 
82
           just redirecting controlfile access into HTTP requests.
 
83
           We would need a RemoteStore working similarly.
 
84
 
 
85
    TODO: Keep the on-disk branch locked while the object exists.
 
86
 
 
87
    TODO: mkdir() method.
90
88
    """
91
 
    # this is really an instance variable - FIXME move it there
92
 
    # - RBC 20060112
93
 
    base = None
94
 
 
95
 
    def __init__(self, *ignored, **ignored_too):
96
 
        raise NotImplementedError('The Branch class is abstract')
97
 
 
98
 
    def break_lock(self):
99
 
        """Break a lock if one is present from another instance.
100
 
 
101
 
        Uses the ui factory to ask for confirmation if the lock may be from
102
 
        an active process.
103
 
 
104
 
        This will probe the repository for its lock as well.
105
 
        """
106
 
        self.control_files.break_lock()
107
 
        self.repository.break_lock()
108
 
        master = self.get_master_branch()
109
 
        if master is not None:
110
 
            master.break_lock()
111
 
 
112
 
    @staticmethod
113
 
    @deprecated_method(zero_eight)
114
 
    def open_downlevel(base):
115
 
        """Open a branch which may be of an old format."""
116
 
        return Branch.open(base, _unsupported=True)
117
 
        
118
 
    @staticmethod
119
 
    def open(base, _unsupported=False):
120
 
        """Open the repository rooted at base.
121
 
 
122
 
        For instance, if the repository is at URL/.bzr/repository,
123
 
        Repository.open(URL) -> a Repository instance.
124
 
        """
125
 
        control = bzrdir.BzrDir.open(base, _unsupported)
126
 
        return control.open_branch(_unsupported)
127
 
 
128
 
    @staticmethod
129
 
    def open_containing(url):
130
 
        """Open an existing branch which contains url.
131
 
        
132
 
        This probes for a branch at url, and searches upwards from there.
133
 
 
134
 
        Basically we keep looking up until we find the control directory or
135
 
        run into the root.  If there isn't one, raises NotBranchError.
136
 
        If there is one and it is either an unrecognised format or an unsupported 
137
 
        format, UnknownFormatError or UnsupportedFormatError are raised.
138
 
        If there is one, it is returned, along with the unused portion of url.
139
 
        """
140
 
        control, relpath = bzrdir.BzrDir.open_containing(url)
141
 
        return control.open_branch(), relpath
142
 
 
143
 
    @staticmethod
144
 
    @deprecated_function(zero_eight)
145
 
    def initialize(base):
146
 
        """Create a new working tree and branch, rooted at 'base' (url)
147
 
 
148
 
        NOTE: This will soon be deprecated in favour of creation
149
 
        through a BzrDir.
150
 
        """
151
 
        return bzrdir.BzrDir.create_standalone_workingtree(base).branch
152
 
 
153
 
    def setup_caching(self, cache_root):
154
 
        """Subclasses that care about caching should override this, and set
155
 
        up cached stores located under cache_root.
156
 
        """
157
 
        # seems to be unused, 2006-01-13 mbp
158
 
        warn('%s is deprecated' % self.setup_caching)
159
 
        self.cache_root = cache_root
160
 
 
161
 
    def _get_nick(self):
162
 
        cfg = self.tree_config()
163
 
        return cfg.get_option(u"nickname", default=self.base.split('/')[-2])
164
 
 
165
 
    def _set_nick(self, nick):
166
 
        cfg = self.tree_config()
167
 
        cfg.set_option(nick, "nickname")
168
 
        assert cfg.get_option("nickname") == nick
169
 
 
170
 
    nick = property(_get_nick, _set_nick)
171
 
 
172
 
    def is_locked(self):
173
 
        raise NotImplementedError('is_locked is abstract')
174
 
 
175
 
    def lock_write(self):
176
 
        raise NotImplementedError('lock_write is abstract')
177
 
 
178
 
    def lock_read(self):
179
 
        raise NotImplementedError('lock_read is abstract')
180
 
 
181
 
    def unlock(self):
182
 
        raise NotImplementedError('unlock is abstract')
183
 
 
184
 
    def peek_lock_mode(self):
185
 
        """Return lock mode for the Branch: 'r', 'w' or None"""
186
 
        raise NotImplementedError(self.peek_lock_mode)
187
 
 
188
 
    def get_physical_lock_status(self):
189
 
        raise NotImplementedError('get_physical_lock_status is abstract')
 
89
    def __init__(self, base, init=False, find_root=True):
 
90
        """Create new branch object at a particular location.
 
91
 
 
92
        base -- Base directory for the branch.
 
93
        
 
94
        init -- If True, create new control files in a previously
 
95
             unversioned directory.  If False, the branch must already
 
96
             be versioned.
 
97
 
 
98
        find_root -- If true and init is false, find the root of the
 
99
             existing branch containing base.
 
100
 
 
101
        In the test suite, creation of new trees is tested using the
 
102
        `ScratchBranch` class.
 
103
        """
 
104
        if init:
 
105
            self.base = os.path.realpath(base)
 
106
            self._make_control()
 
107
        elif find_root:
 
108
            self.base = find_branch_root(base)
 
109
        else:
 
110
            self.base = os.path.realpath(base)
 
111
            if not isdir(self.controlfilename('.')):
 
112
                bailout("not a bzr branch: %s" % quotefn(base),
 
113
                        ['use "bzr init" to initialize a new working tree',
 
114
                         'current bzr can only operate from top-of-tree'])
 
115
        self._check_format()
 
116
 
 
117
        self.text_store = ImmutableStore(self.controlfilename('text-store'))
 
118
        self.revision_store = ImmutableStore(self.controlfilename('revision-store'))
 
119
        self.inventory_store = ImmutableStore(self.controlfilename('inventory-store'))
 
120
 
 
121
 
 
122
    def __str__(self):
 
123
        return '%s(%r)' % (self.__class__.__name__, self.base)
 
124
 
 
125
 
 
126
    __repr__ = __str__
 
127
 
190
128
 
191
129
    def abspath(self, name):
192
 
        """Return absolute filename for something in the branch
193
 
        
194
 
        XXX: Robert Collins 20051017 what is this used for? why is it a branch
195
 
        method and not a tree method.
196
 
        """
197
 
        raise NotImplementedError('abspath is abstract')
198
 
 
199
 
    def bind(self, other):
200
 
        """Bind the local branch the other branch.
201
 
 
202
 
        :param other: The branch to bind to
203
 
        :type other: Branch
204
 
        """
205
 
        raise errors.UpgradeRequired(self.base)
206
 
 
207
 
    @needs_write_lock
208
 
    def fetch(self, from_branch, last_revision=None, pb=None):
209
 
        """Copy revisions from from_branch into this branch.
210
 
 
211
 
        :param from_branch: Where to copy from.
212
 
        :param last_revision: What revision to stop at (None for at the end
213
 
                              of the branch.
214
 
        :param pb: An optional progress bar to use.
215
 
 
216
 
        Returns the copied revision count and the failed revisions in a tuple:
217
 
        (copied, failures).
218
 
        """
219
 
        if self.base == from_branch.base:
220
 
            return (0, [])
221
 
        if pb is None:
222
 
            nested_pb = bzrlib.ui.ui_factory.nested_progress_bar()
223
 
            pb = nested_pb
 
130
        """Return absolute filename for something in the branch"""
 
131
        return os.path.join(self.base, name)
 
132
 
 
133
 
 
134
    def relpath(self, path):
 
135
        """Return path relative to this branch of something inside it.
 
136
 
 
137
        Raises an error if path is not in this branch."""
 
138
        rp = os.path.realpath(path)
 
139
        # FIXME: windows
 
140
        if not rp.startswith(self.base):
 
141
            bailout("path %r is not within branch %r" % (rp, self.base))
 
142
        rp = rp[len(self.base):]
 
143
        rp = rp.lstrip(os.sep)
 
144
        return rp
 
145
 
 
146
 
 
147
    def controlfilename(self, file_or_path):
 
148
        """Return location relative to branch."""
 
149
        if isinstance(file_or_path, types.StringTypes):
 
150
            file_or_path = [file_or_path]
 
151
        return os.path.join(self.base, bzrlib.BZRDIR, *file_or_path)
 
152
 
 
153
 
 
154
    def controlfile(self, file_or_path, mode='r'):
 
155
        """Open a control file for this branch.
 
156
 
 
157
        There are two classes of file in the control directory: text
 
158
        and binary.  binary files are untranslated byte streams.  Text
 
159
        control files are stored with Unix newlines and in UTF-8, even
 
160
        if the platform or locale defaults are different.
 
161
        """
 
162
 
 
163
        fn = self.controlfilename(file_or_path)
 
164
 
 
165
        if mode == 'rb' or mode == 'wb':
 
166
            return file(fn, mode)
 
167
        elif mode == 'r' or mode == 'w':
 
168
            # open in binary mode anyhow so there's no newline translation;
 
169
            # codecs uses line buffering by default; don't want that.
 
170
            import codecs
 
171
            return codecs.open(fn, mode + 'b', 'utf-8',
 
172
                               buffering=60000)
224
173
        else:
225
 
            nested_pb = None
226
 
 
227
 
        from_branch.lock_read()
228
 
        try:
229
 
            if last_revision is None:
230
 
                pb.update('get source history')
231
 
                from_history = from_branch.revision_history()
232
 
                if from_history:
233
 
                    last_revision = from_history[-1]
234
 
                else:
235
 
                    # no history in the source branch
236
 
                    last_revision = NULL_REVISION
237
 
            return self.repository.fetch(from_branch.repository,
238
 
                                         revision_id=last_revision,
239
 
                                         pb=nested_pb)
240
 
        finally:
241
 
            if nested_pb is not None:
242
 
                nested_pb.finished()
243
 
            from_branch.unlock()
244
 
 
245
 
    def get_bound_location(self):
246
 
        """Return the URL of the branch we are bound to.
247
 
 
248
 
        Older format branches cannot bind, please be sure to use a metadir
249
 
        branch.
250
 
        """
251
 
        return None
252
 
 
253
 
    def get_master_branch(self):
254
 
        """Return the branch we are bound to.
 
174
            raise BzrError("invalid controlfile mode %r" % mode)
 
175
 
 
176
 
 
177
 
 
178
    def _make_control(self):
 
179
        os.mkdir(self.controlfilename([]))
 
180
        self.controlfile('README', 'w').write(
 
181
            "This is a Bazaar-NG control directory.\n"
 
182
            "Do not change any files in this directory.")
 
183
        self.controlfile('branch-format', 'w').write(BZR_BRANCH_FORMAT)
 
184
        for d in ('text-store', 'inventory-store', 'revision-store'):
 
185
            os.mkdir(self.controlfilename(d))
 
186
        for f in ('revision-history', 'merged-patches',
 
187
                  'pending-merged-patches', 'branch-name'):
 
188
            self.controlfile(f, 'w').write('')
 
189
        mutter('created control directory in ' + self.base)
 
190
        Inventory().write_xml(self.controlfile('inventory','w'))
 
191
 
 
192
 
 
193
    def _check_format(self):
 
194
        """Check this branch format is supported.
 
195
 
 
196
        The current tool only supports the current unstable format.
 
197
 
 
198
        In the future, we might need different in-memory Branch
 
199
        classes to support downlevel branches.  But not yet.
 
200
        """
 
201
        # This ignores newlines so that we can open branches created
 
202
        # on Windows from Linux and so on.  I think it might be better
 
203
        # to always make all internal files in unix format.
 
204
        fmt = self.controlfile('branch-format', 'r').read()
 
205
        fmt.replace('\r\n', '')
 
206
        if fmt != BZR_BRANCH_FORMAT:
 
207
            bailout('sorry, branch format %r not supported' % fmt,
 
208
                    ['use a different bzr version',
 
209
                     'or remove the .bzr directory and "bzr init" again'])
 
210
 
 
211
 
 
212
    def read_working_inventory(self):
 
213
        """Read the working inventory."""
 
214
        before = time.time()
 
215
        # ElementTree does its own conversion from UTF-8, so open in
 
216
        # binary.
 
217
        inv = Inventory.read_xml(self.controlfile('inventory', 'rb'))
 
218
        mutter("loaded inventory of %d items in %f"
 
219
               % (len(inv), time.time() - before))
 
220
        return inv
 
221
 
 
222
 
 
223
    def _write_inventory(self, inv):
 
224
        """Update the working inventory.
 
225
 
 
226
        That is to say, the inventory describing changes underway, that
 
227
        will be committed to the next revision.
 
228
        """
 
229
        ## TODO: factor out to atomicfile?  is rename safe on windows?
 
230
        ## TODO: Maybe some kind of clean/dirty marker on inventory?
 
231
        tmpfname = self.controlfilename('inventory.tmp')
 
232
        tmpf = file(tmpfname, 'wb')
 
233
        inv.write_xml(tmpf)
 
234
        tmpf.close()
 
235
        inv_fname = self.controlfilename('inventory')
 
236
        if sys.platform == 'win32':
 
237
            os.remove(inv_fname)
 
238
        os.rename(tmpfname, inv_fname)
 
239
        mutter('wrote working inventory')
 
240
 
 
241
 
 
242
    inventory = property(read_working_inventory, _write_inventory, None,
 
243
                         """Inventory for the working copy.""")
 
244
 
 
245
 
 
246
    def add(self, files, verbose=False):
 
247
        """Make files versioned.
 
248
 
 
249
        Note that the command line normally calls smart_add instead.
 
250
 
 
251
        This puts the files in the Added state, so that they will be
 
252
        recorded by the next commit.
 
253
 
 
254
        TODO: Perhaps have an option to add the ids even if the files do
 
255
               not (yet) exist.
 
256
 
 
257
        TODO: Perhaps return the ids of the files?  But then again it
 
258
               is easy to retrieve them if they're needed.
 
259
 
 
260
        TODO: Option to specify file id.
 
261
 
 
262
        TODO: Adding a directory should optionally recurse down and
 
263
               add all non-ignored children.  Perhaps do that in a
 
264
               higher-level method.
 
265
 
 
266
        >>> b = ScratchBranch(files=['foo'])
 
267
        >>> 'foo' in b.unknowns()
 
268
        True
 
269
        >>> b.show_status()
 
270
        ?       foo
 
271
        >>> b.add('foo')
 
272
        >>> 'foo' in b.unknowns()
 
273
        False
 
274
        >>> bool(b.inventory.path2id('foo'))
 
275
        True
 
276
        >>> b.show_status()
 
277
        A       foo
 
278
 
 
279
        >>> b.add('foo')
 
280
        Traceback (most recent call last):
 
281
        ...
 
282
        BzrError: ('foo is already versioned', [])
 
283
 
 
284
        >>> b.add(['nothere'])
 
285
        Traceback (most recent call last):
 
286
        BzrError: ('cannot add: not a regular file or directory: nothere', [])
 
287
        """
 
288
 
 
289
        # TODO: Re-adding a file that is removed in the working copy
 
290
        # should probably put it back with the previous ID.
 
291
        if isinstance(files, types.StringTypes):
 
292
            files = [files]
255
293
        
256
 
        :return: Either a Branch, or None
257
 
        """
258
 
        return None
259
 
 
260
 
    def get_root_id(self):
261
 
        """Return the id of this branches root"""
262
 
        raise NotImplementedError('get_root_id is abstract')
263
 
 
264
 
    def print_file(self, file, revision_id):
 
294
        inv = self.read_working_inventory()
 
295
        for f in files:
 
296
            if is_control_file(f):
 
297
                bailout("cannot add control file %s" % quotefn(f))
 
298
 
 
299
            fp = splitpath(f)
 
300
 
 
301
            if len(fp) == 0:
 
302
                bailout("cannot add top-level %r" % f)
 
303
                
 
304
            fullpath = os.path.normpath(self.abspath(f))
 
305
 
 
306
            try:
 
307
                kind = file_kind(fullpath)
 
308
            except OSError:
 
309
                # maybe something better?
 
310
                bailout('cannot add: not a regular file or directory: %s' % quotefn(f))
 
311
            
 
312
            if kind != 'file' and kind != 'directory':
 
313
                bailout('cannot add: not a regular file or directory: %s' % quotefn(f))
 
314
 
 
315
            file_id = gen_file_id(f)
 
316
            inv.add_path(f, kind=kind, file_id=file_id)
 
317
 
 
318
            if verbose:
 
319
                show_status('A', kind, quotefn(f))
 
320
                
 
321
            mutter("add file %s file_id:{%s} kind=%r" % (f, file_id, kind))
 
322
            
 
323
        self._write_inventory(inv)
 
324
 
 
325
 
 
326
    def print_file(self, file, revno):
265
327
        """Print `file` to stdout."""
266
 
        raise NotImplementedError('print_file is abstract')
267
 
 
268
 
    def append_revision(self, *revision_ids):
269
 
        raise NotImplementedError('append_revision is abstract')
270
 
 
271
 
    def set_revision_history(self, rev_history):
272
 
        raise NotImplementedError('set_revision_history is abstract')
 
328
        tree = self.revision_tree(self.lookup_revision(revno))
 
329
        # use inventory as it was in that revision
 
330
        file_id = tree.inventory.path2id(file)
 
331
        if not file_id:
 
332
            bailout("%r is not present in revision %d" % (file, revno))
 
333
        tree.print_file(file_id)
 
334
        
 
335
 
 
336
    def remove(self, files, verbose=False):
 
337
        """Mark nominated files for removal from the inventory.
 
338
 
 
339
        This does not remove their text.  This does not run on 
 
340
 
 
341
        TODO: Refuse to remove modified files unless --force is given?
 
342
 
 
343
        >>> b = ScratchBranch(files=['foo'])
 
344
        >>> b.add('foo')
 
345
        >>> b.inventory.has_filename('foo')
 
346
        True
 
347
        >>> b.remove('foo')
 
348
        >>> b.working_tree().has_filename('foo')
 
349
        True
 
350
        >>> b.inventory.has_filename('foo')
 
351
        False
 
352
        
 
353
        >>> b = ScratchBranch(files=['foo'])
 
354
        >>> b.add('foo')
 
355
        >>> b.commit('one')
 
356
        >>> b.remove('foo')
 
357
        >>> b.commit('two')
 
358
        >>> b.inventory.has_filename('foo') 
 
359
        False
 
360
        >>> b.basis_tree().has_filename('foo') 
 
361
        False
 
362
        >>> b.working_tree().has_filename('foo') 
 
363
        True
 
364
 
 
365
        TODO: Do something useful with directories.
 
366
 
 
367
        TODO: Should this remove the text or not?  Tough call; not
 
368
        removing may be useful and the user can just use use rm, and
 
369
        is the opposite of add.  Removing it is consistent with most
 
370
        other tools.  Maybe an option.
 
371
        """
 
372
        ## TODO: Normalize names
 
373
        ## TODO: Remove nested loops; better scalability
 
374
 
 
375
        if isinstance(files, types.StringTypes):
 
376
            files = [files]
 
377
        
 
378
        tree = self.working_tree()
 
379
        inv = tree.inventory
 
380
 
 
381
        # do this before any modifications
 
382
        for f in files:
 
383
            fid = inv.path2id(f)
 
384
            if not fid:
 
385
                bailout("cannot remove unversioned file %s" % quotefn(f))
 
386
            mutter("remove inventory entry %s {%s}" % (quotefn(f), fid))
 
387
            if verbose:
 
388
                # having remove it, it must be either ignored or unknown
 
389
                if tree.is_ignored(f):
 
390
                    new_status = 'I'
 
391
                else:
 
392
                    new_status = '?'
 
393
                show_status(new_status, inv[fid].kind, quotefn(f))
 
394
            del inv[fid]
 
395
 
 
396
        self._write_inventory(inv)
 
397
 
 
398
 
 
399
    def unknowns(self):
 
400
        """Return all unknown files.
 
401
 
 
402
        These are files in the working directory that are not versioned or
 
403
        control files or ignored.
 
404
        
 
405
        >>> b = ScratchBranch(files=['foo', 'foo~'])
 
406
        >>> list(b.unknowns())
 
407
        ['foo']
 
408
        >>> b.add('foo')
 
409
        >>> list(b.unknowns())
 
410
        []
 
411
        >>> b.remove('foo')
 
412
        >>> list(b.unknowns())
 
413
        ['foo']
 
414
        """
 
415
        return self.working_tree().unknowns()
 
416
 
 
417
 
 
418
    def commit(self, message, timestamp=None, timezone=None,
 
419
               committer=None,
 
420
               verbose=False):
 
421
        """Commit working copy as a new revision.
 
422
        
 
423
        The basic approach is to add all the file texts into the
 
424
        store, then the inventory, then make a new revision pointing
 
425
        to that inventory and store that.
 
426
        
 
427
        This is not quite safe if the working copy changes during the
 
428
        commit; for the moment that is simply not allowed.  A better
 
429
        approach is to make a temporary copy of the files before
 
430
        computing their hashes, and then add those hashes in turn to
 
431
        the inventory.  This should mean at least that there are no
 
432
        broken hash pointers.  There is no way we can get a snapshot
 
433
        of the whole directory at an instant.  This would also have to
 
434
        be robust against files disappearing, moving, etc.  So the
 
435
        whole thing is a bit hard.
 
436
 
 
437
        timestamp -- if not None, seconds-since-epoch for a
 
438
             postdated/predated commit.
 
439
        """
 
440
 
 
441
        ## TODO: Show branch names
 
442
 
 
443
        # TODO: Don't commit if there are no changes, unless forced?
 
444
 
 
445
        # First walk over the working inventory; and both update that
 
446
        # and also build a new revision inventory.  The revision
 
447
        # inventory needs to hold the text-id, sha1 and size of the
 
448
        # actual file versions committed in the revision.  (These are
 
449
        # not present in the working inventory.)  We also need to
 
450
        # detect missing/deleted files, and remove them from the
 
451
        # working inventory.
 
452
 
 
453
        work_inv = self.read_working_inventory()
 
454
        inv = Inventory()
 
455
        basis = self.basis_tree()
 
456
        basis_inv = basis.inventory
 
457
        missing_ids = []
 
458
        for path, entry in work_inv.iter_entries():
 
459
            ## TODO: Cope with files that have gone missing.
 
460
 
 
461
            ## TODO: Check that the file kind has not changed from the previous
 
462
            ## revision of this file (if any).
 
463
 
 
464
            entry = entry.copy()
 
465
 
 
466
            p = self.abspath(path)
 
467
            file_id = entry.file_id
 
468
            mutter('commit prep file %s, id %r ' % (p, file_id))
 
469
 
 
470
            if not os.path.exists(p):
 
471
                mutter("    file is missing, removing from inventory")
 
472
                if verbose:
 
473
                    show_status('D', entry.kind, quotefn(path))
 
474
                missing_ids.append(file_id)
 
475
                continue
 
476
 
 
477
            # TODO: Handle files that have been deleted
 
478
 
 
479
            # TODO: Maybe a special case for empty files?  Seems a
 
480
            # waste to store them many times.
 
481
 
 
482
            inv.add(entry)
 
483
 
 
484
            if basis_inv.has_id(file_id):
 
485
                old_kind = basis_inv[file_id].kind
 
486
                if old_kind != entry.kind:
 
487
                    bailout("entry %r changed kind from %r to %r"
 
488
                            % (file_id, old_kind, entry.kind))
 
489
 
 
490
            if entry.kind == 'directory':
 
491
                if not isdir(p):
 
492
                    bailout("%s is entered as directory but not a directory" % quotefn(p))
 
493
            elif entry.kind == 'file':
 
494
                if not isfile(p):
 
495
                    bailout("%s is entered as file but is not a file" % quotefn(p))
 
496
 
 
497
                content = file(p, 'rb').read()
 
498
 
 
499
                entry.text_sha1 = sha_string(content)
 
500
                entry.text_size = len(content)
 
501
 
 
502
                old_ie = basis_inv.has_id(file_id) and basis_inv[file_id]
 
503
                if (old_ie
 
504
                    and (old_ie.text_size == entry.text_size)
 
505
                    and (old_ie.text_sha1 == entry.text_sha1)):
 
506
                    ## assert content == basis.get_file(file_id).read()
 
507
                    entry.text_id = basis_inv[file_id].text_id
 
508
                    mutter('    unchanged from previous text_id {%s}' %
 
509
                           entry.text_id)
 
510
                    
 
511
                else:
 
512
                    entry.text_id = gen_file_id(entry.name)
 
513
                    self.text_store.add(content, entry.text_id)
 
514
                    mutter('    stored with text_id {%s}' % entry.text_id)
 
515
                    if verbose:
 
516
                        if not old_ie:
 
517
                            state = 'A'
 
518
                        elif (old_ie.name == entry.name
 
519
                              and old_ie.parent_id == entry.parent_id):
 
520
                            state = 'M'
 
521
                        else:
 
522
                            state = 'R'
 
523
 
 
524
                        show_status(state, entry.kind, quotefn(path))
 
525
 
 
526
        for file_id in missing_ids:
 
527
            # have to do this later so we don't mess up the iterator.
 
528
            # since parents may be removed before their children we
 
529
            # have to test.
 
530
 
 
531
            # FIXME: There's probably a better way to do this; perhaps
 
532
            # the workingtree should know how to filter itself.
 
533
            if work_inv.has_id(file_id):
 
534
                del work_inv[file_id]
 
535
 
 
536
 
 
537
        inv_id = rev_id = _gen_revision_id(time.time())
 
538
        
 
539
        inv_tmp = tempfile.TemporaryFile()
 
540
        inv.write_xml(inv_tmp)
 
541
        inv_tmp.seek(0)
 
542
        self.inventory_store.add(inv_tmp, inv_id)
 
543
        mutter('new inventory_id is {%s}' % inv_id)
 
544
 
 
545
        self._write_inventory(work_inv)
 
546
 
 
547
        if timestamp == None:
 
548
            timestamp = time.time()
 
549
 
 
550
        if committer == None:
 
551
            committer = username()
 
552
 
 
553
        if timezone == None:
 
554
            timezone = local_time_offset()
 
555
 
 
556
        mutter("building commit log message")
 
557
        rev = Revision(timestamp=timestamp,
 
558
                       timezone=timezone,
 
559
                       committer=committer,
 
560
                       precursor = self.last_patch(),
 
561
                       message = message,
 
562
                       inventory_id=inv_id,
 
563
                       revision_id=rev_id)
 
564
 
 
565
        rev_tmp = tempfile.TemporaryFile()
 
566
        rev.write_xml(rev_tmp)
 
567
        rev_tmp.seek(0)
 
568
        self.revision_store.add(rev_tmp, rev_id)
 
569
        mutter("new revision_id is {%s}" % rev_id)
 
570
        
 
571
        ## XXX: Everything up to here can simply be orphaned if we abort
 
572
        ## the commit; it will leave junk files behind but that doesn't
 
573
        ## matter.
 
574
 
 
575
        ## TODO: Read back the just-generated changeset, and make sure it
 
576
        ## applies and recreates the right state.
 
577
 
 
578
        ## TODO: Also calculate and store the inventory SHA1
 
579
        mutter("committing patch r%d" % (self.revno() + 1))
 
580
 
 
581
 
 
582
        self.append_revision(rev_id)
 
583
        
 
584
        if verbose:
 
585
            note("commited r%d" % self.revno())
 
586
 
 
587
 
 
588
    def append_revision(self, revision_id):
 
589
        mutter("add {%s} to revision-history" % revision_id)
 
590
        rev_history = self.revision_history()
 
591
 
 
592
        tmprhname = self.controlfilename('revision-history.tmp')
 
593
        rhname = self.controlfilename('revision-history')
 
594
        
 
595
        f = file(tmprhname, 'wt')
 
596
        rev_history.append(revision_id)
 
597
        f.write('\n'.join(rev_history))
 
598
        f.write('\n')
 
599
        f.close()
 
600
 
 
601
        if sys.platform == 'win32':
 
602
            os.remove(rhname)
 
603
        os.rename(tmprhname, rhname)
 
604
        
 
605
 
 
606
 
 
607
    def get_revision(self, revision_id):
 
608
        """Return the Revision object for a named revision"""
 
609
        r = Revision.read_xml(self.revision_store[revision_id])
 
610
        assert r.revision_id == revision_id
 
611
        return r
 
612
 
 
613
 
 
614
    def get_inventory(self, inventory_id):
 
615
        """Get Inventory object by hash.
 
616
 
 
617
        TODO: Perhaps for this and similar methods, take a revision
 
618
               parameter which can be either an integer revno or a
 
619
               string hash."""
 
620
        i = Inventory.read_xml(self.inventory_store[inventory_id])
 
621
        return i
 
622
 
 
623
 
 
624
    def get_revision_inventory(self, revision_id):
 
625
        """Return inventory of a past revision."""
 
626
        if revision_id == None:
 
627
            return Inventory()
 
628
        else:
 
629
            return self.get_inventory(self.get_revision(revision_id).inventory_id)
 
630
 
273
631
 
274
632
    def revision_history(self):
275
 
        """Return sequence of revision hashes on to this branch."""
276
 
        raise NotImplementedError('revision_history is abstract')
 
633
        """Return sequence of revision hashes on to this branch.
 
634
 
 
635
        >>> ScratchBranch().revision_history()
 
636
        []
 
637
        """
 
638
        return [l.rstrip('\r\n') for l in self.controlfile('revision-history', 'r').readlines()]
 
639
 
277
640
 
278
641
    def revno(self):
279
642
        """Return current revision number for this branch.
280
643
 
281
644
        That is equivalent to the number of revisions committed to
282
645
        this branch.
 
646
 
 
647
        >>> b = ScratchBranch()
 
648
        >>> b.revno()
 
649
        0
 
650
        >>> b.commit('no foo')
 
651
        >>> b.revno()
 
652
        1
283
653
        """
284
654
        return len(self.revision_history())
285
655
 
286
 
    def unbind(self):
287
 
        """Older format branches cannot bind or unbind."""
288
 
        raise errors.UpgradeRequired(self.base)
289
 
 
290
 
    def last_revision(self):
291
 
        """Return last patch hash, or None if no history."""
 
656
 
 
657
    def last_patch(self):
 
658
        """Return last patch hash, or None if no history.
 
659
 
 
660
        >>> ScratchBranch().last_patch() == None
 
661
        True
 
662
        """
292
663
        ph = self.revision_history()
293
664
        if ph:
294
665
            return ph[-1]
295
666
        else:
296
667
            return None
297
 
 
298
 
    def missing_revisions(self, other, stop_revision=None):
299
 
        """Return a list of new revisions that would perfectly fit.
300
668
        
301
 
        If self and other have not diverged, return a list of the revisions
302
 
        present in other, but missing from self.
303
 
 
304
 
        >>> from bzrlib.workingtree import WorkingTree
305
 
        >>> bzrlib.trace.silent = True
306
 
        >>> d1 = bzrdir.ScratchDir()
307
 
        >>> br1 = d1.open_branch()
308
 
        >>> wt1 = d1.open_workingtree()
309
 
        >>> d2 = bzrdir.ScratchDir()
310
 
        >>> br2 = d2.open_branch()
311
 
        >>> wt2 = d2.open_workingtree()
312
 
        >>> br1.missing_revisions(br2)
313
 
        []
314
 
        >>> wt2.commit("lala!", rev_id="REVISION-ID-1")
315
 
        >>> br1.missing_revisions(br2)
316
 
        [u'REVISION-ID-1']
317
 
        >>> br2.missing_revisions(br1)
318
 
        []
319
 
        >>> wt1.commit("lala!", rev_id="REVISION-ID-1")
320
 
        >>> br1.missing_revisions(br2)
321
 
        []
322
 
        >>> wt2.commit("lala!", rev_id="REVISION-ID-2A")
323
 
        >>> br1.missing_revisions(br2)
324
 
        [u'REVISION-ID-2A']
325
 
        >>> wt1.commit("lala!", rev_id="REVISION-ID-2B")
326
 
        >>> br1.missing_revisions(br2)
327
 
        Traceback (most recent call last):
328
 
        DivergedBranches: These branches have diverged.  Try merge.
329
 
        """
330
 
        self_history = self.revision_history()
331
 
        self_len = len(self_history)
332
 
        other_history = other.revision_history()
333
 
        other_len = len(other_history)
334
 
        common_index = min(self_len, other_len) -1
335
 
        if common_index >= 0 and \
336
 
            self_history[common_index] != other_history[common_index]:
337
 
            raise DivergedBranches(self, other)
338
 
 
339
 
        if stop_revision is None:
340
 
            stop_revision = other_len
341
 
        else:
342
 
            assert isinstance(stop_revision, int)
343
 
            if stop_revision > other_len:
344
 
                raise bzrlib.errors.NoSuchRevision(self, stop_revision)
345
 
        return other_history[self_len:stop_revision]
346
 
 
347
 
    def update_revisions(self, other, stop_revision=None):
348
 
        """Pull in new perfect-fit revisions.
349
 
 
350
 
        :param other: Another Branch to pull from
351
 
        :param stop_revision: Updated until the given revision
352
 
        :return: None
353
 
        """
354
 
        raise NotImplementedError('update_revisions is abstract')
355
 
 
356
 
    def revision_id_to_revno(self, revision_id):
357
 
        """Given a revision id, return its revno"""
358
 
        if revision_id is None:
359
 
            return 0
360
 
        history = self.revision_history()
361
 
        try:
362
 
            return history.index(revision_id) + 1
363
 
        except ValueError:
364
 
            raise bzrlib.errors.NoSuchRevision(self, revision_id)
365
 
 
366
 
    def get_rev_id(self, revno, history=None):
367
 
        """Find the revision id of the specified revno."""
 
669
 
 
670
    def lookup_revision(self, revno):
 
671
        """Return revision hash for revision number."""
368
672
        if revno == 0:
369
673
            return None
370
 
        if history is None:
371
 
            history = self.revision_history()
372
 
        elif revno <= 0 or revno > len(history):
373
 
            raise bzrlib.errors.NoSuchRevision(self, revno)
374
 
        return history[revno - 1]
375
 
 
376
 
    def pull(self, source, overwrite=False, stop_revision=None):
377
 
        raise NotImplementedError('pull is abstract')
 
674
 
 
675
        try:
 
676
            # list is 0-based; revisions are 1-based
 
677
            return self.revision_history()[revno-1]
 
678
        except IndexError:
 
679
            raise BzrError("no such revision %s" % revno)
 
680
 
 
681
 
 
682
    def revision_tree(self, revision_id):
 
683
        """Return Tree for a revision on this branch.
 
684
 
 
685
        `revision_id` may be None for the null revision, in which case
 
686
        an `EmptyTree` is returned."""
 
687
 
 
688
        if revision_id == None:
 
689
            return EmptyTree()
 
690
        else:
 
691
            inv = self.get_revision_inventory(revision_id)
 
692
            return RevisionTree(self.text_store, inv)
 
693
 
 
694
 
 
695
    def working_tree(self):
 
696
        """Return a `Tree` for the working copy."""
 
697
        return WorkingTree(self.base, self.read_working_inventory())
 
698
 
378
699
 
379
700
    def basis_tree(self):
380
701
        """Return `Tree` object for last revision.
381
702
 
382
703
        If there are no revisions yet, return an `EmptyTree`.
 
704
 
 
705
        >>> b = ScratchBranch(files=['foo'])
 
706
        >>> b.basis_tree().has_filename('foo')
 
707
        False
 
708
        >>> b.working_tree().has_filename('foo')
 
709
        True
 
710
        >>> b.add('foo')
 
711
        >>> b.commit('add foo')
 
712
        >>> b.basis_tree().has_filename('foo')
 
713
        True
383
714
        """
384
 
        return self.repository.revision_tree(self.last_revision())
 
715
        r = self.last_patch()
 
716
        if r == None:
 
717
            return EmptyTree()
 
718
        else:
 
719
            return RevisionTree(self.text_store, self.get_revision_inventory(r))
 
720
 
 
721
 
 
722
 
 
723
    def write_log(self, show_timezone='original', verbose=False):
 
724
        """Write out human-readable log of commits to this branch
 
725
 
 
726
        utc -- If true, show dates in universal time, not local time."""
 
727
        ## TODO: Option to choose either original, utc or local timezone
 
728
        revno = 1
 
729
        precursor = None
 
730
        for p in self.revision_history():
 
731
            print '-' * 40
 
732
            print 'revno:', revno
 
733
            ## TODO: Show hash if --id is given.
 
734
            ##print 'revision-hash:', p
 
735
            rev = self.get_revision(p)
 
736
            print 'committer:', rev.committer
 
737
            print 'timestamp: %s' % (format_date(rev.timestamp, rev.timezone or 0,
 
738
                                                 show_timezone))
 
739
 
 
740
            ## opportunistic consistency check, same as check_patch_chaining
 
741
            if rev.precursor != precursor:
 
742
                bailout("mismatched precursor!")
 
743
 
 
744
            print 'message:'
 
745
            if not rev.message:
 
746
                print '  (no message)'
 
747
            else:
 
748
                for l in rev.message.split('\n'):
 
749
                    print '  ' + l
 
750
 
 
751
            if verbose == True and precursor != None:
 
752
                print 'changed files:'
 
753
                tree = self.revision_tree(p)
 
754
                prevtree = self.revision_tree(precursor)
 
755
                
 
756
                for file_state, fid, old_name, new_name, kind in \
 
757
                                        diff_trees(prevtree, tree, ):
 
758
                    if file_state == 'A' or file_state == 'M':
 
759
                        show_status(file_state, kind, new_name)
 
760
                    elif file_state == 'D':
 
761
                        show_status(file_state, kind, old_name)
 
762
                    elif file_state == 'R':
 
763
                        show_status(file_state, kind,
 
764
                            old_name + ' => ' + new_name)
 
765
                
 
766
            revno += 1
 
767
            precursor = p
 
768
 
385
769
 
386
770
    def rename_one(self, from_rel, to_rel):
387
771
        """Rename one file.
388
772
 
389
773
        This can change the directory or the filename or both.
390
 
        """
391
 
        raise NotImplementedError('rename_one is abstract')
 
774
         """
 
775
        tree = self.working_tree()
 
776
        inv = tree.inventory
 
777
        if not tree.has_filename(from_rel):
 
778
            bailout("can't rename: old working file %r does not exist" % from_rel)
 
779
        if tree.has_filename(to_rel):
 
780
            bailout("can't rename: new working file %r already exists" % to_rel)
 
781
            
 
782
        file_id = inv.path2id(from_rel)
 
783
        if file_id == None:
 
784
            bailout("can't rename: old name %r is not versioned" % from_rel)
 
785
 
 
786
        if inv.path2id(to_rel):
 
787
            bailout("can't rename: new name %r is already versioned" % to_rel)
 
788
 
 
789
        to_dir, to_tail = os.path.split(to_rel)
 
790
        to_dir_id = inv.path2id(to_dir)
 
791
        if to_dir_id == None and to_dir != '':
 
792
            bailout("can't determine destination directory id for %r" % to_dir)
 
793
 
 
794
        mutter("rename_one:")
 
795
        mutter("  file_id    {%s}" % file_id)
 
796
        mutter("  from_rel   %r" % from_rel)
 
797
        mutter("  to_rel     %r" % to_rel)
 
798
        mutter("  to_dir     %r" % to_dir)
 
799
        mutter("  to_dir_id  {%s}" % to_dir_id)
 
800
            
 
801
        inv.rename(file_id, to_dir_id, to_tail)
 
802
 
 
803
        print "%s => %s" % (from_rel, to_rel)
 
804
        
 
805
        from_abs = self.abspath(from_rel)
 
806
        to_abs = self.abspath(to_rel)
 
807
        try:
 
808
            os.rename(from_abs, to_abs)
 
809
        except OSError, e:
 
810
            bailout("failed to rename %r to %r: %s"
 
811
                    % (from_abs, to_abs, e[1]),
 
812
                    ["rename rolled back"])
 
813
 
 
814
        self._write_inventory(inv)
 
815
            
 
816
 
392
817
 
393
818
    def move(self, from_paths, to_name):
394
819
        """Rename files.
400
825
 
401
826
        Note that to_name is only the last component of the new name;
402
827
        this doesn't change the directory.
403
 
 
404
 
        This returns a list of (from_path, to_path) pairs for each
405
 
        entry that is moved.
406
 
        """
407
 
        raise NotImplementedError('move is abstract')
408
 
 
409
 
    def get_parent(self):
410
 
        """Return the parent location of the branch.
411
 
 
412
 
        This is the default location for push/pull/missing.  The usual
413
 
        pattern is that the user can override it by specifying a
414
 
        location.
415
 
        """
416
 
        raise NotImplementedError('get_parent is abstract')
417
 
 
418
 
    def get_push_location(self):
419
 
        """Return the None or the location to push this branch to."""
420
 
        raise NotImplementedError('get_push_location is abstract')
421
 
 
422
 
    def set_push_location(self, location):
423
 
        """Set a new push location for this branch."""
424
 
        raise NotImplementedError('set_push_location is abstract')
425
 
 
426
 
    def set_parent(self, url):
427
 
        raise NotImplementedError('set_parent is abstract')
428
 
 
429
 
    @needs_write_lock
430
 
    def update(self):
431
 
        """Synchronise this branch with the master branch if any. 
432
 
 
433
 
        :return: None or the last_revision pivoted out during the update.
434
 
        """
435
 
        return None
436
 
 
437
 
    def check_revno(self, revno):
438
 
        """\
439
 
        Check whether a revno corresponds to any revision.
440
 
        Zero (the NULL revision) is considered valid.
441
 
        """
442
 
        if revno != 0:
443
 
            self.check_real_revno(revno)
 
828
        """
 
829
        ## TODO: Option to move IDs only
 
830
        assert not isinstance(from_paths, basestring)
 
831
        tree = self.working_tree()
 
832
        inv = tree.inventory
 
833
        to_abs = self.abspath(to_name)
 
834
        if not isdir(to_abs):
 
835
            bailout("destination %r is not a directory" % to_abs)
 
836
        if not tree.has_filename(to_name):
 
837
            bailout("destination %r not in working directory" % to_abs)
 
838
        to_dir_id = inv.path2id(to_name)
 
839
        if to_dir_id == None and to_name != '':
 
840
            bailout("destination %r is not a versioned directory" % to_name)
 
841
        to_dir_ie = inv[to_dir_id]
 
842
        if to_dir_ie.kind not in ('directory', 'root_directory'):
 
843
            bailout("destination %r is not a directory" % to_abs)
 
844
 
 
845
        to_idpath = Set(inv.get_idpath(to_dir_id))
 
846
 
 
847
        for f in from_paths:
 
848
            if not tree.has_filename(f):
 
849
                bailout("%r does not exist in working tree" % f)
 
850
            f_id = inv.path2id(f)
 
851
            if f_id == None:
 
852
                bailout("%r is not versioned" % f)
 
853
            name_tail = splitpath(f)[-1]
 
854
            dest_path = appendpath(to_name, name_tail)
 
855
            if tree.has_filename(dest_path):
 
856
                bailout("destination %r already exists" % dest_path)
 
857
            if f_id in to_idpath:
 
858
                bailout("can't move %r to a subdirectory of itself" % f)
 
859
 
 
860
        # OK, so there's a race here, it's possible that someone will
 
861
        # create a file in this interval and then the rename might be
 
862
        # left half-done.  But we should have caught most problems.
 
863
 
 
864
        for f in from_paths:
 
865
            name_tail = splitpath(f)[-1]
 
866
            dest_path = appendpath(to_name, name_tail)
 
867
            print "%s => %s" % (f, dest_path)
 
868
            inv.rename(inv.path2id(f), to_dir_id, name_tail)
 
869
            try:
 
870
                os.rename(self.abspath(f), self.abspath(dest_path))
 
871
            except OSError, e:
 
872
                bailout("failed to rename %r to %r: %s" % (f, dest_path, e[1]),
 
873
                        ["rename rolled back"])
 
874
 
 
875
        self._write_inventory(inv)
 
876
 
 
877
 
 
878
 
 
879
    def show_status(self, show_all=False):
 
880
        """Display single-line status for non-ignored working files.
 
881
 
 
882
        The list is show sorted in order by file name.
 
883
 
 
884
        >>> b = ScratchBranch(files=['foo', 'foo~'])
 
885
        >>> b.show_status()
 
886
        ?       foo
 
887
        >>> b.add('foo')
 
888
        >>> b.show_status()
 
889
        A       foo
 
890
        >>> b.commit("add foo")
 
891
        >>> b.show_status()
 
892
        >>> os.unlink(b.abspath('foo'))
 
893
        >>> b.show_status()
 
894
        D       foo
 
895
        
 
896
        TODO: Get state for single files.
 
897
        """
 
898
 
 
899
        # We have to build everything into a list first so that it can
 
900
        # sorted by name, incorporating all the different sources.
 
901
 
 
902
        # FIXME: Rather than getting things in random order and then sorting,
 
903
        # just step through in order.
 
904
 
 
905
        # Interesting case: the old ID for a file has been removed,
 
906
        # but a new file has been created under that name.
 
907
 
 
908
        old = self.basis_tree()
 
909
        new = self.working_tree()
 
910
 
 
911
        for fs, fid, oldname, newname, kind in diff_trees(old, new):
 
912
            if fs == 'R':
 
913
                show_status(fs, kind,
 
914
                            oldname + ' => ' + newname)
 
915
            elif fs == 'A' or fs == 'M':
 
916
                show_status(fs, kind, newname)
 
917
            elif fs == 'D':
 
918
                show_status(fs, kind, oldname)
 
919
            elif fs == '.':
 
920
                if show_all:
 
921
                    show_status(fs, kind, newname)
 
922
            elif fs == 'I':
 
923
                if show_all:
 
924
                    show_status(fs, kind, newname)
 
925
            elif fs == '?':
 
926
                show_status(fs, kind, newname)
 
927
            else:
 
928
                bailout("weird file state %r" % ((fs, fid),))
 
929
                
 
930
 
 
931
 
 
932
class ScratchBranch(Branch):
 
933
    """Special test class: a branch that cleans up after itself.
 
934
 
 
935
    >>> b = ScratchBranch()
 
936
    >>> isdir(b.base)
 
937
    True
 
938
    >>> bd = b.base
 
939
    >>> del b
 
940
    >>> isdir(bd)
 
941
    False
 
942
    """
 
943
    def __init__(self, files=[], dirs=[]):
 
944
        """Make a test branch.
 
945
 
 
946
        This creates a temporary directory and runs init-tree in it.
 
947
 
 
948
        If any files are listed, they are created in the working copy.
 
949
        """
 
950
        Branch.__init__(self, tempfile.mkdtemp(), init=True)
 
951
        for d in dirs:
 
952
            os.mkdir(self.abspath(d))
444
953
            
445
 
    def check_real_revno(self, revno):
446
 
        """\
447
 
        Check whether a revno corresponds to a real revision.
448
 
        Zero (the NULL revision) is considered invalid
449
 
        """
450
 
        if revno < 1 or revno > self.revno():
451
 
            raise InvalidRevisionNumber(revno)
452
 
 
453
 
    @needs_read_lock
454
 
    def clone(self, *args, **kwargs):
455
 
        """Clone this branch into to_bzrdir preserving all semantic values.
456
 
        
457
 
        revision_id: if not None, the revision history in the new branch will
458
 
                     be truncated to end with revision_id.
459
 
        """
460
 
        # for API compatability, until 0.8 releases we provide the old api:
461
 
        # def clone(self, to_location, revision=None, basis_branch=None, to_branch_format=None):
462
 
        # after 0.8 releases, the *args and **kwargs should be changed:
463
 
        # def clone(self, to_bzrdir, revision_id=None):
464
 
        if (kwargs.get('to_location', None) or
465
 
            kwargs.get('revision', None) or
466
 
            kwargs.get('basis_branch', None) or
467
 
            (len(args) and isinstance(args[0], basestring))):
468
 
            # backwards compatability api:
469
 
            warn("Branch.clone() has been deprecated for BzrDir.clone() from"
470
 
                 " bzrlib 0.8.", DeprecationWarning, stacklevel=3)
471
 
            # get basis_branch
472
 
            if len(args) > 2:
473
 
                basis_branch = args[2]
474
 
            else:
475
 
                basis_branch = kwargs.get('basis_branch', None)
476
 
            if basis_branch:
477
 
                basis = basis_branch.bzrdir
478
 
            else:
479
 
                basis = None
480
 
            # get revision
481
 
            if len(args) > 1:
482
 
                revision_id = args[1]
483
 
            else:
484
 
                revision_id = kwargs.get('revision', None)
485
 
            # get location
486
 
            if len(args):
487
 
                url = args[0]
488
 
            else:
489
 
                # no default to raise if not provided.
490
 
                url = kwargs.get('to_location')
491
 
            return self.bzrdir.clone(url,
492
 
                                     revision_id=revision_id,
493
 
                                     basis=basis).open_branch()
494
 
        # new cleaner api.
495
 
        # generate args by hand 
496
 
        if len(args) > 1:
497
 
            revision_id = args[1]
498
 
        else:
499
 
            revision_id = kwargs.get('revision_id', None)
500
 
        if len(args):
501
 
            to_bzrdir = args[0]
502
 
        else:
503
 
            # no default to raise if not provided.
504
 
            to_bzrdir = kwargs.get('to_bzrdir')
505
 
        result = self._format.initialize(to_bzrdir)
506
 
        self.copy_content_into(result, revision_id=revision_id)
507
 
        return  result
508
 
 
509
 
    @needs_read_lock
510
 
    def sprout(self, to_bzrdir, revision_id=None):
511
 
        """Create a new line of development from the branch, into to_bzrdir.
512
 
        
513
 
        revision_id: if not None, the revision history in the new branch will
514
 
                     be truncated to end with revision_id.
515
 
        """
516
 
        result = self._format.initialize(to_bzrdir)
517
 
        self.copy_content_into(result, revision_id=revision_id)
518
 
        result.set_parent(self.bzrdir.root_transport.base)
519
 
        return result
520
 
 
521
 
    @needs_read_lock
522
 
    def copy_content_into(self, destination, revision_id=None):
523
 
        """Copy the content of self into destination.
524
 
 
525
 
        revision_id: if not None, the revision history in the new branch will
526
 
                     be truncated to end with revision_id.
527
 
        """
528
 
        new_history = self.revision_history()
529
 
        if revision_id is not None:
530
 
            try:
531
 
                new_history = new_history[:new_history.index(revision_id) + 1]
532
 
            except ValueError:
533
 
                rev = self.repository.get_revision(revision_id)
534
 
                new_history = rev.get_history(self.repository)[1:]
535
 
        destination.set_revision_history(new_history)
536
 
        parent = self.get_parent()
537
 
        if parent:
538
 
            destination.set_parent(parent)
539
 
 
540
 
 
541
 
class BranchFormat(object):
542
 
    """An encapsulation of the initialization and open routines for a format.
543
 
 
544
 
    Formats provide three things:
545
 
     * An initialization routine,
546
 
     * a format string,
547
 
     * an open routine.
548
 
 
549
 
    Formats are placed in an dict by their format string for reference 
550
 
    during branch opening. Its not required that these be instances, they
551
 
    can be classes themselves with class methods - it simply depends on 
552
 
    whether state is needed for a given format or not.
553
 
 
554
 
    Once a format is deprecated, just deprecate the initialize and open
555
 
    methods on the format class. Do not deprecate the object, as the 
556
 
    object will be created every time regardless.
557
 
    """
558
 
 
559
 
    _default_format = None
560
 
    """The default format used for new branches."""
561
 
 
562
 
    _formats = {}
563
 
    """The known formats."""
564
 
 
565
 
    @classmethod
566
 
    def find_format(klass, a_bzrdir):
567
 
        """Return the format for the branch object in a_bzrdir."""
568
 
        try:
569
 
            transport = a_bzrdir.get_branch_transport(None)
570
 
            format_string = transport.get("format").read()
571
 
            return klass._formats[format_string]
572
 
        except NoSuchFile:
573
 
            raise NotBranchError(path=transport.base)
574
 
        except KeyError:
575
 
            raise errors.UnknownFormatError(format_string)
576
 
 
577
 
    @classmethod
578
 
    def get_default_format(klass):
579
 
        """Return the current default format."""
580
 
        return klass._default_format
581
 
 
582
 
    def get_format_string(self):
583
 
        """Return the ASCII format string that identifies this format."""
584
 
        raise NotImplementedError(self.get_format_string)
585
 
 
586
 
    def get_format_description(self):
587
 
        """Return the short format description for this format."""
588
 
        raise NotImplementedError(self.get_format_string)
589
 
 
590
 
    def initialize(self, a_bzrdir):
591
 
        """Create a branch of this format in a_bzrdir."""
592
 
        raise NotImplementedError(self.initialized)
593
 
 
594
 
    def is_supported(self):
595
 
        """Is this format supported?
596
 
 
597
 
        Supported formats can be initialized and opened.
598
 
        Unsupported formats may not support initialization or committing or 
599
 
        some other features depending on the reason for not being supported.
600
 
        """
601
 
        return True
602
 
 
603
 
    def open(self, a_bzrdir, _found=False):
604
 
        """Return the branch object for a_bzrdir
605
 
 
606
 
        _found is a private parameter, do not use it. It is used to indicate
607
 
               if format probing has already be done.
608
 
        """
609
 
        raise NotImplementedError(self.open)
610
 
 
611
 
    @classmethod
612
 
    def register_format(klass, format):
613
 
        klass._formats[format.get_format_string()] = format
614
 
 
615
 
    @classmethod
616
 
    def set_default_format(klass, format):
617
 
        klass._default_format = format
618
 
 
619
 
    @classmethod
620
 
    def unregister_format(klass, format):
621
 
        assert klass._formats[format.get_format_string()] is format
622
 
        del klass._formats[format.get_format_string()]
623
 
 
624
 
    def __str__(self):
625
 
        return self.get_format_string().rstrip()
626
 
 
627
 
 
628
 
class BzrBranchFormat4(BranchFormat):
629
 
    """Bzr branch format 4.
630
 
 
631
 
    This format has:
632
 
     - a revision-history file.
633
 
     - a branch-lock lock file [ to be shared with the bzrdir ]
634
 
    """
635
 
 
636
 
    def get_format_description(self):
637
 
        """See BranchFormat.get_format_description()."""
638
 
        return "Branch format 4"
639
 
 
640
 
    def initialize(self, a_bzrdir):
641
 
        """Create a branch of this format in a_bzrdir."""
642
 
        mutter('creating branch in %s', a_bzrdir.transport.base)
643
 
        branch_transport = a_bzrdir.get_branch_transport(self)
644
 
        utf8_files = [('revision-history', ''),
645
 
                      ('branch-name', ''),
646
 
                      ]
647
 
        control_files = LockableFiles(branch_transport, 'branch-lock',
648
 
                                      TransportLock)
649
 
        control_files.create_lock()
650
 
        control_files.lock_write()
651
 
        try:
652
 
            for file, content in utf8_files:
653
 
                control_files.put_utf8(file, content)
654
 
        finally:
655
 
            control_files.unlock()
656
 
        return self.open(a_bzrdir, _found=True)
657
 
 
658
 
    def __init__(self):
659
 
        super(BzrBranchFormat4, self).__init__()
660
 
        self._matchingbzrdir = bzrdir.BzrDirFormat6()
661
 
 
662
 
    def open(self, a_bzrdir, _found=False):
663
 
        """Return the branch object for a_bzrdir
664
 
 
665
 
        _found is a private parameter, do not use it. It is used to indicate
666
 
               if format probing has already be done.
667
 
        """
668
 
        if not _found:
669
 
            # we are being called directly and must probe.
670
 
            raise NotImplementedError
671
 
        return BzrBranch(_format=self,
672
 
                         _control_files=a_bzrdir._control_files,
673
 
                         a_bzrdir=a_bzrdir,
674
 
                         _repository=a_bzrdir.open_repository())
675
 
 
676
 
    def __str__(self):
677
 
        return "Bazaar-NG branch format 4"
678
 
 
679
 
 
680
 
class BzrBranchFormat5(BranchFormat):
681
 
    """Bzr branch format 5.
682
 
 
683
 
    This format has:
684
 
     - a revision-history file.
685
 
     - a format string
686
 
     - a lock dir guarding the branch itself
687
 
     - all of this stored in a branch/ subdirectory
688
 
     - works with shared repositories.
689
 
 
690
 
    This format is new in bzr 0.8.
691
 
    """
692
 
 
693
 
    def get_format_string(self):
694
 
        """See BranchFormat.get_format_string()."""
695
 
        return "Bazaar-NG branch format 5\n"
696
 
 
697
 
    def get_format_description(self):
698
 
        """See BranchFormat.get_format_description()."""
699
 
        return "Branch format 5"
700
 
        
701
 
    def initialize(self, a_bzrdir):
702
 
        """Create a branch of this format in a_bzrdir."""
703
 
        mutter('creating branch %r in %s', self, a_bzrdir.transport.base)
704
 
        branch_transport = a_bzrdir.get_branch_transport(self)
705
 
        utf8_files = [('revision-history', ''),
706
 
                      ('branch-name', ''),
707
 
                      ]
708
 
        control_files = LockableFiles(branch_transport, 'lock', LockDir)
709
 
        control_files.create_lock()
710
 
        control_files.lock_write()
711
 
        control_files.put_utf8('format', self.get_format_string())
712
 
        try:
713
 
            for file, content in utf8_files:
714
 
                control_files.put_utf8(file, content)
715
 
        finally:
716
 
            control_files.unlock()
717
 
        return self.open(a_bzrdir, _found=True, )
718
 
 
719
 
    def __init__(self):
720
 
        super(BzrBranchFormat5, self).__init__()
721
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
722
 
 
723
 
    def open(self, a_bzrdir, _found=False):
724
 
        """Return the branch object for a_bzrdir
725
 
 
726
 
        _found is a private parameter, do not use it. It is used to indicate
727
 
               if format probing has already be done.
728
 
        """
729
 
        if not _found:
730
 
            format = BranchFormat.find_format(a_bzrdir)
731
 
            assert format.__class__ == self.__class__
732
 
        transport = a_bzrdir.get_branch_transport(None)
733
 
        control_files = LockableFiles(transport, 'lock', LockDir)
734
 
        return BzrBranch5(_format=self,
735
 
                          _control_files=control_files,
736
 
                          a_bzrdir=a_bzrdir,
737
 
                          _repository=a_bzrdir.find_repository())
738
 
 
739
 
    def __str__(self):
740
 
        return "Bazaar-NG Metadir branch format 5"
741
 
 
742
 
 
743
 
class BranchReferenceFormat(BranchFormat):
744
 
    """Bzr branch reference format.
745
 
 
746
 
    Branch references are used in implementing checkouts, they
747
 
    act as an alias to the real branch which is at some other url.
748
 
 
749
 
    This format has:
750
 
     - A location file
751
 
     - a format string
752
 
    """
753
 
 
754
 
    def get_format_string(self):
755
 
        """See BranchFormat.get_format_string()."""
756
 
        return "Bazaar-NG Branch Reference Format 1\n"
757
 
 
758
 
    def get_format_description(self):
759
 
        """See BranchFormat.get_format_description()."""
760
 
        return "Checkout reference format 1"
761
 
        
762
 
    def initialize(self, a_bzrdir, target_branch=None):
763
 
        """Create a branch of this format in a_bzrdir."""
764
 
        if target_branch is None:
765
 
            # this format does not implement branch itself, thus the implicit
766
 
            # creation contract must see it as uninitializable
767
 
            raise errors.UninitializableFormat(self)
768
 
        mutter('creating branch reference in %s', a_bzrdir.transport.base)
769
 
        branch_transport = a_bzrdir.get_branch_transport(self)
770
 
        # FIXME rbc 20060209 one j-a-ms encoding branch lands this str() cast is not needed.
771
 
        branch_transport.put('location', StringIO(str(target_branch.bzrdir.root_transport.base)))
772
 
        branch_transport.put('format', StringIO(self.get_format_string()))
773
 
        return self.open(a_bzrdir, _found=True)
774
 
 
775
 
    def __init__(self):
776
 
        super(BranchReferenceFormat, self).__init__()
777
 
        self._matchingbzrdir = bzrdir.BzrDirMetaFormat1()
778
 
 
779
 
    def _make_reference_clone_function(format, a_branch):
780
 
        """Create a clone() routine for a branch dynamically."""
781
 
        def clone(to_bzrdir, revision_id=None):
782
 
            """See Branch.clone()."""
783
 
            return format.initialize(to_bzrdir, a_branch)
784
 
            # cannot obey revision_id limits when cloning a reference ...
785
 
            # FIXME RBC 20060210 either nuke revision_id for clone, or
786
 
            # emit some sort of warning/error to the caller ?!
787
 
        return clone
788
 
 
789
 
    def open(self, a_bzrdir, _found=False):
790
 
        """Return the branch that the branch reference in a_bzrdir points at.
791
 
 
792
 
        _found is a private parameter, do not use it. It is used to indicate
793
 
               if format probing has already be done.
794
 
        """
795
 
        if not _found:
796
 
            format = BranchFormat.find_format(a_bzrdir)
797
 
            assert format.__class__ == self.__class__
798
 
        transport = a_bzrdir.get_branch_transport(None)
799
 
        real_bzrdir = bzrdir.BzrDir.open(transport.get('location').read())
800
 
        result = real_bzrdir.open_branch()
801
 
        # this changes the behaviour of result.clone to create a new reference
802
 
        # rather than a copy of the content of the branch.
803
 
        # I did not use a proxy object because that needs much more extensive
804
 
        # testing, and we are only changing one behaviour at the moment.
805
 
        # If we decide to alter more behaviours - i.e. the implicit nickname
806
 
        # then this should be refactored to introduce a tested proxy branch
807
 
        # and a subclass of that for use in overriding clone() and ....
808
 
        # - RBC 20060210
809
 
        result.clone = self._make_reference_clone_function(result)
810
 
        return result
811
 
 
812
 
 
813
 
# formats which have no format string are not discoverable
814
 
# and not independently creatable, so are not registered.
815
 
__default_format = BzrBranchFormat5()
816
 
BranchFormat.register_format(__default_format)
817
 
BranchFormat.register_format(BranchReferenceFormat())
818
 
BranchFormat.set_default_format(__default_format)
819
 
_legacy_formats = [BzrBranchFormat4(),
820
 
                   ]
821
 
 
822
 
class BzrBranch(Branch):
823
 
    """A branch stored in the actual filesystem.
824
 
 
825
 
    Note that it's "local" in the context of the filesystem; it doesn't
826
 
    really matter if it's on an nfs/smb/afs/coda/... share, as long as
827
 
    it's writable, and can be accessed via the normal filesystem API.
828
 
    """
829
 
    
830
 
    def __init__(self, transport=DEPRECATED_PARAMETER, init=DEPRECATED_PARAMETER,
831
 
                 relax_version_check=DEPRECATED_PARAMETER, _format=None,
832
 
                 _control_files=None, a_bzrdir=None, _repository=None):
833
 
        """Create new branch object at a particular location.
834
 
 
835
 
        transport -- A Transport object, defining how to access files.
836
 
        
837
 
        init -- If True, create new control files in a previously
838
 
             unversioned directory.  If False, the branch must already
839
 
             be versioned.
840
 
 
841
 
        relax_version_check -- If true, the usual check for the branch
842
 
            version is not applied.  This is intended only for
843
 
            upgrade/recovery type use; it's not guaranteed that
844
 
            all operations will work on old format branches.
845
 
        """
846
 
        if a_bzrdir is None:
847
 
            self.bzrdir = bzrdir.BzrDir.open(transport.base)
848
 
        else:
849
 
            self.bzrdir = a_bzrdir
850
 
        self._transport = self.bzrdir.transport.clone('..')
851
 
        self._base = self._transport.base
852
 
        self._format = _format
853
 
        if _control_files is None:
854
 
            raise BzrBadParameterMissing('_control_files')
855
 
        self.control_files = _control_files
856
 
        if deprecated_passed(init):
857
 
            warn("BzrBranch.__init__(..., init=XXX): The init parameter is "
858
 
                 "deprecated as of bzr 0.8. Please use Branch.create().",
859
 
                 DeprecationWarning,
860
 
                 stacklevel=2)
861
 
            if init:
862
 
                # this is slower than before deprecation, oh well never mind.
863
 
                # -> its deprecated.
864
 
                self._initialize(transport.base)
865
 
        self._check_format(_format)
866
 
        if deprecated_passed(relax_version_check):
867
 
            warn("BzrBranch.__init__(..., relax_version_check=XXX_: The "
868
 
                 "relax_version_check parameter is deprecated as of bzr 0.8. "
869
 
                 "Please use BzrDir.open_downlevel, or a BzrBranchFormat's "
870
 
                 "open() method.",
871
 
                 DeprecationWarning,
872
 
                 stacklevel=2)
873
 
            if (not relax_version_check
874
 
                and not self._format.is_supported()):
875
 
                raise errors.UnsupportedFormatError(
876
 
                        'sorry, branch format %r not supported' % fmt,
877
 
                        ['use a different bzr version',
878
 
                         'or remove the .bzr directory'
879
 
                         ' and "bzr init" again'])
880
 
        if deprecated_passed(transport):
881
 
            warn("BzrBranch.__init__(transport=XXX...): The transport "
882
 
                 "parameter is deprecated as of bzr 0.8. "
883
 
                 "Please use Branch.open, or bzrdir.open_branch().",
884
 
                 DeprecationWarning,
885
 
                 stacklevel=2)
886
 
        self.repository = _repository
887
 
 
888
 
    def __str__(self):
889
 
        return '%s(%r)' % (self.__class__.__name__, self.base)
890
 
 
891
 
    __repr__ = __str__
 
954
        for f in files:
 
955
            file(os.path.join(self.base, f), 'w').write('content of %s' % f)
 
956
 
892
957
 
893
958
    def __del__(self):
894
 
        # TODO: It might be best to do this somewhere else,
895
 
        # but it is nice for a Branch object to automatically
896
 
        # cache it's information.
897
 
        # Alternatively, we could have the Transport objects cache requests
898
 
        # See the earlier discussion about how major objects (like Branch)
899
 
        # should never expect their __del__ function to run.
900
 
        # XXX: cache_root seems to be unused, 2006-01-13 mbp
901
 
        if hasattr(self, 'cache_root') and self.cache_root is not None:
902
 
            try:
903
 
                rmtree(self.cache_root)
904
 
            except:
905
 
                pass
906
 
            self.cache_root = None
907
 
 
908
 
    def _get_base(self):
909
 
        return self._base
910
 
 
911
 
    base = property(_get_base, doc="The URL for the root of this branch.")
912
 
 
913
 
    def _finish_transaction(self):
914
 
        """Exit the current transaction."""
915
 
        return self.control_files._finish_transaction()
916
 
 
917
 
    def get_transaction(self):
918
 
        """Return the current active transaction.
919
 
 
920
 
        If no transaction is active, this returns a passthrough object
921
 
        for which all data is immediately flushed and no caching happens.
922
 
        """
923
 
        # this is an explicit function so that we can do tricky stuff
924
 
        # when the storage in rev_storage is elsewhere.
925
 
        # we probably need to hook the two 'lock a location' and 
926
 
        # 'have a transaction' together more delicately, so that
927
 
        # we can have two locks (branch and storage) and one transaction
928
 
        # ... and finishing the transaction unlocks both, but unlocking
929
 
        # does not. - RBC 20051121
930
 
        return self.control_files.get_transaction()
931
 
 
932
 
    def _set_transaction(self, transaction):
933
 
        """Set a new active transaction."""
934
 
        return self.control_files._set_transaction(transaction)
935
 
 
936
 
    def abspath(self, name):
937
 
        """See Branch.abspath."""
938
 
        return self.control_files._transport.abspath(name)
939
 
 
940
 
    def _check_format(self, format):
941
 
        """Identify the branch format if needed.
942
 
 
943
 
        The format is stored as a reference to the format object in
944
 
        self._format for code that needs to check it later.
945
 
 
946
 
        The format parameter is either None or the branch format class
947
 
        used to open this branch.
948
 
 
949
 
        FIXME: DELETE THIS METHOD when pre 0.8 support is removed.
950
 
        """
951
 
        if format is None:
952
 
            format = BzrBranchFormat.find_format(self.bzrdir)
953
 
        self._format = format
954
 
        mutter("got branch format %s", self._format)
955
 
 
956
 
    @needs_read_lock
957
 
    def get_root_id(self):
958
 
        """See Branch.get_root_id."""
959
 
        tree = self.repository.revision_tree(self.last_revision())
960
 
        return tree.inventory.root.file_id
961
 
 
962
 
    def is_locked(self):
963
 
        return self.control_files.is_locked()
964
 
 
965
 
    def lock_write(self):
966
 
        # TODO: test for failed two phase locks. This is known broken.
967
 
        self.control_files.lock_write()
968
 
        self.repository.lock_write()
969
 
 
970
 
    def lock_read(self):
971
 
        # TODO: test for failed two phase locks. This is known broken.
972
 
        self.control_files.lock_read()
973
 
        self.repository.lock_read()
974
 
 
975
 
    def unlock(self):
976
 
        # TODO: test for failed two phase locks. This is known broken.
977
 
        try:
978
 
            self.repository.unlock()
979
 
        finally:
980
 
            self.control_files.unlock()
981
 
        
982
 
    def peek_lock_mode(self):
983
 
        if self.control_files._lock_count == 0:
984
 
            return None
985
 
        else:
986
 
            return self.control_files._lock_mode
987
 
 
988
 
    def get_physical_lock_status(self):
989
 
        return self.control_files.get_physical_lock_status()
990
 
 
991
 
    @needs_read_lock
992
 
    def print_file(self, file, revision_id):
993
 
        """See Branch.print_file."""
994
 
        return self.repository.print_file(file, revision_id)
995
 
 
996
 
    @needs_write_lock
997
 
    def append_revision(self, *revision_ids):
998
 
        """See Branch.append_revision."""
999
 
        for revision_id in revision_ids:
1000
 
            mutter("add {%s} to revision-history" % revision_id)
1001
 
        rev_history = self.revision_history()
1002
 
        rev_history.extend(revision_ids)
1003
 
        self.set_revision_history(rev_history)
1004
 
 
1005
 
    @needs_write_lock
1006
 
    def set_revision_history(self, rev_history):
1007
 
        """See Branch.set_revision_history."""
1008
 
        self.control_files.put_utf8(
1009
 
            'revision-history', '\n'.join(rev_history))
1010
 
        transaction = self.get_transaction()
1011
 
        history = transaction.map.find_revision_history()
1012
 
        if history is not None:
1013
 
            # update the revision history in the identity map.
1014
 
            history[:] = list(rev_history)
1015
 
            # this call is disabled because revision_history is 
1016
 
            # not really an object yet, and the transaction is for objects.
1017
 
            # transaction.register_dirty(history)
1018
 
        else:
1019
 
            transaction.map.add_revision_history(rev_history)
1020
 
            # this call is disabled because revision_history is 
1021
 
            # not really an object yet, and the transaction is for objects.
1022
 
            # transaction.register_clean(history)
1023
 
 
1024
 
    def get_revision_delta(self, revno):
1025
 
        """Return the delta for one revision.
1026
 
 
1027
 
        The delta is relative to its mainline predecessor, or the
1028
 
        empty tree for revision 1.
1029
 
        """
1030
 
        assert isinstance(revno, int)
1031
 
        rh = self.revision_history()
1032
 
        if not (1 <= revno <= len(rh)):
1033
 
            raise InvalidRevisionNumber(revno)
1034
 
 
1035
 
        # revno is 1-based; list is 0-based
1036
 
 
1037
 
        new_tree = self.repository.revision_tree(rh[revno-1])
1038
 
        if revno == 1:
1039
 
            old_tree = EmptyTree()
1040
 
        else:
1041
 
            old_tree = self.repository.revision_tree(rh[revno-2])
1042
 
        return compare_trees(old_tree, new_tree)
1043
 
 
1044
 
    @needs_read_lock
1045
 
    def revision_history(self):
1046
 
        """See Branch.revision_history."""
1047
 
        transaction = self.get_transaction()
1048
 
        history = transaction.map.find_revision_history()
1049
 
        if history is not None:
1050
 
            mutter("cache hit for revision-history in %s", self)
1051
 
            return list(history)
1052
 
        history = [l.rstrip('\r\n') for l in
1053
 
                self.control_files.get_utf8('revision-history').readlines()]
1054
 
        transaction.map.add_revision_history(history)
1055
 
        # this call is disabled because revision_history is 
1056
 
        # not really an object yet, and the transaction is for objects.
1057
 
        # transaction.register_clean(history, precious=True)
1058
 
        return list(history)
1059
 
 
1060
 
    @needs_write_lock
1061
 
    def update_revisions(self, other, stop_revision=None):
1062
 
        """See Branch.update_revisions."""
1063
 
        other.lock_read()
1064
 
        try:
1065
 
            if stop_revision is None:
1066
 
                stop_revision = other.last_revision()
1067
 
                if stop_revision is None:
1068
 
                    # if there are no commits, we're done.
1069
 
                    return
1070
 
            # whats the current last revision, before we fetch [and change it
1071
 
            # possibly]
1072
 
            last_rev = self.last_revision()
1073
 
            # we fetch here regardless of whether we need to so that we pickup
1074
 
            # filled in ghosts.
1075
 
            self.fetch(other, stop_revision)
1076
 
            my_ancestry = self.repository.get_ancestry(last_rev)
1077
 
            if stop_revision in my_ancestry:
1078
 
                # last_revision is a descendant of stop_revision
1079
 
                return
1080
 
            # stop_revision must be a descendant of last_revision
1081
 
            stop_graph = self.repository.get_revision_graph(stop_revision)
1082
 
            if last_rev is not None and last_rev not in stop_graph:
1083
 
                # our previous tip is not merged into stop_revision
1084
 
                raise errors.DivergedBranches(self, other)
1085
 
            # make a new revision history from the graph
1086
 
            current_rev_id = stop_revision
1087
 
            new_history = []
1088
 
            while current_rev_id not in (None, NULL_REVISION):
1089
 
                new_history.append(current_rev_id)
1090
 
                current_rev_id_parents = stop_graph[current_rev_id]
1091
 
                try:
1092
 
                    current_rev_id = current_rev_id_parents[0]
1093
 
                except IndexError:
1094
 
                    current_rev_id = None
1095
 
            new_history.reverse()
1096
 
            self.set_revision_history(new_history)
1097
 
        finally:
1098
 
            other.unlock()
1099
 
 
1100
 
    def basis_tree(self):
1101
 
        """See Branch.basis_tree."""
1102
 
        return self.repository.revision_tree(self.last_revision())
1103
 
 
1104
 
    @deprecated_method(zero_eight)
1105
 
    def working_tree(self):
1106
 
        """Create a Working tree object for this branch."""
1107
 
        from bzrlib.workingtree import WorkingTree
1108
 
        from bzrlib.transport.local import LocalTransport
1109
 
        if (self.base.find('://') != -1 or 
1110
 
            not isinstance(self._transport, LocalTransport)):
1111
 
            raise NoWorkingTree(self.base)
1112
 
        return self.bzrdir.open_workingtree()
1113
 
 
1114
 
    @needs_write_lock
1115
 
    def pull(self, source, overwrite=False, stop_revision=None):
1116
 
        """See Branch.pull."""
1117
 
        source.lock_read()
1118
 
        try:
1119
 
            old_count = len(self.revision_history())
1120
 
            try:
1121
 
                self.update_revisions(source,stop_revision)
1122
 
            except DivergedBranches:
1123
 
                if not overwrite:
1124
 
                    raise
1125
 
            if overwrite:
1126
 
                self.set_revision_history(source.revision_history())
1127
 
            new_count = len(self.revision_history())
1128
 
            return new_count - old_count
1129
 
        finally:
1130
 
            source.unlock()
1131
 
 
1132
 
    def get_parent(self):
1133
 
        """See Branch.get_parent."""
1134
 
        import errno
1135
 
        _locs = ['parent', 'pull', 'x-pull']
1136
 
        for l in _locs:
1137
 
            try:
1138
 
                return self.control_files.get_utf8(l).read().strip('\n')
1139
 
            except NoSuchFile:
1140
 
                pass
1141
 
        return None
1142
 
 
1143
 
    def get_push_location(self):
1144
 
        """See Branch.get_push_location."""
1145
 
        config = bzrlib.config.BranchConfig(self)
1146
 
        push_loc = config.get_user_option('push_location')
1147
 
        return push_loc
1148
 
 
1149
 
    def set_push_location(self, location):
1150
 
        """See Branch.set_push_location."""
1151
 
        config = bzrlib.config.LocationConfig(self.base)
1152
 
        config.set_user_option('push_location', location)
1153
 
 
1154
 
    @needs_write_lock
1155
 
    def set_parent(self, url):
1156
 
        """See Branch.set_parent."""
1157
 
        # TODO: Maybe delete old location files?
1158
 
        # URLs should never be unicode, even on the local fs,
1159
 
        # FIXUP this and get_parent in a future branch format bump:
1160
 
        # read and rewrite the file, and have the new format code read
1161
 
        # using .get not .get_utf8. RBC 20060125
1162
 
        if url is None:
1163
 
            self.control_files._transport.delete('parent')
1164
 
        else:
1165
 
            self.control_files.put_utf8('parent', url + '\n')
1166
 
 
1167
 
    def tree_config(self):
1168
 
        return TreeConfig(self)
1169
 
 
1170
 
 
1171
 
class BzrBranch5(BzrBranch):
1172
 
    """A format 5 branch. This supports new features over plan branches.
1173
 
 
1174
 
    It has support for a master_branch which is the data for bound branches.
1175
 
    """
1176
 
 
1177
 
    def __init__(self,
1178
 
                 _format,
1179
 
                 _control_files,
1180
 
                 a_bzrdir,
1181
 
                 _repository):
1182
 
        super(BzrBranch5, self).__init__(_format=_format,
1183
 
                                         _control_files=_control_files,
1184
 
                                         a_bzrdir=a_bzrdir,
1185
 
                                         _repository=_repository)
1186
 
        
1187
 
    @needs_write_lock
1188
 
    def pull(self, source, overwrite=False, stop_revision=None):
1189
 
        """Updates branch.pull to be bound branch aware."""
1190
 
        bound_location = self.get_bound_location()
1191
 
        if source.base != bound_location:
1192
 
            # not pulling from master, so we need to update master.
1193
 
            master_branch = self.get_master_branch()
1194
 
            if master_branch:
1195
 
                master_branch.pull(source)
1196
 
                source = master_branch
1197
 
        return super(BzrBranch5, self).pull(source, overwrite, stop_revision)
1198
 
 
1199
 
    def get_bound_location(self):
1200
 
        try:
1201
 
            return self.control_files.get_utf8('bound').read()[:-1]
1202
 
        except errors.NoSuchFile:
1203
 
            return None
1204
 
 
1205
 
    @needs_read_lock
1206
 
    def get_master_branch(self):
1207
 
        """Return the branch we are bound to.
1208
 
        
1209
 
        :return: Either a Branch, or None
1210
 
 
1211
 
        This could memoise the branch, but if thats done
1212
 
        it must be revalidated on each new lock.
1213
 
        So for now we just dont memoise it.
1214
 
        # RBC 20060304 review this decision.
1215
 
        """
1216
 
        bound_loc = self.get_bound_location()
1217
 
        if not bound_loc:
1218
 
            return None
1219
 
        try:
1220
 
            return Branch.open(bound_loc)
1221
 
        except (errors.NotBranchError, errors.ConnectionError), e:
1222
 
            raise errors.BoundBranchConnectionFailure(
1223
 
                    self, bound_loc, e)
1224
 
 
1225
 
    @needs_write_lock
1226
 
    def set_bound_location(self, location):
1227
 
        """Set the target where this branch is bound to.
1228
 
 
1229
 
        :param location: URL to the target branch
1230
 
        """
1231
 
        if location:
1232
 
            self.control_files.put_utf8('bound', location+'\n')
1233
 
        else:
1234
 
            try:
1235
 
                self.control_files._transport.delete('bound')
1236
 
            except NoSuchFile:
1237
 
                return False
 
959
        """Destroy the test branch, removing the scratch directory."""
 
960
        try:
 
961
            shutil.rmtree(self.base)
 
962
        except OSError:
 
963
            # Work around for shutil.rmtree failing on Windows when
 
964
            # readonly files are encountered
 
965
            for root, dirs, files in os.walk(self.base, topdown=False):
 
966
                for name in files:
 
967
                    os.chmod(os.path.join(root, name), 0700)
 
968
            shutil.rmtree(self.base)
 
969
 
 
970
    
 
971
 
 
972
######################################################################
 
973
# predicates
 
974
 
 
975
 
 
976
def is_control_file(filename):
 
977
    ## FIXME: better check
 
978
    filename = os.path.normpath(filename)
 
979
    while filename != '':
 
980
        head, tail = os.path.split(filename)
 
981
        ## mutter('check %r for control file' % ((head, tail), ))
 
982
        if tail == bzrlib.BZRDIR:
1238
983
            return True
1239
 
 
1240
 
    @needs_write_lock
1241
 
    def bind(self, other):
1242
 
        """Bind the local branch the other branch.
1243
 
 
1244
 
        :param other: The branch to bind to
1245
 
        :type other: Branch
1246
 
        """
1247
 
        # TODO: jam 20051230 Consider checking if the target is bound
1248
 
        #       It is debatable whether you should be able to bind to
1249
 
        #       a branch which is itself bound.
1250
 
        #       Committing is obviously forbidden,
1251
 
        #       but binding itself may not be.
1252
 
        #       Since we *have* to check at commit time, we don't
1253
 
        #       *need* to check here
1254
 
        self.pull(other)
1255
 
 
1256
 
        # we are now equal to or a suffix of other.
1257
 
 
1258
 
        # Since we have 'pulled' from the remote location,
1259
 
        # now we should try to pull in the opposite direction
1260
 
        # in case the local tree has more revisions than the
1261
 
        # remote one.
1262
 
        # There may be a different check you could do here
1263
 
        # rather than actually trying to install revisions remotely.
1264
 
        # TODO: capture an exception which indicates the remote branch
1265
 
        #       is not writeable. 
1266
 
        #       If it is up-to-date, this probably should not be a failure
1267
 
        
1268
 
        # lock other for write so the revision-history syncing cannot race
1269
 
        other.lock_write()
1270
 
        try:
1271
 
            other.pull(self)
1272
 
            # if this does not error, other now has the same last rev we do
1273
 
            # it can only error if the pull from other was concurrent with
1274
 
            # a commit to other from someone else.
1275
 
 
1276
 
            # until we ditch revision-history, we need to sync them up:
1277
 
            self.set_revision_history(other.revision_history())
1278
 
            # now other and self are up to date with each other and have the
1279
 
            # same revision-history.
1280
 
        finally:
1281
 
            other.unlock()
1282
 
 
1283
 
        self.set_bound_location(other.base)
1284
 
 
1285
 
    @needs_write_lock
1286
 
    def unbind(self):
1287
 
        """If bound, unbind"""
1288
 
        return self.set_bound_location(None)
1289
 
 
1290
 
    @needs_write_lock
1291
 
    def update(self):
1292
 
        """Synchronise this branch with the master branch if any. 
1293
 
 
1294
 
        :return: None or the last_revision that was pivoted out during the
1295
 
                 update.
1296
 
        """
1297
 
        master = self.get_master_branch()
1298
 
        if master is not None:
1299
 
            old_tip = self.last_revision()
1300
 
            self.pull(master, overwrite=True)
1301
 
            if old_tip in self.repository.get_ancestry(self.last_revision()):
1302
 
                return None
1303
 
            return old_tip
1304
 
        return None
1305
 
 
1306
 
 
1307
 
class BranchTestProviderAdapter(object):
1308
 
    """A tool to generate a suite testing multiple branch formats at once.
1309
 
 
1310
 
    This is done by copying the test once for each transport and injecting
1311
 
    the transport_server, transport_readonly_server, and branch_format
1312
 
    classes into each copy. Each copy is also given a new id() to make it
1313
 
    easy to identify.
1314
 
    """
1315
 
 
1316
 
    def __init__(self, transport_server, transport_readonly_server, formats):
1317
 
        self._transport_server = transport_server
1318
 
        self._transport_readonly_server = transport_readonly_server
1319
 
        self._formats = formats
1320
 
    
1321
 
    def adapt(self, test):
1322
 
        result = TestSuite()
1323
 
        for branch_format, bzrdir_format in self._formats:
1324
 
            new_test = deepcopy(test)
1325
 
            new_test.transport_server = self._transport_server
1326
 
            new_test.transport_readonly_server = self._transport_readonly_server
1327
 
            new_test.bzrdir_format = bzrdir_format
1328
 
            new_test.branch_format = branch_format
1329
 
            def make_new_test_id():
1330
 
                new_id = "%s(%s)" % (new_test.id(), branch_format.__class__.__name__)
1331
 
                return lambda: new_id
1332
 
            new_test.id = make_new_test_id()
1333
 
            result.addTest(new_test)
1334
 
        return result
1335
 
 
1336
 
 
1337
 
######################################################################
1338
 
# predicates
1339
 
 
1340
 
 
1341
 
@deprecated_function(zero_eight)
1342
 
def ScratchBranch(*args, **kwargs):
1343
 
    """See bzrlib.bzrdir.ScratchDir."""
1344
 
    d = ScratchDir(*args, **kwargs)
1345
 
    return d.open_branch()
1346
 
 
1347
 
 
1348
 
@deprecated_function(zero_eight)
1349
 
def is_control_file(*args, **kwargs):
1350
 
    """See bzrlib.workingtree.is_control_file."""
1351
 
    return bzrlib.workingtree.is_control_file(*args, **kwargs)
 
984
        if filename == head:
 
985
            break
 
986
        filename = head
 
987
    return False
 
988
 
 
989
 
 
990
 
 
991
def _gen_revision_id(when):
 
992
    """Return new revision-id."""
 
993
    s = '%s-%s-' % (user_email(), compact_date(when))
 
994
    s += hexlify(rand_bytes(8))
 
995
    return s
 
996
 
 
997
 
 
998
def gen_file_id(name):
 
999
    """Return new file id.
 
1000
 
 
1001
    This should probably generate proper UUIDs, but for the moment we
 
1002
    cope with just randomness because running uuidgen every time is
 
1003
    slow."""
 
1004
    idx = name.rfind('/')
 
1005
    if idx != -1:
 
1006
        name = name[idx+1 : ]
 
1007
    idx = name.rfind('\\')
 
1008
    if idx != -1:
 
1009
        name = name[idx+1 : ]
 
1010
 
 
1011
    name = name.lstrip('.')
 
1012
 
 
1013
    s = hexlify(rand_bytes(8))
 
1014
    return '-'.join((name, compact_date(time.time()), s))