From 97f8a4834b15a4412f4718ed21eb14279c581072 Mon Sep 17 00:00:00 2001
From: Miccah <m.castorina93@gmail.com>
Date: Tue, 13 Aug 2024 08:50:36 -0700
Subject: [PATCH] Add metrics for command invocation (#3185)

---
 main.go                | 3 ++-
 pkg/updater/updater.go | 7 +++++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 62375ebf2..0dbf1f8d1 100644
--- a/main.go
+++ b/main.go
@@ -299,7 +299,8 @@ func main() {
 	}
 
 	if !*noUpdate {
-		updateCfg.Fetcher = updater.Fetcher(usingTUI)
+		topLevelCmd, _, _ := strings.Cut(cmd, " ")
+		updateCfg.Fetcher = updater.Fetcher(topLevelCmd, usingTUI)
 	}
 	if version.BuildVersion == "dev" {
 		updateCfg.Fetcher = nil
diff --git a/pkg/updater/updater.go b/pkg/updater/updater.go
index 3e2516196..b8ad42020 100644
--- a/pkg/updater/updater.go
+++ b/pkg/updater/updater.go
@@ -19,12 +19,13 @@ import (
 	"github.com/trufflesecurity/trufflehog/v3/pkg/version"
 )
 
-func Fetcher(tui bool) fetcher.Interface {
-	return &OSS{TUI: tui}
+func Fetcher(cmd string, tui bool) fetcher.Interface {
+	return &OSS{Cmd: cmd, TUI: tui}
 }
 
 type OSS struct {
 	Interval time.Duration
+	Cmd      string
 	TUI      bool
 	Updated  bool
 }
@@ -41,6 +42,7 @@ type FormData struct {
 	OS             string
 	Arch           string
 	CurrentVersion string
+	Cmd            string
 	TUI            bool
 	Timezone       string
 	Binary         string
@@ -58,6 +60,7 @@ func (g *OSS) Fetch() (io.Reader, error) {
 		OS:             runtime.GOOS,
 		Arch:           runtime.GOARCH,
 		CurrentVersion: version.BuildVersion,
+		Cmd:            g.Cmd,
 		TUI:            g.TUI,
 		Timezone:       zone,
 		Binary:         "trufflehog",