package sqlstore import ( "database/sql" "encoding/json" "fmt" "log" "time" "github.com/mattermost/focalboard/server/model" sq "github.com/Masterminds/squirrel" ) type UserNotFoundError struct { id string } func (unf UserNotFoundError) Error() string { return fmt.Sprintf("user not found (%s)", unf.id) } func (s *SQLStore) GetRegisteredUserCount() (int, error) { query := s.getQueryBuilder(). Select("count(*)"). From(s.tablePrefix + "users"). Where(sq.Eq{"delete_at": 0}) row := query.QueryRow() var count int err := row.Scan(&count) if err != nil { return 0, err } return count, nil } func (s *SQLStore) getUserByCondition(condition sq.Eq) (*model.User, error) { users, err := s.getUsersByCondition(condition) if err != nil { return nil, err } if len(users) == 0 { return nil, nil } return users[0], nil } func (s *SQLStore) getUsersByCondition(condition sq.Eq) ([]*model.User, error) { query := s.getQueryBuilder(). Select( "id", "username", "email", "password", "mfa_secret", "auth_service", "auth_data", "props", "create_at", "update_at", "delete_at", ). From(s.tablePrefix + "users"). Where(sq.Eq{"delete_at": 0}). Where(condition) rows, err := query.Query() if err != nil { log.Printf("getUsersByCondition ERROR: %v", err) return nil, err } defer s.CloseRows(rows) users, err := s.usersFromRows(rows) if err != nil { return nil, err } if len(users) == 0 { return nil, sql.ErrNoRows } return users, nil } func (s *SQLStore) GetUserByID(userID string) (*model.User, error) { return s.getUserByCondition(sq.Eq{"id": userID}) } func (s *SQLStore) GetUserByEmail(email string) (*model.User, error) { return s.getUserByCondition(sq.Eq{"email": email}) } func (s *SQLStore) GetUserByUsername(username string) (*model.User, error) { return s.getUserByCondition(sq.Eq{"username": username}) } func (s *SQLStore) CreateUser(user *model.User) error { now := time.Now().Unix() propsBytes, err := json.Marshal(user.Props) if err != nil { return err } query := s.getQueryBuilder().Insert(s.tablePrefix+"users"). Columns("id", "username", "email", "password", "mfa_secret", "auth_service", "auth_data", "props", "create_at", "update_at", "delete_at"). Values(user.ID, user.Username, user.Email, user.Password, user.MfaSecret, user.AuthService, user.AuthData, propsBytes, now, now, 0) _, err = query.Exec() return err } func (s *SQLStore) UpdateUser(user *model.User) error { now := time.Now().Unix() propsBytes, err := json.Marshal(user.Props) if err != nil { return err } query := s.getQueryBuilder().Update(s.tablePrefix+"users"). Set("username", user.Username). Set("email", user.Email). Set("props", propsBytes). Set("update_at", now). Where(sq.Eq{"id": user.ID}) result, err := query.Exec() if err != nil { return err } rowCount, err := result.RowsAffected() if err != nil { return err } if rowCount < 1 { return UserNotFoundError{user.ID} } return nil } func (s *SQLStore) UpdateUserPassword(username, password string) error { now := time.Now().Unix() query := s.getQueryBuilder().Update(s.tablePrefix+"users"). Set("password", password). Set("update_at", now). Where(sq.Eq{"username": username}) result, err := query.Exec() if err != nil { return err } rowCount, err := result.RowsAffected() if err != nil { return err } if rowCount < 1 { return UserNotFoundError{username} } return nil } func (s *SQLStore) UpdateUserPasswordByID(userID, password string) error { now := time.Now().Unix() query := s.getQueryBuilder().Update(s.tablePrefix+"users"). Set("password", password). Set("update_at", now). Where(sq.Eq{"id": userID}) result, err := query.Exec() if err != nil { return err } rowCount, err := result.RowsAffected() if err != nil { return err } if rowCount < 1 { return UserNotFoundError{userID} } return nil } func (s *SQLStore) GetUsersByWorkspace(workspaceID string) ([]*model.User, error) { return s.getUsersByCondition(nil) } func (s *SQLStore) 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, &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 = append(users, &user) } return users, nil }