add cli opts

1 files changed, 35 insertions, 0 deletions

diff --git a/src/cli.rs b/src/cli.rs
new file mode 100644
index 0000000..009397d
--- /dev/null
+++ b/src/cli.rs
@@ -0,0 +1,35 @@
+use anyhow::Result;
+use lazy_static::lazy_static;
+
+use std::default::Default;
+use std::path::PathBuf;
+
+pub struct Config {
+    pub port: u16,
+    pub db_path: PathBuf,
+}
+
+impl Default for Config {
+    fn default() -> Self {
+        Config {
+            port: 3000,
+            db_path: "./urls.db_3".into(),
+        }
+    }
+}
+
+lazy_static! {
+    pub static ref CONFIG: Config = parse_args().unwrap_or(Default::default());
+}
+
+fn parse_args() -> Result<Config> {
+    let mut _a = pico_args::Arguments::from_env();
+    return Ok(Config {
+        port: _a
+            .opt_value_from_fn("--port", str::parse::<u16>)?
+            .unwrap_or(7878),
+        db_path: _a
+            .opt_value_from_str("--database")?
+            .unwrap_or(PathBuf::from("./urls.db_3")),
+    });
+}