diff --git a/code.go b/code.go index 475b81d..d15b70a 100644 --- a/code.go +++ b/code.go @@ -4,18 +4,23 @@ import ( "os" "fmt" "time" + "flag" "bufio" "strings" "path/filepath" "text/template" ) -const ( - searchDir = "Code" -) +var searchDir string func main() { fmt.Println("Starting...") + + // Get any arguments + dirPtr := flag.String("d", "Code", "Directory to scan for each user.") + flag.Parse() + searchDir = *dirPtr + generate(findProjects()) }