#!/usr/bin/env python

import pathlib
from aoc import cli, input_file, run, exceptions
import sys
import os
import dotenv


SOLUTION_BASE = pathlib.Path("./src/solution")
INPUT_BASE = pathlib.Path("./data/input")
EXAMPLE_BASE = pathlib.Path("./data/example")


def main() -> None:
    args = cli.argument_parser().parse_args()

    session_token = os.getenv("AOC_SESSION_TOKEN")

    try:
        match args.command:
            case "solve":
                input_data = input_file.get(
                    args.year,
                    args.day,
                    session_token,
                    EXAMPLE_BASE if args.example else INPUT_BASE,
                )
                run.run_day(args.year, args.day, input_data, SOLUTION_BASE)
            case "create":
                input_file.create(
                    args.year,
                    args.day,
                    SOLUTION_BASE,
                    INPUT_BASE,
                    EXAMPLE_BASE,
                )
            case "download":
                input_file.create_input_file(
                    args.year, args.day, session_token, INPUT_BASE, args.force
                )

    except exceptions.AocError as e:
        print(f"error: {e}", file=sys.stderr)
        exit(1)


if __name__ == "__main__":
    dotenv.load_dotenv()
    main()