forgejo/modules/auth/auth.go

167 lines
3.9 KiB
Go
Raw Normal View History

2014-03-06 07:21:44 +00:00
// 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-07-26 04:24:27 +00:00
"strings"
2014-03-06 07:21:44 +00:00
2014-07-26 04:24:27 +00:00
"github.com/macaron-contrib/i18n"
"github.com/macaron-contrib/session"
2014-03-06 07:21:44 +00:00
2014-07-26 04:24:27 +00:00
"github.com/gogits/gogs/models"
"github.com/gogits/gogs/modules/log"
2014-05-05 06:42:52 +00:00
"github.com/gogits/gogs/modules/middleware/binding"
2014-07-26 04:24:27 +00:00
"github.com/gogits/gogs/modules/setting"
2014-03-06 07:21:44 +00:00
)
2014-07-26 04:24:27 +00:00
// SignedInId returns the id of signed in user.
func SignedInId(header http.Header, sess session.Store) int64 {
if !models.HasEngine {
return 0
}
if setting.Service.EnableReverseProxyAuth {
webAuthUser := header.Get(setting.ReverseProxyAuthUser)
if len(webAuthUser) > 0 {
u, err := models.GetUserByName(webAuthUser)
if err != nil {
if err != models.ErrUserNotExist {
log.Error(4, "GetUserByName: %v", err)
}
return 0
}
return u.Id
}
}
uid := sess.Get("uid")
if uid == nil {
return 0
}
if id, ok := uid.(int64); ok {
if _, err := models.GetUserById(id); err != nil {
if err != models.ErrUserNotExist {
log.Error(4, "GetUserById: %v", err)
}
return 0
}
return id
}
return 0
2014-07-02 20:42:16 +00:00
}
2014-07-26 04:24:27 +00:00
// SignedInUser returns the user object of signed user.
func SignedInUser(header http.Header, sess session.Store) *models.User {
uid := SignedInId(header, sess)
if uid <= 0 {
return nil
}
u, err := models.GetUserById(uid)
if err != nil {
log.Error(4, "GetUserById: %v", err)
return nil
}
return u
2014-03-28 22:40:31 +00:00
}
2014-07-26 04:24:27 +00:00
// AssignForm assign form values back to the template data.
func AssignForm(form interface{}, data map[string]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
}
data[fieldName] = val.Field(i).Interface()
}
}
2014-08-01 10:12:14 +00:00
func getSize(field reflect.StructField, prefix string) string {
2014-07-26 04:24:27 +00:00
for _, rule := range strings.Split(field.Tag.Get("binding"), ";") {
2014-08-01 10:12:14 +00:00
if strings.HasPrefix(rule, prefix) {
2014-07-26 04:24:27 +00:00
return rule[8 : len(rule)-1]
}
}
return ""
}
2014-08-01 10:12:14 +00:00
func GetMinSize(field reflect.StructField) string {
return getSize(field, "MinSize(")
}
func GetMaxSize(field reflect.StructField) string {
return getSize(field, "MaxSize(")
}
2014-07-26 04:24:27 +00:00
func validate(errs *binding.Errors, data map[string]interface{}, f interface{}, l i18n.Locale) {
if errs.Count() == 0 {
return
} else if len(errs.Overall) > 0 {
for _, err := range errs.Overall {
log.Error(4, "%s: %v", reflect.TypeOf(f), err)
}
return
}
data["HasError"] = true
AssignForm(f, data)
typ := reflect.TypeOf(f)
val := reflect.ValueOf(f)
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 := errs.Fields[field.Name]; ok {
data["Err_"+field.Name] = true
trName := l.Tr("form." + field.Name)
switch err {
case binding.BindingRequireError:
data["ErrorMsg"] = trName + l.Tr("form.require_error")
case binding.BindingAlphaDashError:
data["ErrorMsg"] = trName + l.Tr("form.alpha_dash_error")
case binding.BindingAlphaDashDotError:
data["ErrorMsg"] = trName + l.Tr("form.alpha_dash_dot_error")
case binding.BindingMinSizeError:
2014-08-01 10:12:14 +00:00
data["ErrorMsg"] = trName + l.Tr("form.min_size_error", GetMinSize(field))
2014-07-26 04:24:27 +00:00
case binding.BindingMaxSizeError:
2014-08-01 10:12:14 +00:00
data["ErrorMsg"] = trName + l.Tr("form.max_size_error", GetMaxSize(field))
2014-07-26 04:24:27 +00:00
case binding.BindingEmailError:
data["ErrorMsg"] = trName + l.Tr("form.email_error")
case binding.BindingUrlError:
data["ErrorMsg"] = trName + l.Tr("form.url_error")
default:
data["ErrorMsg"] = l.Tr("form.unknown_error") + " " + err
}
return
}
}
2014-03-28 22:40:31 +00:00
}