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="artist-metadata",
        description="Luminate artist metadata mapped to Spotify artist IDs",
    )
    parser.add_argument(
        "mode",
        choices=["refresh", "sync", "all"],
        help="Execution mode: refresh (Snowflake append), sync (push to Redshift), all (both)",
    )
    parser.add_argument(
        "--log-file",
        "-l",
        type=str,
        default="artist-metadata-cea.log",
        help="Log file name (stored in LOGDIR)",
    )

    args = parser.parse_args()

    setup_logging(args.log_file)

    try:
        if args.mode in ("refresh", "all"):
            from .services.refresh import run_refresh

            run_refresh()

        if args.mode in ("sync", "all"):
            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()