Add metrics for command invocation (#3185)

This commit is contained in:
Miccah 2024-08-13 08:50:36 -07:00 committed by GitHub
parent 8cf1ec2824
commit 97f8a4834b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 3 deletions

View file

@ -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

View file

@ -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",