added heat notification too

src/main/java/com/toofifty/easygiantsfoundry/EasyGiantsFoundryConfig.java

@@ -14,7 +14,18 @@ public interface EasyGiantsFoundryConfig extends Config {
             description = "Configures whether to notify you when you are about to finish a stage.",
             position = 0
     )
-    default boolean showGiantsFoundryNotifications()
+    default boolean showGiantsFoundryStageNotifications()
+    {
+        return true;
+    }
+
+    @ConfigItem(
+            keyName = "giantsFoundryHeatNotification",
+            name = "Enable heat notifications",
+            description = "Configures whether to notify you when you are about to run out of heat.",
+            position = 1
+    )
+    default boolean showGiantsFoundryHeatNotifications()
     {
         return true;
     }

src/main/java/com/toofifty/easygiantsfoundry/EasyGiantsFoundryPlugin.java

@@ -42,6 +42,8 @@ public class EasyGiantsFoundryPlugin extends Plugin
 
 	private boolean enableStageNotifications;
 
+	private boolean enableHeatNotifications;
+
 	@Inject
 	private EasyGiantsFoundryState state;
 
@@ -71,7 +73,8 @@ public class EasyGiantsFoundryPlugin extends Plugin
 	{
 		overlayManager.add(overlay2d);
 		overlayManager.add(overlay3d);
-		enableStageNotifications = config.showGiantsFoundryNotifications();
+		enableStageNotifications = config.showGiantsFoundryStageNotifications();
+		enableHeatNotifications = config.showGiantsFoundryHeatNotifications();
 	}
 
 	@Override
@@ -124,14 +127,17 @@ public class EasyGiantsFoundryPlugin extends Plugin
 	@Subscribe
 	public void onStatChanged(StatChanged statChanged)
 	{
-		if (state.isEnabled() &&
-			enableStageNotifications &&
-			state.getCurrentStage() != null &&
-			helper.getActionsLeftInStage() == 1 &&
-			(oldStage == null || oldStage != state.getCurrentStage()))
+		if (state.isEnabled() && state.getCurrentStage() != null)
 		{
-			notifier.notify("About to finish the current stage!");
-			oldStage = state.getCurrentStage();
+			if (enableStageNotifications && helper.getActionsLeftInStage() == 1 && (oldStage == null || oldStage != state.getCurrentStage()))
+			{
+				notifier.notify("About to finish the current stage!");
+				oldStage = state.getCurrentStage();
+			}
+			else if (enableHeatNotifications && helper.getActionsForHeatLevel() == 1)
+			{
+				notifier.notify("About to run out of heat!");
+			}
 		}
 	}
 
@@ -215,9 +221,14 @@ public class EasyGiantsFoundryPlugin extends Plugin
 			return;
 		}
 
-		if (config.showGiantsFoundryNotifications() != enableStageNotifications)
+		if (config.showGiantsFoundryStageNotifications() != enableStageNotifications)
 		{
-			enableStageNotifications = config.showGiantsFoundryNotifications();
+			enableStageNotifications = config.showGiantsFoundryStageNotifications();
+		}
+
+		if (config.showGiantsFoundryHeatNotifications() != enableHeatNotifications)
+		{
+			enableHeatNotifications = config.showGiantsFoundryHeatNotifications();
 		}
 	}