Implement wearing_potion

Author: IzzelAliz

src/api/java/io/izzel/mesmerize/api/DefaultStats.java

@@ -4,6 +4,7 @@
 import io.izzel.mesmerize.api.data.NumberValue;
 import io.izzel.mesmerize.api.data.RangeNumberValue;
 import io.izzel.mesmerize.api.data.StatsNumber;
+import io.izzel.mesmerize.api.data.complex.PotionValue;
 import io.izzel.mesmerize.api.data.complex.StatsSetValue;
 import io.izzel.mesmerize.api.data.UUIDValue;
 import io.izzel.mesmerize.api.data.stats.PermissionStats;
@@ -54,6 +55,7 @@ public final class DefaultStats {
     public static final Stats<List<StatsSetValue>> STATS_SET = StatsSetValue.STATS;
     public static final Stats<Map<String, StatsValue<?>>> SLOT = SlotStats.STATS;
     public static final Stats<List<StatsSetValue>> HIDDEN_SET = StatsSetValue.HIDDEN;
+    public static final Stats<List<PotionValue>> WEARING_POTION = PotionValue.stats("wearing_potion");
 
     @SuppressWarnings("deprecation")
     private static NamespacedKey key(String id) {

src/api/java/io/izzel/mesmerize/api/data/complex/PotionValue.java

@@ -1,13 +1,20 @@
 package io.izzel.mesmerize.api.data.complex;
 
 import com.google.common.base.Preconditions;
+import io.izzel.mesmerize.api.Stats;
+import io.izzel.mesmerize.api.data.MultiValue;
+import io.izzel.mesmerize.api.service.ElementFactory;
 import io.izzel.mesmerize.api.util.Coerce;
 import io.izzel.mesmerize.api.visitor.ValueVisitor;
 import io.izzel.mesmerize.api.visitor.VisitMode;
 import io.izzel.mesmerize.api.visitor.impl.AbstractValue;
+import org.bukkit.NamespacedKey;
 import org.bukkit.potion.PotionEffect;
 import org.bukkit.potion.PotionEffectType;
 
+import java.time.Duration;
+import java.util.List;
+
 public class PotionValue extends AbstractValue<PotionEffect> {
 
     private String expr;
@@ -43,4 +50,24 @@ public void visitEnd() {
         boolean icon = split.length <= 5 || Coerce.toBoolean(split[5]);
         this.potionEffect = new PotionEffect(effectType, duration, amplifier, ambient, particle, icon);
     }
+
+    public static Stats<List<PotionValue>> stats(String key) {
+        return Stats.builder().key(new NamespacedKey("mesmerize", key))
+            .supplying(
+                MultiValue.builder()
+                    .supplying(PotionValue::new)
+                    .allowSingleNonListValue()
+                    .buildSupplier()
+            )
+            .merging(MultiValue.concatMerger())
+            .displaying((value, pane) -> {
+                ElementFactory factory = ElementFactory.instance();
+                for (PotionValue potionValue : value.get()) {
+                    pane.addElement(factory.createLocaleElement("stats." + key, potionValue.get().getAmplifier(),
+                        factory.createPotionElement(potionValue.get().getType()),
+                        factory.createDurationElement(Duration.ofMillis(potionValue.get().getDuration() * 50))));
+                }
+            })
+            .build();
+    }
 }

src/api/java/io/izzel/mesmerize/api/service/ElementFactory.java

@@ -5,6 +5,7 @@
 import io.izzel.mesmerize.api.display.Element;
 import io.izzel.mesmerize.api.visitor.StatsHolder;
 import org.bukkit.inventory.ItemStack;
+import org.bukkit.potion.PotionEffectType;
 import org.jetbrains.annotations.NotNull;
 
 import java.time.Duration;
@@ -32,6 +33,8 @@ public interface ElementFactory {
 
     Element createDateElement(Instant instant);
 
+    Element createPotionElement(PotionEffectType potionEffectType);
+
     static ElementFactory instance() {
         return StatsService.instance().getElementFactory();
     }

src/api/java/io/izzel/mesmerize/api/util/Quat.java

@@ -7,6 +7,7 @@
 import io.izzel.mesmerize.impl.event.AttributeListener;
 import io.izzel.mesmerize.impl.event.CombatListener;
 import io.izzel.mesmerize.impl.event.EntityStatsCacheListener;
+import io.izzel.mesmerize.impl.event.PotionApplyTask;
 import io.izzel.mesmerize.impl.event.ProjectileListener;
 import io.izzel.mesmerize.impl.service.SimpleStatsService;
 import io.izzel.mesmerize.impl.util.Updater;
@@ -70,6 +71,7 @@ public void onActivated() {
             LOGGER.info(TLocale.asString("load.paper.absent"));
             this.getServer().getPluginManager().registerEvents(new EntityStatsCacheListener.Spigot(), this);
         }
+        new PotionApplyTask().runTaskTimer(this, 0, ConfigSpec.spec().general().potionApplyInterval());
         Updater.start();
         LOGGER.info(TLocale.asString("general.load", (System.currentTimeMillis() - begin) / 1000D));
     }

src/main/java/io/izzel/mesmerize/impl/Mesmerize.java

@@ -10,6 +10,7 @@ public class GeneralSpec {
     private double defaultAttackSpeed = 4.0D;
     private double defaultMoveSpeed = 0.1D;
     private double defaultFlySpeed = 0.2D;
+    private long potionApplyInterval = 60L;
 
     public boolean debug() {
         return debug;
@@ -74,4 +75,12 @@ public double defaultFlySpeed() {
     public void setDefaultFlySpeed(double defaultFlySpeed) {
         this.defaultFlySpeed = defaultFlySpeed;
     }
+
+    public long potionApplyInterval() {
+        return potionApplyInterval;
+    }
+
+    public void setPotionApplyInterval(long potionApplyInterval) {
+        this.potionApplyInterval = potionApplyInterval;
+    }
 }

src/main/java/io/izzel/mesmerize/impl/config/spec/GeneralSpec.java

@@ -0,0 +1,33 @@
+package io.izzel.mesmerize.impl.event;
+
+import io.izzel.mesmerize.api.DefaultStats;
+import io.izzel.mesmerize.api.data.complex.PotionValue;
+import io.izzel.mesmerize.api.visitor.util.StatsSet;
+import io.izzel.mesmerize.impl.config.spec.ConfigSpec;
+import org.bukkit.Bukkit;
+import org.bukkit.entity.Player;
+import org.bukkit.potion.PotionEffect;
+import org.bukkit.scheduler.BukkitRunnable;
+
+import java.util.List;
+import java.util.Optional;
+
+public class PotionApplyTask extends BukkitRunnable {
+
+    @Override
+    public void run() {
+        for (Player player : Bukkit.getOnlinePlayers()) {
+            StatsSet set = StatsSet.of(player);
+            Optional<List<PotionValue>> optional = DefaultStats.WEARING_POTION.tryApply(set, null);
+            if (optional.isPresent()) {
+                for (PotionValue potionValue : optional.get()) {
+                    PotionEffect potionEffect = player.getPotionEffect(potionValue.get().getType());
+                    if (potionEffect != null && potionEffect.getDuration() > ConfigSpec.spec().general().potionApplyInterval()) {
+                        continue;
+                    }
+                    potionValue.get().apply(player);
+                }
+            }
+        }
+    }
+}

src/main/java/io/izzel/mesmerize/impl/event/PotionApplyTask.java

@@ -1,6 +1,6 @@
 package io.izzel.mesmerize.impl.event;
 
-import io.izzel.mesmerize.api.util.Quat;
+import io.izzel.taboolib.util.Quat;
 import org.bukkit.entity.LivingEntity;
 import org.bukkit.entity.Projectile;
 import org.bukkit.scheduler.BukkitRunnable;

src/main/java/io/izzel/mesmerize/impl/event/TracingTargetTask.java

@@ -11,9 +11,11 @@
 import io.izzel.mesmerize.impl.config.spec.ConfigSpec;
 import io.izzel.mesmerize.impl.element.DisplayVisitor;
 import io.izzel.mesmerize.impl.element.SimplePane;
+import io.izzel.taboolib.module.i18n.I18n;
 import io.izzel.taboolib.module.locale.TLocale;
 import org.bukkit.inventory.ItemStack;
 import org.bukkit.inventory.meta.ItemMeta;
+import org.bukkit.potion.PotionEffectType;
 import org.jetbrains.annotations.NotNull;
 
 import java.text.DecimalFormat;
@@ -90,4 +92,9 @@ public Element createDateElement(Instant instant) {
         // todo
         return Element.of(instant.toString());
     }
+
+    @Override
+    public Element createPotionElement(PotionEffectType potionEffectType) {
+        return Element.of(I18n.get().getName(potionEffectType));
+    }
 }