forked from waltcow/goverter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
runner.go
69 lines (60 loc) · 1.93 KB
/
runner.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
package goverter
import (
"bytes"
"io/ioutil"
"os"
"path"
"github.com/polpo-studio/goverter/comments"
"github.com/polpo-studio/goverter/generator"
)
// GenerateConfig the config for generating a converter.
type GenerateConfig struct {
// PackageName is the package to use for the generated code.
PackageName string
// ScanDir is the package with golang files to scan for goverter tags.
ScanDir string
// ExtendMethods is a list of extensions to load in addition to goverter:extend statements
// declared on the interface itself.
ExtendMethods []string
// WorkingDir is the working directory (usually the location of go.mod file), can be empty.
WorkingDir string
// PackagePath is the full package where the generated code is going to be stored, can be empty.
// PackagePath is needed if goverter:extend tags use exported methods from the package where
// generated code resides, making sure goverter does not create a loop import statement from
// the generated code into its own package.
PackagePath string
}
// GenerateConverter generates converters.
func GenerateConverter(c GenerateConfig) ([]byte, error) {
mapping, err := comments.ParseDocs(comments.ParseDocsConfig{
PackagePattern: c.ScanDir,
WorkingDir: c.WorkingDir,
})
if err != nil {
return nil, err
}
file, err := generator.Generate(c.ScanDir, mapping, generator.Config{
Name: c.PackageName,
PackagePath: c.PackagePath,
ExtendMethods: c.ExtendMethods,
WorkingDir: c.WorkingDir,
})
if err != nil {
return nil, err
}
buf := &bytes.Buffer{}
err = file.Render(buf)
return buf.Bytes(), err
}
// GenerateConverterFile generates converters and writes them to a file.
func GenerateConverterFile(fileName string, c GenerateConfig) error {
file, err := GenerateConverter(c)
if err != nil {
return err
}
err = os.MkdirAll(path.Dir(fileName), os.ModePerm)
if err != nil {
return err
}
return ioutil.WriteFile(fileName, file, os.ModePerm)
}