Electroblob77
116 lines
4.7 KiB
Java
116 lines
4.7 KiB
Java
package electroblob.wizardry.spell;
|
|
|
|
import electroblob.wizardry.Wizardry;
|
|
import electroblob.wizardry.item.SpellActions;
|
|
import electroblob.wizardry.registry.Spells;
|
|
import electroblob.wizardry.util.EntityUtils;
|
|
import electroblob.wizardry.util.MagicDamage;
|
|
import electroblob.wizardry.util.MagicDamage.DamageType;
|
|
import electroblob.wizardry.util.SpellModifiers;
|
|
import net.minecraft.entity.Entity;
|
|
import net.minecraft.entity.EntityLivingBase;
|
|
import net.minecraft.entity.effect.EntityLightningBolt;
|
|
import net.minecraft.util.DamageSource;
|
|
import net.minecraft.util.EnumFacing;
|
|
import net.minecraft.util.math.BlockPos;
|
|
import net.minecraft.util.math.Vec3d;
|
|
import net.minecraft.world.World;
|
|
import net.minecraftforge.event.entity.EntityStruckByLightningEvent;
|
|
import net.minecraftforge.event.entity.living.LivingAttackEvent;
|
|
import net.minecraftforge.fml.common.Mod;
|
|
import net.minecraftforge.fml.common.eventhandler.EventPriority;
|
|
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
|
|
|
|
@Mod.EventBusSubscriber
|
|
public class LightningBolt extends SpellRay {
|
|
|
|
/** The NBT key used to store the UUID of the entity that summoned the lightning bolt. */
|
|
public static final String SUMMONER_NBT_KEY = "summoner";
|
|
/** The NBT key used to store the damage modifier for the lightning bolt. */
|
|
public static final String DAMAGE_MODIFIER_NBT_KEY = "damageModifier";
|
|
/** An NBT key used to cancel vanilla's implementation of lightning damage; since there's no way of extracting the
|
|
* lightning bolt entity from {@code LivingAttackEvent} this has to be done using NBT. */
|
|
public static final String IMMUNE_TO_LIGHTNING_NBT_KEY = "immuneToLightning";
|
|
|
|
public LightningBolt(){
|
|
super("lightning_bolt", SpellActions.POINT, false);
|
|
this.ignoreLivingEntities(true);
|
|
addProperties(DAMAGE);
|
|
}
|
|
|
|
@Override public boolean requiresPacket(){ return false; }
|
|
|
|
@Override
|
|
protected boolean onEntityHit(World world, Entity target, Vec3d hit, EntityLivingBase caster, Vec3d origin, int ticksInUse, SpellModifiers modifiers){
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
protected boolean onBlockHit(World world, BlockPos pos, EnumFacing side, Vec3d hit, EntityLivingBase caster, Vec3d origin, int ticksInUse, SpellModifiers modifiers){
|
|
|
|
if(world.canBlockSeeSky(pos.up())){
|
|
|
|
if(!world.isRemote){
|
|
// Temporarily disable the fire tick gamerule if player block damage is disabled
|
|
// Bit of a hack but it works fine!
|
|
boolean doFireTick = world.getGameRules().getBoolean("doFireTick");
|
|
if(doFireTick && !Wizardry.settings.playerBlockDamage) world.getGameRules().setOrCreateGameRule("doFireTick", "false");
|
|
|
|
EntityLightningBolt lightning = new EntityLightningBolt(world, pos.getX(), pos.getY(), pos.getZ(), false);
|
|
if(caster != null) lightning.getEntityData().setUniqueId(SUMMONER_NBT_KEY, caster.getUniqueID());
|
|
lightning.getEntityData().setFloat(DAMAGE_MODIFIER_NBT_KEY, modifiers.get(SpellModifiers.POTENCY));
|
|
world.addWeatherEffect(lightning);
|
|
|
|
// Reset doFireTick to true if it was true before
|
|
if(doFireTick && !Wizardry.settings.playerBlockDamage) world.getGameRules().setOrCreateGameRule("doFireTick", "true");
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
protected boolean onMiss(World world, EntityLivingBase caster, Vec3d origin, Vec3d direction, int ticksInUse, SpellModifiers modifiers){
|
|
return false;
|
|
}
|
|
|
|
@SubscribeEvent(priority = EventPriority.LOW)
|
|
public static void onEntityStruckByLightningEvent(EntityStruckByLightningEvent event){
|
|
|
|
float damageModifier = event.getLightning().getEntityData().getFloat(DAMAGE_MODIFIER_NBT_KEY);
|
|
|
|
Entity summoner = null;
|
|
|
|
if(event.getLightning().getEntityData().hasUniqueId(SUMMONER_NBT_KEY)){
|
|
|
|
summoner = EntityUtils.getEntityByUUID(event.getLightning().world,
|
|
event.getLightning().getEntityData().getUniqueId(SUMMONER_NBT_KEY));
|
|
|
|
if(!(summoner instanceof EntityLivingBase)) summoner = null;
|
|
}
|
|
|
|
if(damageModifier > 0 || summoner != null){
|
|
|
|
DamageSource source = summoner == null ? DamageSource.LIGHTNING_BOLT
|
|
: MagicDamage.causeIndirectMagicDamage(event.getLightning(), summoner, DamageType.SHOCK);
|
|
float damage = Spells.lightning_bolt.getProperty(DAMAGE).floatValue() * damageModifier;
|
|
|
|
// Don't need DamageSafetyChecker here because this isn't an attack event
|
|
EntityUtils.attackEntityWithoutKnockback(event.getEntity(), source, damage);
|
|
event.getEntity().getEntityData().setBoolean(IMMUNE_TO_LIGHTNING_NBT_KEY, true);
|
|
}
|
|
}
|
|
|
|
@SubscribeEvent(priority = EventPriority.HIGHEST)
|
|
public static void onLivingAttackEvent(LivingAttackEvent event){
|
|
if(event.getEntity().getEntityData().hasKey(IMMUNE_TO_LIGHTNING_NBT_KEY)
|
|
&& event.getSource() == DamageSource.LIGHTNING_BOLT){
|
|
event.setCanceled(true);
|
|
event.getEntity().getEntityData().removeTag(IMMUNE_TO_LIGHTNING_NBT_KEY);
|
|
}
|
|
}
|
|
|
|
}
|