gitleaks/options.go

package main

import (
	"fmt"
	"os"
	"strconv"
)

const usage = `usage: gitleaks [options] <url>

Options:
 -c --concurrency 	Concurrency factor (default is 10)
 -u --user 			Git user url
 -r --repo 			Git repo url
 -o --org 			Git organization url
 -s --since 		Commit to stop at
 -b --b64Entropy 	Base64 entropy cutoff (default is 70)
 -x --hexEntropy  	Hex entropy cutoff (default is 40)
 -e --entropy		Enable entropy		
 -h --help 		Display this message
 --strict 		Enables stopwords
`

// Options for gitleaks
type Options struct {
	Concurrency      int
	B64EntropyCutoff int
	HexEntropyCutoff int
	UserURL          string
	OrgURL           string
	RepoURL          string
	Strict           bool
	Entropy          bool
	SinceCommit      string
	Persist          bool
	IncludeForks     bool
	Tmp              bool
	EnableJSON       bool
	Token            string
	Verbose 		 bool
}

// help prints the usage string and exits
func help() {
	os.Stderr.WriteString(usage)
	os.Exit(1)
}

// optionsNextInt is a parseOptions helper that returns the value (int) of an option if valid
func optionsNextInt(args []string, i *int) int {
	if len(args) > *i+1 {
		*i++
	} else {
		help()
	}
	argInt, err := strconv.Atoi(args[*i])
	if err != nil {
		fmt.Printf("Invalid %s option: %s\n", args[*i-1], args[*i])
		help()
	}
	return argInt
}

// optionsNextString is a parseOptions helper that returns the value (string) of an option if valid
func optionsNextString(args []string, i *int) string {
	if len(args) > *i+1 {
		*i++
	} else {
		fmt.Printf("Invalid %s option: %s\n", args[*i-1], args[*i])
		help()
	}
	return args[*i]
}

// parseOptions
func parseOptions(args []string) *Options {
	opts := &Options{
		Concurrency:      10,
		B64EntropyCutoff: 70,
		HexEntropyCutoff: 40,
	}

	if len(args) == 0 {
		help()
	}

	for i := 0; i < len(args); i++ {
		arg := args[i]
		switch arg {
		case "-s", "--since":
			opts.SinceCommit = optionsNextString(args, &i)
		case "--strict":
			opts.Strict = true
		case "-b", "--b64Entropy":
			opts.B64EntropyCutoff = optionsNextInt(args, &i)
		case "-x", "--hexEntropy":
			opts.HexEntropyCutoff = optionsNextInt(args, &i)
		case "-e", "--entropy":
			opts.Entropy = true
		case "-c", "--concurrency":
			opts.Concurrency = optionsNextInt(args, &i)
		case "-o", "--org":
			opts.OrgURL = optionsNextString(args, &i)
		case "-u", "--user":
			opts.UserURL = optionsNextString(args, &i)
		case "-p", "--persist":
			opts.UserURL = optionsNextString(args, &i)
		case "-r", "--repo":
			opts.RepoURL = optionsNextString(args, &i)
		case "-f", "--forks":
			opts.IncludeForks = true
		case "-t", "--temporary":
			opts.Tmp = true
		case "-gt", "--token":
			opts.Token = optionsNextString(args, &i)
		case "-j", "--json":
			opts.EnableJSON = true
		case "-v", "--verbose":
			opts.Verbose = true
		case "-h", "--help":
			help()
			return nil
		default:
			if i == len(args)-1 && opts.OrgURL == "" && opts.RepoURL == "" &&
				opts.UserURL == "" {
				opts.RepoURL = arg
			} else {
				fmt.Printf("Uknown option %s\n\n", arg)
				help()
			}
		}
	}

	return opts
}