Remove logs_dir when the scan is finished.

Fixes #11
Edu4rdSHL · Aug 24, 2023 · 7c6e717 · 7c6e717
1 parent 36419c3
commit 7c6e717
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/resolver_engine.rs b/src/resolver_engine.rs
@@ -251,6 +251,11 @@ fn parallel_resolver_engine(
         })
         .collect();
 
+    // Delete the args.logs_dir directory if it's empty
+    if args.no_keep_nmap_logs && std::fs::remove_dir(&args.logs_dir).is_err() {
+        error!("Error removing directory {}.", &args.logs_dir)
+    }
+
     resolv_data
         .iter()
         .map(|(target, resolv_data)| {