1 files changed, 57 insertions, 0 deletions
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/util/SidebarUtil.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/util/SidebarUtil.kt
new file mode 100644
index 00000000..ad371b56
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/util/SidebarUtil.kt
@@ -0,0 +1,57 @@
+/*
+ * Copyright (C) 2023 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.util
+
+import net.minecraft.client.Minecraft
+import net.minecraft.scoreboard.Score
+import net.minecraft.scoreboard.ScorePlayerTeam
+
+object SidebarUtil {
+
+    @JvmStatic
+    fun readSidebarLines() = readRawSidebarLines(true)
+
+    @JvmStatic
+    fun readRawSidebarLines(cleanColor: Boolean = false): List<String> {
+        val list = fetchScoreboardLines().reversed()
+        val result = list.map { cleanSB(it) }
+        return if (cleanColor) {
+            result.map { Utils.cleanColour(it) }
+        } else result
+    }
+
+    private fun cleanSB(scoreboard: String) =
+        scoreboard.toCharArray().filter { it.code in 21..126 || it.code == 167 }.joinToString(separator = "")
+
+    private fun fetchScoreboardLines(): List<String> {
+        val scoreboard = Minecraft.getMinecraft().theWorld?.scoreboard ?: return emptyList()
+        val objective = scoreboard.getObjectiveInDisplaySlot(1) ?: return emptyList()
+        var scores = scoreboard.getSortedScores(objective)
+        val list = scores.filter { input: Score? ->
+            input != null && input.playerName != null && !input.playerName.startsWith("#")
+        }
+        scores = if (list.size > 15) {
+            list.drop(15)
+        } else list
+        return scores.map {
+            ScorePlayerTeam.formatPlayerName(scoreboard.getPlayersTeam(it.playerName), it.playerName)
+        }
+    }
+}