import argparse
import sys

from voussoirkit import betterhelp
from voussoirkit import pathclass
from voussoirkit import niceprints
from voussoirkit import spinal
from voussoirkit import vlogging

log = vlogging.getLogger(__name__, 'directory_discrepancy')

def helper(root, recurse):
    root = pathclass.Path(root)
    walk = spinal.walk(root, recurse=recurse, yield_directories=True, yield_files=True)
    for path in walk:
        relative = path.relative_to(root)
        yield relative

def directory_discrepancy_argparse(args):
    files1 = set(helper(args.dir1, recurse=args.recurse))
    files2 = set(helper(args.dir2, recurse=args.recurse))

    print(niceprints.equals_header(f'In "{args.dir1}" but not in "{args.dir2}"'))
    for discrepancy in sorted(files1.difference(files2)):
        print(discrepancy)

    print()

    print(niceprints.equals_header(f'In "{args.dir2}" but not in "{args.dir1}"'))
    for discrepancy in sorted(files2.difference(files1)):
        print(discrepancy)

    return 0

@vlogging.main_decorator
def main(argv):
    parser = argparse.ArgumentParser(
        description='''
        This program compares two directory and shows which files exist in each
        directory that do not exist in the other.
        ''',
    )
    parser.add_argument('dir1')
    parser.add_argument('dir2')
    parser.add_argument(
        '--recurse',
        action='store_true',
        help='''
        Also check subdirectories.
        ''',
    )
    parser.set_defaults(func=directory_discrepancy_argparse)

    return betterhelp.go(parser, argv)

if __name__ == '__main__':
    raise SystemExit(main(sys.argv[1:]))