first commit

stdsrv/src/args.rs

@@ -0,0 +1,58 @@
+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)
+    }
+}