rogueserver/api/savedata.go

package api

import (
	"bytes"
	"encoding/gob"
	"encoding/hex"
	"encoding/json"
	"fmt"
	"log"
	"net/http"
	"os"
	"strconv"

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

const sessionSlotCount = 3

// /savedata/get - get save data

func (s *Server) HandleSavedataGet(w http.ResponseWriter, r *http.Request) {
	uuid, err := GetUuidFromRequest(r)
	if err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	hexUuid := hex.EncodeToString(uuid)

	switch r.URL.Query().Get("datatype") {
	case "0": // System
		save, err := os.ReadFile("userdata/" + hexUuid + "/system.pzs")
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to read save file: %s", err), http.StatusInternalServerError)
			return
		}

		zstdReader, err := zstd.NewReader(nil)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to create zstd reader: %s", err), http.StatusInternalServerError)
			return
		}

		decompressed, err := zstdReader.DecodeAll(save, nil)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to decompress save file: %s", err), http.StatusInternalServerError)
			return
		}

		gobDecoderBuf := bytes.NewBuffer(decompressed)

		var system SystemSaveData
		err = gob.NewDecoder(gobDecoderBuf).Decode(&system)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to deserialize save: %s", err), http.StatusInternalServerError)
			return
		}

		saveJson, err := json.Marshal(system)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to marshal save to json: %s", err), http.StatusInternalServerError)
			return
		}

		w.Write(saveJson)
	case "1": // Session
		slotId, err := strconv.Atoi(r.URL.Query().Get("slot"))
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to convert slot id: %s", err), http.StatusBadRequest)
			return
		}

		if slotId < 0 || slotId >= sessionSlotCount {
			http.Error(w, fmt.Sprintf("slot id %d out of range", slotId), http.StatusBadRequest)
			return
		}

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

		save, err := os.ReadFile(fmt.Sprintf("userdata/%s/%s.pzs", hexUuid, fileName))
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to read save file: %s", err), http.StatusInternalServerError)
			return
		}

		zstdReader, err := zstd.NewReader(nil)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to create zstd reader: %s", err), http.StatusInternalServerError)
			return
		}

		decompressed, err := zstdReader.DecodeAll(save, nil)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to decompress save file: %s", err), http.StatusInternalServerError)
			return
		}

		gobDecoderBuf := bytes.NewBuffer(decompressed)

		var session SessionSaveData
		err = gob.NewDecoder(gobDecoderBuf).Decode(&session)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to deserialize save: %s", err), http.StatusInternalServerError)
			return
		}

		saveJson, err := json.Marshal(session)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to marshal save to json: %s", err), http.StatusInternalServerError)
			return
		}

		w.Write(saveJson)
	default:
		http.Error(w, "invalid data type", http.StatusBadRequest)
		return
	}
}

// /savedata/update - update save data

func (s *Server) HandleSavedataUpdate(w http.ResponseWriter, r *http.Request) {
	uuid, err := GetUuidFromRequest(r)
	if err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	err = db.UpdateAccountLastActivity(uuid)
	if err != nil {
		log.Print("failed to update account last activity")
	}

	hexUuid := hex.EncodeToString(uuid)

	switch r.URL.Query().Get("datatype") {
	case "0": // System
		var system SystemSaveData
		err = json.NewDecoder(r.Body).Decode(&system)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to decode request body: %s", err), http.StatusBadRequest)
			return
		}

		if system.TrainerId == 0 && system.SecretId == 0 {
			http.Error(w, "invalid system data", http.StatusInternalServerError)
			return
		}

		var gobBuffer bytes.Buffer
		err = gob.NewEncoder(&gobBuffer).Encode(system)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to serialize save: %s", err), http.StatusInternalServerError)
			return
		}

		zstdWriter, err := zstd.NewWriter(nil)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to create zstd writer, %s", err), http.StatusInternalServerError)
			return
		}

		compressed := zstdWriter.EncodeAll(gobBuffer.Bytes(), nil)

		err = os.MkdirAll("userdata/"+hexUuid, 0755)
		if err != nil && !os.IsExist(err) {
			http.Error(w, fmt.Sprintf("failed to create userdata folder: %s", err), http.StatusInternalServerError)
			return
		}

		err = os.WriteFile("userdata/"+hexUuid+"/system.pzs", compressed, 0644)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to write save file: %s", err), http.StatusInternalServerError)
			return
		}
	case "1": // Session
		slotId, err := strconv.Atoi(r.URL.Query().Get("slot"))
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to convert slot id: %s", err), http.StatusBadRequest)
			return
		}

		if slotId < 0 || slotId >= sessionSlotCount {
			http.Error(w, fmt.Sprintf("slot id %d out of range", slotId), http.StatusBadRequest)
			return
		}

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

		var session SessionSaveData
		err = json.NewDecoder(r.Body).Decode(&session)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to decode request body: %s", err), http.StatusBadRequest)
			return
		}

		var gobBuffer bytes.Buffer
		err = gob.NewEncoder(&gobBuffer).Encode(session)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to serialize save: %s", err), http.StatusInternalServerError)
			return
		}

		zstdWriter, err := zstd.NewWriter(nil)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to create zstd writer, %s", err), http.StatusInternalServerError)
			return
		}

		compressed := zstdWriter.EncodeAll(gobBuffer.Bytes(), nil)

		err = os.MkdirAll("userdata/"+hexUuid, 0755)
		if err != nil && !os.IsExist(err) {
			http.Error(w, 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)
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to write save file: %s", err), http.StatusInternalServerError)
			return
		}
	default:
		http.Error(w, "invalid data type", http.StatusBadRequest)
		return
	}

	w.WriteHeader(http.StatusOK)
}

// /savedata/delete - delete save data

func (s *Server) HandleSavedataDelete(w http.ResponseWriter, r *http.Request) {
	uuid, err := GetUuidFromRequest(r)
	if err != nil {
		http.Error(w, err.Error(), http.StatusBadRequest)
		return
	}

	err = db.UpdateAccountLastActivity(uuid)
	if err != nil {
		log.Print("failed to update account last activity")
	}

	hexUuid := hex.EncodeToString(uuid)

	switch r.URL.Query().Get("datatype") {
	case "0": // System
		err := os.Remove("userdata/" + hexUuid + "/system.pzs")
		if err != nil && !os.IsNotExist(err) {
			http.Error(w, fmt.Sprintf("failed to delete save file: %s", err), http.StatusInternalServerError)
			return
		}
	case "1": // Session
		slotId, err := strconv.Atoi(r.URL.Query().Get("slot"))
		if err != nil {
			http.Error(w, fmt.Sprintf("failed to convert slot id: %s", err), http.StatusBadRequest)
			return
		}

		if slotId < 0 || slotId >= sessionSlotCount {
			http.Error(w, fmt.Sprintf("slot id %d out of range", slotId), http.StatusBadRequest)
			return
		}

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

		err = os.Remove(fmt.Sprintf("userdata/%s/%s.pzs", hexUuid, fileName))
		if err != nil && !os.IsNotExist(err) {
			http.Error(w, fmt.Sprintf("failed to delete save file: %s", err), http.StatusInternalServerError)
			return
		}
	default:
		http.Error(w, "invalid data type", http.StatusBadRequest)
		return
	}

	w.WriteHeader(http.StatusOK)
}