rogueserver/api/savedata-helper.go

package api

import (
	"bytes"
	"encoding/gob"
	"encoding/hex"
	"fmt"
	"os"
	"strconv"

	"github.com/Flashfyre/pokerogue-server/defs"
	"github.com/klauspost/compress/zstd"
)

func ReadSystemSaveData(uuid []byte) (defs.SystemSaveData, error) {
	var system defs.SystemSaveData

	save, err := os.ReadFile("userdata/" + hex.EncodeToString(uuid) + "/system.pzs")
	if err != nil {
		return system, fmt.Errorf("failed to read save file: %s", err)
	}

	zstdReader, err := zstd.NewReader(nil)
	if err != nil {
		return system, fmt.Errorf("failed to create zstd reader: %s", err)
	}

	decompressed, err := zstdReader.DecodeAll(save, nil)
	if err != nil {
		return system, fmt.Errorf("failed to decompress save file: %s", err)
	}

	gobDecoderBuf := bytes.NewBuffer(decompressed)

	err = gob.NewDecoder(gobDecoderBuf).Decode(&system)
	if err != nil {
		return system, fmt.Errorf("failed to deserialize save: %s", err)
	}

	return system, nil
}

func ReadSessionSaveData(uuid []byte, slotId int) (defs.SessionSaveData, error) {
	var session defs.SessionSaveData

	fileName := "session"
	if slotId != 0 {
		fileName += strconv.Itoa(slotId)
	}

	save, err := os.ReadFile(fmt.Sprintf("userdata/%s/%s.pzs", hex.EncodeToString(uuid), fileName))
	if err != nil {
		return session, fmt.Errorf("failed to read save file: %s", err)
	}

	zstdReader, err := zstd.NewReader(nil)
	if err != nil {
		return session, fmt.Errorf("failed to create zstd reader: %s", err)
	}

	decompressed, err := zstdReader.DecodeAll(save, nil)
	if err != nil {
		return session, fmt.Errorf("failed to decompress save file: %s", err)
	}

	gobDecoderBuf := bytes.NewBuffer(decompressed)

	err = gob.NewDecoder(gobDecoderBuf).Decode(&session)
	if err != nil {
		return session, fmt.Errorf("failed to deserialize save: %s", err)
	}

	return session, nil
}

func ValidateSessionCompleted(session defs.SessionSaveData) bool {
	switch session.GameMode {
	case 0:
		return session.BattleType == 2 && session.WaveIndex == 200
	case 3:
		return session.BattleType == 2 && session.WaveIndex == 50
	}
	return false
}