mirror of
https://github.com/aramperes/nut-rs.git
synced 2025-09-08 05:08:31 -04:00
116 lines
3.6 KiB
Rust
116 lines
3.6 KiB
Rust
//! # rupsc
|
|
//! A demo program to display UPS variables.
|
|
//! This a Rust clone of [upsc](https://github.com/networkupstools/nut/blob/master/clients/upsc.c).
|
|
//!
|
|
//! P.S.: pronounced "r-oopsie".
|
|
use core::convert::TryInto;
|
|
|
|
use anyhow::Context;
|
|
use clap::{App, Arg};
|
|
|
|
use rups::UpsdName;
|
|
|
|
mod cmd;
|
|
|
|
fn main() -> anyhow::Result<()> {
|
|
let args = App::new(clap::crate_name!())
|
|
.version(clap::crate_version!())
|
|
.author(clap::crate_authors!())
|
|
.about(clap::crate_description!())
|
|
.arg(
|
|
Arg::with_name("list")
|
|
.short("l")
|
|
.conflicts_with_all(&["list-full", "clients"])
|
|
.takes_value(false)
|
|
.help("Lists each UPS on <hostname>, one per line."),
|
|
)
|
|
.arg(
|
|
Arg::with_name("list-full")
|
|
.short("L")
|
|
.conflicts_with_all(&["list", "clients"])
|
|
.takes_value(false)
|
|
.help("Lists each UPS followed by its description (from ups.conf)."),
|
|
)
|
|
.arg(
|
|
Arg::with_name("clients")
|
|
.short("c")
|
|
.conflicts_with_all(&["list", "list-full"])
|
|
.takes_value(false)
|
|
.help("Lists each client connected on <upsname>, one per line."),
|
|
)
|
|
.arg(
|
|
Arg::with_name("debug")
|
|
.short("D")
|
|
.long("debug")
|
|
.takes_value(false)
|
|
.help("Enables debug mode (logs network commands to stderr)."),
|
|
)
|
|
.arg(
|
|
Arg::with_name("ssl")
|
|
.short("S")
|
|
.long("ssl")
|
|
.takes_value(false)
|
|
.help("Enables SSL on the connection with upsd."),
|
|
)
|
|
.arg(
|
|
Arg::with_name("insecure-ssl")
|
|
.long("insecure-ssl")
|
|
.takes_value(false)
|
|
.help("Disables SSL verification on the connection with upsd."),
|
|
)
|
|
.arg(
|
|
Arg::with_name("upsd-server")
|
|
.required(false)
|
|
.value_name("[upsname][@<hostname>[:<port>]]")
|
|
.help("upsd server (with optional upsname, if applicable)."),
|
|
)
|
|
.arg(
|
|
Arg::with_name("variable")
|
|
.required(false)
|
|
.value_name("variable")
|
|
.help("Optional, display this variable only."),
|
|
)
|
|
.get_matches();
|
|
|
|
let server: UpsdName = args.value_of("upsd-server").map_or_else(
|
|
|| Ok(UpsdName::default()),
|
|
|s| s.try_into().with_context(|| "Invalid upsd server name"),
|
|
)?;
|
|
|
|
let debug = args.is_present("debug");
|
|
let insecure_ssl = args.is_present("insecure-ssl");
|
|
let ssl = insecure_ssl || args.is_present("ssl");
|
|
|
|
let host = server.try_into()?;
|
|
let config = rups::ConfigBuilder::new()
|
|
.with_host(host)
|
|
.with_debug(debug)
|
|
.with_ssl(ssl)
|
|
.with_insecure_ssl(insecure_ssl)
|
|
.build();
|
|
|
|
if args.is_present("list") {
|
|
return cmd::list_devices(config, false);
|
|
}
|
|
|
|
if args.is_present("list-full") {
|
|
return cmd::list_devices(config, true);
|
|
}
|
|
|
|
if args.is_present("clients") {
|
|
return cmd::list_clients(config, get_ups_name(&server)?);
|
|
}
|
|
|
|
// Fallback: prints one variable (or all of them)
|
|
if let Some(variable) = args.value_of("variable") {
|
|
cmd::print_variable(config, get_ups_name(&server)?, variable)
|
|
} else {
|
|
cmd::list_variables(config, get_ups_name(&server)?)
|
|
}
|
|
}
|
|
|
|
fn get_ups_name<'a>(server: &'a UpsdName) -> anyhow::Result<&'a str> {
|
|
server
|
|
.upsname
|
|
.with_context(|| "ups name must be specified: <upsname>[@<hostname>[:<port>]]")
|
|
}
|