Remove proto (#3489)

Only used for tracking analyzer types

pkg/analyzer/analyzers/airbrake/airbrake.go

@@ -11,7 +11,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -21,7 +20,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_Airbrake }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeAirbrake }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	info, err := AnalyzePermissions(a.Cfg, credInfo["key"])

pkg/analyzer/analyzers/analyzers.go

@@ -5,21 +5,23 @@ import (
 	"encoding/json"
 	"io"
 	"net/http"
+	"sort"
 
 	"github.com/fatih/color"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
 type (
 	Analyzer interface {
-		Type() analyzerpb.AnalyzerType
+		Type() AnalyzerType
 		Analyze(ctx context.Context, credentialInfo map[string]string) (*AnalyzerResult, error)
 	}
 
+	AnalyzerType int
+
 	// AnalyzerResult is the output of analysis.
 	AnalyzerResult struct {
-		AnalyzerType       analyzerpb.AnalyzerType
+		AnalyzerType       AnalyzerType
 		Bindings           []Binding
 		UnboundedResources []Resource
 		Metadata           map[string]any
@@ -56,30 +58,80 @@ const (
 	FullAccess string = "full_access"
 )
 
-// Sorted list of all available analyzers. Used for valid sub-commands and TUI
-// selection. TODO: Change slice type to Analyzer interface when all available
-// analyzers implement it.
-var AvailableAnalyzers = []string{
-	"Airbrake",
-	"Asana",
-	"Bitbucket",
-	"GitHub",
-	"GitLab",
-	"HuggingFace",
-	"Mailchimp",
-	"Mailgun",
-	"MySQL",
-	"OpenAI",
-	"Opsgenie",
-	"Postgres",
-	"Postman",
-	"Sendgrid",
-	"Shopify",
-	"Slack",
-	"Sourcegraph",
-	"Square",
-	"Stripe",
-	"Twilio",
+const (
+	AnalyzerTypeInvalid AnalyzerType = iota
+	AnalyzerTypeAirbrake
+	AnalyzerTypeAsana
+	AnalyzerTypeBitbucket
+	AnalyzerTypeGitHub
+	AnalyzerTypeGitLab
+	AnalyzerTypeHuggingFace
+	AnalyzerTypeMailchimp
+	AnalyzerTypeMailgun
+	AnalyzerTypeMySQL
+	AnalyzerTypeOpenAI
+	AnalyzerTypeOpsgenie
+	AnalyzerTypePostgres
+	AnalyzerTypePostman
+	AnalyzerTypeSendgrid
+	AnalyzerTypeShopify
+	AnalyzerTypeSlack
+	AnalyzerTypeSourcegraph
+	AnalyzerTypeSquare
+	AnalyzerTypeStripe
+	AnalyzerTypeTwilio
+	// Add new items here with AnalyzerType prefix
+)
+
+// analyzerTypeStrings maps the enum to its string representation.
+var analyzerTypeStrings = map[AnalyzerType]string{
+	AnalyzerTypeInvalid:     "Invalid",
+	AnalyzerTypeAirbrake:    "Airbrake",
+	AnalyzerTypeAsana:       "Asana",
+	AnalyzerTypeBitbucket:   "Bitbucket",
+	AnalyzerTypeGitHub:      "GitHub",
+	AnalyzerTypeGitLab:      "GitLab",
+	AnalyzerTypeHuggingFace: "HuggingFace",
+	AnalyzerTypeMailchimp:   "Mailchimp",
+	AnalyzerTypeMailgun:     "Mailgun",
+	AnalyzerTypeMySQL:       "MySQL",
+	AnalyzerTypeOpenAI:      "OpenAI",
+	AnalyzerTypeOpsgenie:    "Opsgenie",
+	AnalyzerTypePostgres:    "Postgres",
+	AnalyzerTypePostman:     "Postman",
+	AnalyzerTypeSendgrid:    "Sendgrid",
+	AnalyzerTypeShopify:     "Shopify",
+	AnalyzerTypeSlack:       "Slack",
+	AnalyzerTypeSourcegraph: "Sourcegraph",
+	AnalyzerTypeSquare:      "Square",
+	AnalyzerTypeStripe:      "Stripe",
+	AnalyzerTypeTwilio:      "Twilio",
+	// Add new mappings here
+}
+
+// String method to get the string representation of an AnalyzerType.
+func (a AnalyzerType) String() string {
+	if str, ok := analyzerTypeStrings[a]; ok {
+		return str
+	}
+	return "Unknown"
+}
+
+// GetSortedAnalyzerTypes returns a sorted slice of AnalyzerType strings, skipping "Invalid".
+func AvailableAnalyzers() []string {
+	var analyzerStrings []string
+
+	// Iterate through the map to collect all string values except "Invalid".
+	for typ, str := range analyzerTypeStrings {
+		if typ != AnalyzerTypeInvalid {
+			analyzerStrings = append(analyzerStrings, str)
+		}
+	}
+
+	// Sort the slice alphabetically.
+	sort.Strings(analyzerStrings)
+
+	return analyzerStrings
 }
 
 type PermissionStatus struct {

pkg/analyzer/analyzers/asana/asana.go

@@ -14,7 +14,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -24,7 +23,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_Asana }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeAsana }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	key, ok := credInfo["key"]

pkg/analyzer/analyzers/bitbucket/bitbucket.go

@@ -13,7 +13,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -59,7 +58,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_Bitbucket }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeBitbucket }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	key, ok := credInfo["key"]
@@ -79,7 +78,7 @@ func secretInfoToAnalyzerResult(info *SecretInfo) *analyzers.AnalyzerResult {
 	}
 
 	result := analyzers.AnalyzerResult{
-		AnalyzerType: analyzerpb.AnalyzerType_Bitbucket,
+		AnalyzerType: analyzers.AnalyzerTypeBitbucket,
 	}
 
 	// add unbounded resources

pkg/analyzer/analyzers/github/github.go

@@ -13,7 +13,6 @@ import (
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers/github/common"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers/github/finegrained"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -23,7 +22,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_GitHub }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeGitHub }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	info, err := AnalyzePermissions(a.Cfg, credInfo["key"])

