mirror of https://github.com/sharkdp/bat.git
82 lines
2.5 KiB
Rust
82 lines
2.5 KiB
Rust
use std::{collections::HashMap, fs, path::Path};
|
|
|
|
fn main() -> anyhow::Result<()> {
|
|
#[cfg(feature = "application")]
|
|
gen_man_and_comp()?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Generate manpage and shell completions for the bat application.
|
|
#[cfg(feature = "application")]
|
|
fn gen_man_and_comp() -> anyhow::Result<()> {
|
|
// Read environment variables.
|
|
let project_name = option_env!("PROJECT_NAME").unwrap_or("bat");
|
|
let executable_name = option_env!("PROJECT_EXECUTABLE").unwrap_or(project_name);
|
|
let executable_name_uppercase = executable_name.to_uppercase();
|
|
static PROJECT_VERSION: &str = env!("CARGO_PKG_VERSION");
|
|
|
|
let mut variables = HashMap::new();
|
|
variables.insert("PROJECT_NAME", project_name);
|
|
variables.insert("PROJECT_EXECUTABLE", executable_name);
|
|
variables.insert("PROJECT_EXECUTABLE_UPPERCASE", &executable_name_uppercase);
|
|
variables.insert("PROJECT_VERSION", PROJECT_VERSION);
|
|
|
|
let Some(out_dir_env) =
|
|
std::env::var_os("BAT_ASSETS_GEN_DIR").or_else(|| std::env::var_os("OUT_DIR"))
|
|
else {
|
|
anyhow::bail!("BAT_ASSETS_GEN_DIR or OUT_DIR should be set for build.rs");
|
|
};
|
|
let out_dir = Path::new(&out_dir_env);
|
|
|
|
fs::create_dir_all(out_dir.join("assets/manual")).unwrap();
|
|
fs::create_dir_all(out_dir.join("assets/completions")).unwrap();
|
|
|
|
render_template(
|
|
&variables,
|
|
"assets/manual/bat.1.in",
|
|
out_dir.join("assets/manual/bat.1"),
|
|
)?;
|
|
render_template(
|
|
&variables,
|
|
"assets/completions/bat.bash.in",
|
|
out_dir.join("assets/completions/bat.bash"),
|
|
)?;
|
|
render_template(
|
|
&variables,
|
|
"assets/completions/bat.fish.in",
|
|
out_dir.join("assets/completions/bat.fish"),
|
|
)?;
|
|
render_template(
|
|
&variables,
|
|
"assets/completions/_bat.ps1.in",
|
|
out_dir.join("assets/completions/_bat.ps1"),
|
|
)?;
|
|
render_template(
|
|
&variables,
|
|
"assets/completions/bat.zsh.in",
|
|
out_dir.join("assets/completions/bat.zsh"),
|
|
)?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Generates a file from a template.
|
|
#[allow(dead_code)]
|
|
fn render_template(
|
|
variables: &HashMap<&str, &str>,
|
|
in_file: &str,
|
|
out_file: impl AsRef<Path>,
|
|
) -> anyhow::Result<()> {
|
|
let mut content = fs::read_to_string(in_file)?;
|
|
|
|
for (variable_name, value) in variables {
|
|
// Replace {{variable_name}} by the value
|
|
let pattern = format!("{{{{{variable_name}}}}}", variable_name = variable_name);
|
|
content = content.replace(&pattern, value);
|
|
}
|
|
|
|
fs::write(out_file, content)?;
|
|
Ok(())
|
|
}
|