import logging
import os
import subprocess
import json
from utils.aux_logging import prepare_logger, log_all
import traceback
from configuration import gtfs_validator


def is_existing_path(path: str) -> bool:
    return os.path.exists(path)


def main(gtfs_file: str, res_folder: str) -> None:
    # Build the shell command
    validator_path = gtfs_validator
    if os.path.exists(validator_path):
        command: str = f"java -jar {validator_path} -i {gtfs_file} -o {res_folder}"
        if not is_existing_path(gtfs_file) or not is_existing_path(res_folder):
            log_all(logging.ERROR, f"{gtfs_file} or  {res_folder} not valid. Stopping")
            raise
        log_all(logging.INFO, command)
        # Execute the command in the shell
        subprocess.run(command, shell=True, check=True)
    else:
        log_all(logging.ERROR, f"validator not found {validator_path}")
        raise Exception(f"validator not found {validator_path}")
    # Check the result files for errors
    report_file: str = os.path.join(res_folder, "report.json")
    system_errors_file: str = os.path.join(res_folder, "system_errors.json")

    # Check system_errors.json for notices
    with open(system_errors_file) as se_file:
        system_errors = json.load(se_file)
        if len(system_errors["notices"]) == 0:
            log_all(logging.INFO, "No notices found in system_errors.json")
        else:
            log_all(logging.ERROR, "Notices found in system_errors.json")
            raise Exception("Notices found in system_errors.json")

    # Check report.json for notices with severity "ERROR"
    with open(report_file, encoding="utf-8") as rf:
        report_data = json.load(rf)
        if "notices" in report_data:
            for notice in report_data["notices"]:
                if notice["severity"] == "ERROR":
                    log_all(
                        logging.ERROR,
                        f"ERROR notice in report.json: {notice.get('code')}",
                    )
                    raise Exception(f"ERROR notice in report.json: {notice.get('code')}")
        else:
            log_all(logging.INFO, "No notices found in report.json")


if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser(description="Validates a gtfs file and shows some stats")
    parser.add_argument("gtfs_file", type=str, help="The input file (gtfs.zip)")
    parser.add_argument("res_folder", type=str, help="The folder for the report.")
    parser.add_argument("--log_file", type=str, required=False, help="the logfile")
    args = parser.parse_args()
    mylogger = prepare_logger(logging.INFO, args.log_file)

    try:
        main(args.gtfs_file, args.res_folder)
    except Exception as e:
        log_all(logging.ERROR, f"{e} {traceback.format_exc()}")
        raise e