main.py

import importlib
import os
import webbrowser
from datetime import date, datetime
from shutil import copyfile
from typing import Literal

import click

import aoc
from infra import get_day_path, get_mod_path


def _store_markdown(year: int, day: int) -> None:
    day_path = get_day_path(year, day)
    markdown = aoc.get_markdown(year=year, day=day)
    with open(os.path.join(day_path, "README.md"), "w") as readme:
        readme.write(markdown.split("\n\n", 1)[0] + "\n")
    with open(os.path.join(day_path, "PUZZLE.md"), "w") as readme:
        readme.write(markdown)


@click.group()
def cli() -> None:
    pass


def _log_entry(year: int, day: int, part_name: Literal["a", "b"], entry: str) -> None:
    day_path = get_day_path(year, day)
    with open(os.path.join(day_path, f"part-{part_name}.log"), "a") as log_file:
        log_file.write(f"{datetime.now().isoformat()} {entry}\n")


def _log_start(year: int, day: int, part_name: Literal["a", "b"]) -> datetime:
    day_path = get_day_path(year, day)
    log_file_path = os.path.join(day_path, f"part-{part_name}.log")

    if not os.path.isfile(log_file_path):
        _log_entry(year, day, part_name, "Log file created")
        return datetime.now()

    with open(log_file_path) as log_file:
        return datetime.fromisoformat(log_file.readline().split()[0])


@cli.command()
@click.option("-y", "--year", type=int, default=date.today().year)
@click.option("-d", "--day", type=int, default=date.today().day)
def new(year: int, day: int) -> None:
    day_path = get_day_path(year, day)

    if os.path.isdir(day_path):
        print("day alreddy exists")
    else:
        os.makedirs(day_path)
        dest_path_file_name = os.path.join(day_path, "a.py")
        if not os.path.exists(dest_path_file_name):
            copyfile("templates/base.py", dest_path_file_name)

        test_year_dir_path = os.path.join(
            os.path.dirname(__file__),
            "tests",
            str(year),
        )
        if not os.path.isdir(test_year_dir_path):
            os.makedirs(test_year_dir_path)
        test_dest_path_file_name = os.path.join(
            test_year_dir_path,
            f"test_{year}_{str(day).zfill(2)}.py",
        )
        if not os.path.exists(test_dest_path_file_name):
            copyfile("templates/test.py", test_dest_path_file_name)

    webbrowser.open(aoc.get_url(year, day))
    _log_entry(year, day, "a", "Started puzzle")
    _store_markdown(year, day)


def _correct_submission(year: int, day: int, part: Literal["1", "2"]) -> None:
    """
    Store markdown whenever a correct submission has been registered.
    """
    _store_markdown(year, day)
    if part == "1":
        # copy a.py to b.py
        _log_entry(year, day, "b", "Started puzzle")

        day_path = get_day_path(year, day)
        source_path_file_name = os.path.join(day_path, "a.py")
        dest_path_file_name = os.path.join(day_path, "b.py")
        copyfile(source_path_file_name, dest_path_file_name)


@cli.command()
@click.argument("part", type=click.Choice(["1", "2"]))
@click.option("-y", "--year", type=int, default=date.today().year)
@click.option("-d", "--day", type=int, default=date.today().day)
@click.option("-i", "--input-string")
@click.option("--submit", is_flag=True)
def run(part: Literal["1", "2"], year: int, day: int, input_string: str, submit: bool) -> None:
    part_name: Literal["a", "b"] = "a" if part == "1" else "b"
    mod = importlib.import_module(get_mod_path(year, day, part_name))

    answer = mod.main(provided_input=input_string)

    print(f"Part {part}:", answer)

    if not submit:
        return

    text, correct = aoc.submit(year=year, day=day, part=int(part), answer=answer)

    start_time = _log_start(year, day, part_name)
    duration = datetime.now() - start_time

    if correct:
        _log_entry(year, day, part_name, f"Correct answer: {answer}! Puzzle completed in {str(duration)}")
        _correct_submission(year, day, part)
    else:
        _log_entry(year, day, part_name, f"Incorrect guess: {answer}. {text} Time passed: {str(duration)}")


@cli.command()
@click.argument("part", type=click.Choice(["1", "2"]))
@click.argument("answer")
@click.option("-y", "--year", type=int, default=date.today().year)
@click.option("-d", "--day", type=int, default=date.today().day)
def submit(part: Literal["1", "2"], answer: str, year: int, day: int) -> None:
    part_name: Literal["a", "b"] = "a" if part == "1" else "b"
    text, correct = aoc.submit(year=year, day=day, part=int(part), answer=answer)

    start_time = _log_start(year, day, part_name)
    duration = datetime.now() - start_time

    if correct or "already been completed" in text:
        _log_entry(year, day, part_name, f"Correct answer: {answer}! Puzzle completed in {str(duration)}")
        _correct_submission(year, day, part)
    else:
        _log_entry(year, day, part_name, f"Incorrect guess: {answer}. {text} Time passed: {str(duration)}")


@cli.command()
@click.option("-y", "--year", type=int, default=date.today().year)
@click.option("-d", "--day", type=int, default=date.today().day)
def markdown(year: int, day: int) -> None:
    _store_markdown(year, day)


@cli.command()
@click.option("-y", "--year", type=int, default=date.today().year)
@click.option("-d", "--day", type=int, default=date.today().day)
def test(year: int, day: int) -> None:
    import subprocess

    subprocess.call(["pytest", f"tests/{year}/test_{year}_{day}.py"])


if __name__ == "__main__":
    cli()