pkg/analyzer/analyzers/gitlab/gitlab.go

@@ -13,7 +13,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -23,7 +22,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_GitLab }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeGitLab }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	_, err := AnalyzePermissions(a.Cfg, credInfo["key"])

pkg/analyzer/analyzers/huggingface/huggingface.go

@@ -13,7 +13,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -29,7 +28,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_HuggingFace }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeHuggingFace }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	key, ok := credInfo["key"]
@@ -239,7 +238,7 @@ func secretInfoToAnalyzerResult(info *SecretInfo) *analyzers.AnalyzerResult {
 	}
 
 	result := analyzers.AnalyzerResult{
-		AnalyzerType: analyzerpb.AnalyzerType_HuggingFace,
+		AnalyzerType: analyzers.AnalyzerTypeHuggingFace,
 		Metadata: map[string]interface{}{
 			"username":   info.Token.Username,
 			"name":       info.Token.Name,

pkg/analyzer/analyzers/mailchimp/mailchimp.go

@@ -13,7 +13,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -25,7 +24,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_Mailchimp }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeMailchimp }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	key, ok := credInfo["key"]
@@ -45,7 +44,7 @@ func secretInfoToAnalyzerResult(info *SecretInfo) *analyzers.AnalyzerResult {
 		return nil
 	}
 	result := analyzers.AnalyzerResult{
-		AnalyzerType:       analyzerpb.AnalyzerType_Mailchimp,
+		AnalyzerType:       analyzers.AnalyzerTypeMailchimp,
 		Bindings:           make([]analyzers.Binding, 0, len(StringToPermission)),
 		UnboundedResources: make([]analyzers.Resource, 0, len(info.Domains.Domains)),
 	}

