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

use std::process::exit;

use confy;
use filenamify::filenamify;
use serde::{Deserialize, Serialize};

const CONFY_NAME: &str = "aors";

#[derive(Default, Debug, Serialize, Deserialize)]
struct YearConfig {
    year: u32,
}

#[derive(Default, Debug, Serialize, Deserialize)]
struct CookieConfig {
    cookie: String,
}

pub fn config(year: Option<u32>, cookie: Option<String>) -> (u32, String) {
    let file = std::env::current_dir().unwrap();
    let file = filenamify(file.to_str().unwrap());

    // year needs folder-scope
    let mut year_config: YearConfig = confy::load(CONFY_NAME, file.as_str()).unwrap();
    let mut cookie_config: CookieConfig = confy::load(CONFY_NAME, None).unwrap();

    if let Some(year) = year {
        year_config.year = year;
        confy::store(CONFY_NAME, file.as_str(), &year_config).unwrap();
    }

    if let Some(cookie) = cookie {
        cookie_config.cookie = cookie.to_owned();
        confy::store(CONFY_NAME, None, &cookie_config).unwrap();
    }

    missing_config_msg(year_config.year, &cookie_config.cookie);

    (year_config.year, cookie_config.cookie)
}

fn missing_config_msg(year: u32, cookie: &str) {
    if year == 0 {
        println!(
            "Please enter the year you want to solve for:\n\t\x1b[1maors --year\x1b[0m <YEAR>\n"
        );
    }
    if cookie == "" {
        println!("Please enter a session cookie:\n\t\x1b[1maors --cookie\x1b[0m <COOKIE>");
        println!("This cookie is acquired like so: https://github.com/wimglenn/advent-of-code-wim/issues/1\n");
    }
    if year == 0 || cookie == "" {
        exit(1);
    }
}