1
0
mirror of https://github.com/astaxie/beego.git synced 2024-12-23 17:20:50 +00:00
Beego/validation/validators.go

422 lines
7.4 KiB
Go
Raw Normal View History

2013-07-20 17:37:24 +00:00
package validation
2013-07-19 08:49:28 +00:00
import (
"fmt"
"reflect"
"regexp"
"time"
)
type Validator interface {
IsSatisfied(interface{}) bool
DefaultMessage() string
2013-07-20 17:37:24 +00:00
GetKey() string
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
type Required struct {
Key string
}
2013-07-19 08:49:28 +00:00
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"
}
2013-07-20 17:37:24 +00:00
func (r Required) GetKey() string {
return r.Key
}
2013-07-19 08:49:28 +00:00
type Min struct {
Min int
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (m Min) IsSatisfied(obj interface{}) bool {
num, ok := obj.(int)
if ok {
return num >= m.Min
}
return false
}
func (m Min) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Minimum is ", m.Min)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (m Min) GetKey() string {
return m.Key
}
2013-07-19 08:49:28 +00:00
type Max struct {
Max int
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (m Max) IsSatisfied(obj interface{}) bool {
num, ok := obj.(int)
if ok {
return num <= m.Max
}
return false
}
func (m Max) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Maximum is ", m.Max)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (m Max) GetKey() string {
return m.Key
}
2013-07-19 08:49:28 +00:00
// Requires an integer to be within Min, Max inclusive.
type Range struct {
Min
Max
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (r Range) IsSatisfied(obj interface{}) bool {
return r.Min.IsSatisfied(obj) && r.Max.IsSatisfied(obj)
}
func (r Range) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Range is ", r.Min.Min, " to ", r.Max.Max)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (r Range) GetKey() string {
return r.Key
}
2013-07-19 08:49:28 +00:00
// Requires an array or string to be at least a given length.
type MinSize struct {
Min int
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
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 {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Minimum size is ", m.Min)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (m MinSize) GetKey() string {
return m.Key
}
2013-07-19 08:49:28 +00:00
// Requires an array or string to be at most a given length.
type MaxSize struct {
Max int
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
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 {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Maximum size is ", m.Max)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (m MaxSize) GetKey() string {
return m.Key
}
2013-07-19 08:49:28 +00:00
// Requires an array or string to be exactly a given length.
type Length struct {
2013-07-20 17:37:24 +00:00
N int
Key string
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (l Length) IsSatisfied(obj interface{}) bool {
2013-07-19 08:49:28 +00:00
if str, ok := obj.(string); ok {
2013-07-20 17:37:24 +00:00
return len(str) == l.N
2013-07-19 08:49:28 +00:00
}
v := reflect.ValueOf(obj)
if v.Kind() == reflect.Slice {
2013-07-20 17:37:24 +00:00
return v.Len() == l.N
2013-07-19 08:49:28 +00:00
}
return false
}
2013-07-20 17:37:24 +00:00
func (l Length) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Required length is ", l.N)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (l Length) GetKey() string {
return l.Key
}
type Alpha struct {
Key string
}
2013-07-19 08:49:28 +00:00
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
}
}
2013-07-21 15:46:18 +00:00
return true
2013-07-19 08:49:28 +00:00
}
return false
}
func (a Alpha) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must be valid alpha characters")
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (a Alpha) GetKey() string {
return a.Key
}
type Numeric struct {
Key string
}
2013-07-19 08:49:28 +00:00
func (n Numeric) IsSatisfied(obj interface{}) bool {
if str, ok := obj.(string); ok {
for _, v := range str {
if '9' < v || v < '0' {
return false
}
}
2013-07-21 15:46:18 +00:00
return true
2013-07-19 08:49:28 +00:00
}
return false
}
func (n Numeric) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must be valid numeric characters")
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (n Numeric) GetKey() string {
return n.Key
}
type AlphaNumeric struct {
Key string
}
2013-07-19 08:49:28 +00:00
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
}
}
2013-07-21 15:46:18 +00:00
return true
2013-07-19 08:49:28 +00:00
}
return false
}
func (a AlphaNumeric) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must be valid alpha or numeric characters")
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (a AlphaNumeric) GetKey() string {
return a.Key
}
2013-07-19 08:49:28 +00:00
// Requires a string to match a given regex.
type Match struct {
Regexp *regexp.Regexp
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (m Match) IsSatisfied(obj interface{}) bool {
2013-07-28 08:59:35 +00:00
return m.Regexp.MatchString(fmt.Sprintf("%v", obj))
2013-07-19 08:49:28 +00:00
}
func (m Match) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must match ", m.Regexp)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (m Match) GetKey() string {
return m.Key
}
2013-07-19 08:49:28 +00:00
// Requires a string to not match a given regex.
type NoMatch struct {
Match
2013-07-20 17:37:24 +00:00
Key string
}
func (n NoMatch) IsSatisfied(obj interface{}) bool {
return !n.Match.IsSatisfied(obj)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (n NoMatch) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must not match ", n.Regexp)
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (n NoMatch) GetKey() string {
return n.Key
2013-07-19 08:49:28 +00:00
}
var alphaDashPattern = regexp.MustCompile("[^\\d\\w-_]")
type AlphaDash struct {
NoMatch
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (a AlphaDash) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must be valid alpha or numeric or dash(-_) characters")
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (a AlphaDash) GetKey() string {
return a.Key
}
2013-07-19 08:49:28 +00:00
var emailPattern = regexp.MustCompile("[\\w!#$%&'*+/=?^_`{|}~-]+(?:\\.[\\w!#$%&'*+/=?^_`{|}~-]+)*@(?:[\\w](?:[\\w-]*[\\w])?\\.)+[a-zA-Z0-9](?:[\\w-]*[\\w])?")
type Email struct {
Match
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (e Email) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must be a valid email address")
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (e Email) GetKey() string {
return e.Key
}
2013-07-21 15:46:18 +00:00
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?)$")
2013-07-19 08:49:28 +00:00
type IP struct {
Match
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (i IP) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must be a valid ip address")
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (i IP) GetKey() string {
return i.Key
}
2013-07-19 08:49:28 +00:00
var base64Pattern = regexp.MustCompile("^(?:[A-Za-z0-99+/]{4})*(?:[A-Za-z0-9+/]{2}==|[A-Za-z0-9+/]{3}=)?$")
type Base64 struct {
Match
2013-07-20 17:37:24 +00:00
Key string
2013-07-19 08:49:28 +00:00
}
func (b Base64) DefaultMessage() string {
2013-07-23 17:20:24 +00:00
return fmt.Sprint("Must be valid base64 characters")
2013-07-19 08:49:28 +00:00
}
2013-07-20 17:37:24 +00:00
func (b Base64) GetKey() string {
return b.Key
}
2013-07-27 12:40:15 +00:00
// 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
}