aors 0.2.0

Useful rs tools for Advent of Code
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use proc_macro::TokenStream;
use quote::quote;
use syn::{parse_macro_input, Ident, ItemFn};

#[proc_macro_attribute]
pub fn main(_: TokenStream, input: TokenStream) -> TokenStream {
    let mut aoc_solution = parse_macro_input!(input as ItemFn);
    aoc_solution.sig.ident = Ident::new("aoc_solution", aoc_solution.sig.ident.span());

    let tokens = quote! {
      /// Returns `actual` when using real problem input, and `example` when using example input
      fn phi<T>(actual: T, example: T) -> T {
        let args: Vec<String> = std::env::args().collect();
        let mut example_input = false;
        if let Some(a) = args.get(1) {
          if a == "example" {
            example_input = true;
          }
        }
        if example_input {
          example
        } else {
          actual
        }
      }

      #aoc_solution
      fn main() {
        let args: Vec<String> = std::env::args().collect();
        let mut example_input = false;
        let day = args.get(1).unwrap();
        let mut input_file = format!("inputs/{}.in", day);
        if let Some(a) = args.get(2) {
          if a == "example" {
            example_input = true;
            input_file = format!("input_examples/{}.in", day);
          }
        }
        let input = &std::fs::read_to_string(&input_file).expect(&format!("failed to read input from {}", input_file));
        let input = input.as_str();
        println!("\x1b[4;1mDay {}:\x1b[0m", day);

        let now = ::std::time::Instant::now();
        let (p1, p2) = aoc_solution(input.trim_end());
        let time = now.elapsed();

        let ans1 = std::fs::read_to_string(format!("answers/{}p1.sol", day)).expect("unable to find answer file");
        let ans2 = std::fs::read_to_string(format!("answers/{}p2.sol", day)).expect("unable to find answer file");

        print!("Part 1: ");
        if ans1 != "" && !example_input {
          if ans1 == p1.to_string() {
            print!("\x1b[32m");
          } else {
            print!("\x1b[31m");
          }
        }
        println!("{}\x1b[0m", p1);

        print!("Part 2: ");
        if ans2 != "" && !example_input {
          if ans2 == p2.to_string() {
            print!("\x1b[32m");
          } else {
            print!("\x1b[31m");
          }
        }
        println!("{}\x1b[0m", p2);
        if !example_input {
          if time.as_millis() <= 10 {
            print!("\x1b[102m"); // green
          } else if time.as_millis() <= 1000 {
            print!("\x1b[103m"); // yellow
          } else {
            print!("\x1b[101m"); // red
          }
          print!("\x1b[30m");
        }
        if time.as_millis() > 0 {
          print!("Time: {}ms", time.as_millis());
        } else {
          print!("Time: {}μs", time.as_micros());
        }
        println!("\x1b[0m");
      }
    };
    TokenStream::from(tokens)
}