pkg/analyzer/analyzers/mailgun/mailgun.go

@@ -13,7 +13,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -23,7 +22,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_Mailgun }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeMailgun }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	key, ok := credInfo["key"]
@@ -43,7 +42,7 @@ func secretInfoToAnalyzerResult(info *DomainsJSON) *analyzers.AnalyzerResult {
 		return nil
 	}
 	result := analyzers.AnalyzerResult{
-		AnalyzerType: analyzerpb.AnalyzerType_Mailgun,
+		AnalyzerType: analyzers.AnalyzerTypeMailgun,
 		Bindings:     make([]analyzers.Binding, len(info.Items)),
 	}
 

pkg/analyzer/analyzers/mysql/mysql.go

@@ -19,7 +19,6 @@ import (
 
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -29,7 +28,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_MySQL }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeMySQL }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	uri, ok := credInfo["connection_string"]
@@ -48,7 +47,7 @@ func secretInfoToAnalyzerResult(info *SecretInfo) *analyzers.AnalyzerResult {
 		return nil
 	}
 	result := analyzers.AnalyzerResult{
-		AnalyzerType: analyzerpb.AnalyzerType_MySQL,
+		AnalyzerType: analyzers.AnalyzerTypeMySQL,
 		Metadata:     nil,
 		Bindings:     []analyzers.Binding{},
 	}

pkg/analyzer/analyzers/openai/openai.go

@@ -16,7 +16,6 @@ import (
 
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -26,7 +25,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_OpenAI }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeOpenAI }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	info, err := AnalyzePermissions(a.Cfg, credInfo["key"])

pkg/analyzer/analyzers/opsgenie/opsgenie.go

@@ -16,7 +16,6 @@ import (
 	"github.com/jedib0t/go-pretty/table"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -26,7 +25,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_Opsgenie }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypeOpsgenie }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	key, ok := credInfo["key"]
@@ -45,7 +44,7 @@ func secretInfoToAnalyzerResult(info *SecretInfo) *analyzers.AnalyzerResult {
 		return nil
 	}
 	result := analyzers.AnalyzerResult{
-		AnalyzerType:       analyzerpb.AnalyzerType_Opsgenie,
+		AnalyzerType:       analyzers.AnalyzerTypeOpsgenie,
 		Metadata:           nil,
 		Bindings:           make([]analyzers.Binding, len(info.Permissions)),
 		UnboundedResources: make([]analyzers.Resource, len(info.Users)),

pkg/analyzer/analyzers/postgres/postgres.go

@@ -16,7 +16,6 @@ import (
 
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/analyzers"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/config"
-	"github.com/trufflesecurity/trufflehog/v3/pkg/analyzer/pb/analyzerpb"
 	"github.com/trufflesecurity/trufflehog/v3/pkg/context"
 )
 
@@ -26,7 +25,7 @@ type Analyzer struct {
 	Cfg *config.Config
 }
 
-func (Analyzer) Type() analyzerpb.AnalyzerType { return analyzerpb.AnalyzerType_Postgres }
+func (Analyzer) Type() analyzers.AnalyzerType { return analyzers.AnalyzerTypePostgres }
 
 func (a Analyzer) Analyze(_ context.Context, credInfo map[string]string) (*analyzers.AnalyzerResult, error) {
 	uri, ok := credInfo["connection_string"]
@@ -46,7 +45,7 @@ func secretInfoToAnalyzerResult(info *SecretInfo) *analyzers.AnalyzerResult {
 		return nil
 	}
 	result := analyzers.AnalyzerResult{
-		AnalyzerType: analyzerpb.AnalyzerType_Postgres,
+		AnalyzerType: analyzers.AnalyzerTypePostgres,
 		Metadata:     nil,
 		Bindings:     []analyzers.Binding{},
 	}