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
|
package main
import (
"bytes"
"flag"
"fmt"
"go/parser"
"go/printer"
"go/token"
"io"
"os"
"strconv"
"strings"
"github.com/mitch000001/go-hbci/generator"
)
var segmentName string
var segmentInterface string
var segmentVersions segmentVersionsFlag
func init() {
flag.StringVar(&segmentName, "segment", "", "'MyAwesomeSegment'")
flag.StringVar(&segmentInterface, "segment_interface", "Segment", "'MyAwesomeInterface'")
flag.Var(&segmentVersions, "segment_versions", "'MyAwesomeSegmentVersion1:1,MyAwesomeSegmentVersion2:2'")
}
func main() {
flag.Parse()
if segmentName == "" {
fmt.Printf("You must provide a segment to generate the unmarshaler\n")
os.Exit(1)
}
filename := os.Getenv("GOFILE")
packageName := os.Getenv("GOPACKAGE")
fileSet := token.NewFileSet()
f, err := parser.ParseFile(fileSet, filename, nil, 0)
if err != nil {
fmt.Println(err)
}
segment := generator.SegmentIdentifier{
Name: segmentName,
InterfaceName: segmentInterface,
Versions: segmentVersions,
}
var generated io.Reader
if len(segmentVersions) != 0 {
segmentGenerator := generator.NewVersionedSegmentUnmarshaler(segment, packageName, fileSet, f)
generated, err = segmentGenerator.Generate()
} else {
segmentGenerator := generator.NewSegmentUnmarshaler(segment, packageName, fileSet, f)
generated, err = segmentGenerator.Generate()
}
if err != nil {
fmt.Printf("Error while generating Unmarshaler: %v\n", err)
os.Exit(1)
}
newFileName := strings.TrimSuffix(filename, ".go") + "_unmarshaler.go"
file, err := os.Create(newFileName)
if err != nil {
fmt.Printf("Error while creating file: %v\n", err)
os.Exit(1)
}
defer file.Close()
fileSet = token.NewFileSet()
newAstFile, err := parser.ParseFile(fileSet, newFileName, generated, parser.ParseComments)
if err != nil {
fmt.Println(err)
}
err = printer.Fprint(file, fileSet, newAstFile)
if err != nil {
fmt.Println(err)
}
}
type segmentVersionsFlag []generator.SegmentIdentifier
func (s *segmentVersionsFlag) String() string {
var buf bytes.Buffer
for _, version := range *s {
fmt.Fprintf(&buf, "%s:%d:%s", version.Name, version.Version, version.InterfaceName)
}
return buf.String()
}
func (s *segmentVersionsFlag) Set(in string) error {
unquoted, err := strconv.Unquote(in)
if err != nil {
return fmt.Errorf("Invalid input: %q (%v)", in, err)
}
segments := strings.FieldsFunc(unquoted, func(r rune) bool {
return r == ','
})
for _, seg := range segments {
parts := strings.Split(seg, ":")
if len(parts) < 2 {
return fmt.Errorf("Malformed versioned segment: %q", seg)
}
version, err := strconv.Atoi(parts[1])
if err != nil {
return fmt.Errorf("Malformed segment version: %v", err)
}
var interfaceName string
if len(parts) == 3 {
interfaceName = parts[2]
} else {
interfaceName = "Segment" // TODO: defaults?
}
*s = append(*s, generator.SegmentIdentifier{Name: parts[0], Version: version, InterfaceName: interfaceName})
}
return nil
}
|
