public class CoolDownCapability
{
    @CapabilityInject(ICoolDown.class)
    public static Capability<ICoolDown> COOLDOWN_CAPABILITY = null;

    public static void register()
    {
        CapabilityManager.INSTANCE.register(ICoolDown.class, new Storage(), DefaultCoolDownCapability::new);
    }

    public static class Storage implements Capability.IStorage<ICoolDown>
    {
        @Nullable
        @Override
        public INBT writeNBT(Capability<ICoolDown> capability, ICoolDown instance, Direction side) {
            CompoundNBT tag = new CompoundNBT();
            tag.putInt("ticksIn", instance.getCoolDown());
            return tag;
        }

        @Override
        public void readNBT(Capability<ICoolDown> capability, ICoolDown instance, Direction side, INBT nbt) {
            int ticksIn = ((CompoundNBT)nbt).getInt("ticksIn");
            instance.setCoolDown(ticksIn);
        }
    }
}