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
124
125
126
127
128
129
130
|
package main
// auto run protoc with the correct args
import (
"os"
"strings"
"go.wit.com/log"
"golang.org/x/text/cases"
"golang.org/x/text/language"
)
// this parses the .proto file and handles anything with `autogenpb: `
// finds autogenpb:marshal and autogenpb:unique in the .proto file
//
// adds fields to []marshal and []unique
func (pb *Files) findAutogenpb(f *File) error {
// log.Info("starting findAutogenpb() on", names["protofile"])
// read in the .proto file
data, err := os.ReadFile(f.Filename)
if err != nil {
// log.Info("open config file :", err)
return err
}
var curmsg *MsgName
// parse the proto file for message struct names
for _, line := range strings.Split(string(data), "\n") {
if strings.HasPrefix(line, "message ") {
curmsg = f.parseForMessage(line)
}
if strings.HasPrefix(line, "}") {
curmsg = nil
}
// log.Info("line:", line)
parts := strings.Fields(line)
if strings.Contains(line, "autogenpb:sort") {
if parts[0] == "repeated" {
newm := parts[1]
if curmsg == nil {
log.Info("Error: Found Sort for:", newm, "however, this struct can't be used")
// log.Info("found marshal", newm)
marshalKeys = append(marshalKeys, newm)
} else {
log.Info("Found Sort for:", newm, "in struct", curmsg.Name)
}
} else {
log.Info("Error:", line)
log.Info("Error: can not sort on non repeated fields")
}
}
if strings.Contains(line, "autogenpb:unique") {
if parts[0] == "repeated" {
// log.Info("Found Unique for:", parts)
newu := parts[1]
newu = cases.Title(language.English, cases.NoLower).String(newu)
log.Info("found unique field", newu, "in struct", curmsg.Name)
// uniqueKeys = append(uniqueKeys, newu)
} else {
log.Info("Error:", line)
log.Info("Error: can not append on non repeated fields")
}
}
}
return nil
}
// looks for mutex and marshal entries
func (f *File) parseForMessage(line string) *MsgName {
fields := strings.Fields(line)
if fields[0] != "message" {
return nil
}
msgName := fields[1]
msg := new(MsgName)
f.MsgNames = append(f.MsgNames, msg)
msg.Name = msgName
if strings.Contains(line, "`autogenpb:mutex`") {
msg.DoMutex = true
log.Info("Found Mutex for:", msg.Name)
}
if strings.Contains(line, "`autogenpb:marshal`") {
msg.DoMarshal = true
log.Info("Found Marshal for:", msg.Name)
}
return msg
}
func (pb *Files) findGlobalAutogenpb(f *File) error {
// log.Info("starting findAutogenpb() on", filename)
// read in the .proto file
data, err := os.ReadFile(f.Filename)
if err != nil {
// log.Info("open config file :", err)
return err
}
lines := strings.Split(string(data), "\n")
for _, line := range lines {
if strings.Contains(line, "autogenpb:ignoreproto") {
// ignore this protofile completely (don't make foo.pb.go)
os.Exit(0)
}
if strings.Contains(line, "autogenpb:no-marshal") {
// don't marshal anything (don't make foo.marshal.pb.go)
argv.NoMarshal = true
}
if strings.Contains(line, "autogenpb:no-sort") {
// don't sort anything (don't make foo.sort.pb.go)
argv.NoSort = true
}
if strings.Contains(line, "autogenpb:mutex") {
// try the mutex hack
argv.Mutex = true
}
if strings.Contains(line, "autogenpb:gover:") {
// todo: parse the output here
parts := strings.Split(line, "autogenpb:gover:")
log.Info("found gover:", parts[1])
argv.Mutex = true
}
}
return nil
}
|
