forked from istio/tools
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
78 lines (70 loc) · 2.63 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
// Copyright 2019 Istio Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package main
import (
"google.golang.org/protobuf/compiler/protogen"
"google.golang.org/protobuf/types/descriptorpb"
)
func main() {
protogen.Options{}.Run(func(gen *protogen.Plugin) error {
for _, f := range gen.Files {
if !f.Generate {
continue
}
generateFile(gen, f)
}
return nil
})
}
func generateFile(gen *protogen.Plugin, file *protogen.File) {
filename := file.GeneratedFilenamePrefix + "_deepcopy.gen.go"
p := gen.NewGeneratedFile(filename, file.GoImportPath)
protoIdent := protogen.GoIdent{
GoName: "Clone",
GoImportPath: "google.golang.org/protobuf/proto",
}
p.P("// Code generated by protoc-gen-deepcopy. DO NOT EDIT.")
p.P("package ", file.GoPackageName)
var process func([]*protogen.Message)
process = func(messages []*protogen.Message) {
for _, message := range messages {
// skip maps in protos.
if message.Desc.Options().(*descriptorpb.MessageOptions).GetMapEntry() {
continue
}
typeName := message.GoIdent.GoName
// Generate DeepCopyInto() method for this type
p.P(`// DeepCopyInto supports using `, typeName, ` within kubernetes types, where deepcopy-gen is used.`)
p.P(`func (in *`, typeName, `) DeepCopyInto(out *`, typeName, `) {`)
p.P(`p := `, protoIdent, `(in).(*`, typeName, `)`)
p.P(`*out = *p`)
p.P(`}`)
// Generate DeepCopy() method for this type
p.P(`// DeepCopy is an autogenerated deepcopy function, copying the receiver, creating a new `, typeName, `. Required by controller-gen.`)
p.P(`func (in *`, typeName, `) DeepCopy() *`, typeName, ` {`)
p.P(`if in == nil { return nil }`)
p.P(`out := new(`, typeName, `)`)
p.P(`in.DeepCopyInto(out)`)
p.P(`return out`)
p.P(`}`)
// Generate DeepCopyInterface() method for this type
p.P(`// DeepCopyInterface is an autogenerated deepcopy function, copying the receiver, creating a new `, typeName, `. Required by controller-gen.`)
p.P(`func (in *`, typeName, `) DeepCopyInterface() interface{} {`)
p.P(`return in.DeepCopy()`)
p.P(`}`)
process(message.Messages)
}
}
process(file.Messages)
}