nyaa-pantsu/util/modelHelper/modelHelper.go

package modelHelper

import (
	"reflect"
	"net/http"
	"github.com/ewhal/nyaa/util/log"
	"strconv"
	"fmt"
)

func IsZeroOfUnderlyingType(x interface{}) bool {
	return x == reflect.Zero(reflect.TypeOf(x)).Interface()
}

// AssignValue assign form values to model.
func AssignValue(model interface{}, form interface{}) {
	modelIndirect := reflect.Indirect(reflect.ValueOf(model))
	formElem := reflect.ValueOf(form).Elem()
	typeOfTForm := formElem.Type()
	for i := 0; i < formElem.NumField(); i++ {
		tag := typeOfTForm.Field(i).Tag
		if tag.Get("omit") != "true" {
			modelField := modelIndirect.FieldByName(typeOfTForm.Field(i).Name)
			if modelField.IsValid() {
				formField := formElem.Field(i)
				modelField.Set(formField)
			} else {
				log.Warnf("modelField : %s - %s", typeOfTForm.Field(i).Name, modelField)
			}
		}
	}
}

// AssignValue assign form values to model.
func BindValueForm(form interface{}, r *http.Request) {
	r.ParseForm()
	formElem := reflect.ValueOf(form).Elem()
	for i := 0; i < formElem.NumField(); i++ {
		typeField := formElem.Type().Field(i)
		tag := typeField.Tag
		switch typeField.Type.Name() {
			case "string" :
				formElem.Field(i).SetString(r.PostFormValue(tag.Get("form")))
			case "int" :
				nbr, _ := strconv.Atoi(r.PostFormValue(tag.Get("form")))
				formElem.Field(i).SetInt(int64(nbr))
			case "float" :
				nbr, _ := strconv.Atoi(r.PostFormValue(tag.Get("form")))
				formElem.Field(i).SetFloat(float64(nbr))
			case "bool" :
				nbr, _ := strconv.ParseBool(r.PostFormValue(tag.Get("form")))
				formElem.Field(i).SetBool(nbr)
		}
	}
}



func ValidateForm(form interface{}, errorForm map[string][]string) (map[string][]string) {
	formElem := reflect.ValueOf(form).Elem()
	for i := 0; i < formElem.NumField(); i++ {
		typeField := formElem.Type().Field(i)
		tag := typeField.Tag
		inputName := typeField.Name
		if (tag.Get("hum_name") != "") { // For more human input name than gibberish
			inputName = tag.Get("hum_name")
		}
		if tag.Get("len_min") != "" { // Check minimum length
			lenMin, _ := strconv.Atoi(tag.Get("len_min"))
			if formElem.Field(i).Len() < lenMin {
				errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Minimal length of %s required for the input: %s", strconv.Itoa(lenMin), inputName))
			}
		}
		if tag.Get("len_max") != "" { // Check minimum length
			lenMax, _ := strconv.Atoi(tag.Get("len_max"))
			if formElem.Field(i).Len() > lenMax {
				errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Maximal length of %s required for the input: %s", strconv.Itoa(lenMax), inputName))
			}
		}
		if tag.Get("equalInput") != "" {
			otherInput := formElem.FieldByName(tag.Get("equalInput"))
			if formElem.Field(i).Interface() != otherInput.Interface() {
				errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Must be same %s", inputName))
			}
		}
		switch typeField.Type.Name() {
			case "string" :
				if tag.Get("equal") != "" && formElem.Field(i).String() != tag.Get("equal") {
					errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Wrong value for the input: %s", inputName))
				}
				if tag.Get("needed") != "" && formElem.Field(i).String() == "" {
					errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Field needed: %s", inputName))
				}
			case "int" :
				if tag.Get("equal") != "" { // Check minimum length
					equal, _ := strconv.Atoi(tag.Get("equal"))
					if formElem.Field(i).Int() > int64(equal) {
						errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Wrong value for the input: %s", inputName))
					}
					if tag.Get("needed") != "" && formElem.Field(i).Int() == 0 {
						errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Field needed: %s", inputName))
					}
				}
			case "float" :
				if tag.Get("equal") != "" { // Check minimum length
					equal, _ := strconv.Atoi(tag.Get("equal"))
					if formElem.Field(i).Float() != float64(equal) {
						errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Wrong value for the input: %s", inputName))
					}
					if tag.Get("needed") != "" && formElem.Field(i).Float() == 0 {
						errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Field needed: %s", inputName))
					}
				}
			case "bool" :
				if tag.Get("equal") != "" { // Check minimum length
					equal, _ := strconv.ParseBool(tag.Get("equal"))
					if formElem.Field(i).Bool() != equal {
						errorForm[tag.Get("form")] = append(errorForm[tag.Get("form")], fmt.Sprintf("Wrong value for the input: %s", inputName))
					}
				}
		}
	}
	return errorForm
}