/brz/remove-bazaar

# Copyright (C) 2008 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

from bzrlib import (

    merge3,

    osutils,

    pack,

    transform,

)

class ShelfCreator(object):

    """Create a transform to shelve objects and its inverse."""

    def __init__(self, work_tree, target_tree, file_list=None):

        """Constructor.

        :param work_tree: The working tree to apply changes to

        :param target_tree: The tree to make the working tree more similar to.

        :param file_list: The files to make more similar to the target.

        """

        self.work_tree = work_tree

        self.work_transform = transform.TreeTransform(work_tree)

        self.target_tree = target_tree

        self.shelf_transform = transform.TransformPreview(self.target_tree)

        self.renames = {}

        self.creation = {}

        self.deletion = {}

        self.iter_changes = work_tree.iter_changes(self.target_tree,

                                                   specific_files=file_list)

    def iter_shelvable(self):

        """Iterable of tuples describing shelvable changes.

        As well as generating the tuples, this updates several members.

        Tuples may be:

           ('add file', file_id, work_kind, work_path)

           ('delete file', file_id, target_kind, target_path)

           ('rename', file_id, target_path, work_path)

           ('change kind', file_id, target_kind, work_kind, target_path)

           ('modify text', file_id)

        """

        for (file_id, paths, changed, versioned, parents, names, kind,

             executable) in self.iter_changes:

            if kind[0] is None or versioned[0] == False:

                self.creation[file_id] = (kind[1], names[1], parents[1],

                                          versioned)

                yield ('add file', file_id, kind[1], paths[1])

            elif kind[1] is None or versioned[0] == False:

                self.deletion[file_id] = (kind[0], names[0], parents[0],

                                          versioned)

                yield ('delete file', file_id, kind[0], paths[0])

            else:

                if names[0] != names[1] or parents[0] != parents[1]:

                    self.renames[file_id] = (names, parents)

                    yield ('rename', file_id) + paths

                if kind[0] != kind [1]:

                    yield ('change kind', file_id, kind[0], kind[1], paths[0])

                elif changed:

                    yield ('modify text', file_id)

    def shelve_rename(self, file_id):

        """Shelve a file rename.

        :param file_id: The file id of the file to shelve the renaming of.

        """

        names, parents = self.renames[file_id]

        w_trans_id = self.work_transform.trans_id_file_id(file_id)

        work_parent = self.work_transform.trans_id_file_id(parents[0])

        self.work_transform.adjust_path(names[0], work_parent, w_trans_id)

        s_trans_id = self.shelf_transform.trans_id_file_id(file_id)

        shelf_parent = self.shelf_transform.trans_id_file_id(parents[1])

        self.shelf_transform.adjust_path(names[1], shelf_parent, s_trans_id)

    def shelve_lines(self, file_id, new_lines):

        """Shelve text changes to a file, using provided lines.

        :param file_id: The file id of the file to shelve the text of.

        :param new_lines: The lines that the file should have due to shelving.

        """

        w_trans_id = self.work_transform.trans_id_file_id(file_id)

        self.work_transform.delete_contents(w_trans_id)

        self.work_transform.create_file(new_lines, w_trans_id)

        s_trans_id = self.shelf_transform.trans_id_file_id(file_id)

        self.shelf_transform.delete_contents(s_trans_id)

        inverse_lines = self._inverse_lines(new_lines, file_id)

        self.shelf_transform.create_file(inverse_lines, s_trans_id)

    @staticmethod

    def _content_from_tree(tt, tree, file_id):

        trans_id = tt.trans_id_file_id(file_id)

        tt.delete_contents(trans_id)

        transform.create_from_tree(tt, trans_id, tree, file_id)

    def shelve_content_change(self, file_id):

        """Shelve a kind change or binary file content change.

        :param file_id: The file id of the file to shelve the content change

            of.

        """

        self._content_from_tree(self.work_transform, self.target_tree, file_id)

        self._content_from_tree(self.shelf_transform, self.work_tree, file_id)

    def shelve_creation(self, file_id):

        """Shelve creation of a file.

        This handles content and inventory id.

        :param file_id: The file_id of the file to shelve creation of.

        """

        kind, name, parent, versioned = self.creation[file_id]

        version = not versioned[0]

        self._shelve_creation(self.work_tree, file_id, self.work_transform,

                              self.shelf_transform, kind, name, parent,

                              version)

    def shelve_deletion(self, file_id):

        """Shelve deletion of a file.

        This handles content and inventory id.

        :param file_id: The file_id of the file to shelve deletion of.

        """

        kind, name, parent, versioned = self.deletion[file_id]

        existing_path = self.target_tree.id2path(file_id)

        if not self.work_tree.has_filename(existing_path):

            existing_path = None

        version = not versioned[1]

        self._shelve_creation(self.target_tree, file_id, self.shelf_transform,

                              self.work_transform, kind, name, parent,

                              version, existing_path=existing_path)

    def _shelve_creation(self, tree, file_id, from_transform, to_transform,

                         kind, name, parent, version, existing_path=None):

        w_trans_id = from_transform.trans_id_file_id(file_id)

        if parent is not None and kind is not None:

            from_transform.delete_contents(w_trans_id)

        from_transform.unversion_file(w_trans_id)

        if existing_path is not None:

            s_trans_id = to_transform.trans_id_tree_path(existing_path)

        else:

            s_trans_id = to_transform.trans_id_file_id(file_id)

        if parent is not None:

            s_parent_id = to_transform.trans_id_file_id(parent)

            to_transform.adjust_path(name, s_parent_id, s_trans_id)

            if existing_path is None:

                if kind is None:

                    to_transform.create_file('', s_trans_id)

                else:

                    transform.create_from_tree(to_transform, s_trans_id,

                                               tree, file_id)

        if version:

            to_transform.version_file(file_id, s_trans_id)

    def read_tree_lines(self, tree, file_id):

        """Read text lines from a tree.

        (Tree.get_file_lines is not an official API)

        """

        return osutils.split_lines(tree.get_file_text(file_id))

    def _inverse_lines(self, new_lines, file_id):

        """Produce a version with only those changes removed from new_lines."""

        target_lines = self.target_tree.get_file_lines(file_id)

        work_lines = self.read_tree_lines(self.work_tree, file_id)

        return merge3.Merge3(new_lines, target_lines, work_lines).merge_lines()

    def finalize(self):

        """Release all resources used by this ShelfCreator."""

        self.work_transform.finalize()

        self.shelf_transform.finalize()

    def transform(self):

        """Shelve changes from working tree."""

        self.work_transform.apply()

    def make_shelf_filename(self):

        """Generate a filename for a shelf."""

        transport = self.work_tree.bzrdir.root_transport.clone('.shelf2')

        transport.ensure_base()

        return transport.local_abspath('01')

    def write_shelf(self):

        """Serialize the shelved changes to a file.

        :return: the filename of the written file.

        """

        filename = self.make_shelf_filename()

        shelf_file = open(filename, 'wb')

        try:

            serializer = pack.ContainerSerialiser()

            shelf_file.write(serializer.begin())

            for bytes in self.shelf_transform.serialize(serializer):

                shelf_file.write(bytes)

            shelf_file.write(serializer.end())

        finally:

            shelf_file.close()

        return filename