gravel/stdsrv/src/args.rs

//! Simple and program specific command line argument parsing solution.

use std::path::PathBuf;

use crate::error::Error;

use crate::error::ErrorKind;

pub struct ProgramArgs {
    pub outdir: PathBuf,
    pub indir: PathBuf,
    pub generate: bool,
    pub force: bool,
    pub addr: String,
}

impl Default for ProgramArgs {
    fn default() -> Self {
        Self {
            indir: PathBuf::from("./web"),
            outdir: PathBuf::from("./html"),
            generate: false,
            force: false,
            addr: "0.0.0.0:8080".to_string(),
        }
    }
}

impl TryFrom<std::env::Args> for ProgramArgs {
    type Error = crate::error::Error;
    fn try_from(mut value: std::env::Args) -> Result<Self, Self::Error> {
        let mut a = Self::default();
        let _ = value.next(); // ignore executable path
        while let Some(v) = value.next() {
            match v.as_str() {
                "-i" => {
                    a.indir = value
                        .next()
                        .ok_or(Error::new(
                            ErrorKind::CommandLineArgsParse,
                            "Expected input directory after option `-i`",
                        ))?
                        .into();
                }
                "-a" => {
                    a.addr = value.next().ok_or(Error::new(
                        ErrorKind::CommandLineArgsParse,
                        "Expected listener address after option `-a`",
                    ))?;
                }
                "-g" => a.generate = true,
                "-f" => a.force = true,
                _ => {
                    a.outdir = v.into();
                }
            }
        }
        Ok(a)
    }
}