* CRT: desktop thread notifications * Fixes go lint * Adds default for desktop CRT notifications * Adds email and push notifications for CRT threads Adds user ids of thread followers with CRT to crtMentions so they will get notified appropriately. * Minor change * Refactor a bit CRTMentions.addMention had a bug on the return and de-duplication. This commit fixes duplicate notifications by looking up if the user is to be notified on CRT on both email and push notifications. * Minor refactor * Changes according to review comments - Fixes adding to followers a user that had explicitly unfollowed a thread. - Simplified send email according to email_threads option - Send mentions and followers in separate arrays via the websocket - Fixes push notifications message for push_threads * Adds a comment on a buggy use case * Updates comment to correct ticket link * Fixes when user notifications is set to all There was a bug where if user had set notifications to all then they would receive desktop notifications even for non following threads. A similar bug existed in push notifications, where if a user has set it to all the threads setting would still be considered. This commit fixes that by adding users to notificationsForCRT StringArray when they have the non thread setting to 'all'. * Fixes notifications to users unfollowing threads Users which had previously explicitly unfollowed a thread should not receive notifications about those threads. * Update store mocks * Fixes push notifications for CRT Push notification about replies for CRT users should have a title of "Reply to Thread". CRT users with global user setting to 'UserNotifyAll' should not get notifications for unfollowed threads. This commit fixes those issues. * Fixes i18n error Co-authored-by: Mattermod <mattermod@users.noreply.github.com>
973 строки
24 KiB
Go
973 строки
24 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package model
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"regexp"
|
|
"sort"
|
|
"strings"
|
|
"unicode/utf8"
|
|
|
|
"golang.org/x/crypto/bcrypt"
|
|
"golang.org/x/text/language"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/services/timezones"
|
|
"github.com/mattermost/mattermost-server/v6/shared/mlog"
|
|
)
|
|
|
|
const (
|
|
Me = "me"
|
|
UserNotifyAll = "all"
|
|
UserNotifyHere = "here"
|
|
UserNotifyMention = "mention"
|
|
UserNotifyNone = "none"
|
|
DesktopNotifyProp = "desktop"
|
|
DesktopSoundNotifyProp = "desktop_sound"
|
|
MarkUnreadNotifyProp = "mark_unread"
|
|
PushNotifyProp = "push"
|
|
PushStatusNotifyProp = "push_status"
|
|
EmailNotifyProp = "email"
|
|
ChannelMentionsNotifyProp = "channel"
|
|
CommentsNotifyProp = "comments"
|
|
MentionKeysNotifyProp = "mention_keys"
|
|
CommentsNotifyNever = "never"
|
|
CommentsNotifyRoot = "root"
|
|
CommentsNotifyAny = "any"
|
|
FirstNameNotifyProp = "first_name"
|
|
AutoResponderActiveNotifyProp = "auto_responder_active"
|
|
AutoResponderMessageNotifyProp = "auto_responder_message"
|
|
DesktopThreadsNotifyProp = "desktop_threads"
|
|
PushThreadsNotifyProp = "push_threads"
|
|
EmailThreadsNotifyProp = "email_threads"
|
|
|
|
DefaultLocale = "en"
|
|
UserAuthServiceEmail = "email"
|
|
|
|
UserEmailMaxLength = 128
|
|
UserNicknameMaxRunes = 64
|
|
UserPositionMaxRunes = 128
|
|
UserFirstNameMaxRunes = 64
|
|
UserLastNameMaxRunes = 64
|
|
UserAuthDataMaxLength = 128
|
|
UserNameMaxLength = 64
|
|
UserNameMinLength = 1
|
|
UserPasswordMaxLength = 72
|
|
UserLocaleMaxLength = 5
|
|
UserTimezoneMaxRunes = 256
|
|
)
|
|
|
|
//msgp:tuple User
|
|
|
|
// User contains the details about the user.
|
|
// This struct's serializer methods are auto-generated. If a new field is added/removed,
|
|
// please run make gen-serialized.
|
|
type User struct {
|
|
Id string `json:"id"`
|
|
CreateAt int64 `json:"create_at,omitempty"`
|
|
UpdateAt int64 `json:"update_at,omitempty"`
|
|
DeleteAt int64 `json:"delete_at"`
|
|
Username string `json:"username"`
|
|
Password string `json:"password,omitempty"`
|
|
AuthData *string `json:"auth_data,omitempty"`
|
|
AuthService string `json:"auth_service"`
|
|
Email string `json:"email"`
|
|
EmailVerified bool `json:"email_verified,omitempty"`
|
|
Nickname string `json:"nickname"`
|
|
FirstName string `json:"first_name"`
|
|
LastName string `json:"last_name"`
|
|
Position string `json:"position"`
|
|
Roles string `json:"roles"`
|
|
AllowMarketing bool `json:"allow_marketing,omitempty"`
|
|
Props StringMap `json:"props,omitempty"`
|
|
NotifyProps StringMap `json:"notify_props,omitempty"`
|
|
LastPasswordUpdate int64 `json:"last_password_update,omitempty"`
|
|
LastPictureUpdate int64 `json:"last_picture_update,omitempty"`
|
|
FailedAttempts int `json:"failed_attempts,omitempty"`
|
|
Locale string `json:"locale"`
|
|
Timezone StringMap `json:"timezone"`
|
|
MfaActive bool `json:"mfa_active,omitempty"`
|
|
MfaSecret string `json:"mfa_secret,omitempty"`
|
|
RemoteId *string `json:"remote_id,omitempty"`
|
|
LastActivityAt int64 `db:"-" json:"last_activity_at,omitempty"`
|
|
IsBot bool `db:"-" json:"is_bot,omitempty"`
|
|
BotDescription string `db:"-" json:"bot_description,omitempty"`
|
|
BotLastIconUpdate int64 `db:"-" json:"bot_last_icon_update,omitempty"`
|
|
TermsOfServiceId string `db:"-" json:"terms_of_service_id,omitempty"`
|
|
TermsOfServiceCreateAt int64 `db:"-" json:"terms_of_service_create_at,omitempty"`
|
|
DisableWelcomeEmail bool `db:"-" json:"disable_welcome_email"`
|
|
}
|
|
|
|
//msgp UserMap
|
|
|
|
// UserMap is a map from a userId to a user object.
|
|
// It is used to generate methods which can be used for fast serialization/de-serialization.
|
|
type UserMap map[string]*User
|
|
|
|
//msgp:ignore UserUpdate
|
|
type UserUpdate struct {
|
|
Old *User
|
|
New *User
|
|
}
|
|
|
|
//msgp:ignore UserPatch
|
|
type UserPatch struct {
|
|
Username *string `json:"username"`
|
|
Password *string `json:"password,omitempty"`
|
|
Nickname *string `json:"nickname"`
|
|
FirstName *string `json:"first_name"`
|
|
LastName *string `json:"last_name"`
|
|
Position *string `json:"position"`
|
|
Email *string `json:"email"`
|
|
Props StringMap `json:"props,omitempty"`
|
|
NotifyProps StringMap `json:"notify_props,omitempty"`
|
|
Locale *string `json:"locale"`
|
|
Timezone StringMap `json:"timezone"`
|
|
RemoteId *string `json:"remote_id"`
|
|
}
|
|
|
|
//msgp:ignore UserAuth
|
|
type UserAuth struct {
|
|
Password string `json:"password,omitempty"` // DEPRECATED: It is not used.
|
|
AuthData *string `json:"auth_data,omitempty"`
|
|
AuthService string `json:"auth_service,omitempty"`
|
|
}
|
|
|
|
//msgp:ignore UserForIndexing
|
|
type UserForIndexing struct {
|
|
Id string `json:"id"`
|
|
Username string `json:"username"`
|
|
Nickname string `json:"nickname"`
|
|
FirstName string `json:"first_name"`
|
|
LastName string `json:"last_name"`
|
|
Roles string `json:"roles"`
|
|
CreateAt int64 `json:"create_at"`
|
|
DeleteAt int64 `json:"delete_at"`
|
|
TeamsIds []string `json:"team_id"`
|
|
ChannelsIds []string `json:"channel_id"`
|
|
}
|
|
|
|
//msgp:ignore ViewUsersRestrictions
|
|
type ViewUsersRestrictions struct {
|
|
Teams []string
|
|
Channels []string
|
|
}
|
|
|
|
func (r *ViewUsersRestrictions) Hash() string {
|
|
if r == nil {
|
|
return ""
|
|
}
|
|
ids := append(r.Teams, r.Channels...)
|
|
sort.Strings(ids)
|
|
hash := sha256.New()
|
|
hash.Write([]byte(strings.Join(ids, "")))
|
|
return fmt.Sprintf("%x", hash.Sum(nil))
|
|
}
|
|
|
|
//msgp:ignore UserSlice
|
|
type UserSlice []*User
|
|
|
|
func (u UserSlice) Usernames() []string {
|
|
usernames := []string{}
|
|
for _, user := range u {
|
|
usernames = append(usernames, user.Username)
|
|
}
|
|
sort.Strings(usernames)
|
|
return usernames
|
|
}
|
|
|
|
func (u UserSlice) IDs() []string {
|
|
ids := []string{}
|
|
for _, user := range u {
|
|
ids = append(ids, user.Id)
|
|
}
|
|
return ids
|
|
}
|
|
|
|
func (u UserSlice) FilterWithoutBots() UserSlice {
|
|
var matches []*User
|
|
|
|
for _, user := range u {
|
|
if !user.IsBot {
|
|
matches = append(matches, user)
|
|
}
|
|
}
|
|
return UserSlice(matches)
|
|
}
|
|
|
|
func (u UserSlice) FilterByActive(active bool) UserSlice {
|
|
var matches []*User
|
|
|
|
for _, user := range u {
|
|
if user.DeleteAt == 0 && active {
|
|
matches = append(matches, user)
|
|
} else if user.DeleteAt != 0 && !active {
|
|
matches = append(matches, user)
|
|
}
|
|
}
|
|
return UserSlice(matches)
|
|
}
|
|
|
|
func (u UserSlice) FilterByID(ids []string) UserSlice {
|
|
var matches []*User
|
|
for _, user := range u {
|
|
for _, id := range ids {
|
|
if id == user.Id {
|
|
matches = append(matches, user)
|
|
}
|
|
}
|
|
}
|
|
return UserSlice(matches)
|
|
}
|
|
|
|
func (u UserSlice) FilterWithoutID(ids []string) UserSlice {
|
|
var keep []*User
|
|
for _, user := range u {
|
|
present := false
|
|
for _, id := range ids {
|
|
if id == user.Id {
|
|
present = true
|
|
}
|
|
}
|
|
if !present {
|
|
keep = append(keep, user)
|
|
}
|
|
}
|
|
return UserSlice(keep)
|
|
}
|
|
|
|
func (u *User) DeepCopy() *User {
|
|
copyUser := *u
|
|
if u.AuthData != nil {
|
|
copyUser.AuthData = NewString(*u.AuthData)
|
|
}
|
|
if u.Props != nil {
|
|
copyUser.Props = CopyStringMap(u.Props)
|
|
}
|
|
if u.NotifyProps != nil {
|
|
copyUser.NotifyProps = CopyStringMap(u.NotifyProps)
|
|
}
|
|
if u.Timezone != nil {
|
|
copyUser.Timezone = CopyStringMap(u.Timezone)
|
|
}
|
|
return ©User
|
|
}
|
|
|
|
// IsValid validates the user and returns an error if it isn't configured
|
|
// correctly.
|
|
func (u *User) IsValid() *AppError {
|
|
|
|
if !IsValidId(u.Id) {
|
|
return InvalidUserError("id", "")
|
|
}
|
|
|
|
if u.CreateAt == 0 {
|
|
return InvalidUserError("create_at", u.Id)
|
|
}
|
|
|
|
if u.UpdateAt == 0 {
|
|
return InvalidUserError("update_at", u.Id)
|
|
}
|
|
|
|
if u.IsRemote() {
|
|
if !IsValidUsernameAllowRemote(u.Username) {
|
|
return InvalidUserError("username", u.Id)
|
|
}
|
|
} else {
|
|
if !IsValidUsername(u.Username) {
|
|
return InvalidUserError("username", u.Id)
|
|
}
|
|
}
|
|
|
|
if len(u.Email) > UserEmailMaxLength || u.Email == "" || !IsValidEmail(u.Email) {
|
|
return InvalidUserError("email", u.Id)
|
|
}
|
|
|
|
if utf8.RuneCountInString(u.Nickname) > UserNicknameMaxRunes {
|
|
return InvalidUserError("nickname", u.Id)
|
|
}
|
|
|
|
if utf8.RuneCountInString(u.Position) > UserPositionMaxRunes {
|
|
return InvalidUserError("position", u.Id)
|
|
}
|
|
|
|
if utf8.RuneCountInString(u.FirstName) > UserFirstNameMaxRunes {
|
|
return InvalidUserError("first_name", u.Id)
|
|
}
|
|
|
|
if utf8.RuneCountInString(u.LastName) > UserLastNameMaxRunes {
|
|
return InvalidUserError("last_name", u.Id)
|
|
}
|
|
|
|
if u.AuthData != nil && len(*u.AuthData) > UserAuthDataMaxLength {
|
|
return InvalidUserError("auth_data", u.Id)
|
|
}
|
|
|
|
if u.AuthData != nil && *u.AuthData != "" && u.AuthService == "" {
|
|
return InvalidUserError("auth_data_type", u.Id)
|
|
}
|
|
|
|
if u.Password != "" && u.AuthData != nil && *u.AuthData != "" {
|
|
return InvalidUserError("auth_data_pwd", u.Id)
|
|
}
|
|
|
|
if len(u.Password) > UserPasswordMaxLength {
|
|
return InvalidUserError("password_limit", u.Id)
|
|
}
|
|
|
|
if !IsValidLocale(u.Locale) {
|
|
return InvalidUserError("locale", u.Id)
|
|
}
|
|
|
|
if len(u.Timezone) > 0 {
|
|
if tzJSON, err := json.Marshal(u.Timezone); err != nil {
|
|
return NewAppError("User.IsValid", "model.user.is_valid.marshal.app_error", nil, err.Error(), http.StatusInternalServerError)
|
|
} else if utf8.RuneCount(tzJSON) > UserTimezoneMaxRunes {
|
|
return InvalidUserError("timezone_limit", u.Id)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func InvalidUserError(fieldName string, userId string) *AppError {
|
|
id := fmt.Sprintf("model.user.is_valid.%s.app_error", fieldName)
|
|
details := ""
|
|
if userId != "" {
|
|
details = "user_id=" + userId
|
|
}
|
|
return NewAppError("User.IsValid", id, nil, details, http.StatusBadRequest)
|
|
}
|
|
|
|
func NormalizeUsername(username string) string {
|
|
return strings.ToLower(username)
|
|
}
|
|
|
|
func NormalizeEmail(email string) string {
|
|
return strings.ToLower(email)
|
|
}
|
|
|
|
// PreSave will set the Id and Username if missing. It will also fill
|
|
// in the CreateAt, UpdateAt times. It will also hash the password. It should
|
|
// be run before saving the user to the db.
|
|
func (u *User) PreSave() {
|
|
if u.Id == "" {
|
|
u.Id = NewId()
|
|
}
|
|
|
|
if u.Username == "" {
|
|
u.Username = NewId()
|
|
}
|
|
|
|
if u.AuthData != nil && *u.AuthData == "" {
|
|
u.AuthData = nil
|
|
}
|
|
|
|
u.Username = SanitizeUnicode(u.Username)
|
|
u.FirstName = SanitizeUnicode(u.FirstName)
|
|
u.LastName = SanitizeUnicode(u.LastName)
|
|
u.Nickname = SanitizeUnicode(u.Nickname)
|
|
|
|
u.Username = NormalizeUsername(u.Username)
|
|
u.Email = NormalizeEmail(u.Email)
|
|
|
|
u.CreateAt = GetMillis()
|
|
u.UpdateAt = u.CreateAt
|
|
|
|
u.LastPasswordUpdate = u.CreateAt
|
|
|
|
u.MfaActive = false
|
|
|
|
if u.Locale == "" {
|
|
u.Locale = DefaultLocale
|
|
}
|
|
|
|
if u.Props == nil {
|
|
u.Props = make(map[string]string)
|
|
}
|
|
|
|
if u.NotifyProps == nil || len(u.NotifyProps) == 0 {
|
|
u.SetDefaultNotifications()
|
|
}
|
|
|
|
if u.Timezone == nil {
|
|
u.Timezone = timezones.DefaultUserTimezone()
|
|
}
|
|
|
|
if u.Password != "" {
|
|
u.Password = HashPassword(u.Password)
|
|
}
|
|
}
|
|
|
|
// PreUpdate should be run before updating the user in the db.
|
|
func (u *User) PreUpdate() {
|
|
u.Username = SanitizeUnicode(u.Username)
|
|
u.FirstName = SanitizeUnicode(u.FirstName)
|
|
u.LastName = SanitizeUnicode(u.LastName)
|
|
u.Nickname = SanitizeUnicode(u.Nickname)
|
|
u.BotDescription = SanitizeUnicode(u.BotDescription)
|
|
|
|
u.Username = NormalizeUsername(u.Username)
|
|
u.Email = NormalizeEmail(u.Email)
|
|
u.UpdateAt = GetMillis()
|
|
|
|
u.FirstName = SanitizeUnicode(u.FirstName)
|
|
u.LastName = SanitizeUnicode(u.LastName)
|
|
u.Nickname = SanitizeUnicode(u.Nickname)
|
|
u.BotDescription = SanitizeUnicode(u.BotDescription)
|
|
|
|
if u.AuthData != nil && *u.AuthData == "" {
|
|
u.AuthData = nil
|
|
}
|
|
|
|
if u.NotifyProps == nil || len(u.NotifyProps) == 0 {
|
|
u.SetDefaultNotifications()
|
|
} else if _, ok := u.NotifyProps[MentionKeysNotifyProp]; ok {
|
|
// Remove any blank mention keys
|
|
splitKeys := strings.Split(u.NotifyProps[MentionKeysNotifyProp], ",")
|
|
goodKeys := []string{}
|
|
for _, key := range splitKeys {
|
|
if key != "" {
|
|
goodKeys = append(goodKeys, strings.ToLower(key))
|
|
}
|
|
}
|
|
u.NotifyProps[MentionKeysNotifyProp] = strings.Join(goodKeys, ",")
|
|
}
|
|
}
|
|
|
|
func (u *User) SetDefaultNotifications() {
|
|
u.NotifyProps = make(map[string]string)
|
|
u.NotifyProps[EmailNotifyProp] = "true"
|
|
u.NotifyProps[PushNotifyProp] = UserNotifyMention
|
|
u.NotifyProps[DesktopNotifyProp] = UserNotifyMention
|
|
u.NotifyProps[DesktopSoundNotifyProp] = "true"
|
|
u.NotifyProps[MentionKeysNotifyProp] = ""
|
|
u.NotifyProps[ChannelMentionsNotifyProp] = "true"
|
|
u.NotifyProps[PushStatusNotifyProp] = StatusAway
|
|
u.NotifyProps[CommentsNotifyProp] = CommentsNotifyNever
|
|
u.NotifyProps[FirstNameNotifyProp] = "false"
|
|
u.NotifyProps[DesktopThreadsNotifyProp] = UserNotifyAll
|
|
u.NotifyProps[EmailThreadsNotifyProp] = UserNotifyAll
|
|
u.NotifyProps[PushThreadsNotifyProp] = UserNotifyAll
|
|
}
|
|
|
|
func (u *User) UpdateMentionKeysFromUsername(oldUsername string) {
|
|
nonUsernameKeys := []string{}
|
|
for _, key := range u.GetMentionKeys() {
|
|
if key != oldUsername && key != "@"+oldUsername {
|
|
nonUsernameKeys = append(nonUsernameKeys, key)
|
|
}
|
|
}
|
|
|
|
u.NotifyProps[MentionKeysNotifyProp] = ""
|
|
if len(nonUsernameKeys) > 0 {
|
|
u.NotifyProps[MentionKeysNotifyProp] += "," + strings.Join(nonUsernameKeys, ",")
|
|
}
|
|
}
|
|
|
|
func (u *User) GetMentionKeys() []string {
|
|
var keys []string
|
|
|
|
for _, key := range strings.Split(u.NotifyProps[MentionKeysNotifyProp], ",") {
|
|
trimmedKey := strings.TrimSpace(key)
|
|
|
|
if trimmedKey == "" {
|
|
continue
|
|
}
|
|
|
|
keys = append(keys, trimmedKey)
|
|
}
|
|
|
|
return keys
|
|
}
|
|
|
|
func (u *User) Patch(patch *UserPatch) {
|
|
if patch.Username != nil {
|
|
u.Username = *patch.Username
|
|
}
|
|
|
|
if patch.Nickname != nil {
|
|
u.Nickname = *patch.Nickname
|
|
}
|
|
|
|
if patch.FirstName != nil {
|
|
u.FirstName = *patch.FirstName
|
|
}
|
|
|
|
if patch.LastName != nil {
|
|
u.LastName = *patch.LastName
|
|
}
|
|
|
|
if patch.Position != nil {
|
|
u.Position = *patch.Position
|
|
}
|
|
|
|
if patch.Email != nil {
|
|
u.Email = *patch.Email
|
|
}
|
|
|
|
if patch.Props != nil {
|
|
u.Props = patch.Props
|
|
}
|
|
|
|
if patch.NotifyProps != nil {
|
|
u.NotifyProps = patch.NotifyProps
|
|
}
|
|
|
|
if patch.Locale != nil {
|
|
u.Locale = *patch.Locale
|
|
}
|
|
|
|
if patch.Timezone != nil {
|
|
u.Timezone = patch.Timezone
|
|
}
|
|
|
|
if patch.RemoteId != nil {
|
|
u.RemoteId = patch.RemoteId
|
|
}
|
|
}
|
|
|
|
// ToJson convert a User to a json string
|
|
func (u *User) ToJson() string {
|
|
b, _ := json.Marshal(u)
|
|
return string(b)
|
|
}
|
|
|
|
func (u *UserPatch) ToJson() string {
|
|
b, _ := json.Marshal(u)
|
|
return string(b)
|
|
}
|
|
|
|
func (u *UserAuth) ToJson() string {
|
|
b, _ := json.Marshal(u)
|
|
return string(b)
|
|
}
|
|
|
|
// Generate a valid strong etag so the browser can cache the results
|
|
func (u *User) Etag(showFullName, showEmail bool) string {
|
|
return Etag(u.Id, u.UpdateAt, u.TermsOfServiceId, u.TermsOfServiceCreateAt, showFullName, showEmail, u.BotLastIconUpdate)
|
|
}
|
|
|
|
// Remove any private data from the user object
|
|
func (u *User) Sanitize(options map[string]bool) {
|
|
u.Password = ""
|
|
u.AuthData = NewString("")
|
|
u.MfaSecret = ""
|
|
|
|
if len(options) != 0 && !options["email"] {
|
|
u.Email = ""
|
|
}
|
|
if len(options) != 0 && !options["fullname"] {
|
|
u.FirstName = ""
|
|
u.LastName = ""
|
|
}
|
|
if len(options) != 0 && !options["passwordupdate"] {
|
|
u.LastPasswordUpdate = 0
|
|
}
|
|
if len(options) != 0 && !options["authservice"] {
|
|
u.AuthService = ""
|
|
}
|
|
}
|
|
|
|
// Remove any input data from the user object that is not user controlled
|
|
func (u *User) SanitizeInput(isAdmin bool) {
|
|
if !isAdmin {
|
|
u.AuthData = NewString("")
|
|
u.AuthService = ""
|
|
u.EmailVerified = false
|
|
}
|
|
u.LastPasswordUpdate = 0
|
|
u.LastPictureUpdate = 0
|
|
u.FailedAttempts = 0
|
|
u.MfaActive = false
|
|
u.MfaSecret = ""
|
|
u.Email = strings.TrimSpace(u.Email)
|
|
}
|
|
|
|
func (u *User) ClearNonProfileFields() {
|
|
u.Password = ""
|
|
u.AuthData = NewString("")
|
|
u.MfaSecret = ""
|
|
u.EmailVerified = false
|
|
u.AllowMarketing = false
|
|
u.NotifyProps = StringMap{}
|
|
u.LastPasswordUpdate = 0
|
|
u.FailedAttempts = 0
|
|
}
|
|
|
|
func (u *User) SanitizeProfile(options map[string]bool) {
|
|
u.ClearNonProfileFields()
|
|
|
|
u.Sanitize(options)
|
|
}
|
|
|
|
func (u *User) MakeNonNil() {
|
|
if u.Props == nil {
|
|
u.Props = make(map[string]string)
|
|
}
|
|
|
|
if u.NotifyProps == nil {
|
|
u.NotifyProps = make(map[string]string)
|
|
}
|
|
}
|
|
|
|
func (u *User) AddNotifyProp(key string, value string) {
|
|
u.MakeNonNil()
|
|
|
|
u.NotifyProps[key] = value
|
|
}
|
|
|
|
func (u *User) SetCustomStatus(cs *CustomStatus) {
|
|
u.MakeNonNil()
|
|
u.Props[UserPropsKeyCustomStatus] = cs.ToJson()
|
|
}
|
|
|
|
func (u *User) ClearCustomStatus() {
|
|
u.MakeNonNil()
|
|
u.Props[UserPropsKeyCustomStatus] = ""
|
|
}
|
|
|
|
func (u *User) GetFullName() string {
|
|
if u.FirstName != "" && u.LastName != "" {
|
|
return u.FirstName + " " + u.LastName
|
|
} else if u.FirstName != "" {
|
|
return u.FirstName
|
|
} else if u.LastName != "" {
|
|
return u.LastName
|
|
} else {
|
|
return ""
|
|
}
|
|
}
|
|
|
|
func (u *User) getDisplayName(baseName, nameFormat string) string {
|
|
displayName := baseName
|
|
|
|
if nameFormat == ShowNicknameFullName {
|
|
if u.Nickname != "" {
|
|
displayName = u.Nickname
|
|
} else if fullName := u.GetFullName(); fullName != "" {
|
|
displayName = fullName
|
|
}
|
|
} else if nameFormat == ShowFullName {
|
|
if fullName := u.GetFullName(); fullName != "" {
|
|
displayName = fullName
|
|
}
|
|
}
|
|
|
|
return displayName
|
|
}
|
|
|
|
func (u *User) GetDisplayName(nameFormat string) string {
|
|
displayName := u.Username
|
|
|
|
return u.getDisplayName(displayName, nameFormat)
|
|
}
|
|
|
|
func (u *User) GetDisplayNameWithPrefix(nameFormat, prefix string) string {
|
|
displayName := prefix + u.Username
|
|
|
|
return u.getDisplayName(displayName, nameFormat)
|
|
}
|
|
|
|
func (u *User) GetRoles() []string {
|
|
return strings.Fields(u.Roles)
|
|
}
|
|
|
|
func (u *User) GetRawRoles() string {
|
|
return u.Roles
|
|
}
|
|
|
|
func IsValidUserRoles(userRoles string) bool {
|
|
|
|
roles := strings.Fields(userRoles)
|
|
|
|
for _, r := range roles {
|
|
if !IsValidRoleName(r) {
|
|
return false
|
|
}
|
|
}
|
|
|
|
// Exclude just the system_admin role explicitly to prevent mistakes
|
|
if len(roles) == 1 && roles[0] == "system_admin" {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// Make sure you acually want to use this function. In context.go there are functions to check permissions
|
|
// This function should not be used to check permissions.
|
|
func (u *User) IsGuest() bool {
|
|
return IsInRole(u.Roles, SystemGuestRoleId)
|
|
}
|
|
|
|
func (u *User) IsSystemAdmin() bool {
|
|
return IsInRole(u.Roles, SystemAdminRoleId)
|
|
}
|
|
|
|
// Make sure you acually want to use this function. In context.go there are functions to check permissions
|
|
// This function should not be used to check permissions.
|
|
func (u *User) IsInRole(inRole string) bool {
|
|
return IsInRole(u.Roles, inRole)
|
|
}
|
|
|
|
// Make sure you acually want to use this function. In context.go there are functions to check permissions
|
|
// This function should not be used to check permissions.
|
|
func IsInRole(userRoles string, inRole string) bool {
|
|
roles := strings.Split(userRoles, " ")
|
|
|
|
for _, r := range roles {
|
|
if r == inRole {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (u *User) IsSSOUser() bool {
|
|
return u.AuthService != "" && u.AuthService != UserAuthServiceEmail
|
|
}
|
|
|
|
func (u *User) IsOAuthUser() bool {
|
|
return u.AuthService == ServiceGitlab ||
|
|
u.AuthService == ServiceGoogle ||
|
|
u.AuthService == ServiceOffice365 ||
|
|
u.AuthService == ServiceOpenid
|
|
}
|
|
|
|
func (u *User) IsLDAPUser() bool {
|
|
return u.AuthService == UserAuthServiceLdap
|
|
}
|
|
|
|
func (u *User) IsSAMLUser() bool {
|
|
return u.AuthService == UserAuthServiceSaml
|
|
}
|
|
|
|
func (u *User) GetPreferredTimezone() string {
|
|
return GetPreferredTimezone(u.Timezone)
|
|
}
|
|
|
|
// IsRemote returns true if the user belongs to a remote cluster (has RemoteId).
|
|
func (u *User) IsRemote() bool {
|
|
return u.RemoteId != nil && *u.RemoteId != ""
|
|
}
|
|
|
|
// GetRemoteID returns the remote id for this user or "" if not a remote user.
|
|
func (u *User) GetRemoteID() string {
|
|
if u.RemoteId != nil {
|
|
return *u.RemoteId
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// GetProp fetches a prop value by name.
|
|
func (u *User) GetProp(name string) (string, bool) {
|
|
val, ok := u.Props[name]
|
|
return val, ok
|
|
}
|
|
|
|
// SetProp sets a prop value by name, creating the map if nil.
|
|
// Not thread safe.
|
|
func (u *User) SetProp(name string, value string) {
|
|
if u.Props == nil {
|
|
u.Props = make(map[string]string)
|
|
}
|
|
u.Props[name] = value
|
|
}
|
|
|
|
func (u *User) ToPatch() *UserPatch {
|
|
return &UserPatch{
|
|
Username: &u.Username, Password: &u.Password,
|
|
Nickname: &u.Nickname, FirstName: &u.FirstName, LastName: &u.LastName,
|
|
Position: &u.Position, Email: &u.Email,
|
|
Props: u.Props, NotifyProps: u.NotifyProps,
|
|
Locale: &u.Locale, Timezone: u.Timezone,
|
|
}
|
|
}
|
|
|
|
func (u *UserPatch) SetField(fieldName string, fieldValue string) {
|
|
switch fieldName {
|
|
case "FirstName":
|
|
u.FirstName = &fieldValue
|
|
case "LastName":
|
|
u.LastName = &fieldValue
|
|
case "Nickname":
|
|
u.Nickname = &fieldValue
|
|
case "Email":
|
|
u.Email = &fieldValue
|
|
case "Position":
|
|
u.Position = &fieldValue
|
|
case "Username":
|
|
u.Username = &fieldValue
|
|
}
|
|
}
|
|
|
|
// UserFromJson will decode the input and return a User
|
|
func UserFromJson(data io.Reader) *User {
|
|
var user *User
|
|
json.NewDecoder(data).Decode(&user)
|
|
return user
|
|
}
|
|
|
|
func UserPatchFromJson(data io.Reader) *UserPatch {
|
|
var user *UserPatch
|
|
json.NewDecoder(data).Decode(&user)
|
|
return user
|
|
}
|
|
|
|
func UserAuthFromJson(data io.Reader) *UserAuth {
|
|
var user *UserAuth
|
|
json.NewDecoder(data).Decode(&user)
|
|
return user
|
|
}
|
|
|
|
func UserMapToJson(u map[string]*User) string {
|
|
b, _ := json.Marshal(u)
|
|
return string(b)
|
|
}
|
|
|
|
func UserMapFromJson(data io.Reader) map[string]*User {
|
|
var users map[string]*User
|
|
json.NewDecoder(data).Decode(&users)
|
|
return users
|
|
}
|
|
|
|
func UserListToJson(u []*User) string {
|
|
b, _ := json.Marshal(u)
|
|
return string(b)
|
|
}
|
|
|
|
func UserListFromJson(data io.Reader) []*User {
|
|
var users []*User
|
|
json.NewDecoder(data).Decode(&users)
|
|
return users
|
|
}
|
|
|
|
// HashPassword generates a hash using the bcrypt.GenerateFromPassword
|
|
func HashPassword(password string) string {
|
|
hash, err := bcrypt.GenerateFromPassword([]byte(password), 10)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return string(hash)
|
|
}
|
|
|
|
var validUsernameChars = regexp.MustCompile(`^[a-z0-9\.\-_]+$`)
|
|
var validUsernameCharsForRemote = regexp.MustCompile(`^[a-z0-9\.\-_:]+$`)
|
|
|
|
var restrictedUsernames = map[string]struct{}{
|
|
"all": {},
|
|
"channel": {},
|
|
"matterbot": {},
|
|
"system": {},
|
|
}
|
|
|
|
func IsValidUsername(s string) bool {
|
|
if len(s) < UserNameMinLength || len(s) > UserNameMaxLength {
|
|
return false
|
|
}
|
|
|
|
if !validUsernameChars.MatchString(s) {
|
|
return false
|
|
}
|
|
|
|
_, found := restrictedUsernames[s]
|
|
return !found
|
|
}
|
|
|
|
func IsValidUsernameAllowRemote(s string) bool {
|
|
if len(s) < UserNameMinLength || len(s) > UserNameMaxLength {
|
|
return false
|
|
}
|
|
|
|
if !validUsernameCharsForRemote.MatchString(s) {
|
|
return false
|
|
}
|
|
|
|
_, found := restrictedUsernames[s]
|
|
return !found
|
|
}
|
|
|
|
func CleanUsername(username string) string {
|
|
s := NormalizeUsername(strings.Replace(username, " ", "-", -1))
|
|
|
|
for _, value := range reservedName {
|
|
if s == value {
|
|
s = strings.Replace(s, value, "", -1)
|
|
}
|
|
}
|
|
|
|
s = strings.TrimSpace(s)
|
|
|
|
for _, c := range s {
|
|
char := fmt.Sprintf("%c", c)
|
|
if !validUsernameChars.MatchString(char) {
|
|
s = strings.Replace(s, char, "-", -1)
|
|
}
|
|
}
|
|
|
|
s = strings.Trim(s, "-")
|
|
|
|
if !IsValidUsername(s) {
|
|
s = "a" + NewId()
|
|
mlog.Warn("Generating new username since provided username was invalid",
|
|
mlog.String("provided_username", username), mlog.String("new_username", s))
|
|
}
|
|
|
|
return s
|
|
}
|
|
|
|
func IsValidLocale(locale string) bool {
|
|
if locale != "" {
|
|
if len(locale) > UserLocaleMaxLength {
|
|
return false
|
|
} else if _, err := language.Parse(locale); err != nil {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
//msgp:ignore UserWithGroups
|
|
type UserWithGroups struct {
|
|
User
|
|
GroupIDs *string `json:"-"`
|
|
Groups []*Group `json:"groups"`
|
|
SchemeGuest bool `json:"scheme_guest"`
|
|
SchemeUser bool `json:"scheme_user"`
|
|
SchemeAdmin bool `json:"scheme_admin"`
|
|
}
|
|
|
|
func (u *UserWithGroups) GetGroupIDs() []string {
|
|
if u.GroupIDs == nil {
|
|
return nil
|
|
}
|
|
trimmed := strings.TrimSpace(*u.GroupIDs)
|
|
if trimmed == "" {
|
|
return nil
|
|
}
|
|
return strings.Split(trimmed, ",")
|
|
}
|
|
|
|
//msgp:ignore UsersWithGroupsAndCount
|
|
type UsersWithGroupsAndCount struct {
|
|
Users []*UserWithGroups `json:"users"`
|
|
Count int64 `json:"total_count"`
|
|
}
|
|
|
|
func UsersWithGroupsAndCountFromJson(data io.Reader) *UsersWithGroupsAndCount {
|
|
uwg := &UsersWithGroupsAndCount{}
|
|
bodyBytes, _ := ioutil.ReadAll(data)
|
|
json.Unmarshal(bodyBytes, uwg)
|
|
return uwg
|
|
}
|