casdoor/object/user.go

896 lines
26 KiB
Go
Raw Normal View History

2022-02-13 23:39:27 +08:00
// Copyright 2021 The Casdoor Authors. All Rights Reserved.
2020-10-20 23:14:03 +08:00
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package object
import (
2021-02-21 23:51:40 +08:00
"fmt"
"strings"
2021-02-21 23:51:40 +08:00
2022-09-13 20:31:22 +08:00
"github.com/casdoor/casdoor/conf"
2022-01-20 14:11:46 +08:00
"github.com/casdoor/casdoor/util"
2023-03-07 13:38:48 +08:00
"github.com/go-webauthn/webauthn/webauthn"
feat: app session control and db migrate (#1539) * feat: integrate application session management into Casdoor's session management (#774) && standardized the database migration process (#1533) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) * feat: integrate application session management into Casdoor's session management (#774) && standardized the database migration process * feat: integrate application session management into Casdoor's session management (#774) && standardized the database migration process * feat: integrate application session management into Casdoor's session management (#774) && standardized the database migration process --------- Co-authored-by: Zayn Xie <84443886+xiaoniuren99@users.noreply.github.com> * fix: migrate err * fix: migrate err * feat: app session control and db migrate * feat: app session control and db migrate * feat: app session control and db migrate --------- Co-authored-by: Zayn Xie <84443886+xiaoniuren99@users.noreply.github.com>
2023-02-12 09:33:24 +08:00
"github.com/xorm-io/core"
2020-10-20 23:14:03 +08:00
)
const (
2022-08-09 16:50:49 +08:00
UserPropertiesWechatUnionId = "wechatUnionId"
UserPropertiesWechatOpenId = "wechatOpenId"
)
const UserEnforcerId = "built-in/user-enforcer-built-in"
var userEnforcer *UserGroupEnforcer
func InitUserManager() {
enforcer, err := GetInitializedEnforcer(UserEnforcerId)
if err != nil {
panic(err)
}
userEnforcer = NewUserGroupEnforcer(enforcer.Enforcer)
}
2020-10-20 23:14:03 +08:00
type User struct {
Owner string `xorm:"varchar(100) notnull pk" json:"owner"`
Name string `xorm:"varchar(100) notnull pk" json:"name"`
2023-06-01 22:51:01 +08:00
CreatedTime string `xorm:"varchar(100) index" json:"createdTime"`
2021-06-01 20:37:00 +08:00
UpdatedTime string `xorm:"varchar(100)" json:"updatedTime"`
2020-10-20 23:14:03 +08:00
2022-01-14 17:42:11 +08:00
Id string `xorm:"varchar(100) index" json:"id"`
2021-06-21 10:22:47 +08:00
Type string `xorm:"varchar(100)" json:"type"`
Password string `xorm:"varchar(100)" json:"password"`
2021-11-04 21:08:43 +08:00
PasswordSalt string `xorm:"varchar(100)" json:"passwordSalt"`
PasswordType string `xorm:"varchar(100)" json:"passwordType"`
2021-06-21 10:22:47 +08:00
DisplayName string `xorm:"varchar(100)" json:"displayName"`
FirstName string `xorm:"varchar(100)" json:"firstName"`
LastName string `xorm:"varchar(100)" json:"lastName"`
2021-12-24 00:36:53 +08:00
Avatar string `xorm:"varchar(500)" json:"avatar"`
2023-06-17 11:43:46 +08:00
AvatarType string `xorm:"varchar(100)" json:"avatarType"`
2021-12-24 00:36:53 +08:00
PermanentAvatar string `xorm:"varchar(500)" json:"permanentAvatar"`
2022-01-04 19:52:29 +08:00
Email string `xorm:"varchar(100) index" json:"email"`
EmailVerified bool `json:"emailVerified"`
Phone string `xorm:"varchar(20) index" json:"phone"`
CountryCode string `xorm:"varchar(6)" json:"countryCode"`
2023-04-15 16:05:33 +08:00
Region string `xorm:"varchar(100)" json:"region"`
2021-08-21 10:58:34 +08:00
Location string `xorm:"varchar(100)" json:"location"`
2021-06-21 10:22:47 +08:00
Address []string `json:"address"`
Affiliation string `xorm:"varchar(100)" json:"affiliation"`
2021-08-21 10:58:34 +08:00
Title string `xorm:"varchar(100)" json:"title"`
2021-11-19 21:37:13 +08:00
IdCardType string `xorm:"varchar(100)" json:"idCardType"`
2022-01-04 19:52:29 +08:00
IdCard string `xorm:"varchar(100) index" json:"idCard"`
2021-08-21 10:58:34 +08:00
Homepage string `xorm:"varchar(100)" json:"homepage"`
2021-08-25 08:07:08 +08:00
Bio string `xorm:"varchar(100)" json:"bio"`
2021-06-21 10:22:47 +08:00
Tag string `xorm:"varchar(100)" json:"tag"`
Language string `xorm:"varchar(100)" json:"language"`
2021-11-19 21:37:13 +08:00
Gender string `xorm:"varchar(100)" json:"gender"`
Birthday string `xorm:"varchar(100)" json:"birthday"`
Education string `xorm:"varchar(100)" json:"education"`
2021-06-21 10:22:47 +08:00
Score int `json:"score"`
Karma int `json:"karma"`
Ranking int `json:"ranking"`
2021-12-01 21:26:28 +08:00
IsDefaultAvatar bool `json:"isDefaultAvatar"`
2021-08-21 10:58:34 +08:00
IsOnline bool `json:"isOnline"`
2021-06-21 10:22:47 +08:00
IsAdmin bool `json:"isAdmin"`
IsGlobalAdmin bool `json:"isGlobalAdmin"`
IsForbidden bool `json:"isForbidden"`
2021-11-06 15:52:03 +08:00
IsDeleted bool `json:"isDeleted"`
2021-06-21 10:22:47 +08:00
SignupApplication string `xorm:"varchar(100)" json:"signupApplication"`
Hash string `xorm:"varchar(100)" json:"hash"`
PreHash string `xorm:"varchar(100)" json:"preHash"`
AccessKey string `xorm:"varchar(100)" json:"accessKey"`
2023-06-16 20:42:15 +08:00
AccessSecret string `xorm:"varchar(100)" json:"accessSecret"`
2021-02-14 00:22:24 +08:00
2021-11-19 21:37:13 +08:00
CreatedIp string `xorm:"varchar(100)" json:"createdIp"`
LastSigninTime string `xorm:"varchar(100)" json:"lastSigninTime"`
LastSigninIp string `xorm:"varchar(100)" json:"lastSigninIp"`
GitHub string `xorm:"github varchar(100)" json:"github"`
Google string `xorm:"varchar(100)" json:"google"`
QQ string `xorm:"qq varchar(100)" json:"qq"`
WeChat string `xorm:"wechat varchar(100)" json:"wechat"`
Facebook string `xorm:"facebook varchar(100)" json:"facebook"`
DingTalk string `xorm:"dingtalk varchar(100)" json:"dingtalk"`
Weibo string `xorm:"weibo varchar(100)" json:"weibo"`
Gitee string `xorm:"gitee varchar(100)" json:"gitee"`
LinkedIn string `xorm:"linkedin varchar(100)" json:"linkedin"`
Wecom string `xorm:"wecom varchar(100)" json:"wecom"`
Lark string `xorm:"lark varchar(100)" json:"lark"`
Gitlab string `xorm:"gitlab varchar(100)" json:"gitlab"`
Adfs string `xorm:"adfs varchar(100)" json:"adfs"`
Baidu string `xorm:"baidu varchar(100)" json:"baidu"`
Alipay string `xorm:"alipay varchar(100)" json:"alipay"`
Casdoor string `xorm:"casdoor varchar(100)" json:"casdoor"`
Infoflow string `xorm:"infoflow varchar(100)" json:"infoflow"`
Apple string `xorm:"apple varchar(100)" json:"apple"`
AzureAD string `xorm:"azuread varchar(100)" json:"azuread"`
Slack string `xorm:"slack varchar(100)" json:"slack"`
Steam string `xorm:"steam varchar(100)" json:"steam"`
Bilibili string `xorm:"bilibili varchar(100)" json:"bilibili"`
Okta string `xorm:"okta varchar(100)" json:"okta"`
Douyin string `xorm:"douyin varchar(100)" json:"douyin"`
Line string `xorm:"line varchar(100)" json:"line"`
Amazon string `xorm:"amazon varchar(100)" json:"amazon"`
Auth0 string `xorm:"auth0 varchar(100)" json:"auth0"`
BattleNet string `xorm:"battlenet varchar(100)" json:"battlenet"`
Bitbucket string `xorm:"bitbucket varchar(100)" json:"bitbucket"`
Box string `xorm:"box varchar(100)" json:"box"`
CloudFoundry string `xorm:"cloudfoundry varchar(100)" json:"cloudfoundry"`
Dailymotion string `xorm:"dailymotion varchar(100)" json:"dailymotion"`
Deezer string `xorm:"deezer varchar(100)" json:"deezer"`
DigitalOcean string `xorm:"digitalocean varchar(100)" json:"digitalocean"`
Discord string `xorm:"discord varchar(100)" json:"discord"`
Dropbox string `xorm:"dropbox varchar(100)" json:"dropbox"`
EveOnline string `xorm:"eveonline varchar(100)" json:"eveonline"`
Fitbit string `xorm:"fitbit varchar(100)" json:"fitbit"`
Gitea string `xorm:"gitea varchar(100)" json:"gitea"`
Heroku string `xorm:"heroku varchar(100)" json:"heroku"`
InfluxCloud string `xorm:"influxcloud varchar(100)" json:"influxcloud"`
Instagram string `xorm:"instagram varchar(100)" json:"instagram"`
Intercom string `xorm:"intercom varchar(100)" json:"intercom"`
Kakao string `xorm:"kakao varchar(100)" json:"kakao"`
Lastfm string `xorm:"lastfm varchar(100)" json:"lastfm"`
Mailru string `xorm:"mailru varchar(100)" json:"mailru"`
Meetup string `xorm:"meetup varchar(100)" json:"meetup"`
MicrosoftOnline string `xorm:"microsoftonline varchar(100)" json:"microsoftonline"`
Naver string `xorm:"naver varchar(100)" json:"naver"`
Nextcloud string `xorm:"nextcloud varchar(100)" json:"nextcloud"`
OneDrive string `xorm:"onedrive varchar(100)" json:"onedrive"`
Oura string `xorm:"oura varchar(100)" json:"oura"`
Patreon string `xorm:"patreon varchar(100)" json:"patreon"`
Paypal string `xorm:"paypal varchar(100)" json:"paypal"`
SalesForce string `xorm:"salesforce varchar(100)" json:"salesforce"`
Shopify string `xorm:"shopify varchar(100)" json:"shopify"`
Soundcloud string `xorm:"soundcloud varchar(100)" json:"soundcloud"`
Spotify string `xorm:"spotify varchar(100)" json:"spotify"`
Strava string `xorm:"strava varchar(100)" json:"strava"`
Stripe string `xorm:"stripe varchar(100)" json:"stripe"`
TikTok string `xorm:"tiktok varchar(100)" json:"tiktok"`
Tumblr string `xorm:"tumblr varchar(100)" json:"tumblr"`
Twitch string `xorm:"twitch varchar(100)" json:"twitch"`
Twitter string `xorm:"twitter varchar(100)" json:"twitter"`
Typetalk string `xorm:"typetalk varchar(100)" json:"typetalk"`
Uber string `xorm:"uber varchar(100)" json:"uber"`
VK string `xorm:"vk varchar(100)" json:"vk"`
Wepay string `xorm:"wepay varchar(100)" json:"wepay"`
Xero string `xorm:"xero varchar(100)" json:"xero"`
Yahoo string `xorm:"yahoo varchar(100)" json:"yahoo"`
Yammer string `xorm:"yammer varchar(100)" json:"yammer"`
Yandex string `xorm:"yandex varchar(100)" json:"yandex"`
Zoom string `xorm:"zoom varchar(100)" json:"zoom"`
MetaMask string `xorm:"metamask varchar(100)" json:"metamask"`
Web3Onboard string `xorm:"web3onboard varchar(100)" json:"web3onboard"`
Custom string `xorm:"custom varchar(100)" json:"custom"`
2021-05-30 15:13:43 +08:00
WebauthnCredentials []webauthn.Credential `xorm:"webauthnCredentials blob" json:"webauthnCredentials"`
PreferredMfaType string `xorm:"varchar(100)" json:"preferredMfaType"`
RecoveryCodes []string `xorm:"varchar(1000)" json:"recoveryCodes"`
TotpSecret string `xorm:"varchar(100)" json:"totpSecret"`
MfaPhoneEnabled bool `json:"mfaPhoneEnabled"`
MfaEmailEnabled bool `json:"mfaEmailEnabled"`
MultiFactorAuths []*MfaProps `xorm:"-" json:"multiFactorAuths,omitempty"`
Ldap string `xorm:"ldap varchar(100)" json:"ldap"`
2021-05-30 15:13:43 +08:00
Properties map[string]string `json:"properties"`
Roles []*Role `json:"roles"`
Permissions []*Permission `json:"permissions"`
Groups []string `xorm:"groups varchar(1000)" json:"groups"`
LastSigninWrongTime string `xorm:"varchar(100)" json:"lastSigninWrongTime"`
SigninWrongTimes int `json:"signinWrongTimes"`
ManagedAccounts []ManagedAccount `xorm:"managedAccounts blob" json:"managedAccounts"`
2020-10-20 23:14:03 +08:00
}
type Userinfo struct {
2023-06-03 18:24:02 +08:00
Sub string `json:"sub"`
Iss string `json:"iss"`
Aud string `json:"aud"`
Name string `json:"preferred_username,omitempty"`
DisplayName string `json:"name,omitempty"`
Email string `json:"email,omitempty"`
Avatar string `json:"picture,omitempty"`
Address string `json:"address,omitempty"`
Phone string `json:"phone,omitempty"`
Groups []string `json:"groups,omitempty"`
}
type ManagedAccount struct {
Application string `xorm:"varchar(100)" json:"application"`
Username string `xorm:"varchar(100)" json:"username"`
Password string `xorm:"varchar(100)" json:"password"`
SigninUrl string `xorm:"varchar(200)" json:"signinUrl"`
}
func GetGlobalUserCount(field, value string) (int64, error) {
session := GetSession("", -1, -1, field, value, "", "")
return session.Count(&User{})
}
func GetGlobalUsers() ([]*User, error) {
2021-02-13 00:11:12 +08:00
users := []*User{}
err := ormer.Engine.Desc("created_time").Find(&users)
2021-02-13 00:11:12 +08:00
if err != nil {
return nil, err
2021-02-13 00:11:12 +08:00
}
return users, nil
2021-02-13 00:11:12 +08:00
}
func GetPaginationGlobalUsers(offset, limit int, field, value, sortField, sortOrder string) ([]*User, error) {
users := []*User{}
2023-06-01 22:51:01 +08:00
session := GetSessionForUser("", offset, limit, field, value, sortField, sortOrder)
err := session.Find(&users)
if err != nil {
return nil, err
}
return users, nil
}
func GetUserCount(owner, field, value string, groupName string) (int64, error) {
session := GetSession(owner, -1, -1, field, value, "", "")
if groupName != "" {
return GetGroupUserCount(util.GetId(owner, groupName), field, value)
}
return session.Count(&User{})
}
func GetOnlineUserCount(owner string, isOnline int) (int64, error) {
return ormer.Engine.Where("is_online = ?", isOnline).Count(&User{Owner: owner})
}
func GetUsers(owner string) ([]*User, error) {
2020-10-20 23:14:03 +08:00
users := []*User{}
err := ormer.Engine.Desc("created_time").Find(&users, &User{Owner: owner})
2020-10-20 23:14:03 +08:00
if err != nil {
return nil, err
2020-10-20 23:14:03 +08:00
}
return users, nil
2020-10-20 23:14:03 +08:00
}
func GetUsersByTag(owner string, tag string) ([]*User, error) {
2023-05-12 13:03:43 +08:00
users := []*User{}
err := ormer.Engine.Desc("created_time").Find(&users, &User{Owner: owner, Tag: tag})
2023-05-12 13:03:43 +08:00
if err != nil {
return nil, err
2023-05-12 13:03:43 +08:00
}
return users, nil
2023-05-12 13:03:43 +08:00
}
func GetSortedUsers(owner string, sorter string, limit int) ([]*User, error) {
users := []*User{}
err := ormer.Engine.Desc(sorter).Limit(limit, 0).Find(&users, &User{Owner: owner})
if err != nil {
return nil, err
}
return users, nil
}
func GetPaginationUsers(owner string, offset, limit int, field, value, sortField, sortOrder string, groupName string) ([]*User, error) {
users := []*User{}
if groupName != "" {
return GetPaginationGroupUsers(util.GetId(owner, groupName), offset, limit, field, value, sortField, sortOrder)
}
2023-06-01 22:51:01 +08:00
session := GetSessionForUser(owner, offset, limit, field, value, sortField, sortOrder)
err := session.Find(&users)
if err != nil {
return nil, err
}
return users, nil
}
func getUser(owner string, name string) (*User, error) {
2021-06-21 01:01:16 +08:00
if owner == "" || name == "" {
return nil, nil
2021-06-21 01:01:16 +08:00
}
2020-10-20 23:14:03 +08:00
user := User{Owner: owner, Name: name}
existed, err := ormer.Engine.Get(&user)
2020-10-20 23:14:03 +08:00
if err != nil {
return nil, err
2020-10-20 23:14:03 +08:00
}
if existed {
return &user, nil
} else {
return nil, nil
}
}
func getUserById(owner string, id string) (*User, error) {
if owner == "" || id == "" {
return nil, nil
}
user := User{Owner: owner, Id: id}
existed, err := ormer.Engine.Get(&user)
if err != nil {
return nil, err
}
if existed {
return &user, nil
} else {
return nil, nil
}
}
2023-06-21 20:54:19 +08:00
func getUserByWechatId(owner string, wechatOpenId string, wechatUnionId string) (*User, error) {
if wechatUnionId == "" {
wechatUnionId = wechatOpenId
}
user := &User{}
existed, err := ormer.Engine.Where("owner = ?", owner).Where("wechat = ? OR wechat = ?", wechatOpenId, wechatUnionId).Get(user)
if err != nil {
return nil, err
}
if existed {
return user, nil
} else {
return nil, nil
}
}
func GetUserByEmail(owner string, email string) (*User, error) {
if owner == "" || email == "" {
return nil, nil
}
user := User{Owner: owner, Email: email}
existed, err := ormer.Engine.Get(&user)
if err != nil {
return nil, err
}
2020-10-20 23:14:03 +08:00
if existed {
return &user, nil
2020-10-20 23:14:03 +08:00
} else {
return nil, nil
2020-10-20 23:14:03 +08:00
}
}
func GetUserByPhone(owner string, phone string) (*User, error) {
if owner == "" || phone == "" {
return nil, nil
}
user := User{Owner: owner, Phone: phone}
existed, err := ormer.Engine.Get(&user)
if err != nil {
return nil, err
}
if existed {
return &user, nil
} else {
return nil, nil
}
}
func GetUserByUserId(owner string, userId string) (*User, error) {
if owner == "" || userId == "" {
return nil, nil
}
user := User{Owner: owner, Id: userId}
existed, err := ormer.Engine.Get(&user)
if err != nil {
return nil, err
}
if existed {
return &user, nil
} else {
return nil, nil
}
}
func GetUserByAccessKey(accessKey string) (*User, error) {
if accessKey == "" {
return nil, nil
}
user := User{AccessKey: accessKey}
existed, err := ormer.Engine.Get(&user)
if err != nil {
return nil, err
}
if existed {
return &user, nil
} else {
return nil, nil
}
}
func GetUser(id string) (*User, error) {
2020-10-20 23:14:03 +08:00
owner, name := util.GetOwnerAndNameFromId(id)
return getUser(owner, name)
}
func GetUserNoCheck(id string) (*User, error) {
2021-11-20 15:46:54 +08:00
owner, name := util.GetOwnerAndNameFromIdNoCheck(id)
return getUser(owner, name)
}
func GetMaskedUser(user *User, isAdminOrSelf bool, errs ...error) (*User, error) {
if len(errs) > 0 && errs[0] != nil {
return nil, errs[0]
}
2021-06-12 12:27:13 +08:00
if user == nil {
return nil, nil
2021-06-12 12:27:13 +08:00
}
2021-05-30 13:37:34 +08:00
if user.Password != "" {
user.Password = "***"
}
if !isAdminOrSelf {
if user.AccessSecret != "" {
user.AccessSecret = "***"
}
}
if user.ManagedAccounts != nil {
for _, manageAccount := range user.ManagedAccounts {
manageAccount.Password = "***"
}
}
if user.TotpSecret != "" {
user.TotpSecret = ""
}
if user.RecoveryCodes != nil {
user.RecoveryCodes = nil
}
return user, nil
2021-05-30 13:37:34 +08:00
}
func GetMaskedUsers(users []*User, errs ...error) ([]*User, error) {
if len(errs) > 0 && errs[0] != nil {
return nil, errs[0]
}
var err error
2021-05-30 13:37:34 +08:00
for _, user := range users {
user, err = GetMaskedUser(user, false)
if err != nil {
return nil, err
}
2021-05-30 13:37:34 +08:00
}
return users, nil
2021-05-30 13:37:34 +08:00
}
func GetLastUser(owner string) (*User, error) {
2021-06-17 00:49:02 +08:00
user := User{Owner: owner}
existed, err := ormer.Engine.Desc("created_time", "id").Get(&user)
2021-06-17 00:49:02 +08:00
if err != nil {
return nil, err
2021-06-17 00:49:02 +08:00
}
if existed {
return &user, nil
2021-06-17 00:49:02 +08:00
}
2021-08-07 22:02:56 +08:00
return nil, nil
2021-06-17 00:49:02 +08:00
}
func UpdateUser(id string, user *User, columns []string, isAdmin bool) (bool, error) {
var err error
2021-11-20 15:46:54 +08:00
owner, name := util.GetOwnerAndNameFromIdNoCheck(id)
oldUser, err := getUser(owner, name)
if err != nil {
return false, err
}
2021-08-21 22:52:29 +08:00
if oldUser == nil {
return false, nil
2020-10-20 23:14:03 +08:00
}
if name != user.Name {
err := userChangeTrigger(name, user.Name)
if err != nil {
return false, err
}
}
if user.Password == "***" {
user.Password = oldUser.Password
}
2021-05-08 18:51:37 +08:00
2021-11-20 16:21:15 +08:00
if user.Avatar != oldUser.Avatar && user.Avatar != "" && user.PermanentAvatar != "*" {
user.PermanentAvatar, err = getPermanentAvatarUrl(user.Owner, user.Name, user.Avatar, false)
if err != nil {
return false, err
}
2021-08-21 23:17:33 +08:00
}
2021-12-11 14:45:08 +08:00
if len(columns) == 0 {
columns = []string{
"owner", "display_name", "avatar",
2023-04-15 16:05:33 +08:00
"location", "address", "country_code", "region", "language", "affiliation", "title", "homepage", "bio", "tag", "language", "gender", "birthday", "education", "score", "karma", "ranking", "signup_application",
"is_admin", "is_global_admin", "is_forbidden", "is_deleted", "hash", "is_default_avatar", "properties", "webauthnCredentials", "managedAccounts",
2023-06-16 20:42:15 +08:00
"signin_wrong_times", "last_signin_wrong_time", "groups", "access_key", "access_secret",
2023-05-18 15:58:41 +08:00
"github", "google", "qq", "wechat", "facebook", "dingtalk", "weibo", "gitee", "linkedin", "wecom", "lark", "gitlab", "adfs",
"baidu", "alipay", "casdoor", "infoflow", "apple", "azuread", "slack", "steam", "bilibili", "okta", "douyin", "line", "amazon",
"auth0", "battlenet", "bitbucket", "box", "cloudfoundry", "dailymotion", "deezer", "digitalocean", "discord", "dropbox",
"eveonline", "fitbit", "gitea", "heroku", "influxcloud", "instagram", "intercom", "kakao", "lastfm", "mailru", "meetup",
"microsoftonline", "naver", "nextcloud", "onedrive", "oura", "patreon", "paypal", "salesforce", "shopify", "soundcloud",
"spotify", "strava", "stripe", "tiktok", "tumblr", "twitch", "twitter", "typetalk", "uber", "vk", "wepay", "xero", "yahoo",
"yammer", "yandex", "zoom", "custom",
}
2021-12-11 14:45:08 +08:00
}
if isAdmin {
columns = append(columns, "name", "email", "phone", "country_code")
2022-01-13 23:19:36 +08:00
}
2021-12-11 14:45:08 +08:00
if util.ContainsString(columns, "groups") {
_, err := userEnforcer.UpdateGroupsForUser(user.GetId(), user.Groups)
if err != nil {
return false, err
}
}
affected, err := updateUser(id, user, columns)
2020-10-20 23:14:03 +08:00
if err != nil {
return false, err
2020-10-20 23:14:03 +08:00
}
return affected != 0, nil
2021-05-09 15:44:12 +08:00
}
func updateUser(id string, user *User, columns []string) (int64, error) {
owner, name := util.GetOwnerAndNameFromIdNoCheck(id)
err := user.UpdateUserHash()
if err != nil {
return 0, err
}
affected, err := ormer.Engine.ID(core.PK{owner, name}).Cols(columns...).Update(user)
if err != nil {
return 0, err
}
return affected, nil
}
func UpdateUserForAllFields(id string, user *User) (bool, error) {
var err error
2021-06-09 21:41:27 +08:00
owner, name := util.GetOwnerAndNameFromId(id)
oldUser, err := getUser(owner, name)
if err != nil {
return false, err
}
2021-08-21 22:52:29 +08:00
if oldUser == nil {
return false, nil
2021-06-09 21:41:27 +08:00
}
if name != user.Name {
err := userChangeTrigger(name, user.Name)
if err != nil {
return false, err
}
}
err = user.UpdateUserHash()
if err != nil {
return false, err
}
2021-06-09 21:41:27 +08:00
2021-08-21 23:17:33 +08:00
if user.Avatar != oldUser.Avatar && user.Avatar != "" {
user.PermanentAvatar, err = getPermanentAvatarUrl(user.Owner, user.Name, user.Avatar, false)
if err != nil {
return false, err
}
2021-08-21 23:17:33 +08:00
}
affected, err := ormer.Engine.ID(core.PK{owner, name}).AllCols().Update(user)
2021-06-09 21:41:27 +08:00
if err != nil {
return false, err
2021-06-09 21:41:27 +08:00
}
return affected != 0, nil
2021-06-09 21:41:27 +08:00
}
func AddUser(user *User) (bool, error) {
var err error
2021-06-17 01:12:03 +08:00
if user.Id == "" {
user.Id = util.GenerateId()
}
2021-05-16 21:22:20 +08:00
2021-12-12 19:59:55 +08:00
if user.Owner == "" || user.Name == "" {
return false, nil
2021-12-12 19:59:55 +08:00
}
organization, _ := GetOrganizationByUser(user)
if organization == nil {
return false, nil
}
2023-06-01 22:25:19 +08:00
if user.PasswordType == "" && organization.PasswordType != "" {
user.PasswordType = organization.PasswordType
}
2021-05-16 21:04:26 +08:00
user.UpdateUserPassword(organization)
2021-05-08 18:51:37 +08:00
err = user.UpdateUserHash()
if err != nil {
return false, err
}
2021-05-16 21:22:20 +08:00
user.PreHash = user.Hash
updated, err := user.refreshAvatar()
if err != nil {
return false, err
}
2023-04-16 00:59:20 +08:00
if updated && user.PermanentAvatar != "*" {
user.PermanentAvatar, err = getPermanentAvatarUrl(user.Owner, user.Name, user.Avatar, false)
if err != nil {
return false, err
}
2023-04-16 00:59:20 +08:00
}
2021-08-21 23:17:33 +08:00
count, err := GetUserCount(user.Owner, "", "", "")
if err != nil {
return false, err
}
user.Ranking = int(count + 1)
affected, err := ormer.Engine.Insert(user)
2020-10-20 23:14:03 +08:00
if err != nil {
return false, err
2020-10-20 23:14:03 +08:00
}
return affected != 0, nil
2020-10-20 23:14:03 +08:00
}
func AddUsers(users []*User) (bool, error) {
var err error
2021-05-16 21:04:26 +08:00
if len(users) == 0 {
return false, nil
2021-05-16 21:04:26 +08:00
}
// organization := GetOrganizationByUser(users[0])
2021-05-08 18:51:37 +08:00
for _, user := range users {
2022-01-17 13:26:30 +08:00
// this function is only used for syncer or batch upload, so no need to encrypt the password
// user.UpdateUserPassword(organization)
2021-05-16 21:22:20 +08:00
err = user.UpdateUserHash()
if err != nil {
return false, err
}
2021-05-09 14:54:17 +08:00
user.PreHash = user.Hash
2021-08-21 23:17:33 +08:00
user.PermanentAvatar, err = getPermanentAvatarUrl(user.Owner, user.Name, user.Avatar, true)
if err != nil {
return false, err
}
2021-05-08 18:51:37 +08:00
}
affected, err := ormer.Engine.Insert(users)
2021-05-02 23:06:08 +08:00
if err != nil {
2022-03-13 20:53:05 +08:00
if !strings.Contains(err.Error(), "Duplicate entry") {
return false, err
2022-03-13 20:53:05 +08:00
}
2021-05-02 23:06:08 +08:00
}
return affected != 0, nil
2021-05-02 23:06:08 +08:00
}
func AddUsersInBatch(users []*User) (bool, error) {
2022-09-13 20:31:22 +08:00
batchSize := conf.GetConfigBatchSize()
2021-05-02 23:06:08 +08:00
if len(users) == 0 {
return false, nil
2021-05-02 23:06:08 +08:00
}
affected := false
for i := 0; i < (len(users)-1)/batchSize+1; i++ {
start := i * batchSize
end := (i + 1) * batchSize
if end > len(users) {
end = len(users)
}
tmp := users[start:end]
// TODO: save to log instead of standard output
// fmt.Printf("Add users: [%d - %d].\n", start, end)
if ok, err := AddUsers(tmp); err != nil {
return false, err
} else if ok {
2021-05-02 23:06:08 +08:00
affected = true
}
}
return affected, nil
2021-05-02 23:06:08 +08:00
}
func DeleteUser(user *User) (bool, error) {
// Forced offline the user first
_, err := DeleteSession(util.GetSessionId(user.Owner, user.Name, CasdoorApplication))
if err != nil {
return false, err
}
affected, err := ormer.Engine.ID(core.PK{user.Owner, user.Name}).Delete(&User{})
2020-10-20 23:14:03 +08:00
if err != nil {
return false, err
2020-10-20 23:14:03 +08:00
}
return affected != 0, nil
2020-10-20 23:14:03 +08:00
}
2021-02-14 00:22:24 +08:00
2022-09-18 15:43:49 +08:00
func GetUserInfo(user *User, scope string, aud string, host string) *Userinfo {
_, originBackend := getOriginFromHost(host)
resp := Userinfo{
Sub: user.Id,
Iss: originBackend,
Aud: aud,
}
if strings.Contains(scope, "profile") {
resp.Name = user.Name
resp.DisplayName = user.DisplayName
resp.Avatar = user.Avatar
2023-06-19 11:06:55 +08:00
resp.Groups = user.Groups
}
if strings.Contains(scope, "email") {
resp.Email = user.Email
}
if strings.Contains(scope, "address") {
resp.Address = user.Location
}
if strings.Contains(scope, "phone") {
resp.Phone = user.Phone
}
2022-09-18 15:43:49 +08:00
return &resp
}
func LinkUserAccount(user *User, field string, value string) (bool, error) {
2021-04-27 19:35:40 +08:00
return SetUserField(user, field, value)
}
2021-05-08 22:04:45 +08:00
func (user *User) GetId() string {
return fmt.Sprintf("%s/%s", user.Owner, user.Name)
}
func isUserIdGlobalAdmin(userId string) bool {
return strings.HasPrefix(userId, "built-in/")
}
func ExtendUserWithRolesAndPermissions(user *User) (err error) {
if user == nil {
return
}
user.Permissions, user.Roles, err = GetPermissionsAndRolesByUser(user.GetId())
if err != nil {
return err
}
2023-06-19 09:42:17 +08:00
if user.Groups == nil {
user.Groups = []string{}
}
return
}
func DeleteGroupForUser(user string, group string) (bool, error) {
return userEnforcer.DeleteGroupForUser(user, group)
}
func userChangeTrigger(oldName string, newName string) error {
session := ormer.Engine.NewSession()
defer session.Close()
err := session.Begin()
if err != nil {
return err
}
var roles []*Role
err = ormer.Engine.Find(&roles)
if err != nil {
return err
}
for _, role := range roles {
for j, u := range role.Users {
// u = organization/username
owner, name := util.GetOwnerAndNameFromId(u)
if name == oldName {
role.Users[j] = util.GetId(owner, newName)
}
}
_, err = session.Where("name=?", role.Name).And("owner=?", role.Owner).Update(role)
if err != nil {
return err
}
}
var permissions []*Permission
err = ormer.Engine.Find(&permissions)
if err != nil {
return err
}
for _, permission := range permissions {
for j, u := range permission.Users {
// u = organization/username
owner, name := util.GetOwnerAndNameFromId(u)
if name == oldName {
permission.Users[j] = util.GetId(owner, newName)
}
}
_, err = session.Where("name=?", permission.Name).And("owner=?", permission.Owner).Update(permission)
if err != nil {
return err
}
}
resource := new(Resource)
resource.User = newName
_, err = session.Where("user=?", oldName).Update(resource)
if err != nil {
return err
}
return session.Commit()
}
2023-04-16 00:29:56 +08:00
func (user *User) IsMfaEnabled() bool {
if user == nil {
return false
}
return user.PreferredMfaType != ""
}
func (user *User) GetPreferredMfaProps(masked bool) *MfaProps {
if user == nil || user.PreferredMfaType == "" {
return nil
}
return user.GetMfaProps(user.PreferredMfaType, masked)
}
func AddUserkeys(user *User, isAdmin bool) (bool, error) {
if user == nil {
return false, nil
}
user.AccessKey = util.GenerateId()
2023-06-16 20:42:15 +08:00
user.AccessSecret = util.GenerateId()
return UpdateUser(user.GetId(), user, []string{}, isAdmin)
}
func (user *User) IsApplicationAdmin(application *Application) bool {
if user == nil {
return false
}
return (user.Owner == application.Organization && user.IsAdmin) || user.IsGlobalAdmin
}