use clap::CommandFactory;
use clap_complete::generate_to;
use clap_complete::shells::Bash;
use clap_complete::shells::Fish;
use clap_complete::shells::Zsh;
use color_eyre::{Result, eyre::eyre};
use man::prelude::*;
use std::env;
use std::error::Error;
use std::fs::{self, File, OpenOptions};
use std::io::Write;
use std::path;
include!("src/cli.rs");
fn generate_man_page<P: AsRef<path::Path>>(outdir: P) -> Result<()> {
let outdir = outdir.as_ref();
let man_path = outdir.join("ogle.1");
let cmd = Cli::command();
let manpage: Manual = clap2man::Manual::try_from(&cmd)
.map_err(|e| eyre!(e))?
.into();
let manpage = manpage
.example(
Example::new()
.text("Monitor the current directory for changes")
.command("ogle ls"),
)
.example(
Example::new()
.text("Show the current time every 3 seconds")
.command("ogle -p 3 date"),
)
.example(
Example::new()
.text("Try to connect to an ssh server until it succeeds")
.command("ogle -z ssh server true"),
)
.example(
Example::new()
.text("Poor man's top using ps")
.command("ogle -- /bin/bash -c 'ps -eo %cpu,args --sort -%cpu | head'"),
)
.render();
File::create(man_path)?.write_all(manpage.as_bytes())?;
Ok(())
}
fn main() -> Result<(), Box<dyn Error>> {
color_eyre::install()?;
let mut outdir =
path::PathBuf::from(env::var_os("OUT_DIR").ok_or_else(|| eyre!("error getting OUT_DIR"))?);
fs::create_dir_all(&outdir)?;
generate_man_page(&outdir)?;
outdir.pop();
outdir.pop();
outdir.pop();
generate_man_page(&outdir)?;
let mut cmd = Cli::command();
generate_to(Bash, &mut cmd, "ogle", &outdir)?;
let path = generate_to(Fish, &mut cmd, "ogle", &outdir)?;
let mut fd = OpenOptions::new().append(true).open(path)?;
writeln!(fd, "complete -c ogle --wraps command")?;
writeln!(fd, "complete -c ogle --no-files")?;
generate_to(Zsh, &mut cmd, "ogle", &outdir)?;
Ok(())
}