#!/usr/bin/env python3
import sys
from os import rename
from argparse import ArgumentParser

from normalisename import Normalisename


if __name__ == '__main__':
    # parse arguments
    parser = ArgumentParser(description='Eliminate funky stuff from filenames.')
    parser.add_argument(
        '--separator', '-s', type=str, nargs='?', default='_',
        help='Set separator to use (to replace spaces with).'
    )
    parser.add_argument(
        '--allow', '-a', type=str, nargs='?', action='append',
        help='Specify a special character to allow (it will not be removed).'
    )
    parser.add_argument(
        '--disallow', '-d', type=str, nargs='?', action='append',
        help='Specify a special character to disallow (it will removed).'
    )
    parser.add_argument(
        '--whitelist', '-w', type=str,
        help='Overwrite default whitelist (format: whitespace separated string).'
    )

    mgroup = parser.add_mutually_exclusive_group()
    mgroup.add_argument(
        '--dryrun', '-n', action='store_true',
        help='Do not touch anything, just tell me what would change!'
    )
    mgroup.add_argument(
        '--funky', '-f', action='store_true',
        help='List filenames with special characters'
    )
    parser.add_argument(
        'files', type=str, nargs='+',
        help='File(s) to normalise the name of (relative or absolute path).'
    )

    args = parser.parse_args()

    # verify arguments
    if args.disallow:
        if args.separator in args.disallow:
            sys.exit('Disallowing your chosen separator makes no sense!')

    # declare special characters that will not be removed (spaces are handled elsewhere)
    whitelist = {' ', '.', '-'}
    if args.whitelist:
        whitelist = set(args.whitelist.split())

    # modify whitelist based on arguments
    whitelist.add(args.separator)
    if args.allow:
        whitelist = whitelist.union(set(args.allow))
    if args.disallow:
        whitelist = whitelist.difference(set(args.disallow))

    # define operations
    listchanges = lambda a, b: print('{}{} -->    {}'.format(a, ' '*(60-len(a)), b))
    listfunky = lambda a, b: print(a)

    operation = rename
    if args.dryrun:
        operation = listchanges
    elif args.funky:
        operation = listfunky

    # do what needs to be done
    Normalisename(operation, args.separator, whitelist)(args.files)