/
main.rs
57 lines (49 loc) · 1.63 KB
/
main.rs
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
use rand::{rngs::StdRng, seq::SliceRandom, SeedableRng};
use serde::Deserialize;
use std::process::ExitCode;
const RESPONSES: &'static str = include_str!("../responses.json");
const AFFECTIONATE_TERM_PLACEHOLDER: &'static str = "AFFECTIONATE_TERM";
#[derive(Deserialize)]
struct Responses {
positive: Vec<String>,
negative: Vec<String>,
}
enum ResponseType {
Positive,
Negative,
}
fn main() -> Result<ExitCode, Box<dyn std::error::Error>> {
let cargo = std::env::var("CARGO")?;
let mut arg_iter = std::env::args();
let _cargo = arg_iter.next();
let _mommy = arg_iter.next();
let mut cmd = std::process::Command::new(cargo);
cmd.args(arg_iter);
let status = cmd.status()?;
eprintln!("\x1b[1m");
if status.success() {
eprintln!("{}", select_response(ResponseType::Positive))
} else {
eprintln!("{}", select_response(ResponseType::Negative));
}
eprintln!("\x1b[0m");
Ok(ExitCode::from(status.code().unwrap_or(-1) as u8))
}
fn select_response(response_type: ResponseType) -> String {
let affectionate_term =
std::env::var("CARGO_MOMMYS_LITTLE").unwrap_or_else(|_| "girl".to_owned());
let mut rng = StdRng::from_entropy();
let responses: Responses = serde_json::from_str(RESPONSES).expect("RESPONSES to be valid JSON");
return match response_type {
ResponseType::Positive => responses.positive,
ResponseType::Negative => responses.negative,
}
.choose(&mut rng)
.expect("non-zero amount of responses")
.replace(AFFECTIONATE_TERM_PLACEHOLDER, &affectionate_term);
}
#[cfg(test)]
#[test]
fn test() {
panic!("oops!!");
}