Merge pull request #7 from sho-hata/feature/ignore-option

Support ignore specified directories with cli option -i/-ignore

README.md

@@ -121,7 +121,7 @@ func SampleTest(t *testing.T) {
 
 - Support when called `t.Setenv()` in test
 - Support when called `time.Now()` in test
-- Ignore specified directories with cli option -i/-ignore.
+- [x] Ignore specified directories with cli option -i/-ignore.
 - Able to ignore main/sub test function by tparagen:ignore comment.
 
 ## Synopsis
@@ -130,8 +130,19 @@ $ tparagen
 ```
 
 ## Options
-wip
+```
+$ tparagen -h
+tparagen inserts `testing.T.Parallel()` in a test function in a specific source file or in an entire directory.
+
+Usage of tparagen:
+  -i string
+        ignore directory names. ex: foo,bar,baz
+        (testdata directory is always ignored.)
+  -ignore string
+        ignore directory names. ex: foo,bar,baz
+        (testdata directory is always ignored.)
 
+```
 ## Installation
 ```
 go install -v github.com/sho-hata/tparagen/cmd/tparagen@latest

tparagen.go

@@ -43,15 +43,22 @@ func fill(args []string, outStream, errStream io.Writer) (*tparagen, error) {
 		flags.PrintDefaults()
 	}
 
-	var destDir string
+	var ignoreDirsString string
+	idesc := "ignore directory names. ex: foo,bar,baz\n(testdata directory is always ignored.)"
+	flags.StringVar(&ignoreDirsString, "ignore", "", idesc)
+	flags.StringVar(&ignoreDirsString, "i", "", idesc)
 
-	odesc := "destination directory."
-	flags.StringVar(&destDir, "destination", "", odesc)
+	var destDir string
 
 	if err := flags.Parse(args[1:]); err != nil {
 		return nil, err
 	}
 
+	ignoreDirs := []string{"testdata"}
+	if len(ignoreDirs) != 0 {
+		ignoreDirs = append(ignoreDirs, strings.Split(ignoreDirsString, ",")...)
+	}
+
 	targetDir := "./"
 
 	nargs := flags.Args()
@@ -64,16 +71,18 @@ func fill(args []string, outStream, errStream io.Writer) (*tparagen, error) {
 	}
 
 	return &tparagen{
-		in:        targetDir,
-		dest:      destDir,
-		outStream: outStream,
-		errStream: errStream,
+		in:         targetDir,
+		dest:       destDir,
+		outStream:  outStream,
+		errStream:  errStream,
+		ignoreDirs: ignoreDirs,
 	}, nil
 }
 
 type tparagen struct {
 	in, dest             string
 	outStream, errStream io.Writer
+	ignoreDirs           []string
 	errFlag              bool
 }
 
@@ -83,6 +92,10 @@ func (t *tparagen) run() error {
 			return err
 		}
 
+		if info.IsDir() && t.skipDir(path) {
+			return filepath.SkipDir
+		}
+
 		if info.IsDir() {
 			return nil
 		}
@@ -161,3 +174,13 @@ func (t *tparagen) writeOtherPath(in, dist, path string, got []byte) error {
 
 	return nil
 }
+
+func (t *tparagen) skipDir(p string) bool {
+	for _, dir := range t.ignoreDirs {
+		if filepath.Base(p) == dir {
+			return true
+		}
+	}
+
+	return false
+}