package-grokker.py

import argparse
import os

import pacdb
import grokkermod


parser = argparse.ArgumentParser(description='Search packages for problematic imports')
parser.add_argument('-e', '--repo', default='mingw64', help='pacman repo name to search')
parser.add_argument('-p', '--package', required=True, help='package from which to find dependents')
parser.add_argument('-d', '--dll', required=True, action='append', help='dll(s) from which problematic symbols are imported')
parser.add_argument('-l', '--local-mirror', help='root directory of local mirror')
parser.add_argument('-v', '--verbose', action='count', help='output additional information')
parser.add_argument('symbol', nargs='*', help='problematic symbol(s)')

options = parser.parse_args()

local_mirror = None
if options.local_mirror:
    if options.repo == 'msys':
        local_mirror = os.path.join(options.local_mirror, options.repo, 'x86_64')
    else:
        local_mirror = os.path.join(options.local_mirror, 'mingw', options.repo)

    repo = pacdb.Database(options.repo, filename=os.path.join(local_mirror, '{}.files'.format(options.repo)))
elif options.repo == 'msys':
    repo = pacdb.msys_db_by_arch('x86_64', 'files')
else:
    repo = pacdb.mingw_db_by_name(options.repo, 'files')

# TODO: revamp args to allow specifying per-dll symbol list
bytes_symbols = set(sym.encode('ascii') for sym in options.symbol)
package_handler = grokkermod.ProblematicImportSearcher({dll.encode('ascii').lower(): bytes_symbols for dll in options.dll}, local_mirror)

seen = set()
for pkgbase in grokkermod.grok_dependency_tree(repo, options.package, package_handler):
    if pkgbase not in seen:
        print(pkgbase)
        seen.add(pkgbase)