1 files changed, 72 insertions, 2 deletions

diff --git a/backend/controllers/controllers.go b/backend/controllers/controllers.go
index 6a38e8c..712b33c 100644
--- a/backend/controllers/controllers.go
+++ b/backend/controllers/controllers.go
@@ -4,11 +4,13 @@ import (
         "log"
         "net/http"
         "os"
+        "regexp"
         "time"
 
         "github.com/gin-gonic/gin"
         "github.com/golang-jwt/jwt/v4"
         "github.com/pektezol/leastportals/backend/database"
+        "github.com/pektezol/leastportals/backend/models"
         "github.com/solovev/steam_go"
 )
 
@@ -46,7 +48,7 @@ func Login(c *gin.Context) {
                         }
                         // Insert new user to database
                         database.DB.Exec(`INSERT INTO users (steam_id, username, avatar_link, country_code, created_at, updated_at, user_type)
-                        VALUES ($1, $2, $3, $4, $5, $6, $7)`, steamID, user.PersonaName, user.Avatar, user.LocCountryCode, time.Now().UTC(), time.Now().UTC(), 0)
+                        VALUES ($1, $2, $3, $4, $5, $6, $7)`, steamID, user.PersonaName, user.AvatarFull, user.LocCountryCode, time.Now().UTC(), time.Now().UTC(), 0)
                 }
                 // Update updated_at
                 database.DB.Exec(`UPDATE users SET updated_at = $1 WHERE steam_id = $2`, time.Now().UTC(), steamID)
@@ -84,6 +86,74 @@ func Logout(c *gin.Context) {
                 c.JSON(http.StatusOK, gin.H{
                         "output": "logout success",
                 })
-                //c.Redirect(http.StatusPermanentRedirect, "/")
         }
 }
+
+func Profile(c *gin.Context) {
+        // Check if user exists
+        user, exists := c.Get("user")
+        if !exists {
+                c.JSON(http.StatusUnauthorized, gin.H{
+                        "code": http.StatusUnauthorized,
+                        "output": gin.H{
+                                "error": "User not logged in. Could be invalid token.",
+                        },
+                })
+        } else {
+                user := user.(models.User)
+                c.JSON(http.StatusOK, gin.H{
+                        "code": http.StatusOK,
+                        "output": gin.H{
+                                "username": user.Username,
+                                "avatar":   user.AvatarLink,
+                                "types":    user.TypeToString(),
+                        },
+                        "profile": true,
+                })
+        }
+}
+
+func User(c *gin.Context) {
+        id := c.Param("id")
+        // Check if id is all numbers and 17 length
+        match, _ := regexp.MatchString("^[0-9]{17}$", id)
+        if !match {
+                c.JSON(http.StatusNotFound, gin.H{
+                        "code": http.StatusNotFound,
+                        "output": gin.H{
+                                "error": "User not found.",
+                        },
+                })
+                return
+        }
+        // Check if user exists
+        var targetUser models.User
+        database.DB.QueryRow(`SELECT * FROM users WHERE steam_id = $1;`, id).Scan(
+                &targetUser.SteamID, &targetUser.Username, &targetUser.AvatarLink, &targetUser.CountryCode,
+                &targetUser.CreatedAt, &targetUser.UpdatedAt, &targetUser.UserType)
+        if targetUser.SteamID == "" {
+                // User does not exist
+                c.JSON(http.StatusNotFound, gin.H{
+                        "code": http.StatusNotFound,
+                        "output": gin.H{
+                                "error": "User not found.",
+                        },
+                })
+                return
+        }
+        // Target user exists
+        _, exists := c.Get("user")
+        if exists {
+                c.Redirect(http.StatusFound, "/api/v1/profile")
+                return
+        }
+        c.JSON(http.StatusOK, gin.H{
+                "code": http.StatusOK,
+                "output": gin.H{
+                        "username": targetUser.Username,
+                        "avatar":   targetUser.AvatarLink,
+                        "types":    targetUser.TypeToString(),
+                },
+                "profile": false,
+        })
+}