fix folia support

Author: MagicTeaMC

src/main/java/org/milkteamc/autotreechop/AutoTreeChop.java

@@ -1,5 +1,6 @@
 package org.milkteamc.autotreechop;
 
+import com.github.Anon8281.universalScheduler.UniversalScheduler;
 import java.io.File;
 import java.util.HashSet;
 import java.util.Locale;
@@ -158,8 +159,7 @@ public void onEnable() {
                 config.getPassword());
 
         saveTask = new PlayerDataSaveTask(this, SAVE_THRESHOLD);
-        saveTask.runTaskTimerAsynchronously(this, SAVE_INTERVAL, SAVE_INTERVAL);
-
+        UniversalScheduler.getScheduler(this).runTaskTimerAsynchronously(saveTask, SAVE_INTERVAL, SAVE_INTERVAL);
         autoTreeChopAPI = new AutoTreeChopAPI(this);
         playerConfigs = new ConcurrentHashMap<>();
         initializeHooks();
@@ -268,7 +268,11 @@ public void onDisable() {
         getLogger().info("Saving all player data before shutdown...");
 
         if (saveTask != null) {
-            saveTask.cancel();
+            try {
+                saveTask.cancel();
+            } catch (IllegalStateException ignored) {
+                // Task was never scheduled or already cancelled (e.g. Folia shutdown)
+            }
         }
 
         for (Map.Entry<UUID, PlayerConfig> entry : playerConfigs.entrySet()) {

src/main/java/org/milkteamc/autotreechop/ModrinthUpdateChecker.java

@@ -177,9 +177,8 @@ public ModrinthUpdateChecker startPeriodicCheck() {
         checkNow();
 
         long intervalTicks = checkIntervalHours * 60 * 60 * 20L; // hours to ticks
-        plugin.getServer()
-                .getScheduler()
-                .runTaskTimerAsynchronously(plugin, this::performCheck, intervalTicks, intervalTicks);
+        UniversalScheduler.getScheduler(plugin)
+                .runTaskTimerAsynchronously(this::performCheck, intervalTicks, intervalTicks);
 
         return this;
     }
@@ -322,9 +321,8 @@ public void onPlayerJoin(PlayerJoinEvent event) {
         }
 
         if (shouldNotify) {
-            plugin.getServer()
-                    .getScheduler()
-                    .runTaskLater(plugin, () -> printCheckResultToPlayer(player, false), 40L); // 2s
+            UniversalScheduler.getScheduler(plugin)
+                    .runTaskLater(() -> printCheckResultToPlayer(player, false), 40L); // 2s
         }
     }
 

src/main/java/org/milkteamc/autotreechop/events/BlockBreakListener.java

@@ -132,7 +132,7 @@ public void onBlockBreak(BlockBreakEvent event) {
         Location frozenLocation = location.clone();
         ItemStack frozenTool = tool.clone();
 
-        plugin.getServer().getScheduler().runTaskAsynchronously(plugin, () -> {
+        scheduler.runTaskAsync(() -> {
             boolean hasLeaves = hasNearbyLeaves(block, config, snapshots);
 
             // Return to the main/region thread to act on the result.