miasma 0.3.0

Trap malicious AI web scrapers in an endless pit of poisoned training data.
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
105

use std::error::Error;
use std::io::Write;
use std::time::Duration;

use colored::Colorize;
use serde::Deserialize;

use miasma::MIASMA_USER_AGENT;

#[derive(Deserialize)]
struct CrateInfo {
    max_stable_version: String,
}

#[derive(Deserialize)]
struct CratesIOResponse {
    #[serde(rename = "crate")]
    crate_info: CrateInfo,
}

const VERSION: &str = env!("CARGO_PKG_VERSION");

pub async fn check_for_new_version() {
    let _res: Result<(), Box<dyn Error>> = async {
        let resp = reqwest::Client::builder()
            .user_agent(MIASMA_USER_AGENT)
            .timeout(Duration::from_secs(5))
            .build()?
            .get("https://crates.io/api/v1/crates/miasma")
            .send()
            .await?
            .json::<CratesIOResponse>()
            .await?;

        let latest = resp.crate_info.max_stable_version;
        print_update_message(VERSION, &latest);

        Ok(())
    }
    .await;

    #[cfg(debug_assertions)]
    #[allow(clippy::used_underscore_binding)]
    if let Err(e) = _res {
        eprintln!("Failed to check for latest version: {e}");
    }
}

fn print_update_message(current: &str, latest: &str) {
    if current != latest {
        // fd redirection problems with eprintln! and gag require writeln! to the live stderr handle
        let mut stderr = std::io::stderr().lock();
        let _ = writeln!(stderr, "\n------- New Version Available -------");
        let _ = writeln!(
            stderr,
            "Installed ({}) -> Latest ({})",
            current.red(),
            latest.green()
        );
        let _ = writeln!(stderr, "-------------------------------------\n");
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use gag::BufferRedirect;
    use serial_test::serial;
    use std::io::Read;

    #[test]
    #[serial]
    fn version_mismatch_prints_to_stderr() {
        let capture = BufferRedirect::stderr().unwrap();
        print_update_message("0.1.0", "0.1.1");
        let _ = std::io::stderr().flush();
        let mut buffer = capture.into_inner();
        let mut output = String::new();
        buffer.read_to_string(&mut output).unwrap();
        assert!(
            output.contains("New Version Available"),
            "expected update banner: {output:?}"
        );
        assert!(
            output.contains("0.1.0"),
            "expected current version: {output:?}"
        );
        assert!(
            output.contains("0.1.1"),
            "expected latest version: {output:?}"
        );
    }

    #[test]
    #[serial]
    fn version_match_prints_nothing_to_stderr() {
        let capture = BufferRedirect::stderr().unwrap();
        print_update_message("0.1.0", "0.1.0");
        let _ = std::io::stderr().flush();
        let mut buffer = capture.into_inner();
        let mut output = String::new();
        buffer.read_to_string(&mut output).unwrap();
        assert!(output.is_empty(), "expected no stderr: {output:?}");
    }
}