Merge pull request #105 from miraclesu/form

Add ParseForm Function & utils test
2024-06-25 05:54:13 +00:00 · 2013-07-26 01:36:13 -07:00 · 2013-07-26 01:36:13 -07:00 · da5dd4d173
commit da5dd4d173
parent 6373379da6 6b5dc3b7d5
3 changed files with 224 additions and 0 deletions
--- a/controller.go
+++ b/controller.go
@ -264,6 +264,10 @@ func (c *Controller) Input() url.Values {
 	return c.Ctx.Request.Form
 }

+func (c *Controller) ParseForm(obj interface{}) error {
+	return ParseForm(c.Input(), obj)
+}
+
 func (c *Controller) GetString(key string) string {
 	return c.Input().Get(key)
 }
--- a/utils.go
+++ b/utils.go
@ -3,7 +3,10 @@ package beego
 import (
 	"fmt"
 	"html/template"
+	"net/url"
+	"reflect"
 	"regexp"
+	"strconv"
 	"strings"
 	"time"
 )
@ -170,3 +173,80 @@ func inSlice(v string, sl []string) bool {
 	}
 	return false
 }
+
+// parse form values to struct via tag
+func ParseForm(form url.Values, obj interface{}) error {
+	objT := reflect.TypeOf(obj)
+	objV := reflect.ValueOf(obj)
+	if !(objT.Kind() == reflect.Ptr && objT.Elem().Kind() == reflect.Struct) {
+		return fmt.Errorf("%v must be  a struct pointer", obj)
+	}
+	objT = objT.Elem()
+	objV = objV.Elem()
+
+	for i := 0; i < objT.NumField(); i++ {
+		fieldV := objV.Field(i)
+		if !fieldV.CanSet() {
+			continue
+		}
+		fieldT := objT.Field(i)
+		tag := fieldT.Tag.Get("form")
+		if len(tag) == 0 {
+			tag = fieldT.Name
+		}
+		value := form.Get(tag)
+		if len(value) == 0 {
+			continue
+		}
+
+		switch fieldT.Type.Kind() {
+		case reflect.Bool:
+			b, err := strconv.ParseBool(value)
+			if err != nil {
+				return err
+			}
+			fieldV.SetBool(b)
+		case reflect.Int:
+			fallthrough
+		case reflect.Int8:
+			fallthrough
+		case reflect.Int16:
+			fallthrough
+		case reflect.Int32:
+			fallthrough
+		case reflect.Int64:
+			x, err := strconv.ParseInt(value, 10, 64)
+			if err != nil {
+				return err
+			}
+			fieldV.SetInt(x)
+		case reflect.Uint:
+			fallthrough
+		case reflect.Uint8:
+			fallthrough
+		case reflect.Uint16:
+			fallthrough
+		case reflect.Uint32:
+			fallthrough
+		case reflect.Uint64:
+			x, err := strconv.ParseUint(value, 10, 64)
+			if err != nil {
+				return err
+			}
+			fieldV.SetUint(x)
+		case reflect.Float32:
+			fallthrough
+		case reflect.Float64:
+			x, err := strconv.ParseFloat(value, 64)
+			if err != nil {
+				return err
+			}
+			fieldV.SetFloat(x)
+		case reflect.Interface:
+			fieldV.Set(reflect.ValueOf(value))
+		case reflect.String:
+			fieldV.SetString(value)
+		}
+	}
+	return nil
+}
--- a/utils_test.go
+++ b/utils_test.go
@ -0,0 +1,140 @@
+package beego
+
+import (
+	"net/url"
+	"testing"
+	"time"
+)
+
+func TestWebTime(t *testing.T) {
+	ts := "Fri, 26 Jul 2013 12:27:42 CST"
+	tt, _ := time.Parse(time.RFC1123, ts)
+	if ts != webTime(tt) {
+		t.Error("should be equal")
+	}
+	if "Fri, 26 Jul 2013 04:27:42 GMT" != webTime(tt.UTC()) {
+		t.Error("should be equal")
+	}
+}
+
+func TestSubstr(t *testing.T) {
+	s := `012345`
+	if Substr(s, 0, 2) != "01" {
+		t.Error("should be equal")
+	}
+	if Substr(s, 0, 100) != "012345" {
+		t.Error("should be equal")
+	}
+}
+
+func TestHtml2str(t *testing.T) {
+	h := `<HTML><style></style><script>x<x</script></HTML><123>  123\n
+
+
+	\n`
+	if Html2str(h) != "123\\n\n\\n" {
+		t.Error("should be equal")
+	}
+}
+
+func TestDateFormat(t *testing.T) {
+	ts := "Mon, 01 Jul 2013 13:27:42 CST"
+	tt, _ := time.Parse(time.RFC1123, ts)
+	if DateFormat(tt, "2006-01-02 15:04:05") != "2013-07-01 13:27:42" {
+		t.Error("should be equal")
+	}
+}
+
+func TestDate(t *testing.T) {
+	ts := "Mon, 01 Jul 2013 13:27:42 CST"
+	tt, _ := time.Parse(time.RFC1123, ts)
+	if Date(tt, "Y-m-d H:i:s") != "2013-07-01 13:27:42" {
+		t.Error("should be equal")
+	}
+	if Date(tt, "y-n-j h:i:s A") != "13-7-1 01:27:42 PM" {
+		t.Error("should be equal")
+	}
+	if Date(tt, "D, d M Y g:i:s a") != "Mon, 01 Jul 2013 1:27:42 pm" {
+		t.Error("should be equal")
+	}
+	if Date(tt, "l, d F Y G:i:s") != "Monday, 01 July 2013 13:27:42" {
+		t.Error("should be equal")
+	}
+}
+
+func TestCompare(t *testing.T) {
+	if !Compare("abc", "abc") {
+		t.Error("should be equal")
+	}
+	if Compare("abc", "aBc") {
+		t.Error("should be not equal")
+	}
+	if !Compare("1", 1) {
+		t.Error("should be equal")
+	}
+}
+
+func TestHtmlquote(t *testing.T) {
+	h := `&lt;&#39;&nbsp;&rdquo;&ldquo;&amp;&quot;&gt;`
+	s := `<' ”“&">`
+	if Htmlquote(s) != h {
+		t.Error("should be equal")
+	}
+}
+
+func TestHtmlunquote(t *testing.T) {
+	h := `&lt;&#39;&nbsp;&rdquo;&ldquo;&amp;&quot;&gt;`
+	s := `<' ”“&">`
+	if Htmlunquote(h) != s {
+		t.Error("should be equal")
+	}
+}
+
+func TestInSlice(t *testing.T) {
+	sl := []string{"A", "b"}
+	if !inSlice("A", sl) {
+		t.Error("should be true")
+	}
+	if inSlice("B", sl) {
+		t.Error("should be false")
+	}
+}
+
+func TestParseForm(t *testing.T) {
+	type user struct {
+		Id    int
+		tag   string      `form:tag`
+		Name  interface{} `form:"username"`
+		Age   int         `form:"age"`
+		Email string
+	}
+
+	u := user{}
+	form := url.Values{
+		"tag":      []string{"no"},
+		"username": []string{"test"},
+		"age":      []string{"40"},
+		"Email":    []string{"test@gmail.com"},
+	}
+	if err := ParseForm(form, u); err == nil {
+		t.Fatal("nothing will be changed")
+	}
+	if err := ParseForm(form, &u); err != nil {
+		t.Fatal(err)
+	}
+	if u.Id != 0 {
+		t.Error("Id should not be changed")
+	}
+	if len(u.tag) != 0 {
+		t.Error("tag should not be changed")
+	}
+	if u.Name.(string) != "test" {
+		t.Error("should be equal")
+	}
+	if u.Age != 40 {
+		t.Error("should be equal")
+	}
+	if u.Email != "test@gmail.com" {
+		t.Error("should be equal")
+	}
+}