/
main.go
90 lines (71 loc) · 2.11 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package main
import (
"fmt"
"log"
"strings"
"github.com/derision-test/go-mockgen/v2/internal/mockgen/generation"
"github.com/derision-test/go-mockgen/v2/internal/mockgen/types"
"golang.org/x/tools/go/packages"
)
func init() {
log.SetFlags(0)
log.SetPrefix("go-mockgen: ")
}
func main() {
if err := mainErr(); err != nil {
message := fmt.Sprintf("error: %s\n", err.Error())
if solvableError, ok := err.(solvableError); ok {
message += "\nPossible solutions:\n"
for _, hint := range solvableError.Solutions() {
message += fmt.Sprintf(" - %s\n", hint)
}
message += "\n"
}
log.Fatalf(message)
}
}
type solvableError interface {
Solutions() []string
}
func mainErr() error {
allOptions, err := parseAndValidateOptions()
if err != nil {
return err
}
var importPaths []string
for _, opts := range allOptions {
for _, packageOpts := range opts.PackageOptions {
importPaths = append(importPaths, packageOpts.ImportPaths...)
}
}
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 | packages.NeedDeps}, importPaths...)
if err != nil {
return fmt.Errorf("could not load packages %s (%s)", strings.Join(importPaths, ","), err.Error())
}
for _, opts := range allOptions {
typePackageOpts := make([]types.PackageOptions, 0, len(opts.PackageOptions))
for _, packageOpts := range opts.PackageOptions {
typePackageOpts = append(typePackageOpts, types.PackageOptions(packageOpts))
}
ifaces, err := types.Extract(pkgs, typePackageOpts)
if err != nil {
return err
}
nameMap := make(map[string]struct{}, len(ifaces))
for _, t := range ifaces {
nameMap[strings.ToLower(t.Name)] = struct{}{}
}
for _, packageOpts := range opts.PackageOptions {
for _, name := range packageOpts.Interfaces {
if _, ok := nameMap[strings.ToLower(name)]; !ok {
return fmt.Errorf("type '%s' not found in supplied import paths", name)
}
}
}
if err := generation.Generate(ifaces, opts); err != nil {
return err
}
}
return nil
}