mirror of
https://github.com/astaxie/beego.git
synced 2024-12-23 09:40:50 +00:00
422 lines
7.4 KiB
Go
422 lines
7.4 KiB
Go
package validation
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"regexp"
|
|
"time"
|
|
)
|
|
|
|
type Validator interface {
|
|
IsSatisfied(interface{}) bool
|
|
DefaultMessage() string
|
|
GetKey() string
|
|
}
|
|
|
|
type Required struct {
|
|
Key string
|
|
}
|
|
|
|
func (r Required) IsSatisfied(obj interface{}) bool {
|
|
if obj == nil {
|
|
return false
|
|
}
|
|
|
|
if str, ok := obj.(string); ok {
|
|
return len(str) > 0
|
|
}
|
|
if b, ok := obj.(bool); ok {
|
|
return b
|
|
}
|
|
if i, ok := obj.(int); ok {
|
|
return i != 0
|
|
}
|
|
if t, ok := obj.(time.Time); ok {
|
|
return !t.IsZero()
|
|
}
|
|
v := reflect.ValueOf(obj)
|
|
if v.Kind() == reflect.Slice {
|
|
return v.Len() > 0
|
|
}
|
|
return true
|
|
}
|
|
|
|
func (r Required) DefaultMessage() string {
|
|
return "Required"
|
|
}
|
|
|
|
func (r Required) GetKey() string {
|
|
return r.Key
|
|
}
|
|
|
|
type Min struct {
|
|
Min int
|
|
Key string
|
|
}
|
|
|
|
func (m Min) IsSatisfied(obj interface{}) bool {
|
|
num, ok := obj.(int)
|
|
if ok {
|
|
return num >= m.Min
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (m Min) DefaultMessage() string {
|
|
return fmt.Sprint("Minimum is ", m.Min)
|
|
}
|
|
|
|
func (m Min) GetKey() string {
|
|
return m.Key
|
|
}
|
|
|
|
type Max struct {
|
|
Max int
|
|
Key string
|
|
}
|
|
|
|
func (m Max) IsSatisfied(obj interface{}) bool {
|
|
num, ok := obj.(int)
|
|
if ok {
|
|
return num <= m.Max
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (m Max) DefaultMessage() string {
|
|
return fmt.Sprint("Maximum is ", m.Max)
|
|
}
|
|
|
|
func (m Max) GetKey() string {
|
|
return m.Key
|
|
}
|
|
|
|
// Requires an integer to be within Min, Max inclusive.
|
|
type Range struct {
|
|
Min
|
|
Max
|
|
Key string
|
|
}
|
|
|
|
func (r Range) IsSatisfied(obj interface{}) bool {
|
|
return r.Min.IsSatisfied(obj) && r.Max.IsSatisfied(obj)
|
|
}
|
|
|
|
func (r Range) DefaultMessage() string {
|
|
return fmt.Sprint("Range is ", r.Min.Min, " to ", r.Max.Max)
|
|
}
|
|
|
|
func (r Range) GetKey() string {
|
|
return r.Key
|
|
}
|
|
|
|
// Requires an array or string to be at least a given length.
|
|
type MinSize struct {
|
|
Min int
|
|
Key string
|
|
}
|
|
|
|
func (m MinSize) IsSatisfied(obj interface{}) bool {
|
|
if str, ok := obj.(string); ok {
|
|
return len(str) >= m.Min
|
|
}
|
|
v := reflect.ValueOf(obj)
|
|
if v.Kind() == reflect.Slice {
|
|
return v.Len() >= m.Min
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (m MinSize) DefaultMessage() string {
|
|
return fmt.Sprint("Minimum size is ", m.Min)
|
|
}
|
|
|
|
func (m MinSize) GetKey() string {
|
|
return m.Key
|
|
}
|
|
|
|
// Requires an array or string to be at most a given length.
|
|
type MaxSize struct {
|
|
Max int
|
|
Key string
|
|
}
|
|
|
|
func (m MaxSize) IsSatisfied(obj interface{}) bool {
|
|
if str, ok := obj.(string); ok {
|
|
return len(str) <= m.Max
|
|
}
|
|
v := reflect.ValueOf(obj)
|
|
if v.Kind() == reflect.Slice {
|
|
return v.Len() <= m.Max
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (m MaxSize) DefaultMessage() string {
|
|
return fmt.Sprint("Maximum size is ", m.Max)
|
|
}
|
|
|
|
func (m MaxSize) GetKey() string {
|
|
return m.Key
|
|
}
|
|
|
|
// Requires an array or string to be exactly a given length.
|
|
type Length struct {
|
|
N int
|
|
Key string
|
|
}
|
|
|
|
func (l Length) IsSatisfied(obj interface{}) bool {
|
|
if str, ok := obj.(string); ok {
|
|
return len(str) == l.N
|
|
}
|
|
v := reflect.ValueOf(obj)
|
|
if v.Kind() == reflect.Slice {
|
|
return v.Len() == l.N
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (l Length) DefaultMessage() string {
|
|
return fmt.Sprint("Required length is ", l.N)
|
|
}
|
|
|
|
func (l Length) GetKey() string {
|
|
return l.Key
|
|
}
|
|
|
|
type Alpha struct {
|
|
Key string
|
|
}
|
|
|
|
func (a Alpha) IsSatisfied(obj interface{}) bool {
|
|
if str, ok := obj.(string); ok {
|
|
for _, v := range str {
|
|
if ('Z' < v || v < 'A') && ('z' < v || v < 'a') {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (a Alpha) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid alpha characters")
|
|
}
|
|
|
|
func (a Alpha) GetKey() string {
|
|
return a.Key
|
|
}
|
|
|
|
type Numeric struct {
|
|
Key string
|
|
}
|
|
|
|
func (n Numeric) IsSatisfied(obj interface{}) bool {
|
|
if str, ok := obj.(string); ok {
|
|
for _, v := range str {
|
|
if '9' < v || v < '0' {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (n Numeric) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid numeric characters")
|
|
}
|
|
|
|
func (n Numeric) GetKey() string {
|
|
return n.Key
|
|
}
|
|
|
|
type AlphaNumeric struct {
|
|
Key string
|
|
}
|
|
|
|
func (a AlphaNumeric) IsSatisfied(obj interface{}) bool {
|
|
if str, ok := obj.(string); ok {
|
|
for _, v := range str {
|
|
if ('Z' < v || v < 'A') && ('z' < v || v < 'a') && ('9' < v || v < '0') {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (a AlphaNumeric) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid alpha or numeric characters")
|
|
}
|
|
|
|
func (a AlphaNumeric) GetKey() string {
|
|
return a.Key
|
|
}
|
|
|
|
// Requires a string to match a given regex.
|
|
type Match struct {
|
|
Regexp *regexp.Regexp
|
|
Key string
|
|
}
|
|
|
|
func (m Match) IsSatisfied(obj interface{}) bool {
|
|
return m.Regexp.MatchString(fmt.Sprintf("%v", obj))
|
|
}
|
|
|
|
func (m Match) DefaultMessage() string {
|
|
return fmt.Sprint("Must match ", m.Regexp)
|
|
}
|
|
|
|
func (m Match) GetKey() string {
|
|
return m.Key
|
|
}
|
|
|
|
// Requires a string to not match a given regex.
|
|
type NoMatch struct {
|
|
Match
|
|
Key string
|
|
}
|
|
|
|
func (n NoMatch) IsSatisfied(obj interface{}) bool {
|
|
return !n.Match.IsSatisfied(obj)
|
|
}
|
|
|
|
func (n NoMatch) DefaultMessage() string {
|
|
return fmt.Sprint("Must not match ", n.Regexp)
|
|
}
|
|
|
|
func (n NoMatch) GetKey() string {
|
|
return n.Key
|
|
}
|
|
|
|
var alphaDashPattern = regexp.MustCompile("[^\\d\\w-_]")
|
|
|
|
type AlphaDash struct {
|
|
NoMatch
|
|
Key string
|
|
}
|
|
|
|
func (a AlphaDash) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid alpha or numeric or dash(-_) characters")
|
|
}
|
|
|
|
func (a AlphaDash) GetKey() string {
|
|
return a.Key
|
|
}
|
|
|
|
var emailPattern = regexp.MustCompile("[\\w!#$%&'*+/=?^_`{|}~-]+(?:\\.[\\w!#$%&'*+/=?^_`{|}~-]+)*@(?:[\\w](?:[\\w-]*[\\w])?\\.)+[a-zA-Z0-9](?:[\\w-]*[\\w])?")
|
|
|
|
type Email struct {
|
|
Match
|
|
Key string
|
|
}
|
|
|
|
func (e Email) DefaultMessage() string {
|
|
return fmt.Sprint("Must be a valid email address")
|
|
}
|
|
|
|
func (e Email) GetKey() string {
|
|
return e.Key
|
|
}
|
|
|
|
var ipPattern = regexp.MustCompile("^((2[0-4]\\d|25[0-5]|[01]?\\d\\d?)\\.){3}(2[0-4]\\d|25[0-5]|[01]?\\d\\d?)$")
|
|
|
|
type IP struct {
|
|
Match
|
|
Key string
|
|
}
|
|
|
|
func (i IP) DefaultMessage() string {
|
|
return fmt.Sprint("Must be a valid ip address")
|
|
}
|
|
|
|
func (i IP) GetKey() string {
|
|
return i.Key
|
|
}
|
|
|
|
var base64Pattern = regexp.MustCompile("^(?:[A-Za-z0-99+/]{4})*(?:[A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=)?$")
|
|
|
|
type Base64 struct {
|
|
Match
|
|
Key string
|
|
}
|
|
|
|
func (b Base64) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid base64 characters")
|
|
}
|
|
|
|
func (b Base64) GetKey() string {
|
|
return b.Key
|
|
}
|
|
|
|
// just for chinese mobile phone number
|
|
var mobilePattern = regexp.MustCompile("^((\\+86)|(86))?(1(([35][0-9])|(47)|[8][01236789]))\\d{8}$")
|
|
|
|
type Mobile struct {
|
|
Match
|
|
Key string
|
|
}
|
|
|
|
func (m Mobile) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid mobile number")
|
|
}
|
|
|
|
func (m Mobile) GetKey() string {
|
|
return m.Key
|
|
}
|
|
|
|
// just for chinese telephone number
|
|
var telPattern = regexp.MustCompile("^(0\\d{2,3}(\\-)?)?\\d{7,8}$")
|
|
|
|
type Tel struct {
|
|
Match
|
|
Key string
|
|
}
|
|
|
|
func (t Tel) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid telephone number")
|
|
}
|
|
|
|
func (t Tel) GetKey() string {
|
|
return t.Key
|
|
}
|
|
|
|
// just for chinese telephone or mobile phone number
|
|
type Phone struct {
|
|
Mobile
|
|
Tel
|
|
Key string
|
|
}
|
|
|
|
func (p Phone) IsSatisfied(obj interface{}) bool {
|
|
return p.Mobile.IsSatisfied(obj) || p.Tel.IsSatisfied(obj)
|
|
}
|
|
|
|
func (p Phone) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid telephone or mobile phone number")
|
|
}
|
|
|
|
func (p Phone) GetKey() string {
|
|
return p.Key
|
|
}
|
|
|
|
// just for chinese zipcode
|
|
var zipCodePattern = regexp.MustCompile("^[1-9]\\d{5}$")
|
|
|
|
type ZipCode struct {
|
|
Match
|
|
Key string
|
|
}
|
|
|
|
func (z ZipCode) DefaultMessage() string {
|
|
return fmt.Sprint("Must be valid zipcode")
|
|
}
|
|
|
|
func (z ZipCode) GetKey() string {
|
|
return z.Key
|
|
}
|