Fixed config loading and default config generation

.vscode/launch.json

@@ -0,0 +1,7 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": []
+}

src/app.rs

@@ -1,6 +1,7 @@
 use ratatui::widgets::ListState;
 
 use crate::config::*;
+use crate::logger::*;
 
 pub enum AppRenderDir {
     Vertical,
@@ -37,7 +38,7 @@ impl App {
             render_direction: AppRenderDir::Vertical,
             current_state: CurrentState::Lookup,
             menu_state: MenuState::Main,
-            config: Config::from_file("test.ini".to_string()),
+            config: Config::load(),
             state: ListState::default(),
         }
     }

src/config.rs

@@ -1,5 +1,7 @@
 use configparser::ini::Ini;
 use core::fmt;
+use std::env;
+use std::fs;
 
 pub struct Config {
     pub subdomains: Vec<String>,
@@ -20,21 +22,72 @@ impl fmt::Display for Config {
 impl Config {
     pub fn from_file(file_path: String) -> Config {
         let mut config = Ini::new();
-        config.load(file_path).unwrap();
+        match config.load(file_path) {
+            Ok(_) => {
+                let subdomains = config.get("General", "subdomains").unwrap();
+                let mut subvec: Vec<String> = vec![];
 
-        let subdomains = config.get("General", "subdomains").unwrap();
+                for x in subdomains.split(',') {
-        let mut subvec: Vec<String> = vec![];
+                    subvec.push(x.to_string());
+                }
 
-        for x in subdomains.split(',') {
+                let wildcard_test = config.get("General", "wildcard");
-            subvec.push(x.to_string());
+                let datadir = config.get("General", "data_directory");
-        }
 
-        let wildcard_test = config.get("General", "wildcard_test");
+                Config {
-        let datadir = config.get("General", "data_directory");
+                    subdomains: subvec,
-        Config {
+                    wildcard_test: wildcard_test.unwrap(),
-            subdomains: subvec,
+                    data_dir: datadir.unwrap(),
-            wildcard_test: wildcard_test.unwrap(),
+                }
-            data_dir: datadir.unwrap(),
+            }
+            Err(err) => {
+                println!("{:?}", err);
+                Config::new()
+            }
         }
     }
+
+    pub fn new() -> Config {
+        let mut config = Ini::new();
+        let mut default_path = env::var_os("HOME").unwrap();
+        default_path.push("/.config/dnslookup");
+
+        // Create the directories if they don't exist
+        match fs::create_dir_all(&default_path) {
+            Ok(_) => {}
+            Err(err) => {
+                println!("Unable to create directories, got {}", err)
+            }
+        }
+
+        config.set(
+            "General",
+            "wildcard",
+            Some(String::from("dfjgnkdfjngkdfngjkd")),
+        );
+        config.set(
+            "General",
+            "subdomains",
+            Some(String::from("www,ftp,mail,files")),
+        );
+        config.set(
+            "General",
+            "data_directory",
+            Some(String::from(default_path.to_str().unwrap())),
+        );
+
+        let mut conf_path = default_path.clone();
+        conf_path.push("/config.ini");
+        println!("{:?}", conf_path);
+        config.write(&conf_path.clone()).unwrap();
+        Config::from_file(conf_path.into_string().unwrap())
+    }
+
+    pub fn load() -> Config {
+        // Default path is $HOME/.config/dnslookup/config.ini
+        let mut conf_path = env::var_os("HOME").unwrap();
+        conf_path.push("/.config/dnslookup/config.ini");
+
+        Config::from_file(conf_path.into_string().unwrap())
+    }
 }

src/fsutil.rs

@@ -37,7 +37,8 @@ impl DomainData {
             date.format("%Y-%m-%d_%H-%M-%S"),
             self.name
         );
-        let data_dir = Path::new(&config.data_dir.clone()).exists();
+        let path = format!("{}/lookups", config.data_dir);
+        let data_dir = Path::new(&path).exists();
         if data_dir != true {
             match fs::create_dir_all(format!("{}/lookups", config.data_dir)) {
                 Ok(_val) => {}

src/main.rs

@@ -39,9 +39,8 @@ fn main() -> Result<(), Box<dyn Error>> {
     let backend = CrosstermBackend::new(stdout);
     let mut terminal = Terminal::new(backend)?;
 
-    // Initialise the App and config
+    // Initialise the App
     let mut app = App::new();
-    let config = Config::from_file("test.ini".to_string());
 
     // Run the app
     let _res = run_app(&mut terminal, &mut app);