1
# Copyright (C) 2005 by Canonical Ltd
4
# Johan Rydberg <jrydberg@gnu.org>
6
# This program is free software; you can redistribute it and/or modify
7
# it under the terms of the GNU General Public License as published by
8
# the Free Software Foundation; either version 2 of the License, or
9
# (at your option) any later version.
11
# This program is distributed in the hope that it will be useful,
12
# but WITHOUT ANY WARRANTY; without even the implied warranty of
13
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
# GNU General Public License for more details.
16
# You should have received a copy of the GNU General Public License
17
# along with this program; if not, write to the Free Software
18
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20
# Remaing to do is to figure out if get_graph should return a simple
21
# map, or a graph object of some kind.
24
"""Versioned text file storage api."""
27
from copy import deepcopy
28
from unittest import TestSuite
31
from bzrlib.inter import InterObject
32
from bzrlib.symbol_versioning import *
33
from bzrlib.transport.memory import MemoryTransport
34
from bzrlib.tsort import topo_sort
37
class VersionedFile(object):
38
"""Versioned text file storage.
40
A versioned file manages versions of line-based text files,
41
keeping track of the originating version for each line.
43
To clients the "lines" of the file are represented as a list of
44
strings. These strings will typically have terminal newline
45
characters, but this is not required. In particular files commonly
46
do not have a newline at the end of the file.
48
Texts are identified by a version-id string.
51
def copy_to(self, name, transport):
52
"""Copy this versioned file to name on transport."""
53
raise NotImplementedError(self.copy_to)
55
@deprecated_method(zero_eight)
57
"""Return a list of all the versions in this versioned file.
59
Please use versionedfile.versions() now.
61
return self.versions()
64
"""Return a unsorted list of versions."""
65
raise NotImplementedError(self.versions)
67
def has_version(self, version_id):
68
"""Returns whether version is present."""
69
raise NotImplementedError(self.has_version)
71
def add_lines(self, version_id, parents, lines):
72
"""Add a single text on top of the versioned file.
74
Must raise RevisionAlreadyPresent if the new version is
75
already present in file history.
77
Must raise RevisionNotPresent if any of the given parents are
78
not present in file history."""
79
raise NotImplementedError(self.add_lines)
81
def check(self, progress_bar=None):
82
"""Check the versioned file for integrity."""
83
raise NotImplementedError(self.check)
85
def clear_cache(self):
86
"""Remove any data cached in the versioned file object."""
88
def clone_text(self, new_version_id, old_version_id, parents):
89
"""Add an identical text to old_version_id as new_version_id.
91
Must raise RevisionNotPresent if the old version or any of the
92
parents are not present in file history.
94
Must raise RevisionAlreadyPresent if the new version is
95
already present in file history."""
96
raise NotImplementedError(self.clone_text)
98
def create_empty(self, name, transport, mode=None):
99
"""Create a new versioned file of this exact type.
101
:param name: the file name
102
:param transport: the transport
103
:param mode: optional file mode.
105
raise NotImplementedError(self.create_empty)
107
def get_suffixes(self):
108
"""Return the file suffixes associated with this versioned file."""
109
raise NotImplementedError(self.get_suffixes)
111
def get_text(self, version_id):
112
"""Return version contents as a text string.
114
Raises RevisionNotPresent if version is not present in
117
return ''.join(self.get_lines(version_id))
118
get_string = get_text
120
def get_lines(self, version_id):
121
"""Return version contents as a sequence of lines.
123
Raises RevisionNotPresent if version is not present in
126
raise NotImplementedError(self.get_lines)
128
def get_ancestry(self, version_ids):
129
"""Return a list of all ancestors of given version(s). This
130
will not include the null revision.
132
Must raise RevisionNotPresent if any of the given versions are
133
not present in file history."""
134
if isinstance(version_ids, basestring):
135
version_ids = [version_ids]
136
raise NotImplementedError(self.get_ancestry)
139
"""Return a graph for the entire versioned file."""
141
for version in self.versions():
142
result[version] = self.get_parents(version)
145
@deprecated_method(zero_eight)
146
def parent_names(self, version):
147
"""Return version names for parents of a version.
149
See get_parents for the current api.
151
return self.get_parents(version)
153
def get_parents(self, version_id):
154
"""Return version names for parents of a version.
156
Must raise RevisionNotPresent if version is not present in
159
raise NotImplementedError(self.get_parents)
161
def annotate_iter(self, version_id):
162
"""Yield list of (version-id, line) pairs for the specified
165
Must raise RevisionNotPresent if any of the given versions are
166
not present in file history.
168
raise NotImplementedError(self.annotate_iter)
170
def annotate(self, version_id):
171
return list(self.annotate_iter(version_id))
173
def join(self, other, pb=None, msg=None, version_ids=None):
174
"""Integrate versions from other into this versioned file.
176
If version_ids is None all versions from other should be
177
incorporated into this versioned file.
179
Must raise RevisionNotPresent if any of the specified versions
180
are not present in the other files history."""
181
return InterVersionedFile.get(other, self).join(pb, msg, version_ids)
183
def walk(self, version_ids=None):
184
"""Walk the versioned file as a weave-like structure, for
185
versions relative to version_ids. Yields sequence of (lineno,
186
insert, deletes, text) for each relevant line.
188
Must raise RevisionNotPresent if any of the specified versions
189
are not present in the file history.
191
:param version_ids: the version_ids to walk with respect to. If not
192
supplied the entire weave-like structure is walked.
194
raise NotImplementedError(self.walk)
196
@deprecated_method(zero_eight)
197
def iter_names(self):
198
"""Walk the names list."""
199
return iter(self.versions())
201
def plan_merge(self, ver_a, ver_b):
202
"""Return pseudo-annotation indicating how the two versions merge.
204
This is computed between versions a and b and their common
207
Weave lines present in none of them are skipped entirely.
209
inc_a = set(self.inclusions([ver_a]))
210
inc_b = set(self.inclusions([ver_b]))
211
inc_c = inc_a & inc_b
213
for lineno, insert, deleteset, line in self.walk():
214
if deleteset & inc_c:
215
# killed in parent; can't be in either a or b
216
# not relevant to our work
217
yield 'killed-base', line
218
elif insert in inc_c:
219
# was inserted in base
220
killed_a = bool(deleteset & inc_a)
221
killed_b = bool(deleteset & inc_b)
222
if killed_a and killed_b:
223
yield 'killed-both', line
225
yield 'killed-a', line
227
yield 'killed-b', line
229
yield 'unchanged', line
230
elif insert in inc_a:
231
if deleteset & inc_a:
232
yield 'ghost-a', line
236
elif insert in inc_b:
237
if deleteset & inc_b:
238
yield 'ghost-b', line
242
# not in either revision
243
yield 'irrelevant', line
245
yield 'unchanged', '' # terminator
247
def weave_merge(self, plan, a_marker='<<<<<<< \n', b_marker='>>>>>>> \n'):
251
# TODO: Return a structured form of the conflicts (e.g. 2-tuples for
252
# conflicted regions), rather than just inserting the markers.
254
# TODO: Show some version information (e.g. author, date) on
255
# conflicted regions.
256
for state, line in plan:
257
if state == 'unchanged' or state == 'killed-both':
258
# resync and flush queued conflicts changes if any
259
if not lines_a and not lines_b:
261
elif ch_a and not ch_b:
263
for l in lines_a: yield l
264
elif ch_b and not ch_a:
265
for l in lines_b: yield l
266
elif lines_a == lines_b:
267
for l in lines_a: yield l
270
for l in lines_a: yield l
272
for l in lines_b: yield l
279
if state == 'unchanged':
282
elif state == 'killed-a':
285
elif state == 'killed-b':
288
elif state == 'new-a':
291
elif state == 'new-b':
295
assert state in ('irrelevant', 'ghost-a', 'ghost-b', 'killed-base',
300
class InterVersionedFile(InterObject):
301
"""This class represents operations taking place between two versionedfiles..
303
Its instances have methods like join, and contain
304
references to the source and target versionedfiles these operations can be
307
Often we will provide convenience methods on 'versionedfile' which carry out
308
operations with another versionedfile - they will always forward to
309
InterVersionedFile.get(other).method_name(parameters).
313
"""The available optimised InterVersionedFile types."""
315
def join(self, pb=None, msg=None, version_ids=None):
316
"""Integrate versions from self.source into self.target.
318
If version_ids is None all versions from source should be
319
incorporated into this versioned file.
321
Must raise RevisionNotPresent if any of the specified versions
322
are not present in the other files history.
325
# - make a temporary versioned file of type target
326
# - insert the source content into it one at a time
328
# Make a new target-format versioned file.
329
temp_source = self.target.create_empty("temp", MemoryTransport())
330
graph = self.source.get_graph()
331
order = topo_sort(graph.items())
332
for version in order:
333
temp_source.add_lines(version,
334
self.source.get_parents(version),
335
self.source.get_lines(version))
337
# this should hit the native code path for target
338
return self.target.join(temp_source, pb, msg, version_ids)
341
class InterVersionedFileTestProviderAdapter(object):
342
"""A tool to generate a suite testing multiple inter versioned-file classes.
344
This is done by copying the test once for each interversionedfile provider
345
and injecting the transport_server, transport_readonly_server,
346
versionedfile_factory and versionedfile_factory_to classes into each copy.
347
Each copy is also given a new id() to make it easy to identify.
350
def __init__(self, transport_server, transport_readonly_server, formats):
351
self._transport_server = transport_server
352
self._transport_readonly_server = transport_readonly_server
353
self._formats = formats
355
def adapt(self, test):
357
for (interversionedfile_class,
358
versionedfile_factory,
359
versionedfile_factory_to) in self._formats:
360
new_test = deepcopy(test)
361
new_test.transport_server = self._transport_server
362
new_test.transport_readonly_server = self._transport_readonly_server
363
new_test.interversionedfile_class = interversionedfile_class
364
new_test.versionedfile_factory = versionedfile_factory
365
new_test.versionedfile_factory_to = versionedfile_factory_to
366
def make_new_test_id():
367
new_id = "%s(%s)" % (new_test.id(), interversionedfile_class.__name__)
368
return lambda: new_id
369
new_test.id = make_new_test_id()
370
result.addTest(new_test)
374
def default_test_list():
375
"""Generate the default list of interversionedfile permutations to test."""
376
from bzrlib.weave import WeaveFile
377
from bzrlib.knit import KnitVersionedFile
379
# test the fallback InterVersionedFile from weave to annotated knits
380
result.append((InterVersionedFile,
383
for optimiser in InterVersionedFile._optimisers:
384
result.append((optimiser,
385
optimiser._matching_file_factory,
386
optimiser._matching_file_factory
388
# if there are specific combinations we want to use, we can add them