Add --include-internal flag

README.md

@@ -35,3 +35,9 @@ If you want to ignore files (e.g. proto generated files), pass the `--skip-files
 ```bash
 porto --skip-files ".*\\.pb\\.go$" path/to/library
 ```
+
+If you want to check `internal` folders too
+
+```bash
+porto --check-internal path/to/library
+```

cmd/porto/main.go

@@ -16,6 +16,7 @@ func main() {
 	flagWriteOutputToFile := flag.Bool("w", false, "write result to (source) file instead of stdout")
 	flagListDiff := flag.Bool("l", false, "list files whose vanity import differs from porto's")
 	flagSkipFiles := flag.String("skip-files", "", "Regexps of files to skip")
+	flagCheckInternal := flag.Bool("check-internal", false, "check internal folders")
 	flag.Parse()
 
 	baseDir := flag.Arg(0)
@@ -24,9 +25,10 @@ func main() {
 usage: porto [options] <target-path>
 
 Options:
--w             Write result to (source) file instead of stdout (default: false)
--l             List files whose vanity import differs from porto's (default: false)
---skip-files   Regexps of files to skip
+-w                Write result to (source) file instead of stdout (default: false)
+-l                List files whose vanity import differs from porto's (default: false)
+--skip-files      Regexps of files to skip
+--check-internal  Check internal folders
 
 Examples:
 
@@ -61,6 +63,7 @@ Add import path to a folder
 		WriteResultToFile: *flagWriteOutputToFile,
 		ListDiffFiles:     *flagListDiff,
 		SkipFilesRegexes:  skipFilesRegex,
+		CheckInternal:     *flagCheckInternal,
 	})
 	if err != nil {
 		log.Fatal(err)

files.go

@@ -20,11 +20,10 @@ func isGoTestFile(filename string) bool {
 	return strings.HasSuffix(filename, "_test.go")
 }
 
-// isUnexportedDir checks if a dirname is a known unexported directory. Notice, we also
-// ignore "internal" because it is only exported inside the codebase. This situation can
-// become optional in the future.
-func isUnexportedDir(dirname string) bool {
-	return dirname == "testdata" || dirname == "internal"
+// isUnexportedDir checks if a dirname is a known unexported directory.
+// If checkInternal is false, we also ignore "internal".
+func isUnexportedDir(dirname string, checkInternal bool) bool {
+	return dirname == "testdata" || (!checkInternal && dirname == "internal")
 }
 
 // writeContentToFile writes the content in bytes to a given file.

import.go

@@ -67,7 +67,7 @@ func isUnexportedModule(moduleName string) bool {
 }
 
 func findAndAddVanityImportForModuleDir(workingDir, absDir string, moduleName string, opts Options) (int, error) {
-	if isUnexportedModule(moduleName) {
+	if !opts.CheckInternal && isUnexportedModule(moduleName) {
 		return 0, nil
 	}
 
@@ -83,7 +83,7 @@ func findAndAddVanityImportForModuleDir(workingDir, absDir string, moduleName st
 				c   int
 				err error
 			)
-			if isUnexportedDir(dirName) {
+			if isUnexportedDir(dirName, opts.CheckInternal) {
 				continue
 			} else if newModuleName, ok := findGoModule(absDir + pathSeparator + dirName); ok {
 				// if folder contains go.mod we use it from now on to build the vanity import
@@ -165,7 +165,7 @@ func findAndAddVanityImportForNonModuleDir(workingDir, absDir string, opts Optio
 		}
 
 		dirName := f.Name()
-		if isUnexportedDir(dirName) {
+		if isUnexportedDir(dirName, opts.CheckInternal) {
 			continue
 		}
 
@@ -199,6 +199,8 @@ type Options struct {
 	ListDiffFiles bool
 	// Set of regex for matching files to be skipped
 	SkipFilesRegexes []*regexp.Regexp
+	// Check internal files
+	CheckInternal bool
 }
 
 // FindAndAddVanityImportForDir scans all files in a folder and based on go.mod files
@@ -222,7 +224,7 @@ func FindAndAddVanityImportForDir(workingDir, absDir string, opts Options) (int,
 		}
 
 		dirName := f.Name()
-		if isUnexportedDir(dirName) {
+		if isUnexportedDir(dirName, opts.CheckInternal) {
 			continue
 		}