forked from block-vision/sui-go-sdk
/
validate.go
84 lines (70 loc) · 1.84 KB
/
validate.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
79
80
81
82
83
84
// Copyright (c) BlockVision, Inc. All rights reserved.
// SPDX-License-Identifier: Apache-2.0
package sui
import (
"fmt"
"log"
"reflect"
"regexp"
"sync"
"github.com/go-playground/validator/v10"
)
const ValidSuiAddressLength = 66
var validate *defaultValidator
type defaultValidator struct {
validate *validator.Validate
once sync.Once
}
func init() {
validate = &defaultValidator{
validate: validator.New(),
}
validate.registerCheckFn("checkAddress", checkAddress)
}
func (v *defaultValidator) ValidateStruct(obj interface{}) error {
if kindOfData(obj) == reflect.Struct {
if err := v.validate.Struct(obj); err != nil {
return v.handleErr(err)
}
}
return nil
}
func (v *defaultValidator) handleErr(err error) error {
validationErrs := err.(validator.ValidationErrors)
for _, e := range validationErrs {
if e.Tag() == "lte" {
err = fmt.Errorf("%v, field `%s` must be less than or equal to %s", err, e.Field(), e.Param())
} else if e.Tag() == "gte" {
err = fmt.Errorf("%v, field `%s` must be greater than or equal to %s", err, e.Field(), e.Param())
}
}
return err
}
func kindOfData(data interface{}) reflect.Kind {
value := reflect.ValueOf(data)
valueType := value.Kind()
if valueType == reflect.Ptr {
valueType = value.Elem().Kind()
}
return valueType
}
func (v *defaultValidator) registerCheckFn(tag string, fn validator.Func) {
if err := v.validate.RegisterValidation(tag, fn); err != nil {
log.Fatalf("register validation %s failed", tag)
}
}
func checkAddress(fl validator.FieldLevel) bool {
return isHex(fl.Field().String()) && len(fl.Field().String()) == ValidSuiAddressLength
}
func isHex(value string) bool {
pattern := "^(0x|0X)?[a-fA-F0-9]+$"
rex := regexp.MustCompile(pattern)
ok := rex.Match([]byte(value))
if ok {
length := len(value)
if length%2 != 0 {
return false
}
}
return true
}