#!/usr/bin/python3
# Given as input a comps definition and a .ks file, list
# the packages which would be installed.  This list is
# NOT depsolved, it's merely a textual list of explicitly
# added or removed packages from the combined set.
#
# Licensed under the new-BSD license (http://www.opensource.org/licenses/bsd-license.php)
# Copyright (C) 2010 Red Hat, Inc.
# Written by Colin Walters <walters@verbum.org>

import os
import sys
import getopt
import xml.etree.ElementTree as ElementTree

import pykickstart
import pykickstart.parser
import pykickstart.version


def usage(ecode):
    print("Usage: {} [--version=VERSION] COMPS.xml KICKSTART.ks".format(sys.argv[0]))
    print("List packages installed by KICKSTART.")
    sys.exit(ecode)


def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], 'h', ['help', 'version=', 'debug'])
    except getopt.GetoptError as e:
        usage(1)

    version = None
    debug = False
    for o, a in opts:
        if o in ('-h', '--help'):
            usage(0)
        elif o in ('--debug',):
            debug = True
        elif o in ('--version',):
            version = a

    if len(args) != 2:
        usage(1)
    comps_filename = args[0]
    kickstart_filename = args[1]

    if version:
        version_cls = pykickstart.version.stringToVersion(version)
    else:
        version_cls = pykickstart.version.makeVersion()

    comps = ElementTree.parse(comps_filename)

    # pykickstart is going to look for files relative to
    # the current directory of the kickstart file
    working_directory = os.path.dirname(kickstart_filename)
    if working_directory:
        os.chdir(working_directory)

    parser = pykickstart.parser.KickstartParser(version_cls)
    parser.readKickstart(kickstart_filename)

    comps_groups = comps.findall('group')
    packages_for_group = {}
    for group in comps_groups:
        group_id = group.find('id')
        assert group_id is not None
        pkglist_node = group.find('packagelist')
        assert pkglist_node is not None
        reqs = pkglist_node.findall('packagereq')
        pkglist = []
        for req in reqs:
            if req.attrib['type'] in ('default', 'mandatory'):
                pkglist.append(req.text)
        packages_for_group[group_id.text] = pkglist

    pkg_list = set()

    for group in parser.handler.packages.groupList:
        if debug:
            print("# Including {} packages from group '{}'".format(len(packages_for_group.get(group.name, [])),
                                                                   group.name), file=sys.stderr)
        for pkg in packages_for_group.get(group.name, []):
            pkg_list.add(pkg)
    if debug:
        print("# Adding {} explicitly specified packages".format(len(parser.handler.packages.packageList)),
              file=sys.stderr)
    for pkg in parser.handler.packages.packageList:
        pkg_list.add(pkg)
    if debug:
        print("# Processing {} explicitly excluded packages".format(len(parser.handler.packages.excludedList)),
              file=sys.stderr)
    for pkg in parser.handler.packages.excludedList:
        pkg_list.discard(pkg)

    for pkg in sorted(pkg_list):
        print(pkg)

    sys.exit(0)


if __name__ == '__main__':
    main()