added ignoring interfaces via -exclude flag (#68)

mockgen/internal/tests/exclude/interfaces.go

@@ -0,0 +1,15 @@
+package exclude
+
+//go:generate mockgen -source=interfaces.go -destination=mock.go -package=ignore -exclude=IgnoreMe,IgnoreMe2
+
+type IgnoreMe interface {
+	A() bool
+}
+
+type IgnoreMe2 interface {
+	~int
+}
+
+type GenerateMockForMe interface {
+	B() int
+}

mockgen/mockgen.go

@@ -65,6 +65,7 @@ var (
 	typed                  = flag.Bool("typed", false, "Generate Type-safe 'Return', 'Do', 'DoAndReturn' function")
 	imports                = flag.String("imports", "", "(source mode) Comma-separated name=path pairs of explicit imports to use.")
 	auxFiles               = flag.String("aux_files", "", "(source mode) Comma-separated pkg=path pairs of auxiliary Go source files.")
+	exclude                = flag.String("exclude", "", "Comma-separated names of interfaces to be excluded")
 
 	debugParser = flag.Bool("debug_parser", false, "Print out parser results only.")
 	showVersion = flag.Bool("version", false, "Print version.")
@@ -200,6 +201,20 @@ func parseMockNames(names string) map[string]string {
 	return mocksMap
 }
 
+func parseExclude(names string) map[string]struct{} {
+	splitNames := strings.Split(names, ",")
+	if len(splitNames) == 0 {
+		return nil
+	}
+
+	namesSet := make(map[string]struct{}, len(splitNames))
+	for _, name := range splitNames {
+		namesSet[name] = struct{}{}
+	}
+
+	return namesSet
+}
+
 func usage() {
 	_, _ = io.WriteString(os.Stderr, usageText)
 	flag.PrintDefaults()

mockgen/parse.go

@@ -75,6 +75,10 @@ func sourceMode(source string) (*model.Package, error) {
 		}
 	}
 
+	if *exclude != "" {
+		p.excludeNamesSet = parseExclude(*exclude)
+	}
+
 	// Handle -aux_files.
 	if err := p.parseAuxFiles(*auxFiles); err != nil {
 		return nil, err
@@ -163,6 +167,7 @@ type fileParser struct {
 	auxFiles           []*ast.File
 	auxInterfaces      *interfaceCache
 	srcDir             string
+	excludeNamesSet    map[string]struct{}
 }
 
 func (p *fileParser) errorf(pos token.Pos, format string, args ...any) error {
@@ -223,6 +228,9 @@ func (p *fileParser) parseFile(importPath string, file *ast.File) (*model.Packag
 
 	var is []*model.Interface
 	for ni := range iterInterfaces(file) {
+		if _, ok := p.excludeNamesSet[ni.name.String()]; ok {
+			continue
+		}
 		i, err := p.parseInterface(ni.name.String(), importPath, ni)
 		if err != nil {
 			return nil, err