Merge pull request #41 from 0xCC00FFEE/master

Fixed the indexing issue of "args" and looking up secrets in a given repo

.gitignore

@@ -1 +1,2 @@
-config.yml
+config.yml
+git-hound

cmd/root.go

@@ -24,6 +24,7 @@ func InitializeFlags() {
 	rootCmd.PersistentFlags().StringVar(&app.GetFlags().LanguageFile, "language-file", "", "Supply your own list of languages to search (java, python).")
 	rootCmd.PersistentFlags().StringVar(&app.GetFlags().ConfigFile, "config-file", "", "Supply the path to a config file.")
 	rootCmd.PersistentFlags().IntVar(&app.GetFlags().Pages, "pages", 100, "Maximum pages to search per query")
+	rootCmd.PersistentFlags().BoolVar(&app.GetFlags().GithubRepo, "github-repo", false, "Search in a specific Github Repo only.")
 	rootCmd.PersistentFlags().BoolVar(&app.GetFlags().ResultsOnly, "results-only", false, "Only print match strings.")
 	rootCmd.PersistentFlags().BoolVar(&app.GetFlags().NoAPIKeys, "no-api-keys", false, "Don't search for generic API keys.")
 	rootCmd.PersistentFlags().BoolVar(&app.GetFlags().NoScoring, "no-scoring", false, "Don't use scoring to filter out false positives.")
@@ -109,11 +110,12 @@ var rootCmd = &cobra.Command{
 }
 
 func getScanner(args []string) *bufio.Scanner {
-	if args[0] == "searchKeyword" {
-		return bufio.NewScanner(strings.NewReader(args[1]))
-	} else {
-		return bufio.NewScanner(os.Stdin)
+	if len(args) == 2 {
+		if args[0] == "searchKeyword" {
+			return bufio.NewScanner(strings.NewReader(args[1]))
+		}
 	}
+	return bufio.NewScanner(os.Stdin)
 }
 
 // Execute command

internal/app/options.go

@@ -9,6 +9,7 @@ type Flags struct {
 	LanguageFile  string
 	ConfigFile    string
 	Pages         int
+	GithubRepo    bool
 	ResultsOnly   bool
 	NoAPIKeys     bool
 	NoScoring     bool

internal/app/search.go

@@ -80,7 +80,13 @@ func Search(query string, client *http.Client) (results []RepoSearchResult, err
 func SearchGitHub(query string, options SearchOptions, client *http.Client, results *[]RepoSearchResult, resultSet map[string]bool) (err error) {
 	// TODO: A lot of this code is shared between GitHub and Gist searches,
 	// so we should rework the logic
-	base := "https://github.com/search"
+	base := ""
+	if GetFlags().GithubRepo {
+		base = "https://github.com/" + query + "/search"
+	} else {
+		base = "https://github.com/search"
+	}
+
 	page, pages := 0, 1
 	var delay = 5
 	orders := []string{"asc"}