gitui/src/args.rs

use crate::bug_report;
use anyhow::{anyhow, Result};
use asyncgit::sync::RepoPath;
use clap::{
	crate_authors, crate_description, crate_name, crate_version,
	App as ClapApp, Arg,
};
use simplelog::{Config, LevelFilter, WriteLogger};
use std::{
	env,
	fs::{self, File},
	path::PathBuf,
};

pub struct CliArgs {
	pub theme: PathBuf,
	pub repo_path: RepoPath,
}

pub fn process_cmdline() -> Result<CliArgs> {
	let app = app();

	let arg_matches = app.get_matches();
	if arg_matches.is_present("bugreport") {
		bug_report::generate_bugreport();
		std::process::exit(0);
	}
	if arg_matches.is_present("logging") {
		setup_logging()?;
	}

	let workdir = arg_matches.value_of("workdir").map(PathBuf::from);
	let gitdir = arg_matches
		.value_of("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.value_of("theme").unwrap_or("theme.ron");

	if get_app_config_path()?.join(arg_theme).is_file() {
		Ok(CliArgs {
			theme: get_app_config_path()?.join(arg_theme),
			repo_path,
		})
	} else {
		Ok(CliArgs {
			theme: get_app_config_path()?.join("theme.ron"),
			repo_path,
		})
	}
}

fn app() -> ClapApp<'static> {
	let app = ClapApp::new(crate_name!())
		.author(crate_authors!())
		.version(crate_version!())
		.about(crate_description!())
		.arg(
			Arg::with_name("theme")
				.help("Set the color theme (defaults to theme.ron)")
				.short('t')
				.long("theme")
				.value_name("THEME")
				.takes_value(true),
		)
		.arg(
			Arg::with_name("logging")
				.help("Stores logging output into a cache directory")
				.short('l')
				.long("logging"),
		)
		.arg(
			Arg::with_name("bugreport")
				.help("Generate a bug report")
				.long("bugreport"),
		)
		.arg(
			Arg::with_name("directory")
				.help("Set the git directory")
				.short('d')
				.long("directory")
				.env("GIT_DIR")
				.takes_value(true),
		)
		.arg(
			Arg::with_name("workdir")
				.help("Set the working directory")
				.short('w')
				.long("workdir")
				.env("GIT_WORK_TREE")
				.takes_value(true),
		);
	app
}

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();
}