aoc-companion-codegen 0.1.2

use proc_macro::TokenStream;
use std::collections::HashMap;
use syn::{
    parse::{Parse, ParseStream},
    Ident, LitInt, Token,
};

struct Args {
    day: LitInt,
    parser: Option<Ident>,
    part1: Option<Ident>,
    part2: Option<Ident>,
}

impl Parse for Args {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let day = input.parse()?;
        let mut args = HashMap::new();

        input.parse::<Token![,]>()?;
        while let Ok(id) = input.parse::<Ident>() {
            match id.to_string().as_str() {
                "parser" => {
                    input.parse::<Token![=]>()?;
                    let parser: Ident = input.parse()?;
                    args.insert("parser", parser);
                }
                "part1" => {
                    input.parse::<Token![=]>()?;
                    let part1: Ident = input.parse()?;
                    args.insert("part1", part1);
                }

                "part2" => {
                    input.parse::<Token![=]>()?;
                    let part2: Ident = input.parse()?;
                    args.insert("part2", part2);
                }
                _ => return Err(syn::Error::new(id.span(), "unknown argument")),
            }

            if input.peek(Token![,]) {
                input.parse::<Token![,]>()?;
            } else {
                break;
            }
        }

        Ok(Args {
            day,
            parser: args.get("parser").cloned(),
            part1: args.get("part1").cloned(),
            part2: args.get("part2").cloned(),
        })
    }
}

/// Generates a main function for the day.
/// # Panics
/// Panics if the day is not a number.
#[proc_macro_attribute]
pub fn day(attr: TokenStream, _item: TokenStream) -> TokenStream {
    let args = syn::parse_macro_input!(attr as Args);
    let day: u8 = args.day.base10_parse().unwrap();
    let parser = args.parser;
    let part1 = args.part1;
    let part2 = args.part2;

    quote::quote! {
        fn main() {
            println!("Running Advent of Code Day {}", #day);
            _ = aoc_companion::input::apply(&format!("input-day-{:02}.txt", #day), |input| {
                aoc_companion::timed(|| {
                    let input = #parser(&input);
                    aoc_companion::timed(|| println!("Part 1: {}", #part1(&input)));
                    aoc_companion::timed(|| println!("Part 2: {}", #part2(&input)));
                });
            });
        }
    }
    .into()
}