fix and refactor

Yamato-Security · Nov 19, 2020 · 76566d3 · 76566d3
1 parent e6aa458
commit 76566d3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/src/detections/detection.rs b/src/detections/detection.rs
@@ -67,7 +67,10 @@ impl Detection {
                                 //&other.detection();
                             }
                         }
-                        Err(_) => break,
+                        Err(err) => {
+                            println!("{}", err);
+                            break;
+                        },
                     }
                 }
                 Err(e) => eprintln!("{}", e),

diff --git a/src/main.rs b/src/main.rs
@@ -7,13 +7,8 @@ use rusty_blue::detections::detection;
 use std::{fs, path::PathBuf, process};
 
 fn main() -> Result<(), DeError> {
-    let filepath: String = configs::singleton()
-        .args
-        .value_of("filepath")
-        .unwrap_or("")
-        .to_string();
-    if filepath != "" {
-        parse_file(&filepath);
+    if let Some(filepath) = configs::singleton().args.value_of("filepath") {
+        parse_file(&filepath.to_string());
     }
 
     Ok(())