add prefix match exclude feature

doc.go

@@ -2,7 +2,7 @@
 //
 // Usage:
 //
-//  gocmt [-i] [-p] [-t "comment template"] [-d dir]
+//  gocmt [-i] [-p] [-t "comment template"] [-d dir] [-e exclude_dirs]
 //
 // This tools exists because I have to work with some existed code base, which
 // is lack of documentation for many exported identifiers. Iterating over them

helper.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"go/scanner"
 	"os"
 	"strings"
@@ -17,6 +18,14 @@ func printError(err error) {
 
 func walkFunc(path string, fi os.FileInfo, err error) error {
 	if err == nil && isGoFile(fi) {
+		// exclude path
+		for _, e := range excludeDirs {
+			if strings.HasPrefix(path, e) {
+				fmt.Fprintf(os.Stdout, "ignore %s\n", path)
+				return nil
+			}
+		}
+
 		err = processFile(path, *template, *inPlace)
 	}
 

main.go

@@ -22,13 +22,16 @@ var (
 	defaultMode         = os.FileMode(0644)
 	tralingWsRegex      = regexp.MustCompile(`(?m)[\t ]+$`)
 	newlinesRegex       = regexp.MustCompile(`(?m)\n{3,}`)
+	excludeDirs         []string
 )
 
 var (
 	inPlace      = flag.Bool("i", false, "Make in-place editing")
 	template     = flag.String("t", "...", "Comment template")
 	dir          = flag.String("d", "", "Directory to process")
 	parenComment = flag.Bool("p", false, "Add comments to all const inside the parens if true")
+	exclude      = flag.String("e", "",
+		"Path to exclude(only support prefix match), such as 'pkg/xxx/', 'pkg/xxx/,pkg/yyy/', and so on")
 )
 
 func main() {
@@ -43,6 +46,10 @@ func usage() {
 func gocmtRun() int {
 	flag.Parse()
 
+	if *exclude != "" {
+		excludeDirs = strings.Split(*exclude, ",")
+	}
+
 	if *dir != "" {
 		if err := filepath.Walk(*dir, walkFunc); err != nil {
 			printError(err)

testdata/pkg/xxx/xxx.go

@@ -0,0 +1,2 @@
+// auto generate, should ignore
+package xxx

testdata/pkg/yyy/yyy.go

@@ -0,0 +1 @@
+package yyy