Add customizable post step actions

Author: HydrolienF

build.gradle.kts

@@ -11,7 +11,7 @@ plugins {
 }
 
 group = "fr.formiko.mc.underilla"
-version = "2.1.1"
+version = "2.1.2"
 description="Generate vanilla cave in custom world."
 val mainMinecraftVersion = "1.21.4"
 val supportedMinecraftVersions = "1.21.3 - 1.21.4"

src/main/java/fr/formiko/mc/underilla/paper/Underilla.java

@@ -16,6 +16,7 @@
 import fr.formiko.mc.underilla.paper.listener.WorldListener;
 import fr.formiko.mc.underilla.paper.preparing.ServerSetup;
 import fr.formiko.mc.underilla.paper.selector.Selector;
+import java.util.EnumMap;
 import java.util.Map;
 import java.util.function.Consumer;
 import java.util.function.Function;
@@ -52,6 +53,7 @@ public final class Underilla extends JavaPlugin {
     private Function<org.bukkit.block.Biome, org.bukkit.block.Biome> endBiomeTransformer;
     private Consumer<Block> endBlockTransformer;
     private Consumer<Entity> endEntityTransformer;
+    private Map<StringKeys, Runnable> endTaskActions = new EnumMap<>(StringKeys.class);
 
     @Override
     public ChunkGenerator getDefaultWorldGenerator(String worldName, String id) {
@@ -211,6 +213,10 @@ public boolean allStepsDone() {
     }
     public void validateTask(StringKeys taskKey, boolean done) {
         getUnderillaConfig().saveNewValue(taskKey, done ? DONE : FAILED);
+        if (done && endTaskActions.containsKey(taskKey)) {
+            Underilla.info("Running post action for task " + taskKey);
+            endTaskActions.get(taskKey).run();
+        }
         runNextStepsAfterWorldInit();
     }
     public void validateInitServerTask(StringKeys taskKey, boolean done) {
@@ -232,6 +238,7 @@ public void setEndBiomeTransformer(Function<org.bukkit.block.Biome, org.bukkit.b
     public Consumer<Entity> getEndEntityTransformer() { return endEntityTransformer; }
     public void setEndEntityTransformer(Consumer<Entity> endEntityTransformer) { this.endEntityTransformer = endEntityTransformer; }
     public boolean hasEndEntityTransformer() { return endEntityTransformer != null; }
+    public void setPostTaskAction(Runnable action, StringKeys taskKey) { endTaskActions.put(taskKey, action); }
 
     // run tasks ------------------------------------------------------------------------------------------------------
     private void runChunky(boolean restart) {