Add endpoints for stats

api/game.go

@@ -8,25 +8,36 @@ import (
 	"time"
 
 	"github.com/Flashfyre/pokerogue-server/db"
+	"github.com/Flashfyre/pokerogue-server/defs"
 	"github.com/go-co-op/gocron"
 )
 
 var (
-	playerCountScheduler = gocron.NewScheduler(time.UTC)
-	playerCount          int
+	statScheduler       = gocron.NewScheduler(time.UTC)
+	playerCount         int
+	battleCount         int
+	classicSessionCount int
 )
 
-func SchedulePlayerCountRefresh() {
-	playerCountScheduler.Every(10).Second().Do(updatePlayerCount)
-	playerCountScheduler.StartAsync()
+func ScheduleStatRefresh() {
+	statScheduler.Every(10).Second().Do(updateStats)
+	statScheduler.StartAsync()
 }
 
-func updatePlayerCount() {
+func updateStats() {
 	var err error
 	playerCount, err = db.FetchPlayerCount()
 	if err != nil {
 		log.Print(err)
 	}
+	battleCount, err = db.FetchBattleCount()
+	if err != nil {
+		log.Print(err)
+	}
+	classicSessionCount, err = db.FetchClassicSessionCount()
+	if err != nil {
+		log.Print(err)
+	}
 }
 
 // /game/playercount - get player count
@@ -39,3 +50,29 @@ func (s *Server) handlePlayerCountGet(w http.ResponseWriter) {
 
 	w.Write(response)
 }
+
+// /game/titlestats - get title stats
+func (s *Server) handleTitleStatsGet(w http.ResponseWriter) {
+	titleStats := &defs.TitleStats{
+		PlayerCount: playerCount,
+		BattleCount: battleCount,
+	}
+	response, err := json.Marshal(titleStats)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
+}
+
+// /game/classicsessioncount - get classic session count
+func (s *Server) handleClassicSessionCountGet(w http.ResponseWriter) {
+	response, err := json.Marshal(classicSessionCount)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
+}

api/generic.go

@@ -36,6 +36,10 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
 	case "/game/playercount":
 		s.handlePlayerCountGet(w)
+	case "/game/titlestats":
+		s.handleTitleStatsGet(w)
+	case "/game/classicsessioncount":
+		s.handleClassicSessionCountGet(w)
 
 	case "/savedata/get":
 		s.handleSavedataGet(w, r)