Update random daily seed logic

api/account-helper.go

@@ -22,7 +22,7 @@ func GetUsernameFromRequest(request *http.Request) (string, error) {
 		return "", fmt.Errorf("invalid token length: got %d, expected 32", len(token))
 	}
 
-	username, err := db.GetUsernameFromToken(token)
+	username, err := db.FetchUsernameFromToken(token)
 	if err != nil {
 		return "", fmt.Errorf("failed to validate token: %s", err)
 	}
@@ -44,7 +44,7 @@ func GetUuidFromRequest(request *http.Request) ([]byte, error) {
 		return nil, fmt.Errorf("invalid token length: got %d, expected 32", len(token))
 	}
 
-	uuid, err := db.GetUuidFromToken(token)
+	uuid, err := db.FetchUuidFromToken(token)
 	if err != nil {
 		return nil, fmt.Errorf("failed to validate token: %s", err)
 	}

api/account.go

@@ -144,7 +144,7 @@ func (s *Server) HandleAccountLogin(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	key, salt, err := db.GetAccountKeySaltFromUsername(request.Username)
+	key, salt, err := db.FetchAccountKeySaltFromUsername(request.Username)
 	if err != nil {
 		if err == sql.ErrNoRows {
 			http.Error(w, "account doesn't exist", http.StatusBadRequest)

api/daily.go

@@ -1,17 +1,26 @@
 package api
 
 import (
+	"crypto/md5"
+	"crypto/rand"
+	"encoding/base64"
+	"encoding/binary"
 	"encoding/json"
 	"fmt"
 	"log"
 	"net/http"
+	"os"
 	"strconv"
+	"time"
 
 	"github.com/Flashfyre/pokerogue-server/db"
 )
 
+const secondsPerDay = 60 * 60 * 24
+
 var (
-	dailyRunSeed string
+	dailyRunSecret []byte
+	dailyRunSeed   string
 )
 
 func ScheduleDailyRunRefresh() {
@@ -21,24 +30,48 @@ func ScheduleDailyRunRefresh() {
 }
 
 func InitDailyRun() {
-	var err error
-	dailyRunSeed, err = db.GetDailyRunSeed()
+	secret, err := os.ReadFile("secret.key")
 	if err != nil {
-		log.Printf("failed to generated daily run seed: %s", err.Error())
+		if !os.IsNotExist(err) {
+			log.Fatalf("failed to read daily seed secret: %s", err)
+		}
+
+		newSecret := make([]byte, 32)
+		_, err := rand.Read(newSecret)
+		if err != nil {
+			log.Fatalf("failed to generate daily seed secret: %s", err)
+		}
+
+		err = os.WriteFile("secret.key", newSecret, 0400)
+		if err != nil {
+			log.Fatalf("failed to write daily seed secret: %s", err)
+		}
+
+		secret = newSecret
 	}
 
-	if dailyRunSeed == "" {
-		dailyRunSeed = RandString(24)
-		err := db.TryAddDailyRun(dailyRunSeed)
-		if err != nil {
-			log.Print(err.Error())
-		} else {
-			log.Printf("Daily Run Seed: %s", dailyRunSeed)
-		}
+	dailyRunSecret = secret
+
+	dailyRunSeed = base64.StdEncoding.EncodeToString(DeriveDailyRunSeed(time.Now().UTC()))
+
+	err = db.TryAddDailyRun(dailyRunSeed)
+	if err != nil {
+		log.Print(err.Error())
+	} else {
+		log.Printf("Daily Run Seed: %s", dailyRunSeed)
 	}
 }
 
-// /daily/seed - get daily run seed
+func DeriveDailyRunSeed(seedTime time.Time) []byte {
+	day := make([]byte, 8)
+	binary.BigEndian.PutUint64(day, uint64(seedTime.Unix()/secondsPerDay))
+
+	hashedSeed := md5.Sum(append(day, dailyRunSecret...))
+
+	return hashedSeed[:]
+}
+
+// /daily/seed - fetch daily run seed
 
 func (s *Server) HandleSeed(w http.ResponseWriter, r *http.Request) {
 	w.Write([]byte(dailyRunSeed))
@@ -60,7 +93,7 @@ func (s *Server) HandleRankings(w http.ResponseWriter, r *http.Request) {
 		page = 1
 	}
 
-	rankings, err := db.GetRankings(page)
+	rankings, err := db.FetchRankings(page)
 	if err != nil {
 		log.Print("failed to retrieve rankings")
 	}

api/savedata-helper.go

@@ -12,7 +12,7 @@ import (
 	"github.com/klauspost/compress/zstd"
 )
 
-func GetSystemSaveData(uuid []byte) (defs.SystemSaveData, error) {
+func ReadSystemSaveData(uuid []byte) (defs.SystemSaveData, error) {
 	var system defs.SystemSaveData
 
 	save, err := os.ReadFile("userdata/" + hex.EncodeToString(uuid) + "/system.pzs")
@@ -40,7 +40,7 @@ func GetSystemSaveData(uuid []byte) (defs.SystemSaveData, error) {
 	return system, nil
 }
 
-func GetSessionSaveData(uuid []byte, slotId int) (defs.SessionSaveData, error) {
+func ReadSessionSaveData(uuid []byte, slotId int) (defs.SessionSaveData, error) {
 	var session defs.SessionSaveData
 
 	fileName := "session"
@@ -76,9 +76,9 @@ func GetSessionSaveData(uuid []byte, slotId int) (defs.SessionSaveData, error) {
 func ValidateSessionCompleted(session defs.SessionSaveData) bool {
 	switch session.GameMode {
 	case 0:
-		return session.WaveIndex == 200
+		return session.BattleType == 2 && session.WaveIndex == 200
 	case 3:
-		return session.WaveIndex == 50
+		return session.BattleType == 2 && session.WaveIndex == 50
 	}
 	return false
 }

api/savedata.go

@@ -29,7 +29,7 @@ func (s *Server) HandleSavedataGet(w http.ResponseWriter, r *http.Request) {
 
 	switch r.URL.Query().Get("datatype") {
 	case "0": // System
-		system, err := GetSystemSaveData(uuid)
+		system, err := ReadSystemSaveData(uuid)
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusInternalServerError)
 			return
@@ -54,7 +54,7 @@ func (s *Server) HandleSavedataGet(w http.ResponseWriter, r *http.Request) {
 			return
 		}
 
-		session, err := GetSessionSaveData(uuid, slotId)
+		session, err := ReadSessionSaveData(uuid, slotId)
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusInternalServerError)
 			return
@@ -276,12 +276,12 @@ func (s *Server) HandleSavedataClear(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
-	sessionCompleted := session.BattleType == 2 && ValidateSessionCompleted(session)
+	sessionCompleted := ValidateSessionCompleted(session)
 	newCompletion := false
 
 	if session.GameMode == 3 {
 		waveCompleted := session.WaveIndex
-		if session.BattleType != 2 {
+		if !sessionCompleted {
 			waveCompleted--
 		}
 		err = db.AddOrUpdateAccountDailyRun(uuid, session.Score, waveCompleted)

api/utils.go

@@ -1,20 +0,0 @@
-package api
-
-import (
-	"crypto/rand"
-)
-
-const randRunes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890"
-const lenRandRunes = len(randRunes)
-
-func RandString(length int) string {
-	b := make([]byte, length)
-
-	rand.Read(b)
-
-	for i := range b {
-		b[i] = randRunes[int(b[i])%lenRandRunes]
-	}
-
-	return string(b)
-}