/b-gtk/fix-viz

# Copyright (C) 2006 by Szilveszter Farkas (Phanatic) <szilveszter.farkas@gmail.com>

# Some parts of the code are:

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

import time

import bzrlib

import bzrlib.osutils as osutils

from bzrlib.info import _repo_relpath

from bzrlib.missing import find_unmerged

def get_location_info(repository, branch=None, working=None):

    """ Get known locations for working, branch and repository.

    :return: a dictionary containing the needed infos

    """

    ret = {}

    repository_path = repository.bzrdir.root_transport.base

    if working and branch:

        working_path = working.bzrdir.root_transport.base

        branch_path = branch.bzrdir.root_transport.base

        if working_path != branch_path:

            # lightweight checkout

            ret['lightcoroot'] = working_path

            if repository.is_shared():

                # lightweight checkout of branch in shared repository

                ret['sharedrepo'] = repository_path

                ret['repobranch'] = _repo_relpath(repository_path, branch_path)

            else:

                # lightweight checkout of standalone branch

                ret['cobranch'] = branch_path

        elif repository.is_shared():

            # branch with tree inside shared repository

            ret['sharedrepo'] = repository_path

            ret['repoco'] = _repo_relpath(repository_path, branch_path)

        elif branch.get_bound_location():

            # normal checkout

            ret['coroot'] = working_path

            ret['cobranch'] = branch.get_bound_location()

        else:

            # standalone

            ret['branchroot'] = working_path

    elif branch:

        branch_path = branch.bzrdir.root_transport.base

        if repository.is_shared():

            # branch is part of shared repository

            ret['sharedrepo'] = repository_path

            ret['repobranch'] = _repo_relpath(repository_path, branch_path)

        else:

            # standalone branch

            ret['branchroot'] = branch_path

    else:

        # shared repository

        assert repository.is_shared()

        ret['sharedrepo'] = repository_path

    return ret

def get_related_info(branch):

    """ Get parent and push location of branch.

    :return: a dictionary containing the needed infos

    """

    ret = {}

    if branch.get_parent() or branch.get_push_location():

        if branch.get_parent():

            ret['parentbranch'] = branch.get_parent()

        if branch.get_push_location():

            ret['publishbranch'] = branch.get_push_location()

    return ret

def get_format_info(control=None, repository=None, branch=None, working=None):

    """ Get known formats for control, working, branch and repository.

    :return: a dictionary containing the needed infos

    """

    ret = {}

    if control:

        ret['control'] = control._format.get_format_description()

    if working:

        ret['workingtree'] = working._format.get_format_description()

    if branch:

        ret['branch'] = branch._format.get_format_description()

    if repository:

        ret['repository'] = repository._format.get_format_description()

    return ret

def get_locking_info(repository, branch=None, working=None):

    """ Get locking status of working, branch and repository.

    :return: a dictionary containing the needed infos

    """

    ret = {}

    if (repository.get_physical_lock_status() or

        (branch and branch.get_physical_lock_status()) or

        (working and working.get_physical_lock_status())):

        if working:

            if working.get_physical_lock_status():

                status = 'locked'

            else:

                status = 'unlocked'

            ret['workingtree'] = status

        if branch:

            if branch.get_physical_lock_status():

                status = 'locked'

            else:

                status = 'unlocked'

            ret['branch'] = status

        if repository:

            if repository.get_physical_lock_status():

                status = 'locked'

            else:

                status = 'unlocked'

            ret['repository'] = status

    return ret

def get_missing_revisions_branch(branch):

    """ Get missing master revisions in branch.

    :return: a dictionary containing the needed infos

    """

    ret = {}

    # Try with inaccessible branch ?

    master = branch.get_master_branch()

    if master:

        local_extra, remote_extra = find_unmerged(branch, master)

        if remote_extra:

            ret = len(remote_extra)

    return ret

