mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
* Add atomic login-attempt counter primitives to UserStore
Two new store methods back the upcoming switch from a global
per-node mutex to per-user atomic slot claiming:
TryIncrementFailedPasswordAttempts(userID, maxAttempts) (bool, error)
UPDATE Users SET FailedAttempts = FailedAttempts + 1
WHERE Id = ? AND FailedAttempts < maxAttempts
Returns true when a slot was claimed (rows affected == 1) and
false when the cap was already reached. The conditional UPDATE
serialises concurrent attempts on the same user via the row
lock, so the cap is enforced without any application-level
locking and without serialising attempts across users.
DecrementFailedPasswordAttempts(userID) error
UPDATE Users SET FailedAttempts = FailedAttempts - 1
WHERE Id = ? AND FailedAttempts > 0
Releases a slot previously claimed by TryIncrement when the
in-flight authentication turns out not to be a credential
failure. The conditional UPDATE means concurrent decrements
cannot underflow.
Storetest covers both primitives: claim-below-cap, reject-at-cap,
reject-above-cap, no-op for unknown user, and a 50-goroutine
concurrent test with a start barrier asserting exactly
maxAttempts slots are ever claimed and that decrement clamps at
zero under contention.
The testify mock is regenerated here so the storetest package
that returns *mocks.UserStore as a store.UserStore still satisfies
the interface; the wrapper layers are regenerated in the next
commit.
------
AI assisted commit
* Regenerate store layers for the new primitives
Pick up TryIncrementFailedPasswordAttempts and
DecrementFailedPasswordAttempts in every generated wrapper:
- retrylayer: retry on repeatable errors using the standard
three-attempt loop.
- timerlayer: record store-method duration metrics under
UserStore.TryIncrementFailedPasswordAttempts and
UserStore.DecrementFailedPasswordAttempts.
- localcachelayer: invalidate the profile cache only after the
underlying conditional UPDATE actually changes a row; an
at-cap no-op return on TryIncrement no longer produces
unnecessary cluster invalidation traffic.
------
AI assisted commit
* Drop login-attempt mutex; use per-user slot claiming
Replace the global per-node mutex that serialised every login
attempt with the database-side atomic slot machine added on the
Users row. Each of the three authentication entry points now
pre-claims a slot via TryIncrementFailedPasswordAttempts before
running the expensive password / LDAP / MFA check, and releases
the slot when the failure path is not a real credential mismatch:
- CheckPasswordAndAllCriteria (email/password): refunds the
slot on backend errors during the password check (malformed
stored hash, hasher misc failure, password-migration write
failure) so a transient infra issue cannot ratchet
FailedAttempts to a lockout for a user with valid credentials;
refunds on the MFA pre-flight probe (empty mfaToken on an
MFA-enabled user) so the probe is not counted as a real
attempt.
- DoubleCheckPassword: same backend-error refund predicate.
- checkLdapUserPasswordAndAllCriteria: pre-claims only for
existing users (first-time LDAP users have no local row to
claim against); refunds non-credential DoLogin errors (server
unreachable, transient) so an LDAP outage cannot lock out
everyone; refunds the MFA pre-flight probe; for first-time
users, explicitly bumps the counter via UpdateFailedPasswordAttempts
on a real bad-password or bad-MFA attempt, matching the
pre-refactor counting behaviour.
If the refund itself fails the underlying authentication error is
preserved and returned to the caller (the failure is logged); a
leaked slot is annoying, but masking the real failure with a
generic store 500 would be a clear observability regression.
Cluster-wide behaviour also changes: the previous design honoured
MaximumLoginAttempts per node, so an n-node cluster effectively
permitted n * MaximumLoginAttempts attempts. The cap is now
enforced globally.
------
AI assisted commit
* Cover app-layer behaviors of the new login slot machine
The store-layer tests already exercise TryIncrement and Decrement
under concurrency and at the cap boundary. The new behavioural
contracts at the app layer were not covered, so a regression that
flipped a refund predicate, a probe condition, or a first-time
LDAP path would have slipped through type checking and existing
unit tests.
Add tests around the three callers of the new path:
- CheckPasswordAndAllCriteria: an MFA pre-flight probe (empty
token) does not consume a slot; a real attempt with a wrong
non-empty token does; a backend error during the password
check (malformed stored hash) refunds the slot; the happy
path also asserts FailedAttempts resets to zero.
- DoubleCheckPassword: gets its first test coverage, covering
the happy path, rate-limit rejection once max attempts is
reached, and the backend-error refund path.
- checkLdapUserPasswordAndAllCriteria: covers paths the table
loop did not exercise, first-time LDAP user with a bad
password (uses GetUserByAuth to reach the freshly created
row), first-time LDAP user with a wrong MFA token, existing
LDAP user with a non-credential DoLogin error (slot
refunded), and the existing LDAP user MFA pre-flight probe
(slot refunded).
------
AI assisted commit
* Address coderabbit review
------
AI assisted commit
* Fix race in first-time LDAP failed-attempt counter
For first-time LDAP users we have no local row to pre-claim, so
the bad-password and bad-MFA branches fell back to an absolute
UpdateFailedPasswordAttempts(id, ldapUser.FailedAttempts+1) based
on a snapshot from GetUserByAuth. Concurrent first-attempt
requests for the same user could all read FailedAttempts == 0 and
all write 1, losing increments. As a secondary issue the absolute
set did not enforce MaximumLoginAttempts, so the counter could
also drift past the cap.
Switch both branches to TryIncrementFailedPasswordAttempts, the
atomic conditional UPDATE already used on every other path. The
row lock serialises concurrent increments and the predicate caps
at MaximumLoginAttempts.
A new concurrent storetest-style subtest runs
3 * maxFailedLoginAttempts goroutines through the first-time
bad-password path against the same fresh LDAP row and asserts
FailedAttempts lands at exactly maxFailedLoginAttempts. Against
the previous absolute-set implementation the test fails (observed
FailedAttempts = 4 with maxFailedLoginAttempts = 3, either a lost
increment or a cap overshoot).
The first-time bad-password branch also switches from a wrapped
500 return on store error to log-and-continue, matching the rest
of the file's refund/probe error handling: the underlying LDAP
authentication failure is the more useful error for the caller.
------
AI assisted commit
* Address review comments
------
AI assisted commit
---------
Co-authored-by: Mattermost Build <[email protected]>
2597 lines
80 KiB
Go
2597 lines
80 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package sqlstore
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
"unicode/utf8"
|
|
|
|
sq "github.com/mattermost/squirrel"
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
"github.com/mattermost/mattermost/server/v8/channels/app/password/hashers"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
"github.com/mattermost/mattermost/server/v8/einterfaces"
|
|
)
|
|
|
|
const (
|
|
MaxGroupChannelsForProfiles = 50
|
|
ContentReviewerSearchLimit = 50
|
|
)
|
|
|
|
var (
|
|
UserSearchTypeNamesNoFullName = []string{"Username", "Nickname"}
|
|
UserSearchTypeNames = []string{"Username", "FirstName", "LastName", "Nickname"}
|
|
UserSearchTypeAllNoFullName = []string{"Username", "Nickname", "Email"}
|
|
UserSearchTypeAll = []string{"Username", "FirstName", "LastName", "Nickname", "Email"}
|
|
)
|
|
|
|
type SqlUserStore struct {
|
|
*SqlStore
|
|
metrics einterfaces.MetricsInterface
|
|
|
|
// usersQuery is a starting point for all queries that return one or more Users.
|
|
usersQuery sq.SelectBuilder
|
|
}
|
|
|
|
func (us *SqlUserStore) ClearCaches() {}
|
|
|
|
func (us SqlUserStore) InvalidateProfileCacheForUser(userId string) {}
|
|
|
|
// getUsersColumnsWithName returns the user columns prefixed with the given name,
|
|
// with any extra columns appended.
|
|
//
|
|
// Note that the order of these columns must match the order in
|
|
// [SqlUserStore.Get] and [SqlUserStore.GetAllProfilesInChannel].
|
|
func getUsersColumnsWithName(name string, extraColumns ...string) []string {
|
|
columns := []string{
|
|
"Id",
|
|
"CreateAt",
|
|
"UpdateAt",
|
|
"DeleteAt",
|
|
"Username",
|
|
"Password",
|
|
"AuthData",
|
|
"AuthService",
|
|
"Email",
|
|
"EmailVerified",
|
|
"Nickname",
|
|
"FirstName",
|
|
"LastName",
|
|
"Position",
|
|
"Roles",
|
|
"AllowMarketing",
|
|
"Props",
|
|
"NotifyProps",
|
|
"LastPasswordUpdate",
|
|
"LastPictureUpdate",
|
|
"FailedAttempts",
|
|
"Locale",
|
|
"Timezone",
|
|
"MfaActive",
|
|
"MfaSecret",
|
|
"MfaUsedTimestamps",
|
|
"RemoteId",
|
|
"LastLogin",
|
|
}
|
|
columns = append(columns, extraColumns...)
|
|
result := make([]string, len(columns))
|
|
for i, col := range columns {
|
|
result[i] = name + "." + col
|
|
}
|
|
return result
|
|
}
|
|
|
|
// getUsersColumns returns the user columns for the Users table.
|
|
// This is primarily useful for other stores who choose to directly query
|
|
// and return [model.User] data.
|
|
//
|
|
// Note that the order of these columns must match the order in
|
|
// [SqlUserStore.Get] and [SqlUserStore.GetAllProfilesInChannel].
|
|
func getUsersColumns() []string {
|
|
return getUsersColumnsWithName("Users")
|
|
}
|
|
|
|
func getBotInfoColumns() []string {
|
|
return []string{
|
|
"b.UserId IS NOT NULL AS IsBot",
|
|
"COALESCE(b.Description, '') AS BotDescription",
|
|
"COALESCE(b.LastIconUpdate, 0) AS BotLastIconUpdate",
|
|
}
|
|
}
|
|
|
|
func newSqlUserStore(sqlStore *SqlStore, metrics einterfaces.MetricsInterface) store.UserStore {
|
|
us := &SqlUserStore{
|
|
SqlStore: sqlStore,
|
|
metrics: metrics,
|
|
}
|
|
|
|
us.usersQuery = us.getQueryBuilder().
|
|
// Together with getUsersColumns, the order specified here must match
|
|
// with [SqlUserStore.Get] and [SqlUserStore.GetAllProfilesInChannel].
|
|
Select(getUsersColumns()...).
|
|
Columns(getBotInfoColumns()...).
|
|
From("Users").
|
|
LeftJoin("Bots b ON ( b.UserId = Users.Id )")
|
|
|
|
return us
|
|
}
|
|
|
|
func (us SqlUserStore) validateAutoResponderMessageSize(notifyProps model.StringMap) error {
|
|
if notifyProps != nil {
|
|
maxPostSize := us.Post().GetMaxPostSize()
|
|
msg := notifyProps[model.AutoResponderMessageNotifyProp]
|
|
msgSize := utf8.RuneCountInString(msg)
|
|
if msgSize > maxPostSize {
|
|
mlog.Warn("auto_responder_message has size restrictions", mlog.Int("max_characters", maxPostSize), mlog.Int("received_size", msgSize))
|
|
return errors.New("Auto responder message size can't be more than the allowed Post size")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) insert(user *model.User) (sql.Result, error) {
|
|
if err := us.validateAutoResponderMessageSize(user.NotifyProps); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
query := `INSERT INTO Users
|
|
(Id, CreateAt, UpdateAt, DeleteAt, Username, Password, AuthData, AuthService,
|
|
Email, EmailVerified, Nickname, FirstName, LastName, Position, Roles, AllowMarketing,
|
|
Props, NotifyProps, LastPasswordUpdate, LastPictureUpdate, FailedAttempts,
|
|
Locale, Timezone, MfaActive, MfaSecret, RemoteId, MfaUsedTimestamps)
|
|
VALUES
|
|
(:Id, :CreateAt, :UpdateAt, :DeleteAt, :Username, :Password, :AuthData, :AuthService,
|
|
:Email, :EmailVerified, :Nickname, :FirstName, :LastName, :Position, :Roles, :AllowMarketing,
|
|
:Props, :NotifyProps, :LastPasswordUpdate, :LastPictureUpdate, :FailedAttempts,
|
|
:Locale, :Timezone, :MfaActive, :MfaSecret, :RemoteId, :MfaUsedTimestamps)`
|
|
|
|
user.Props = wrapBinaryParamStringMap(us.IsBinaryParamEnabled(), user.Props)
|
|
return us.GetMaster().NamedExec(query, user)
|
|
}
|
|
|
|
func (us SqlUserStore) InsertUsers(users []*model.User) error {
|
|
for _, user := range users {
|
|
_, err := us.insert(user)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) Save(rctx request.CTX, user *model.User) (*model.User, error) {
|
|
if user.Id != "" && !user.IsRemote() {
|
|
return nil, store.NewErrInvalidInput("User", "id", user.Id)
|
|
}
|
|
|
|
if err := user.PreSave(hashers.GetLatestHasher()); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := user.IsValid(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if _, err := us.insert(user); err != nil {
|
|
if IsUniqueConstraintError(err, []string{"Email", "users_email_key", "idx_users_email_unique"}) {
|
|
return nil, store.NewErrInvalidInput("User", "email", user.Email)
|
|
}
|
|
if IsUniqueConstraintError(err, []string{"Username", "users_username_key", "idx_users_username_unique"}) {
|
|
return nil, store.NewErrInvalidInput("User", "username", user.Username)
|
|
}
|
|
return nil, errors.Wrapf(err, "failed to save User with userId=%s", user.Id)
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) DeactivateGuests() ([]string, error) {
|
|
curTime := model.GetMillis()
|
|
updateQuery := us.getQueryBuilder().Update("Users").
|
|
Set("UpdateAt", curTime).
|
|
Set("DeleteAt", curTime).
|
|
Where(sq.Eq{"Roles": "system_guest"}).
|
|
Where(sq.Eq{"DeleteAt": 0})
|
|
|
|
_, err := us.GetMaster().ExecBuilder(updateQuery)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to update Users with roles=system_guest")
|
|
}
|
|
|
|
selectQuery := us.getQueryBuilder().
|
|
Select("Id").
|
|
From("Users").
|
|
Where(sq.Eq{"DeleteAt": curTime})
|
|
|
|
userIds := []string{}
|
|
err = us.GetMaster().SelectBuilder(&userIds, selectQuery)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
return userIds, nil
|
|
}
|
|
|
|
func (us SqlUserStore) DeactivateMagicLinkGuests() ([]string, error) {
|
|
curTime := model.GetMillis()
|
|
updateQuery := us.getQueryBuilder().Update("Users").
|
|
Set("UpdateAt", curTime).
|
|
Set("DeleteAt", curTime).
|
|
Set("Roles", "system_user").
|
|
Where(sq.Eq{"DeleteAt": 0}).
|
|
Where(sq.Eq{"AuthService": model.UserAuthServiceMagicLink})
|
|
|
|
_, err := us.GetMaster().ExecBuilder(updateQuery)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to update Users with auth_service=magic_link")
|
|
}
|
|
|
|
selectQuery := us.getQueryBuilder().
|
|
Select("Id").
|
|
From("Users").
|
|
Where(sq.Eq{"DeleteAt": curTime})
|
|
|
|
userIds := []string{}
|
|
err = us.GetMaster().SelectBuilder(&userIds, selectQuery)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
return userIds, nil
|
|
}
|
|
|
|
func (us SqlUserStore) Update(rctx request.CTX, user *model.User, trustedUpdateData bool) (*model.UserUpdate, error) {
|
|
user.PreUpdate()
|
|
|
|
if err := user.IsValid(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := us.validateAutoResponderMessageSize(user.NotifyProps); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
oldUser := model.User{}
|
|
query := us.usersQuery.Where(sq.Eq{"Users.Id": user.Id})
|
|
err := us.GetMaster().GetBuilder(&oldUser, query)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get User with userId=%s", user.Id)
|
|
}
|
|
|
|
if oldUser.Id == "" {
|
|
return nil, store.NewErrInvalidInput("User", "id", user.Id)
|
|
}
|
|
|
|
user.CreateAt = oldUser.CreateAt
|
|
user.AuthData = oldUser.AuthData
|
|
user.AuthService = oldUser.AuthService
|
|
user.Password = oldUser.Password
|
|
user.LastPasswordUpdate = oldUser.LastPasswordUpdate
|
|
user.LastPictureUpdate = oldUser.LastPictureUpdate
|
|
user.EmailVerified = oldUser.EmailVerified
|
|
user.FailedAttempts = oldUser.FailedAttempts
|
|
user.MfaSecret = oldUser.MfaSecret
|
|
user.MfaActive = oldUser.MfaActive
|
|
user.MfaUsedTimestamps = oldUser.MfaUsedTimestamps
|
|
user.LastLogin = oldUser.LastLogin
|
|
user.RemoteId = oldUser.RemoteId
|
|
|
|
if !trustedUpdateData {
|
|
user.Roles = oldUser.Roles
|
|
user.DeleteAt = oldUser.DeleteAt
|
|
|
|
if user.IsOAuthUser() {
|
|
user.Email = oldUser.Email
|
|
}
|
|
|
|
if user.IsLDAPUser() {
|
|
if user.Username != oldUser.Username {
|
|
return nil, store.NewErrInvalidInput("User", "id", user.Id)
|
|
}
|
|
if user.Email != oldUser.Email {
|
|
return nil, store.NewErrInvalidInput("User", "email", user.Id)
|
|
}
|
|
}
|
|
|
|
if user.Email != oldUser.Email {
|
|
user.EmailVerified = false
|
|
}
|
|
}
|
|
|
|
// In the past, changing the email of a SSO user would mark the email as unverified.
|
|
// This is a lazy migration to fix broken records.
|
|
if user.IsSSOUser() {
|
|
user.EmailVerified = true
|
|
}
|
|
|
|
if user.Username != oldUser.Username {
|
|
user.UpdateMentionKeysFromUsername(oldUser.Username)
|
|
}
|
|
|
|
updateQuery := `UPDATE Users
|
|
SET CreateAt=:CreateAt, UpdateAt=:UpdateAt, DeleteAt=:DeleteAt, Username=:Username, Password=:Password,
|
|
AuthData=:AuthData, AuthService=:AuthService,Email=:Email, EmailVerified=:EmailVerified,
|
|
Nickname=:Nickname, FirstName=:FirstName, LastName=:LastName, Position=:Position, Roles=:Roles,
|
|
AllowMarketing=:AllowMarketing, Props=:Props, NotifyProps=:NotifyProps,
|
|
LastPasswordUpdate=:LastPasswordUpdate, LastPictureUpdate=:LastPictureUpdate,
|
|
FailedAttempts=:FailedAttempts,Locale=:Locale, Timezone=:Timezone, MfaActive=:MfaActive,
|
|
MfaSecret=:MfaSecret, RemoteId=:RemoteId, LastLogin=:LastLogin, MfaUsedTimestamps=:MfaUsedTimestamps
|
|
WHERE Id=:Id`
|
|
|
|
user.Props = wrapBinaryParamStringMap(us.IsBinaryParamEnabled(), user.Props)
|
|
res, err := us.GetMaster().NamedExec(updateQuery, user)
|
|
if err != nil {
|
|
if IsUniqueConstraintError(err, []string{"Email", "users_email_key", "idx_users_email_unique"}) {
|
|
return nil, store.NewErrConflict("Email", err, user.Email)
|
|
}
|
|
if IsUniqueConstraintError(err, []string{"Username", "users_username_key", "idx_users_username_unique"}) {
|
|
return nil, store.NewErrConflict("Username", err, user.Username)
|
|
}
|
|
return nil, errors.Wrapf(err, "failed to update User with userId=%s", user.Id)
|
|
}
|
|
|
|
count, err := res.RowsAffected()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to get rows_affected")
|
|
}
|
|
if count > 1 {
|
|
return nil, fmt.Errorf("multiple users were update: userId=%s, count=%d", user.Id, count)
|
|
}
|
|
|
|
user.Sanitize(map[string]bool{})
|
|
oldUser.Sanitize(map[string]bool{})
|
|
return &model.UserUpdate{New: user.DeepCopy(), Old: &oldUser}, nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateNotifyProps(userID string, props map[string]string) error {
|
|
if err := us.validateAutoResponderMessageSize(props); err != nil {
|
|
return err
|
|
}
|
|
|
|
buf, err := json.Marshal(props)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed marshalling session props")
|
|
}
|
|
if us.IsBinaryParamEnabled() {
|
|
buf = AppendBinaryFlag(buf)
|
|
}
|
|
|
|
if _, err := us.GetMaster().Exec(`UPDATE Users
|
|
SET NotifyProps = ?
|
|
WHERE Id = ?`, buf, userID); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userID)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateLastPictureUpdate(userId string) error {
|
|
curTime := model.GetMillis()
|
|
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET LastPictureUpdate = ?, UpdateAt = ? WHERE Id = ?", curTime, curTime, userId); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) ResetLastPictureUpdate(userId string) error {
|
|
curTime := model.GetMillis()
|
|
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET LastPictureUpdate = ?, UpdateAt = ? WHERE Id = ?", -curTime, curTime, userId); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateUpdateAt(userId string) (int64, error) {
|
|
curTime := model.GetMillis()
|
|
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET UpdateAt = ? WHERE Id = ?", curTime, userId); err != nil {
|
|
return curTime, errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return curTime, nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdatePassword(userId, hashedPassword string) error {
|
|
updateAt := model.GetMillis()
|
|
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET Password = ?, LastPasswordUpdate = ?, UpdateAt = ?, AuthData = NULL, AuthService = '', FailedAttempts = 0 WHERE Id = ?", hashedPassword, updateAt, updateAt, userId); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateFailedPasswordAttempts(userId string, attempts int) error {
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET FailedAttempts = ? WHERE Id = ?", attempts, userId); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// TryIncrementFailedPasswordAttempts atomically increments FailedAttempts by one
|
|
// for the given user, only if FailedAttempts is strictly less than maxAttempts.
|
|
// Returns true if the row was updated (a slot was claimed), false if the cap had
|
|
// already been reached (or the user does not exist). The row lock taken by the
|
|
// UPDATE serializes concurrent attempts on the same user, so the cap predicate
|
|
// is enforced without any application-level locking.
|
|
func (us SqlUserStore) TryIncrementFailedPasswordAttempts(userId string, maxAttempts int) (bool, error) {
|
|
res, err := us.GetMaster().Exec(
|
|
"UPDATE Users SET FailedAttempts = FailedAttempts + 1 WHERE Id = ? AND FailedAttempts < ?",
|
|
userId, maxAttempts,
|
|
)
|
|
if err != nil {
|
|
return false, errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
rows, err := res.RowsAffected()
|
|
if err != nil {
|
|
return false, errors.Wrapf(err, "failed to read rows affected for userId=%s", userId)
|
|
}
|
|
|
|
return rows == 1, nil
|
|
}
|
|
|
|
// DecrementFailedPasswordAttempts atomically decrements FailedAttempts by one
|
|
// for the given user, only if FailedAttempts is strictly greater than zero. It
|
|
// is used to refund a slot previously claimed by TryIncrementFailedPasswordAttempts
|
|
// when the in-flight authentication turns out not to be a credential-failure
|
|
// event (e.g. a backend error or an MFA pre-flight probe).
|
|
func (us SqlUserStore) DecrementFailedPasswordAttempts(userId string) error {
|
|
_, err := us.GetMaster().Exec(
|
|
"UPDATE Users SET FailedAttempts = FailedAttempts - 1 WHERE Id = ? AND FailedAttempts > 0",
|
|
userId,
|
|
)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateAuthData(userId string, service string, authData *string, email string, resetMfa bool) (string, error) {
|
|
updateAt := model.GetMillis()
|
|
|
|
updateQuery := us.getQueryBuilder().Update("Users").
|
|
Set("Password", "").
|
|
Set("LastPasswordUpdate", updateAt).
|
|
Set("UpdateAt", updateAt).
|
|
Set("FailedAttempts", 0).
|
|
Set("AuthService", service).
|
|
Set("AuthData", authData).
|
|
Where(sq.Eq{"Id": userId})
|
|
|
|
if email != "" {
|
|
updateQuery = updateQuery.Set("Email", sq.Expr("lower(?)", email))
|
|
}
|
|
|
|
if resetMfa {
|
|
updateQuery = updateQuery.Set("MfaActive", false).
|
|
Set("MfaSecret", "").
|
|
Set("MfaUsedTimestamps", model.StringArray{})
|
|
}
|
|
|
|
if _, err := us.GetMaster().ExecBuilder(updateQuery); err != nil {
|
|
if IsUniqueConstraintError(err, []string{"Email", "users_email_key", "idx_users_email_unique", "AuthData", "users_authdata_key"}) {
|
|
return "", store.NewErrInvalidInput("User", "id", userId)
|
|
}
|
|
return "", errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
return userId, nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateLastLogin(userId string, lastLogin int64) error {
|
|
updateQuery := us.getQueryBuilder().
|
|
Update("Users").
|
|
Set("LastLogin", lastLogin).
|
|
Set("UpdateAt", model.GetMillis()).
|
|
Where(sq.Eq{"Id": userId})
|
|
|
|
if _, err := us.GetMaster().ExecBuilder(updateQuery); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// ResetAuthDataToEmailForUsers resets the AuthData of users whose AuthService
|
|
// is |service| to their Email. If userIDs is non-empty, only the users whose
|
|
// IDs are in userIDs will be affected. If dryRun is true, only the number
|
|
// of users who *would* be affected is returned; otherwise, the number of
|
|
// users who actually were affected is returned.
|
|
func (us SqlUserStore) ResetAuthDataToEmailForUsers(service string, userIDs []string, includeDeleted bool, dryRun bool) (int, error) {
|
|
whereEquals := sq.Eq{"AuthService": service}
|
|
if len(userIDs) > 0 {
|
|
whereEquals["Id"] = userIDs
|
|
}
|
|
if !includeDeleted {
|
|
whereEquals["DeleteAt"] = 0
|
|
}
|
|
|
|
if dryRun {
|
|
builder := us.getQueryBuilder().
|
|
Select("COUNT(*)").
|
|
From("Users").
|
|
Where(whereEquals)
|
|
var numAffected int
|
|
err := us.GetReplica().GetBuilder(&numAffected, builder)
|
|
return numAffected, err
|
|
}
|
|
builder := us.getQueryBuilder().
|
|
Update("Users").
|
|
Set("AuthData", sq.Expr("Email")).
|
|
Where(whereEquals)
|
|
result, err := us.GetMaster().ExecBuilder(builder)
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "failed to update users' AuthData")
|
|
}
|
|
numAffected, err := result.RowsAffected()
|
|
return int(numAffected), err
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateMfaSecret(userId, secret string) error {
|
|
updateAt := model.GetMillis()
|
|
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET MfaSecret = ?, MfaUsedTimestamps = ?, UpdateAt = ? WHERE Id = ?", secret, model.StringArray{}, updateAt, userId); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) UpdateMfaActive(userId string, active bool) error {
|
|
updateAt := model.GetMillis()
|
|
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET MfaActive = ?, UpdateAt = ? WHERE Id = ?", active, updateAt, userId); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) StoreMfaUsedTimestamps(userId string, ts []int) error {
|
|
tSStrArray := model.StringArray{}
|
|
for _, t := range ts {
|
|
tSStrArray = append(tSStrArray, fmt.Sprintf("%d", t))
|
|
}
|
|
|
|
updateAt := model.GetMillis()
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET MfaUsedTimestamps = ?, UpdateAt = ? WHERE Id = ?", tSStrArray, updateAt, userId); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetMfaUsedTimestamps(userId string) ([]int, error) {
|
|
tsStrArray := model.StringArray{}
|
|
err := us.GetReplica().Get(&tsStrArray, "SELECT MfaUsedTimestamps FROM Users WHERE Id = ?", userId)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to get MFA used timestamps for user with ID %s", userId)
|
|
}
|
|
|
|
ts := make([]int, len(tsStrArray))
|
|
for i, t := range tsStrArray {
|
|
ts[i], err = strconv.Atoi(t)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to parse MFA used timestamp %s for user with ID %s", t, userId)
|
|
}
|
|
}
|
|
|
|
return ts, nil
|
|
}
|
|
|
|
// GetMany returns a list of users for the provided list of ids
|
|
func (us SqlUserStore) GetMany(rctx request.CTX, ids []string) ([]*model.User, error) {
|
|
query := us.usersQuery.Where(sq.Eq{"Id": ids})
|
|
users := []*model.User{}
|
|
if err := us.SqlStore.DBXFromContext(rctx.Context()).SelectBuilderCtx(rctx.Context(), &users, query); err != nil {
|
|
return nil, errors.Wrap(err, "users_get_many_select")
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) Get(ctx context.Context, id string) (*model.User, error) {
|
|
query := us.usersQuery.Where("Id = ?", id)
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "users_get_tosql")
|
|
}
|
|
row := us.SqlStore.DBXFromContext(ctx).QueryRow(queryString, args...)
|
|
|
|
var user model.User
|
|
var props, notifyProps, timezone []byte
|
|
err = row.Scan(&user.Id, &user.CreateAt, &user.UpdateAt, &user.DeleteAt, &user.Username,
|
|
&user.Password, &user.AuthData, &user.AuthService, &user.Email, &user.EmailVerified,
|
|
&user.Nickname, &user.FirstName, &user.LastName, &user.Position, &user.Roles,
|
|
&user.AllowMarketing, &props, ¬ifyProps, &user.LastPasswordUpdate, &user.LastPictureUpdate,
|
|
&user.FailedAttempts, &user.Locale, &timezone, &user.MfaActive, &user.MfaSecret, &user.MfaUsedTimestamps,
|
|
&user.RemoteId, &user.LastLogin, &user.IsBot, &user.BotDescription, &user.BotLastIconUpdate)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, store.NewErrNotFound("User", id)
|
|
}
|
|
return nil, errors.Wrapf(err, "failed to get User with userId=%s", id)
|
|
}
|
|
if err = json.Unmarshal(props, &user.Props); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal user props")
|
|
}
|
|
if err = json.Unmarshal(notifyProps, &user.NotifyProps); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal user notify props")
|
|
}
|
|
if err = json.Unmarshal(timezone, &user.Timezone); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal user timezone")
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetAll() ([]*model.User, error) {
|
|
query := us.usersQuery.OrderBy("Username ASC")
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_all_users_tosql")
|
|
}
|
|
|
|
data := []*model.User{}
|
|
if err := us.GetReplica().Select(&data, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetAllAfter(limit int, afterId string) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Where("Id > ?", afterId).
|
|
OrderBy("Id ASC").
|
|
Limit(uint64(limit))
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().SelectBuilder(&users, query); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetEtagForAllProfiles() string {
|
|
var updateAt int64
|
|
err := us.GetReplica().Get(&updateAt, "SELECT UpdateAt FROM Users ORDER BY UpdateAt DESC LIMIT 1")
|
|
if err != nil {
|
|
return fmt.Sprintf("%v.%v", model.CurrentVersion, model.GetMillis())
|
|
}
|
|
return fmt.Sprintf("%v.%v", model.CurrentVersion, updateAt)
|
|
}
|
|
|
|
func (us SqlUserStore) GetAllProfiles(options *model.UserGetOptions) ([]*model.User, error) {
|
|
// Determine ordering based on Sort option - default to Username ASC for backwards compatibility
|
|
orderBy := "Users.Username ASC"
|
|
if options.Sort == "update_at_asc" {
|
|
orderBy = "Users.UpdateAt ASC"
|
|
}
|
|
|
|
query := us.usersQuery.
|
|
OrderBy(orderBy).
|
|
Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
|
|
|
|
query = applyViewRestrictionsFilter(query, options.ViewRestrictions, true)
|
|
|
|
query = applyRoleFilter(query, options.Role)
|
|
query = applyMultiRoleFilters(query, options.Roles, []string{}, []string{})
|
|
|
|
if options.Inactive {
|
|
query = query.Where("Users.DeleteAt != 0")
|
|
} else if options.Active {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
if options.UpdatedAfter > 0 {
|
|
query = query.Where(sq.Gt{"Users.UpdateAt": options.UpdatedAfter})
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().SelectBuilder(&users, query); err != nil {
|
|
return nil, errors.Wrap(err, "failed to get User profiles")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func applyRoleFilter(query sq.SelectBuilder, role string) sq.SelectBuilder {
|
|
if role == "" {
|
|
return query
|
|
}
|
|
|
|
roleParam := fmt.Sprintf("%%%s%%", sanitizeSearchTerm(role, "\\"))
|
|
return query.Where("Users.Roles LIKE LOWER(?)", roleParam)
|
|
}
|
|
|
|
func applyMultiRoleFilters(query sq.SelectBuilder, systemRoles []string, teamRoles []string, channelRoles []string) sq.SelectBuilder {
|
|
sqOr := sq.Or{}
|
|
|
|
if len(systemRoles) > 0 && systemRoles[0] != "" {
|
|
for _, role := range systemRoles {
|
|
queryRole := wildcardSearchTerm(role)
|
|
switch role {
|
|
case model.SystemUserRoleId:
|
|
// If querying for a `system_user` ensure that the user is only a system_user.
|
|
sqOr = append(sqOr, sq.Eq{"Users.Roles": role})
|
|
case model.SystemGuestRoleId, model.SystemAdminRoleId, model.SystemUserManagerRoleId, model.SystemReadOnlyAdminRoleId, model.SystemManagerRoleId, model.SystemCustomGroupAdminRoleId, model.SharedChannelManagerRoleId:
|
|
// If querying for any other roles search using a wildcard.
|
|
sqOr = append(sqOr, sq.ILike{"Users.Roles": queryRole})
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(channelRoles) > 0 && channelRoles[0] != "" {
|
|
for _, channelRole := range channelRoles {
|
|
switch channelRole {
|
|
case model.ChannelAdminRoleId:
|
|
sqOr = append(sqOr, sq.And{sq.Eq{"cm.SchemeAdmin": true}, sq.NotILike{"Users.Roles": wildcardSearchTerm(model.SystemAdminRoleId)}})
|
|
case model.ChannelUserRoleId:
|
|
sqOr = append(sqOr, sq.And{sq.Eq{"cm.SchemeUser": true}, sq.Eq{"cm.SchemeAdmin": false}, sq.NotILike{"Users.Roles": wildcardSearchTerm(model.SystemAdminRoleId)}})
|
|
case model.ChannelGuestRoleId:
|
|
sqOr = append(sqOr, sq.Eq{"cm.SchemeGuest": true})
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(teamRoles) > 0 && teamRoles[0] != "" {
|
|
for _, teamRole := range teamRoles {
|
|
switch teamRole {
|
|
case model.TeamAdminRoleId:
|
|
sqOr = append(sqOr, sq.And{sq.Eq{"tm.SchemeAdmin": true}, sq.NotILike{"Users.Roles": wildcardSearchTerm(model.SystemAdminRoleId)}})
|
|
case model.TeamUserRoleId:
|
|
sqOr = append(sqOr, sq.And{sq.Eq{"tm.SchemeUser": true}, sq.Eq{"tm.SchemeAdmin": false}, sq.NotILike{"Users.Roles": wildcardSearchTerm(model.SystemAdminRoleId)}})
|
|
case model.TeamGuestRoleId:
|
|
sqOr = append(sqOr, sq.Eq{"tm.SchemeGuest": true})
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(sqOr) > 0 {
|
|
return query.Where(sqOr)
|
|
}
|
|
return query
|
|
}
|
|
|
|
func applyChannelGroupConstrainedFilter(query sq.SelectBuilder, channelId string) sq.SelectBuilder {
|
|
if channelId == "" {
|
|
return query
|
|
}
|
|
|
|
return query.
|
|
Where(`Users.Id IN (
|
|
SELECT
|
|
GroupMembers.UserId
|
|
FROM
|
|
Channels
|
|
JOIN GroupChannels ON GroupChannels.ChannelId = Channels.Id
|
|
JOIN UserGroups ON UserGroups.Id = GroupChannels.GroupId
|
|
JOIN GroupMembers ON GroupMembers.GroupId = UserGroups.Id
|
|
WHERE
|
|
Channels.Id = ?
|
|
AND GroupChannels.DeleteAt = 0
|
|
AND UserGroups.DeleteAt = 0
|
|
AND GroupMembers.DeleteAt = 0
|
|
GROUP BY
|
|
GroupMembers.UserId
|
|
)`, channelId)
|
|
}
|
|
|
|
func applyTeamGroupConstrainedFilter(query sq.SelectBuilder, teamId string) sq.SelectBuilder {
|
|
if teamId == "" {
|
|
return query
|
|
}
|
|
|
|
return query.
|
|
Where(`Users.Id IN (
|
|
SELECT
|
|
GroupMembers.UserId
|
|
FROM
|
|
Teams
|
|
JOIN GroupTeams ON GroupTeams.TeamId = Teams.Id
|
|
JOIN UserGroups ON UserGroups.Id = GroupTeams.GroupId
|
|
JOIN GroupMembers ON GroupMembers.GroupId = UserGroups.Id
|
|
WHERE
|
|
Teams.Id = ?
|
|
AND GroupTeams.DeleteAt = 0
|
|
AND UserGroups.DeleteAt = 0
|
|
AND GroupMembers.DeleteAt = 0
|
|
GROUP BY
|
|
GroupMembers.UserId
|
|
)`, teamId)
|
|
}
|
|
|
|
func (us SqlUserStore) GetEtagForProfiles(teamId string) string {
|
|
var updateAt int64
|
|
err := us.GetReplica().Get(&updateAt, "SELECT UpdateAt FROM Users, TeamMembers WHERE TeamMembers.TeamId = ? AND Users.Id = TeamMembers.UserId ORDER BY UpdateAt DESC LIMIT 1", teamId)
|
|
if err != nil {
|
|
return fmt.Sprintf("%v.%v", model.CurrentVersion, model.GetMillis())
|
|
}
|
|
return fmt.Sprintf("%v.%v", model.CurrentVersion, updateAt)
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfiles(options *model.UserGetOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("TeamMembers tm ON ( tm.UserId = Users.Id AND tm.DeleteAt = 0 )").
|
|
Where("tm.TeamId = ?", options.InTeamId).
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
|
|
|
|
query = applyViewRestrictionsFilter(query, options.ViewRestrictions, true)
|
|
|
|
query = applyRoleFilter(query, options.Role)
|
|
query = applyMultiRoleFilters(query, options.Roles, options.TeamRoles, options.ChannelRoles)
|
|
|
|
if options.Inactive {
|
|
query = query.Where("Users.DeleteAt != 0")
|
|
} else if options.Active {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().SelectBuilder(&users, query); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) InvalidateProfilesInChannelCacheByUser(userId string) {}
|
|
|
|
func (us SqlUserStore) InvalidateProfilesInChannelCache(channelId string) {}
|
|
|
|
func (us SqlUserStore) GetProfilesInChannel(options *model.UserGetOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("ChannelMembers cm ON ( cm.UserId = Users.Id )").
|
|
Where("cm.ChannelId = ?", options.InChannelId).
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
|
|
|
|
if options.Inactive {
|
|
query = query.Where("Users.DeleteAt != 0")
|
|
} else if options.Active {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
query = applyMultiRoleFilters(query, options.Roles, options.TeamRoles, options.ChannelRoles)
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().SelectBuilder(&users, query); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfilesInChannelByStatus(options *model.UserGetOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("ChannelMembers cm ON ( cm.UserId = Users.Id )").
|
|
LeftJoin("Status s ON ( s.UserId = Users.Id )").
|
|
Where("cm.ChannelId = ?", options.InChannelId).
|
|
OrderBy(`
|
|
CASE s.Status
|
|
WHEN 'online' THEN 1
|
|
WHEN 'away' THEN 2
|
|
WHEN 'dnd' THEN 3
|
|
ELSE 4
|
|
END
|
|
`).
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
|
|
|
|
if options.Inactive && !options.Active {
|
|
query = query.Where("Users.DeleteAt != 0")
|
|
} else if options.Active && !options.Inactive {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().SelectBuilder(&users, query); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfilesInChannelByAdmin(options *model.UserGetOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("ChannelMembers cm ON ( cm.UserId = Users.Id )").
|
|
Where("cm.ChannelId = ?", options.InChannelId).
|
|
OrderBy(`cm.SchemeAdmin DESC`).
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
|
|
|
|
if options.Inactive && !options.Active {
|
|
query = query.Where("Users.DeleteAt != 0")
|
|
} else if options.Active && !options.Inactive {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_profiles_in_channel_by_admin_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetAllProfilesInChannel(ctx context.Context, channelID string, allowFromCache bool) (map[string]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("ChannelMembers cm ON ( cm.UserId = Users.Id )").
|
|
Where("cm.ChannelId = ?", channelID).
|
|
Where("Users.DeleteAt = 0").
|
|
OrderBy("Users.Username ASC")
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_all_profiles_in_channel_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
rows, err := us.SqlStore.DBXFromContext(ctx).Query(queryString, args...)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var user model.User
|
|
var props, notifyProps, timezone []byte
|
|
if err = rows.Scan(&user.Id, &user.CreateAt, &user.UpdateAt, &user.DeleteAt, &user.Username, &user.Password, &user.AuthData, &user.AuthService, &user.Email, &user.EmailVerified, &user.Nickname, &user.FirstName, &user.LastName, &user.Position, &user.Roles, &user.AllowMarketing, &props, ¬ifyProps, &user.LastPasswordUpdate, &user.LastPictureUpdate, &user.FailedAttempts, &user.Locale, &timezone, &user.MfaActive, &user.MfaSecret, &user.MfaUsedTimestamps, &user.RemoteId, &user.LastLogin, &user.IsBot, &user.BotDescription, &user.BotLastIconUpdate); err != nil {
|
|
return nil, errors.Wrap(err, "failed to scan values from rows into User entity")
|
|
}
|
|
if err = json.Unmarshal(props, &user.Props); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal user props")
|
|
}
|
|
if err = json.Unmarshal(notifyProps, &user.NotifyProps); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal user notify props")
|
|
}
|
|
if err = json.Unmarshal(timezone, &user.Timezone); err != nil {
|
|
return nil, errors.Wrap(err, "failed to unmarshal user timezone")
|
|
}
|
|
users = append(users, &user)
|
|
}
|
|
err = rows.Err()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error while iterating over rows")
|
|
}
|
|
|
|
userMap := make(map[string]*model.User)
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
userMap[u.Id] = u
|
|
}
|
|
|
|
return userMap, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfilesNotInChannel(teamId string, channelId string, groupConstrained bool, offset int, limit int, viewRestrictions *model.ViewUsersRestrictions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("TeamMembers tm ON ( tm.UserId = Users.Id AND tm.DeleteAt = 0 AND tm.TeamId = ? )", teamId).
|
|
LeftJoin("ChannelMembers cm ON ( cm.UserId = Users.Id AND cm.ChannelId = ? )", channelId).
|
|
Where("cm.UserId IS NULL").
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(offset)).Limit(uint64(limit))
|
|
|
|
query = applyViewRestrictionsFilter(query, viewRestrictions, true)
|
|
|
|
if groupConstrained {
|
|
query = applyChannelGroupConstrainedFilter(query, channelId)
|
|
}
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_profiles_not_in_channel_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfilesWithoutTeam(options *model.UserGetOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Where(`(
|
|
SELECT
|
|
COUNT(0)
|
|
FROM
|
|
TeamMembers
|
|
WHERE
|
|
TeamMembers.UserId = Users.Id
|
|
AND TeamMembers.DeleteAt = 0
|
|
) = 0`).
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(options.Page * options.PerPage)).Limit(uint64(options.PerPage))
|
|
|
|
query = applyViewRestrictionsFilter(query, options.ViewRestrictions, true)
|
|
|
|
query = applyRoleFilter(query, options.Role)
|
|
|
|
if options.Inactive {
|
|
query = query.Where("Users.DeleteAt != 0")
|
|
} else if options.Active {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_profiles_without_team_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfilesByUsernames(usernames []string, viewRestrictions *model.ViewUsersRestrictions) ([]*model.User, error) {
|
|
query := us.usersQuery
|
|
|
|
query = applyViewRestrictionsFilter(query, viewRestrictions, true)
|
|
|
|
query = query.
|
|
Where(map[string]any{
|
|
"Username": usernames,
|
|
}).
|
|
OrderBy("Users.Username ASC")
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_profiles_by_usernames")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
type UserWithLastActivityAt struct {
|
|
model.User
|
|
LastActivityAt int64
|
|
}
|
|
|
|
func (us SqlUserStore) GetRecentlyActiveUsersForTeam(teamId string, offset, limit int, viewRestrictions *model.ViewUsersRestrictions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Column("s.LastActivityAt").
|
|
Join("TeamMembers tm ON (tm.UserId = Users.Id AND tm.TeamId = ?)", teamId).
|
|
Join("Status s ON (s.UserId = Users.Id)").
|
|
OrderBy("s.LastActivityAt DESC").
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(offset)).Limit(uint64(limit))
|
|
|
|
query = applyViewRestrictionsFilter(query, viewRestrictions, true)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_recently_active_users_for_team_tosql")
|
|
}
|
|
|
|
users := []*UserWithLastActivityAt{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
userList := []*model.User{}
|
|
|
|
for _, userWithLastActivityAt := range users {
|
|
u := userWithLastActivityAt.User
|
|
u.Sanitize(map[string]bool{})
|
|
u.LastActivityAt = userWithLastActivityAt.LastActivityAt
|
|
userList = append(userList, &u)
|
|
}
|
|
|
|
return userList, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetNewUsersForTeam(teamId string, offset, limit int, viewRestrictions *model.ViewUsersRestrictions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("TeamMembers tm ON (tm.UserId = Users.Id AND tm.TeamId = ?)", teamId).
|
|
OrderBy("Users.CreateAt DESC").
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(offset)).Limit(uint64(limit))
|
|
|
|
query = applyViewRestrictionsFilter(query, viewRestrictions, true)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_new_users_for_team_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfileByIds(rctx request.CTX, userIds []string, options *store.UserGetByIdsOpts, allowFromCache bool) ([]*model.User, error) {
|
|
if options == nil {
|
|
options = &store.UserGetByIdsOpts{}
|
|
}
|
|
|
|
users := []*model.User{}
|
|
query := us.usersQuery.
|
|
Where(map[string]any{
|
|
"Users.Id": userIds,
|
|
}).
|
|
OrderBy("Users.Username ASC")
|
|
|
|
if options.Since > 0 {
|
|
query = query.Where(sq.Gt(map[string]any{
|
|
"Users.UpdateAt": options.Since,
|
|
}))
|
|
}
|
|
|
|
query = applyViewRestrictionsFilter(query, options.ViewRestrictions, true)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_profile_by_ids_tosql")
|
|
}
|
|
|
|
if err := us.SqlStore.DBXFromContext(rctx.Context()).Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
type UserWithChannel struct {
|
|
model.User
|
|
ChannelId string
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfileByGroupChannelIdsForUser(userId string, channelIds []string) (map[string][]*model.User, error) {
|
|
if len(channelIds) > MaxGroupChannelsForProfiles {
|
|
channelIds = channelIds[0:MaxGroupChannelsForProfiles]
|
|
}
|
|
|
|
isMemberQuery := fmt.Sprintf(`
|
|
EXISTS(
|
|
SELECT
|
|
1
|
|
FROM
|
|
ChannelMembers
|
|
WHERE
|
|
UserId = '%s'
|
|
AND
|
|
ChannelId = cm.ChannelId
|
|
)`, userId)
|
|
|
|
query := us.usersQuery.
|
|
Columns("cm.ChannelId").
|
|
Join("ChannelMembers cm ON Users.Id = cm.UserId").
|
|
Join("Channels c ON cm.ChannelId = c.Id").
|
|
Where(sq.Eq{"c.Type": model.ChannelTypeGroup, "cm.ChannelId": channelIds}).
|
|
Where(isMemberQuery).
|
|
Where(sq.NotEq{"Users.Id": userId}).
|
|
OrderBy("Users.Username ASC")
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_profiles_by_group_channel_ids_for_user_tosql")
|
|
}
|
|
|
|
usersWithChannel := []*UserWithChannel{}
|
|
if err := us.GetReplica().Select(&usersWithChannel, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
usersByChannelId := map[string][]*model.User{}
|
|
for _, user := range usersWithChannel {
|
|
if val, ok := usersByChannelId[user.ChannelId]; ok {
|
|
usersByChannelId[user.ChannelId] = append(val, &user.User)
|
|
} else {
|
|
usersByChannelId[user.ChannelId] = []*model.User{&user.User}
|
|
}
|
|
}
|
|
|
|
return usersByChannelId, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetSystemAdminProfiles() (map[string]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Where("Roles LIKE ?", "%system_admin%").
|
|
OrderBy("Users.Username ASC")
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_system_admin_profiles_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
userMap := make(map[string]*model.User)
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
userMap[u.Id] = u
|
|
}
|
|
|
|
return userMap, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetByEmail(email string) (*model.User, error) {
|
|
query := us.usersQuery.Where("Email = lower(?)", email)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_by_email_tosql")
|
|
}
|
|
|
|
user := model.User{}
|
|
if err := us.GetReplica().Get(&user, queryString, args...); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, errors.Wrap(store.NewErrNotFound("User", fmt.Sprintf("email=%s", email)), "failed to find User")
|
|
}
|
|
|
|
return nil, errors.Wrapf(err, "failed to get User with email=%s", email)
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetByRemoteID(remoteID string) (*model.User, error) {
|
|
query := us.usersQuery.Where(sq.Eq{"RemoteId": remoteID})
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_by_remote_id_tosql")
|
|
}
|
|
|
|
user := model.User{}
|
|
if err := us.GetReplica().Get(&user, queryString, args...); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, errors.Wrap(store.NewErrNotFound("User", fmt.Sprintf("remoteid=%s", remoteID)), "failed to find User")
|
|
}
|
|
|
|
return nil, errors.Wrapf(err, "failed to get User with RemoteId=%s", remoteID)
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetByAuthData(authData *string) (*model.User, error) {
|
|
if authData == nil || *authData == "" {
|
|
return nil, store.NewErrInvalidInput("User", "<authData>", "empty or nil")
|
|
}
|
|
|
|
query := us.usersQuery.Where("Users.AuthData = ?", authData)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_by_auth_data_tosql")
|
|
}
|
|
|
|
user := model.User{}
|
|
if err := us.GetReplica().Get(&user, queryString, args...); err == sql.ErrNoRows {
|
|
return nil, store.NewErrNotFound("User", fmt.Sprintf("authData=%s", *authData))
|
|
} else if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to find User with authData=%s", *authData)
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetByAuth(authData *string, authService string) (*model.User, error) {
|
|
if authData == nil || *authData == "" {
|
|
return nil, store.NewErrInvalidInput("User", "<authData>", "empty or nil")
|
|
}
|
|
|
|
query := us.usersQuery.
|
|
Where("Users.AuthData = ?", authData).
|
|
Where("Users.AuthService = ?", authService)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_by_auth_tosql")
|
|
}
|
|
|
|
user := model.User{}
|
|
if err := us.GetReplica().Get(&user, queryString, args...); err == sql.ErrNoRows {
|
|
return nil, store.NewErrNotFound("User", fmt.Sprintf("authData=%s, authService=%s", *authData, authService))
|
|
} else if err != nil {
|
|
return nil, errors.Wrapf(err, "failed to find User with authData=%s and authService=%s", *authData, authService)
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetAllUsingAuthService(authService string) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Where("Users.AuthService = ?", authService).
|
|
OrderBy("Users.Username ASC")
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_all_using_auth_service_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to find Users with authService=%s", authService)
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetAllNotInAuthService(authServices []string) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Where(sq.NotEq{"Users.AuthService": authServices}).
|
|
OrderBy("Users.Username ASC")
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_all_not_in_auth_service_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to find Users with authServices in %v", authServices)
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetByUsername(username string) (*model.User, error) {
|
|
query := us.usersQuery.Where("Users.Username = lower(?)", username)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_by_username_tosql")
|
|
}
|
|
|
|
user := model.User{}
|
|
if err := us.GetReplica().Get(&user, queryString, args...); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, errors.Wrap(store.NewErrNotFound("User", fmt.Sprintf("username=%s", username)), "failed to find User")
|
|
}
|
|
|
|
return nil, errors.Wrapf(err, "failed to find User with username=%s", username)
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetForLogin(loginId string, allowSignInWithUsername, allowSignInWithEmail bool) (*model.User, error) {
|
|
query := us.usersQuery
|
|
if allowSignInWithUsername && allowSignInWithEmail {
|
|
query = query.Where("Username = lower(?) OR Email = lower(?)", loginId, loginId)
|
|
} else if allowSignInWithUsername {
|
|
query = query.Where("Username = lower(?)", loginId)
|
|
} else if allowSignInWithEmail {
|
|
query = query.Where("Email = lower(?)", loginId)
|
|
} else {
|
|
return nil, errors.New("sign in with username and email are disabled")
|
|
}
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_for_login_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
if len(users) == 0 {
|
|
return nil, errors.New("user not found")
|
|
}
|
|
|
|
if len(users) > 1 {
|
|
return nil, errors.New("multiple users found")
|
|
}
|
|
|
|
return users[0], nil
|
|
}
|
|
|
|
func (us SqlUserStore) VerifyEmail(userId, email string) (string, error) {
|
|
curTime := model.GetMillis()
|
|
if _, err := us.GetMaster().Exec("UPDATE Users SET Email = lower(?), EmailVerified = true, UpdateAt = ? WHERE Id = ?", email, curTime, userId); err != nil {
|
|
return "", errors.Wrapf(err, "failed to update Users with userId=%s and email=%s", userId, email)
|
|
}
|
|
|
|
return userId, nil
|
|
}
|
|
|
|
func (us SqlUserStore) PermanentDelete(rctx request.CTX, userId string) error {
|
|
if _, err := us.GetMaster().Exec("DELETE FROM Users WHERE Id = ?", userId); err != nil {
|
|
return errors.Wrapf(err, "failed to delete User with userId=%s", userId)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) Count(options model.UserCountOptions) (int64, error) {
|
|
query := us.getQueryBuilder().Select("COUNT(*)").From("Users")
|
|
|
|
if !options.IncludeDeleted {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
if !options.IncludeRemoteUsers {
|
|
query = query.Where(sq.Or{sq.Eq{"Users.RemoteId": ""}, sq.Eq{"Users.RemoteId": nil}})
|
|
}
|
|
|
|
if options.IncludeBotAccounts {
|
|
if options.ExcludeRegularUsers {
|
|
query = query.Join("Bots ON Users.Id = Bots.UserId")
|
|
}
|
|
} else {
|
|
query = query.LeftJoin("Bots ON Users.Id = Bots.UserId").Where("Bots.UserId IS NULL")
|
|
|
|
if options.ExcludeRegularUsers {
|
|
// Currently this doesn't make sense because it will always return 0
|
|
return int64(0), errors.New("query with IncludeBotAccounts=false and excludeRegularUsers=true always return 0")
|
|
}
|
|
}
|
|
|
|
if options.TeamId != "" {
|
|
query = query.LeftJoin("TeamMembers AS tm ON Users.Id = tm.UserId").Where("tm.TeamId = ? AND tm.DeleteAt = 0", options.TeamId)
|
|
} else if options.ChannelId != "" {
|
|
query = query.LeftJoin("ChannelMembers AS cm ON Users.Id = cm.UserId").Where("cm.ChannelId = ?", options.ChannelId)
|
|
}
|
|
query = applyViewRestrictionsFilter(query, options.ViewRestrictions, false)
|
|
query = applyMultiRoleFilters(query, options.Roles, options.TeamRoles, options.ChannelRoles)
|
|
|
|
query = query.PlaceholderFormat(sq.Dollar)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return int64(0), errors.Wrap(err, "count_tosql")
|
|
}
|
|
|
|
var count int64
|
|
err = us.GetReplica().Get(&count, queryString, args...)
|
|
if err != nil {
|
|
return int64(0), errors.Wrap(err, "failed to count Users")
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AnalyticsActiveCount(timePeriod int64, options model.UserCountOptions) (int64, error) {
|
|
time := model.GetMillis() - timePeriod
|
|
query := us.getQueryBuilder().Select("COUNT(*)").From("Status AS s").Where("LastActivityAt > ?", time)
|
|
|
|
if !options.IncludeBotAccounts {
|
|
query = query.LeftJoin("Bots ON s.UserId = Bots.UserId").Where("Bots.UserId IS NULL")
|
|
}
|
|
|
|
if !options.IncludeRemoteUsers || !options.IncludeDeleted {
|
|
query = query.LeftJoin("Users ON s.UserId = Users.Id")
|
|
}
|
|
|
|
if !options.IncludeRemoteUsers {
|
|
query = query.Where(sq.Or{sq.Eq{"Users.RemoteId": ""}, sq.Eq{"Users.RemoteId": nil}})
|
|
}
|
|
|
|
if !options.IncludeDeleted {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
queryStr, args, err := query.ToSql()
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "analytics_active_count_tosql")
|
|
}
|
|
|
|
var v int64
|
|
err = us.GetReplica().Get(&v, queryStr, args...)
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "failed to count Users")
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AnalyticsActiveCountForPeriod(startTime int64, endTime int64, options model.UserCountOptions) (int32, error) {
|
|
query := us.getQueryBuilder().Select("COUNT(*)").From("Status AS s").Where("LastActivityAt > ? AND LastActivityAt <= ?", startTime, endTime)
|
|
|
|
if !options.IncludeBotAccounts {
|
|
query = query.LeftJoin("Bots ON s.UserId = Bots.UserId").Where("Bots.UserId IS NULL")
|
|
}
|
|
|
|
if !options.IncludeRemoteUsers || !options.IncludeDeleted {
|
|
query = query.LeftJoin("Users ON s.UserId = Users.Id")
|
|
}
|
|
|
|
if !options.IncludeRemoteUsers {
|
|
query = query.Where(sq.Or{sq.Eq{"Users.RemoteId": ""}, sq.Eq{"Users.RemoteId": nil}})
|
|
}
|
|
|
|
if !options.IncludeDeleted {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
queryStr, args, err := query.ToSql()
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "Failed to build query.")
|
|
}
|
|
|
|
var v int32
|
|
err = us.GetReplica().Get(&v, queryStr, args...)
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "Unable to get the active users during the requested period.")
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetUnreadCount(userId string, isCRTEnabled bool) (int64, error) {
|
|
mentionCountColumn := "cm.MentionCount"
|
|
if isCRTEnabled {
|
|
mentionCountColumn = "cm.MentionCountRoot"
|
|
}
|
|
|
|
query := `
|
|
SELECT SUM(` + mentionCountColumn + `)
|
|
FROM Channels c
|
|
INNER JOIN ChannelMembers cm
|
|
ON cm.ChannelId = c.Id
|
|
AND cm.UserId = ?
|
|
AND c.DeleteAt = 0
|
|
`
|
|
|
|
var count int64
|
|
err := us.GetReplica().Get(&count, query, userId)
|
|
if err != nil {
|
|
return count, errors.Wrapf(err, "failed to count unread Channels for userId=%s", userId)
|
|
}
|
|
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetUnreadCountForChannel(userId string, channelId string) (int64, error) {
|
|
var count int64
|
|
err := us.GetReplica().Get(&count, "SELECT SUM(CASE WHEN c.Type = ? THEN (c.TotalMsgCount - cm.MsgCount) ELSE cm.MentionCount END) FROM Channels c INNER JOIN ChannelMembers cm ON c.Id = cm.ChannelId AND cm.ChannelId = ? AND cm.UserId = ?", model.ChannelTypeDirect, channelId, userId)
|
|
if err != nil {
|
|
return 0, errors.Wrapf(err, "failed to get unread count for channelId=%s and userId=%s", channelId, userId)
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetAnyUnreadPostCountForChannel(userId string, channelId string) (int64, error) {
|
|
var count int64
|
|
err := us.GetReplica().Get(&count, "SELECT SUM(c.TotalMsgCount - cm.MsgCount) FROM Channels c INNER JOIN ChannelMembers cm ON c.Id = cm.ChannelId AND cm.ChannelId = ? AND cm.UserId = ?", channelId, userId)
|
|
if err != nil {
|
|
return count, errors.Wrapf(err, "failed to get any unread count for channelId=%s and userId=%s", channelId, userId)
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) Search(rctx request.CTX, teamId string, term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
OrderBy("Username ASC").
|
|
Limit(uint64(options.Limit))
|
|
|
|
if teamId != "" {
|
|
query = query.Join("TeamMembers tm ON ( tm.UserId = Users.Id AND tm.DeleteAt = 0 AND tm.TeamId = ? )", teamId)
|
|
}
|
|
return us.performSearch(query, term, options)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchCommonContentFlaggingReviewers(term string) ([]*model.User, error) {
|
|
query := us.getQueryBuilder().
|
|
Select(getUsersColumns()...).
|
|
Columns(getBotInfoColumns()...).
|
|
From("ContentFlaggingCommonReviewers").
|
|
LeftJoin("Users ON Users.Id = ContentFlaggingCommonReviewers.UserId").
|
|
LeftJoin("Bots b ON (b.UserId = Users.Id)").
|
|
OrderBy("Users.Username ASC").
|
|
Limit(ContentReviewerSearchLimit)
|
|
|
|
searchOptions := &model.UserSearchOptions{
|
|
AllowEmails: false,
|
|
AllowFullNames: true,
|
|
AllowInactive: false,
|
|
Limit: 50,
|
|
}
|
|
|
|
return us.performSearch(query, term, searchOptions)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchTeamContentFlaggingReviewers(teamId, term string) ([]*model.User, error) {
|
|
query := us.getQueryBuilder().
|
|
Select(getUsersColumns()...).
|
|
Columns(getBotInfoColumns()...).
|
|
From("ContentFlaggingTeamReviewers").
|
|
LeftJoin("Users ON Users.Id = ContentFlaggingTeamReviewers.UserId").
|
|
LeftJoin("Bots b ON (b.UserId = Users.Id)").
|
|
Where("ContentFlaggingTeamReviewers.TeamId = ?", teamId).
|
|
OrderBy("Users.Username ASC").
|
|
Limit(ContentReviewerSearchLimit)
|
|
|
|
searchOptions := &model.UserSearchOptions{
|
|
AllowEmails: false,
|
|
AllowFullNames: true,
|
|
AllowInactive: false,
|
|
Limit: 50,
|
|
}
|
|
|
|
return us.performSearch(query, term, searchOptions)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchWithoutTeam(term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Where(`(
|
|
SELECT
|
|
COUNT(0)
|
|
FROM
|
|
TeamMembers
|
|
WHERE
|
|
TeamMembers.UserId = Users.Id
|
|
AND TeamMembers.DeleteAt = 0
|
|
) = 0`).
|
|
OrderBy("Users.Username ASC").
|
|
Limit(uint64(options.Limit))
|
|
|
|
return us.performSearch(query, term, options)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchNotInTeam(notInTeamId string, term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
LeftJoin("TeamMembers tm ON ( tm.UserId = Users.Id AND tm.DeleteAt = 0 AND tm.TeamId = ? )", notInTeamId).
|
|
Where("tm.UserId IS NULL").
|
|
OrderBy("Users.Username ASC").
|
|
Limit(uint64(options.Limit))
|
|
|
|
if options.GroupConstrained {
|
|
query = applyTeamGroupConstrainedFilter(query, notInTeamId)
|
|
}
|
|
|
|
return us.performSearch(query, term, options)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchNotInChannel(teamId string, channelId string, term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
LeftJoin("ChannelMembers cm ON ( cm.UserId = Users.Id AND cm.ChannelId = ? )", channelId).
|
|
Where("cm.UserId IS NULL").
|
|
OrderBy("Username ASC").
|
|
Limit(uint64(options.Limit))
|
|
|
|
if teamId != "" {
|
|
query = query.Join("TeamMembers tm ON ( tm.UserId = Users.Id AND tm.DeleteAt = 0 AND tm.TeamId = ? )", teamId)
|
|
}
|
|
|
|
if options.GroupConstrained {
|
|
query = applyChannelGroupConstrainedFilter(query, channelId)
|
|
}
|
|
|
|
return us.performSearch(query, term, options)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchInChannel(channelId string, term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("ChannelMembers cm ON ( cm.UserId = Users.Id AND cm.ChannelId = ? )", channelId).
|
|
OrderBy("Username ASC").
|
|
Limit(uint64(options.Limit))
|
|
|
|
return us.performSearch(query, term, options)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchInGroup(groupID string, term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
Join("GroupMembers gm ON ( gm.UserId = Users.Id AND gm.GroupId = ? AND gm.DeleteAt = 0 )", groupID).
|
|
OrderBy("Username ASC").
|
|
Limit(uint64(options.Limit))
|
|
|
|
return us.performSearch(query, term, options)
|
|
}
|
|
|
|
func (us SqlUserStore) SearchNotInGroup(groupID string, term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
query := us.usersQuery.
|
|
LeftJoin("GroupMembers gm ON ( gm.UserId = Users.Id AND gm.GroupId = ? )", groupID).
|
|
Where("(gm.UserId IS NULL OR gm.deleteat != 0)").
|
|
OrderBy("Username ASC").
|
|
Limit(uint64(options.Limit))
|
|
|
|
return us.performSearch(query, term, options)
|
|
}
|
|
|
|
func generateSearchQuery(query sq.SelectBuilder, terms []string, fields []string) sq.SelectBuilder {
|
|
for _, term := range terms {
|
|
searchFields := []string{}
|
|
termArgs := []any{}
|
|
for _, field := range fields {
|
|
searchFields = append(searchFields, fmt.Sprintf("lower(%s) LIKE lower(?) escape '*' ", field))
|
|
termArgs = append(termArgs, fmt.Sprintf("%%%s%%", strings.TrimLeft(term, "@")))
|
|
}
|
|
searchFields = append(searchFields, "Id = ?")
|
|
termArgs = append(termArgs, strings.TrimLeft(term, "@"))
|
|
query = query.Where(fmt.Sprintf("(%s)", strings.Join(searchFields, " OR ")), termArgs...)
|
|
}
|
|
|
|
return query
|
|
}
|
|
|
|
func (us SqlUserStore) performSearch(query sq.SelectBuilder, term string, options *model.UserSearchOptions) ([]*model.User, error) {
|
|
term = sanitizeSearchTerm(term, "*")
|
|
|
|
var searchType []string
|
|
if options.AllowEmails {
|
|
if options.AllowFullNames {
|
|
searchType = UserSearchTypeAll
|
|
} else {
|
|
searchType = UserSearchTypeAllNoFullName
|
|
}
|
|
} else {
|
|
if options.AllowFullNames {
|
|
searchType = UserSearchTypeNames
|
|
} else {
|
|
searchType = UserSearchTypeNamesNoFullName
|
|
}
|
|
}
|
|
|
|
query = applyRoleFilter(query, options.Role)
|
|
query = applyMultiRoleFilters(query, options.Roles, options.TeamRoles, options.ChannelRoles)
|
|
|
|
if !options.AllowInactive {
|
|
query = query.Where("Users.DeleteAt = 0")
|
|
}
|
|
|
|
if strings.TrimSpace(term) != "" {
|
|
query = generateSearchQuery(query, strings.Fields(term), searchType)
|
|
}
|
|
|
|
query = applyViewRestrictionsFilter(query, options.ViewRestrictions, true)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "perform_search_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to find Users with term=%s and searchType=%v", term, searchType)
|
|
}
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AnalyticsGetInactiveUsersCount() (int64, error) {
|
|
query := us.getQueryBuilder().
|
|
Select("COUNT(Id)").
|
|
From("Users").
|
|
LeftJoin("Bots ON Users.ID = Bots.UserId").
|
|
Where(sq.And{
|
|
sq.Gt{"Users.DeleteAt": 0},
|
|
sq.Eq{"Bots.UserId": nil},
|
|
})
|
|
|
|
var count int64
|
|
err := us.GetReplica().GetBuilder(&count, query)
|
|
if err != nil {
|
|
return int64(0), errors.Wrap(err, "failed to count inactive Users")
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AnalyticsGetExternalUsers(hostDomain string) (bool, error) {
|
|
var count int64
|
|
err := us.GetReplica().Get(&count, "SELECT COUNT(Id) FROM Users WHERE LOWER(Email) NOT LIKE ?", "%@"+strings.ToLower(hostDomain))
|
|
if err != nil {
|
|
return false, errors.Wrap(err, "failed to count inactive Users")
|
|
}
|
|
return count > 0, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AnalyticsGetGuestCount() (int64, error) {
|
|
var count int64
|
|
err := us.GetReplica().Get(&count, "SELECT count(*) FROM Users WHERE Roles LIKE ? and DeleteAt = 0", "%system_guest%")
|
|
if err != nil {
|
|
return int64(0), errors.Wrap(err, "failed to count guest Users")
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AnalyticsGetSingleChannelGuestCount() (int64, error) {
|
|
subQuery := us.getQueryBuilder().
|
|
Select("cm.UserId").
|
|
From("ChannelMembers cm").
|
|
Join("Users u ON cm.UserId = u.Id").
|
|
Join("Channels c ON cm.ChannelId = c.Id").
|
|
Where(sq.ILike{"u.Roles": "%system_guest%"}).
|
|
Where(sq.Eq{"u.DeleteAt": 0}).
|
|
Where(sq.Eq{"c.DeleteAt": 0}).
|
|
Where(sq.Eq{"c.Type": []model.ChannelType{model.ChannelTypeOpen, model.ChannelTypePrivate}}).
|
|
GroupBy("cm.UserId").
|
|
Having("COUNT(cm.ChannelId) = 1")
|
|
|
|
query := us.getQueryBuilder().
|
|
Select("COUNT(*)").
|
|
FromSelect(subQuery, "single_channel_guests")
|
|
|
|
var count int64
|
|
err := us.GetReplica().GetBuilder(&count, query)
|
|
if err != nil {
|
|
return int64(0), errors.Wrap(err, "failed to count single-channel guest Users")
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AnalyticsGetSystemAdminCount() (int64, error) {
|
|
var count int64
|
|
err := us.GetReplica().Get(&count, "SELECT count(*) FROM Users WHERE Roles LIKE ? and DeleteAt = 0", "%system_admin%")
|
|
if err != nil {
|
|
return int64(0), errors.Wrap(err, "failed to count system admin Users")
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetProfilesNotInTeam(teamId string, groupConstrained bool, offset int, limit int, viewRestrictions *model.ViewUsersRestrictions) ([]*model.User, error) {
|
|
users := []*model.User{}
|
|
query := us.usersQuery.
|
|
LeftJoin("TeamMembers tm ON ( tm.UserId = Users.Id AND tm.DeleteAt = 0 AND tm.TeamId = ? )", teamId).
|
|
Where("tm.UserId IS NULL").
|
|
OrderBy("Users.Username ASC").
|
|
Offset(uint64(offset)).Limit(uint64(limit))
|
|
|
|
query = applyViewRestrictionsFilter(query, viewRestrictions, true)
|
|
|
|
if groupConstrained {
|
|
query = applyTeamGroupConstrainedFilter(query, teamId)
|
|
}
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_profiles_not_in_team_tosql")
|
|
}
|
|
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetEtagForProfilesNotInTeam(teamId string) string {
|
|
querystr := `
|
|
SELECT
|
|
CONCAT(MAX(UpdateAt), '.', COUNT(Id)) as etag
|
|
FROM
|
|
Users
|
|
LEFT JOIN TeamMembers tm
|
|
ON tm.UserId = Users.Id
|
|
AND tm.TeamId = ?
|
|
AND tm.DeleteAt = 0
|
|
WHERE
|
|
tm.UserId IS NULL
|
|
`
|
|
var etag string
|
|
err := us.GetReplica().Get(&etag, querystr, teamId)
|
|
if err != nil {
|
|
return fmt.Sprintf("%v.%v", model.CurrentVersion, model.GetMillis())
|
|
}
|
|
|
|
return fmt.Sprintf("%v.%v", model.CurrentVersion, etag)
|
|
}
|
|
|
|
func (us SqlUserStore) ClearAllCustomRoleAssignments() (err error) {
|
|
builtInRoles := model.MakeDefaultRoles()
|
|
lastUserId := strings.Repeat("0", 26)
|
|
|
|
for {
|
|
var transaction *sqlxTxWrapper
|
|
var err error
|
|
|
|
if transaction, err = us.GetMaster().Begin(); err != nil {
|
|
return errors.Wrap(err, "begin_transaction")
|
|
}
|
|
defer finalizeTransactionX(transaction, &err)
|
|
|
|
users := []*model.User{}
|
|
query := us.usersQuery.Where(sq.Gt{"Users.Id": lastUserId}).OrderBy("Users.Id").Limit(1000)
|
|
if err := transaction.SelectBuilder(&users, query); err != nil {
|
|
return errors.Wrapf(err, "failed to find Users with id > %s", lastUserId)
|
|
}
|
|
|
|
if len(users) == 0 {
|
|
break
|
|
}
|
|
|
|
for _, user := range users {
|
|
lastUserId = user.Id
|
|
|
|
var newRoles []string
|
|
|
|
for role := range strings.FieldsSeq(user.Roles) {
|
|
for name := range builtInRoles {
|
|
if name == role {
|
|
newRoles = append(newRoles, role)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
newRolesString := strings.Join(newRoles, " ")
|
|
if newRolesString != user.Roles {
|
|
if _, err := transaction.Exec("UPDATE Users SET Roles = ? WHERE Id = ?", newRolesString, user.Id); err != nil {
|
|
return errors.Wrap(err, "failed to update Users")
|
|
}
|
|
}
|
|
}
|
|
|
|
if err := transaction.Commit(); err != nil {
|
|
return errors.Wrap(err, "commit_transaction")
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) InferSystemInstallDate() (int64, error) {
|
|
var createAt int64
|
|
err := us.GetReplica().Get(&createAt, "SELECT CreateAt FROM Users WHERE CreateAt IS NOT NULL ORDER BY CreateAt ASC LIMIT 1")
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "failed to infer system install date")
|
|
}
|
|
|
|
return createAt, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetUsersBatchForIndexing(startTime int64, startFileID string, limit int) ([]*model.UserForIndexing, error) {
|
|
users := []*model.User{}
|
|
usersQuery, args, err := us.usersQuery.
|
|
Where(sq.Or{
|
|
sq.Gt{"Users.CreateAt": startTime},
|
|
sq.And{
|
|
sq.Eq{"Users.CreateAt": startTime},
|
|
sq.Gt{"Users.Id": startFileID},
|
|
},
|
|
}).
|
|
OrderBy("Users.CreateAt ASC, Users.Id ASC").
|
|
Limit(uint64(limit)).
|
|
ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "GetUsersBatchForIndexing_ToSql1")
|
|
}
|
|
|
|
err = us.GetSearchReplicaX().Select(&users, usersQuery, args...)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
userIds := []string{}
|
|
for _, user := range users {
|
|
userIds = append(userIds, user.Id)
|
|
}
|
|
|
|
channelMembers := []*model.ChannelMember{}
|
|
channelMembersQuery, args, err := us.getQueryBuilder().
|
|
Select(`
|
|
cm.ChannelId,
|
|
cm.UserId,
|
|
cm.Roles,
|
|
cm.LastViewedAt,
|
|
cm.MsgCount,
|
|
cm.MentionCount,
|
|
cm.MentionCountRoot,
|
|
cm.NotifyProps,
|
|
cm.LastUpdateAt,
|
|
cm.SchemeUser,
|
|
cm.SchemeAdmin,
|
|
(cm.SchemeGuest IS NOT NULL AND cm.SchemeGuest) as SchemeGuest
|
|
`).
|
|
From("ChannelMembers cm").
|
|
Join("Channels c ON cm.ChannelId = c.Id").
|
|
Where(sq.And{
|
|
sq.Eq{
|
|
"cm.UserId": userIds,
|
|
},
|
|
sq.Or{
|
|
sq.Eq{"c.Type": model.ChannelTypeOpen},
|
|
sq.Eq{"c.Type": model.ChannelTypeDirect},
|
|
sq.Eq{"c.Type": model.ChannelTypeGroup},
|
|
},
|
|
}).
|
|
ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "GetUsersBatchForIndexing_ToSql2")
|
|
}
|
|
|
|
err = us.GetSearchReplicaX().Select(&channelMembers, channelMembersQuery, args...)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to find ChannelMembers")
|
|
}
|
|
|
|
teamMembers := []*model.TeamMember{}
|
|
teamMembersQuery, args, err := us.getQueryBuilder().
|
|
Select("TeamId, UserId, Roles, DeleteAt, (SchemeGuest IS NOT NULL AND SchemeGuest) as SchemeGuest, SchemeUser, SchemeAdmin").
|
|
From("TeamMembers").
|
|
Where(sq.Eq{"UserId": userIds, "DeleteAt": 0}).
|
|
ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "GetUsersBatchForIndexing_ToSql3")
|
|
}
|
|
|
|
err = us.GetSearchReplicaX().Select(&teamMembers, teamMembersQuery, args...)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to find TeamMembers")
|
|
}
|
|
|
|
userMap := map[string]*model.UserForIndexing{}
|
|
for _, user := range users {
|
|
userMap[user.Id] = &model.UserForIndexing{
|
|
Id: user.Id,
|
|
Username: user.Username,
|
|
Nickname: user.Nickname,
|
|
FirstName: user.FirstName,
|
|
LastName: user.LastName,
|
|
Roles: user.Roles,
|
|
CreateAt: user.CreateAt,
|
|
DeleteAt: user.DeleteAt,
|
|
TeamsIds: []string{},
|
|
ChannelsIds: []string{},
|
|
}
|
|
}
|
|
|
|
for _, c := range channelMembers {
|
|
if userMap[c.UserId] != nil {
|
|
userMap[c.UserId].ChannelsIds = append(userMap[c.UserId].ChannelsIds, c.ChannelId)
|
|
}
|
|
}
|
|
for _, t := range teamMembers {
|
|
if userMap[t.UserId] != nil {
|
|
userMap[t.UserId].TeamsIds = append(userMap[t.UserId].TeamsIds, t.TeamId)
|
|
}
|
|
}
|
|
|
|
usersForIndexing := []*model.UserForIndexing{}
|
|
for _, user := range userMap {
|
|
usersForIndexing = append(usersForIndexing, user)
|
|
}
|
|
sort.Slice(usersForIndexing, func(i, j int) bool {
|
|
return usersForIndexing[i].CreateAt < usersForIndexing[j].CreateAt
|
|
})
|
|
|
|
return usersForIndexing, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetTeamGroupUsers(teamID string) ([]*model.User, error) {
|
|
query := applyTeamGroupConstrainedFilter(us.usersQuery, teamID)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_team_group_users_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetChannelGroupUsers(channelID string) ([]*model.User, error) {
|
|
query := applyChannelGroupConstrainedFilter(us.usersQuery, channelID)
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "get_channel_group_users_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "failed to find Users")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func applyViewRestrictionsFilter(query sq.SelectBuilder, restrictions *model.ViewUsersRestrictions, distinct bool) sq.SelectBuilder {
|
|
if restrictions == nil {
|
|
return query
|
|
}
|
|
|
|
// If you have no access to teams or channels, return and empty result.
|
|
if restrictions.Teams != nil && len(restrictions.Teams) == 0 && restrictions.Channels != nil && len(restrictions.Channels) == 0 {
|
|
return query.Where("1 = 0")
|
|
}
|
|
|
|
teams := make([]any, len(restrictions.Teams))
|
|
for i, v := range restrictions.Teams {
|
|
teams[i] = v
|
|
}
|
|
channels := make([]any, len(restrictions.Channels))
|
|
for i, v := range restrictions.Channels {
|
|
channels[i] = v
|
|
}
|
|
resultQuery := query
|
|
if len(restrictions.Teams) > 0 {
|
|
resultQuery = resultQuery.Join(fmt.Sprintf("TeamMembers rtm ON ( rtm.UserId = Users.Id AND rtm.DeleteAt = 0 AND rtm.TeamId IN (%s))", sq.Placeholders(len(teams))), teams...)
|
|
}
|
|
if len(restrictions.Channels) > 0 {
|
|
resultQuery = resultQuery.Join(fmt.Sprintf("ChannelMembers rcm ON ( rcm.UserId = Users.Id AND rcm.ChannelId IN (%s))", sq.Placeholders(len(channels))), channels...)
|
|
}
|
|
|
|
if distinct {
|
|
return resultQuery.Distinct()
|
|
}
|
|
|
|
return resultQuery
|
|
}
|
|
|
|
func (us SqlUserStore) PromoteGuestToUser(userId string) (err error) {
|
|
transaction, err := us.GetMaster().Begin()
|
|
if err != nil {
|
|
return errors.Wrap(err, "begin_transaction")
|
|
}
|
|
defer finalizeTransactionX(transaction, &err)
|
|
|
|
user, err := us.Get(context.Background(), userId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
roles := user.GetRoles()
|
|
|
|
for idx, role := range roles {
|
|
if role == "system_guest" {
|
|
roles[idx] = "system_user"
|
|
}
|
|
}
|
|
|
|
curTime := model.GetMillis()
|
|
query := us.getQueryBuilder().Update("Users").
|
|
Set("Roles", strings.Join(roles, " ")).
|
|
Set("UpdateAt", curTime).
|
|
Where(sq.Eq{"Id": userId})
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "promote_guest_to_user_tosql")
|
|
}
|
|
|
|
if _, err = transaction.Exec(queryString, args...); err != nil {
|
|
return errors.Wrapf(err, "failed to update User with userId=%s", userId)
|
|
}
|
|
|
|
query = us.getQueryBuilder().Update("ChannelMembers").
|
|
Set("SchemeUser", true).
|
|
Set("SchemeGuest", false).
|
|
Where(sq.Eq{"UserId": userId})
|
|
|
|
queryString, args, err = query.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "promote_guest_to_user_tosql")
|
|
}
|
|
|
|
if _, err = transaction.Exec(queryString, args...); err != nil {
|
|
return errors.Wrapf(err, "failed to update ChannelMembers with userId=%s", userId)
|
|
}
|
|
|
|
query = us.getQueryBuilder().Update("TeamMembers").
|
|
Set("SchemeUser", true).
|
|
Set("SchemeGuest", false).
|
|
Where(sq.Eq{"UserId": userId})
|
|
|
|
queryString, args, err = query.ToSql()
|
|
if err != nil {
|
|
return errors.Wrap(err, "promote_guest_to_user_tosql")
|
|
}
|
|
|
|
if _, err := transaction.Exec(queryString, args...); err != nil {
|
|
return errors.Wrapf(err, "failed to update TeamMembers with userId=%s", userId)
|
|
}
|
|
|
|
if err := transaction.Commit(); err != nil {
|
|
return errors.Wrap(err, "commit_transaction")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (us SqlUserStore) DemoteUserToGuest(userID string) (_ *model.User, err error) {
|
|
transaction, err := us.GetMaster().Begin()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "begin_transaction")
|
|
}
|
|
defer finalizeTransactionX(transaction, &err)
|
|
|
|
user, err := us.Get(context.Background(), userID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
curTime := model.GetMillis()
|
|
newRolesDBStr := model.SystemGuestRoleId
|
|
|
|
query := us.getQueryBuilder().Update("Users").
|
|
Set("Roles", newRolesDBStr).
|
|
Set("UpdateAt", curTime).
|
|
Where(sq.Eq{"Id": userID})
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "demote_user_to_guest_tosql")
|
|
}
|
|
|
|
if _, err = transaction.Exec(queryString, args...); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to update User with userId=%s", userID)
|
|
}
|
|
|
|
user.Roles = newRolesDBStr
|
|
user.UpdateAt = curTime
|
|
|
|
query = us.getQueryBuilder().Update("ChannelMembers").
|
|
Set("SchemeUser", false).
|
|
Set("SchemeAdmin", false).
|
|
Set("SchemeGuest", true).
|
|
Where(sq.Eq{"UserId": userID})
|
|
|
|
queryString, args, err = query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "demote_user_to_guest_tosql")
|
|
}
|
|
|
|
if _, err = transaction.Exec(queryString, args...); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to update ChannelMembers with userId=%s", userID)
|
|
}
|
|
|
|
query = us.getQueryBuilder().Update("TeamMembers").
|
|
Set("SchemeUser", false).
|
|
Set("SchemeAdmin", false).
|
|
Set("SchemeGuest", true).
|
|
Where(sq.Eq{"UserId": userID})
|
|
|
|
queryString, args, err = query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "demote_user_to_guest_tosql")
|
|
}
|
|
|
|
if _, err := transaction.Exec(queryString, args...); err != nil {
|
|
return nil, errors.Wrapf(err, "failed to update TeamMembers with userId=%s", userID)
|
|
}
|
|
|
|
if err := transaction.Commit(); err != nil {
|
|
return nil, errors.Wrap(err, "commit_transaction")
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func (us SqlUserStore) AutocompleteUsersInChannel(rctx request.CTX, teamId, channelId, term string, options *model.UserSearchOptions) (*model.UserAutocompleteInChannel, error) {
|
|
var usersInChannel, usersNotInChannel []*model.User
|
|
g := errgroup.Group{}
|
|
g.Go(func() (err error) {
|
|
usersInChannel, err = us.SearchInChannel(channelId, term, options)
|
|
return err
|
|
})
|
|
g.Go(func() (err error) {
|
|
usersNotInChannel, err = us.SearchNotInChannel(teamId, channelId, term, options)
|
|
return err
|
|
})
|
|
err := g.Wait()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &model.UserAutocompleteInChannel{
|
|
InChannel: usersInChannel,
|
|
OutOfChannel: usersNotInChannel,
|
|
}, nil
|
|
}
|
|
|
|
// GetKnownUsers returns the list of user ids of users with any direct
|
|
// relationship with a user. That means any user sharing any channel, including
|
|
// direct and group channels.
|
|
func (us SqlUserStore) GetKnownUsers(userId string) ([]string, error) {
|
|
userIds := []string{}
|
|
usersQuery, args, err := us.getQueryBuilder().
|
|
Select("DISTINCT ocm.UserId").
|
|
From("ChannelMembers AS cm").
|
|
Join("ChannelMembers AS ocm ON ocm.ChannelId = cm.ChannelId").
|
|
Where(sq.NotEq{"ocm.UserId": userId}).
|
|
Where(sq.Eq{"cm.UserId": userId}).
|
|
ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "GetKnownUsers_ToSql")
|
|
}
|
|
|
|
err = us.GetSearchReplicaX().Select(&userIds, usersQuery, args...)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to find ChannelMembers")
|
|
}
|
|
|
|
return userIds, nil
|
|
}
|
|
|
|
// IsEmpty returns whether or not the Users table is empty.
|
|
func (us SqlUserStore) IsEmpty(excludeBots bool) (bool, error) {
|
|
var hasRows bool
|
|
|
|
builder := us.getQueryBuilder().
|
|
Select("1").
|
|
Prefix("SELECT EXISTS (").
|
|
From("Users")
|
|
|
|
if excludeBots {
|
|
builder = builder.LeftJoin("Bots ON Users.Id = Bots.UserId").Where("Bots.UserId IS NULL")
|
|
}
|
|
|
|
builder = builder.Suffix(")")
|
|
|
|
query, args, err := builder.ToSql()
|
|
if err != nil {
|
|
return false, errors.Wrapf(err, "users_is_empty_to_sql")
|
|
}
|
|
|
|
if err = us.GetReplica().Get(&hasRows, query, args...); err != nil {
|
|
return false, errors.Wrap(err, "failed to check if table is empty")
|
|
}
|
|
return !hasRows, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetUsersWithInvalidEmails(page int, perPage int, restrictedDomains string) ([]*model.User, error) {
|
|
domainArray := strings.Split(restrictedDomains, ",")
|
|
query := us.usersQuery.
|
|
LeftJoin("Bots ON Users.Id = Bots.UserId").
|
|
Where("Bots.UserId IS NULL").
|
|
Where("Users.Roles != 'system_guest'").
|
|
Where("Users.DeleteAt = 0").
|
|
Where("(Users.AuthService = '' OR Users.AuthService IS NULL)")
|
|
|
|
for _, d := range domainArray {
|
|
if d != "" {
|
|
query = query.Where("Users.Email NOT LIKE LOWER(?)", wildcardSearchTerm(d))
|
|
}
|
|
}
|
|
|
|
query = query.Offset(uint64(page * perPage)).Limit(uint64(perPage))
|
|
|
|
queryString, args, err := query.ToSql()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "users_get_many_tosql")
|
|
}
|
|
|
|
users := []*model.User{}
|
|
if err := us.GetReplica().Select(&users, queryString, args...); err != nil {
|
|
return nil, errors.Wrap(err, "users_get_many_select")
|
|
}
|
|
|
|
for _, u := range users {
|
|
u.Sanitize(map[string]bool{})
|
|
}
|
|
|
|
return users, nil
|
|
}
|
|
|
|
func (us SqlUserStore) RefreshPostStatsForUsers() error {
|
|
ctx, cancel := us.analyticsContext()
|
|
defer cancel()
|
|
|
|
if _, err := us.GetMaster().ExecContext(ctx, "REFRESH MATERIALIZED VIEW poststats"); err != nil {
|
|
return errors.Wrap(err, "users_refresh_post_stats_exec")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func applyUserReportFilter(query sq.SelectBuilder, filter *model.UserReportOptions) sq.SelectBuilder {
|
|
switch filter.GuestFilter {
|
|
case model.GuestFilterAll:
|
|
query = applyRoleFilter(query, "system_guest")
|
|
case model.GuestFilterSingleChannel:
|
|
query = applyRoleFilter(query, "system_guest")
|
|
query = query.Where(sq.Expr("(SELECT COUNT(*) FROM ChannelMembers cm INNER JOIN Channels c ON c.Id = cm.ChannelId AND c.DeleteAt = 0 AND c.Type IN ('O','P') WHERE cm.UserId = Users.Id) = 1"))
|
|
case model.GuestFilterMultipleChannel:
|
|
query = applyRoleFilter(query, "system_guest")
|
|
query = query.Where(sq.Expr("(SELECT COUNT(*) FROM ChannelMembers cm INNER JOIN Channels c ON c.Id = cm.ChannelId AND c.DeleteAt = 0 AND c.Type IN ('O','P') WHERE cm.UserId = Users.Id) > 1"))
|
|
default:
|
|
query = applyRoleFilter(query, filter.Role)
|
|
}
|
|
|
|
if filter.HasNoTeam {
|
|
query = query.Where(sq.Expr("Users.Id NOT IN (SELECT UserId FROM TeamMembers WHERE DeleteAt = 0)"))
|
|
} else if filter.Team != "" {
|
|
query = query.Join("TeamMembers tm ON (tm.UserId = Users.Id AND tm.DeleteAt = 0)").
|
|
Where(sq.Eq{"tm.TeamId": filter.Team})
|
|
}
|
|
if filter.HideActive {
|
|
query = query.Where(sq.Gt{"Users.DeleteAt": 0})
|
|
}
|
|
if filter.HideInactive {
|
|
query = query.Where(sq.Eq{"Users.DeleteAt": 0})
|
|
}
|
|
|
|
if strings.TrimSpace(filter.SearchTerm) != "" {
|
|
query = generateSearchQuery(query, strings.Fields(sanitizeSearchTerm(filter.SearchTerm, "*")), UserSearchTypeAll)
|
|
}
|
|
|
|
return query
|
|
}
|
|
|
|
func (us SqlUserStore) GetUserCountForReport(filter *model.UserReportOptions) (int64, error) {
|
|
query := us.getQueryBuilder().
|
|
Select("COUNT(Users.Id)").
|
|
From("Users").
|
|
LeftJoin("Bots ON Users.Id = Bots.UserId").
|
|
Where("Bots.UserId IS NULL")
|
|
|
|
query = applyUserReportFilter(query, filter)
|
|
queryStr, args, err := query.ToSql()
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "user_count_report_tosql")
|
|
}
|
|
var v int64
|
|
err = us.GetReplica().Get(&v, queryStr, args...)
|
|
if err != nil {
|
|
return 0, errors.Wrap(err, "failed to count Users for report")
|
|
}
|
|
return v, nil
|
|
}
|
|
|
|
func (us SqlUserStore) GetUserReport(filter *model.UserReportOptions) ([]*model.UserReportQuery, error) {
|
|
selectColumns := append(getUsersColumns(),
|
|
"MAX(s.LastActivityAt) AS LastStatusAt",
|
|
"MAX(ps.LastPostDate) AS LastPostDate",
|
|
"COUNT(ps.Day) AS DaysActive",
|
|
"SUM(ps.NumPosts) AS TotalPosts",
|
|
"(SELECT COUNT(*) FROM ChannelMembers cm INNER JOIN Channels c ON c.Id = cm.ChannelId AND c.DeleteAt = 0 AND c.Type IN ('O','P') WHERE cm.UserId = Users.Id) AS ChannelCount",
|
|
)
|
|
|
|
sortDirection := "ASC"
|
|
if filter.SortDesc {
|
|
sortDirection = "DESC"
|
|
}
|
|
|
|
query := us.getQueryBuilder().
|
|
Select(selectColumns...).
|
|
From("Users").
|
|
LeftJoin("Status s ON s.UserId = Users.Id").
|
|
Where(sq.Expr("Users.Id NOT IN (SELECT UserId FROM Bots)")).
|
|
GroupBy("Users.Id")
|
|
|
|
// no need to apply any filtering and pagination if there are no
|
|
// previous element ID and value provided.
|
|
if filter.FromId != "" && filter.FromColumnValue != "" {
|
|
if (filter.Direction == "prev" && !filter.SortDesc) || (filter.Direction == "next" && filter.SortDesc) {
|
|
sortDirection = "DESC"
|
|
|
|
query = query.Where(sq.Or{
|
|
sq.Lt{filter.SortColumn: filter.FromColumnValue},
|
|
sq.And{
|
|
sq.Eq{filter.SortColumn: filter.FromColumnValue},
|
|
sq.Lt{"Users.Id": filter.FromId},
|
|
},
|
|
})
|
|
} else {
|
|
sortDirection = "ASC"
|
|
|
|
query = query.Where(sq.Or{
|
|
sq.Gt{filter.SortColumn: filter.FromColumnValue},
|
|
sq.And{
|
|
sq.Eq{filter.SortColumn: filter.FromColumnValue},
|
|
sq.Gt{"Users.Id": filter.FromId},
|
|
},
|
|
})
|
|
}
|
|
}
|
|
|
|
query = query.OrderBy(filter.SortColumn+" "+sortDirection, "Users.Id")
|
|
|
|
if filter.PageSize > 0 {
|
|
query = query.Limit(uint64(filter.PageSize))
|
|
}
|
|
|
|
joinSql := sq.And{}
|
|
if filter.StartAt > 0 {
|
|
startDate := time.UnixMilli(filter.StartAt)
|
|
joinSql = append(joinSql, sq.GtOrEq{"ps.Day": startDate.Format("2006-01-02")})
|
|
}
|
|
if filter.EndAt > 0 {
|
|
endDate := time.UnixMilli(filter.EndAt)
|
|
joinSql = append(joinSql, sq.Lt{"ps.Day": endDate.Format("2006-01-02")})
|
|
}
|
|
sql, args, err := joinSql.ToSql()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
query = query.LeftJoin("PostStats ps ON ps.UserId = Users.Id AND "+sql, args...)
|
|
|
|
query = applyUserReportFilter(query, filter)
|
|
|
|
parentQuery := query
|
|
// If we're going a page back...
|
|
//
|
|
// The way pagination works, we get the previous page's rows
|
|
// in reverse order. So, we use parent query on it to
|
|
// reverse the order in database itself.
|
|
if filter.Direction == "prev" {
|
|
reverseSortDirection := "ASC"
|
|
if sortDirection == "ASC" {
|
|
reverseSortDirection = "DESC"
|
|
}
|
|
|
|
parentQuery = us.getQueryBuilder().
|
|
Select(getUsersColumnsWithName("data", "LastStatusAt", "LastPostDate", "DaysActive", "TotalPosts", "ChannelCount")...).
|
|
FromSelect(query, "data").
|
|
OrderBy(filter.SortColumn+" "+reverseSortDirection, "Id")
|
|
}
|
|
|
|
userResults := []*model.UserReportQuery{}
|
|
err = us.GetReplica().SelectBuilder(&userResults, parentQuery)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "failed to get users for reporting")
|
|
}
|
|
|
|
return userResults, nil
|
|
}
|