1
# Copyright (C) 2005, 2006, 2007 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 
17
from bzrlib.lazy_import import lazy_import
 
 
18
lazy_import(globals(), """
 
 
26
    revision as _mod_revision,
 
 
37
from bzrlib.decorators import needs_read_lock, needs_write_lock
 
 
38
from bzrlib.repository import (
 
 
41
    MetaDirRepositoryFormat,
 
 
45
from bzrlib.trace import mutter, mutter_callsite
 
 
48
class _KnitParentsProvider(object):
 
 
50
    def __init__(self, knit):
 
 
54
        return 'KnitParentsProvider(%r)' % self._knit
 
 
56
    @symbol_versioning.deprecated_method(symbol_versioning.one_one)
 
 
57
    def get_parents(self, revision_ids):
 
 
58
        """See graph._StackedParentsProvider.get_parents"""
 
 
59
        parent_map = self.get_parent_map(revision_ids)
 
 
60
        return [parent_map.get(r, None) for r in revision_ids]
 
 
62
    def get_parent_map(self, keys):
 
 
63
        """See graph._StackedParentsProvider.get_parent_map"""
 
 
65
        for revision_id in keys:
 
 
66
            if revision_id is None:
 
 
67
                raise ValueError('get_parent_map(None) is not valid')
 
 
68
            if revision_id == _mod_revision.NULL_REVISION:
 
 
69
                parent_map[revision_id] = ()
 
 
73
                        self._knit.get_parents_with_ghosts(revision_id))
 
 
74
                except errors.RevisionNotPresent:
 
 
78
                        parents = (_mod_revision.NULL_REVISION,)
 
 
79
                parent_map[revision_id] = parents
 
 
83
class _KnitsParentsProvider(object):
 
 
85
    def __init__(self, knit, prefix=()):
 
 
86
        """Create a parent provider for string keys mapped to tuple keys."""
 
 
91
        return 'KnitsParentsProvider(%r)' % self._knit
 
 
93
    def get_parent_map(self, keys):
 
 
94
        """See graph._StackedParentsProvider.get_parent_map"""
 
 
95
        parent_map = self._knit.get_parent_map(
 
 
96
            [self._prefix + (key,) for key in keys])
 
 
98
        for key, parents in parent_map.items():
 
 
100
            if len(parents) == 0:
 
 
101
                parents = (_mod_revision.NULL_REVISION,)
 
 
103
                parents = tuple(parent[-1] for parent in parents)
 
 
104
            result[revid] = parents
 
 
105
        for revision_id in keys:
 
 
106
            if revision_id == _mod_revision.NULL_REVISION:
 
 
107
                result[revision_id] = ()
 
 
111
class KnitRepository(MetaDirRepository):
 
 
112
    """Knit format repository."""
 
 
114
    # These attributes are inherited from the Repository base class. Setting
 
 
115
    # them to None ensures that if the constructor is changed to not initialize
 
 
116
    # them, or a subclass fails to call the constructor, that an error will
 
 
117
    # occur rather than the system working but generating incorrect data.
 
 
118
    _commit_builder_class = None
 
 
121
    def __init__(self, _format, a_bzrdir, control_files, _commit_builder_class,
 
 
123
        MetaDirRepository.__init__(self, _format, a_bzrdir, control_files)
 
 
124
        self._commit_builder_class = _commit_builder_class
 
 
125
        self._serializer = _serializer
 
 
126
        self._reconcile_fixes_text_parents = True
 
 
127
        self._fetch_uses_deltas = True
 
 
128
        self._fetch_order = 'topological'
 
 
131
    def _all_revision_ids(self):
 
 
132
        """See Repository.all_revision_ids()."""
 
 
133
        return [key[0] for key in self.revisions.keys()]
 
 
135
    def _activate_new_inventory(self):
 
 
136
        """Put a replacement inventory.new into use as inventories."""
 
 
137
        # Copy the content across
 
 
139
        t.copy('inventory.new.kndx', 'inventory.kndx')
 
 
141
            t.copy('inventory.new.knit', 'inventory.knit')
 
 
142
        except errors.NoSuchFile:
 
 
143
            # empty inventories knit
 
 
144
            t.delete('inventory.knit')
 
 
145
        # delete the temp inventory
 
 
146
        t.delete('inventory.new.kndx')
 
 
148
            t.delete('inventory.new.knit')
 
 
149
        except errors.NoSuchFile:
 
 
150
            # empty inventories knit
 
 
152
        # Force index reload (sanity check)
 
 
153
        self.inventories._index._reset_cache()
 
 
154
        self.inventories.keys()
 
 
156
    def _backup_inventory(self):
 
 
158
        t.copy('inventory.kndx', 'inventory.backup.kndx')
 
 
159
        t.copy('inventory.knit', 'inventory.backup.knit')
 
 
161
    def _move_file_id(self, from_id, to_id):
 
 
162
        t = self._transport.clone('knits')
 
 
163
        from_rel_url = self.texts._index._mapper.map((from_id, None))
 
 
164
        to_rel_url = self.texts._index._mapper.map((to_id, None))
 
 
165
        # We expect both files to always exist in this case.
 
 
166
        for suffix in ('.knit', '.kndx'):
 
 
167
            t.rename(from_rel_url + suffix, to_rel_url + suffix)
 
 
169
    def _remove_file_id(self, file_id):
 
 
170
        t = self._transport.clone('knits')
 
 
171
        rel_url = self.texts._index._mapper.map((file_id, None))
 
 
172
        for suffix in ('.kndx', '.knit'):
 
 
174
                t.delete(rel_url + suffix)
 
 
175
            except errors.NoSuchFile:
 
 
178
    def _temp_inventories(self):
 
 
179
        result = self._format._get_inventories(self._transport, self,
 
 
181
        # Reconciling when the output has no revisions would result in no
 
 
182
        # writes - but we want to ensure there is an inventory for
 
 
183
        # compatibility with older clients that don't lazy-load.
 
 
184
        result.get_parent_map([('A',)])
 
 
187
    def fileid_involved_between_revs(self, from_revid, to_revid):
 
 
188
        """Find file_id(s) which are involved in the changes between revisions.
 
 
190
        This determines the set of revisions which are involved, and then
 
 
191
        finds all file ids affected by those revisions.
 
 
193
        vf = self._get_revision_vf()
 
 
194
        from_set = set(vf.get_ancestry(from_revid))
 
 
195
        to_set = set(vf.get_ancestry(to_revid))
 
 
196
        changed = to_set.difference(from_set)
 
 
197
        return self._fileid_involved_by_set(changed)
 
 
199
    def fileid_involved(self, last_revid=None):
 
 
200
        """Find all file_ids modified in the ancestry of last_revid.
 
 
202
        :param last_revid: If None, last_revision() will be used.
 
 
205
            changed = set(self.all_revision_ids())
 
 
207
            changed = set(self.get_ancestry(last_revid))
 
 
210
        return self._fileid_involved_by_set(changed)
 
 
213
    def get_revision(self, revision_id):
 
 
214
        """Return the Revision object for a named revision"""
 
 
215
        revision_id = osutils.safe_revision_id(revision_id)
 
 
216
        return self.get_revision_reconcile(revision_id)
 
 
219
    def reconcile(self, other=None, thorough=False):
 
 
220
        """Reconcile this repository."""
 
 
221
        from bzrlib.reconcile import KnitReconciler
 
 
222
        reconciler = KnitReconciler(self, thorough=thorough)
 
 
223
        reconciler.reconcile()
 
 
226
    def _make_parents_provider(self):
 
 
227
        return _KnitsParentsProvider(self.revisions)
 
 
229
    def _find_inconsistent_revision_parents(self):
 
 
230
        """Find revisions with different parent lists in the revision object
 
 
231
        and in the index graph.
 
 
233
        :returns: an iterator yielding tuples of (revison-id, parents-in-index,
 
 
234
            parents-in-revision).
 
 
236
        if not self.is_locked():
 
 
237
            raise AssertionError()
 
 
239
        for index_version in vf.keys():
 
 
240
            parent_map = vf.get_parent_map([index_version])
 
 
241
            parents_according_to_index = tuple(parent[-1] for parent in
 
 
242
                parent_map[index_version])
 
 
243
            revision = self.get_revision(index_version[-1])
 
 
244
            parents_according_to_revision = tuple(revision.parent_ids)
 
 
245
            if parents_according_to_index != parents_according_to_revision:
 
 
246
                yield (index_version[-1], parents_according_to_index,
 
 
247
                    parents_according_to_revision)
 
 
249
    def _check_for_inconsistent_revision_parents(self):
 
 
250
        inconsistencies = list(self._find_inconsistent_revision_parents())
 
 
252
            raise errors.BzrCheckError(
 
 
253
                "Revision knit has inconsistent parents.")
 
 
255
    def revision_graph_can_have_wrong_parents(self):
 
 
256
        # The revision.kndx could potentially claim a revision has a different
 
 
257
        # parent to the revision text.
 
 
261
class RepositoryFormatKnit(MetaDirRepositoryFormat):
 
 
262
    """Bzr repository knit format (generalized). 
 
 
264
    This repository format has:
 
 
265
     - knits for file texts and inventory
 
 
266
     - hash subdirectory based stores.
 
 
267
     - knits for revisions and signatures
 
 
268
     - TextStores for revisions and signatures.
 
 
269
     - a format marker of its own
 
 
270
     - an optional 'shared-storage' flag
 
 
271
     - an optional 'no-working-trees' flag
 
 
275
    # Set this attribute in derived classes to control the repository class
 
 
276
    # created by open and initialize.
 
 
277
    repository_class = None
 
 
278
    # Set this attribute in derived classes to control the
 
 
279
    # _commit_builder_class that the repository objects will have passed to
 
 
281
    _commit_builder_class = None
 
 
282
    # Set this attribute in derived clases to control the _serializer that the
 
 
283
    # repository objects will have passed to their constructor.
 
 
285
    def _serializer(self):
 
 
286
        return xml5.serializer_v5
 
 
287
    # Knit based repositories handle ghosts reasonably well.
 
 
288
    supports_ghosts = True
 
 
289
    # External lookups are not supported in this format.
 
 
290
    supports_external_lookups = False
 
 
292
    def _get_inventories(self, repo_transport, repo, name='inventory'):
 
 
293
        mapper = versionedfile.ConstantMapper(name)
 
 
294
        index = _mod_knit._KndxIndex(repo_transport, mapper,
 
 
295
            repo.get_transaction, repo.is_write_locked, repo.is_locked)
 
 
296
        access = _mod_knit._KnitKeyAccess(repo_transport, mapper)
 
 
297
        return _mod_knit.KnitVersionedFiles(index, access, annotated=False)
 
 
299
    def _get_revisions(self, repo_transport, repo):
 
 
300
        mapper = versionedfile.ConstantMapper('revisions')
 
 
301
        index = _mod_knit._KndxIndex(repo_transport, mapper,
 
 
302
            repo.get_transaction, repo.is_write_locked, repo.is_locked)
 
 
303
        access = _mod_knit._KnitKeyAccess(repo_transport, mapper)
 
 
304
        return _mod_knit.KnitVersionedFiles(index, access, max_delta_chain=0,
 
 
307
    def _get_signatures(self, repo_transport, repo):
 
 
308
        mapper = versionedfile.ConstantMapper('signatures')
 
 
309
        index = _mod_knit._KndxIndex(repo_transport, mapper,
 
 
310
            repo.get_transaction, repo.is_write_locked, repo.is_locked)
 
 
311
        access = _mod_knit._KnitKeyAccess(repo_transport, mapper)
 
 
312
        return _mod_knit.KnitVersionedFiles(index, access, max_delta_chain=0,
 
 
315
    def _get_texts(self, repo_transport, repo):
 
 
316
        mapper = versionedfile.HashEscapedPrefixMapper()
 
 
317
        base_transport = repo_transport.clone('knits')
 
 
318
        index = _mod_knit._KndxIndex(base_transport, mapper,
 
 
319
            repo.get_transaction, repo.is_write_locked, repo.is_locked)
 
 
320
        access = _mod_knit._KnitKeyAccess(base_transport, mapper)
 
 
321
        return _mod_knit.KnitVersionedFiles(index, access, max_delta_chain=200,
 
 
324
    def initialize(self, a_bzrdir, shared=False):
 
 
325
        """Create a knit format 1 repository.
 
 
327
        :param a_bzrdir: bzrdir to contain the new repository; must already
 
 
329
        :param shared: If true the repository will be initialized as a shared
 
 
332
        mutter('creating repository in %s.', a_bzrdir.transport.base)
 
 
335
        utf8_files = [('format', self.get_format_string())]
 
 
337
        self._upload_blank_content(a_bzrdir, dirs, files, utf8_files, shared)
 
 
338
        repo_transport = a_bzrdir.get_repository_transport(None)
 
 
339
        control_files = lockable_files.LockableFiles(repo_transport,
 
 
340
                                'lock', lockdir.LockDir)
 
 
341
        transaction = transactions.WriteTransaction()
 
 
342
        result = self.open(a_bzrdir=a_bzrdir, _found=True)
 
 
344
        # the revision id here is irrelevant: it will not be stored, and cannot
 
 
345
        # already exist, we do this to create files on disk for older clients.
 
 
346
        result.inventories.get_parent_map([('A',)])
 
 
347
        result.revisions.get_parent_map([('A',)])
 
 
348
        result.signatures.get_parent_map([('A',)])
 
 
352
    def open(self, a_bzrdir, _found=False, _override_transport=None):
 
 
353
        """See RepositoryFormat.open().
 
 
355
        :param _override_transport: INTERNAL USE ONLY. Allows opening the
 
 
356
                                    repository at a slightly different url
 
 
357
                                    than normal. I.e. during 'upgrade'.
 
 
360
            format = RepositoryFormat.find_format(a_bzrdir)
 
 
361
        if _override_transport is not None:
 
 
362
            repo_transport = _override_transport
 
 
364
            repo_transport = a_bzrdir.get_repository_transport(None)
 
 
365
        control_files = lockable_files.LockableFiles(repo_transport,
 
 
366
                                'lock', lockdir.LockDir)
 
 
367
        repo = self.repository_class(_format=self,
 
 
369
                              control_files=control_files,
 
 
370
                              _commit_builder_class=self._commit_builder_class,
 
 
371
                              _serializer=self._serializer)
 
 
372
        repo.revisions = self._get_revisions(repo_transport, repo)
 
 
373
        repo.signatures = self._get_signatures(repo_transport, repo)
 
 
374
        repo.inventories = self._get_inventories(repo_transport, repo)
 
 
375
        repo.texts = self._get_texts(repo_transport, repo)
 
 
376
        repo._transport = repo_transport
 
 
380
class RepositoryFormatKnit1(RepositoryFormatKnit):
 
 
381
    """Bzr repository knit format 1.
 
 
383
    This repository format has:
 
 
384
     - knits for file texts and inventory
 
 
385
     - hash subdirectory based stores.
 
 
386
     - knits for revisions and signatures
 
 
387
     - TextStores for revisions and signatures.
 
 
388
     - a format marker of its own
 
 
389
     - an optional 'shared-storage' flag
 
 
390
     - an optional 'no-working-trees' flag
 
 
393
    This format was introduced in bzr 0.8.
 
 
396
    repository_class = KnitRepository
 
 
397
    _commit_builder_class = CommitBuilder
 
 
399
    def _serializer(self):
 
 
400
        return xml5.serializer_v5
 
 
402
    def __ne__(self, other):
 
 
403
        return self.__class__ is not other.__class__
 
 
405
    def get_format_string(self):
 
 
406
        """See RepositoryFormat.get_format_string()."""
 
 
407
        return "Bazaar-NG Knit Repository Format 1"
 
 
409
    def get_format_description(self):
 
 
410
        """See RepositoryFormat.get_format_description()."""
 
 
411
        return "Knit repository format 1"
 
 
413
    def check_conversion_target(self, target_format):
 
 
417
class RepositoryFormatKnit3(RepositoryFormatKnit):
 
 
418
    """Bzr repository knit format 3.
 
 
420
    This repository format has:
 
 
421
     - knits for file texts and inventory
 
 
422
     - hash subdirectory based stores.
 
 
423
     - knits for revisions and signatures
 
 
424
     - TextStores for revisions and signatures.
 
 
425
     - a format marker of its own
 
 
426
     - an optional 'shared-storage' flag
 
 
427
     - an optional 'no-working-trees' flag
 
 
429
     - support for recording full info about the tree root
 
 
430
     - support for recording tree-references
 
 
433
    repository_class = KnitRepository
 
 
434
    _commit_builder_class = RootCommitBuilder
 
 
435
    rich_root_data = True
 
 
436
    supports_tree_reference = True
 
 
438
    def _serializer(self):
 
 
439
        return xml7.serializer_v7
 
 
441
    def _get_matching_bzrdir(self):
 
 
442
        return bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
 
 
444
    def _ignore_setting_bzrdir(self, format):
 
 
447
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
 
 
449
    def check_conversion_target(self, target_format):
 
 
450
        if not target_format.rich_root_data:
 
 
451
            raise errors.BadConversionTarget(
 
 
452
                'Does not support rich root data.', target_format)
 
 
453
        if not getattr(target_format, 'supports_tree_reference', False):
 
 
454
            raise errors.BadConversionTarget(
 
 
455
                'Does not support nested trees', target_format)
 
 
457
    def get_format_string(self):
 
 
458
        """See RepositoryFormat.get_format_string()."""
 
 
459
        return "Bazaar Knit Repository Format 3 (bzr 0.15)\n"
 
 
461
    def get_format_description(self):
 
 
462
        """See RepositoryFormat.get_format_description()."""
 
 
463
        return "Knit repository format 3"
 
 
466
class RepositoryFormatKnit4(RepositoryFormatKnit):
 
 
467
    """Bzr repository knit format 4.
 
 
469
    This repository format has everything in format 3, except for
 
 
471
     - knits for file texts and inventory
 
 
472
     - hash subdirectory based stores.
 
 
473
     - knits for revisions and signatures
 
 
474
     - TextStores for revisions and signatures.
 
 
475
     - a format marker of its own
 
 
476
     - an optional 'shared-storage' flag
 
 
477
     - an optional 'no-working-trees' flag
 
 
479
     - support for recording full info about the tree root
 
 
482
    repository_class = KnitRepository
 
 
483
    _commit_builder_class = RootCommitBuilder
 
 
484
    rich_root_data = True
 
 
485
    supports_tree_reference = False
 
 
487
    def _serializer(self):
 
 
488
        return xml6.serializer_v6
 
 
490
    def _get_matching_bzrdir(self):
 
 
491
        return bzrdir.format_registry.make_bzrdir('rich-root')
 
 
493
    def _ignore_setting_bzrdir(self, format):
 
 
496
    _matchingbzrdir = property(_get_matching_bzrdir, _ignore_setting_bzrdir)
 
 
498
    def check_conversion_target(self, target_format):
 
 
499
        if not target_format.rich_root_data:
 
 
500
            raise errors.BadConversionTarget(
 
 
501
                'Does not support rich root data.', target_format)
 
 
503
    def get_format_string(self):
 
 
504
        """See RepositoryFormat.get_format_string()."""
 
 
505
        return 'Bazaar Knit Repository Format 4 (bzr 1.0)\n'
 
 
507
    def get_format_description(self):
 
 
508
        """See RepositoryFormat.get_format_description()."""
 
 
509
        return "Knit repository format 4"