def get_missing_revisions_working(working):

    """ Get missing revisions in working tree.

    :return: a dictionary containing the needed infos

    """

    if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):

        # function deprecated after 0.9

        from bzrlib.delta import compare_trees

    ret = {}

    branch = working.branch

    basis = working.basis_tree()

    work_inv = working.inventory

    if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):

        delta = compare_trees(basis, working, want_unchanged=True)

    else:

        delta = working.changes_from(basis, want_unchanged=True)

    history = branch.revision_history()

    tree_last_id = working.last_revision()

    if len(history) and tree_last_id != history[-1]:

        tree_last_revno = branch.revision_id_to_revno(tree_last_id)

        missing_count = len(history) - tree_last_revno

        ret = missing_count

    return ret

def get_working_stats(working):

    """ Get statistics about a working tree.

    :return: a dictionary containing the needed infos

    """

    if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):

        # function deprecated after 0.9

        from bzrlib.delta import compare_trees

    ret = {}

    basis = working.basis_tree()

    work_inv = working.inventory

    if (bzrlib.version_info[0] == 0) and (bzrlib.version_info[1] < 9):

        delta = compare_trees(basis, working, want_unchanged=True)

    else:

        delta = working.changes_from(basis, want_unchanged=True)

    ret['unchanged'] = len(delta.unchanged)

    ret['modified'] = len(delta.modified)

    ret['added'] = len(delta.added)

    ret['removed'] = len(delta.removed)

    ret['renamed'] = len(delta.renamed)

    ignore_cnt = unknown_cnt = 0

    for path in working.extras():

        if working.is_ignored(path):

            ignore_cnt += 1

        else:

            unknown_cnt += 1

    ret['unknown'] = unknown_cnt

    ret['ignored'] = ignore_cnt

    dir_cnt = 0

    for file_id in work_inv:

        if work_inv.get_file_kind(file_id) == 'directory':

            dir_cnt += 1

    ret['subdirs'] = dir_cnt

    return ret

def get_branch_stats(branch):

    """ Get statistics about a branch.

    :return: a dictionary containing the needed infos

    """

    ret = {}

    repository = branch.repository

    history = branch.revision_history()

    revno = len(history)

    ret['revno'] = revno 

    committers = {}

    for rev in history:

        committers[repository.get_revision(rev).committer] = True

    ret['commiters'] = len(committers)

    if revno > 0:

        firstrev = repository.get_revision(history[0])

        age = int((time.time() - firstrev.timestamp) / 3600 / 24)

        ret['age'] = age

        ret['firstrev'] = osutils.format_date(firstrev.timestamp,

                                              firstrev.timezone)

        lastrev = repository.get_revision(history[-1])

        ret['lastrev'] = osutils.format_date(lastrev.timestamp,

                                             lastrev.timezone)

    return ret

def get_repository_stats(repository):

    """ Get statistics about a repository.

    :return: a dictionary containing the needed infos

    """

    ret = {}

    if repository.bzrdir.root_transport.listable():

        c, t = repository._revision_store.total_size(repository.get_transaction())

        ret['revisions'] = c

        ret['size'] = t

    return ret

def diff_helper(tree, specific_files, external_diff_options, 

                    old_revision_spec=None, new_revision_spec=None,

                    old_label='a/', new_label='b/', output=None):

    """ Helper for diff.

    :param tree: a WorkingTree

    :param specific_files: the specific files to compare, or None

    :param external_diff_options: if non-None, run an external diff, and pass it these options

    :param old_revision_spec: if None, use basis tree as old revision, otherwise use the tree for the specified revision. 

    :param new_revision_spec:  if None, use working tree as new revision, otherwise use the tree for the specified revision.

    """

    import sys

    from bzrlib.diff import show_diff_trees

    if output == None:

        output = sys.stdout

    def spec_tree(spec):

        revision_id = spec.in_store(tree.branch).rev_id

        return tree.branch.repository.revision_tree(revision_id)

    if old_revision_spec is None:

        old_tree = tree.basis_tree()

    else:

        old_tree = spec_tree(old_revision_spec)

    if new_revision_spec is None:

        new_tree = tree

    else:

        new_tree = spec_tree(new_revision_spec)

    return show_diff_trees(old_tree, new_tree, output, specific_files,

                           external_diff_options,

                           old_label=old_label, new_label=new_label)