diff options
Diffstat (limited to 'backend/middleware')
| -rw-r--r-- | backend/middleware/auth.go | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/backend/middleware/auth.go b/backend/middleware/auth.go index 14a0b78..86e79d9 100644 --- a/backend/middleware/auth.go +++ b/backend/middleware/auth.go | |||
| @@ -36,13 +36,21 @@ func CheckAuth(c *gin.Context) { | |||
| 36 | } | 36 | } |
| 37 | // Get user from DB | 37 | // Get user from DB |
| 38 | var user models.User | 38 | var user models.User |
| 39 | database.DB.QueryRow(`SELECT * FROM users WHERE steam_id = $1`, claims["sub"]).Scan( | 39 | database.DB.QueryRow(`SELECT u.steam_id, u.user_name, u.avatar_link, u.country_code, u.created_at, u.updated_at, array_agg(t.) FROM users u WHERE steam_id = $1`, claims["sub"]).Scan( |
| 40 | &user.SteamID, &user.UserName, &user.AvatarLink, | 40 | &user.SteamID, &user.UserName, &user.AvatarLink, |
| 41 | &user.CountryCode, &user.CreatedAt, &user.UpdatedAt) | 41 | &user.CountryCode, &user.CreatedAt, &user.UpdatedAt) |
| 42 | if user.SteamID == "" { | 42 | if user.SteamID == "" { |
| 43 | c.Next() | 43 | c.Next() |
| 44 | return | 44 | return |
| 45 | } | 45 | } |
| 46 | // Get user titles from DB | ||
| 47 | user.Titles = []string{} | ||
| 48 | rows, _ := database.DB.Query(`SELECT t.title_name FROM titles t WHERE t.user_id = $1`, user.SteamID) | ||
| 49 | for rows.Next() { | ||
| 50 | var title string | ||
| 51 | rows.Scan(&title) | ||
| 52 | user.Titles = append(user.Titles, title) | ||
| 53 | } | ||
| 46 | c.Set("user", user) | 54 | c.Set("user", user) |
| 47 | c.Next() | 55 | c.Next() |
| 48 | } else { | 56 | } else { |