mirror of
https://github.com/ClementTsang/bottom.git
synced 2024-09-11 03:35:35 +03:00
* clean up Cargo.toml * some small cleanup * refactor: group together similar args in the help generation and code This groups together related arguments in both the help text and the code itself. * update changelog * clippy * builder pattern instead
119 lines
3.8 KiB
Rust
119 lines
3.8 KiB
Rust
#[allow(dead_code)]
|
|
#[path = "src/options/args.rs"]
|
|
mod args;
|
|
|
|
use std::{
|
|
env, fs, io,
|
|
path::{Path, PathBuf},
|
|
};
|
|
|
|
use clap::Command;
|
|
use clap_complete::{generate_to, shells::Shell, Generator};
|
|
use clap_complete_fig::Fig;
|
|
use clap_complete_nushell::Nushell;
|
|
|
|
use crate::args::build_app;
|
|
|
|
fn create_dir(dir: &Path) -> io::Result<()> {
|
|
let res = fs::create_dir_all(dir);
|
|
match &res {
|
|
Ok(()) => {}
|
|
Err(err) => {
|
|
eprintln!("Failed to create a directory at location {dir:?}, encountered error {err:?}. Aborting...",);
|
|
}
|
|
}
|
|
|
|
res
|
|
}
|
|
|
|
fn generate_completions<G>(to_generate: G, cmd: &mut Command, out_dir: &Path) -> io::Result<PathBuf>
|
|
where
|
|
G: Generator,
|
|
{
|
|
generate_to(to_generate, cmd, "btm", out_dir)
|
|
}
|
|
|
|
fn btm_generate() -> io::Result<()> {
|
|
const ENV_KEY: &str = "BTM_GENERATE";
|
|
|
|
match env::var_os(ENV_KEY) {
|
|
Some(var) if !var.is_empty() => {
|
|
const COMPLETION_DIR: &str = "./target/tmp/bottom/completion/";
|
|
const MANPAGE_DIR: &str = "./target/tmp/bottom/manpage/";
|
|
|
|
let completion_out_dir = PathBuf::from(COMPLETION_DIR);
|
|
let manpage_out_dir = PathBuf::from(MANPAGE_DIR);
|
|
|
|
create_dir(&completion_out_dir)?;
|
|
create_dir(&manpage_out_dir)?;
|
|
|
|
// Generate completions
|
|
let mut app = build_app();
|
|
generate_completions(Shell::Bash, &mut app, &completion_out_dir)?;
|
|
generate_completions(Shell::Zsh, &mut app, &completion_out_dir)?;
|
|
generate_completions(Shell::Fish, &mut app, &completion_out_dir)?;
|
|
generate_completions(Shell::PowerShell, &mut app, &completion_out_dir)?;
|
|
generate_completions(Shell::Elvish, &mut app, &completion_out_dir)?;
|
|
generate_completions(Fig, &mut app, &completion_out_dir)?;
|
|
generate_completions(Nushell, &mut app, &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)?;
|
|
fs::write(manpage_out_dir.join("btm.1"), buffer)?;
|
|
}
|
|
_ => {}
|
|
}
|
|
|
|
println!("cargo:rerun-if-env-changed={ENV_KEY}");
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn extract_sha(sha: Option<&str>) -> Option<&str> {
|
|
sha.and_then(|sha: &str| sha.get(0..8))
|
|
}
|
|
|
|
fn output_nightly_version(version: &str, git_hash: &str) {
|
|
println!("cargo:rustc-env=NIGHTLY_VERSION={version}-nightly-{git_hash}");
|
|
}
|
|
|
|
fn nightly_version() {
|
|
const ENV_KEY: &str = "BTM_BUILD_RELEASE_CALLER";
|
|
|
|
match env::var_os(ENV_KEY) {
|
|
Some(var) if !var.is_empty() && var == "nightly" => {
|
|
let version = env!("CARGO_PKG_VERSION");
|
|
|
|
if let Some(hash) = extract_sha(option_env!("CIRRUS_CHANGE_IN_REPO")) {
|
|
// May be set if we're building with Cirrus CI.
|
|
output_nightly_version(version, hash);
|
|
} else if let Some(hash) = extract_sha(option_env!("GITHUB_SHA")) {
|
|
// May be set if we're building with GHA.
|
|
output_nightly_version(version, hash);
|
|
} else if let Ok(output) = std::process::Command::new("git")
|
|
.args(["rev-parse", "--short=8", "HEAD"])
|
|
.output()
|
|
{
|
|
// If we're not building in either, we do the lazy thing and fall back to
|
|
// manually grabbing info using git as a command.
|
|
let hash = String::from_utf8(output.stdout).unwrap();
|
|
output_nightly_version(version, &hash);
|
|
}
|
|
}
|
|
_ => {}
|
|
}
|
|
|
|
println!("cargo:rerun-if-env-changed={ENV_KEY}");
|
|
println!("cargo:rerun-if-env-changed=CIRRUS_CHANGE_IN_REPO");
|
|
}
|
|
|
|
fn main() -> io::Result<()> {
|
|
btm_generate()?;
|
|
nightly_version();
|
|
|
|
Ok(())
|
|
}
|