ActuallyAdditions/src/main/java/de/ellpeck/actuallyadditions/common/data/PlayerData.java

package de.ellpeck.actuallyadditions.common.data;

import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.ConcurrentHashMap;

import de.ellpeck.actuallyadditions.api.booklet.IBookletPage;
import de.ellpeck.actuallyadditions.booklet.gui.GuiBooklet;
import de.ellpeck.actuallyadditions.booklet.misc.BookletUtils;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.nbt.NBTTagString;
import net.minecraftforge.fml.relauncher.Side;
import net.minecraftforge.fml.relauncher.SideOnly;

public final class PlayerData {

    public static PlayerSave getDataFromPlayer(EntityPlayer player) {
        WorldData worldData = WorldData.get(player.getEntityWorld());
        ConcurrentHashMap<UUID, PlayerSave> data = worldData.playerSaveData;
        UUID id = player.getUniqueID();

        if (data.containsKey(id)) {
            PlayerSave save = data.get(id);
            if (save != null && save.id != null && save.id.equals(id)) { return save; }
        }

        //Add Data if none is existant
        PlayerSave save = new PlayerSave(id);
        data.put(id, save);
        worldData.markDirty();
        return save;
    }

    public static class PlayerSave {

        public UUID id;

        public boolean bookGottenAlready;
        public boolean didBookTutorial;
        public boolean hasBatWings;
        public boolean shouldDisableBatWings;
        public int batWingsFlyTime;

        public IBookletPage[] bookmarks = new IBookletPage[12];
        public List<String> completedTrials = new ArrayList<>();

        @SideOnly(Side.CLIENT)
        public GuiBooklet lastOpenBooklet;

        public PlayerSave(UUID id) {
            this.id = id;
        }

        public void readFromNBT(NBTTagCompound compound, boolean savingToFile) {
            this.bookGottenAlready = compound.getBoolean("BookGotten");
            this.didBookTutorial = compound.getBoolean("DidTutorial");

            this.hasBatWings = compound.getBoolean("HasBatWings");
            this.batWingsFlyTime = compound.getInteger("BatWingsFlyTime");

            NBTTagList bookmarks = compound.getTagList("Bookmarks", 8);
            this.loadBookmarks(bookmarks);

            NBTTagList trials = compound.getTagList("Trials", 8);
            this.loadTrials(trials);

            if (!savingToFile) {
                this.shouldDisableBatWings = compound.getBoolean("ShouldDisableWings");
            }
        }

        public void writeToNBT(NBTTagCompound compound, boolean savingToFile) {
            compound.setBoolean("BookGotten", this.bookGottenAlready);
            compound.setBoolean("DidTutorial", this.didBookTutorial);

            compound.setBoolean("HasBatWings", this.hasBatWings);
            compound.setInteger("BatWingsFlyTime", this.batWingsFlyTime);

            compound.setTag("Bookmarks", this.saveBookmarks());
            compound.setTag("Trials", this.saveTrials());

            if (!savingToFile) {
                compound.setBoolean("ShouldDisableWings", this.shouldDisableBatWings);
            }
        }

        public NBTTagList saveBookmarks() {
            NBTTagList bookmarks = new NBTTagList();
            for (IBookletPage bookmark : this.bookmarks) {
                bookmarks.appendTag(new NBTTagString(bookmark == null ? "" : bookmark.getIdentifier()));
            }
            return bookmarks;
        }

        public void loadBookmarks(NBTTagList bookmarks) {
            for (int i = 0; i < bookmarks.tagCount(); i++) {
                String strg = bookmarks.getStringTagAt(i);
                if (strg != null && !strg.isEmpty()) {
                    IBookletPage page = BookletUtils.getBookletPageById(strg);
                    this.bookmarks[i] = page;
                } else {
                    this.bookmarks[i] = null;
                }
            }
        }

        public NBTTagList saveTrials() {
            NBTTagList trials = new NBTTagList();
            for (String trial : this.completedTrials) {
                trials.appendTag(new NBTTagString(trial));
            }
            return trials;
        }

        public void loadTrials(NBTTagList trials) {
            this.completedTrials.clear();

            for (int i = 0; i < trials.tagCount(); i++) {
                String strg = trials.getStringTagAt(i);
                if (strg != null && !strg.isEmpty()) {
                    this.completedTrials.add(strg);
                }
            }
        }
    }

}