From ba49aa806a48839b61fb261f7ccd007a507d8d50 Mon Sep 17 00:00:00 2001
From: 53hornet <atc@53hor.net>
Date: Sat, 22 Jan 2022 08:45:24 -0500
Subject: feat: experimental cgi fork

---
 src/config.rs | 73 +++++++++++++++++++++++++----------------------------------
 1 file changed, 31 insertions(+), 42 deletions(-)

(limited to 'src/config.rs')

diff --git a/src/config.rs b/src/config.rs
index c4396d4..f7d391f 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -1,48 +1,37 @@
-use std::{net::SocketAddrV4, path::PathBuf};
-use structopt::StructOpt;
+use anyhow::{Context, Result};
+use std::{
+    collections::HashMap,
+    env,
+    fs::File,
+    io::{BufRead, BufReader},
+};
 
-lazy_static! {
-    pub static ref INSTANCE: Config = Config::from_args();
-}
-
-#[derive(Debug, StructOpt)]
-#[structopt(author, about)]
-pub struct Config {
-    #[structopt(
-        name = "DIR",
-        required = true,
-        parse(from_str),
-        help = "The path to your data directory"
-    )]
-    pub data_dir: PathBuf,
-
-    #[structopt(
-        name = "ADDR",
-        long = "bind-addr",
-        required = false,
-        default_value = "127.0.0.1:5353",
-        help = "Sets the IP address and port to bind to"
-    )]
-    pub bind_addr: SocketAddrV4,
+const DEFAULT_RCFILE: &str = "/usr/local/etc/twinhrc"; // TODO: windows users?
+const RCFILE: &str = "TWINHRC";
+const PREFIX: &str = "TWINH_";
+pub const LOGLEVEL: &str = "TWINH_LOGLEVEL";
 
-    #[structopt(
-        long,
-        short,
-        multiple = true,
-        parse(from_occurrences),
-        help = "Increases log level"
-    )]
-    pub verbose: usize,
+pub fn init() -> Result<HashMap<String, String>> {
+    let rcpath = env::var(RCFILE).unwrap_or_else(|_| DEFAULT_RCFILE.to_owned());
+    let rcfile = File::open(rcpath).with_context(|| "Failed to open rcfile")?;
 
-    #[structopt(
-        long,
-        help = "Creates a fresh data directory at DIR and exits; Fails if DIR exists"
-    )]
-    pub create_dir: bool,
+    // read rcfile into config map
+    let mut config = HashMap::new();
+    for line in BufReader::new(rcfile).lines() {
+        let line = line.with_context(|| "Failed to parse line in rcfile")?;
+        if !line.starts_with('#') {
+            if let Some((key, val)) = line.split_once('=') {
+                config.insert(key.into(), val.into());
+            }
+        }
+    }
 
-    #[structopt(long, help = "Imports CSV car data into the data directory")]
-    pub import_cars: bool,
+    // override rcfile config with any present env vars
+    for (key, val) in env::vars() {
+        if let Some(key) = key.strip_prefix(PREFIX) {
+            config.insert(key.into(), val);
+        }
+    }
 
-    #[structopt(long, help = "Imports CSV parts data into the data directory")]
-    pub import_parts: bool,
+    Ok(config)
 }
-- 
cgit v1.2.3