ToolTrackProvider

package net.railsofwar.row.track.capability;

import net.minecraft.nbt.NBTBase;
import net.minecraft.util.EnumFacing;
import net.minecraftforge.common.capabilities.Capability;
import net.minecraftforge.common.capabilities.CapabilityInject;
import net.minecraftforge.common.capabilities.ICapabilitySerializable;

public class ToolTrackProvider implements ICapabilitySerializable<NBTBase>{
    @CapabilityInject(CapabilityToolTrack.IToolTrackCapability.class)
    public static Capability<CapabilityToolTrack.IToolTrackCapability> CAP_TOOL_TRACK = null;
    public static CapabilityToolTrack.IToolTrackCapability capToolInstance = new CapabilityToolTrack.DefaultToolTrack();

    @Override
    public boolean hasCapability(Capability<?> capability, EnumFacing facing){
        return capability == CAP_TOOL_TRACK;
    }

    @Override
    public <T> T getCapability(Capability<T> capability, EnumFacing facing){
        return capability == CAP_TOOL_TRACK ? CAP_TOOL_TRACK.cast(capToolInstance) : null;
    }

    @Override
    public NBTBase serializeNBT(){
        return CAP_TOOL_TRACK.getStorage().writeNBT(CAP_TOOL_TRACK, capToolInstance, null);
    }

    @Override
    public void deserializeNBT(NBTBase nbt){
        CAP_TOOL_TRACK.getStorage().readNBT(CAP_TOOL_TRACK, capToolInstance, null, nbt);
    }
}