From 258ded8e781248f22fc251920de451a80e58f037 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Sun, 2 Aug 2020 01:37:01 +0800
Subject: [PATCH] feat: add count outputs

---
 src/main.rs | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index e9aa915..8464110 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -82,6 +82,8 @@ fn main() -> std::io::Result<()> {
             }
         }
 
+        let total_count = tokens.len();
+
         if let Some(members) = matches.values_of("member") {
             let members = members.collect::<HashSet<&str>>();
             tokens = tokens.into_iter()
@@ -113,10 +115,18 @@ fn main() -> std::io::Result<()> {
             tokens = tokens.into_iter().take(head_count).collect::<Vec<Token>>();
         }
 
-        if matches.is_present("inline") {
-            tokens.iter().for_each(|i| i.inline());
+        if tokens.is_empty() {
+            println!("{}", "Cool, cannot find any TODO in this project".green().bold())
         } else {
-            tokens.iter().for_each(|i| println!("{}", i));
+            if matches.is_present("inline") {
+                tokens.iter().for_each(|i| i.inline());
+            } else {
+                tokens.iter().for_each(|i| println!("{}", i));
+            }
+            println!("Found total {} TODOs, {} displayed",
+                format!(" {} ", total_count.to_string()).bold().underline(),
+                format!(" {} ", tokens.len().to_string()).bold().green().underline(),
+            );
         }
 
         Ok(())