/
impp.go
81 lines (64 loc) · 1.54 KB
/
impp.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
package impp
import (
"go/ast"
"log"
"os"
"strings"
"slices"
"golang.org/x/tools/go/analysis"
"golang.org/x/tools/go/analysis/passes/inspect"
"golang.org/x/tools/go/ast/inspector"
"gopkg.in/yaml.v3"
)
const doc = "impp check imported packages"
// Analyzer is ...
var Analyzer = &analysis.Analyzer{
Name: "impp",
Doc: doc,
Run: run,
Requires: []*analysis.Analyzer{
inspect.Analyzer,
},
}
type CheckConfig struct {
PackageNames []string `yaml:"packages"`
}
func run(pass *analysis.Pass) (any, error) {
fileName := "impp.yaml"
fb, err := os.ReadFile("impp.yaml")
if err != nil {
log.Fatalf("failed to open file: %s, err: %+v\n", fileName, err)
return nil, err
}
var cfg CheckConfig
if err := yaml.Unmarshal(fb, &cfg); err != nil {
log.Fatalf("failed to read yaml file.\nerr:%+v\n", err)
return nil, err
}
if len(cfg.PackageNames) == 0 {
return nil, nil
}
inspect := pass.ResultOf[inspect.Analyzer].(*inspector.Inspector)
nodeFilter := []ast.Node{
(*ast.File)(nil),
}
allowedImportedPackages := map[string]struct{}{}
inspect.Preorder(nodeFilter, func(n ast.Node) {
f, _ := n.(*ast.File)
if len(f.Imports) == 0 {
return
}
for _, imspec := range f.Imports {
pkgName := strings.Trim(imspec.Path.Value, "\"")
if _, ok := allowedImportedPackages[pkgName]; ok {
continue
}
if slices.Contains(cfg.PackageNames, pkgName) {
pass.Reportf(imspec.Pos(), "%s is not allowed to be imported", imspec.Path.Value)
continue
}
allowedImportedPackages[pkgName] = struct{}{}
}
})
return nil, nil
}