# @https://github.com/ue71603, 2024
import logging
import os
from pathlib import Path

from utils.aux_logging import prepare_logger, log_all
import traceback
from lxml import etree
from storage.lxml.core.implementation import XmlStorage
from typing import Any, IO
from configuration import work_dir


def get_absolute_path(file_name: str, work_dir: str) -> str:
    if os.path.isabs(file_name):
        # If the file_name is already an absolute path, use it as is
        return file_name
    else:
        # If the file_name is a relative path, prepend it with the work_dir
        return os.path.join(work_dir, file_name)


def check_xsd_validity(xsd_file: str) -> bool:
    try:
        etree.parse(xsd_file)
        log_all(logging.INFO, f"{xsd_file} is a valid XSD file")
    except etree.XMLSyntaxError as e:
        log_all(logging.ERROR, f"{xsd_file} is an invalid XSD file: {str(e)}")
    except Exception as e:
        log_all(logging.ERROR, f"An error occurred while checking {xsd_file}: {str(e)}")
    return True


def validate_xml(xml_file: IO[Any], xmlschema: etree.XMLSchema) -> bool:
    try:

        # Validate the XML file against the schema
        xmlschema.assertValid(etree.parse(xml_file))
        # Check for any errors captured by the handler
        log_all(logging.INFO, f"{xml_file} is valid against the schema")
    except etree.XMLSchemaError as e:
        log_all(logging.ERROR, f"{xml_file} is invalid: {str(e)}")
        raise e
    except Exception as e:
        log_all(logging.ERROR, f"An error occurred while validating {xml_file}: {str(e)}")
        raise e

    return True



def main(folder: str, xsd_schema: str) -> None:
    xsd_schema = get_absolute_path(xsd_schema, work_dir)
    xmlschema = etree.XMLSchema(etree.parse(xsd_schema))

    def process_file_path(file_full_path: str):
        filename = os.path.basename(file_full_path)
        if filename.endswith(".xml") or filename.endswith(
                ".xml.gz"):  # TODO zips are not processed (because might be GTFS)
            log_all(logging.INFO, f"Processing file: {filename}")
            for sub_file, real_filename in XmlStorage.open_netex_file(XmlStorage(Path(file_full_path))):
                validate_xml(sub_file, xmlschema)
        elif filename.endswith(".xsd"):
            log_all(logging.INFO, f"Processing XSD-file: {filename}")
            check_xsd_validity(file_full_path)
        else:
            # If you want to skip silently, remove this line
            log_all(logging.INFO, f"Skipping unsupported file type: {filename}")

    if not os.path.exists(folder):
        log_all(logging.ERROR, f"Provided path does not exist: {folder}")
        return

    if os.path.isfile(folder):
        process_file_path(folder)
    elif os.path.isdir(folder):
        for root, dirs, files in os.walk(folder):
            for filename in files:
                process_file_path(os.path.join(root, filename))
    else:
        log_all(logging.ERROR, f"Path is neither a file nor a directory: {folder}")


if __name__ == "__main__":
    import argparse

    argument_parser = argparse.ArgumentParser(description="Validates each file in a folder (xsd and xml)")
    argument_parser.add_argument("folder", type=str, help="The folder containing the xml files to validate")
    # argument_parser.add_argument('xmlschema', type=str, help='The schema file to use as a basis for validation')
    xmlschema = "./netex-xsd/xsd/NeTEx_publication.xsd"
    argument_parser.add_argument("--log_file", type=str, required=False, help="the logfile")
    args = argument_parser.parse_args()
    # Fetching the data based on command-line arguments
    mylogger = prepare_logger(logging.INFO, args.log_file)
    try:
        main(args.folder, xmlschema)
    except Exception as e:
        log_all(logging.ERROR, f"{e} {traceback.format_exc()}")
        raise e