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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use fancy_regex::Regex;
use std::fs;
use std::path::Path;
use std::process::exit;

const TOO_FAST: &str = "(You gave an answer too recently.*to wait.)";
const INCORRECT: &str = r"(That's not the right answer[^\.]*.)";
const ALREADY_DONE: &str = r"(You don't seem to be solving.*\.)";
const CORRECT: &str = "(That's the right answer!)";

pub fn post(day: u8, year: u32, example_input: bool, cookie: &str, p1: &str, p2: &str) {
    let day_str = format!("{:0>2}", day);

    if day < 1 || day > 25 {
        eprintln!("\x1b[31;1mIncorrect day. Should be between 1 and 25\x1b[0m");
        exit(1);
    }

    if !Path::new(&format!("src/bin/{}.rs", day_str)).exists() {
        eprintln!("\x1b[31;1mYou do not have a solution for this day\x1b[0m");
        exit(1);
    }

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

    let part = if ans1 == "" { 1 } else { 2 };

    if part == 2 && ans2 != "" {
        println!(
            "⭐ \x1b[103;30mYou've already solved day {}!\x1b[0m ⭐",
            day_str
        );
        return;
    }

    if example_input {
        eprintln!("\x1b[41;30mTried to submit with example input\x1b[0m");
        exit(1);
    }

    let answer = match part {
        1 => p1,
        _ => p2,
    };

    println!(
        "\x1b[4;1mPosting {} to day {} part {} ({})\x1b[0m\n",
        answer, day_str, part, year
    );

    let html = post_req(year, day, answer, part, cookie);

    for err in [TOO_FAST, INCORRECT, ALREADY_DONE] {
        let err_re = Regex::new(err).unwrap();
        if let Ok(c) = err_re.captures(&html) {
            if let Some(c) = c {
                eprintln!("\x1b[41;30m{}\x1b[0m", c.get(1).unwrap().as_str());
            }
        }
    }

    let corr_re = Regex::new(CORRECT).unwrap();
    if let Ok(c) = corr_re.captures(&html) {
        if let Some(c) = c {
            println!("\x1b[102;30m{}\x1b[0m", c.get(1).unwrap().as_str());
            if part == 1 {
                write_ans(&day_str, answer, "one".to_string());
            } else {
                write_ans(&day_str, answer, "two".to_string());
            }
        }
    }
}

fn post_req(year: u32, day: u8, answer: &str, part: i32, cookie: &str) -> String {
    ureq::post(&format!(
        "https://adventofcode.com/{}/day/{}/answer",
        year, day
    ))
    .set("Cookie", &format!("session={}", cookie))
    .send_form(&[("answer", answer), ("level", &part.to_string())])
    .unwrap()
    .into_string()
    .unwrap()
}

fn write_ans(day_str: &str, answer: &str, part_string: String) {
    let ans_path = format!("answers/{}.sol", day_str);
    let ans_file = fs::read_to_string(&ans_path).unwrap();
    let re = Regex::new(&(format!("part {}: ", part_string).to_owned() + r"([^\n]*)")).unwrap();
    let new_ans_file = re
        .replace(
            &ans_file,
            format!("part {}: {}", part_string, answer).as_str(),
        )
        .to_string();
    let ansi_escape = Regex::new(r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])").unwrap();
    let new_ans_file = ansi_escape.replace_all(&new_ans_file, "").to_string();

    fs::write(&ans_path, new_ans_file).unwrap();
}