Add metrics for command invocation (#3185)

trufflesecurity · Aug 13, 2024 · 97f8a48 · 97f8a48
1 parent 8cf1ec2
commit 97f8a48
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git 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
@@ -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",