Add voucher compensation logic

api/account-helper.go

@@ -30,7 +30,7 @@ func getUsernameFromRequest(r *http.Request) (string, error) {
 	return username, nil
 }
 
-func getUUIDFromRequest(r *http.Request) ([]byte, error) {
+func getUuidFromRequest(r *http.Request) ([]byte, error) {
 	if r.Header.Get("Authorization") == "" {
 		return nil, fmt.Errorf("missing token")
 	}
@@ -44,7 +44,7 @@ func getUUIDFromRequest(r *http.Request) ([]byte, error) {
 		return nil, fmt.Errorf("invalid token length: got %d, expected 32", len(token))
 	}
 
-	uuid, err := db.FetchUUIDFromToken(token)
+	uuid, err := db.FetchUuidFromToken(token)
 	if err != nil {
 		return nil, fmt.Errorf("failed to validate token: %s", err)
 	}

api/account.go

@@ -5,7 +5,9 @@ import (
 	"crypto/rand"
 	"database/sql"
 	"encoding/base64"
+	"encoding/json"
 	"fmt"
+	"net/http"
 	"os"
 	"regexp"
 	"strconv"
@@ -16,12 +18,10 @@ import (
 )
 
 const (
-	UUIDSize      = 16
-	ArgonTime     = 1
-	ArgonMemory   = 256 * 1024
-	ArgonThreads  = 4
-	ArgonKeySize  = 32
-	ArgonSaltSize = 16
+	argonTime      = 1
+	argonMemory    = 256 * 1024
+	argonThreads   = 4
+	argonKeyLength = 32
 )
 
 var isValidUsername = regexp.MustCompile(`^\w{1,16}$`).MatchString
@@ -32,9 +32,21 @@ type AccountInfoResponse struct {
 }
 
 // /account/info - get account info
-func handleAccountInfo(username string, uuid []byte) (AccountInfoResponse, error) {
-	var latestSave time.Time
-	latestSaveID := -1
+func (s *Server) handleAccountInfo(w http.ResponseWriter, r *http.Request) {
+	username, err := getUsernameFromRequest(r)
+	if err != nil {
+		httpError(w, r, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	uuid, err := getUuidFromRequest(r) // lazy
+	if err != nil {
+		httpError(w, r, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	var latestSaveTime time.Time
+	latestSaveId := -1
 	for id := range sessionSlotCount {
 		fileName := "session"
 		if id != 0 {
@@ -46,101 +58,148 @@ func handleAccountInfo(username string, uuid []byte) (AccountInfoResponse, error
 			continue
 		}
 
-		if stat.ModTime().After(latestSave) {
-			latestSave = stat.ModTime()
-			latestSaveID = id
+		if stat.ModTime().After(latestSaveTime) {
+			latestSaveTime = stat.ModTime()
+			latestSaveId = id
 		}
 	}
 
-	return AccountInfoResponse{Username: username, LastSessionSlot: latestSaveID}, nil
+	response, err := json.Marshal(AccountInfoResponse{Username: username, LastSessionSlot: latestSaveId})
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
 }
 
 type AccountRegisterRequest GenericAuthRequest
 
 // /account/register - register account
-func handleAccountRegister(username, password string) error {
-	if !isValidUsername(username) {
-		return fmt.Errorf("invalid username")
-	}
-
-	if len(password) < 6 {
-		return fmt.Errorf("invalid password")
-	}
-
-	uuid := make([]byte, UUIDSize)
-	_, err := rand.Read(uuid)
+func (s *Server) handleAccountRegister(w http.ResponseWriter, r *http.Request) {
+	var request AccountRegisterRequest
+	err := json.NewDecoder(r.Body).Decode(&request)
 	if err != nil {
-		return fmt.Errorf("failed to generate uuid: %s", err)
+		httpError(w, r, fmt.Sprintf("failed to decode request body: %s", err), http.StatusBadRequest)
+		return
 	}
 
-	salt := make([]byte, ArgonSaltSize)
+	if !isValidUsername(request.Username) {
+		httpError(w, r, "invalid username", http.StatusBadRequest)
+		return
+	}
+
+	if len(request.Password) < 6 {
+		httpError(w, r, "invalid password", http.StatusBadRequest)
+		return
+	}
+
+	uuid := make([]byte, 16)
+	_, err = rand.Read(uuid)
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to generate uuid: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	salt := make([]byte, 16)
 	_, err = rand.Read(salt)
 	if err != nil {
-		return fmt.Errorf(fmt.Sprintf("failed to generate salt: %s", err))
+		httpError(w, r, fmt.Sprintf("failed to generate salt: %s", err), http.StatusInternalServerError)
+		return
 	}
 
-	err = db.AddAccountRecord(uuid, username, argon2.IDKey([]byte(password), salt, ArgonTime, ArgonMemory, ArgonThreads, ArgonKeySize), salt)
+	err = db.AddAccountRecord(uuid, request.Username, argon2.IDKey([]byte(request.Password), salt, argonTime, argonMemory, argonThreads, argonKeyLength), salt)
 	if err != nil {
-		return fmt.Errorf("failed to add account record: %s", err)
+		httpError(w, r, fmt.Sprintf("failed to add account record: %s", err), http.StatusInternalServerError)
+		return
 	}
 
-	return nil
+	w.WriteHeader(http.StatusOK)
 }
 
 type AccountLoginRequest GenericAuthRequest
 type AccountLoginResponse GenericAuthResponse
 
 // /account/login - log into account
-func handleAccountLogin(username, password string) (AccountLoginResponse, error) {
-	if !isValidUsername(username) {
-		return AccountLoginResponse{}, fmt.Errorf("invalid username")
+func (s *Server) handleAccountLogin(w http.ResponseWriter, r *http.Request) {
+	var request AccountLoginRequest
+	err := json.NewDecoder(r.Body).Decode(&request)
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to decode request body: %s", err), http.StatusBadRequest)
+		return
 	}
 
-	if len(password) < 6 {
-		return AccountLoginResponse{}, fmt.Errorf("invalid password")
+	if !isValidUsername(request.Username) {
+		httpError(w, r, "invalid username", http.StatusBadRequest)
+		return
 	}
 
-	key, salt, err := db.FetchAccountKeySaltFromUsername(username)
+	if len(request.Password) < 6 {
+		httpError(w, r, "invalid password", http.StatusBadRequest)
+		return
+	}
+
+	key, salt, err := db.FetchAccountKeySaltFromUsername(request.Username)
 	if err != nil {
 		if err == sql.ErrNoRows {
-			return AccountLoginResponse{}, fmt.Errorf("account doesn't exist")
+			httpError(w, r, "account doesn't exist", http.StatusBadRequest)
+			return
 		}
 
-		return AccountLoginResponse{}, err
+		httpError(w, r, err.Error(), http.StatusInternalServerError)
+		return
 	}
 
-	if !bytes.Equal(key, argon2.IDKey([]byte(password), salt, ArgonTime, ArgonMemory, ArgonThreads, ArgonKeySize)) {
-		return AccountLoginResponse{}, fmt.Errorf("password doesn't match")
+	if !bytes.Equal(key, argon2.IDKey([]byte(request.Password), salt, argonTime, argonMemory, argonThreads, argonKeyLength)) {
+		httpError(w, r, "password doesn't match", http.StatusBadRequest)
+		return
 	}
 
 	token := make([]byte, 32)
 	_, err = rand.Read(token)
 	if err != nil {
-		return AccountLoginResponse{}, fmt.Errorf("failed to generate token: %s", err)
+		httpError(w, r, fmt.Sprintf("failed to generate token: %s", err), http.StatusInternalServerError)
+		return
 	}
 
-	err = db.AddAccountSession(username, token)
+	err = db.AddAccountSession(request.Username, token)
 	if err != nil {
-		return AccountLoginResponse{}, fmt.Errorf("failed to add account session")
+		httpError(w, r, "failed to add account session", http.StatusInternalServerError)
+		return
 	}
 
-	return AccountLoginResponse{Token: base64.StdEncoding.EncodeToString(token)}, nil
+	response, err := json.Marshal(AccountLoginResponse{Token: base64.StdEncoding.EncodeToString(token)})
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
 }
 
 // /account/logout - log out of account
-func handleAccountLogout(token []byte) error {
-	if len(token) != 32 {
-		return fmt.Errorf("invalid token")
+func (s *Server) handleAccountLogout(w http.ResponseWriter, r *http.Request) {
+	token, err := base64.StdEncoding.DecodeString(r.Header.Get("Authorization"))
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to decode token: %s", err), http.StatusBadRequest)
+		return
 	}
 
-	err := db.RemoveSessionFromToken(token)
+	if len(token) != 32 {
+		httpError(w, r, "invalid token", http.StatusBadRequest)
+		return
+	}
+
+	err = db.RemoveSessionFromToken(token)
 	if err != nil {
 		if err == sql.ErrNoRows {
-			return fmt.Errorf("token not found")
+			httpError(w, r, "token not found", http.StatusBadRequest)
+			return
 		}
 
-		return fmt.Errorf("failed to remove account session")
+		httpError(w, r, "failed to remove account session", http.StatusInternalServerError)
+		return
 	}
 
-	return nil
+	w.WriteHeader(http.StatusOK)
 }

api/daily.go

@@ -5,13 +5,15 @@ import (
 	"crypto/rand"
 	"encoding/base64"
 	"encoding/binary"
+	"encoding/json"
 	"fmt"
 	"log"
+	"net/http"
 	"os"
+	"strconv"
 	"time"
 
 	"github.com/Flashfyre/pokerogue-server/db"
-	"github.com/Flashfyre/pokerogue-server/defs"
 	"github.com/go-co-op/gocron"
 )
 
@@ -24,33 +26,26 @@ var (
 )
 
 func ScheduleDailyRunRefresh() {
-	dailyRunScheduler.Every(1).Day().At("00:00").Do(func() error {
-		err := InitDailyRun()
-		if err != nil {
-			log.Fatal(err)
-		}
-
-		return nil
-	}())
+	dailyRunScheduler.Every(1).Day().At("00:00").Do(InitDailyRun)
 	dailyRunScheduler.StartAsync()
 }
 
-func InitDailyRun() error {
+func InitDailyRun() {
 	secret, err := os.ReadFile("secret.key")
 	if err != nil {
 		if !os.IsNotExist(err) {
-			return fmt.Errorf("failed to read daily seed secret: %s", err)
+			log.Fatalf("failed to read daily seed secret: %s", err)
 		}
 
 		newSecret := make([]byte, 32)
 		_, err := rand.Read(newSecret)
 		if err != nil {
-			return fmt.Errorf("failed to generate daily seed secret: %s", err)
+			log.Fatalf("failed to generate daily seed secret: %s", err)
 		}
 
 		err = os.WriteFile("secret.key", newSecret, 0400)
 		if err != nil {
-			return fmt.Errorf("failed to write daily seed secret: %s", err)
+			log.Fatalf("failed to write daily seed secret: %s", err)
 		}
 
 		secret = newSecret
@@ -66,8 +61,6 @@ func InitDailyRun() error {
 	}
 
 	log.Printf("Daily Run Seed: %s", dailyRunSeed)
-
-	return nil
 }
 
 func deriveDailyRunSeed(seedTime time.Time) []byte {
@@ -79,27 +72,79 @@ func deriveDailyRunSeed(seedTime time.Time) []byte {
 	return hashedSeed[:]
 }
 
+// /daily/seed - fetch daily run seed
+func (s *Server) handleSeed(w http.ResponseWriter) {
+	w.Write([]byte(dailyRunSeed))
+}
+
 // /daily/rankings - fetch daily rankings
-func handleRankings(uuid []byte, category, page int) ([]defs.DailyRanking, error) {
-	err := db.UpdateAccountLastActivity(uuid)
+func (s *Server) handleRankings(w http.ResponseWriter, r *http.Request) {
+	uuid, err := getUuidFromRequest(r)
+	if err != nil {
+		httpError(w, r, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	err = db.UpdateAccountLastActivity(uuid)
 	if err != nil {
 		log.Print("failed to update account last activity")
 	}
 
+	var category int
+	if r.URL.Query().Has("category") {
+		category, err = strconv.Atoi(r.URL.Query().Get("category"))
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to convert category: %s", err), http.StatusBadRequest)
+			return
+		}
+	}
+
+	page := 1
+	if r.URL.Query().Has("page") {
+		page, err = strconv.Atoi(r.URL.Query().Get("page"))
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to convert page: %s", err), http.StatusBadRequest)
+			return
+		}
+	}
+
 	rankings, err := db.FetchRankings(category, page)
 	if err != nil {
 		log.Print("failed to retrieve rankings")
 	}
 
-	return rankings, nil
+	response, err := json.Marshal(rankings)
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
 }
 
 // /daily/rankingpagecount - fetch daily ranking page count
-func handleRankingPageCount(category int) (int, error) {
+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 {
+			httpError(w, r, fmt.Sprintf("failed to convert category: %s", err), http.StatusBadRequest)
+			return
+		}
+	}
+
 	pageCount, err := db.FetchRankingPageCount(category)
 	if err != nil {
 		log.Print("failed to retrieve ranking page count")
 	}
 
-	return pageCount, nil
+	response, err := json.Marshal(pageCount)
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
 }

api/error.go

@@ -5,7 +5,7 @@ import (
 	"net/http"
 )
 
-func httpError(w http.ResponseWriter, r *http.Request, err error, code int) {
-	log.Printf("%s: %s\n", r.URL.Path, err)
-	http.Error(w, err.Error(), code)
+func httpError(w http.ResponseWriter, r *http.Request, error string, code int) {
+	log.Printf("%s: %s\n", r.URL.Path, error)
+	http.Error(w, error, code)
 }

api/game.go

@@ -1,10 +1,14 @@
 package api
 
 import (
+	"encoding/json"
+	"fmt"
 	"log"
+	"net/http"
 	"time"
 
 	"github.com/Flashfyre/pokerogue-server/db"
+	"github.com/Flashfyre/pokerogue-server/defs"
 	"github.com/go-co-op/gocron"
 )
 
@@ -26,14 +30,49 @@ func updateStats() {
 	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
+func (s *Server) handlePlayerCountGet(w http.ResponseWriter, r *http.Request) {
+	response, err := json.Marshal(playerCount)
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
+}
+
+// /game/titlestats - get title stats
+func (s *Server) handleTitleStatsGet(w http.ResponseWriter, r *http.Request) {
+	titleStats := &defs.TitleStats{
+		PlayerCount: playerCount,
+		BattleCount: battleCount,
+	}
+	response, err := json.Marshal(titleStats)
+	if err != nil {
+		httpError(w, r, 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, r *http.Request) {
+	response, err := json.Marshal(classicSessionCount)
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
+	w.Write(response)
+}

api/generic.go

@@ -1,26 +1,14 @@
 package api
 
 import (
-	"encoding/base64"
 	"encoding/gob"
-	"encoding/json"
-	"fmt"
 	"net/http"
-	"strconv"
-
-	"github.com/Flashfyre/pokerogue-server/defs"
 )
 
 type Server struct {
 	Debug bool
 }
 
-/*
-	The caller of endpoint handler functions are responsible for extracting the necessary data from the request.
-	Handler functions are responsible for checking the validity of this data and returning a result or error.
-	Handlers should not return serialized JSON, instead return the struct itself.
-*/
-
 func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	gob.Register([]interface{}{})
 	gob.Register(map[string]interface{}{})
@@ -37,239 +25,37 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	}
 
 	switch r.URL.Path {
-	// /account
 	case "/account/info":
-		username, err := getUsernameFromRequest(r)
-		if err != nil {
-			httpError(w, r, err, http.StatusBadRequest)
-			return
-		}
-
-		uuid, err := getUUIDFromRequest(r) // lazy
-		if err != nil {
-			httpError(w, r, err, http.StatusBadRequest)
-			return
-		}
-
-		info, err := handleAccountInfo(username, uuid)
-		if err != nil {
-			httpError(w, r, err, http.StatusInternalServerError)
-			return
-		}
-
-		response, err := json.Marshal(info)
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to marshal response json: %s", err), http.StatusInternalServerError)
-			return
-		}
-
-		w.Write(response)
+		s.handleAccountInfo(w, r)
 	case "/account/register":
-		var request AccountRegisterRequest
-		err := json.NewDecoder(r.Body).Decode(&request)
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to decode request body: %s", err), http.StatusBadRequest)
-			return
-		}
-
-		err = handleAccountRegister(request.Username, request.Password)
-		if err != nil {
-			httpError(w, r, err, http.StatusInternalServerError)
-			return
-		}
-
-		w.WriteHeader(http.StatusOK)
+		s.handleAccountRegister(w, r)
 	case "/account/login":
-		var request AccountLoginRequest
-		err := json.NewDecoder(r.Body).Decode(&request)
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to decode request body: %s", err), http.StatusBadRequest)
-			return
-		}
-
-		token, err := handleAccountLogin(request.Username, request.Password)
-		if err != nil {
-			httpError(w, r, err, http.StatusInternalServerError)
-			return
-		}
-
-		response, err := json.Marshal(token)
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to marshal response json: %s", err), http.StatusInternalServerError)
-			return
-		}
-
-		w.Write(response)
+		s.handleAccountLogin(w, r)
 	case "/account/logout":
-		token, err := base64.StdEncoding.DecodeString(r.Header.Get("Authorization"))
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to decode token: %s", err), http.StatusBadRequest)
-			return
-		}
+		s.handleAccountLogout(w, r)
 
-		err = handleAccountLogout(token)
-		if err != nil {
-			httpError(w, r, err, http.StatusInternalServerError)
-			return
-		}
-
-		w.WriteHeader(http.StatusOK)
-
-		// /game
 	case "/game/playercount":
-		w.Write([]byte(strconv.Itoa(playerCount)))
+		s.handlePlayerCountGet(w, r)
 	case "/game/titlestats":
-		response, err := json.Marshal(&defs.TitleStats{
-			PlayerCount: playerCount,
-			BattleCount: battleCount,
-		})
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to marshal response json: %s", err), http.StatusInternalServerError)
-			return
-		}
-
-		w.Write(response)
+		s.handleTitleStatsGet(w, r)
 	case "/game/classicsessioncount":
-		w.Write([]byte(strconv.Itoa(classicSessionCount)))
+		s.handleClassicSessionCountGet(w, r)
 
-		// /savedata
-	case "/savedata/get", "/savedata/update", "/savedata/delete", "/savedata/clear":
-		uuid, err := getUUIDFromRequest(r)
-		if err != nil {
-			httpError(w, r, err, http.StatusBadRequest)
-			return
-		}
+	case "/savedata/get":
+		s.handleSavedataGet(w, r)
+	case "/savedata/update":
+		s.handleSavedataUpdate(w, r)
+	case "/savedata/delete":
+		s.handleSavedataDelete(w, r)
+	case "/savedata/clear":
+		s.handleSavedataClear(w, r)
 
-		datatype := -1
-		if r.URL.Query().Has("datatype") {
-			datatype, err = strconv.Atoi(r.URL.Query().Get("datatype"))
-			if err != nil {
-				httpError(w, r, err, http.StatusBadRequest)
-				return
-			}
-		}
-
-		var slot int
-		if r.URL.Query().Has("slot") {
-			slot, err = strconv.Atoi(r.URL.Query().Get("slot"))
-			if err != nil {
-				httpError(w, r, err, http.StatusBadRequest)
-				return
-			}
-		}
-
-		var save any
-		// /savedata/get and /savedata/delete specify datatype, but don't expect data in body
-		if r.URL.Path != "/savedata/get" && r.URL.Path != "/savedata/delete" {
-			if datatype == 0 {
-				var system defs.SystemSaveData
-				err = json.NewDecoder(r.Body).Decode(&system)
-				if err != nil {
-					httpError(w, r, fmt.Errorf("failed to decode request body: %s", err), http.StatusBadRequest)
-					return
-				}
-
-				save = system
-				// /savedata/clear doesn't specify datatype, it is assumed to be 1 (session)
-			} else if datatype == 1 || r.URL.Path == "/savedata/clear" {
-				var session defs.SessionSaveData
-				err = json.NewDecoder(r.Body).Decode(&session)
-				if err != nil {
-					httpError(w, r, fmt.Errorf("failed to decode request body: %s", err), http.StatusBadRequest)
-					return
-				}
-
-				save = session
-			}
-		}
-
-		switch r.URL.Path {
-		case "/savedata/get":
-			save, err = handleSavedataGet(uuid, datatype, slot)
-		case "/savedata/update":
-			err = handleSavedataUpdate(uuid, slot, save)
-		case "/savedata/delete":
-			err = handleSavedataDelete(uuid, datatype, slot)
-		case "/savedata/clear":
-			// doesn't return a save, but it works
-			save, err = handleSavedataClear(uuid, slot, save.(defs.SessionSaveData))
-		}
-		if err != nil {
-			httpError(w, r, err, http.StatusInternalServerError)
-			return
-		}
-
-		if save == nil || r.URL.Path == "/savedata/update" {
-			w.WriteHeader(http.StatusOK)
-			return
-		}
-
-		response, err := json.Marshal(save)
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to marshal response json: %s", err), http.StatusInternalServerError)
-			return
-		}
-
-		w.Write(response)
-
-		// /daily
 	case "/daily/seed":
-		w.Write([]byte(dailyRunSeed))
+		s.handleSeed(w)
 	case "/daily/rankings":
-		uuid, err := getUUIDFromRequest(r)
-		if err != nil {
-			httpError(w, r, err, http.StatusBadRequest)
-			return
-		}
-
-		var category int
-		if r.URL.Query().Has("category") {
-			category, err = strconv.Atoi(r.URL.Query().Get("category"))
-			if err != nil {
-				httpError(w, r, fmt.Errorf("failed to convert category: %s", err), http.StatusBadRequest)
-				return
-			}
-		}
-
-		page := 1
-		if r.URL.Query().Has("page") {
-			page, err = strconv.Atoi(r.URL.Query().Get("page"))
-			if err != nil {
-				httpError(w, r, fmt.Errorf("failed to convert page: %s", err), http.StatusBadRequest)
-				return
-			}
-		}
-
-		rankings, err := handleRankings(uuid, category, page)
-		if err != nil {
-			httpError(w, r, err, http.StatusInternalServerError)
-			return
-		}
-
-		response, err := json.Marshal(rankings)
-		if err != nil {
-			httpError(w, r, fmt.Errorf("failed to marshal response json: %s", err), http.StatusInternalServerError)
-			return
-		}
-
-		w.Write(response)
+		s.handleRankings(w, r)
 	case "/daily/rankingpagecount":
-		var category int
-		if r.URL.Query().Has("category") {
-			var err error
-			category, err = strconv.Atoi(r.URL.Query().Get("category"))
-			if err != nil {
-				httpError(w, r, fmt.Errorf("failed to convert category: %s", err), http.StatusBadRequest)
-				return
-			}
-		}
-
-		count, err := handleRankingPageCount(category)
-		if err != nil {
-			httpError(w, r, err, http.StatusInternalServerError)
-		}
-
-		w.Write([]byte(strconv.Itoa(count)))
+		s.handleRankingPageCount(w, r)
 	}
 }
 

api/savedata-helper.go

@@ -40,12 +40,12 @@ func readSystemSaveData(uuid []byte) (defs.SystemSaveData, error) {
 	return system, nil
 }
 
-func readSessionSaveData(uuid []byte, slotID int) (defs.SessionSaveData, error) {
+func readSessionSaveData(uuid []byte, slotId int) (defs.SessionSaveData, error) {
 	var session defs.SessionSaveData
 
 	fileName := "session"
-	if slotID != 0 {
-		fileName += strconv.Itoa(slotID)
+	if slotId != 0 {
+		fileName += strconv.Itoa(slotId)
 	}
 
 	save, err := os.ReadFile(fmt.Sprintf("userdata/%s/%s.pzs", hex.EncodeToString(uuid), fileName))

api/savedata.go

@@ -4,8 +4,10 @@ import (
 	"bytes"
 	"encoding/gob"
 	"encoding/hex"
+	"encoding/json"
 	"fmt"
 	"log"
+	"net/http"
 	"os"
 	"strconv"
 
@@ -17,17 +19,25 @@ import (
 const sessionSlotCount = 3
 
 // /savedata/get - get save data
-func handleSavedataGet(uuid []byte, datatype, slot int) (any, error) {
-	switch datatype {
-	case 0: // System
+func (s *Server) handleSavedataGet(w http.ResponseWriter, r *http.Request) {
+	uuid, err := getUuidFromRequest(r)
+	if err != nil {
+		httpError(w, r, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	switch r.URL.Query().Get("datatype") {
+	case "0": // System
 		system, err := readSystemSaveData(uuid)
 		if err != nil {
-			return nil, err
+			httpError(w, r, err.Error(), http.StatusInternalServerError)
+			return
 		}
 
 		compensations, err := db.FetchAndClaimAccountCompensations(uuid)
 		if err != nil {
-			return nil, fmt.Errorf("failed to fetch compensations: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to fetch compensations: %s", err), http.StatusInternalServerError)
+			return
 		}
 
 		for k, v := range compensations {
@@ -35,140 +45,215 @@ func handleSavedataGet(uuid []byte, datatype, slot int) (any, error) {
 			system.VoucherCounts[typeKey] += v
 		}
 
-		return system, nil
-	case 1: // Session
-		if slot < 0 || slot >= sessionSlotCount {
-			return nil, fmt.Errorf("slot id %d out of range", slot)
-		}
-
-		session, err := readSessionSaveData(uuid, slot)
+		saveJson, err := json.Marshal(system)
 		if err != nil {
-			return nil, err
+			httpError(w, r, fmt.Sprintf("failed to marshal save to json: %s", err), http.StatusInternalServerError)
+			return
 		}
 
-		return session, nil
+		w.Write(saveJson)
+	case "1": // Session
+		slotId, err := strconv.Atoi(r.URL.Query().Get("slot"))
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to convert slot id: %s", err), http.StatusBadRequest)
+			return
+		}
+
+		if slotId < 0 || slotId >= sessionSlotCount {
+			httpError(w, r, fmt.Sprintf("slot id %d out of range", slotId), http.StatusBadRequest)
+			return
+		}
+
+		session, err := readSessionSaveData(uuid, slotId)
+		if err != nil {
+			httpError(w, r, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
+		saveJson, err := json.Marshal(session)
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to marshal save to json: %s", err), http.StatusInternalServerError)
+			return
+		}
+
+		w.Write(saveJson)
 	default:
-		return nil, fmt.Errorf("invalid data type")
+		httpError(w, r, "invalid data type", http.StatusBadRequest)
+		return
 	}
 }
 
 // /savedata/update - update save data
-func handleSavedataUpdate(uuid []byte, slot int, save any) error {
-	err := db.UpdateAccountLastActivity(uuid)
+func (s *Server) handleSavedataUpdate(w http.ResponseWriter, r *http.Request) {
+	uuid, err := getUuidFromRequest(r)
+	if err != nil {
+		httpError(w, r, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	err = db.UpdateAccountLastActivity(uuid)
 	if err != nil {
 		log.Print("failed to update account last activity")
 	}
 
-	hexUUID := hex.EncodeToString(uuid)
+	hexUuid := hex.EncodeToString(uuid)
 
-	switch save := save.(type) {
-	case defs.SystemSaveData: // System
-		if save.TrainerId == 0 && save.SecretId == 0 {
-			return fmt.Errorf("invalid system data")
+	switch r.URL.Query().Get("datatype") {
+	case "0": // System
+		var system defs.SystemSaveData
+		err = json.NewDecoder(r.Body).Decode(&system)
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to decode request body: %s", err), http.StatusBadRequest)
+			return
 		}
 
-		err = db.UpdateAccountStats(uuid, save.GameStats)
+		if system.TrainerId == 0 && system.SecretId == 0 {
+			httpError(w, r, "invalid system data", http.StatusInternalServerError)
+			return
+		}
+
+		err = db.UpdateAccountStats(uuid, system.GameStats)
 		if err != nil {
-			return fmt.Errorf("failed to update account stats: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to update account stats: %s", err), http.StatusBadRequest)
+			return
 		}
 
 		var gobBuffer bytes.Buffer
-		err = gob.NewEncoder(&gobBuffer).Encode(save)
+		err = gob.NewEncoder(&gobBuffer).Encode(system)
 		if err != nil {
-			return fmt.Errorf("failed to serialize save: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to serialize save: %s", err), http.StatusInternalServerError)
+			return
 		}
 
 		zstdWriter, err := zstd.NewWriter(nil)
 		if err != nil {
-			return fmt.Errorf("failed to create zstd writer, %s", err)
+			httpError(w, r, fmt.Sprintf("failed to create zstd writer, %s", err), http.StatusInternalServerError)
+			return
 		}
 
 		compressed := zstdWriter.EncodeAll(gobBuffer.Bytes(), nil)
 
-		err = os.MkdirAll("userdata/"+hexUUID, 0755)
+		err = os.MkdirAll("userdata/"+hexUuid, 0755)
 		if err != nil && !os.IsExist(err) {
-			return fmt.Errorf("failed to create userdata folder: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to create userdata folder: %s", err), http.StatusInternalServerError)
+			return
 		}
 
-		err = os.WriteFile("userdata/"+hexUUID+"/system.pzs", compressed, 0644)
+		err = os.WriteFile("userdata/"+hexUuid+"/system.pzs", compressed, 0644)
 		if err != nil {
-			return fmt.Errorf("failed to write save file: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to write save file: %s", err), http.StatusInternalServerError)
+			return
 		}
 
 		db.DeleteClaimedAccountCompensations(uuid)
-	case defs.SessionSaveData: // Session
-		if slot < 0 || slot >= sessionSlotCount {
-			return fmt.Errorf("slot id %d out of range", slot)
+	case "1": // Session
+		slotId, err := strconv.Atoi(r.URL.Query().Get("slot"))
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to convert slot id: %s", err), http.StatusBadRequest)
+			return
+		}
+
+		if slotId < 0 || slotId >= sessionSlotCount {
+			httpError(w, r, fmt.Sprintf("slot id %d out of range", slotId), http.StatusBadRequest)
+			return
 		}
 
 		fileName := "session"
-		if slot != 0 {
-			fileName += strconv.Itoa(slot)
+		if slotId != 0 {
+			fileName += strconv.Itoa(slotId)
+		}
+
+		var session defs.SessionSaveData
+		err = json.NewDecoder(r.Body).Decode(&session)
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to decode request body: %s", err), http.StatusBadRequest)
+			return
 		}
 
 		var gobBuffer bytes.Buffer
-		err = gob.NewEncoder(&gobBuffer).Encode(save)
+		err = gob.NewEncoder(&gobBuffer).Encode(session)
 		if err != nil {
-			return fmt.Errorf("failed to serialize save: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to serialize save: %s", err), http.StatusInternalServerError)
+			return
 		}
 
 		zstdWriter, err := zstd.NewWriter(nil)
 		if err != nil {
-			return fmt.Errorf("failed to create zstd writer, %s", err)
+			httpError(w, r, fmt.Sprintf("failed to create zstd writer, %s", err), http.StatusInternalServerError)
+			return
 		}
 
 		compressed := zstdWriter.EncodeAll(gobBuffer.Bytes(), nil)
 
-		err = os.MkdirAll("userdata/"+hexUUID, 0755)
+		err = os.MkdirAll("userdata/"+hexUuid, 0755)
 		if err != nil && !os.IsExist(err) {
-			return fmt.Errorf(fmt.Sprintf("failed to create userdata folder: %s", err))
+			httpError(w, r, fmt.Sprintf("failed to create userdata folder: %s", err), http.StatusInternalServerError)
+			return
 		}
 
-		err = os.WriteFile(fmt.Sprintf("userdata/%s/%s.pzs", hexUUID, fileName), compressed, 0644)
+		err = os.WriteFile(fmt.Sprintf("userdata/%s/%s.pzs", hexUuid, fileName), compressed, 0644)
 		if err != nil {
-			return fmt.Errorf("failed to write save file: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to write save file: %s", err), http.StatusInternalServerError)
+			return
 		}
 	default:
-		return fmt.Errorf("invalid data type")
+		httpError(w, r, "invalid data type", http.StatusBadRequest)
+		return
 	}
 
-	return nil
+	w.WriteHeader(http.StatusOK)
 }
 
 // /savedata/delete - delete save data
-func handleSavedataDelete(uuid []byte, datatype, slot int) error {
-	err := db.UpdateAccountLastActivity(uuid)
+func (s *Server) handleSavedataDelete(w http.ResponseWriter, r *http.Request) {
+	uuid, err := getUuidFromRequest(r)
+	if err != nil {
+		httpError(w, r, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	err = db.UpdateAccountLastActivity(uuid)
 	if err != nil {
 		log.Print("failed to update account last activity")
 	}
 
-	hexUUID := hex.EncodeToString(uuid)
+	hexUuid := hex.EncodeToString(uuid)
 
-	switch datatype {
-	case 0: // System
-		err := os.Remove("userdata/" + hexUUID + "/system.pzs")
+	switch r.URL.Query().Get("datatype") {
+	case "0": // System
+		err := os.Remove("userdata/" + hexUuid + "/system.pzs")
 		if err != nil && !os.IsNotExist(err) {
-			return fmt.Errorf("failed to delete save file: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to delete save file: %s", err), http.StatusInternalServerError)
+			return
 		}
-	case 1: // Session
-		if slot < 0 || slot >= sessionSlotCount {
-			return fmt.Errorf("slot id %d out of range", slot)
+	case "1": // Session
+		slotId, err := strconv.Atoi(r.URL.Query().Get("slot"))
+		if err != nil {
+			httpError(w, r, fmt.Sprintf("failed to convert slot id: %s", err), http.StatusBadRequest)
+			return
+		}
+
+		if slotId < 0 || slotId >= sessionSlotCount {
+			httpError(w, r, fmt.Sprintf("slot id %d out of range", slotId), http.StatusBadRequest)
+			return
 		}
 
 		fileName := "session"
-		if slot != 0 {
-			fileName += strconv.Itoa(slot)
+		if slotId != 0 {
+			fileName += strconv.Itoa(slotId)
 		}
 
-		err = os.Remove(fmt.Sprintf("userdata/%s/%s.pzs", hexUUID, fileName))
+		err = os.Remove(fmt.Sprintf("userdata/%s/%s.pzs", hexUuid, fileName))
 		if err != nil && !os.IsNotExist(err) {
-			return fmt.Errorf("failed to delete save file: %s", err)
+			httpError(w, r, fmt.Sprintf("failed to delete save file: %s", err), http.StatusInternalServerError)
+			return
 		}
 	default:
-		return fmt.Errorf("invalid data type")
+		httpError(w, r, "invalid data type", http.StatusBadRequest)
+		return
 	}
 
-	return nil
+	w.WriteHeader(http.StatusOK)
 }
 
 type SavedataClearResponse struct {
@@ -176,46 +261,73 @@ type SavedataClearResponse struct {
 }
 
 // /savedata/clear - mark session save data as cleared and delete
-func handleSavedataClear(uuid []byte, slot int, save defs.SessionSaveData) (SavedataClearResponse, error) {
-	err := db.UpdateAccountLastActivity(uuid)
+func (s *Server) handleSavedataClear(w http.ResponseWriter, r *http.Request) {
+	uuid, err := getUuidFromRequest(r)
+	if err != nil {
+		httpError(w, r, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	err = db.UpdateAccountLastActivity(uuid)
 	if err != nil {
 		log.Print("failed to update account last activity")
 	}
 
-	if slot < 0 || slot >= sessionSlotCount {
-		return SavedataClearResponse{}, fmt.Errorf("slot id %d out of range", slot)
+	slotId, err := strconv.Atoi(r.URL.Query().Get("slot"))
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to convert slot id: %s", err), http.StatusBadRequest)
+		return
 	}
 
-	sessionCompleted := validateSessionCompleted(save)
+	if slotId < 0 || slotId >= sessionSlotCount {
+		httpError(w, r, fmt.Sprintf("slot id %d out of range", slotId), http.StatusBadRequest)
+		return
+	}
+
+	var session defs.SessionSaveData
+	err = json.NewDecoder(r.Body).Decode(&session)
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to decode request body: %s", err), http.StatusBadRequest)
+		return
+	}
+
+	sessionCompleted := validateSessionCompleted(session)
 	newCompletion := false
 
-	if save.GameMode == 3 && save.Seed == dailyRunSeed {
-		waveCompleted := save.WaveIndex
+	if session.GameMode == 3 && session.Seed == dailyRunSeed {
+		waveCompleted := session.WaveIndex
 		if !sessionCompleted {
 			waveCompleted--
 		}
-		err = db.AddOrUpdateAccountDailyRun(uuid, save.Score, waveCompleted)
+		err = db.AddOrUpdateAccountDailyRun(uuid, session.Score, waveCompleted)
 		if err != nil {
 			log.Printf("failed to add or update daily run record: %s", err)
 		}
 	}
 
 	if sessionCompleted {
-		newCompletion, err = db.TryAddSeedCompletion(uuid, save.Seed, int(save.GameMode))
+		newCompletion, err = db.TryAddSeedCompletion(uuid, session.Seed, int(session.GameMode))
 		if err != nil {
 			log.Printf("failed to mark seed as completed: %s", err)
 		}
 	}
 
+	response, err := json.Marshal(SavedataClearResponse{Success: newCompletion})
+	if err != nil {
+		httpError(w, r, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
+		return
+	}
+
 	fileName := "session"
-	if slot != 0 {
-		fileName += strconv.Itoa(slot)
+	if slotId != 0 {
+		fileName += strconv.Itoa(slotId)
 	}
 
 	err = os.Remove(fmt.Sprintf("userdata/%s/%s.pzs", hex.EncodeToString(uuid), fileName))
 	if err != nil && !os.IsNotExist(err) {
-		return SavedataClearResponse{}, fmt.Errorf("failed to delete save file: %s", err)
+		httpError(w, r, fmt.Sprintf("failed to delete save file: %s", err), http.StatusInternalServerError)
+		return
 	}
 
-	return SavedataClearResponse{Success: newCompletion}, nil
+	w.Write(response)
 }

db/account.go

@@ -157,7 +157,7 @@ func FetchAccountKeySaltFromUsername(username string) ([]byte, []byte, error) {
 	return key, salt, nil
 }
 
-func FetchUUIDFromToken(token []byte) ([]byte, error) {
+func FetchUuidFromToken(token []byte) ([]byte, error) {
 	var uuid []byte
 	err := handle.QueryRow("SELECT uuid FROM sessions WHERE token = ? AND expire > UTC_TIMESTAMP()", token).Scan(&uuid)
 	if err != nil {