81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package sources
|
|
|
|
import (
|
|
"strings"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/circleci"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/docker"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/elasticsearch"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/filesystem"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/gcs"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/git"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/github"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/gitlab"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/huggingface"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/jenkins"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/postman"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/s3"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/sources/syslog"
|
|
)
|
|
|
|
func GetSourceNotes(sourceName string) string {
|
|
source := strings.ToLower(sourceName)
|
|
switch source {
|
|
case "github":
|
|
return github.GetNote()
|
|
case "postman":
|
|
return postman.GetNote()
|
|
case "elasticsearch":
|
|
return elasticsearch.GetNote()
|
|
case "huggingface":
|
|
return huggingface.GetNote()
|
|
case "jenkins":
|
|
return jenkins.GetNote()
|
|
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
type CmdModel interface {
|
|
tea.Model
|
|
Cmd() string
|
|
Summary() string
|
|
}
|
|
|
|
func GetSourceFields(sourceName string) CmdModel {
|
|
source := strings.ToLower(sourceName)
|
|
|
|
switch source {
|
|
case "aws s3":
|
|
return s3.GetFields()
|
|
case "circleci":
|
|
return circleci.GetFields()
|
|
case "docker":
|
|
return docker.GetFields()
|
|
case "elasticsearch":
|
|
return elasticsearch.GetFields()
|
|
case "filesystem":
|
|
return filesystem.GetFields()
|
|
case "gcs (google cloud storage)":
|
|
return gcs.GetFields()
|
|
case "git":
|
|
return git.GetFields()
|
|
case "github":
|
|
return github.GetFields()
|
|
case "gitlab":
|
|
return gitlab.GetFields()
|
|
case "hugging face":
|
|
return huggingface.GetFields()
|
|
case "jenkins":
|
|
return jenkins.GetFields()
|
|
case "postman":
|
|
return postman.GetFields()
|
|
case "syslog":
|
|
return syslog.GetFields()
|
|
}
|
|
|
|
return nil
|
|
}
|