form.go 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package auth
  5. import (
  6. "net/http"
  7. "reflect"
  8. "strings"
  9. "github.com/codegangsta/martini"
  10. "github.com/gogits/binding"
  11. "github.com/gogits/gogs/modules/base"
  12. "github.com/gogits/gogs/utils/log"
  13. )
  14. type RegisterForm struct {
  15. Username string `form:"username" binding:"Required;AlphaDash;MinSize(5);MaxSize(30)"`
  16. Email string `form:"email" binding:"Required;Email;MaxSize(50)"`
  17. Password string `form:"passwd" binding:"Required;MinSize(6);MaxSize(30)"`
  18. }
  19. func getMinMaxSize(field reflect.StructField) string {
  20. for _, rule := range strings.Split(field.Tag.Get("binding"), ";") {
  21. if strings.HasPrefix(rule, "MinSize(") || strings.HasPrefix(rule, "MaxSize(") {
  22. return rule[8 : len(rule)-1]
  23. }
  24. }
  25. return ""
  26. }
  27. func validate(errors *binding.Errors, data base.TmplData, form interface{}) {
  28. typ := reflect.TypeOf(form)
  29. val := reflect.ValueOf(form)
  30. if typ.Kind() == reflect.Ptr {
  31. typ = typ.Elem()
  32. val = val.Elem()
  33. }
  34. for i := 0; i < typ.NumField(); i++ {
  35. field := typ.Field(i)
  36. fieldName := field.Tag.Get("form")
  37. // Allow ignored fields in the struct
  38. if fieldName == "-" {
  39. continue
  40. }
  41. if err, ok := errors.Fields[field.Name]; ok {
  42. data["Err_"+field.Name] = true
  43. switch err {
  44. case binding.RequireError:
  45. data["ErrorMsg"] = field.Name + " cannot be empty"
  46. case binding.AlphaDashError:
  47. data["ErrorMsg"] = field.Name + " must be valid alpha or numeric or dash(-_) characters"
  48. case binding.MinSizeError:
  49. data["ErrorMsg"] = field.Name + " must contain at least has " + getMinMaxSize(field) + " characters"
  50. case binding.MaxSizeError:
  51. data["ErrorMsg"] = field.Name + " must contain at most has " + getMinMaxSize(field) + " characters"
  52. case binding.EmailError:
  53. data["ErrorMsg"] = field.Name + " is not valid"
  54. default:
  55. data["ErrorMsg"] = "Unknown error: " + err
  56. }
  57. return
  58. }
  59. }
  60. }
  61. func (r *RegisterForm) Validate(errors *binding.Errors, req *http.Request, context martini.Context) {
  62. if req.Method == "GET" || errors.Count() == 0 {
  63. return
  64. }
  65. data := context.Get(reflect.TypeOf(base.TmplData{})).Interface().(base.TmplData)
  66. data["HasError"] = true
  67. AssignForm(r, data)
  68. if len(errors.Overall) > 0 {
  69. for _, err := range errors.Overall {
  70. log.Error("RegisterForm.Validate: %v", err)
  71. }
  72. return
  73. }
  74. validate(errors, data, r)
  75. }
  76. // AssignForm assign form values back to the template data.
  77. func AssignForm(form interface{}, data base.TmplData) {
  78. typ := reflect.TypeOf(form)
  79. val := reflect.ValueOf(form)
  80. if typ.Kind() == reflect.Ptr {
  81. typ = typ.Elem()
  82. val = val.Elem()
  83. }
  84. for i := 0; i < typ.NumField(); i++ {
  85. field := typ.Field(i)
  86. fieldName := field.Tag.Get("form")
  87. // Allow ignored fields in the struct
  88. if fieldName == "-" {
  89. continue
  90. }
  91. data[fieldName] = val.Field(i).Interface()
  92. }
  93. }