~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

"""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(), """

from bisect import bisect_left

import collections

from copy import deepcopy

import errno

import itertools

import operator

import stat

from time import time

import warnings

import re

import bzrlib

from bzrlib import (

    branch,

    bzrdir,

    conflicts as _mod_conflicts,

    dirstate,

    errors,

    generate_ids,

    globbing,

    hashcache,

    ignores,

    merge,

    osutils,

    revisiontree,

    repository,

    textui,

    trace,

    transform,

    ui,

    urlutils,

    xml5,

    xml6,

    xml7,

    )

import bzrlib.branch

from bzrlib.transport import get_transport

import bzrlib.ui

from bzrlib.workingtree_4 import WorkingTreeFormat4

""")

from bzrlib import symbol_versioning

from bzrlib.decorators import needs_read_lock, needs_write_lock

from bzrlib.inventory import InventoryEntry, Inventory, ROOT_ID, TreeReference

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,

    normpath,

    pathjoin,

    rand_chars,

    realpath,

    safe_unicode,

    splitpath,

    supports_executable,

    )

from bzrlib.trace import mutter, note

from bzrlib.transport.local import LocalTransport

from bzrlib.progress import DummyProgress, ProgressPhase

from bzrlib.revision import NULL_REVISION, CURRENT_REVISION

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 instance. This is not a public API.

        :param branch: A branch to override probing for the branch.

        """

        self._format = _format

        self.bzrdir = _bzrdir

        if not _internal:

            raise errors.BzrError("Please use bzrdir.open_workingtree or "

                "WorkingTree.open() to obtain a WorkingTree.")

        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):

            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.

        wt_trans = self.bzrdir.get_workingtree_transport(None)

        cache_filename = wt_trans.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:

            # This will be acquired on lock_read() or lock_write()

            self._inventory_is_modified = False

            self._inventory = None

        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 requires_rich_root(self):

        return self._format.requires_rich_root

    def supports_tree_reference(self):

        return False

    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)

    # should be deprecated - this is slow and in any case treating them as a

    # container is (we now know) bad style -- mbp 20070302

    ## @deprecated_method(zero_fifteen)

    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)

        try:

            return self.revision_tree(revision_id)

        except errors.NoSuchRevision:

            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):

        file_id = osutils.safe_file_id(file_id)

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

    def get_file_text(self, file_id):

        file_id = osutils.safe_file_id(file_id)

        return self.get_file(file_id).read()

    def get_file_byname(self, filename):

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

    @needs_read_lock

    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).

        """

        file_id = osutils.safe_file_id(file_id)

        basis = self.basis_tree()

        basis.lock_read()

        try:

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

                require_versioned=True).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)

        finally:

            basis.unlock()

    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('pending-merges')

        except errors.NoSuchFile:

            pass

        else:

            for l in merges_file.readlines():

                revision_id = osutils.safe_revision_id(l.rstrip('\n'))

                parents.append(revision_id)

        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

        file_id = osutils.safe_file_id(file_id)

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

    @needs_read_lock

    def clone(self, to_bzrdir, revision_id=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.

        """

        # 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):

        file_id = osutils.safe_file_id(file_id)

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

    def has_id(self, file_id):

        # files that have been deleted are excluded

        file_id = osutils.safe_file_id(file_id)

        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):

        file_id = osutils.safe_file_id(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):

        file_id = osutils.safe_file_id(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):

        file_id = osutils.safe_file_id(file_id)

        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):

        file_id = osutils.safe_file_id(file_id)

        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):

            file_id = osutils.safe_file_id(file_id)

            return self._inventory[file_id].executable

    else:

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

            if not path:

                file_id = osutils.safe_file_id(file_id)

                path = self.id2path(file_id)

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

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

    @needs_tree_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:

                file_id = osutils.safe_file_id(file_id)

                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 errors.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: