1 files changed, 115 insertions, 0 deletions
diff --git a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CookieWarning.java b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CookieWarning.java
new file mode 100644
index 00000000..f21d0c50
--- /dev/null
+++ b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CookieWarning.java
@@ -0,0 +1,115 @@
+/*
+ * Copyright (C) 2022 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.miscfeatures;
+
+import com.google.common.collect.Lists;
+import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
+import io.github.moulberry.notenoughupdates.mixins.AccessorGuiPlayerTabOverlay;
+import io.github.moulberry.notenoughupdates.util.NotificationHandler;
+import net.minecraft.client.Minecraft;
+import net.minecraft.util.ChatComponentText;
+import net.minecraft.util.EnumChatFormatting;
+
+public class CookieWarning {
+
+	private static boolean hasNotified;
+
+	public static void resetNotification() {
+		hasNotified = false;
+		NotificationHandler.cancelNotification();
+	}
+
+	/**
+	 * Checks the tab list for a cookie timer, and sends a notification if the timer is within the tolerance
+	 */
+	public static void checkCookie() {
+		if (NotEnoughUpdates.INSTANCE.config.notifications.doBoosterNotif &&
+			NotEnoughUpdates.INSTANCE.hasSkyblockScoreboard()) {
+			String[] lines;
+			try {
+				lines = ((AccessorGuiPlayerTabOverlay) Minecraft.getMinecraft().ingameGUI.getTabList())
+					.getFooter()
+					.getUnformattedText()
+					.split("\n");
+			} catch (NullPointerException e) {
+				return; // if the footer is null or somehow doesn't exist, stop
+			}
+			boolean hasCookie = true;
+			String timeLine = null; // the line that contains the cookie timer
+			for (int i = 0; i < lines.length; i++) {
+				if (lines[i].startsWith("Cookie Buff")) {
+					timeLine = lines[i + 1]; // the line after the "Cookie Buff" line
+				}
+				if (lines[i].startsWith("Not active! Obtain booster cookies from the")) {
+					hasCookie = false;
+				}
+			}
+			if (!hasCookie) {
+				if (!hasNotified) {
+					NotificationHandler.displayNotification(Lists.newArrayList(
+						"\u00a7cBooster Cookie Ran Out!",
+						"\u00a77Your Booster Cookie expired!",
+						"\u00a77",
+						"\u00a77Press X on your keyboard to close this notification"
+					), true, true);
+					hasNotified = true;
+				}
+				return;
+			}
+			if (timeLine != null) {
+				String[] digits = timeLine.split(" ");
+				int minutes = 0;
+				try {
+					for (String digit : digits) {
+						if (digit.endsWith("y")) {
+							digit = digit.substring(0, digit.length() - 1);
+							minutes += Integer.parseInt(digit) * 525600;
+						} else if (digit.endsWith("d")) {
+							digit = digit.substring(0, digit.length() - 1);
+							minutes += Integer.parseInt(digit) * 1440;
+						} else if (digit.endsWith("h")) {
+							digit = digit.substring(0, digit.length() - 1);
+							minutes += Integer.parseInt(digit) * 60;
+						} else if (digit.endsWith("m")) {
+							digit = digit.substring(0, digit.length() - 1);
+							minutes += Integer.parseInt(digit);
+						} // ignore seconds
+					}
+				} catch (NumberFormatException e) {
+					e.printStackTrace();
+					Minecraft.getMinecraft().thePlayer.addChatMessage(new ChatComponentText(
+						EnumChatFormatting.RED +
+							"NEU ran into an issue when retrieving the Booster Cookie Timer. Check the logs for details."));
+					hasNotified = true;
+				}
+				if (minutes < NotEnoughUpdates.INSTANCE.config.notifications.boosterCookieWarningMins && !hasNotified) {
+					NotificationHandler.displayNotification(Lists.newArrayList(
+						"\u00a7cBooster Cookie Running Low!",
+						"\u00a77Your Booster Cookie will expire in " + timeLine,
+						"\u00a77",
+						"\u00a77Press X on your keyboard to close this notification"
+					), true, true);
+					hasNotified = true;
+				}
+			}
+		}
+
+	}
+}