3 files changed, 102 insertions, 0 deletions

diff --git a/backend/controllers/homeController.go b/backend/controllers/homeController.go
index 635038b..42dfaa7 100644
--- a/backend/controllers/homeController.go
+++ b/backend/controllers/homeController.go
@@ -75,3 +75,93 @@ func Login(c *gin.Context) {
                 return
         }
 }
+
+func Rankings(c *gin.Context) {
+        rows, err := database.DB.Query(`SELECT steam_id, username FROM users;`)
+        if err != nil {
+                c.JSON(http.StatusBadRequest, models.ErrorResponse(err.Error()))
+                return
+        }
+        var spRankings []models.UserRanking
+        var mpRankings []models.UserRanking
+        for rows.Next() {
+                var userID, username string
+                err := rows.Scan(&userID, &username)
+                if err != nil {
+                        c.JSON(http.StatusBadRequest, models.ErrorResponse(err.Error()))
+                        return
+                }
+                // Getting all sp records for each user
+                var uniqueSingleUserRecords, totalSingleMaps int
+                sql := `SELECT COUNT(DISTINCT map_id), (SELECT COUNT(map_name) FROM maps 
+                WHERE is_coop = FALSE AND is_disabled = false) FROM records_sp WHERE user_id = $1;`
+                err = database.DB.QueryRow(sql, userID).Scan(&uniqueSingleUserRecords, &totalSingleMaps)
+                if err != nil {
+                        c.JSON(http.StatusBadRequest, models.ErrorResponse(err.Error()))
+                        return
+                }
+                // Has all singleplayer records
+                if uniqueSingleUserRecords == totalSingleMaps {
+                        var ranking models.UserRanking
+                        ranking.UserID = userID
+                        ranking.Username = username
+                        sql := `SELECT DISTINCT map_id, score_count FROM records_sp WHERE user_id = $1 ORDER BY map_id, score_count;`
+                        rows, err := database.DB.Query(sql, userID)
+                        if err != nil {
+                                c.JSON(http.StatusBadRequest, models.ErrorResponse(err.Error()))
+                                return
+                        }
+                        totalScore := 0
+                        var maps []int
+                        for rows.Next() {
+                                var mapID, scoreCount int
+                                rows.Scan(&mapID, &scoreCount)
+                                if len(maps) != 0 && maps[len(maps)-1] == mapID {
+                                        continue
+                                }
+                                totalScore += scoreCount
+                                maps = append(maps, mapID)
+                        }
+                        ranking.TotalScore = totalScore
+                        spRankings = append(spRankings, ranking)
+                }
+                // Getting all mp records for each user
+                var uniqueMultiUserRecords, totalMultiMaps int
+                sql = `SELECT COUNT(DISTINCT map_id), (SELECT COUNT(map_name) FROM maps 
+                WHERE is_coop = TRUE AND is_disabled = false) FROM records_mp WHERE host_id = $1 OR partner_id = $2;`
+                err = database.DB.QueryRow(sql, userID, userID).Scan(&uniqueMultiUserRecords, &totalMultiMaps)
+                if err != nil {
+                        c.JSON(http.StatusBadRequest, models.ErrorResponse(err.Error()))
+                        return
+                }
+                // Has all singleplayer records
+                if uniqueMultiUserRecords == totalMultiMaps {
+                        var ranking models.UserRanking
+                        ranking.UserID = userID
+                        ranking.Username = username
+                        sql := `SELECT DISTINCT map_id, score_count FROM records_mp WHERE host_id = $1 OR partner_id = $2 ORDER BY map_id, score_count;`
+                        rows, err := database.DB.Query(sql, userID, userID)
+                        if err != nil {
+                                c.JSON(http.StatusBadRequest, models.ErrorResponse(err.Error()))
+                                return
+                        }
+                        totalScore := 0
+                        var maps []int
+                        for rows.Next() {
+                                var mapID, scoreCount int
+                                rows.Scan(&mapID, &scoreCount)
+                                if len(maps) != 0 && maps[len(maps)-1] == mapID {
+                                        continue
+                                }
+                                totalScore += scoreCount
+                                maps = append(maps, mapID)
+                        }
+                        ranking.TotalScore = totalScore
+                        mpRankings = append(mpRankings, ranking)
+                }
+        }
+        c.JSON(http.StatusOK, models.RankingsResponse{
+                RankingsSP: spRankings,
+                RankingsMP: mpRankings,
+        })
+}
diff --git a/backend/models/models.go b/backend/models/models.go
index 0c380ff..5ce3ebf 100644
--- a/backend/models/models.go
+++ b/backend/models/models.go
@@ -61,6 +61,17 @@ type RecordRequest struct {
         IsPartnerOrange bool   `json:"is_partner_orange" form:"is_partner_orange" binding:"required"`
 }
 
+type UserRanking struct {
+        UserID     string `json:"user_id"`
+        Username   string `json:"username"`
+        TotalScore int    `json:"total_score"`
+}
+
+type RankingsResponse struct {
+        RankingsSP []UserRanking `json:"rankings_sp"`
+        RankingsMP []UserRanking `json:"rankings_mp"`
+}
+
 type ProfileResponse struct {
         Profile     bool            `json:"profile"`
         SteamID     string          `json:"steam_id"`
diff --git a/backend/routes/routes.go b/backend/routes/routes.go
index b872b77..b42167a 100644
--- a/backend/routes/routes.go
+++ b/backend/routes/routes.go
@@ -21,5 +21,6 @@ func InitRoutes(router *gin.Engine) {
                 v1.GET("/demo", controllers.DownloadDemoWithID)
                 v1.GET("/maps/:id", middleware.CheckAuth, controllers.FetchMap)
                 v1.POST("/maps/:id/record", middleware.CheckAuth, controllers.CreateRecordWithDemo)
+                v1.GET("/rankings", middleware.CheckAuth, controllers.Rankings)
         }
 }