Setting an item's NBT tag after a period of time

[65_7a]

I'm creating an item that strikes the target with lightning. I want to grant the attacker with temporary immunity to lightning. The way I plan to do this is setting an NBT tag on the ItemStack when an entity is attacked and removing it after 20 ticks or after the player is struck by lightning. I need help putting some sort of timer after an entity is attacked to set the NBT tag to false.

Event Handler class:

@Mod.EventBusSubscriber(modid = Main.MODID, bus = Mod.EventBusSubscriber.Bus.FORGE)
public class EventHandler {
    @SubscribeEvent
    public static void onStruckByLightning(EntityStruckByLightningEvent e) {
        if (e.getEntity() instanceof PlayerEntity) {
            if (((PlayerEntity) e.getEntity()).isHolding(ItemInit.LIGHTNING_AXE.get())) {
                ItemStack stack = ((PlayerEntity) e.getEntity()).getMainHandItem();
                if (stack.hasTag() && stack.getTag().contains(LIGHTNING_IMMUNITY_TAG) && stack.getTag().getBoolean(LIGHTNING_IMMUNITY_TAG)) {
                    e.setCanceled(true);
                    stack.getTag().putBoolean(LIGHTNING_IMMUNITY_TAG, false);
                }
            }
        }
    }

    @SubscribeEvent
    public static void onAttackEntity(AttackEntityEvent e) {
        if (e.getTarget() instanceof LivingEntity) {
            PlayerEntity attacker = e.getPlayer();

            if (attacker.level.dimension() == World.OVERWORLD && attacker.isHolding(ItemInit.LIGHTNING_AXE.get())) {
                attacker.getMainHandItem().getCapability(CapabilityEnergy.ENERGY, null).ifPresent(energyStorage -> {
                    if (energyStorage.getEnergyStored() >= 500 && attacker.getAttackStrengthScale(0) >= 1.0F) {
                        CompoundNBT nbt = attacker.getMainHandItem().getTag();

                        if (nbt == null) {
                            nbt = new CompoundNBT();
                            attacker.getMainHandItem().setTag(nbt);
                        }

                        nbt.putBoolean(LIGHTNING_IMMUNITY_TAG, true); // I want to put some sort of timer here to eventually set this to false.

                        LightningBoltEntity lightningBoltEntity = new LightningBoltEntity(EntityType.LIGHTNING_BOLT, e.getTarget().level);
                        lightningBoltEntity.setPos(e.getTarget().getX(), e.getTarget().getY(), e.getTarget().getZ());
                        e.getTarget().level.addFreshEntity(lightningBoltEntity);
                        e.getTarget().setSecondsOnFire(2);

                        energyStorage.extractEnergy(500, false);

                        attacker.addEffect(new EffectInstance(Effects.FIRE_RESISTANCE, 120, 0, false, false));
                    }
                });
            }
        }
    }
}

 

Other code can be found on my GitHub repo: https://github.com/65-7a/EnergyBasedGear/tree/lightning-axe

Thanks for reading.

[Draco18s]

Store the world time of the time you want the condition to end.

When the entity is struck by lightning, check to see if that time is in the future, or the past.

[65_7a]

51 minutes ago, Draco18s said:

Store the world time of the time you want the condition to end.

When the entity is struck by lightning, check to see if that time is in the future, or the past.

Thanks for responding.

Is this the correct way to do it?

    @SubscribeEvent
    public static void onStruckByLightning(EntityStruckByLightningEvent e) {
        if (e.getEntity() instanceof PlayerEntity) {
            if (((PlayerEntity) e.getEntity()).isHolding(ItemInit.LIGHTNING_AXE.get())) {
                ItemStack stack = ((PlayerEntity) e.getEntity()).getMainHandItem();
                if (stack.hasTag() && stack.getTag().contains(LIGHTNING_IMMUNITY_TAG) && stack.getTag().getLong(LIGHTNING_IMMUNITY_TAG) > e.getEntity().level.getGameTime()) {
                    e.setCanceled(true);
                }
            }
        }
    }

    @SubscribeEvent
    public static void onAttackEntity(AttackEntityEvent e) {
        if (e.getTarget() instanceof LivingEntity) {
            PlayerEntity attacker = e.getPlayer();

            if (attacker.level.dimension() == World.OVERWORLD && attacker.isHolding(ItemInit.LIGHTNING_AXE.get())) {
                attacker.getMainHandItem().getCapability(CapabilityEnergy.ENERGY, null).ifPresent(energyStorage -> {
                    if (energyStorage.getEnergyStored() >= 500 && attacker.getAttackStrengthScale(0) >= 1.0F) {
                        CompoundNBT nbt = attacker.getMainHandItem().getOrCreateTag();
                        nbt.putLong(LIGHTNING_IMMUNITY_TAG, e.getTarget().level.getGameTime() + 20L);

                        LightningBoltEntity lightningBoltEntity = new LightningBoltEntity(EntityType.LIGHTNING_BOLT, e.getTarget().level);
                        lightningBoltEntity.setPos(e.getTarget().getX(), e.getTarget().getY(), e.getTarget().getZ());
                        e.getTarget().level.addFreshEntity(lightningBoltEntity);
                        e.getTarget().setSecondsOnFire(2);

                        energyStorage.extractEnergy(500, false);

                        attacker.addEffect(new EffectInstance(Effects.FIRE_RESISTANCE, 120, 0, false, false));
                    }
                });
            }
        }
    }

 

[Draco18s]

And to me.