1
# Copyright (C) 2007-2011 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""WorkingTree3 format and implementation.
21
from __future__ import absolute_import
30
revision as _mod_revision,
34
from bzrlib.decorators import (
37
from bzrlib.lockable_files import LockableFiles
38
from bzrlib.lockdir import LockDir
39
from bzrlib.mutabletree import MutableTree
40
from bzrlib.transport.local import LocalTransport
41
from bzrlib.workingtree import (
43
WorkingTreeFormatMetaDir,
47
class PreDirStateWorkingTree(InventoryWorkingTree):
49
def __init__(self, basedir='.', *args, **kwargs):
50
super(PreDirStateWorkingTree, self).__init__(basedir, *args, **kwargs)
51
# update the whole cache up front and write to disk if anything changed;
52
# in the future we might want to do this more selectively
53
# two possible ways offer themselves : in self._unlock, write the cache
54
# if needed, or, when the cache sees a change, append it to the hash
55
# cache file, and have the parser take the most recent entry for a
57
wt_trans = self.bzrdir.get_workingtree_transport(None)
58
cache_filename = wt_trans.local_abspath('stat-cache')
59
self._hashcache = hashcache.HashCache(basedir, cache_filename,
60
self.bzrdir._get_file_mode(),
61
self._content_filter_stack_provider())
64
# is this scan needed ? it makes things kinda slow.
68
trace.mutter("write hc")
71
def _write_hashcache_if_dirty(self):
72
"""Write out the hashcache if it is dirty."""
73
if self._hashcache.needs_write:
75
self._hashcache.write()
77
if e.errno not in (errno.EPERM, errno.EACCES):
79
# TODO: jam 20061219 Should this be a warning? A single line
80
# warning might be sufficient to let the user know what
82
trace.mutter('Could not write hashcache for %s\nError: %s',
83
self._hashcache.cache_file_name(), e)
86
def get_file_sha1(self, file_id, path=None, stat_value=None):
88
path = self._inventory.id2path(file_id)
89
return self._hashcache.get_sha1(path, stat_value)
92
class WorkingTree3(PreDirStateWorkingTree):
93
"""This is the Format 3 working tree.
95
This differs from the base WorkingTree by:
96
- having its own file lock
97
- having its own last-revision property.
99
This is new in bzr 0.8
103
def _last_revision(self):
104
"""See Mutable.last_revision."""
106
return self._transport.get_bytes('last-revision')
107
except errors.NoSuchFile:
108
return _mod_revision.NULL_REVISION
110
def _change_last_revision(self, revision_id):
111
"""See WorkingTree._change_last_revision."""
112
if revision_id is None or revision_id == _mod_revision.NULL_REVISION:
114
self._transport.delete('last-revision')
115
except errors.NoSuchFile:
119
self._transport.put_bytes('last-revision', revision_id,
120
mode=self.bzrdir._get_file_mode())
123
def _get_check_refs(self):
124
"""Return the references needed to perform a check of this tree."""
125
return [('trees', self.last_revision())]
128
if self._control_files._lock_count == 1:
129
# do non-implementation specific cleanup
131
# _inventory_is_modified is always False during a read lock.
132
if self._inventory_is_modified:
134
self._write_hashcache_if_dirty()
135
# reverse order of locking.
137
return self._control_files.unlock()
142
class WorkingTreeFormat3(WorkingTreeFormatMetaDir):
143
"""The second working tree format updated to record a format marker.
146
- exists within a metadir controlling .bzr
147
- includes an explicit version marker for the workingtree control
148
files, separate from the ControlDir format
149
- modifies the hash cache format
151
- uses a LockDir to guard access for writes.
154
upgrade_recommended = True
156
missing_parent_conflicts = True
158
supports_versioned_directories = True
161
def get_format_string(cls):
162
"""See WorkingTreeFormat.get_format_string()."""
163
return "Bazaar-NG Working Tree format 3"
165
def get_format_description(self):
166
"""See WorkingTreeFormat.get_format_description()."""
167
return "Working tree format 3"
169
_tree_class = WorkingTree3
171
def __get_matchingbzrdir(self):
172
return bzrdir.BzrDirMetaFormat1()
174
_matchingbzrdir = property(__get_matchingbzrdir)
176
def _open_control_files(self, a_bzrdir):
177
transport = a_bzrdir.get_workingtree_transport(None)
178
return LockableFiles(transport, 'lock', LockDir)
180
def initialize(self, a_bzrdir, revision_id=None, from_branch=None,
181
accelerator_tree=None, hardlink=False):
182
"""See WorkingTreeFormat.initialize().
184
:param revision_id: if supplied, create a working tree at a different
185
revision than the branch is at.
186
:param accelerator_tree: A tree which can be used for retrieving file
187
contents more quickly than the revision tree, i.e. a workingtree.
188
The revision tree will be used for cases where accelerator_tree's
189
content is different.
190
:param hardlink: If true, hard-link files from accelerator_tree,
193
if not isinstance(a_bzrdir.transport, LocalTransport):
194
raise errors.NotLocalUrl(a_bzrdir.transport.base)
195
transport = a_bzrdir.get_workingtree_transport(self)
196
control_files = self._open_control_files(a_bzrdir)
197
control_files.create_lock()
198
control_files.lock_write()
199
transport.put_bytes('format', self.as_string(),
200
mode=a_bzrdir._get_file_mode())
201
if from_branch is not None:
204
branch = a_bzrdir.open_branch()
205
if revision_id is None:
206
revision_id = _mod_revision.ensure_null(branch.last_revision())
207
# WorkingTree3 can handle an inventory which has a unique root id.
208
# as of bzr 0.12. However, bzr 0.11 and earlier fail to handle
209
# those trees. And because there isn't a format bump inbetween, we
210
# are maintaining compatibility with older clients.
211
# inv = Inventory(root_id=gen_root_id())
212
inv = self._initial_inventory()
213
wt = self._tree_class(a_bzrdir.root_transport.local_abspath('.'),
219
_control_files=control_files)
222
basis_tree = branch.repository.revision_tree(revision_id)
223
# only set an explicit root id if there is one to set.
224
if basis_tree.inventory.root is not None:
225
wt.set_root_id(basis_tree.get_root_id())
226
if revision_id == _mod_revision.NULL_REVISION:
227
wt.set_parent_trees([])
229
wt.set_parent_trees([(revision_id, basis_tree)])
230
transform.build_tree(basis_tree, wt)
231
for hook in MutableTree.hooks['post_build_tree']:
234
# Unlock in this order so that the unlock-triggers-flush in
235
# WorkingTree is given a chance to fire.
236
control_files.unlock()
240
def _initial_inventory(self):
241
return inventory.Inventory()
243
def open(self, a_bzrdir, _found=False):
244
"""Return the WorkingTree object for a_bzrdir
246
_found is a private parameter, do not use it. It is used to indicate
247
if format probing has already been done.
250
# we are being called directly and must probe.
251
raise NotImplementedError
252
if not isinstance(a_bzrdir.transport, LocalTransport):
253
raise errors.NotLocalUrl(a_bzrdir.transport.base)
254
wt = self._open(a_bzrdir, self._open_control_files(a_bzrdir))
257
def _open(self, a_bzrdir, control_files):
258
"""Open the tree itself.
260
:param a_bzrdir: the dir for the tree.
261
:param control_files: the control files for the tree.
263
return self._tree_class(a_bzrdir.root_transport.local_abspath('.'),
267
_control_files=control_files)