#! /usr/bin/python3
#
# wget http://bootstrap.debian.net/importance_metric_noall.txt
# ssh buildd_arm64@buildd.debian-ports.org  wanna-build -A arm64 \
#        -d unstable -l  build-attempted > build-attempted
# botch-wanna-build-findblockers build-attempted importance_metric_noall.txt

from __future__ import print_function

__copyright__ = """
Copyright (C) 2014, Chen Baozi <baozich@gmail.com>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License version 2 as
published by the Free Software Foundation.

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
"""


def read_build_attempted(fname):
    candidates = set()
    with open(fname) as items:
        for line in items:
            if line.startswith('Total'):
                continue
            if line.startswith(' '):
                continue
            candidates.add(line.split()[0].split('/')[1].split('_')[0])
    return candidates


def read_importance_metric(fname):
    priorities = list()
    with open(fname) as metrics:
        for line in metrics:
            priorities.append(line.split())
    return priorities


def main(candidates, priorities, order, vols):
    priorities.sort()
    priorities.sort(key=lambda x: int(x[vols]), reverse=order)
    for i in priorities:
        pkg = i[0].split(':')[0]
        if pkg in candidates:
            # popcon values go into the nine-digit range
            print('%9s %s' % ((i[vols], pkg)))

if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser(
        description=("Filter the output of botch-calcportsmetric for " +
                     "\"build-attempted\" packages from wanna-build"))
    parser.add_argument('-r', '--reverse', action='store_false',
                        help='reverse the result of comparisons')
    parser.add_argument('candidates', metavar='build-attempt-list',
                        type=read_build_attempted,
                        help='output of "wanna-build -l build-attempted"')
    parser.add_argument('priorities', metavar='importance-metric',
                        type=read_importance_metric,
                        help="output of botch-calcportsmetric or from " +
                             "http://bootstrap.debian.net/" +
                             "importance_metric.html")
    group = parser.add_mutually_exclusive_group()
    group.add_argument('-m', '--min', action='store_true',
                       help="sort by minimum number of blocked source " +
                            "packages (default)")
    group.add_argument('-M', '--max', action='store_true',
                       help="sort by maximum number of blocked source " +
                            "packages")
    group.add_argument('--min-popcon', action='store_true',
                       help="sort by minimum sum of popcon values of the " +
                            "blocked source packages")
    group.add_argument('--max-popcon', action='store_true',
                       help="sort by maximum sum of popcon values of the " +
                            "blocked source packages")
    parser.add_argument(
        '-v', '--verbose', action='store_true', help='be verbose')
    args = parser.parse_args()
    # turn the selected option into the column number in the importance-metric
    # file if no option is set then the default is -m (the second column)
    try:
        vols = [args.min, args.max, args.min_popcon,
                args.max_popcon].index(True) + 1
    except ValueError:
        vols = 1
    main(args.candidates, args.priorities, args.reverse, vols)
