Entity Potion Effect Event

src/main/java/ch/njol/skript/events/EvtEntityPotion.java

@@ -45,16 +45,17 @@ public class EvtEntityPotion extends SkriptEvent {
 
 	@SuppressWarnings("unchecked")
 	private Expression<PotionEffectType> potionEffects;
+	private Expression<EntityPotionEffectEvent.Cause> cause;
 
 	@Override
 	public boolean init(Literal<?>[] args, int matchedPattern, ParseResult parseResult) {
 		if (args.length > 0) {
 			potionEffects = (Expression<PotionEffectType>) args[0];
+			cause = (Expression<EntityPotionEffectEvent.Cause>) args[1];
 		}
 		return true;
 	}
 
-
 	@Override
 	public boolean check(Event event) {
 		if (!(event instanceof EntityPotionEffectEvent)) {
@@ -69,14 +70,15 @@ public boolean check(Event event) {
 		PotionEffectType effectType = potionEvent.getNewEffect().getType();
 		for (PotionEffectType potionEffectType : potionEffects.getArray(event)) {
 			if (potionEffectType.equals(effectType)) {
-				return true;
+				if (cause == null || cause.getSingle(event).equals(potionEvent.getCause())) {
+					return true;
+				}
 			}
 		}
 
 		return false;
 	}
 
-
 	@Override
 	public String toString(@Nullable Event event, boolean debug) {
 		return "on entity potion effect modification";