85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package api
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/gob"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/Flashfyre/pokerogue-server/defs"
|
|
"github.com/klauspost/compress/zstd"
|
|
)
|
|
|
|
func GetSystemSaveData(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 GetSessionSaveData(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.WaveIndex == 200
|
|
case 3:
|
|
return session.WaveIndex == 50
|
|
}
|
|
return false
|
|
}
|