From 5959e344a9133074cf2c65bd5e5d759be7895daf Mon Sep 17 00:00:00 2001
From: Unknown <joe2010xtmf@163.com>
Date: Thu, 6 Mar 2014 02:59:44 -0500
Subject: [PATCH] Optimize validate

---
 modules/auth/form.go       | 99 ++++++++++++++++++++------------------
 routers/user/user.go       |  2 +-
 templates/user/signup.tmpl |  2 +-
 3 files changed, 53 insertions(+), 50 deletions(-)

diff --git a/modules/auth/form.go b/modules/auth/form.go
index 9094ef5c..f918543f 100644
--- a/modules/auth/form.go
+++ b/modules/auth/form.go
@@ -7,6 +7,7 @@ package auth
 import (
 	"net/http"
 	"reflect"
+	"strings"
 
 	"github.com/codegangsta/martini"
 
@@ -17,11 +18,59 @@ import (
 )
 
 type RegisterForm struct {
-	UserName string `form:"username" binding:"Required;AlphaDash;MinSize(5);MaxSize(30)"`
+	Username string `form:"username" binding:"Required;AlphaDash;MinSize(5);MaxSize(30)"`
 	Email    string `form:"email" binding:"Required;Email;MaxSize(50)"`
 	Password string `form:"passwd" binding:"Required;MinSize(6);MaxSize(30)"`
 }
 
+func getMinMaxSize(field reflect.StructField) string {
+	for _, rule := range strings.Split(field.Tag.Get("binding"), ";") {
+		if strings.HasPrefix(rule, "MinSize(") || strings.HasPrefix(rule, "MaxSize(") {
+			return rule[8 : len(rule)-1]
+		}
+	}
+	return ""
+}
+
+func validate(errors *binding.Errors, data base.TmplData, form interface{}) {
+	typ := reflect.TypeOf(form)
+	val := reflect.ValueOf(form)
+
+	if typ.Kind() == reflect.Ptr {
+		typ = typ.Elem()
+		val = val.Elem()
+	}
+
+	for i := 0; i < typ.NumField(); i++ {
+		field := typ.Field(i)
+
+		fieldName := field.Tag.Get("form")
+		// Allow ignored fields in the struct
+		if fieldName == "-" {
+			continue
+		}
+
+		if err, ok := errors.Fields[field.Name]; ok {
+			data["Err_"+field.Name] = true
+			switch err {
+			case binding.RequireError:
+				data["ErrorMsg"] = field.Name + " cannot be empty"
+			case binding.AlphaDashError:
+				data["ErrorMsg"] = field.Name + " must be valid alpha or numeric or dash(-_) characters"
+			case binding.MinSizeError:
+				data["ErrorMsg"] = field.Name + " must contain at least has " + getMinMaxSize(field) + " characters"
+			case binding.MaxSizeError:
+				data["ErrorMsg"] = field.Name + " must contain at most has " + getMinMaxSize(field) + " characters"
+			case binding.EmailError:
+				data["ErrorMsg"] = field.Name + " is not valid"
+			default:
+				data["ErrorMsg"] = "Unknown error: " + err
+			}
+			return
+		}
+	}
+}
+
 func (r *RegisterForm) Validate(errors *binding.Errors, req *http.Request, context martini.Context) {
 	if req.Method == "GET" || errors.Count() == 0 {
 		return
@@ -38,53 +87,7 @@ func (r *RegisterForm) Validate(errors *binding.Errors, req *http.Request, conte
 		return
 	}
 
-	if err, ok := errors.Fields["UserName"]; ok {
-		data["Err_Username"] = true
-		switch err {
-		case binding.RequireError:
-			data["ErrorMsg"] = "Username cannot be empty"
-		case binding.AlphaDashError:
-			data["ErrorMsg"] = "Username must be valid alpha or numeric or dash(-_) characters"
-		case binding.MinSizeError:
-			data["ErrorMsg"] = "Username at least has 5 characters"
-		case binding.MaxSizeError:
-			data["ErrorMsg"] = "Username at most has 30 characters"
-		default:
-			data["ErrorMsg"] = "Unknown error: " + err
-		}
-		return
-	}
-
-	if err, ok := errors.Fields["Email"]; ok {
-		data["Err_Email"] = true
-		switch err {
-		case binding.RequireError:
-			data["ErrorMsg"] = "E-mail address cannot be empty"
-		case binding.EmailError:
-			data["ErrorMsg"] = "E-mail address is not valid"
-		case binding.MaxSizeError:
-			data["ErrorMsg"] = "E-mail address at most has 50 characters"
-		default:
-			data["ErrorMsg"] = "Unknown error: " + err
-		}
-		return
-	}
-
-	if err, ok := errors.Fields["Password"]; ok {
-		data["Err_Passwd"] = true
-		switch err {
-		case binding.RequireError:
-			data["ErrorMsg"] = "Password cannot be empty"
-		case binding.MinSizeError:
-			data["ErrorMsg"] = "Password at least has 6 characters"
-		case binding.MaxSizeError:
-			data["ErrorMsg"] = "Password at most has 30 characters"
-		default:
-			data["ErrorMsg"] = "Unknown error: " + err
-		}
-		return
-	}
-
+	validate(errors, data, r)
 }
 
 // AssignForm assign form values back to the template data.
diff --git a/routers/user/user.go b/routers/user/user.go
index 503ebf54..657bf95f 100644
--- a/routers/user/user.go
+++ b/routers/user/user.go
@@ -65,7 +65,7 @@ func SignUp(form auth.RegisterForm, data base.TmplData, req *http.Request, r ren
 
 	//Front-end should do double check of password.
 	u := &models.User{
-		Name:   form.UserName,
+		Name:   form.Username,
 		Email:  form.Email,
 		Passwd: form.Password,
 	}
diff --git a/templates/user/signup.tmpl b/templates/user/signup.tmpl
index 52e8fe18..265365bf 100644
--- a/templates/user/signup.tmpl
+++ b/templates/user/signup.tmpl
@@ -19,7 +19,7 @@
 			</div>
 		</div>
 
-		<div class="form-group {{if .Err_Passwd}}has-error has-feedback{{end}}">
+		<div class="form-group {{if .Err_Password}}has-error has-feedback{{end}}">
 			<label class="col-md-4 control-label">Password: </label>
 			<div class="col-md-6">
 				<input name="passwd" type="password" class="form-control" placeholder="Type your password">