aors 0.5.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

use crate::commands;
use fancy_regex::Regex;
use std::error::Error;

pub fn run_day(day_str: &str, example_input: bool, quiet: bool) -> Option<(String, String, usize)> {
    let part_one: Regex = Regex::new(r"Part 1: ([^\n]+)").ok()?;
    let part_two: Regex = Regex::new(r"Part 2: ([^\n]+)").ok()?;
    let mut args = vec!["run"];
    if quiet {
        args.push("--quiet");
    }
    args.append(&mut vec!["--release", "--bin", day_str, day_str]);
    if example_input {
        args.push("example");
    }

    let output = commands::cmd("cargo", args)?;

    let p1 = get_answer(part_one, &output)?;
    let p2 = get_answer(part_two, &output)?;
    let time = extract_microseconds(&output).ok()?;
    Some((p1, p2, time))
}

fn get_answer(r: Regex, text: &str) -> Option<String> {
    let ans = r.captures(text).ok()??.get(1)?.as_str().to_string();
    let ansi_escape = Regex::new(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])").ok()?;
    Some(ansi_escape.replace_all(&ans, "").to_string())
}

fn extract_microseconds(output: &str) -> Result<usize, Box<dyn Error>> {
    let out = output.lines().last().expect(
        "Execution failed. Make sure all existing DAY.rs files have corresponding DAY.in files",
    );
    let ansi_escape = Regex::new(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])").unwrap();
    let out = ansi_escape.replace_all(out, "").to_string();
    let time = if out.ends_with("ms") {
        out["Time: ".len()..out.len() - 2].parse::<usize>()? * 1000
    } else {
        out["Time: ".len()..out.len() - 3].parse::<usize>()?
    };
    Ok(time)
}