ec1f89217a
Full Mattermost server source with integrated Community Enterprise features. Includes vendor directory for offline/air-gapped builds. Structure: - enterprise-impl/: Enterprise feature implementations - enterprise-community/: Init files that register implementations - enterprise/: Bridge imports (community_imports.go) - vendor/: All dependencies for offline builds Build (online): go build ./cmd/mattermost Build (offline/air-gapped): go build -mod=vendor ./cmd/mattermost 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
124 lines
2.9 KiB
Go
124 lines
2.9 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package sqlstore
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
"github.com/mattermost/mattermost/server/v8/einterfaces"
|
|
sq "github.com/mattermost/squirrel"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
type SqlDesktopTokensStore struct {
|
|
*SqlStore
|
|
metrics einterfaces.MetricsInterface
|
|
}
|
|
|
|
func newSqlDesktopTokensStore(sqlStore *SqlStore, metrics einterfaces.MetricsInterface) store.DesktopTokensStore {
|
|
return &SqlDesktopTokensStore{
|
|
SqlStore: sqlStore,
|
|
metrics: metrics,
|
|
}
|
|
}
|
|
|
|
func (s *SqlDesktopTokensStore) GetUserId(token string, minCreateAt int64) (*string, error) {
|
|
query := s.getQueryBuilder().
|
|
Select("UserId").
|
|
From("DesktopTokens").
|
|
Where(sq.And{
|
|
sq.Eq{"Token": token},
|
|
sq.GtOrEq{"CreateAt": minCreateAt},
|
|
})
|
|
|
|
dt := struct{ UserId string }{}
|
|
err := s.GetReplica().GetBuilder(&dt, query)
|
|
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, store.NewErrNotFound("DesktopTokens", token)
|
|
}
|
|
return nil, errors.Wrapf(err, "No token for %s", token)
|
|
}
|
|
|
|
return &dt.UserId, nil
|
|
}
|
|
|
|
func (s *SqlDesktopTokensStore) Insert(token string, createAt int64, userId string) error {
|
|
builder := s.getQueryBuilder().
|
|
Insert("DesktopTokens").
|
|
Columns("Token", "CreateAt", "UserId").
|
|
Values(token, createAt, userId)
|
|
|
|
query, args, err := builder.ToSql()
|
|
|
|
if err != nil {
|
|
return errors.Wrap(err, "insert_desktoptokens_tosql")
|
|
}
|
|
|
|
if _, err = s.GetMaster().Exec(query, args...); err != nil {
|
|
return errors.Wrap(err, "failed to insert token row")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *SqlDesktopTokensStore) Delete(token string) error {
|
|
builder := s.getQueryBuilder().
|
|
Delete("DesktopTokens").
|
|
Where(sq.Eq{
|
|
"Token": token,
|
|
})
|
|
|
|
query, args, err := builder.ToSql()
|
|
|
|
if err != nil {
|
|
return errors.Wrap(err, "delete_desktoptokens_tosql")
|
|
}
|
|
|
|
if _, err = s.GetMaster().Exec(query, args...); err != nil {
|
|
return errors.Wrap(err, "failed to delete token row")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *SqlDesktopTokensStore) DeleteByUserId(userId string) error {
|
|
builder := s.getQueryBuilder().
|
|
Delete("DesktopTokens").
|
|
Where(sq.Eq{
|
|
"UserId": userId,
|
|
})
|
|
|
|
query, args, err := builder.ToSql()
|
|
|
|
if err != nil {
|
|
return errors.Wrap(err, "delete_by_userid_desktoptokens_tosql")
|
|
}
|
|
|
|
if _, err = s.GetMaster().Exec(query, args...); err != nil {
|
|
return errors.Wrap(err, "failed to delete token row")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *SqlDesktopTokensStore) DeleteOlderThan(minCreateAt int64) error {
|
|
builder := s.getQueryBuilder().
|
|
Delete("DesktopTokens").
|
|
Where(sq.Lt{
|
|
"CreateAt": minCreateAt,
|
|
})
|
|
|
|
query, args, err := builder.ToSql()
|
|
|
|
if err != nil {
|
|
return errors.Wrap(err, "delete_old_desktoptokens_tosql")
|
|
}
|
|
|
|
if _, err = s.GetMaster().Exec(query, args...); err != nil {
|
|
return errors.Wrap(err, "failed to delete token row")
|
|
}
|
|
return nil
|
|
}
|