From fb91ff415cdf7c20a114b75bcac9caa6c36adb2e Mon Sep 17 00:00:00 2001
From: Phillip Glanz <p.glanz@madfix.me>
Date: Sun, 2 Apr 2023 17:34:08 +0200
Subject: [PATCH] Port taskmanager for folia

---
 .../bukkit/util/FoliaTaskManager.java         | 24 ++++++++++++++-----
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/worldedit-bukkit/src/main/java/com/fastasyncworldedit/bukkit/util/FoliaTaskManager.java b/worldedit-bukkit/src/main/java/com/fastasyncworldedit/bukkit/util/FoliaTaskManager.java
index f92859349..f1d96ca1d 100644
--- a/worldedit-bukkit/src/main/java/com/fastasyncworldedit/bukkit/util/FoliaTaskManager.java
+++ b/worldedit-bukkit/src/main/java/com/fastasyncworldedit/bukkit/util/FoliaTaskManager.java
@@ -34,23 +34,35 @@ public class FoliaTaskManager extends TaskManager {
 
     @Override
     public int repeat(@NotNull final Runnable runnable, final int interval) {
-        return fail();
+        Bukkit.getGlobalRegionScheduler().runAtFixedRate(
+                WorldEditPlugin.getInstance(),
+                scheduledTask -> runnable.run(),
+                1,
+                interval
+        );
+        return 0;
     }
 
     @Override
     public int repeatAsync(@NotNull final Runnable runnable, final int interval) {
-        backgroundExecutor.scheduleAtFixedRate(runnable, 0, ticksToMs(interval), TimeUnit.MILLISECONDS);
+        Bukkit.getAsyncScheduler().runAtFixedRate(
+                WorldEditPlugin.getInstance(),
+                scheduledTask -> runnable.run(),
+                0,
+                ticksToMs(interval),
+                TimeUnit.MILLISECONDS
+        );
         return idCounter.getAndIncrement();
     }
 
     @Override
     public void async(@NotNull final Runnable runnable) {
-        backgroundExecutor.submit(runnable);
+        Bukkit.getAsyncScheduler().runNow(WorldEditPlugin.getInstance(), (s) -> runnable.run());
     }
 
     @Override
     public void task(@NotNull final Runnable runnable) {
-        fail();
+        Bukkit.getGlobalRegionScheduler().execute(WorldEditPlugin.getInstance(), runnable);
     }
 
     @Override
@@ -60,7 +72,7 @@ public class FoliaTaskManager extends TaskManager {
 
     @Override
     public void later(@NotNull final Runnable runnable, final int delay) {
-        fail();
+        Bukkit.getGlobalRegionScheduler().runDelayed(WorldEditPlugin.getInstance(), scheduledTask -> runnable.run(), delay);
     }
 
     @Override
@@ -142,7 +154,7 @@ public class FoliaTaskManager extends TaskManager {
 
             MethodHandle executeForPlayer;
             try {
-                Class<?> regionisedSchedulerClass = Class.forName("io.papermc.paper.threadedregions.scheduler.RegionisedScheduler");
+                Class<?> regionisedSchedulerClass = Class.forName("io.papermc.paper.threadedregions.scheduler.FoliaRegionScheduler");
                 final Method method = Bukkit.class.getDeclaredMethod("getRegionScheduler");
                 executeForLocation = lookup.findVirtual(
                         regionisedSchedulerClass,