~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2005, 2006, 2007 Canonical Ltd

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""Old weave-based repository formats"""

from StringIO import StringIO

from bzrlib import (

    bzrdir,

    lockable_files,

    lockdir,

    weave,

    weavefile,

    xml5,

    )

from bzrlib.decorators import needs_read_lock, needs_write_lock

from bzrlib.repository import (

    MetaDirRepository,

    MetaDirRepositoryFormat,

    Repository,

    RepositoryFormat,

    )

from bzrlib.store.text import TextStore

from bzrlib.trace import mutter

class AllInOneRepository(Repository):

    """Legacy support - the repository behaviour for all-in-one branches."""

    _serializer = xml5.serializer_v5

    def __init__(self, _format, a_bzrdir, _revision_store, control_store, text_store):

        # we reuse one control files instance.

        dir_mode = a_bzrdir._control_files._dir_mode

        file_mode = a_bzrdir._control_files._file_mode

        def get_store(name, compressed=True, prefixed=False):

            # FIXME: This approach of assuming stores are all entirely compressed

            # or entirely uncompressed is tidy, but breaks upgrade from 

            # some existing branches where there's a mixture; we probably 

            # still want the option to look for both.

            relpath = a_bzrdir._control_files._escape(name)

            store = TextStore(a_bzrdir._control_files._transport.clone(relpath),

                              prefixed=prefixed, compressed=compressed,

                              dir_mode=dir_mode,

                              file_mode=file_mode)

            #if self._transport.should_cache():

            #    cache_path = os.path.join(self.cache_root, name)

            #    os.mkdir(cache_path)

            #    store = bzrlib.store.CachedStore(store, cache_path)

            return store

        # not broken out yet because the controlweaves|inventory_store

        # and text_store | weave_store bits are still different.

        if isinstance(_format, RepositoryFormat4):

            # cannot remove these - there is still no consistent api 

            # which allows access to this old info.

            self.inventory_store = get_store('inventory-store')

            text_store = get_store('text-store')

        super(AllInOneRepository, self).__init__(_format, a_bzrdir, a_bzrdir._control_files, _revision_store, control_store, text_store)

    def get_commit_builder(self, branch, parents, config, timestamp=None,

                           timezone=None, committer=None, revprops=None,

                           revision_id=None):

        self._check_ascii_revisionid(revision_id, self.get_commit_builder)

        return Repository.get_commit_builder(self, branch, parents, config,

            timestamp, timezone, committer, revprops, revision_id)

    @needs_read_lock

    def is_shared(self):

        """AllInOne repositories cannot be shared."""

        return False

    @needs_write_lock

    def set_make_working_trees(self, new_value):

        """Set the policy flag for making working trees when creating branches.

        This only applies to branches that use this repository.

        The default is 'True'.

        :param new_value: True to restore the default, False to disable making

                          working trees.

        """

        raise NotImplementedError(self.set_make_working_trees)

    def make_working_trees(self):

        """Returns the policy for making working trees on new branches."""

        return True

class WeaveMetaDirRepository(MetaDirRepository):

    """A subclass of MetaDirRepository to set weave specific policy."""

    _serializer = xml5.serializer_v5

    def get_commit_builder(self, branch, parents, config, timestamp=None,

                           timezone=None, committer=None, revprops=None,

                           revision_id=None):

        self._check_ascii_revisionid(revision_id, self.get_commit_builder)

        return MetaDirRepository.get_commit_builder(self, branch, parents,

            config, timestamp, timezone, committer, revprops, revision_id)

class PreSplitOutRepositoryFormat(RepositoryFormat):

    """Base class for the pre split out repository formats."""

    rich_root_data = False

    supports_tree_reference = False

    def initialize(self, a_bzrdir, shared=False, _internal=False):

        """Create a weave repository.

        TODO: when creating split out bzr branch formats, move this to a common

        base for Format5, Format6. or something like that.

        """

        if shared:

            raise errors.IncompatibleFormat(self, a_bzrdir._format)

        if not _internal:

            # always initialized when the bzrdir is.

            return self.open(a_bzrdir, _found=True)

        # Create an empty weave

        sio = StringIO()

        weavefile.write_weave_v5(weave.Weave(), sio)

        empty_weave = sio.getvalue()

        mutter('creating repository in %s.', a_bzrdir.transport.base)

        dirs = ['revision-store', 'weaves']

        files = [('inventory.weave', StringIO(empty_weave)),

                 ]

        # FIXME: RBC 20060125 don't peek under the covers

        # NB: no need to escape relative paths that are url safe.

        control_files = lockable_files.LockableFiles(a_bzrdir.transport,

                                'branch-lock', lockable_files.TransportLock)

        control_files.create_lock()

        control_files.lock_write()

        control_files._transport.mkdir_multi(dirs,

                mode=control_files._dir_mode)

        try:

            for file, content in files:

                control_files.put(file, content)

        finally:

            control_files.unlock()

        return self.open(a_bzrdir, _found=True)

    def _get_control_store(self, repo_transport, control_files):

        """Return the control store for this repository."""

        return self._get_versioned_file_store('',

                                              repo_transport,

                                              control_files,

                                              prefixed=False)

    def _get_text_store(self, transport, control_files):

        """Get a store for file texts for this format."""

        raise NotImplementedError(self._get_text_store)

    def open(self, a_bzrdir, _found=False):

        """See RepositoryFormat.open()."""

        if not _found:

            # we are being called directly and must probe.

            raise NotImplementedError

        repo_transport = a_bzrdir.get_repository_transport(None)

        control_files = a_bzrdir._control_files

        text_store = self._get_text_store(repo_transport, control_files)

        control_store = self._get_control_store(repo_transport, control_files)

        _revision_store = self._get_revision_store(repo_transport, control_files)

        return AllInOneRepository(_format=self,

                                  a_bzrdir=a_bzrdir,

                                  _revision_store=_revision_store,

                                  control_store=control_store,

                                  text_store=text_store)

    def check_conversion_target(self, target_format):

        pass

class RepositoryFormat4(PreSplitOutRepositoryFormat):

    """Bzr repository format 4.

    This repository format has:

     - flat stores

     - TextStores for texts, inventories,revisions.

    This format is deprecated: it indexes texts using a text id which is

    removed in format 5; initialization and write support for this format

    has been removed.

    """

    _matchingbzrdir = bzrdir.BzrDirFormat4()

    def __init__(self):

        super(RepositoryFormat4, self).__init__()

    def get_format_description(self):

        """See RepositoryFormat.get_format_description()."""

        return "Repository format 4"

    def initialize(self, url, shared=False, _internal=False):

        """Format 4 branches cannot be created."""

        raise errors.UninitializableFormat(self)

    def is_supported(self):

        """Format 4 is not supported.

        It is not supported because the model changed from 4 to 5 and the

        conversion logic is expensive - so doing it on the fly was not 

        feasible.

        """

        return False

    def _get_control_store(self, repo_transport, control_files):

        """Format 4 repositories have no formal control store at this point.

        This will cause any control-file-needing apis to fail - this is desired.

        """

        return None

    def _get_revision_store(self, repo_transport, control_files):

        """See RepositoryFormat._get_revision_store()."""

        from bzrlib.xml4 import serializer_v4

        return self._get_text_rev_store(repo_transport,

                                        control_files,

                                        'revision-store',

                                        serializer=serializer_v4)

    def _get_text_store(self, transport, control_files):

        """See RepositoryFormat._get_text_store()."""

class RepositoryFormat5(PreSplitOutRepositoryFormat):

    """Bzr control format 5.

    This repository format has:

     - weaves for file texts and inventory

     - flat stores

     - TextStores for revisions and signatures.

    """

    _versionedfile_class = weave.WeaveFile

    _matchingbzrdir = bzrdir.BzrDirFormat5()

    def __init__(self):

        super(RepositoryFormat5, self).__init__()

    def get_format_description(self):

        """See RepositoryFormat.get_format_description()."""

        return "Weave repository format 5"

    def _get_revision_store(self, repo_transport, control_files):

        """See RepositoryFormat._get_revision_store()."""

        """Return the revision store object for this a_bzrdir."""

        return self._get_text_rev_store(repo_transport,

                                        control_files,

                                        'revision-store',

                                        compressed=False)

    def _get_text_store(self, transport, control_files):

        """See RepositoryFormat._get_text_store()."""

        return self._get_versioned_file_store('weaves', transport, control_files, prefixed=False)

class RepositoryFormat6(PreSplitOutRepositoryFormat):

    """Bzr control format 6.

    This repository format has:

     - weaves for file texts and inventory

     - hash subdirectory based stores.

     - TextStores for revisions and signatures.

    """

    _versionedfile_class = weave.WeaveFile

    _matchingbzrdir = bzrdir.BzrDirFormat6()

    def __init__(self):

        super(RepositoryFormat6, self).__init__()

    def get_format_description(self):

        """See RepositoryFormat.get_format_description()."""

        return "Weave repository format 6"

    def _get_revision_store(self, repo_transport, control_files):

        """See RepositoryFormat._get_revision_store()."""

        return self._get_text_rev_store(repo_transport,

                                        control_files,

                                        'revision-store',

                                        compressed=False,

                                        prefixed=True)

    def _get_text_store(self, transport, control_files):

        """See RepositoryFormat._get_text_store()."""

        return self._get_versioned_file_store('weaves', transport, control_files)

class RepositoryFormat7(MetaDirRepositoryFormat):

    """Bzr repository 7.

    This repository format has:

     - weaves for file texts and inventory

     - hash subdirectory based stores.

     - TextStores for revisions and signatures.

     - a format marker of its own

     - an optional 'shared-storage' flag

     - an optional 'no-working-trees' flag

    """

    _versionedfile_class = weave.WeaveFile

    def _get_control_store(self, repo_transport, control_files):

        """Return the control store for this repository."""

        return self._get_versioned_file_store('',

                                              repo_transport,

                                              control_files,

                                              prefixed=False)

    def get_format_string(self):

        """See RepositoryFormat.get_format_string()."""

        return "Bazaar-NG Repository format 7"

    def get_format_description(self):

        """See RepositoryFormat.get_format_description()."""

        return "Weave repository format 7"

    def check_conversion_target(self, target_format):

        pass

    def _get_revision_store(self, repo_transport, control_files):

        """See RepositoryFormat._get_revision_store()."""

        return self._get_text_rev_store(repo_transport,

                                        control_files,

                                        'revision-store',

                                        compressed=False,

                                        prefixed=True,

                                        )

    def _get_text_store(self, transport, control_files):

        """See RepositoryFormat._get_text_store()."""

        return self._get_versioned_file_store('weaves',

                                              transport,

                                              control_files)

    def initialize(self, a_bzrdir, shared=False):

        """Create a weave repository.

        :param shared: If true the repository will be initialized as a shared

                       repository.

        """

        # Create an empty weave

        sio = StringIO()

        weavefile.write_weave_v5(weave.Weave(), sio)

        empty_weave = sio.getvalue()

        mutter('creating repository in %s.', a_bzrdir.transport.base)

        dirs = ['revision-store', 'weaves']

        files = [('inventory.weave', StringIO(empty_weave)), 

                 ]

        utf8_files = [('format', self.get_format_string())]

        self._upload_blank_content(a_bzrdir, dirs, files, utf8_files, shared)

        return self.open(a_bzrdir=a_bzrdir, _found=True)

    def open(self, a_bzrdir, _found=False, _override_transport=None):

        """See RepositoryFormat.open().

        :param _override_transport: INTERNAL USE ONLY. Allows opening the

                                    repository at a slightly different url

                                    than normal. I.e. during 'upgrade'.

        """

        if not _found:

            format = RepositoryFormat.find_format(a_bzrdir)

            assert format.__class__ ==  self.__class__

        if _override_transport is not None:

            repo_transport = _override_transport

        else:

            repo_transport = a_bzrdir.get_repository_transport(None)

        control_files = lockable_files.LockableFiles(repo_transport,

                                'lock', lockdir.LockDir)

        text_store = self._get_text_store(repo_transport, control_files)

        control_store = self._get_control_store(repo_transport, control_files)

        _revision_store = self._get_revision_store(repo_transport, control_files)

        return WeaveMetaDirRepository(_format=self,

            a_bzrdir=a_bzrdir,

            control_files=control_files,

            _revision_store=_revision_store,

            control_store=control_store,

            text_store=text_store)

_legacy_formats = [RepositoryFormat4(),

                   RepositoryFormat5(),

                   RepositoryFormat6()]