Backport #24867 by @lunny Fix #23211 Replace #23496 --------- Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
This commit is contained in:
parent
d19d5bc5b8
commit
275abd6593
2 changed files with 35 additions and 9 deletions
|
@ -613,7 +613,7 @@ func CreateUser(u *User, overwriteDefault ...*CreateUserOverwriteOptions) (err e
|
||||||
}
|
}
|
||||||
|
|
||||||
// validate data
|
// validate data
|
||||||
if err := validateUser(u); err != nil {
|
if err := ValidateUser(u); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -803,19 +803,26 @@ func checkDupEmail(ctx context.Context, u *User) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// validateUser check if user is valid to insert / update into database
|
// ValidateUser check if user is valid to insert / update into database
|
||||||
func validateUser(u *User) error {
|
func ValidateUser(u *User, cols ...string) error {
|
||||||
|
if len(cols) == 0 || util.SliceContainsString(cols, "visibility", true) {
|
||||||
if !setting.Service.AllowedUserVisibilityModesSlice.IsAllowedVisibility(u.Visibility) && !u.IsOrganization() {
|
if !setting.Service.AllowedUserVisibilityModesSlice.IsAllowedVisibility(u.Visibility) && !u.IsOrganization() {
|
||||||
return fmt.Errorf("visibility Mode not allowed: %s", u.Visibility.String())
|
return fmt.Errorf("visibility Mode not allowed: %s", u.Visibility.String())
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(cols) == 0 || util.SliceContainsString(cols, "email", true) {
|
||||||
u.Email = strings.ToLower(u.Email)
|
u.Email = strings.ToLower(u.Email)
|
||||||
return ValidateEmail(u.Email)
|
if err := ValidateEmail(u.Email); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpdateUser updates user's information.
|
// UpdateUser updates user's information.
|
||||||
func UpdateUser(ctx context.Context, u *User, changePrimaryEmail bool, cols ...string) error {
|
func UpdateUser(ctx context.Context, u *User, changePrimaryEmail bool, cols ...string) error {
|
||||||
err := validateUser(u)
|
err := ValidateUser(u, cols...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -881,7 +888,7 @@ func UpdateUser(ctx context.Context, u *User, changePrimaryEmail bool, cols ...s
|
||||||
|
|
||||||
// UpdateUserCols update user according special columns
|
// UpdateUserCols update user according special columns
|
||||||
func UpdateUserCols(ctx context.Context, u *User, cols ...string) error {
|
func UpdateUserCols(ctx context.Context, u *User, cols ...string) error {
|
||||||
if err := validateUser(u); err != nil {
|
if err := ValidateUser(u, cols...); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -5,6 +5,7 @@ package user_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
@ -524,3 +525,21 @@ func TestIsUserVisibleToViewer(t *testing.T) {
|
||||||
test(user31, user33, true)
|
test(user31, user33, true)
|
||||||
test(user31, nil, false)
|
test(user31, nil, false)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Test_ValidateUser(t *testing.T) {
|
||||||
|
oldSetting := setting.Service.AllowedUserVisibilityModesSlice
|
||||||
|
defer func() {
|
||||||
|
setting.Service.AllowedUserVisibilityModesSlice = oldSetting
|
||||||
|
}()
|
||||||
|
setting.Service.AllowedUserVisibilityModesSlice = []bool{true, false, true}
|
||||||
|
kases := map[*user_model.User]bool{
|
||||||
|
{ID: 1, Visibility: structs.VisibleTypePublic}: true,
|
||||||
|
{ID: 2, Visibility: structs.VisibleTypeLimited}: false,
|
||||||
|
{ID: 2, Visibility: structs.VisibleTypeLimited, Email: "invalid"}: false,
|
||||||
|
{ID: 2, Visibility: structs.VisibleTypePrivate, Email: "valid@valid.com"}: true,
|
||||||
|
}
|
||||||
|
for kase, expected := range kases {
|
||||||
|
err := user_model.ValidateUser(kase)
|
||||||
|
assert.EqualValues(t, expected, err == nil, fmt.Sprintf("case: %+v", kase))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue