/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 sys

import bzrlib

from bzrlib import (

    bzrdir,

    diff,

    errors,

    osutils,

    urlutils,

    )

from bzrlib.diff import show_diff_trees

from bzrlib.missing import find_unmerged

def _repo_rel_url(repo_url, inner_url):

    """Return path with common prefix of repository path removed.

    If path is not part of the repository, the original path is returned.

    If path is equal to the repository, the current directory marker '.' is

    returned.

    Otherwise, a relative path is returned, with trailing '/' stripped.

    """

    inner_url = urlutils.normalize_url(inner_url)

    repo_url = urlutils.normalize_url(repo_url)

    if inner_url == repo_url:

        return '.'

    result = urlutils.relative_url(repo_url, inner_url)

    if result != inner_url:

        result = result.rstrip('/')

    return result

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_rel_url(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_rel_url(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_rel_url(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()

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

    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 path, ie in working.iter_entries_by_dir():

        if ie.kind == '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'] = '%d days'%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'] = '%d KiB'%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.

    """

    if output == None:

        output = sys.stdout

    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)

def spec_tree(spec):

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

        return tree.branch.repository.revision_tree(revision_id)