exclude files too

Signed-off-by: Guillem Bonet <[email protected]>
swaggo · Jul 28, 2023 · 8ae8ca0 · 8ae8ca0
1 parent 27b27bd
commit 8ae8ca0
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/parser.go b/parser.go
@@ -1673,6 +1673,12 @@ func (parser *Parser) getAllGoFileInfo(packageDir, searchDir string) error {
 			return err
 		}
 
+		if parser.excludes != nil {
+			if _, ok := parser.excludes[relPath]; ok {
+				return nil
+			}
+		}
+
 		return parser.parseFile(filepath.ToSlash(filepath.Dir(filepath.Clean(filepath.Join(packageDir, relPath)))), path, nil, ParseAll)
 	})
 }

diff --git a/parser_test.go b/parser_test.go
@@ -3772,6 +3772,20 @@ func TestParser_Skip(t *testing.T) {
 	assert.Error(t, parser.Skip(filepath.Clean("admin/release"), &mockFS{IsDirectory: true}))
 }
 
+func TestParser_ExcludeFilesAndFolders(t *testing.T) {
+	t.Parallel()
+
+	searchDir := "testdata/conflict_name"
+
+	p := New(SetExcludedDirsAndFiles("api/api1.go,model"))
+	err := p.getAllGoFileInfo("testdata", searchDir)
+
+	assert.NoError(t, err)
+	assert.Equal(t, 4, len(p.packages.files))
+
+	assert.Error(t, p.Skip("model", &mockFS{IsDirectory: true}))
+}
+
 func TestGetFieldType(t *testing.T) {
 	t.Parallel()