~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2005, 2006 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

"""WorkingTree object and friends.

A WorkingTree represents the editable working copy of a branch.

Operations which represent the WorkingTree are also done here, 

such as renaming or adding files.  The WorkingTree has an inventory 

which is updated by these operations.  A commit produces a 

new revision based on the workingtree and its inventory.

At the moment every WorkingTree has its own branch.  Remote

WorkingTrees aren't supported.

To get a WorkingTree, call bzrdir.open_workingtree() or

WorkingTree.open(dir).

"""

# TODO: Give the workingtree sole responsibility for the working inventory;

# remove the variable and references to it from the branch.  This may require

# updating the commit code so as to update the inventory within the working

# copy, and making sure there's only one WorkingTree for any directory on disk.

# At the moment they may alias the inventory and have old copies of it in

# memory.  (Now done? -- mbp 20060309)

from cStringIO import StringIO

import os

from bzrlib.lazy_import import lazy_import

lazy_import(globals(), """

import collections

from copy import deepcopy

import errno

import stat

from time import time

import warnings

import bzrlib

from bzrlib import (

    bzrdir,

    conflicts as _mod_conflicts,

    errors,

    generate_ids,

    globbing,

    hashcache,

    ignores,

    merge,

    osutils,

    textui,

    transform,

    urlutils,

    xml5,

    xml6,

    )

import bzrlib.branch

from bzrlib.transport import get_transport

import bzrlib.ui

""")

from bzrlib import symbol_versioning

from bzrlib.decorators import needs_read_lock, needs_write_lock

from bzrlib.errors import (BzrCheckError,

                           BzrError,

                           ConflictFormatError,

                           WeaveRevisionNotPresent,

                           NotBranchError,

                           NoSuchFile,

                           NotVersionedError,

                           MergeModifiedFormatError,

                           UnsupportedOperation,

                           )

from bzrlib.inventory import InventoryEntry, Inventory, ROOT_ID

from bzrlib.lockable_files import LockableFiles, TransportLock

from bzrlib.lockdir import LockDir

import bzrlib.mutabletree

from bzrlib.mutabletree import needs_tree_write_lock

from bzrlib.osutils import (

    compact_date,

    file_kind,

    isdir,

    pathjoin,

    safe_unicode,

    splitpath,

    rand_chars,

    normpath,

    realpath,

    supports_executable,

    )

from bzrlib.trace import mutter, note

from bzrlib.transport.local import LocalTransport

import bzrlib.tree

from bzrlib.progress import DummyProgress, ProgressPhase

from bzrlib.revision import NULL_REVISION, CURRENT_REVISION

import bzrlib.revisiontree

from bzrlib.rio import RioReader, rio_file, Stanza

from bzrlib.symbol_versioning import (deprecated_passed,

        deprecated_method,

        deprecated_function,

        DEPRECATED_PARAMETER,

        zero_eight,

        zero_eleven,

        zero_thirteen,

        )

MERGE_MODIFIED_HEADER_1 = "BZR merge-modified list format 1"

CONFLICT_HEADER_1 = "BZR conflict list format 1"

@deprecated_function(zero_thirteen)

def gen_file_id(name):

    """Return new file id for the basename 'name'.

    Use bzrlib.generate_ids.gen_file_id() instead

    """

    return generate_ids.gen_file_id(name)

@deprecated_function(zero_thirteen)

def gen_root_id():

    """Return a new tree-root file id.

    This has been deprecated in favor of bzrlib.generate_ids.gen_root_id()

    """

    return generate_ids.gen_root_id()

class TreeEntry(object):

    """An entry that implements the minimum interface used by commands.

    This needs further inspection, it may be better to have 

    InventoryEntries without ids - though that seems wrong. For now,

    this is a parallel hierarchy to InventoryEntry, and needs to become

    one of several things: decorates to that hierarchy, children of, or

    parents of it.

    Another note is that these objects are currently only used when there is

    no InventoryEntry available - i.e. for unversioned objects.

    Perhaps they should be UnversionedEntry et al. ? - RBC 20051003

    """

    def __eq__(self, other):

        # yes, this us ugly, TODO: best practice __eq__ style.

        return (isinstance(other, TreeEntry)

                and other.__class__ == self.__class__)

    def kind_character(self):

        return "???"

