From 53ccdc592173f829b33e9c150e651c7cfa662f17 Mon Sep 17 00:00:00 2001
From: Flashfyre <flashfireex@gmail.com>
Date: Sat, 23 Mar 2024 16:26:00 -0400
Subject: [PATCH] Add weekly rankings and ranking pagination support

---
 api/daily.go   | 43 ++++++++++++++++++++++++++++++++++++++++++-
 api/generic.go |  2 ++
 db/daily.go    | 33 +++++++++++++++++++++++++++++++--
 3 files changed, 75 insertions(+), 3 deletions(-)

diff --git a/api/daily.go b/api/daily.go
index 9a46af4..290c0f0 100644
--- a/api/daily.go
+++ b/api/daily.go
@@ -81,8 +81,19 @@ func (s *Server) HandleSeed(w http.ResponseWriter, r *http.Request) {
 
 func (s *Server) HandleRankings(w http.ResponseWriter, r *http.Request) {
 	var err error
+	var category int
 	var page int
 
+	if r.URL.Query().Has("category") {
+		category, err = strconv.Atoi(r.URL.Query().Get("category"))
+		if err != nil {
+			http.Error(w, fmt.Sprintf("failed to convert category: %s", err), http.StatusBadRequest)
+			return
+		}
+	} else {
+		category = 0
+	}
+
 	if r.URL.Query().Has("page") {
 		page, err = strconv.Atoi(r.URL.Query().Get("page"))
 		if err != nil {
@@ -93,7 +104,7 @@ func (s *Server) HandleRankings(w http.ResponseWriter, r *http.Request) {
 		page = 1
 	}
 
-	rankings, err := db.FetchRankings(page)
+	rankings, err := db.FetchRankings(category, page)
 	if err != nil {
 		log.Print("failed to retrieve rankings")
 	}
@@ -106,3 +117,33 @@ func (s *Server) HandleRankings(w http.ResponseWriter, r *http.Request) {
 
 	w.Write(response)
 }
+
+// /daily/rankingpagecount - fetch daily ranking page count
+
+func (s *Server) HandleRankingPageCount(w http.ResponseWriter, r *http.Request) {
+	var err error
+	var category int
+
+	if r.URL.Query().Has("category") {
+		category, err = strconv.Atoi(r.URL.Query().Get("category"))
+		if err != nil {
+			http.Error(w, fmt.Sprintf("failed to convert category: %s", err), http.StatusBadRequest)
+			return
+		}
+	} else {
+		category = 0
+	}
+
+	pageCount, err := db.FetchRankingPageCount(category)
+	if err != nil {
+		log.Print("failed to retrieve ranking page count")
+	}
+
+	response, err := json.Marshal(pageCount)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
+}
diff --git a/api/generic.go b/api/generic.go
index bf93179..9aef79d 100644
--- a/api/generic.go
+++ b/api/generic.go
@@ -47,6 +47,8 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		s.HandleSeed(w, r)
 	case "/daily/rankings":
 		s.HandleRankings(w, r)
+	case "/daily/rankingpagecount":
+		s.HandleRankingPageCount(w, r)
 	}
 }
 
diff --git a/db/daily.go b/db/daily.go
index c4d0f5e..aa24636 100644
--- a/db/daily.go
+++ b/db/daily.go
@@ -1,6 +1,8 @@
 package db
 
 import (
+	"math"
+
 	"github.com/Flashfyre/pokerogue-server/defs"
 )
 
@@ -22,12 +24,20 @@ func AddOrUpdateAccountDailyRun(uuid []byte, score int, wave int) error {
 	return nil
 }
 
-func FetchRankings(page int) ([]defs.DailyRanking, error) {
+func FetchRankings(category int, page int) ([]defs.DailyRanking, error) {
 	var rankings []defs.DailyRanking
 
 	offset := (page - 1) * 10
 
-	results, err := handle.Query("SELECT RANK() OVER (ORDER BY adr.score DESC, adr.timestamp), a.username, adr.score, adr.wave FROM accountDailyRuns adr JOIN dailyRuns dr ON dr.date = adr.date JOIN accounts a ON adr.uuid = a.uuid WHERE dr.date = UTC_DATE() LIMIT 10 OFFSET ?", offset)
+	var query string
+	switch category {
+	case 0:
+		query = "SELECT RANK() OVER (ORDER BY adr.score DESC, adr.timestamp), a.username, adr.score, adr.wave FROM accountDailyRuns adr JOIN dailyRuns dr ON dr.date = adr.date JOIN accounts a ON adr.uuid = a.uuid WHERE dr.date = UTC_DATE() LIMIT 10 OFFSET ?"
+	case 1:
+		query = "SELECT RANK() OVER (ORDER BY SUM(adr.score) DESC, adr.timestamp), a.username, SUM(adr.score), 0 FROM accountDailyRuns adr JOIN dailyRuns dr ON dr.date = adr.date JOIN accounts a ON adr.uuid = a.uuid WHERE dr.date >= DATE_SUB(DATE(UTC_TIMESTAMP()), INTERVAL DAYOFWEEK(UTC_TIMESTAMP()) - 1 DAY) GROUP BY a.username ORDER BY 1 LIMIT 10 OFFSET ?"
+	}
+
+	results, err := handle.Query(query, offset)
 	if err != nil {
 		return rankings, err
 	}
@@ -46,3 +56,22 @@ func FetchRankings(page int) ([]defs.DailyRanking, error) {
 
 	return rankings, nil
 }
+
+func FetchRankingPageCount(category int) (int, error) {
+	var recordCount int
+
+	var query string
+	switch category {
+	case 0:
+		query = "SELECT COUNT(a.username) FROM accountDailyRuns adr JOIN dailyRuns dr ON dr.date = adr.date JOIN accounts a ON adr.uuid = a.uuid WHERE dr.date = UTC_DATE()"
+	case 1:
+		query = "SELECT COUNT(a.username) FROM accountDailyRuns adr JOIN dailyRuns dr ON dr.date = adr.date JOIN accounts a ON adr.uuid = a.uuid WHERE dr.date >= DATE_SUB(DATE(UTC_TIMESTAMP()), INTERVAL DAYOFWEEK(UTC_TIMESTAMP()) - 1 DAY)"
+	}
+
+	err := handle.QueryRow(query).Scan(&recordCount)
+	if err != nil {
+		return 0, err
+	}
+
+	return int(math.Ceil(float64(recordCount) / 10)), nil
+}