use crate::bug_report;
use anyhow::{anyhow, Result};
use asyncgit::sync::RepoPath;
use clap::{
crate_authors, crate_description, crate_name, crate_version,
value_parser, Arg, Command as ClapApp,
};
use simplelog::{Config, LevelFilter, WriteLogger};
use std::{
env,
fs::{self, File},
path::PathBuf,
};
pub struct CliArgs {
pub theme: PathBuf,
pub repo_path: RepoPath,
pub poll_watcher: bool,
}
pub fn process_cmdline() -> Result<CliArgs> {
let app = app();
let arg_matches = app.get_matches();
if arg_matches.contains_id("bugreport") {
bug_report::generate_bugreport();
std::process::exit(0);
}
if arg_matches.contains_id("logging") {
setup_logging()?;
}
let workdir =
arg_matches.get_one::<String>("workdir").map(PathBuf::from);
let gitdir = arg_matches
.get_one::<String>("directory")
.map_or_else(|| PathBuf::from("."), PathBuf::from);
#[allow(clippy::option_if_let_else)]
let repo_path = if let Some(w) = workdir {
RepoPath::Workdir { gitdir, workdir: w }
} else {
RepoPath::Path(gitdir)
};
let arg_theme = arg_matches
.get_one::<String>("theme")
.map_or_else(|| PathBuf::from("theme.ron"), PathBuf::from);
let theme = if get_app_config_path()?.join(&arg_theme).is_file() {
get_app_config_path()?.join(arg_theme)
} else {
get_app_config_path()?.join("theme.ron")
};
let arg_poll: bool =
*arg_matches.get_one("poll").unwrap_or(&false);
Ok(CliArgs {
theme,
poll_watcher: arg_poll,
repo_path,
})
}
fn app() -> ClapApp {
ClapApp::new(crate_name!())
.author(crate_authors!())
.version(crate_version!())
.about(crate_description!())
.help_template(
"\
{before-help}gitui {version}
{author}
{about}
{usage-heading} {usage}
{all-args}{after-help}
",
)
.arg(
Arg::new("theme")
.help("Set the color theme (defaults to theme.ron)")
.short('t')
.long("theme")
.value_name("THEME")
.num_args(1),
)
.arg(
Arg::new("logging")
.help("Stores logging output into a cache directory")
.short('l')
.long("logging")
.num_args(0),
)
.arg(
Arg::new("poll")
.help("Poll folder for changes instead of using file system events. This can be useful if you run into issues with maximum # of file descriptors")
.long("polling")
.num_args(0)
.value_parser(value_parser!(bool)),
)
.arg(
Arg::new("bugreport")
.help("Generate a bug report")
.long("bugreport"),
)
.arg(
Arg::new("directory")
.help("Set the git directory")
.short('d')
.long("directory")
.env("GIT_DIR")
.num_args(1),
)
.arg(
Arg::new("workdir")
.help("Set the working directory")
.short('w')
.long("workdir")
.env("GIT_WORK_TREE")
.num_args(1),
)
}
fn setup_logging() -> Result<()> {
let mut path = get_app_cache_path()?;
path.push("gitui.log");
let _ = WriteLogger::init(
LevelFilter::Trace,
Config::default(),
File::create(path)?,
);
Ok(())
}
fn get_app_cache_path() -> Result<PathBuf> {
let mut path = dirs_next::cache_dir()
.ok_or_else(|| anyhow!("failed to find os cache dir."))?;
path.push("gitui");
fs::create_dir_all(&path)?;
Ok(path)
}
pub fn get_app_config_path() -> Result<PathBuf> {
let mut path = if cfg!(target_os = "macos") {
dirs_next::home_dir().map(|h| h.join(".config"))
} else {
dirs_next::config_dir()
}
.ok_or_else(|| anyhow!("failed to find os config dir."))?;
path.push("gitui");
fs::create_dir_all(&path)?;
Ok(path)
}
#[test]
fn verify_app() {
app().debug_assert();
}