Feat: added close consumer

Author: danorris709

commons/src/main/java/com/envyful/api/gui/Gui.java

@@ -4,6 +4,8 @@
 import com.envyful.api.player.EnvyPlayer;
 import com.envyful.api.player.PlayerManager;
 
+import java.util.function.Consumer;
+
 /**
  *
  * An interface representing chest GUIs for the platform specific implementation
@@ -63,6 +65,15 @@ interface Builder {
          */
         Builder setPlayerManager(PlayerManager<?, ?> playerManager);
 
+        /**
+         *
+         * Sets the consumer for when the GUI is closed.
+         *
+         * @param consumer The close consumer
+         * @return The builder
+         */
+        Builder setCloseConsumer(Consumer<EnvyPlayer<?>> consumer);
+
         /**
          *
          * Builds the GUI from the given specifications

forge/src/main/java/com/envyful/api/forge/gui/ForgeGui.java

@@ -24,6 +24,7 @@
 import scala.xml.dtd.REQUIRED;
 
 import java.util.List;
+import java.util.function.Consumer;
 
 /**
  *
@@ -35,15 +36,18 @@ public class ForgeGui implements Gui {
     private final ITextComponent title;
     private final int height;
     private final PlayerManager<ForgeEnvyPlayer, EntityPlayerMP> playerManager;
+    private final Consumer<ForgeEnvyPlayer> closeConsumer;
     private final ForgeSimplePane parentPane;
     private final ForgeSimplePane[] panes;
 
     private final List<ForgeGuiContainer> containers = Lists.newArrayList();
 
-    ForgeGui(String title, int height, PlayerManager<ForgeEnvyPlayer, EntityPlayerMP> playerManager, Pane... panes) {
+    ForgeGui(String title, int height, PlayerManager<ForgeEnvyPlayer, EntityPlayerMP> playerManager,
+             Consumer<ForgeEnvyPlayer> closeConsumer, Pane... panes) {
         this.title = new TextComponentString(title);
         this.height = height;
         this.playerManager = playerManager;
+        this.closeConsumer = closeConsumer;
         this.parentPane = (ForgeSimplePane) new ForgeSimplePane.Builder().height(height).topLeftX(0).topLeftY(0).width(9).build();
         this.panes = new ForgeSimplePane[panes.length];
         int i = 0;
@@ -254,6 +258,11 @@ public void onContainerClosed(EntityPlayer playerIn) {
             super.onContainerClosed(playerIn);
 
             EnvyPlayer<?> player = this.gui.playerManager.getPlayer(playerIn.getUniqueID());
+
+            if (this.gui.closeConsumer != null) {
+                this.gui.closeConsumer.accept((ForgeEnvyPlayer) player);
+            }
+
             ForgeGuiTracker.removePlayer(player);
         }
     }

forge/src/main/java/com/envyful/api/forge/gui/ForgeGuiBuilder.java

@@ -3,11 +3,13 @@
 import com.envyful.api.forge.player.ForgeEnvyPlayer;
 import com.envyful.api.gui.Gui;
 import com.envyful.api.gui.pane.Pane;
+import com.envyful.api.player.EnvyPlayer;
 import com.envyful.api.player.PlayerManager;
 import com.google.common.collect.Lists;
 import net.minecraft.entity.player.EntityPlayerMP;
 
 import java.util.List;
+import java.util.function.Consumer;
 
 /**
  *
@@ -20,6 +22,7 @@ public class ForgeGuiBuilder implements Gui.Builder {
     private int height = 5;
     private List<Pane> panes = Lists.newArrayList();
     private PlayerManager<ForgeEnvyPlayer, EntityPlayerMP> playerManager;
+    private Consumer<EnvyPlayer<?>> closeConsumer = null;
 
     @Override
     public Gui.Builder title(String title) {
@@ -45,12 +48,19 @@ public Gui.Builder setPlayerManager(PlayerManager<?, ?> playerManager) {
         return this;
     }
 
+    @Override
+    public Gui.Builder setCloseConsumer(Consumer<EnvyPlayer<?>> consumer) {
+        this.closeConsumer = consumer;
+        return this;
+    }
+
     @Override
     public Gui build() {
         if (this.playerManager == null) {
             throw new RuntimeException("Cannot build GUI without PlayerManager being set");
         }
 
-        return new ForgeGui(this.title, this.height, this.playerManager, this.panes.toArray(new Pane[0]));
+        return new ForgeGui(this.title, this.height, this.playerManager,
+                forgeEnvyPlayer -> this.closeConsumer.accept(forgeEnvyPlayer), this.panes.toArray(new Pane[0]));
     }
 }