|
|
|
package api
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"encoding/gob"
|
|
|
|
"encoding/hex"
|
|
|
|
"encoding/json"
|
|
|
|
"fmt"
|
|
|
|
"log"
|
|
|
|
"net/http"
|
|
|
|
"os"
|
|
|
|
"strconv"
|
|
|
|
|
|
|
|
"github.com/Flashfyre/pokerogue-server/db"
|
|
|
|
"github.com/Flashfyre/pokerogue-server/defs"
|
|
|
|
"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
|
|
|
|
}
|
|
|
|
|
|
|
|
switch r.URL.Query().Get("datatype") {
|
|
|
|
case "0": // System
|
|
|
|
system, err := ReadSystemSaveData(uuid)
|
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), 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
|
|
|
|
}
|
|
|
|
|
|
|
|
session, err := ReadSessionSaveData(uuid, slotId)
|
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), 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 defs.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 defs.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)
|
|
|
|
}
|
|
|
|
|
|
|
|
type SavedataClearResponse struct {
|
|
|
|
Success bool `json:"success"`
|
|
|
|
}
|
|
|
|
|
|
|
|
// /savedata/clear - mark session save data as cleared and delete
|
|
|
|
|
|
|
|
func (s *Server) HandleSavedataClear(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")
|
|
|
|
}
|
|
|
|
|
|
|
|
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
|
|
|
|
}
|
|
|
|
|
|
|
|
var session defs.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
|
|
|
|
}
|
|
|
|
|
|
|
|
sessionCompleted := ValidateSessionCompleted(session)
|
|
|
|
newCompletion := false
|
|
|
|
|
|
|
|
if session.GameMode == 3 {
|
|
|
|
waveCompleted := session.WaveIndex
|
|
|
|
if !sessionCompleted {
|
|
|
|
waveCompleted--
|
|
|
|
}
|
|
|
|
err = db.AddOrUpdateAccountDailyRun(uuid, session.Score, waveCompleted)
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("failed to add or update daily run record: %s", err.Error())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if sessionCompleted {
|
|
|
|
newCompletion, err = db.TryAddSeedCompletion(uuid, session.Seed, int(session.GameMode))
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("failed to mark seed as completed: %s", err.Error())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
response, err := json.Marshal(SavedataClearResponse{Success: newCompletion})
|
|
|
|
if err != nil {
|
|
|
|
http.Error(w, fmt.Sprintf("failed to marshal response json: %s", err), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
fileName := "session"
|
|
|
|
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) {
|
|
|
|
http.Error(w, fmt.Sprintf("failed to delete save file: %s", err), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
w.Write(response)
|
|
|
|
}
|