From 75e06b36b06b472aa69ac6c143c99c24e317801a Mon Sep 17 00:00:00 2001
From: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
Date: Thu, 28 Apr 2022 16:24:26 +0000
Subject: [PATCH] Fix GetUser

---
 database/sql/sql.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/database/sql/sql.go b/database/sql/sql.go
index 8dbbbc79..0ea44dfa 100644
--- a/database/sql/sql.go
+++ b/database/sql/sql.go
@@ -941,6 +941,18 @@ func (s *sqlDatabase) getUserByUsernameOrEmail(user string) (User, error) {
 	return dbUser, nil
 }
 
+func (s *sqlDatabase) getUserByID(userID string) (User, error) {
+	var dbUser User
+	q := s.conn.Model(&User{}).Where("id = ?", userID).First(&dbUser)
+	if q.Error != nil {
+		if errors.Is(q.Error, gorm.ErrRecordNotFound) {
+			return User{}, runnerErrors.ErrNotFound
+		}
+		return User{}, errors.Wrap(q.Error, "fetching user")
+	}
+	return dbUser, nil
+}
+
 func (s *sqlDatabase) CreateUser(ctx context.Context, user params.NewUserParams) (params.User, error) {
 	if user.Username == "" || user.Email == "" {
 		return params.User{}, runnerErrors.NewBadRequestError("missing username or email")
@@ -978,7 +990,7 @@ func (s *sqlDatabase) HasAdminUser(ctx context.Context) bool {
 }
 
 func (s *sqlDatabase) GetUser(ctx context.Context, user string) (params.User, error) {
-	dbUser, err := s.getUserByUsernameOrEmail(user)
+	dbUser, err := s.getUserByID(user)
 	if err != nil {
 		return params.User{}, errors.Wrap(err, "fetching user")
 	}