Issue with ShieldBlockEvent handling and projectiles

[Braquen]

So I'm creating a shield enchantment that reacts only to projectiles, but I've run into a problem. I'm using the .isProjectile() method on damage source to see if event handler should continue or return, which works great for projectiles from a dispenser. However, for projectiles from a mob, the damageSource.getEntity() returns the MOB that shot the projectile, rather than the projectile itself. Any ideas on how to get around this?

[Choonster]

Use DamageSource#getDirectEntity to get the projectile entity.

[Braquen]

31 minutes ago, Choonster said:

Use DamageSource#getDirectEntity to get the projectile entity.

Thanks, I was curious what the difference was!