#!/usr/bin/env python3

from sys import argv
from os import mkdir, chdir

def main():
    day = int(argv[1])
    with open("Cargo.toml", "r") as f:
        cargo_toml = f.read()
    new_cargo_toml = []
    for line in cargo_toml.splitlines():
        if line == "]":
            new_cargo_toml.append(f"    \"day{day}\",")
        new_cargo_toml.append(line)
    with open("Cargo.toml", "w") as f:
        for line in new_cargo_toml:
            f.write(line)
            f.write("\n")

    mkdir(f"day{day}")
    chdir(f"day{day}")

    with open("Cargo.toml", "w") as f:
        f.write(f"""
[package]
name = "day{day}"
version = "0.1.0"
edition = "2021"

[dependencies]
lib = {{ path = "../lib" }}
        """.strip())
        f.write("\n")

    mkdir("src")

    with open("src/main.rs", "w") as f:
        f.write(f"""
fn main() {{
    let input = lib::read_input({day});

    part1(&input);
    part2(&input);
}}

fn part1(input: &str) {{
}}

fn part2(input: &str) {{
}}
        """.strip())
        f.write("\n")

if __name__ == "__main__":
    main()