package savedata import ( "bytes" "encoding/gob" "fmt" "log" "os" "strconv" "github.com/klauspost/compress/zstd" "github.com/pagefaultgames/pokerogue-server/db" "github.com/pagefaultgames/pokerogue-server/defs" ) var zstdEncoder, _ = zstd.NewWriter(nil) // /savedata/update - update save data func Update(uuid []byte, slot int, save any) error { err := db.UpdateAccountLastActivity(uuid) if err != nil { log.Print("failed to update account last activity") } // ideally should have been done at account creation err = os.MkdirAll(fmt.Sprintf("userdata/%x", uuid), 0755) if err != nil && !os.IsExist(err) { return fmt.Errorf(fmt.Sprintf("failed to create userdata folder: %s", err)) } var filename string switch save := save.(type) { case defs.SystemSaveData: // System if save.TrainerId == 0 && save.SecretId == 0 { return fmt.Errorf("invalid system data") } if save.GameVersion != "1.0.4" { return fmt.Errorf("client version out of date") } if save.VoucherCounts["0"] > 300 || save.VoucherCounts["1"] > 150 || save.VoucherCounts["2"] > 100 || save.VoucherCounts["3"] > 10 { db.UpdateAccountBanned(uuid, true) } err = db.UpdateAccountStats(uuid, save.GameStats, save.VoucherCounts) if err != nil { return fmt.Errorf("failed to update account stats: %s", err) } filename = "system" db.DeleteClaimedAccountCompensations(uuid) case defs.SessionSaveData: // Session if slot < 0 || slot >= defs.SessionSlotCount { return fmt.Errorf("slot id %d out of range", slot) } filename = "session" if slot != 0 { filename += strconv.Itoa(slot) } default: return fmt.Errorf("invalid data type") } var buf bytes.Buffer err = gob.NewEncoder(&buf).Encode(save) if err != nil { return fmt.Errorf("failed to serialize save: %s", err) } if buf.Len() == 0 { return fmt.Errorf("tried to write empty save file") } err = os.WriteFile(fmt.Sprintf("userdata/%x/%s.pzs", uuid, filename), zstdEncoder.EncodeAll(buf.Bytes(), nil), 0644) if err != nil { return fmt.Errorf("failed to write save to disk: %s", err) } return nil }