check-macos-updates 0.3.3

A monitoring plugin to check if macOS system updates are available.
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

use check_macos_updates::*;
use clap::Parser;
use nagios_range::NagiosRange as ThresholdRange;
use plist::from_file;
use std::process;

const ABOUT_TEXT: &str = r#"
A monitoring plugin that checks for available MacOS updates.

Thresholds are defined using monitoring plugin range syntax. Examples:
+------------------+-------------------------------------------------+
| Range definition | Generate an alert if x...                       |
+------------------+-------------------------------------------------+
| 10               | < 0 or > 10, (outside the range of {0 .. 10})   |
+------------------+-------------------------------------------------+
| 10:              | < 10, (outside {10 .. ∞})                       |
+------------------+-------------------------------------------------+
| ~:10             | > 10, (outside the range of {-∞ .. 10})         |
+------------------+-------------------------------------------------+
| 10:20            | < 10 or > 20, (outside the range of {10 .. 20}) |
+------------------+-------------------------------------------------+
| @10:20           | ≥ 10 and ≤ 20, (inside the range of {10 .. 20}) |
+------------------+-------------------------------------------------+
"#;

#[derive(Parser, Debug)]
#[command(author, version, long_about = None, about = ABOUT_TEXT)]
struct Args {
    /// Force manual check with `softwareupdate -l` (slow)
    #[arg(short, long)]
    force_manual: bool,

    /// Warning limit for number of updates available
    #[arg(short, long, default_value = "0")]
    warning: Option<String>,

    /// Critical limit for number of updates available
    #[arg(short, long)]
    critical: Option<String>,
}

fn exit_with_message(status: Status) {
    println!("{}", status);
    process::exit(status.to_int());
}

/// Check for macOS updates. Returns warning if updates are available.
fn main() {
    if !cfg!(target_os = "macos") {
        exit_with_message(Status::Unknown(UnkownVariant::NotMacOS))
    }

    let args = Args::parse();

    if args.warning.is_none() && args.critical.is_none() {
        exit_with_message(Status::Unknown(UnkownVariant::NoThresholds))
    }

    let mut warning: Option<ThresholdRange> = None;

    if let Some(w) = args.warning {
        let w_range = ThresholdRange::from(&w);
        match w_range {
            Ok(r) => warning = Some(r),
            Err(e) => exit_with_message(Status::Unknown(UnkownVariant::RangeParseError(w, e))),
        }
    }

    let mut critical: Option<ThresholdRange> = None;

    if let Some(c) = args.critical {
        let c_range = ThresholdRange::from(&c);
        match c_range {
            Ok(r) => critical = Some(r),
            Err(e) => exit_with_message(Status::Unknown(UnkownVariant::RangeParseError(c, e))),
        }
    }

    let thresholds = Thresholds { warning, critical };

    if args.force_manual {
        let check_status = check_softwareupdate_output(&softwareupdate_output(), &thresholds);
        exit_with_message(check_status)
    } else {
        let software_update_plist = from_file(PLIST_FILE);
        let check_status = match software_update_plist {
            Ok(u) => determine_updates(&u, &thresholds),
            Err(_) => Status::Unknown(UnkownVariant::UnableToParsePlist),
        };
        exit_with_message(check_status)
    }
}