From db5c57c35daea17f0c3d6867b3e0a141cdf38793 Mon Sep 17 00:00:00 2001
From: DustInDark <nextsasasa@gmail.com>
Date: Wed, 22 Jun 2022 00:13:33 +0900
Subject: [PATCH] added output saved file size #595

---
 src/afterfact.rs | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/afterfact.rs b/src/afterfact.rs
index a6edaa50..65d009fb 100644
--- a/src/afterfact.rs
+++ b/src/afterfact.rs
@@ -4,6 +4,7 @@ use crate::detections::print;
 use crate::detections::print::{AlertMessage, IS_HIDE_RECORD_ID};
 use crate::detections::utils;
 use crate::detections::utils::write_color_buffer;
+use bytesize::ByteSize;
 use chrono::{DateTime, Local, TimeZone, Utc};
 use csv::QuoteStyle;
 use hashbrown::HashMap;
@@ -17,6 +18,7 @@ use std::fs::File;
 use std::io;
 use std::io::BufWriter;
 use std::io::Write;
+use std::fs;
 use std::process;
 use termcolor::{BufferWriter, Color, ColorChoice, ColorSpec, WriteColor};
 use terminal_size::Width;
@@ -358,6 +360,17 @@ fn emit_csv<W: std::io::Write>(
         wtr.flush()?;
     }
 
+    let output_path = configs::CONFIG.read().unwrap().args.output.clone();
+    if let Some(path) = output_path {
+        if let Ok(metadata) = fs::metadata(path) {
+            println!(
+                "Saved file: {} ({})", configs::CONFIG.read().unwrap().args.output.as_ref().unwrap().display(), ByteSize::b(metadata.len()).to_string_as(false)
+            );
+            println!();
+        }
+    };
+
+
     disp_wtr_buf.clear();
     disp_wtr_buf.set_color(ColorSpec::new().set_fg(None)).ok();
     writeln!(disp_wtr_buf, "Results Summary:").ok();