class TreeDirectory(TreeEntry):

    """See TreeEntry. This is a directory in a working tree."""

    def __eq__(self, other):

        return (isinstance(other, TreeDirectory)

                and other.__class__ == self.__class__)

    def kind_character(self):

        return "/"

class TreeFile(TreeEntry):

    """See TreeEntry. This is a regular file in a working tree."""

    def __eq__(self, other):

        return (isinstance(other, TreeFile)

                and other.__class__ == self.__class__)

    def kind_character(self):

        return ''

class TreeLink(TreeEntry):

    """See TreeEntry. This is a symlink in a working tree."""

    def __eq__(self, other):

        return (isinstance(other, TreeLink)

                and other.__class__ == self.__class__)

    def kind_character(self):

        return ''

class WorkingTree(bzrlib.mutabletree.MutableTree):

    """Working copy tree.

    The inventory is held in the `Branch` working-inventory, and the

    files are in a directory on disk.

    It is possible for a `WorkingTree` to have a filename which is

    not listed in the Inventory and vice versa.

    """

    def __init__(self, basedir='.',

                 branch=DEPRECATED_PARAMETER,

                 _inventory=None,

                 _control_files=None,

                 _internal=False,

                 _format=None,

                 _bzrdir=None):

        """Construct a WorkingTree for basedir.

        If the branch is not supplied, it is opened automatically.

        If the branch is supplied, it must be the branch for this basedir.

        (branch.base is not cross checked, because for remote branches that

        would be meaningless).

        """

        self._format = _format

        self.bzrdir = _bzrdir

        if not _internal:

            # not created via open etc.

            warnings.warn("WorkingTree() is deprecated as of bzr version 0.8. "

                 "Please use bzrdir.open_workingtree or WorkingTree.open().",

                 DeprecationWarning,

                 stacklevel=2)

            wt = WorkingTree.open(basedir)

            self._branch = wt.branch

            self.basedir = wt.basedir

            self._control_files = wt._control_files

            self._hashcache = wt._hashcache

            self._set_inventory(wt._inventory, dirty=False)

            self._format = wt._format

            self.bzrdir = wt.bzrdir

        assert isinstance(basedir, basestring), \

            "base directory %r is not a string" % basedir

        basedir = safe_unicode(basedir)

        mutter("opening working tree %r", basedir)

        if deprecated_passed(branch):

            if not _internal:

                warnings.warn("WorkingTree(..., branch=XXX) is deprecated as of bzr 0.8."

                     " Please use bzrdir.open_workingtree() or"

                     " WorkingTree.open().",

                     DeprecationWarning,

                     stacklevel=2

                     )

            self._branch = branch

        else:

            self._branch = self.bzrdir.open_branch()

        self.basedir = realpath(basedir)

        # if branch is at our basedir and is a format 6 or less

        if isinstance(self._format, WorkingTreeFormat2):

            # share control object

            self._control_files = self.branch.control_files

        else:

            # assume all other formats have their own control files.

            assert isinstance(_control_files, LockableFiles), \

                    "_control_files must be a LockableFiles, not %r" \

                    % _control_files

            self._control_files = _control_files

        # update the whole cache up front and write to disk if anything changed;

        # in the future we might want to do this more selectively

        # two possible ways offer themselves : in self._unlock, write the cache

        # if needed, or, when the cache sees a change, append it to the hash

        # cache file, and have the parser take the most recent entry for a

        # given path only.

        cache_filename = self.bzrdir.get_workingtree_transport(None).local_abspath('stat-cache')

        self._hashcache = hashcache.HashCache(basedir, cache_filename,

                                              self._control_files._file_mode)

        hc = self._hashcache

        hc.read()

        # is this scan needed ? it makes things kinda slow.

        #hc.scan()

        if hc.needs_write:

            mutter("write hc")

            hc.write()

        if _inventory is None:

            self._inventory_is_modified = False

            self.read_working_inventory()

        else:

            # the caller of __init__ has provided an inventory,

            # we assume they know what they are doing - as its only

            # the Format factory and creation methods that are

            # permitted to do this.

            self._set_inventory(_inventory, dirty=False)

    branch = property(

        fget=lambda self: self._branch,

        doc="""The branch this WorkingTree is connected to.

            This cannot be set - it is reflective of the actual disk structure

            the working tree has been constructed from.

            """)

    def break_lock(self):

        """Break a lock if one is present from another instance.

        Uses the ui factory to ask for confirmation if the lock may be from

        an active process.

        This will probe the repository for its lock as well.

        """

        self._control_files.break_lock()

        self.branch.break_lock()

    def _set_inventory(self, inv, dirty):

        """Set the internal cached inventory.

        :param inv: The inventory to set.

        :param dirty: A boolean indicating whether the inventory is the same

            logical inventory as whats on disk. If True the inventory is not

            the same and should be written to disk or data will be lost, if

            False then the inventory is the same as that on disk and any

            serialisation would be unneeded overhead.

        """

        assert inv.root is not None

        self._inventory = inv

        self._inventory_is_modified = dirty

    @staticmethod

    def open(path=None, _unsupported=False):

        """Open an existing working tree at path.

        """

        if path is None:

            path = os.path.getcwdu()

        control = bzrdir.BzrDir.open(path, _unsupported)

        return control.open_workingtree(_unsupported)

    @staticmethod

    def open_containing(path=None):

        """Open an existing working tree which has its root about path.

        This probes for a working tree at path and searches upwards from there.

        Basically we keep looking up until we find the control directory or

        run into /.  If there isn't one, raises NotBranchError.

        TODO: give this a new exception.

        If there is one, it is returned, along with the unused portion of path.

        :return: The WorkingTree that contains 'path', and the rest of path

        """

        if path is None:

            path = osutils.getcwd()

        control, relpath = bzrdir.BzrDir.open_containing(path)

        return control.open_workingtree(), relpath

    @staticmethod

    def open_downlevel(path=None):

        """Open an unsupported working tree.

        Only intended for advanced situations like upgrading part of a bzrdir.

        """

        return WorkingTree.open(path, _unsupported=True)

    def __iter__(self):

        """Iterate through file_ids for this tree.

        file_ids are in a WorkingTree if they are in the working inventory

        and the working file exists.

        """

        inv = self._inventory

        for path, ie in inv.iter_entries():

            if osutils.lexists(self.abspath(path)):

                yield ie.file_id

    def __repr__(self):

        return "<%s of %s>" % (self.__class__.__name__,

                               getattr(self, 'basedir', None))

    def abspath(self, filename):

        return pathjoin(self.basedir, filename)

    def basis_tree(self):

        """Return RevisionTree for the current last revision.

        If the left most parent is a ghost then the returned tree will be an

        empty tree - one obtained by calling repository.revision_tree(None).

        """

        try:

            revision_id = self.get_parent_ids()[0]

        except IndexError:

            # no parents, return an empty revision tree.

            # in the future this should return the tree for

            # 'empty:' - the implicit root empty tree.

            return self.branch.repository.revision_tree(None)

        else:

            try:

                xml = self.read_basis_inventory()

                inv = xml6.serializer_v6.read_inventory_from_string(xml)

                if inv is not None and inv.revision_id == revision_id:

                    return bzrlib.revisiontree.RevisionTree(

                        self.branch.repository, inv, revision_id)

            except (NoSuchFile, errors.BadInventoryFormat):

                pass

        # No cached copy available, retrieve from the repository.

        # FIXME? RBC 20060403 should we cache the inventory locally

        # at this point ?

        try:

            return self.branch.repository.revision_tree(revision_id)

        except errors.RevisionNotPresent:

            # the basis tree *may* be a ghost or a low level error may have

            # occured. If the revision is present, its a problem, if its not

            # its a ghost.

            if self.branch.repository.has_revision(revision_id):

                raise

            # the basis tree is a ghost so return an empty tree.

            return self.branch.repository.revision_tree(None)

    @staticmethod

    @deprecated_method(zero_eight)

    def create(branch, directory):

        """Create a workingtree for branch at directory.

        If existing_directory already exists it must have a .bzr directory.

        If it does not exist, it will be created.

        This returns a new WorkingTree object for the new checkout.

        TODO FIXME RBC 20060124 when we have checkout formats in place this

        should accept an optional revisionid to checkout [and reject this if

        checking out into the same dir as a pre-checkout-aware branch format.]

        XXX: When BzrDir is present, these should be created through that 

        interface instead.

        """

        warnings.warn('delete WorkingTree.create', stacklevel=3)

        transport = get_transport(directory)

        if branch.bzrdir.root_transport.base == transport.base:

            # same dir 

            return branch.bzrdir.create_workingtree()

        # different directory, 

        # create a branch reference

        # and now a working tree.

        raise NotImplementedError

    @staticmethod

    @deprecated_method(zero_eight)

    def create_standalone(directory):

        """Create a checkout and a branch and a repo at directory.

        Directory must exist and be empty.

        please use BzrDir.create_standalone_workingtree

        """

        return bzrdir.BzrDir.create_standalone_workingtree(directory)

    def relpath(self, path):

        """Return the local path portion from a given path.

        The path may be absolute or relative. If its a relative path it is 

        interpreted relative to the python current working directory.

        """

        return osutils.relpath(self.basedir, path)

    def has_filename(self, filename):

        return osutils.lexists(self.abspath(filename))

    def get_file(self, file_id):

        return self.get_file_byname(self.id2path(file_id))

    def get_file_text(self, file_id):

        return self.get_file(file_id).read()

    def get_file_byname(self, filename):

        return file(self.abspath(filename), 'rb')

    def annotate_iter(self, file_id):

        """See Tree.annotate_iter

        This implementation will use the basis tree implementation if possible.

        Lines not in the basis are attributed to CURRENT_REVISION

        If there are pending merges, lines added by those merges will be

        incorrectly attributed to CURRENT_REVISION (but after committing, the

        attribution will be correct).

        """

        basis = self.basis_tree()

        changes = self._iter_changes(basis, True, [file_id]).next()

        changed_content, kind = changes[2], changes[6]

        if not changed_content:

            return basis.annotate_iter(file_id)

        if kind[1] is None:

            return None

        import annotate

        if kind[0] != 'file':

            old_lines = []

        else:

            old_lines = list(basis.annotate_iter(file_id))

        old = [old_lines]

        for tree in self.branch.repository.revision_trees(

            self.get_parent_ids()[1:]):

            if file_id not in tree:

                continue

            old.append(list(tree.annotate_iter(file_id)))

        return annotate.reannotate(old, self.get_file(file_id).readlines(),

                                   CURRENT_REVISION)

    def get_parent_ids(self):

        """See Tree.get_parent_ids.

        This implementation reads the pending merges list and last_revision

        value and uses that to decide what the parents list should be.

        """

        last_rev = self._last_revision()

        if last_rev is None:

            parents = []

        else:

            parents = [last_rev]

        try:

            merges_file = self._control_files.get_utf8('pending-merges')

        except NoSuchFile:

            pass

        else:

            for l in merges_file.readlines():

                parents.append(l.rstrip('\n'))

        return parents

    @needs_read_lock

    def get_root_id(self):

        """Return the id of this trees root"""

        return self._inventory.root.file_id

    def _get_store_filename(self, file_id):

        ## XXX: badly named; this is not in the store at all

        return self.abspath(self.id2path(file_id))

    @needs_read_lock

    def clone(self, to_bzrdir, revision_id=None, basis=None):

        """Duplicate this working tree into to_bzr, including all state.

        Specifically modified files are kept as modified, but

        ignored and unknown files are discarded.

        If you want to make a new line of development, see bzrdir.sprout()

        revision

            If not None, the cloned tree will have its last revision set to 

            revision, and and difference between the source trees last revision

            and this one merged in.

        basis

            If not None, a closer copy of a tree which may have some files in

            common, and which file content should be preferentially copied from.

        """

        # assumes the target bzr dir format is compatible.

        result = self._format.initialize(to_bzrdir)

        self.copy_content_into(result, revision_id)

        return result

    @needs_read_lock

    def copy_content_into(self, tree, revision_id=None):

        """Copy the current content and user files of this tree into tree."""

        tree.set_root_id(self.get_root_id())

        if revision_id is None:

            merge.transform_tree(tree, self)

        else:

            # TODO now merge from tree.last_revision to revision (to preserve

            # user local changes)

            merge.transform_tree(tree, self)

            tree.set_parent_ids([revision_id])

    def id2abspath(self, file_id):

        return self.abspath(self.id2path(file_id))

    def has_id(self, file_id):

        # files that have been deleted are excluded

        inv = self._inventory

        if not inv.has_id(file_id):

            return False

        path = inv.id2path(file_id)

        return osutils.lexists(self.abspath(path))

    def has_or_had_id(self, file_id):

        if file_id == self.inventory.root.file_id:

            return True

        return self.inventory.has_id(file_id)

    __contains__ = has_id

    def get_file_size(self, file_id):

        return os.path.getsize(self.id2abspath(file_id))

    @needs_read_lock

    def get_file_sha1(self, file_id, path=None, stat_value=None):

        if not path:

            path = self._inventory.id2path(file_id)

        return self._hashcache.get_sha1(path, stat_value)

    def get_file_mtime(self, file_id, path=None):

        if not path:

            path = self._inventory.id2path(file_id)

        return os.lstat(self.abspath(path)).st_mtime

    if not supports_executable():

        def is_executable(self, file_id, path=None):

            return self._inventory[file_id].executable

    else:

        def is_executable(self, file_id, path=None):

            if not path:

                path = self._inventory.id2path(file_id)

            mode = os.lstat(self.abspath(path)).st_mode

            return bool(stat.S_ISREG(mode) and stat.S_IEXEC & mode)

    @needs_write_lock

    def _add(self, files, ids, kinds):

        """See MutableTree._add."""

        # TODO: Re-adding a file that is removed in the working copy

        # should probably put it back with the previous ID.

        # the read and write working inventory should not occur in this 

        # function - they should be part of lock_write and unlock.

        inv = self.read_working_inventory()

        for f, file_id, kind in zip(files, ids, kinds):

            assert kind is not None

            if file_id is None:

                inv.add_path(f, kind=kind)

            else:

                inv.add_path(f, kind=kind, file_id=file_id)

        self._write_inventory(inv)

    @needs_tree_write_lock

    def _gather_kinds(self, files, kinds):

        """See MutableTree._gather_kinds."""

        for pos, f in enumerate(files):

            if kinds[pos] is None:

                fullpath = normpath(self.abspath(f))

                try:

                    kinds[pos] = file_kind(fullpath)

                except OSError, e:

                    if e.errno == errno.ENOENT:

                        raise NoSuchFile(fullpath)

    @needs_write_lock

    def add_parent_tree_id(self, revision_id, allow_leftmost_as_ghost=False):

        """Add revision_id as a parent.

        This is equivalent to retrieving the current list of parent ids

        and setting the list to its value plus revision_id.

        :param revision_id: The revision id to add to the parent list. It may

        be a ghost revision as long as its not the first parent to be added,

        or the allow_leftmost_as_ghost parameter is set True.

        :param allow_leftmost_as_ghost: Allow the first parent to be a ghost.

        """

        parents = self.get_parent_ids() + [revision_id]

        self.set_parent_ids(parents,

            allow_leftmost_as_ghost=len(parents) > 1 or allow_leftmost_as_ghost)

    @needs_tree_write_lock

    def add_parent_tree(self, parent_tuple, allow_leftmost_as_ghost=False):

        """Add revision_id, tree tuple as a parent.

        This is equivalent to retrieving the current list of parent trees

        and setting the list to its value plus parent_tuple. See also

        add_parent_tree_id - if you only have a parent id available it will be

        simpler to use that api. If you have the parent already available, using

        this api is preferred.

        :param parent_tuple: The (revision id, tree) to add to the parent list.

            If the revision_id is a ghost, pass None for the tree.

        :param allow_leftmost_as_ghost: Allow the first parent to be a ghost.

        """

        parent_ids = self.get_parent_ids() + [parent_tuple[0]]

        if len(parent_ids) > 1:

            # the leftmost may have already been a ghost, preserve that if it

            # was.

            allow_leftmost_as_ghost = True

        self.set_parent_ids(parent_ids,

            allow_leftmost_as_ghost=allow_leftmost_as_ghost)

    @needs_tree_write_lock

    def add_pending_merge(self, *revision_ids):

        # TODO: Perhaps should check at this point that the

        # history of the revision is actually present?

        parents = self.get_parent_ids()

        updated = False

        for rev_id in revision_ids:

            if rev_id in parents:

                continue

            parents.append(rev_id)

            updated = True

        if updated:

            self.set_parent_ids(parents, allow_leftmost_as_ghost=True)

    @deprecated_method(zero_eleven)

    @needs_read_lock

    def pending_merges(self):

        """Return a list of pending merges.

        These are revisions that have been merged into the working

        directory but not yet committed.

        As of 0.11 this is deprecated. Please see WorkingTree.get_parent_ids()

        instead - which is available on all tree objects.

        """

        return self.get_parent_ids()[1:]