/
main.go
123 lines (105 loc) · 2.46 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
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package main
import (
"io/fs"
"math"
"os"
"path/filepath"
"regexp"
"strings"
"github.com/urfave/cli"
"github.com/xsean2020/deep-copy/deepcopy"
"golang.org/x/tools/go/packages"
)
var gofileReg = regexp.MustCompile(`.*\.go$`)
func main() {
app := cli.NewApp()
app.Name = "deep-copy"
app.Usage = "generator deep-copy code"
app.Flags = []cli.Flag{
cli.StringFlag{
Name: "root",
Usage: "root directory",
Value: ".",
},
cli.BoolFlag{
Name: "reverse",
Usage: "root directory",
},
cli.StringSliceFlag{
Name: "method",
Usage: "copy method",
Value: &cli.StringSlice{"copy", "deepcopy"},
},
cli.StringSliceFlag{
Name: "tags",
Usage: "build tags",
},
}
app.Action = func(c *cli.Context) error {
root := c.String("root")
methods := c.StringSlice("method")
reverse := c.Bool("reverse")
buildTagsF := c.StringSlice("tags")
// fmt.Println("root:", root, methods, reverse, buildTagsF)
maxDep := math.MaxInt32
if !reverse {
maxDep = 2
}
return filepath.WalkDir(root, func(path string, d fs.DirEntry, err error) error {
if err != nil {
return err
}
if !d.IsDir() {
return nil
}
if strings.Count(path, string(os.PathListSeparator)) >= maxDep {
return filepath.SkipAll
}
files, err := os.ReadDir(path)
if err != nil {
return err
}
if len(files) == 0 {
return nil
}
var patterns []string
for _, file := range files {
if !gofileReg.MatchString(file.Name()) {
continue
}
patterns = append(patterns, path+"/"+file.Name())
}
packages, err := load(patterns...)
if err != nil {
return err
}
if len(packages) < 1 {
return nil
}
p := packages[0]
groups := deepcopy.ParseGeneratorOptions(p)
for i, group := range groups {
if len(group) == 0 {
continue
}
out := strings.ReplaceAll(p.CompiledGoFiles[i], ".go", "_gen_deepcopy.go")
outF, err := os.Create(out)
if err != nil {
return err
}
defer outF.Close()
generator := deepcopy.NewGenerator(buildTagsF, methods)
if err := generator.Generate(outF, p, group, groups); err != nil {
return err
}
}
return nil
})
}
app.Run(os.Args)
}
func load(patterns ...string) ([]*packages.Package, error) {
return packages.Load(&packages.Config{
Mode: packages.NeedSyntax | packages.NeedName | packages.NeedCompiledGoFiles | packages.NeedTypes | packages.NeedTypesInfo | packages.NeedDeps | packages.NeedImports,
}, patterns...)
}