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
|
package extensions
import (
"encoding/json"
"reflect"
"strings"
"github.com/golang/protobuf/proto"
"github.com/golang/protobuf/protoc-gen-go/descriptor"
validator "github.com/mwitkow/go-proto-validators"
"github.com/pseudomuto/protoc-gen-doc/extensions"
)
func init() {
// NOTE: mwitkow/go-proto-validators uses gogo/profobuf/proto and therefore
// only registers the extension under gogo. We need to register it under
// golang/protobuf/proto with the same properties, except using the
// golang/protobuf FieldOptions descriptor.
proto.RegisterExtension(&proto.ExtensionDesc{
ExtendedType: (*descriptor.FieldOptions)(nil),
ExtensionType: validator.E_Field.ExtensionType,
Field: validator.E_Field.Field,
Name: validator.E_Field.Name,
Tag: validator.E_Field.Tag,
Filename: validator.E_Field.Filename,
})
}
// ValidatorRule represents a single validator rule from the (validator.field) method option extension.
type ValidatorRule struct {
Name string `json:"name"`
Value interface{} `json:"value"`
}
// ValidatorExtension contains the rules set by the (validator.field) method option extension.
type ValidatorExtension struct {
*validator.FieldValidator
rules []ValidatorRule // memoized so that we don't have to use reflection more than we need.
}
// MarshalJSON implements the json.Marshaler interface.
func (v ValidatorExtension) MarshalJSON() ([]byte, error) { return json.Marshal(v.Rules()) }
// Rules returns all active rules
func (v ValidatorExtension) Rules() []ValidatorRule {
if v.FieldValidator == nil {
return nil
}
if v.rules != nil {
return v.rules
}
vv := reflect.ValueOf(*v.FieldValidator)
vt := vv.Type()
for i := 0; i < vt.NumField(); i++ {
tag, ok := vt.Field(i).Tag.Lookup("protobuf")
if !ok {
continue
}
for _, opt := range strings.Split(tag, ",") {
if strings.HasPrefix(opt, "name=") {
tag = strings.TrimPrefix(opt, "name=")
break
}
}
value := vv.Field(i)
if value.IsNil() {
continue
}
value = reflect.Indirect(value)
v.rules = append(v.rules, ValidatorRule{Name: tag, Value: value.Interface()})
}
return v.rules
}
func init() {
extensions.SetTransformer("validator.field", func(payload interface{}) interface{} {
validator, ok := payload.(*validator.FieldValidator)
if !ok {
return nil
}
return ValidatorExtension{FieldValidator: validator}
})
}
|
