import argparse
import os
from logger import logger 

from table_handler.serializer import TableSerializer
import instructions.countings_formater as countings_formater
import instructions.ring_formater as ring_formater


def process_file(input_file, output_file, formatter, row_instructions):
    logger.info(f"Processing: {input_file}")

    serializer = TableSerializer(
        input_file,
        formatter.EXPECTED_INPUT_TABLE_SCHEMA,
        formatter.EXPECTED_OUTPUT_TABLE_SCHEMA
    )

    serializer.run(
        formatter.COL_INSTRUCTIONS,
        row_instructions
    )

    serializer.write_output(output_file)


def main():
    parser = argparse.ArgumentParser(description="Serialize Excel tables.")

    # Single file mode (optional fallback)
    parser.add_argument("-i", "--input", help="Input Excel file")
    parser.add_argument("-o", "--output", help="Output Excel file")

    # Directory mode
    parser.add_argument("--input-dir", help="Input directory")
    parser.add_argument("--output-dir", help="Output directory")

    parser.add_argument(
        "-t", "--type",
        required=True,
        choices=["ring", "counting"],
        help="Type of table to process"
    )

    args = parser.parse_args()

    # Select formatter
    if args.type == "ring":
        formatter = ring_formater
        row_instructions = ring_formater.ROW_INSTRUCTIONS
    else:
        formatter = countings_formater
        row_instructions = countings_formater.ROW_INSTRUCTION

    # directory
    if args.input_dir and args.output_dir:
        logger.info("Running in directory mode")

        for root, _, files in os.walk(args.input_dir):
            for file in files:
                if not file.endswith((".xlsx", ".xls")):
                    continue

                input_path = os.path.join(root, file)

                # Create corresponding output path
                relative_path = os.path.relpath(root, args.input_dir)
                output_root = os.path.join(args.output_dir, relative_path)
                os.makedirs(output_root, exist_ok=True)

                output_path = os.path.join(output_root, file)

                process_file(input_path, output_path, formatter, row_instructions)

    # single file
    elif args.input and args.output:
        logger.info("Running in single file mode")
        process_file(args.input, args.output, formatter, row_instructions)

    else:
        parser.error("You must provide either input/output files or input-dir/output-dir")

    logger.info("Done")


if __name__ == "__main__":
    main()