19
19
See MemoryTree for more details.
28
28
revision as _mod_revision,
30
from bzrlib.decorators import needs_read_lock
31
from bzrlib.osutils import sha_file
32
from bzrlib.mutabletree import needs_tree_write_lock
33
from bzrlib.transport.memory import MemoryTransport
36
class MemoryTree(mutabletree.MutableTree):
30
from .bzr.inventory import Inventory
31
from .bzr.inventorytree import MutableInventoryTree
32
from .osutils import sha_file
33
from .transport.memory import MemoryTransport
36
class MemoryTree(MutableInventoryTree):
37
37
"""A MemoryTree is a specialisation of MutableTree.
39
39
It maintains nearly no state outside of read_lock and write_lock
44
44
def __init__(self, branch, revision_id):
45
45
"""Construct a MemoryTree for branch using revision_id."""
46
46
self.branch = branch
47
self.bzrdir = branch.bzrdir
47
self.controldir = branch.controldir
48
48
self._branch_revision_id = revision_id
50
50
self._lock_mode = None
52
@needs_tree_write_lock
52
def supports_symlinks(self):
55
def supports_tree_reference(self):
58
def get_config_stack(self):
59
return self.branch.get_config_stack()
61
def is_control_filename(self, filename):
62
# Memory tree doesn't have any control filenames
53
65
def _add(self, files, ids, kinds):
54
66
"""See MutableTree._add."""
55
for f, file_id, kind in zip(files, ids, kinds):
59
self._inventory.add_path(f, kind=kind)
61
self._inventory.add_path(f, kind=kind, file_id=file_id)
67
with self.lock_tree_write():
68
for f, file_id, kind in zip(files, ids, kinds):
70
st_mode = self._file_transport.stat(f).st_mode
71
if stat.S_ISREG(st_mode):
73
elif stat.S_ISLNK(st_mode):
75
elif stat.S_ISDIR(st_mode):
78
raise AssertionError('Unknown file kind')
80
self._inventory.add_path(f, kind=kind)
82
self._inventory.add_path(f, kind=kind, file_id=file_id)
63
84
def basis_tree(self):
64
85
"""See Tree.basis_tree()."""
77
98
missing files, so is a no-op.
80
def get_file(self, file_id, path=None):
101
def get_file(self, path):
81
102
"""See Tree.get_file."""
83
path = self.id2path(file_id)
84
103
return self._file_transport.get(path)
86
def get_file_sha1(self, file_id, path=None, stat_value=None):
105
def get_file_sha1(self, path, stat_value=None):
87
106
"""See Tree.get_file_sha1()."""
89
path = self.id2path(file_id)
90
107
stream = self._file_transport.get(path)
91
108
return sha_file(stream)
93
def get_root_id(self):
94
return self.path2id('')
96
110
def _comparison_data(self, entry, path):
97
111
"""See Tree._comparison_data."""
99
113
return None, False, None
100
114
return entry.kind, entry.executable, None
102
@needs_tree_write_lock
103
116
def rename_one(self, from_rel, to_rel):
104
file_id = self.path2id(from_rel)
105
to_dir, to_tail = os.path.split(to_rel)
106
to_parent_id = self.path2id(to_dir)
107
self._file_transport.move(from_rel, to_rel)
108
self._inventory.rename(file_id, to_parent_id, to_tail)
117
with self.lock_tree_write():
118
file_id = self.path2id(from_rel)
119
to_dir, to_tail = os.path.split(to_rel)
120
to_parent_id = self.path2id(to_dir)
121
self._file_transport.move(from_rel, to_rel)
122
self._inventory.rename(file_id, to_parent_id, to_tail)
110
124
def path_content_summary(self, path):
111
125
"""See Tree.path_content_summary."""
112
126
id = self.path2id(path)
114
128
return 'missing', None, None, None
129
kind = self.kind(path, id)
116
130
if kind == 'file':
117
131
bytes = self._file_transport.get_bytes(path)
118
132
size = len(bytes)
119
133
executable = self._inventory[id].executable
120
sha1 = None # no stat cache
134
sha1 = None # no stat cache
121
135
return (kind, size, executable, sha1)
122
136
elif kind == 'directory':
123
137
# memory tree does not support nested trees yet.
124
138
return kind, None, None, None
125
139
elif kind == 'symlink':
126
raise NotImplementedError('symlink support')
140
return kind, None, None, self._inventory[id].symlink_target
128
142
raise NotImplementedError('unknown kind')
130
def _file_size(self, entry, stat_value):
131
"""See Tree._file_size."""
134
return entry.text_size
137
144
def get_parent_ids(self):
138
145
"""See Tree.get_parent_ids.
140
147
This implementation returns the current cached value from
141
148
self._parent_ids.
143
return list(self._parent_ids)
150
with self.lock_read():
151
return list(self._parent_ids)
145
153
def has_filename(self, filename):
146
154
"""See Tree.has_filename()."""
147
155
return self._file_transport.has(filename)
149
def is_executable(self, file_id, path=None):
150
return self._inventory[file_id].executable
157
def is_executable(self, path):
158
return self._inventory.get_entry_by_path(path).executable
152
def kind(self, file_id):
153
return self._inventory[file_id].kind
160
def kind(self, path):
161
return self._inventory.get_entry_by_path(path).kind
155
163
def mkdir(self, path, file_id=None):
156
164
"""See MutableTree.mkdir()."""
215
226
self._parent_ids = []
217
228
self._parent_ids = [self._branch_revision_id]
218
self._inventory = self._basis_tree._inventory._get_mutable_inventory()
229
self._inventory = Inventory(None, self._basis_tree.get_revision_id())
219
230
self._file_transport = MemoryTransport()
220
231
# TODO copy the revision trees content, or do it lazy, or something.
221
inventory_entries = self._inventory.iter_entries()
232
inventory_entries = self._basis_tree.iter_entries_by_dir()
222
233
for path, entry in inventory_entries:
234
self._inventory.add(entry.copy())
225
237
if entry.kind == 'directory':
226
238
self._file_transport.mkdir(path)
239
elif entry.kind == 'symlink':
240
self._file_transport.symlink(entry.symlink_target, path)
227
241
elif entry.kind == 'file':
228
self._file_transport.put_file(path,
229
self._basis_tree.get_file(entry.file_id))
242
self._file_transport.put_file(
243
path, self._basis_tree.get_file(path))
231
245
raise NotImplementedError(self._populate_from_branch)
233
def put_file_bytes_non_atomic(self, file_id, bytes):
247
def put_file_bytes_non_atomic(self, path, bytes):
234
248
"""See MutableTree.put_file_bytes_non_atomic."""
235
self._file_transport.put_bytes(self.id2path(file_id), bytes)
249
self._file_transport.put_bytes(path, bytes)
237
251
def unlock(self):
238
252
"""Release a lock.
255
@needs_tree_write_lock
256
def unversion(self, file_ids):
257
"""Remove the file ids in file_ids from the current versioned set.
269
def unversion(self, paths):
270
"""Remove the paths from the current versioned set.
259
272
When a file_id is unversioned, all of its children are automatically
262
:param file_ids: The file ids to stop versioning.
275
:param paths: The paths to stop versioning.
263
276
:raises: NoSuchId if any fileid is not currently versioned.
265
# XXX: This should be in mutabletree, but the inventory-save action
266
# is not relevant to memory tree. Until that is done in unlock by
267
# working tree, we cannot share the implementation.
268
for file_id in file_ids:
269
if self._inventory.has_id(file_id):
270
self._inventory.remove_recursive_id(file_id)
272
raise errors.NoSuchId(self, file_id)
278
with self.lock_tree_write():
279
# XXX: This should be in mutabletree, but the inventory-save action
280
# is not relevant to memory tree. Until that is done in unlock by
281
# working tree, we cannot share the implementation.
284
file_id = self.path2id(path)
286
raise errors.NoSuchFile(path)
287
file_ids.add(file_id)
288
for file_id in file_ids:
289
if self._inventory.has_id(file_id):
290
self._inventory.remove_recursive_id(file_id)
274
292
def set_parent_ids(self, revision_ids, allow_leftmost_as_ghost=False):
275
293
"""See MutableTree.set_parent_trees()."""
316
def get_symlink_target(self, path):
317
with self.lock_read():
318
return self._file_transport.readlink(path)
298
320
def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
299
321
"""See MutableTree.set_parent_trees()."""
300
322
if len(parents_list) == 0:
301
323
self._parent_ids = []
302
324
self._basis_tree = self.branch.repository.revision_tree(
303
_mod_revision.NULL_REVISION)
325
_mod_revision.NULL_REVISION)
305
327
if parents_list[0][1] is None and not allow_leftmost_as_ghost:
306
328
# a ghost in the left most parent
307
329
raise errors.GhostRevisionUnusableHere(parents_list[0][0])
308
330
self._parent_ids = [parent_id for parent_id, tree in parents_list]
309
if parents_list[0][1] is None or parents_list[0][1] == 'null:':
331
if parents_list[0][1] is None or parents_list[0][1] == b'null:':
310
332
self._basis_tree = self.branch.repository.revision_tree(
311
_mod_revision.NULL_REVISION)
333
_mod_revision.NULL_REVISION)
313
335
self._basis_tree = parents_list[0][1]
314
336
self._branch_revision_id = parents_list[0][0]