Hon Kwokandmcastorina
4598244167
* Use placeholder as default if field left empty and is required Co-authored-by: mcastorina <[email protected]> * Drop unused func * uncomment azure source * update wording --------- Co-authored-by: mcastorina <[email protected]>
76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package github
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/common"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/components/textinputs"
|
|
)
|
|
|
|
type githubCmdModel struct {
|
|
textinputs.Model
|
|
}
|
|
|
|
func GetNote() string {
|
|
return "Please enter an organization OR repository."
|
|
}
|
|
|
|
func GetFields() githubCmdModel {
|
|
org := textinputs.InputConfig{
|
|
Label: "Organization",
|
|
Key: "org",
|
|
Required: true,
|
|
Help: "GitHub organization to scan.",
|
|
Placeholder: "trufflesecurity",
|
|
}
|
|
|
|
repo := textinputs.InputConfig{
|
|
Label: "Repository",
|
|
Key: "repo",
|
|
Required: true,
|
|
Help: "GitHub repo to scan.",
|
|
Placeholder: "https://github.com/trufflesecurity/test_keys",
|
|
}
|
|
|
|
return githubCmdModel{textinputs.New([]textinputs.InputConfig{org, repo})}
|
|
}
|
|
|
|
// Handle default values since GitHub flags are OR operations
|
|
func (m githubCmdModel) GetSpecialInputs() map[string]textinputs.Input {
|
|
inputs := m.GetInputs()
|
|
if inputs["org"].IsDefault != inputs["repo"].IsDefault {
|
|
if inputs["org"].IsDefault {
|
|
delete(inputs, "org")
|
|
}
|
|
if inputs["repo"].IsDefault {
|
|
delete(inputs, "repo")
|
|
}
|
|
}
|
|
|
|
return inputs
|
|
}
|
|
|
|
func (m githubCmdModel) Cmd() string {
|
|
var command []string
|
|
command = append(command, "trufflehog", "github")
|
|
inputs := m.GetSpecialInputs()
|
|
|
|
if inputs["org"].Value != "" {
|
|
command = append(command, "--org="+inputs["org"].Value)
|
|
}
|
|
|
|
if inputs["repo"].Value != "" {
|
|
command = append(command, "--repo="+inputs["repo"].Value)
|
|
}
|
|
|
|
return strings.Join(command, " ")
|
|
}
|
|
|
|
func (m githubCmdModel) Summary() string {
|
|
inputs := m.GetSpecialInputs()
|
|
labels := m.GetLabels()
|
|
|
|
keys := []string{"org", "repo"}
|
|
return common.SummarizeSource(keys, inputs, labels)
|
|
}
|