|
|
@@ -1,472 +0,0 @@
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-package binding
|
|
|
-
|
|
|
-import (
|
|
|
- "encoding/json"
|
|
|
- "fmt"
|
|
|
- "io"
|
|
|
- "net/http"
|
|
|
- "reflect"
|
|
|
- "regexp"
|
|
|
- "strconv"
|
|
|
- "strings"
|
|
|
- "unicode/utf8"
|
|
|
-
|
|
|
- "github.com/Unknwon/macaron"
|
|
|
- "github.com/macaron-contrib/i18n"
|
|
|
-)
|
|
|
-
|
|
|
-
|
|
|
- To the land of Middle-ware Earth:
|
|
|
-
|
|
|
- One func to rule them all,
|
|
|
- One func to find them,
|
|
|
- One func to bring them all,
|
|
|
- And in this package BIND them.
|
|
|
-*/
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func Bind(obj interface{}, ifacePtr ...interface{}) macaron.Handler {
|
|
|
- return func(ctx *macaron.Context) {
|
|
|
- contentType := ctx.Req.Header.Get("Content-Type")
|
|
|
-
|
|
|
- if strings.Contains(contentType, "form-urlencoded") {
|
|
|
- ctx.Invoke(Form(obj, ifacePtr...))
|
|
|
- } else if strings.Contains(contentType, "multipart/form-data") {
|
|
|
- ctx.Invoke(MultipartForm(obj, ifacePtr...))
|
|
|
- } else if strings.Contains(contentType, "json") {
|
|
|
- ctx.Invoke(Json(obj, ifacePtr...))
|
|
|
- } else {
|
|
|
- ctx.Invoke(Json(obj, ifacePtr...))
|
|
|
- if getErrors(ctx).Count() > 0 {
|
|
|
- ctx.Invoke(Form(obj, ifacePtr...))
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- ctx.Invoke(ErrorHandler)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func BindIgnErr(obj interface{}, ifacePtr ...interface{}) macaron.Handler {
|
|
|
- return func(ctx *macaron.Context, req *http.Request) {
|
|
|
- contentType := req.Header.Get("Content-Type")
|
|
|
-
|
|
|
- if strings.Contains(contentType, "form-urlencoded") {
|
|
|
- ctx.Invoke(Form(obj, ifacePtr...))
|
|
|
- } else if strings.Contains(contentType, "multipart/form-data") {
|
|
|
- ctx.Invoke(MultipartForm(obj, ifacePtr...))
|
|
|
- } else if strings.Contains(contentType, "json") {
|
|
|
- ctx.Invoke(Json(obj, ifacePtr...))
|
|
|
- } else {
|
|
|
- ctx.Invoke(Json(obj, ifacePtr...))
|
|
|
- if getErrors(ctx).Count() > 0 {
|
|
|
- ctx.Invoke(Form(obj, ifacePtr...))
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func Form(formStruct interface{}, ifacePtr ...interface{}) macaron.Handler {
|
|
|
- return func(ctx *macaron.Context) {
|
|
|
- ensureNotPointer(formStruct)
|
|
|
- formStruct := reflect.New(reflect.TypeOf(formStruct))
|
|
|
- errors := newErrors()
|
|
|
- parseErr := ctx.Req.ParseForm()
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
- if parseErr != nil {
|
|
|
- errors.Overall[BindingDeserializationError] = parseErr.Error()
|
|
|
- }
|
|
|
-
|
|
|
- mapForm(formStruct, ctx.Req.Form, errors)
|
|
|
-
|
|
|
- validateAndMap(formStruct, ctx, errors, ifacePtr...)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func MultipartForm(formStruct interface{}, ifacePtr ...interface{}) macaron.Handler {
|
|
|
- return func(ctx *macaron.Context) {
|
|
|
- ensureNotPointer(formStruct)
|
|
|
- formStruct := reflect.New(reflect.TypeOf(formStruct))
|
|
|
- errors := newErrors()
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
- multipartReader, err := ctx.Req.MultipartReader()
|
|
|
- if err != nil {
|
|
|
- errors.Overall[BindingDeserializationError] = err.Error()
|
|
|
- } else {
|
|
|
- form, parseErr := multipartReader.ReadForm(MaxMemory)
|
|
|
-
|
|
|
- if parseErr != nil {
|
|
|
- errors.Overall[BindingDeserializationError] = parseErr.Error()
|
|
|
- }
|
|
|
-
|
|
|
- ctx.Req.MultipartForm = form
|
|
|
- }
|
|
|
-
|
|
|
- mapForm(formStruct, ctx.Req.MultipartForm.Value, errors)
|
|
|
-
|
|
|
- validateAndMap(formStruct, ctx, errors, ifacePtr...)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func Json(jsonStruct interface{}, ifacePtr ...interface{}) macaron.Handler {
|
|
|
- return func(ctx *macaron.Context) {
|
|
|
- ensureNotPointer(jsonStruct)
|
|
|
- jsonStruct := reflect.New(reflect.TypeOf(jsonStruct))
|
|
|
- errors := newErrors()
|
|
|
-
|
|
|
- if ctx.Req.Body != nil {
|
|
|
- defer ctx.Req.Body.Close()
|
|
|
- }
|
|
|
-
|
|
|
- if err := json.NewDecoder(ctx.Req.Body).Decode(jsonStruct.Interface()); err != nil && err != io.EOF {
|
|
|
- errors.Overall[BindingDeserializationError] = err.Error()
|
|
|
- }
|
|
|
-
|
|
|
- validateAndMap(jsonStruct, ctx, errors, ifacePtr...)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func Validate(obj interface{}) macaron.Handler {
|
|
|
- return func(ctx *macaron.Context, l i18n.Locale) {
|
|
|
- errors := newErrors()
|
|
|
- validateStruct(errors, obj)
|
|
|
-
|
|
|
- if validator, ok := obj.(Validator); ok {
|
|
|
- validator.Validate(ctx, errors, l)
|
|
|
- }
|
|
|
- ctx.Map(*errors)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-var (
|
|
|
- alphaDashPattern = regexp.MustCompile("[^\\d\\w-_]")
|
|
|
- alphaDashDotPattern = regexp.MustCompile("[^\\d\\w-_\\.]")
|
|
|
- emailPattern = regexp.MustCompile("[\\w!#$%&'*+/=?^_`{|}~-]+(?:\\.[\\w!#$%&'*+/=?^_`{|}~-]+)*@(?:[\\w](?:[\\w-]*[\\w])?\\.)+[a-zA-Z0-9](?:[\\w-]*[\\w])?")
|
|
|
- urlPattern = regexp.MustCompile(`(http|https):\/\/[\w\-_]+(\.[\w\-_]+)+([\w\-\.,@?^=%&:/~\+#]*[\w\-\@?^=%&/~\+#])?`)
|
|
|
-)
|
|
|
-
|
|
|
-func validateStruct(errors *Errors, obj interface{}) {
|
|
|
- typ := reflect.TypeOf(obj)
|
|
|
- val := reflect.ValueOf(obj)
|
|
|
-
|
|
|
- if typ.Kind() == reflect.Ptr {
|
|
|
- typ = typ.Elem()
|
|
|
- val = val.Elem()
|
|
|
- }
|
|
|
-
|
|
|
- for i := 0; i < typ.NumField(); i++ {
|
|
|
- field := typ.Field(i)
|
|
|
-
|
|
|
-
|
|
|
- if field.Tag.Get("form") == "-" {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- fieldValue := val.Field(i).Interface()
|
|
|
- if field.Type.Kind() == reflect.Struct {
|
|
|
- validateStruct(errors, fieldValue)
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- zero := reflect.Zero(field.Type).Interface()
|
|
|
-
|
|
|
-
|
|
|
- for _, rule := range strings.Split(field.Tag.Get("binding"), ";") {
|
|
|
- if len(rule) == 0 {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- switch {
|
|
|
- case rule == "Required":
|
|
|
- if reflect.DeepEqual(zero, fieldValue) {
|
|
|
- errors.Fields[field.Name] = BindingRequireError
|
|
|
- break
|
|
|
- }
|
|
|
- case rule == "AlphaDash":
|
|
|
- if alphaDashPattern.MatchString(fmt.Sprintf("%v", fieldValue)) {
|
|
|
- errors.Fields[field.Name] = BindingAlphaDashError
|
|
|
- break
|
|
|
- }
|
|
|
- case rule == "AlphaDashDot":
|
|
|
- if alphaDashDotPattern.MatchString(fmt.Sprintf("%v", fieldValue)) {
|
|
|
- errors.Fields[field.Name] = BindingAlphaDashDotError
|
|
|
- break
|
|
|
- }
|
|
|
- case strings.HasPrefix(rule, "MinSize("):
|
|
|
- min, err := strconv.Atoi(rule[8 : len(rule)-1])
|
|
|
- if err != nil {
|
|
|
- errors.Overall["MinSize"] = err.Error()
|
|
|
- break
|
|
|
- }
|
|
|
- if str, ok := fieldValue.(string); ok && utf8.RuneCountInString(str) < min {
|
|
|
- errors.Fields[field.Name] = BindingMinSizeError
|
|
|
- break
|
|
|
- }
|
|
|
- v := reflect.ValueOf(fieldValue)
|
|
|
- if v.Kind() == reflect.Slice && v.Len() < min {
|
|
|
- errors.Fields[field.Name] = BindingMinSizeError
|
|
|
- break
|
|
|
- }
|
|
|
- case strings.HasPrefix(rule, "MaxSize("):
|
|
|
- max, err := strconv.Atoi(rule[8 : len(rule)-1])
|
|
|
- if err != nil {
|
|
|
- errors.Overall["MaxSize"] = err.Error()
|
|
|
- break
|
|
|
- }
|
|
|
- if str, ok := fieldValue.(string); ok && utf8.RuneCountInString(str) > max {
|
|
|
- errors.Fields[field.Name] = BindingMaxSizeError
|
|
|
- break
|
|
|
- }
|
|
|
- v := reflect.ValueOf(fieldValue)
|
|
|
- if v.Kind() == reflect.Slice && v.Len() > max {
|
|
|
- errors.Fields[field.Name] = BindingMinSizeError
|
|
|
- break
|
|
|
- }
|
|
|
- case rule == "Email":
|
|
|
- if !emailPattern.MatchString(fmt.Sprintf("%v", fieldValue)) {
|
|
|
- errors.Fields[field.Name] = BindingEmailError
|
|
|
- break
|
|
|
- }
|
|
|
- case rule == "Url":
|
|
|
- str := fmt.Sprintf("%v", fieldValue)
|
|
|
- if len(str) == 0 {
|
|
|
- continue
|
|
|
- } else if !urlPattern.MatchString(str) {
|
|
|
- errors.Fields[field.Name] = BindingUrlError
|
|
|
- break
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func mapForm(formStruct reflect.Value, form map[string][]string, errors *Errors) {
|
|
|
- typ := formStruct.Elem().Type()
|
|
|
-
|
|
|
- for i := 0; i < typ.NumField(); i++ {
|
|
|
- typeField := typ.Field(i)
|
|
|
- if inputFieldName := typeField.Tag.Get("form"); inputFieldName != "" {
|
|
|
- structField := formStruct.Elem().Field(i)
|
|
|
- if !structField.CanSet() {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- inputValue, exists := form[inputFieldName]
|
|
|
-
|
|
|
- if !exists {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- numElems := len(inputValue)
|
|
|
- if structField.Kind() == reflect.Slice && numElems > 0 {
|
|
|
- sliceOf := structField.Type().Elem().Kind()
|
|
|
- slice := reflect.MakeSlice(structField.Type(), numElems, numElems)
|
|
|
- for i := 0; i < numElems; i++ {
|
|
|
- setWithProperType(sliceOf, inputValue[i], slice.Index(i), inputFieldName, errors)
|
|
|
- }
|
|
|
- formStruct.Elem().Field(i).Set(slice)
|
|
|
- } else {
|
|
|
- setWithProperType(typeField.Type.Kind(), inputValue[0], structField, inputFieldName, errors)
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func ErrorHandler(errs Errors, resp http.ResponseWriter) {
|
|
|
- if errs.Count() > 0 {
|
|
|
- resp.Header().Set("Content-Type", "application/json; charset=utf-8")
|
|
|
- if _, ok := errs.Overall[BindingDeserializationError]; ok {
|
|
|
- resp.WriteHeader(http.StatusBadRequest)
|
|
|
- } else {
|
|
|
- resp.WriteHeader(422)
|
|
|
- }
|
|
|
- errOutput, _ := json.Marshal(errs)
|
|
|
- resp.Write(errOutput)
|
|
|
- return
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func setWithProperType(valueKind reflect.Kind, val string, structField reflect.Value, nameInTag string, errors *Errors) {
|
|
|
- switch valueKind {
|
|
|
- case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
|
|
- if val == "" {
|
|
|
- val = "0"
|
|
|
- }
|
|
|
- intVal, err := strconv.ParseInt(val, 10, 64)
|
|
|
- if err != nil {
|
|
|
- errors.Fields[nameInTag] = BindingIntegerTypeError
|
|
|
- } else {
|
|
|
- structField.SetInt(intVal)
|
|
|
- }
|
|
|
- case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
|
- if val == "" {
|
|
|
- val = "0"
|
|
|
- }
|
|
|
- uintVal, err := strconv.ParseUint(val, 10, 64)
|
|
|
- if err != nil {
|
|
|
- errors.Fields[nameInTag] = BindingIntegerTypeError
|
|
|
- } else {
|
|
|
- structField.SetUint(uintVal)
|
|
|
- }
|
|
|
- case reflect.Bool:
|
|
|
- structField.SetBool(val == "on")
|
|
|
- case reflect.Float32:
|
|
|
- if val == "" {
|
|
|
- val = "0.0"
|
|
|
- }
|
|
|
- floatVal, err := strconv.ParseFloat(val, 32)
|
|
|
- if err != nil {
|
|
|
- errors.Fields[nameInTag] = BindingFloatTypeError
|
|
|
- } else {
|
|
|
- structField.SetFloat(floatVal)
|
|
|
- }
|
|
|
- case reflect.Float64:
|
|
|
- if val == "" {
|
|
|
- val = "0.0"
|
|
|
- }
|
|
|
- floatVal, err := strconv.ParseFloat(val, 64)
|
|
|
- if err != nil {
|
|
|
- errors.Fields[nameInTag] = BindingFloatTypeError
|
|
|
- } else {
|
|
|
- structField.SetFloat(floatVal)
|
|
|
- }
|
|
|
- case reflect.String:
|
|
|
- structField.SetString(val)
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-func ensureNotPointer(obj interface{}) {
|
|
|
- if reflect.TypeOf(obj).Kind() == reflect.Ptr {
|
|
|
- panic("Pointers are not accepted as binding models")
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func validateAndMap(obj reflect.Value, ctx *macaron.Context, errors *Errors, ifacePtr ...interface{}) {
|
|
|
- ctx.Invoke(Validate(obj.Interface()))
|
|
|
- errors.Combine(getErrors(ctx))
|
|
|
- ctx.Map(*errors)
|
|
|
- ctx.Map(obj.Elem().Interface())
|
|
|
- if len(ifacePtr) > 0 {
|
|
|
- ctx.MapTo(obj.Elem().Interface(), ifacePtr[0])
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-func newErrors() *Errors {
|
|
|
- return &Errors{make(map[string]string), make(map[string]string)}
|
|
|
-}
|
|
|
-
|
|
|
-func getErrors(ctx *macaron.Context) Errors {
|
|
|
- return ctx.GetVal(reflect.TypeOf(Errors{})).Interface().(Errors)
|
|
|
-}
|
|
|
-
|
|
|
-type (
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
- Validator interface {
|
|
|
- Validate(*macaron.Context, *Errors, i18n.Locale)
|
|
|
- }
|
|
|
-)
|
|
|
-
|
|
|
-var (
|
|
|
-
|
|
|
-
|
|
|
- MaxMemory = int64(1024 * 1024 * 10)
|
|
|
-)
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-type Errors struct {
|
|
|
- Overall map[string]string `json:"overall"`
|
|
|
- Fields map[string]string `json:"fields"`
|
|
|
-}
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-func (err Errors) Count() int {
|
|
|
- return len(err.Overall) + len(err.Fields)
|
|
|
-}
|
|
|
-
|
|
|
-func (this *Errors) Combine(other Errors) {
|
|
|
- for key, val := range other.Fields {
|
|
|
- if _, exists := this.Fields[key]; !exists {
|
|
|
- this.Fields[key] = val
|
|
|
- }
|
|
|
- }
|
|
|
- for key, val := range other.Overall {
|
|
|
- if _, exists := this.Overall[key]; !exists {
|
|
|
- this.Overall[key] = val
|
|
|
- }
|
|
|
- }
|
|
|
-}
|
|
|
-
|
|
|
-const (
|
|
|
- BindingRequireError string = "Required"
|
|
|
- BindingAlphaDashError string = "AlphaDash"
|
|
|
- BindingAlphaDashDotError string = "AlphaDashDot"
|
|
|
- BindingMinSizeError string = "MinSize"
|
|
|
- BindingMaxSizeError string = "MaxSize"
|
|
|
- BindingEmailError string = "Email"
|
|
|
- BindingUrlError string = "Url"
|
|
|
- BindingDeserializationError string = "DeserializationError"
|
|
|
- BindingIntegerTypeError string = "IntegerTypeError"
|
|
|
- BindingBooleanTypeError string = "BooleanTypeError"
|
|
|
- BindingFloatTypeError string = "FloatTypeError"
|
|
|
-)
|
Unknwon