import os
import sys
import argparse

from dotenv import load_dotenv
from loguru import logger

load_dotenv()


def setup_logging(log_file: str | None = None) -> None:
    logger.remove()
    logger.add(
        sys.stderr,
        level="INFO",
        format="{time:YYYY-MM-DD HH:mm:ss} | <level>{level: <8}</level> | {message}",
    )

    if log_file:
        log_dir = os.getenv("LOGDIR")
        if log_dir:
            logger.add(
                f"{log_dir}/{log_file}",
                rotation="7 days",
                retention="30 days",
                compression="zip",
                level="DEBUG",
            )


def main() -> None:
    parser = argparse.ArgumentParser(
        prog="isrc-origin",
        description="ISRC country of origin determination",
    )
    parser.add_argument(
        "mode",
        choices=["full", "overrides", "sync"],
        help="Execution mode: full (weekly refresh), overrides (hourly check), sync (push to Redshift)",
    )
    parser.add_argument(
        "--log-file",
        "-l",
        type=str,
        default="isrc-origin-cea.log",
        help="Log file name (stored in LOGDIR)",
    )

    args = parser.parse_args()

    setup_logging(args.log_file)

    try:
        if args.mode == "full":
            from .services.refresh import run_full_refresh

            run_full_refresh()

        elif args.mode == "overrides":
            from .services.overrides import run_overrides_check

            run_overrides_check()

        elif args.mode == "sync":
            from .services.sync import run_sync

            run_sync()

    except KeyboardInterrupt:
        logger.info("Interrupted by user")
        sys.exit(130)
    except Exception as e:
        logger.exception(f"Fatal error: {e}")
        sys.exit(1)


if __name__ == "__main__":
    main()