56 lines
1.7 KiB
Rust
56 lines
1.7 KiB
Rust
use clap_complete::{generate_to, shells::Shell};
|
|
use std::{
|
|
env, fs,
|
|
io::Result,
|
|
path::{Path, PathBuf},
|
|
};
|
|
|
|
include!("src/clap.rs");
|
|
|
|
fn create_dir(dir: &Path) -> Result<()> {
|
|
let res = fs::create_dir_all(&dir);
|
|
match &res {
|
|
Ok(()) => {}
|
|
Err(err) => {
|
|
eprintln!(
|
|
"Failed to create a directory at location {:?}, encountered error {:?}. Aborting...",
|
|
dir, err
|
|
);
|
|
}
|
|
}
|
|
|
|
res
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
if env::var_os("GENERATE").is_some() {
|
|
// OUT_DIR is where extra build files are written to for Cargo.
|
|
let completion_out_dir = PathBuf::from("completion");
|
|
let manpage_out_dir = PathBuf::from("manpage");
|
|
|
|
create_dir(&completion_out_dir)?;
|
|
create_dir(&manpage_out_dir)?;
|
|
|
|
// Generate completions
|
|
let mut app = build_app();
|
|
generate_to(Shell::Bash, &mut app, "btm", &completion_out_dir)?;
|
|
generate_to(Shell::Zsh, &mut app, "btm", &completion_out_dir)?;
|
|
generate_to(Shell::Fish, &mut app, "btm", &completion_out_dir)?;
|
|
generate_to(Shell::PowerShell, &mut app, "btm", &completion_out_dir)?;
|
|
generate_to(Shell::Elvish, &mut app, "btm", &completion_out_dir)?;
|
|
|
|
// Generate manpage
|
|
let app = app.name("btm");
|
|
let man = clap_mangen::Man::new(app);
|
|
let mut buffer: Vec<u8> = Default::default();
|
|
man.render(&mut buffer)?;
|
|
std::fs::write(manpage_out_dir.join("btm.1"), buffer)?;
|
|
}
|
|
|
|
println!("cargo:rerun-if-changed=build.rs");
|
|
println!("cargo:rerun-if-changed=./src/clap.rs");
|
|
println!("cargo:rerun-if-env-changed=GENERATE");
|
|
|
|
Ok(())
|
|
}
|