package mattermostauthlayer import ( "database/sql" "encoding/json" "fmt" "strings" "github.com/pkg/errors" sq "github.com/Masterminds/squirrel" "github.com/mattermost/focalboard/server/model" "github.com/mattermost/focalboard/server/services/store" "github.com/mattermost/focalboard/server/utils" "github.com/mattermost/mattermost-server/v6/shared/mlog" ) const ( sqliteDBType = "sqlite3" postgresDBType = "postgres" mysqlDBType = "mysql" directChannelType = "D" ) var ( errUnsupportedDatabaseError = errors.New("method is unsupported on current database. Supported databases are - MySQL and PostgreSQL") ) type NotSupportedError struct { msg string } func (pe NotSupportedError) Error() string { return pe.msg } // Store represents the abstraction of the data storage. type MattermostAuthLayer struct { store.Store dbType string mmDB *sql.DB logger *mlog.Logger } // New creates a new SQL implementation of the store. func New(dbType string, db *sql.DB, store store.Store, logger *mlog.Logger) (*MattermostAuthLayer, error) { layer := &MattermostAuthLayer{ Store: store, dbType: dbType, mmDB: db, logger: logger, } return layer, nil } // Shutdown close the connection with the store. func (s *MattermostAuthLayer) Shutdown() error { return s.Store.Shutdown() } func (s *MattermostAuthLayer) GetRegisteredUserCount() (int, error) { query := s.getQueryBuilder(). Select("count(*)"). From("Users"). Where(sq.Eq{"deleteAt": 0}) row := query.QueryRow() var count int err := row.Scan(&count) if err != nil { return 0, err } return count, nil } func (s *MattermostAuthLayer) getUserByCondition(condition sq.Eq) (*model.User, error) { users, err := s.getUsersByCondition(condition) if err != nil { return nil, err } var user *model.User for _, u := range users { user = u break } return user, nil } func (s *MattermostAuthLayer) getUsersByCondition(condition sq.Eq) (map[string]*model.User, error) { query := s.getQueryBuilder(). Select("id", "username", "email", "password", "MFASecret as mfa_secret", "AuthService as auth_service", "COALESCE(AuthData, '') as auth_data", "props", "CreateAt as create_at", "UpdateAt as update_at", "DeleteAt as delete_at"). From("Users"). Where(sq.Eq{"deleteAt": 0}). Where(condition) row, err := query.Query() if err != nil { return nil, err } users := map[string]*model.User{} for row.Next() { user := model.User{} var propsBytes []byte err := row.Scan(&user.ID, &user.Username, &user.Email, &user.Password, &user.MfaSecret, &user.AuthService, &user.AuthData, &propsBytes, &user.CreateAt, &user.UpdateAt, &user.DeleteAt) if err != nil { return nil, err } err = json.Unmarshal(propsBytes, &user.Props) if err != nil { return nil, err } users[user.ID] = &user } return users, nil } func (s *MattermostAuthLayer) GetUserByID(userID string) (*model.User, error) { return s.getUserByCondition(sq.Eq{"id": userID}) } func (s *MattermostAuthLayer) GetUserByEmail(email string) (*model.User, error) { return s.getUserByCondition(sq.Eq{"email": email}) } func (s *MattermostAuthLayer) GetUserByUsername(username string) (*model.User, error) { return s.getUserByCondition(sq.Eq{"username": username}) } func (s *MattermostAuthLayer) CreateUser(user *model.User) error { return NotSupportedError{"no user creation allowed from focalboard, create it using mattermost"} } func (s *MattermostAuthLayer) UpdateUser(user *model.User) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } func (s *MattermostAuthLayer) UpdateUserPassword(username, password string) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } func (s *MattermostAuthLayer) UpdateUserPasswordByID(userID, password string) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } // GetActiveUserCount returns the number of users with active sessions within N seconds ago. func (s *MattermostAuthLayer) GetActiveUserCount(updatedSecondsAgo int64) (int, error) { query := s.getQueryBuilder(). Select("count(distinct userId)"). From("Sessions"). Where(sq.Gt{"LastActivityAt": utils.GetMillis() - utils.SecondsToMillis(updatedSecondsAgo)}) row := query.QueryRow() var count int err := row.Scan(&count) if err != nil { return 0, err } return count, nil } func (s *MattermostAuthLayer) GetSession(token string, expireTime int64) (*model.Session, error) { return nil, NotSupportedError{"sessions not used when using mattermost"} } func (s *MattermostAuthLayer) CreateSession(session *model.Session) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } func (s *MattermostAuthLayer) RefreshSession(session *model.Session) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } func (s *MattermostAuthLayer) UpdateSession(session *model.Session) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } func (s *MattermostAuthLayer) DeleteSession(sessionID string) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } func (s *MattermostAuthLayer) CleanUpSessions(expireTime int64) error { return NotSupportedError{"no update allowed from focalboard, update it using mattermost"} } func (s *MattermostAuthLayer) GetWorkspace(id string) (*model.Workspace, error) { if id == "0" { workspace := model.Workspace{ ID: id, Title: "", } return &workspace, nil } query := s.getQueryBuilder(). Select("DisplayName, Type"). From("Channels"). Where(sq.Eq{"ID": id}) row := query.QueryRow() var displayName string var channelType string err := row.Scan(&displayName, &channelType) if err != nil { return nil, err } if channelType != "D" && channelType != "G" { return &model.Workspace{ID: id, Title: displayName}, nil } query = s.getQueryBuilder(). Select("Username"). From("ChannelMembers"). Join("Users ON Users.ID=ChannelMembers.UserID"). Where(sq.Eq{"ChannelID": id}) var sb strings.Builder rows, err := query.Query() if err != nil { return nil, err } defer s.CloseRows(rows) first := true for rows.Next() { if first { first = false } else { sb.WriteString(", ") } var name string if err := rows.Scan(&name); err != nil { return nil, err } sb.WriteString(name) } return &model.Workspace{ID: id, Title: sb.String()}, nil } func (s *MattermostAuthLayer) HasWorkspaceAccess(userID string, workspaceID string) (bool, error) { query := s.getQueryBuilder(). Select("count(*)"). From("ChannelMembers"). Where(sq.Eq{"ChannelID": workspaceID}). Where(sq.Eq{"UserID": userID}) row := query.QueryRow() var count int err := row.Scan(&count) if err != nil { return false, err } return count > 0, nil } func (s *MattermostAuthLayer) getQueryBuilder() sq.StatementBuilderType { builder := sq.StatementBuilder if s.dbType == postgresDBType || s.dbType == sqliteDBType { builder = builder.PlaceholderFormat(sq.Dollar) } return builder.RunWith(s.mmDB) } func (s *MattermostAuthLayer) GetUsersByWorkspace(workspaceID string) ([]*model.User, error) { query := s.getQueryBuilder(). Select("id", "username", "props", "Users.CreateAt as create_at", "Users.UpdateAt as update_at", "Users.DeleteAt as delete_at", "b.UserId IS NOT NULL AS is_bot"). From("Users"). Join("ChannelMembers ON ChannelMembers.UserID = Users.ID"). LeftJoin("Bots b ON ( b.UserId = Users.ID )"). Where(sq.Eq{"Users.deleteAt": 0}). Where(sq.Eq{"ChannelMembers.ChannelId": workspaceID}) rows, err := query.Query() if err != nil { return nil, err } defer s.CloseRows(rows) users, err := s.usersFromRows(rows) if err != nil { return nil, err } return users, nil } func (s *MattermostAuthLayer) usersFromRows(rows *sql.Rows) ([]*model.User, error) { users := []*model.User{} for rows.Next() { var user model.User var propsBytes []byte err := rows.Scan( &user.ID, &user.Username, &propsBytes, &user.CreateAt, &user.UpdateAt, &user.DeleteAt, &user.IsBot, ) if err != nil { return nil, err } err = json.Unmarshal(propsBytes, &user.Props) if err != nil { return nil, err } users = append(users, &user) } return users, nil } func (s *MattermostAuthLayer) CloseRows(rows *sql.Rows) { if err := rows.Close(); err != nil { s.logger.Error("error closing MattermostAuthLayer row set", mlog.Err(err)) } } func (s *MattermostAuthLayer) GetUserWorkspaces(userID string) ([]model.UserWorkspace, error) { var query sq.SelectBuilder var nonTemplateFilter string switch s.dbType { case mysqlDBType: nonTemplateFilter = "focalboard_blocks.fields LIKE '%\"isTemplate\":false%'" case postgresDBType: nonTemplateFilter = "focalboard_blocks.fields ->> 'isTemplate' = 'false'" default: return nil, fmt.Errorf("GetUserWorkspaces - %w", errUnsupportedDatabaseError) } query = s.getQueryBuilder(). Select("Channels.ID", "Channels.DisplayName", "COUNT(focalboard_blocks.id), Channels.Type, Channels.Name"). From("ChannelMembers"). // select channels without a corresponding workspace LeftJoin( "focalboard_blocks ON focalboard_blocks.workspace_id = ChannelMembers.ChannelId AND "+ "focalboard_blocks.type = 'board' AND "+ nonTemplateFilter, ). Join("Channels ON ChannelMembers.ChannelId = Channels.Id"). Where(sq.Eq{"ChannelMembers.UserId": userID}). GroupBy("Channels.Id", "Channels.DisplayName") rows, err := query.Query() if err != nil { s.logger.Error("ERROR GetUserWorkspaces", mlog.Err(err)) return nil, err } defer s.CloseRows(rows) return s.userWorkspacesFromRows(rows) } type UserWorkspaceRawModel struct { ID string `json:"id"` Title string `json:"title"` BoardCount int `json:"boardCount"` Type string `json:"type"` Name string `json:"name"` } func (s *MattermostAuthLayer) userWorkspacesFromRows(rows *sql.Rows) ([]model.UserWorkspace, error) { rawUserWorkspaces := []UserWorkspaceRawModel{} usersToFetch := []string{} for rows.Next() { var rawUserWorkspace UserWorkspaceRawModel err := rows.Scan( &rawUserWorkspace.ID, &rawUserWorkspace.Title, &rawUserWorkspace.BoardCount, &rawUserWorkspace.Type, &rawUserWorkspace.Name, ) if err != nil { s.logger.Error("ERROR userWorkspacesFromRows", mlog.Err(err)) return nil, err } if rawUserWorkspace.Type == directChannelType { userIDs := strings.Split(rawUserWorkspace.Name, "__") usersToFetch = append(usersToFetch, userIDs...) } rawUserWorkspaces = append(rawUserWorkspaces, rawUserWorkspace) } var users map[string]*model.User if len(usersToFetch) > 0 { var err error users, err = s.getUsersByCondition(sq.Eq{"id": usersToFetch}) if err != nil { return nil, err } } userWorkspaces := []model.UserWorkspace{} for i := range rawUserWorkspaces { if rawUserWorkspaces[i].Type == directChannelType { userIDs := strings.Split(rawUserWorkspaces[i].Name, "__") names := []string{} for _, userID := range userIDs { user, exists := users[userID] username := userID if exists { username = user.Username } names = append(names, username) } rawUserWorkspaces[i].Title = strings.Join(names, ", ") } userWorkspace := model.UserWorkspace{ ID: rawUserWorkspaces[i].ID, Title: rawUserWorkspaces[i].Title, BoardCount: rawUserWorkspaces[i].BoardCount, } userWorkspaces = append(userWorkspaces, userWorkspace) } return userWorkspaces, nil }