1.8

1 files changed, 63 insertions, 0 deletions

diff --git a/src/main/java/io/github/moulberry/notenoughupdates/StreamerMode.java b/src/main/java/io/github/moulberry/notenoughupdates/StreamerMode.java
new file mode 100644
index 00000000..c7d5e8ae
--- /dev/null
+++ b/src/main/java/io/github/moulberry/notenoughupdates/StreamerMode.java
@@ -0,0 +1,63 @@
+package io.github.moulberry.notenoughupdates;
+
+import java.io.File;
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+public class StreamerMode {
+
+    private static final Pattern lobbyPattern = Pattern.compile("(mini|mega)([0-9]{1,3}[A-Z])");
+
+    public static String filterLobbyNames(String line) {
+        Matcher matcher = lobbyPattern.matcher(line);
+        if(matcher.find() && matcher.groupCount() == 2) {
+            String lobbyType = matcher.group(1);
+            String lobbyId = matcher.group(2);
+            long lobbyNum = Long.parseLong(lobbyId.substring(0, lobbyId.length()-1));
+
+            long obfLobbyNum = (lobbyNum*9182739 + 11) % 500;
+            char obfLobbyLetter = "ABCDEFGHIJKLMNOPQRSTUVWXYZ".charAt((int)(obfLobbyNum%26));
+
+            line = line.replaceAll("(mini|mega)([0-9]{1,3}[A-Z])", lobbyType+obfLobbyNum+obfLobbyLetter);
+        }
+        return line;
+    }
+
+    public static String filterScoreboard(String line) {
+        line = filterLobbyNames(cleanDuplicateColourCodes(line));
+        return line;
+    }
+
+    public static String filterChat(String line) {
+        line = filterLobbyNames(line);
+        return line;
+    }
+
+    private static String cleanDuplicateColourCodes(String line) {
+        StringBuilder sb = new StringBuilder();
+        char currentColourCode = 'r';
+        boolean sectionSymbolLast = false;
+        for(char c : line.toCharArray()) {
+            if((int)c > 50000) continue;
+
+            if(c == '\u00a7') {
+                sectionSymbolLast = true;
+            } else {
+                if(sectionSymbolLast) {
+                    if(currentColourCode != c) {
+                        sb.append('\u00a7');
+                        sb.append(c);
+                        currentColourCode = c;
+                    }
+                    sectionSymbolLast = false;
+                } else {
+                    sb.append(c);
+                }
+            }
+        }
+        return sb.toString();
+    }
+
+}