Made relevant changes (Blame drunk me)

src/main/java/dev/plex/extras/command/OrbitCommand.java

@@ -14,11 +14,16 @@ import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
 import java.util.List;
+import java.util.ArrayList;
+import java.util.UUID;
 
 @CommandParameters(name = "orbit", description = "Accelerates the player at a super fast rate", usage = "/<command> <target> [<<power> | stop>]")
 @CommandPermissions(permission = "plex.tfmextras.orbit")
 public class OrbitCommand extends PlexCommand
 {
+
+    private static final List<UUID> isOrbited = new ArrayList<>();
+
     @Override
     protected Component execute(@NotNull CommandSender sender, @Nullable Player playerSender, String[] args)
     {
@@ -63,9 +68,15 @@ public class OrbitCommand extends PlexCommand
     private void startOrbiting(Player player, int strength) {
         player.setGameMode(org.bukkit.GameMode.SURVIVAL);
         player.addPotionEffect(new PotionEffect(PotionEffectType.LEVITATION, Integer.MAX_VALUE, strength, false, false));
+        isOrbited.add(player.getUniqueId());
     }
 
     private void stopOrbiting(Player player) {
         player.removePotionEffect(PotionEffectType.LEVITATION);
+        isOrbited.remove(player.getUniqueId());
+    }
+
+    public static boolean isPlayerOrbited(UUID playerId) {
+        return isOrbited.contains(playerId);
     }
 }

src/main/java/dev/plex/extras/listener/OrbitEffectListener.java

@@ -0,0 +1,31 @@
+package dev.plex.extras.listener;
+
+import dev.plex.Plex;
+import dev.plex.extras.command.OrbitCommand;
+import org.bukkit.Bukkit;
+import org.bukkit.entity.Player;
+import org.bukkit.event.EventHandler;
+import org.bukkit.event.EventPriority;
+import dev.plex.listener.PlexListener;
+import org.bukkit.event.entity.EntityPotionEffectEvent;
+import org.bukkit.potion.PotionEffect;
+import org.bukkit.potion.PotionEffectType;
+
+public class OrbitEffectListener extends PlexListener
+{
+    @EventHandler(priority = EventPriority.LOWEST)
+    public void onPotionEffectRemove(EntityPotionEffectEvent event)
+    {
+        if (event.getEntity() instanceof Player player)
+        {
+            if ((event.getAction() == EntityPotionEffectEvent.Action.CLEARED || event.getAction() == EntityPotionEffectEvent.Action.REMOVED)
+                    && event.getModifiedType() == PotionEffectType.LEVITATION) {
+                if (OrbitCommand.isPlayerOrbited(player.getUniqueId()))
+                {
+                    Bukkit.getScheduler().runTaskLater(Plex.get(), () ->
+                            player.addPotionEffect(new PotionEffect(PotionEffectType.LEVITATION, Integer.MAX_VALUE, 100, false, false)), 2);
+                }
+            }
+        }
+    }
+}