From f2a336f0fad51c6efea571928ab3f6c30ef5cad2 Mon Sep 17 00:00:00 2001
From: Neocraftr <51837582+Neocraftr@users.noreply.github.com>
Date: Mon, 24 Jul 2023 22:36:00 +0200
Subject: [PATCH] Added basic support for Griefergames Cloud

---
 .../neocraftr/griefergames/GrieferGames.java  | 11 ++++
 .../griefergames/enums/SubServerType.java     |  6 ++
 .../griefergames/listener/GGKeyListener.java  | 61 ++++++++++---------
 .../listener/GGMessageReceiveListener.java    | 25 ++++----
 .../listener/GGMessageSendListener.java       | 25 ++++----
 .../listener/GGScoreboardListener.java        |  2 +
 .../listener/GGServerJoinListener.java        | 11 ----
 .../listener/GGSubServerChangeListener.java   | 12 +++-
 .../listener/GGTablistListener.java           | 41 +++++++++++++
 .../griefergames/listener/GGTickListener.java | 19 +++---
 .../neocraftr/griefergames/utils/Helper.java  |  4 ++
 11 files changed, 146 insertions(+), 71 deletions(-)
 create mode 100644 core/src/main/java/de/neocraftr/griefergames/enums/SubServerType.java
 create mode 100644 core/src/main/java/de/neocraftr/griefergames/listener/GGTablistListener.java

diff --git a/core/src/main/java/de/neocraftr/griefergames/GrieferGames.java b/core/src/main/java/de/neocraftr/griefergames/GrieferGames.java
index 0b10306..77cf5ca 100644
--- a/core/src/main/java/de/neocraftr/griefergames/GrieferGames.java
+++ b/core/src/main/java/de/neocraftr/griefergames/GrieferGames.java
@@ -21,6 +21,7 @@
 import de.neocraftr.griefergames.chat.modules.Vote;
 import de.neocraftr.griefergames.commands.GGMessageCommand;
 import de.neocraftr.griefergames.core.generated.DefaultReferenceStorage;
+import de.neocraftr.griefergames.enums.SubServerType;
 import de.neocraftr.griefergames.hud.BoosterHudWidget;
 import de.neocraftr.griefergames.hud.DelayHudWidget;
 import de.neocraftr.griefergames.hud.FlyHudWidget;
@@ -35,6 +36,7 @@
 import de.neocraftr.griefergames.listener.GGServerMessageListener;
 import de.neocraftr.griefergames.listener.GGServerQuitListener;
 import de.neocraftr.griefergames.listener.GGSubServerChangeListener;
+import de.neocraftr.griefergames.listener.GGTablistListener;
 import de.neocraftr.griefergames.listener.GGTickListener;
 import de.neocraftr.griefergames.settings.GrieferGamesConfig;
 import de.neocraftr.griefergames.utils.FileManager;
