61 lines
2.7 KiB
Rust
61 lines
2.7 KiB
Rust
/********************************************************************************
|
|
* Prometheus exporter for monitoring network connectivity using icmp pings *
|
|
* *
|
|
* Copyright (C) 2019-2022 Jan Christian Grünhage *
|
|
* Copyright (C) 2020-2021 Famedly GmbH *
|
|
* *
|
|
* This program is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the GNU Affero General Public License as *
|
|
* published by the Free Software Foundation, either version 3 of the *
|
|
* License, or (at your option) any later version. *
|
|
* *
|
|
* This program is distributed in the hope that it will be useful, *
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
* GNU Affero General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Affero General Public License *
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>. *
|
|
********************************************************************************/
|
|
use std::path::PathBuf;
|
|
|
|
use clap::{ArgEnum, CommandFactory};
|
|
use clap_complete::{generate_to, Shell};
|
|
use cli::Cli;
|
|
|
|
#[path = "src/cli.rs"] mod cli;
|
|
|
|
fn main() -> std::io::Result<()> {
|
|
let mut cli = Cli::command();
|
|
|
|
if let Some(completions_dir) = std::env::var_os("PESHMING_COMPLETIONS_DIR") {
|
|
let completions_dir: PathBuf = completions_dir.into();
|
|
|
|
std::fs::create_dir_all(&completions_dir)
|
|
.expect("Could not create shell completions output folder.");
|
|
|
|
for shell in Shell::value_variants() {
|
|
generate_to(*shell, &mut cli, "peshming", &completions_dir).unwrap_or_else(|err| {
|
|
panic!(
|
|
"Failed to generate shell completions for {}: {}.",
|
|
shell, err
|
|
)
|
|
});
|
|
}
|
|
}
|
|
|
|
if let Some(man_dir) = std::env::var_os("PESHMING_MAN_DIR") {
|
|
let man_dir: PathBuf = man_dir.into();
|
|
|
|
std::fs::create_dir_all(&man_dir).expect("Could not create man page output folder.");
|
|
|
|
let man = clap_mangen::Man::new(cli);
|
|
let mut buffer: Vec<u8> = Default::default();
|
|
man.render(&mut buffer)?;
|
|
|
|
std::fs::write(man_dir.join("peshming.1"), buffer)?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|