# -*- coding: utf-8 -*-
import argparse
import importlib
import os
from pathlib import Path
from shutil import copytree
from time import perf_counter
from typing import List, Callable


def day(assignment_part: str) -> str:
    return {
        "1": "One",
        "2": "Two",
    }[assignment_part]


def kwargs(kwarg: str) -> List:
    return kwarg.split("=")


def run(args: argparse.Namespace):
    assignment_day = importlib.import_module(args.day)

    Assignment = getattr(assignment_day, f"Assignment{args.part}")
    assignment = Assignment(
        path=os.path.dirname(assignment_day.__file__), **dict(args.kwargs)
    )

    start = perf_counter()
    print(
        assignment.run(
            input=assignment.read_input(example=args.example),
        )
    )
    end = perf_counter()
    delta = end - start
    print()
    print(f'{(delta if delta > 1 else delta * 1000):.3f}{"s" if delta > 1 else "ms"}')


def new(args: argparse.Namespace):
    self = importlib.import_module(".", package="aoc")
    path = Path(self.__file__)
    template = path.parent.joinpath("template")
    target = path.parent.parent.joinpath(args.day)

    try:
        copytree(template, target)
    except FileExistsError:
        print(f"{args.day} already exists")


parser = argparse.ArgumentParser(description="Advent of Code")

parser.add_argument("command", type=str, choices=["run", "new"])
parser.add_argument("day", type=str, help="Assignment day")
parser.add_argument(
    "-p",
    "--part",
    type=day,
    nargs="?",
    default="1",
    help="Assingment part. Defaults to one.",
)
parser.add_argument("--example", default=False, action="store_true")
parser.add_argument("kwargs", type=kwargs, nargs="*")


if __name__ == "__main__":
    args = parser.parse_args()

    match args.command:
        case "run":
            run(args)
        case "new":
            new(args)