14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21
from .lazy_import import lazy_import
22
lazy_import(globals(), """
29
from breezy.bzr import (
35
class ShelfCorrupt(errors.BzrError):
37
_fmt = "Shelf corrupt."
40
class NoSuchShelfId(errors.BzrError):
42
_fmt = 'No changes are shelved with id "%(shelf_id)d".'
44
def __init__(self, shelf_id):
45
errors.BzrError.__init__(self, shelf_id=shelf_id)
48
class InvalidShelfId(errors.BzrError):
50
_fmt = '"%(invalid_id)s" is not a valid shelf id, try a number instead.'
52
def __init__(self, invalid_id):
53
errors.BzrError.__init__(self, invalid_id=invalid_id)
31
56
class ShelfCreator(object):
70
96
('modify text', file_id)
71
97
('modify target', file_id, target_target, work_target)
73
for (file_id, paths, changed, versioned, parents, names, kind,
74
executable) in self.iter_changes:
99
for change in self.iter_changes:
75
100
# don't shelve add of tree root. Working tree should never
76
101
# lack roots, and bzr misbehaves when they do.
77
102
# FIXME ADHB (2009-08-09): should still shelve adds of tree roots
78
103
# when a tree root was deleted / renamed.
79
if kind[0] is None and names[1] == '':
81
if kind[0] is None or versioned[0] == False:
82
self.creation[file_id] = (kind[1], names[1], parents[1],
84
yield ('add file', file_id, kind[1], paths[1])
85
elif kind[1] is None or versioned[0] == False:
86
self.deletion[file_id] = (kind[0], names[0], parents[0],
88
yield ('delete file', file_id, kind[0], paths[0])
104
if change.kind[0] is None and change.name[1] == '':
106
# Also don't shelve deletion of tree root.
107
if change.kind[1] is None and change.name[0] == '':
109
if change.kind[0] is None or change.versioned[0] is False:
110
self.creation[change.file_id] = (
111
change.kind[1], change.name[1], change.parent_id[1], change.versioned)
112
yield ('add file', change.file_id, change.kind[1], change.path[1])
113
elif change.kind[1] is None or change.versioned[0] is False:
114
self.deletion[change.file_id] = (
115
change.kind[0], change.name[0], change.parent_id[0], change.versioned)
116
yield ('delete file', change.file_id, change.kind[0], change.path[0])
90
if names[0] != names[1] or parents[0] != parents[1]:
91
self.renames[file_id] = (names, parents)
92
yield ('rename', file_id) + paths
118
if change.name[0] != change.name[1] or change.parent_id[0] != change.parent_id[1]:
119
self.renames[change.file_id] = (change.name, change.parent_id)
120
yield ('rename', change.file_id) + change.path
94
if kind[0] != kind [1]:
95
yield ('change kind', file_id, kind[0], kind[1], paths[0])
96
elif kind[0] == 'symlink':
97
t_target = self.target_tree.get_symlink_target(file_id)
98
w_target = self.work_tree.get_symlink_target(file_id)
99
yield ('modify target', file_id, paths[0], t_target,
102
yield ('modify text', file_id)
122
if change.kind[0] != change.kind[1]:
123
yield ('change kind', change.file_id, change.kind[0], change.kind[1], change.path[0])
124
elif change.kind[0] == 'symlink':
125
t_target = self.target_tree.get_symlink_target(change.path[0])
126
w_target = self.work_tree.get_symlink_target(change.path[1])
127
yield ('modify target', change.file_id, change.path[0], t_target,
129
elif change.changed_content:
130
yield ('modify text', change.file_id)
104
132
def shelve_change(self, change):
105
133
"""Shelve a change in the iter_shelvable format."""
142
175
:param new_target: The target that the symlink should have due
145
new_target = self.target_tree.get_symlink_target(file_id)
178
new_path = self.target_tree.id2path(file_id)
179
new_target = self.target_tree.get_symlink_target(new_path)
146
180
w_trans_id = self.work_transform.trans_id_file_id(file_id)
147
181
self.work_transform.delete_contents(w_trans_id)
148
182
self.work_transform.create_symlink(new_target, w_trans_id)
150
old_target = self.work_tree.get_symlink_target(file_id)
184
old_path = self.work_tree.id2path(file_id)
185
old_target = self.work_tree.get_symlink_target(old_path)
151
186
s_trans_id = self.shelf_transform.trans_id_file_id(file_id)
152
187
self.shelf_transform.delete_contents(s_trans_id)
153
188
self.shelf_transform.create_symlink(old_target, s_trans_id)
225
260
to_transform.adjust_path(name, s_parent_id, s_trans_id)
226
261
if existing_path is None:
228
to_transform.create_file('', s_trans_id)
263
to_transform.create_file([b''], s_trans_id)
230
transform.create_from_tree(to_transform, s_trans_id,
265
transform.create_from_tree(
266
to_transform, s_trans_id, tree,
267
tree.id2path(file_id))
233
to_transform.version_file(file_id, s_trans_id)
269
to_transform.version_file(s_trans_id, file_id=file_id)
235
271
def _inverse_lines(self, new_lines, file_id):
236
272
"""Produce a version with only those changes removed from new_lines."""
237
target_lines = self.target_tree.get_file_lines(file_id)
238
work_lines = self.work_tree.get_file_lines(file_id)
273
target_path = self.target_tree.id2path(file_id)
274
target_lines = self.target_tree.get_file_lines(target_path)
275
work_path = self.work_tree.id2path(file_id)
276
work_lines = self.work_tree.get_file_lines(work_path)
239
277
return merge3.Merge3(new_lines, target_lines, work_lines).merge_lines()
241
279
def finalize(self):
247
285
"""Shelve changes from working tree."""
248
286
self.work_transform.apply()
289
def metadata_record(serializer, revision_id, message=None):
290
metadata = {b'revision_id': revision_id}
291
if message is not None:
292
metadata[b'message'] = message.encode('utf-8')
293
return serializer.bytes_record(
294
bencode.bencode(metadata), ((b'metadata',),))
250
296
def write_shelf(self, shelf_file, message=None):
251
297
"""Serialize the shelved changes to a file.
255
301
:return: the filename of the written file.
257
303
transform.resolve_conflicts(self.shelf_transform)
304
revision_id = self.target_tree.get_revision_id()
305
return self._write_shelf(shelf_file, self.shelf_transform, revision_id,
309
def _write_shelf(cls, shelf_file, transform, revision_id, message=None):
258
310
serializer = pack.ContainerSerialiser()
259
311
shelf_file.write(serializer.begin())
261
'revision_id': self.target_tree.get_revision_id(),
263
if message is not None:
264
metadata['message'] = message.encode('utf-8')
265
shelf_file.write(serializer.bytes_record(
266
bencode.bencode(metadata), (('metadata',),)))
267
for bytes in self.shelf_transform.serialize(serializer):
312
metadata = cls.metadata_record(serializer, revision_id, message)
313
shelf_file.write(metadata)
314
for bytes in transform.serialize(serializer):
268
315
shelf_file.write(bytes)
269
316
shelf_file.write(serializer.end())
294
341
def parse_metadata(records):
295
names, metadata_bytes = records.next()
296
if names[0] != ('metadata',):
297
raise errors.ShelfCorrupt
342
names, metadata_bytes = next(records)
343
if names[0] != (b'metadata',):
298
345
metadata = bencode.bdecode(metadata_bytes)
299
message = metadata.get('message')
346
message = metadata.get(b'message')
300
347
if message is not None:
301
metadata['message'] = message.decode('utf-8')
348
metadata[b'message'] = message.decode('utf-8')
312
359
records = klass.iter_records(shelf_file)
313
360
metadata = klass.parse_metadata(records)
314
base_revision_id = metadata['revision_id']
361
base_revision_id = metadata[b'revision_id']
316
363
base_tree = tree.revision_tree(base_revision_id)
317
364
except errors.NoSuchRevisionInTree:
318
365
base_tree = tree.branch.repository.revision_tree(base_revision_id)
319
tt = transform.TransformPreview(base_tree)
366
tt = base_tree.preview_transform()
320
367
tt.deserialize(records)
321
return klass(tree, base_tree, tt, metadata.get('message'))
368
return klass(tree, base_tree, tt, metadata.get(b'message'))
323
def make_merger(self, task=None):
370
def make_merger(self):
324
371
"""Return a merger that can unshelve the changes."""
325
372
target_tree = self.transform.get_preview_tree()
326
373
merger = merge.Merger.from_uncommitted(self.tree, target_tree,
327
task, self.base_tree)
328
375
merger.merge_type = merge.Merge3Merger