|
|
|
use anyhow::{bail, Result};
|
|
|
|
use std::collections::VecDeque;
|
|
|
|
use std::env;
|
|
|
|
use std::io::{BufRead, BufReader};
|
|
|
|
use std::path::PathBuf;
|
|
|
|
|
|
|
|
use crate::app;
|
|
|
|
|
|
|
|
/// The arguments to pass
|
|
|
|
#[derive(Debug, Clone, Default)]
|
|
|
|
pub struct Cli {
|
|
|
|
pub version: bool,
|
|
|
|
pub help: bool,
|
|
|
|
pub read_only: bool,
|
|
|
|
pub force_focus: bool,
|
|
|
|
pub print_pwd_as_result: bool,
|
|
|
|
pub config: Option<PathBuf>,
|
|
|
|
pub extra_config: Vec<PathBuf>,
|
|
|
|
pub on_load: Vec<app::ExternalMsg>,
|
|
|
|
pub paths: Vec<PathBuf>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Cli {
|
|
|
|
fn read_path(&mut self, arg: &str) -> Result<()> {
|
|
|
|
if arg.is_empty() {
|
|
|
|
bail!("empty string passed")
|
|
|
|
};
|
|
|
|
|
|
|
|
let path = PathBuf::from(arg);
|
|
|
|
if path.exists() {
|
|
|
|
self.paths.push(path);
|
|
|
|
Ok(())
|
|
|
|
} else {
|
|
|
|
bail!("path doesn't exist: {}", path.to_string_lossy().to_string())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Parse arguments from the command-line
|
|
|
|
pub fn parse(args: env::Args) -> Result<Self> {
|
|
|
|
let mut args: VecDeque<String> = args.skip(1).collect();
|
|
|
|
let mut cli = Self::default();
|
|
|
|
|
|
|
|
let mut flag_ends = false;
|
|
|
|
|
|
|
|
while let Some(arg) = args.pop_front() {
|
|
|
|
if flag_ends {
|
|
|
|
cli.read_path(&arg)?;
|
|
|
|
} else {
|
|
|
|
match arg.as_str() {
|
|
|
|
// Flags
|
|
|
|
"-" => {
|
|
|
|
for path in BufReader::new(std::io::stdin()).lines() {
|
|
|
|
cli.read_path(&path?)?;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
"-h" | "--help" => {
|
|
|
|
cli.help = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
"-V" | "--version" => {
|
|
|
|
cli.version = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
"--" => {
|
|
|
|
flag_ends = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Options
|
|
|
|
"-c" | "--config" => {
|
|
|
|
cli.config = args.pop_front().map(PathBuf::from)
|
|
|
|
}
|
|
|
|
|
|
|
|
"-C" | "--extra-config" => {
|
|
|
|
while let Some(path) = args.pop_front() {
|
|
|
|
if path.starts_with('-') {
|
|
|
|
args.push_front(path);
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
cli.extra_config.push(PathBuf::from(path));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
"--read-only" => cli.read_only = true,
|
|
|
|
|
|
|
|
"--on-load" => {
|
|
|
|
while let Some(msg) = args.pop_front() {
|
|
|
|
if msg.starts_with('-') {
|
|
|
|
args.push_front(msg);
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
cli.on_load.push(serde_yaml::from_str(&msg)?);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
"--force-focus" => {
|
|
|
|
cli.force_focus = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
"--print-pwd-as-result" => {
|
|
|
|
cli.print_pwd_as_result = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// path
|
|
|
|
path => {
|
|
|
|
cli.read_path(path)?;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Ok(cli)
|
|
|
|
}
|
|
|
|
}
|