#! /usr/bin/python3

import sys
sys.path.append('/usr/share/botch')
from util import read_graphml, read_tag_file, sort_pkgs_key
import re


def extract_name_arch_ver(pkgrel):
    m = re.match(r"([^:]*):([^ ]*) \(= ([^\)]*)\)", pkgrel)
    return (m.groups())


def buildgraph2packages(buildgraph, inPackages):
    universe = dict()

    for packages in inPackages:
        for pkg in packages:
            key = (pkg['Package'], pkg['Architecture'], pkg['Version'])
            universe[key] = pkg

    result = set()

    for n, d in buildgraph.nodes(data=True):
        if d.get('type') == 'bin':
            # Do not add the binary package this IS belongs to.
            # Only add the binary packages in the installation set.
            # For normal graphs and closure graphs this has no
            # effect because the binary package is part of the IS
            # but for strong graphs the binary package might not
            # be part of the IS
            for pkg in d['binaries'].split(','):
                name, arch, version = extract_name_arch_ver(pkg)
                result.add((name, arch, version))

    for pkg in sorted(list(result), key=sort_pkgs_key):
        universe[pkg].dump(sys.stdout.buffer)
        sys.stdout.buffer.write(b"\n")

    return True


if __name__ == '__main__':
    import argparse
    parser = argparse.ArgumentParser(
        description=("Output all binary packages occurring in all" +
                     " installation sets in a build graph"))
    parser.add_argument('buildgraph', type=read_graphml,
                        help="Input buildgraph in GraphML format")
    parser.add_argument('packages', type=read_tag_file, nargs='+',
                        help="input Packages files")
    parser.add_argument(
        '-v', '--verbose', action='store_true', help='be verbose')
    args = parser.parse_args()
    ret = buildgraph2packages(args.buildgraph, args.packages)
    exit(not ret)
