From 039592414981283793e10e6406fd4835701969be Mon Sep 17 00:00:00 2001
From: Anthony Weems <amlweems@gmail.com>
Date: Mon, 28 Jan 2019 17:54:33 -0600
Subject: [PATCH] add path arg to allow search over arbitrary directories

---
 main.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index b7cfa9d..7c03ec6 100644
--- a/main.go
+++ b/main.go
@@ -6,6 +6,7 @@ import (
 	"crypto/x509"
 	"encoding/pem"
 	"errors"
+	"flag"
 	"fmt"
 	"io/ioutil"
 	"log"
@@ -131,10 +132,17 @@ func (n *Node) Print(indent int) {
 	}
 }
 
+var (
+	flagPath string
+)
+
 func main() {
+	flag.StringVar(&flagPath, "path", ".", "directory to search for certificates")
+	flag.Parse()
+
 	pool := NewCertPool()
 
-	err := filepath.Walk(".",
+	err := filepath.Walk(flagPath,
 		func(path string, info os.FileInfo, err error) error {
 			if err != nil {
 				return err