bench-forgejo/modules/auth/auth.go

215 lines
5.8 KiB
Go
Raw Normal View History

2014-03-06 12:51:44 +05:30
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package auth
import (
"net/http"
"reflect"
2014-03-06 13:29:44 +05:30
"strings"
2014-03-06 12:51:44 +05:30
2014-03-30 21:41:28 +05:30
"github.com/go-martini/martini"
2014-03-06 12:51:44 +05:30
"github.com/gogits/gogs/modules/base"
2014-03-08 03:52:15 +05:30
"github.com/gogits/gogs/modules/log"
2014-03-06 12:51:44 +05:30
)
2014-03-09 07:55:38 +05:30
// Web form interface.
2014-03-06 21:40:35 +05:30
type Form interface {
Name(field string) string
}
2014-03-06 12:51:44 +05:30
type RegisterForm struct {
2014-03-14 01:51:16 +05:30
UserName string `form:"username" binding:"Required;AlphaDash;MaxSize(30)"`
2014-03-06 21:40:35 +05:30
Email string `form:"email" binding:"Required;Email;MaxSize(50)"`
Password string `form:"passwd" binding:"Required;MinSize(6);MaxSize(30)"`
RetypePasswd string `form:"retypepasswd"`
}
2014-03-06 22:12:14 +05:30
func (f *RegisterForm) Name(field string) string {
2014-03-06 21:40:35 +05:30
names := map[string]string{
"UserName": "Username",
"Email": "E-mail address",
"Password": "Password",
"RetypePasswd": "Re-type password",
}
return names[field]
2014-03-06 12:51:44 +05:30
}
2014-04-13 11:27:42 +05:30
func (f *RegisterForm) Validate(errors *base.BindingErrors, req *http.Request, context martini.Context) {
2014-03-06 22:12:14 +05:30
if req.Method == "GET" || errors.Count() == 0 {
return
}
data := context.Get(reflect.TypeOf(base.TmplData{})).Interface().(base.TmplData)
data["HasError"] = true
AssignForm(f, data)
if len(errors.Overall) > 0 {
for _, err := range errors.Overall {
log.Error("RegisterForm.Validate: %v", err)
}
return
}
validate(errors, data, f)
}
type LogInForm struct {
2014-03-14 01:51:16 +05:30
UserName string `form:"username" binding:"Required;AlphaDash;MaxSize(30)"`
2014-03-06 22:12:14 +05:30
Password string `form:"passwd" binding:"Required;MinSize(6);MaxSize(30)"`
2014-03-23 02:10:09 +05:30
Remember string `form:"remember"`
2014-03-06 22:12:14 +05:30
}
func (f *LogInForm) Name(field string) string {
names := map[string]string{
"UserName": "Username",
"Password": "Password",
}
return names[field]
}
2014-04-13 11:27:42 +05:30
func (f *LogInForm) Validate(errors *base.BindingErrors, req *http.Request, context martini.Context) {
2014-03-06 22:12:14 +05:30
if req.Method == "GET" || errors.Count() == 0 {
return
}
data := context.Get(reflect.TypeOf(base.TmplData{})).Interface().(base.TmplData)
data["HasError"] = true
AssignForm(f, data)
if len(errors.Overall) > 0 {
for _, err := range errors.Overall {
log.Error("LogInForm.Validate: %v", err)
}
return
}
validate(errors, data, f)
}
2014-03-06 13:29:44 +05: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 ""
}
2014-04-13 11:27:42 +05:30
func validate(errors *base.BindingErrors, data base.TmplData, form Form) {
2014-03-06 13:29:44 +05:30
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 {
2014-04-13 11:27:42 +05:30
case base.BindingRequireError:
2014-03-06 21:40:35 +05:30
data["ErrorMsg"] = form.Name(field.Name) + " cannot be empty"
2014-04-13 11:27:42 +05:30
case base.BindingAlphaDashError:
2014-03-06 21:40:35 +05:30
data["ErrorMsg"] = form.Name(field.Name) + " must be valid alpha or numeric or dash(-_) characters"
2014-04-13 11:27:42 +05:30
case base.BindingMinSizeError:
2014-03-06 21:40:35 +05:30
data["ErrorMsg"] = form.Name(field.Name) + " must contain at least " + getMinMaxSize(field) + " characters"
2014-04-13 11:27:42 +05:30
case base.BindingMaxSizeError:
2014-03-06 21:40:35 +05:30
data["ErrorMsg"] = form.Name(field.Name) + " must contain at most " + getMinMaxSize(field) + " characters"
2014-04-13 11:27:42 +05:30
case base.BindingEmailError:
2014-04-13 06:05:35 +05:30
data["ErrorMsg"] = form.Name(field.Name) + " is not a valid e-mail address"
2014-04-13 11:27:42 +05:30
case base.BindingUrlError:
2014-04-13 06:05:35 +05:30
data["ErrorMsg"] = form.Name(field.Name) + " is not a valid URL"
2014-03-06 13:29:44 +05:30
default:
data["ErrorMsg"] = "Unknown error: " + err
}
return
}
}
}
2014-03-06 12:51:44 +05:30
// AssignForm assign form values back to the template data.
func AssignForm(form interface{}, data base.TmplData) {
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
}
data[fieldName] = val.Field(i).Interface()
}
}
2014-03-29 04:10:31 +05:30
type InstallForm struct {
Database string `form:"database" binding:"Required"`
Host string `form:"host"`
User string `form:"user"`
Passwd string `form:"passwd"`
DatabaseName string `form:"database_name"`
SslMode string `form:"ssl_mode"`
DatabasePath string `form:"database_path"`
RepoRootPath string `form:"repo_path"`
RunUser string `form:"run_user"`
Domain string `form:"domain"`
2014-03-29 04:10:31 +05:30
AppUrl string `form:"app_url"`
AdminName string `form:"admin_name" binding:"Required"`
AdminPasswd string `form:"admin_pwd" binding:"Required;MinSize(6);MaxSize(30)"`
AdminEmail string `form:"admin_email" binding:"Required;Email;MaxSize(50)"`
SmtpHost string `form:"smtp_host"`
SmtpEmail string `form:"mailer_user"`
SmtpPasswd string `form:"mailer_pwd"`
RegisterConfirm string `form:"register_confirm"`
MailNotify string `form:"mail_notify"`
}
func (f *InstallForm) Name(field string) string {
names := map[string]string{
"Database": "Database name",
"AdminName": "Admin user name",
"AdminPasswd": "Admin password",
"AdminEmail": "Admin e-maill address",
}
return names[field]
}
2014-04-13 11:27:42 +05:30
func (f *InstallForm) Validate(errors *base.BindingErrors, req *http.Request, context martini.Context) {
2014-03-29 04:10:31 +05:30
if req.Method == "GET" || errors.Count() == 0 {
return
}
data := context.Get(reflect.TypeOf(base.TmplData{})).Interface().(base.TmplData)
data["HasError"] = true
AssignForm(f, data)
if len(errors.Overall) > 0 {
for _, err := range errors.Overall {
log.Error("InstallForm.Validate: %v", err)
}
return
}
validate(errors, data, f)
}