@@ -76,6 +78,7 @@ public class GrieferGames extends LabyAddon<GrieferGamesConfig> {
   private long waitTime = 0;
   private boolean citybuildDelay = false;
   private String subServer = "";
+  private SubServerType subServerType = SubServerType.REGULAR;
   private long lastActivety = 0;
   private boolean afk = false;
   private boolean hideBoosterMenu = false;
@@ -101,6 +104,7 @@ protected void enable() {
     registerListener(new GGKeyListener(this));
     registerListener(new GGServerMessageListener(this));
     registerListener(new GGScoreboardListener(this));
+    registerListener(new GGTablistListener(this));
     registerListener(new GGSubServerChangeListener(this));
     registerListener(new GGTickListener(this));
 
@@ -253,6 +257,13 @@ public void setSubServer(String subServer) {
     this.subServer = subServer;
   }
 
+  public SubServerType getSubServerType() {
+    return subServerType;
+  }
+  public void setSubServerType(SubServerType subServerType) {
+    this.subServerType = subServerType;
+  }
+
   public long getLastActivety() {
     return lastActivety;
   }
diff --git a/core/src/main/java/de/neocraftr/griefergames/enums/SubServerType.java b/core/src/main/java/de/neocraftr/griefergames/enums/SubServerType.java
new file mode 100644
index 0000000..08f4ae1
--- /dev/null
+++ b/core/src/main/java/de/neocraftr/griefergames/enums/SubServerType.java
@@ -0,0 +1,6 @@
+package de.neocraftr.griefergames.enums;
+
+public enum SubServerType {
+  REGULAR,
+  CLOUD
+}
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGKeyListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGKeyListener.java
index 8a67166..3c96b2c 100644
--- a/core/src/main/java/de/neocraftr/griefergames/listener/GGKeyListener.java
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGKeyListener.java
@@ -1,6 +1,7 @@
 package de.neocraftr.griefergames.listener;
 
 import de.neocraftr.griefergames.GrieferGames;
+import de.neocraftr.griefergames.enums.SubServerType;
 import de.neocraftr.griefergames.settings.GrieferGamesConfig;
 import net.labymod.api.Laby;
 import net.labymod.api.client.component.Component;
@@ -26,43 +27,45 @@ public GGKeyListener(GrieferGames griefergames) {
   public void onKeyInput(KeyEvent event) {
     if(!griefergames.isOnGrieferGames()) return;
 
-    if(event.state() == State.PRESS) {
-      griefergames.setLastActivety(System.currentTimeMillis());
-      if(griefergames.isAfk()) {
-        griefergames.setAfk(false);
-        griefergames.helper().performAfkActions(false);
-      }
+    if(griefergames.getSubServerType() == SubServerType.REGULAR) {
+      if(event.state() == State.PRESS) {
+        griefergames.setLastActivety(System.currentTimeMillis());
+        if(griefergames.isAfk()) {
+          griefergames.setAfk(false);
+          griefergames.helper().performAfkActions(false);
+        }
 
-      if(griefergames.configuration().chatConfig().ampEnabled().get() &&
-          event.key().getId() == Laby.labyAPI().minecraft().options().getInputMapping("key.playerlist").getKeyCode()) {
+        if(griefergames.configuration().chatConfig().ampEnabled().get() &&
+            event.key().getId() == Laby.labyAPI().minecraft().options().getInputMapping("key.playerlist").getKeyCode()) {
 
-        for(NetworkPlayerInfo playerInfo : Laby.labyAPI().minecraft().getClientPacketListener().getNetworkPlayerInfos()) {
-          if(playerInfo == null) continue;
-          if(!griefergames.helper().componentToFormattedText(playerInfo.displayName()).contains("§k")) continue;
+          for(NetworkPlayerInfo playerInfo : Laby.labyAPI().minecraft().getClientPacketListener().getNetworkPlayerInfos()) {
+            if(playerInfo == null) continue;
+            if(!griefergames.helper().componentToFormattedText(playerInfo.displayName()).contains("§k")) continue;
 
-          Matcher matcher = antiMagicPrefixRegex.matcher(griefergames.helper().componentToPlainText(playerInfo.displayName()));
-          if(!matcher.find()) continue;
+            Matcher matcher = antiMagicPrefixRegex.matcher(griefergames.helper().componentToPlainText(playerInfo.displayName()));
+            if(!matcher.find()) continue;
 
-          String ampReplacement = griefergames.configuration().chatConfig().ampReplacement().get();
-          if(ampReplacement.isBlank()) {
-            ampReplacement = GrieferGamesConfig.DEFAULT_AMP_REPLACEMENT;
-          }
+            String ampReplacement = griefergames.configuration().chatConfig().ampReplacement().get();
+            if(ampReplacement.isBlank()) {
+              ampReplacement = GrieferGamesConfig.DEFAULT_AMP_REPLACEMENT;
+            }
 
-          List<Component> children = new ArrayList<>(playerInfo.displayName().getChildren());
-          for(int i=0; i<children.size(); i++) {
-            Component component = children.get(i);
-            String plain = griefergames.helper().componentToPlainText(component);
+            List<Component> children = new ArrayList<>(playerInfo.displayName().getChildren());
+            for(int i=0; i<children.size(); i++) {
+              Component component = children.get(i);
+              String plain = griefergames.helper().componentToPlainText(component);
 
-            if(plain.equals(matcher.group(1))) {
-              component.style(component.style().undecorate(TextDecoration.OBFUSCATED));
-              Component ampPrefix = Component.text(ampReplacement+" ", component.style());
-              children.add(i, ampPrefix);
-              i++;
-            } else if(plain.equals(matcher.group(2))) {
-              component.style(component.style().undecorate(TextDecoration.OBFUSCATED));
+              if(plain.equals(matcher.group(1))) {
+                component.style(component.style().undecorate(TextDecoration.OBFUSCATED));
+                Component ampPrefix = Component.text(ampReplacement+" ", component.style());
+                children.add(i, ampPrefix);
+                i++;
+              } else if(plain.equals(matcher.group(2))) {
+                component.style(component.style().undecorate(TextDecoration.OBFUSCATED));
+              }
             }
+            playerInfo.displayName().setChildren(children);
           }
-          playerInfo.displayName().setChildren(children);
         }
       }
     }
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageReceiveListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageReceiveListener.java
index 501e9af..cd25a41 100644
--- a/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageReceiveListener.java
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageReceiveListener.java
@@ -2,6 +2,7 @@
 
 import de.neocraftr.griefergames.GrieferGames;
 import de.neocraftr.griefergames.chat.events.GGChatProcessEvent;
+import de.neocraftr.griefergames.enums.SubServerType;
 import net.labymod.api.Laby;
 import net.labymod.api.configuration.labymod.chat.AdvancedChatMessage;
 import net.labymod.api.event.Subscribe;
@@ -20,18 +21,20 @@ public void onMessage(ChatReceiveEvent event) {
     if(!griefergames.isOnGrieferGames()) return;
     //System.out.println(GsonComponentSerializer.gson().serialize(event.message()));
 
-    GGChatProcessEvent processEvent = new GGChatProcessEvent(event.chatMessage());
-    Laby.labyAPI().eventBus().fire(processEvent);
-    if(processEvent.isCancelled()) {
-      event.setCancelled(true);
-    } else if(processEvent.isSecondChat()) {
-      griefergames.helper().displayInSecondChat(AdvancedChatMessage.chat(processEvent.getMessage()));
-      if(!processEvent.isKeepInRegularChat()) event.setCancelled(true);
-    }
+    if(griefergames.getSubServerType() == SubServerType.REGULAR) {
+      GGChatProcessEvent processEvent = new GGChatProcessEvent(event.chatMessage());
+      Laby.labyAPI().eventBus().fire(processEvent);
+      if(processEvent.isCancelled()) {
+        event.setCancelled(true);
+      } else if(processEvent.isSecondChat()) {
+        griefergames.helper().displayInSecondChat(AdvancedChatMessage.chat(processEvent.getMessage()));
+        if(!processEvent.isKeepInRegularChat()) event.setCancelled(true);
+      }
 
-    if(event.chatMessage().getPlainText().equals("[Switcher] Daten heruntergeladen!")) {
-      griefergames.setHideBoosterMenu(true);
-      griefergames.sendMessage("/booster");
+      if(event.chatMessage().getPlainText().equals("[Switcher] Daten heruntergeladen!")) {
+        griefergames.setHideBoosterMenu(true);
+        griefergames.sendMessage("/booster");
+      }
     }
   }
 
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageSendListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageSendListener.java
index 750f3cf..f8c637a 100644
--- a/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageSendListener.java
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGMessageSendListener.java
@@ -2,6 +2,7 @@
 
 import de.neocraftr.griefergames.GrieferGames;
 import de.neocraftr.griefergames.enums.ChatColor;
+import de.neocraftr.griefergames.enums.SubServerType;
 import net.labymod.api.client.component.Component;
 import net.labymod.api.client.component.format.NamedTextColor;
 import net.labymod.api.event.Subscribe;
@@ -21,19 +22,21 @@ public void onSend(ChatMessageSendEvent event) {
     if(!griefergames.isOnGrieferGames()) return;
     String msg = event.getMessage();
 
-    if(griefergames.configuration().chatConfig().preventCommandFailure().get()) {
-      if(msg.startsWith("7") && !msg.equalsIgnoreCase(lastMessage)) {
-        griefergames.displayAddonMessage(Component.text(I18n.translate(griefergames.namespace()+".messages.commandFailure"), NamedTextColor.RED));
-        lastMessage = msg;
-        event.setCancelled(true);
-      } else {
-        lastMessage = "";
+    if(griefergames.getSubServerType() == SubServerType.REGULAR) {
+      if(griefergames.configuration().chatConfig().preventCommandFailure().get()) {
+        if(msg.startsWith("7") && !msg.equalsIgnoreCase(lastMessage)) {
+          griefergames.displayAddonMessage(Component.text(I18n.translate(griefergames.namespace()+".messages.commandFailure"), NamedTextColor.RED));
+          lastMessage = msg;
+          event.setCancelled(true);
+        } else {
+          lastMessage = "";
+        }
       }
-    }
 
-    if(griefergames.configuration().automations().autoColor().get() != ChatColor.NONE &&
-        !msg.startsWith("/") && !msg.startsWith(".") && !msg.startsWith("-")) {
-      event.changeMessage("&"+griefergames.configuration().automations().autoColor().get().getColorCode()+msg);
+      if(griefergames.configuration().automations().autoColor().get() != ChatColor.NONE &&
+          !msg.startsWith("/") && !msg.startsWith(".") && !msg.startsWith("-")) {
+        event.changeMessage("&"+griefergames.configuration().automations().autoColor().get().getColorCode()+msg);
+      }
     }
   }
 }
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGScoreboardListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGScoreboardListener.java
index cabed9f..8e22f4f 100644
--- a/core/src/main/java/de/neocraftr/griefergames/listener/GGScoreboardListener.java
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGScoreboardListener.java
@@ -2,6 +2,7 @@
 
 import de.neocraftr.griefergames.GrieferGames;
 import de.neocraftr.griefergames.chat.events.GGSubServerChangeEvent;
+import de.neocraftr.griefergames.enums.SubServerType;
 import net.labymod.api.Laby;
 import net.labymod.api.event.Subscribe;
 import net.labymod.api.event.client.scoreboard.ScoreboardTeamUpdateEvent;
@@ -21,6 +22,7 @@ public void onScoreboardTeams(ScoreboardTeamUpdateEvent event) {
       String subServerName = griefergames.helper().componentToPlainText(event.team().getPrefix()).toLowerCase();
       if(subServerName.isBlank() || subServerName.contains("lade")) return;
 
+      griefergames.setSubServerType(SubServerType.REGULAR);
       if(!griefergames.getSubServer().equals(subServerName)) {
         griefergames.setSubServer(subServerName);
         GGSubServerChangeEvent changeEvent = new GGSubServerChangeEvent(subServerName);
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGServerJoinListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGServerJoinListener.java
index 5454734..66790f8 100644
--- a/core/src/main/java/de/neocraftr/griefergames/listener/GGServerJoinListener.java
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGServerJoinListener.java
@@ -12,8 +12,6 @@
 import net.labymod.api.event.client.network.server.ServerJoinEvent;
 import net.labymod.api.notification.Notification;
 import net.labymod.api.util.I18n;
-import java.util.Timer;
-import java.util.TimerTask;
 
 public class GGServerJoinListener {
   private final GrieferGames griefergames;
@@ -63,15 +61,6 @@ public void onServerJoin(ServerJoinEvent event) {
               .icon(Icon.texture(ResourceLocation.create(griefergames.namespace(), "textures/update.png"))).build());
         }
       }
-
-      if(griefergames.configuration().automations().autoPortal().get()) {
-        new Timer().schedule(new TimerTask() {
-          @Override
-          public void run() {
-            griefergames.sendMessage("/portal");
-          }
-        }, 500);
-      }
     }
   }
 }
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGSubServerChangeListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGSubServerChangeListener.java
index d07afd2..eeed767 100644
--- a/core/src/main/java/de/neocraftr/griefergames/listener/GGSubServerChangeListener.java
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGSubServerChangeListener.java
@@ -12,6 +12,8 @@
 import net.labymod.core.labyconnect.DefaultLabyConnect;
 import net.labymod.core.labyconnect.protocol.packets.PacketPlayServerStatusUpdate;
 import net.labymod.core.thirdparty.discord.DefaultDiscordApp;
+import java.util.Timer;
+import java.util.TimerTask;
 import java.util.concurrent.TimeUnit;
 
 public class GGSubServerChangeListener {
@@ -46,7 +48,6 @@ public void onSubServerChange(GGSubServerChangeEvent event) {
       labyConnect.sendPacket(packet);
     }
 
-
     if(griefergames.helper().isCityBuild(event.subServerName())) {
       if(!griefergames.isCitybuildDelay()) griefergames.setWaitTime(System.currentTimeMillis() + TimeUnit.SECONDS.toMillis(15));
       griefergames.setCitybuildDelay(false);
@@ -59,6 +60,15 @@ public void onSubServerChange(GGSubServerChangeEvent event) {
       if(!griefergames.isCitybuildDelay()) griefergames.setWaitTime(System.currentTimeMillis() + TimeUnit.SECONDS.toMillis(12));
     } else if(event.subServerName().equals("skyblock")) {
       if(!griefergames.isCitybuildDelay()) griefergames.setWaitTime(System.currentTimeMillis() + TimeUnit.SECONDS.toMillis(15));
+    } else if(event.subServerName().equals("lobby")) {
+      if(griefergames.configuration().automations().autoPortal().get()) {
+        new Timer().schedule(new TimerTask() {
+          @Override
+          public void run() {
+            griefergames.sendMessage("/portal");
+          }
+        }, 500);
+      }
     }
   }
 }
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGTablistListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGTablistListener.java
new file mode 100644
index 0000000..4cc45aa
--- /dev/null
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGTablistListener.java
@@ -0,0 +1,41 @@
+package de.neocraftr.griefergames.listener;
+
+import de.neocraftr.griefergames.GrieferGames;
+import de.neocraftr.griefergames.chat.events.GGSubServerChangeEvent;
+import de.neocraftr.griefergames.enums.SubServerType;
+import net.labymod.api.Laby;
+import net.labymod.api.client.component.Component;
+import net.labymod.api.client.scoreboard.TabList;
+import net.labymod.api.event.Subscribe;
+import net.labymod.api.event.client.scoreboard.TabListUpdateEvent;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public class GGTablistListener {
+  private final GrieferGames griefergames;
+  private final Pattern tabListServerRegex = Pattern.compile("Current Server: (.+) ", Pattern.MULTILINE);
+
+  public GGTablistListener(GrieferGames griefergames) {
+    this.griefergames = griefergames;
+  }
+
+  @Subscribe
+  public void onTablistUpdate(TabListUpdateEvent event) {
+    TabList tablist = Laby.labyAPI().minecraft().getTabList();
+    if(tablist == null) return;
+    Component header = tablist.header();
+    if(header != null) {
+      String headerText = griefergames.helper().componentToPlainText(header);
+      Matcher matcher = tabListServerRegex.matcher(headerText);
+      if(matcher.find()) {
+        String subServerName = "cloud_"+matcher.group(1).toLowerCase().split("-")[0];
+        griefergames.setSubServerType(SubServerType.CLOUD);
+        if(!griefergames.getSubServer().equals(subServerName)) {
+          griefergames.setSubServer(subServerName);
+          GGSubServerChangeEvent changeEvent = new GGSubServerChangeEvent(subServerName);
+          Laby.labyAPI().eventBus().fire(changeEvent);
+        }
+      }
+    }
+  }
+}
diff --git a/core/src/main/java/de/neocraftr/griefergames/listener/GGTickListener.java b/core/src/main/java/de/neocraftr/griefergames/listener/GGTickListener.java
index cd9d647..20d513a 100644
--- a/core/src/main/java/de/neocraftr/griefergames/listener/GGTickListener.java
+++ b/core/src/main/java/de/neocraftr/griefergames/listener/GGTickListener.java
@@ -1,6 +1,7 @@
 package de.neocraftr.griefergames.listener;
 
 import de.neocraftr.griefergames.GrieferGames;
+import de.neocraftr.griefergames.enums.SubServerType;
 import net.labymod.api.event.Phase;
 import net.labymod.api.event.Subscribe;
 import net.labymod.api.event.client.lifecycle.GameTickEvent;
@@ -18,21 +19,23 @@ public void onTick(GameTickEvent event) {
     if(!griefergames.isOnGrieferGames()) return;
 
     if(event.phase() == Phase.POST) {
-      long currentTime = System.currentTimeMillis();
+      if(griefergames.getSubServerType() == SubServerType.REGULAR) {
+        //long currentTime = System.currentTimeMillis();
 
       /*if(currentTime > nextNameColorize) {
         nextNameColorize = currentTime + 20000L;
         griefergames.helper().colorizePlayerNames();
       }*/
 
-      if(!griefergames.isAfk() && griefergames.getLastActivety() + (griefergames.configuration().automations().afkTime().get() * 60000) < System.currentTimeMillis()) {
-        griefergames.setAfk(true);
-        griefergames.helper().performAfkActions(true);
-      }
+        if(!griefergames.isAfk() && griefergames.getLastActivety() + (griefergames.configuration().automations().afkTime().get() * 60000) < System.currentTimeMillis()) {
+          griefergames.setAfk(true);
+          griefergames.helper().performAfkActions(true);
+        }
 
-      if(griefergames.isHideBoosterMenu() || griefergames.configuration().automations().hideBoosterMenu().get()) {
-        if(griefergames.controller().hideBoosterMenu()) {
-          griefergames.setHideBoosterMenu(false);
+        if(griefergames.isHideBoosterMenu() || griefergames.configuration().automations().hideBoosterMenu().get()) {
+          if(griefergames.controller().hideBoosterMenu()) {
+            griefergames.setHideBoosterMenu(false);
+          }
         }
       }
     }
diff --git a/core/src/main/java/de/neocraftr/griefergames/utils/Helper.java b/core/src/main/java/de/neocraftr/griefergames/utils/Helper.java
index 028563f..25bd7f9 100644
--- a/core/src/main/java/de/neocraftr/griefergames/utils/Helper.java
+++ b/core/src/main/java/de/neocraftr/griefergames/utils/Helper.java
@@ -117,6 +117,10 @@ public String formatServerName(String serverName) {
       return "CityBuild " + capitalize(serverName);
     } else if (serverName.equals("lava") || serverName.equals("wasser")) {
       return "Farmserver " + capitalize(serverName);
+    } else if(serverName.startsWith("cloud_")) {
+      String cloudServerName = serverName.replace("cloud_", "");
+      cloudServerName = cloudServerName.startsWith("cb") ? cloudServerName.toUpperCase() : capitalize(cloudServerName);
+      return "Cloud "+cloudServerName;
     } else {
       Matcher matcher = serverCityBuildRegex.matcher(serverName.toLowerCase());
       if (matcher.find()) {