import argparse
import json
from obfuscator.csv_reader import CSVReader
from obfuscator.obfuscate import obfuscate
from obfuscator.logger import get_logger

logger = get_logger("CLI")


def main():
    parser = argparse.ArgumentParser(description="gdpr-obfuscator")
    # Require user to either choose a local file or an S3 object
    loc = parser.add_mutually_exclusive_group(required=True)
    loc.add_argument("--local")
    loc.add_argument("--s3")

    parser.add_argument("--pii", nargs="+", required=True)

    args = parser.parse_args()

    if args.local and not args.s3:
        logger.debug("User chose to read CSV from local path")
        reader = CSVReader()
        data = reader.read_local(args.local)
        logger.debug(data)
    else:
        logger.debug("User chose to read CSV from S3")

    obfuscated_data = obfuscate(data, args.pii)
    logger.debug(json.dumps(obfuscated_data, indent=4))


if __name__ == "__main__":
    main()