SorceryCraft/src/main/java/com/thebrokenrail/sorcerycraft/mixin/MixinPlayerEntity.java

package com.thebrokenrail.sorcerycraft.mixin;

import com.thebrokenrail.sorcerycraft.spell.SpellPlayerEntity;
import com.thebrokenrail.sorcerycraft.spell.SpellTag;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.nbt.CompoundTag;
import net.minecraft.util.Identifier;
import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

import java.util.HashMap;
import java.util.Map;

@SuppressWarnings("unused")
@Mixin(PlayerEntity.class)
public class MixinPlayerEntity implements SpellPlayerEntity {
    private Map<Identifier, Integer> spells = new HashMap<>();

    @Inject(at = @At("HEAD"), method = "readCustomDataFromTag")
    public void readCustomDataFromTag(CompoundTag tag, CallbackInfo info) {
        spells = SpellTag.getSpells(tag);
    }

    @Inject(at = @At("HEAD"), method = "writeCustomDataToTag")
    public void writeCustomDataToTag(CompoundTag tag, CallbackInfo info) {
        tag.put(SpellTag.SPELL_TAG, SpellTag.createSpellsTag(spells));
    }

    @Override
    public void setSpells(Map<Identifier, Integer> spells) {
        this.spells = spells;
    }

    @Override
    public Map<Identifier, Integer> getSpells() {
        return spells;
    }
}