xplr/src/cli.rs

use anyhow::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 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) {
        if arg.is_empty() {
            return;
        };

        let path = PathBuf::from(arg);
        if path.exists() {
            self.paths.push(path);
        };
    }

    /// 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()
                            .filter_map(|l| l.ok())
                        {
                            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;
                    }

                    // path
                    path => {
                        cli.read_path(path);
                    }
                }
            }
        }
        Ok(cli)
    }
}