Merge pull request #44 from pkwarren/issue-43

Only scan files found in package

pkg/bidichk/bidichk.go

@@ -142,25 +142,28 @@ func NewAnalyzer() *analysis.Analyzer {
 }
 
 func (b bidichk) run(pass *analysis.Pass) (interface{}, error) {
-	var err error
+	readFile := pass.ReadFile
+	if readFile == nil {
+		readFile = os.ReadFile
+	}
 
-	pass.Fset.Iterate(func(f *token.File) bool {
-		if strings.HasPrefix(f.Name(), "$GOROOT") {
-			return true
+	for _, astFile := range pass.Files {
+		f := pass.Fset.File(astFile.FileStart)
+		if f == nil {
+			continue
 		}
 
-		return b.check(f.Name(), f.Pos(0), pass) == nil
-	})
-
-	return nil, err
-}
+		body, err := readFile(f.Name())
+		if err != nil {
+			return nil, err
+		}
 
-func (b bidichk) check(filename string, pos token.Pos, pass *analysis.Pass) error {
-	body, err := os.ReadFile(filename)
-	if err != nil {
-		return err
+		b.check(body, f.Pos(0), pass)
 	}
+	return nil, nil
+}
 
+func (b bidichk) check(body []byte, pos token.Pos, pass *analysis.Pass) {
 	for name, r := range b.disallowedRunes {
 		start := 0
 		for {
@@ -175,6 +178,4 @@ func (b bidichk) check(filename string, pos token.Pos, pass *analysis.Pass) erro
 			start += utf8.RuneLen(r)
 		}
 	}
-
-	return nil
 }