#!/usr/bin/python3
import argparse
import copy
from distutils.version import LooseVersion
import sys

import apt

# Virtual package mapping
virtual_mapping = {'awk':               'gawk', 
                   'perlapi-5.18.1':    'perl-base',
                   'perlapi-5.18.2':    'perl-base',
                   'upstart-job':       'upstart',
                   'python:any':        'python2.7',
                   'python3:any':       'python3.4',
                   'whiptail-provider': 'whiptail'}

def main():
    
    parser = argparse.ArgumentParser(description='Process some integers.')
    parser.add_argument('-v', '--verbose', action='store_true',
                        help='write additional information to stderr')
    parser.add_argument('input_package_files', action='store',
                        help='a comma separated list of files containing '
                             'package names, one per line')
    args = parser.parse_args()
    
    # Input packages
    in_packages = []
    
    # Parse input package files
    for filename in args.input_package_files.split(','):
        with open(filename) as in_file:
            for line in in_file:
                line = line.strip()
                if line:
                    in_packages.append(line)
    
    # Open APT cache
    cache = apt.Cache(memonly=True)
    cache.open()
    
    # Create a set to hold the names of all packages and their dependencies
    packages = set()

    while True:
        try:
            # Grab next package 
            pkg_name = in_packages.pop()
            if args.verbose:    
                sys.stderr.write('Package: {:s}\n'.format(pkg_name))
                sys.stderr.flush()
      
            # Check if the package is in the cache, if not check if it is in
            # our list of known virtual packages
            if pkg_name not in cache:
                if pkg_name in virtual_mapping:
                    pkg_name = virtual_mapping[pkg_name]
                else:
                    sys.stderr.write('ERROR: unkown package "{:s}"; if this is '
                                     'a virtual package please add it to the '
                                     'virtual package mapping\n'.format(pkg_name))
                    continue
      
            # Add this package (with version) to the list of packages
            packages.add(pkg_name)

            # Get package dependencies
            for pdep in cache[pkg_name].versions[0].dependencies:
                dep_name = pdep[0].name
                if args.verbose:
                    sys.stderr.write('\tDependency: {:s}\n'.format(dep_name))
                    sys.stderr.flush()
                if dep_name not in packages:
                    if dep_name in virtual_mapping:
                        dep_name = virtual_mapping[dep_name]
                    packages.add(dep_name)
                    in_packages.insert(0, dep_name)
        except IndexError:
            break

    # Print out the gathered package names
    for package in sorted(packages):
        print(package)
    
if __name__ == '__main__':
    sys.exit(main())