Merge pull request #2288 from domwhewell-sage/shhhh_trufflehog

Change trufflehog output messages

Author: TheTechromancer

bbot/modules/trufflehog.py

@@ -1,4 +1,5 @@
 import json
+from functools import partial
 from bbot.modules.base import BaseModule
 
 
@@ -174,7 +175,7 @@ async def execute_trufflehog(self, module, path=None, string=None):
             command.append("--delete-cached-data")
             command.append("--token=" + self.github_token)
 
-        stats_file = self.helpers.tempfile_tail(callback=self.log_trufflehog_status)
+        stats_file = self.helpers.tempfile_tail(callback=partial(self.log_trufflehog_status, path))
         try:
             with open(stats_file, "w") as stats_fh:
                 async for line in self.helpers.run_live(command, stderr=stats_fh):
@@ -200,7 +201,7 @@ async def execute_trufflehog(self, module, path=None, string=None):
         finally:
             stats_file.unlink()
 
-    def log_trufflehog_status(self, line):
+    def log_trufflehog_status(self, path, line):
         try:
             line = json.loads(line)
         except Exception:
@@ -209,4 +210,5 @@ def log_trufflehog_status(self, line):
         message = line.get("msg", "")
         ts = line.get("ts", "")
         status = f"Message: {message} | Timestamp: {ts}"
-        self.info(status)
+        self.verbose(f"Current scan target: {path}")
+        self.verbose(status)