From df6e83c1e19fcd39bd99a25951210a1ff6fd4e7e Mon Sep 17 00:00:00 2001
From: MetallicGoat <ecazzam3@gmail.com>
Date: Wed, 18 Dec 2024 14:58:10 -0500
Subject: [PATCH] Remove sponge after particles config

---
 .../metallicgoat/tweaksaddon/config/MainConfig.java   |  1 +
 .../tweaksaddon/tweaks/cosmetic/SpongeParticles.java  | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/src/main/java/me/metallicgoat/tweaksaddon/config/MainConfig.java b/src/main/java/me/metallicgoat/tweaksaddon/config/MainConfig.java
index c654bf7..cde91d9 100644
--- a/src/main/java/me/metallicgoat/tweaksaddon/config/MainConfig.java
+++ b/src/main/java/me/metallicgoat/tweaksaddon/config/MainConfig.java
@@ -161,6 +161,7 @@ public class MainConfig {
       }
   )
   public static boolean sponge_particles_enabled = true;
+  @Config public static boolean sponge_particles_remove_sponge_after_complete = true;
 
   @Config(
       description = {
diff --git a/src/main/java/me/metallicgoat/tweaksaddon/tweaks/cosmetic/SpongeParticles.java b/src/main/java/me/metallicgoat/tweaksaddon/tweaks/cosmetic/SpongeParticles.java
index 001bfc3..c43f270 100644
--- a/src/main/java/me/metallicgoat/tweaksaddon/tweaks/cosmetic/SpongeParticles.java
+++ b/src/main/java/me/metallicgoat/tweaksaddon/tweaks/cosmetic/SpongeParticles.java
@@ -57,6 +57,17 @@ public void run() {
       radius++;
     }
 
+    @Override
+    public void cancel() {
+      if (MainConfig.sponge_particles_remove_sponge_after_complete
+          && block.getType().name().contains("SPONGE")) { // SPONGE or WET_SPONGE (also changes across versions)
+
+        block.setType(Material.AIR);
+      }
+
+      super.cancel();
+    }
+
     public List<Location> getParticles(Location start, int radius) {
       final List<Location> locations = new ArrayList<>();