WIP.

cmd/go-mockgen/main.go

@@ -50,6 +50,8 @@ func mainErr() error {
 		}
 	}
 
+	log.Printf("loading data for %d packages\n", len(importPaths))
+
 	pkgs, err := packages.Load(&packages.Config{Mode: packages.NeedName | packages.NeedImports | packages.NeedSyntax | packages.NeedTypes}, importPaths...)
 	if err != nil {
 		return fmt.Errorf("could not load packages %s (%s)", strings.Join(importPaths, ","), err.Error())

internal/mockgen/types/extract.go

@@ -25,12 +25,17 @@ func Extract(pkgs []*packages.Package, packageOptions []PackageOptions) (ifaces
 		return nil, fmt.Errorf("failed to get working directory (%s)", err.Error())
 	}
 
+	var importPaths []string
 	for _, packageOpts := range packageOptions {
-		packageTypes, err := gatherAllPackageTypes(pkgs, workingDirectory, packageOpts.ImportPaths)
-		if err != nil {
-			return nil, err
-		}
+		importPaths = append(importPaths, packageOpts.ImportPaths...)
+	}
 
+	packageTypes, err := gatherAllPackageTypes(pkgs, workingDirectory, importPaths)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, packageOpts := range packageOptions {
 		for _, name := range gatherAllPackageTypeNames(packageTypes) {
 			iface, err := extractInterface(packageTypes, name, packageOpts.Interfaces, packageOpts.Exclude)
 			if err != nil {