aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/at/hannibal2/skyhanni/data/ChatManager.kt
blob: 167efdb3b1dadb9375335e2c5494ab14aad905f6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
package at.hannibal2.skyhanni.data

import at.hannibal2.skyhanni.SkyHanniMod
import at.hannibal2.skyhanni.events.LorenzActionBarEvent
import at.hannibal2.skyhanni.events.LorenzChatEvent
import at.hannibal2.skyhanni.events.PacketEvent
import at.hannibal2.skyhanni.events.SeaCreatureFishEvent
import at.hannibal2.skyhanni.features.chat.ChatFilterGui
import at.hannibal2.skyhanni.features.fishing.SeaCreatureManager
import at.hannibal2.skyhanni.utils.IdentityCharacteristics
import at.hannibal2.skyhanni.utils.LorenzLogger
import at.hannibal2.skyhanni.utils.LorenzUtils
import at.hannibal2.skyhanni.utils.LorenzUtils.makeAccessible
import net.minecraft.client.Minecraft
import net.minecraft.client.gui.ChatLine
import net.minecraft.client.gui.GuiNewChat
import net.minecraft.event.HoverEvent
import net.minecraft.network.play.server.S02PacketChat
import net.minecraft.util.EnumChatFormatting
import net.minecraft.util.IChatComponent
import net.minecraftforge.client.event.ClientChatReceivedEvent
import net.minecraftforge.fml.common.eventhandler.EventPriority
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
import net.minecraftforge.fml.relauncher.ReflectionHelper
import java.lang.invoke.MethodHandles
import java.util.LinkedHashMap

object ChatManager {

    private val loggerAll = LorenzLogger("chat/all")
    private val loggerFiltered = LorenzLogger("chat/blocked")
    private val loggerAllowed = LorenzLogger("chat/allowed")
    private val loggerModified = LorenzLogger("chat/modified")
    private val loggerFilteredTypes = mutableMapOf<String, LorenzLogger>()
    private val messageHistory =
        object : LinkedHashMap<IdentityCharacteristics<IChatComponent>, MessageFilteringResult>() {
            override fun removeEldestEntry(eldest: MutableMap.MutableEntry<IdentityCharacteristics<IChatComponent>, MessageFilteringResult>?): Boolean {
                return size > 100
            }
        }

    fun getRecentMessageHistory(): List<MessageFilteringResult> {
        return messageHistory.toList().map { it.second }
    }

    enum class ActionKind(format: Any) {
        BLOCKED(EnumChatFormatting.RED.toString() + EnumChatFormatting.BOLD),
        RETRACTED(EnumChatFormatting.DARK_PURPLE.toString() + EnumChatFormatting.BOLD),
        MODIFIED(EnumChatFormatting.YELLOW.toString() + EnumChatFormatting.BOLD),
        ALLOWED(EnumChatFormatting.GREEN),
        ;

        val renderedString = "$format$name"

        companion object {
            val maxLength by lazy {
                entries.maxOf { Minecraft.getMinecraft().fontRendererObj.getStringWidth(it.renderedString) }
            }
        }
    }

    data class MessageFilteringResult(
        val message: IChatComponent,
        var actionKind: ActionKind,
        var actionReason: String?,
        val modified: IChatComponent?
    )

    @SubscribeEvent(priority = EventPriority.LOW, receiveCanceled = true)
    fun onActionBarPacket(event: PacketEvent.ReceiveEvent) {
        val packet = event.packet
        if (packet !is S02PacketChat) return
        val messageComponent = packet.chatComponent

        val message = LorenzUtils.stripVanillaMessage(messageComponent.formattedText)
        if (packet.type.toInt() == 2) {
            val actionBarEvent = LorenzActionBarEvent(message)
            actionBarEvent.postAndCatch()
        }
    }

    @SubscribeEvent(receiveCanceled = true)
    fun onChatReceive(event: ClientChatReceivedEvent) {
        if (event.type.toInt() == 2) return

        val original = event.message
        val message = LorenzUtils.stripVanillaMessage(original.formattedText)

        if (message.startsWith("§f{\"server\":\"")) return
        val key = IdentityCharacteristics(original)
        val chatEvent = LorenzChatEvent(message, original)
        if (!isSoopyMessage(event.message)) {
            chatEvent.postAndCatch()
        }

        val blockReason = chatEvent.blockedReason.uppercase()
        if (blockReason != "") {
            event.isCanceled = true
            loggerFiltered.log("[$blockReason] $message")
            loggerAll.log("[$blockReason] $message")
            loggerFilteredTypes.getOrPut(blockReason) { LorenzLogger("chat/filter_blocked/$blockReason") }
                .log(message)
            messageHistory.put(key, MessageFilteringResult(original, ActionKind.BLOCKED, blockReason, null))
            return
        }

        val modified = chatEvent.chatComponent
        loggerAllowed.log("[allowed] $message")
        loggerAll.log("[allowed] $message")
        if (modified.formattedText != original.formattedText) {
            event.message = chatEvent.chatComponent
            loggerModified.log(" ")
            loggerModified.log("[original] " + original.formattedText)
            loggerModified.log("[modified] " + modified.formattedText)
            messageHistory.put(key, MessageFilteringResult(original, ActionKind.MODIFIED, null, modified))
        } else {
            messageHistory.put(key, MessageFilteringResult(original, ActionKind.ALLOWED, null, null))
        }
    }

    private fun isSoopyMessage(message: IChatComponent): Boolean {
        for (sibling in message.siblings) {
            if (isSoopyMessage(sibling)) return true
        }

        val style = message.chatStyle ?: return false
        val hoverEvent = style.chatHoverEvent ?: return false
        if (hoverEvent.action != HoverEvent.Action.SHOW_TEXT) return false
        val text = hoverEvent.value?.formattedText ?: return false

        val lines = text.split("\n")
        if (lines.isEmpty()) return false

        val last = lines.last()
        if (last.startsWith("§f§lCOMMON")) return true
        if (last.startsWith("§a§lUNCOMMON")) return true
        if (last.startsWith("§9§lRARE")) return true
        if (last.startsWith("§5§lEPIC")) return true
        if (last.startsWith("§6§lLEGENDARY")) return true
        if (last.startsWith("§d§lMYTHIC")) return true
        if (last.startsWith("§c§lSPECIAL")) return true

        // TODO confirm this format is correct
        if (last.startsWith("§c§lVERY SPECIAL")) return true
        return false
    }

    @SubscribeEvent
    fun onChatMessage(chatEvent: LorenzChatEvent) {
        if (!LorenzUtils.inSkyBlock) return

        val seaCreature = SeaCreatureManager.getSeaCreature(chatEvent.message) ?: return
        SeaCreatureFishEvent(seaCreature, chatEvent).postAndCatch()
    }

    fun openChatFilterGUI() {
        SkyHanniMod.screenToOpen = (ChatFilterGui(getRecentMessageHistory()))
    }

    val chatLinesField by lazy {
        MethodHandles.publicLookup().unreflectGetter(
            ReflectionHelper.findField(GuiNewChat::class.java, "chatLines", "field_146252_h", "h")
                .makeAccessible()
        )
    }

    fun retractMessage(message: IChatComponent?, reason: String) {
        if (message == null) return
        val chatGUI = Minecraft.getMinecraft().ingameGUI.chatGUI
        val chatLines = chatLinesField.invokeExact(chatGUI) as MutableList<ChatLine>
        if (chatLines.removeIf { it.chatComponent === message }) {
            val history = messageHistory[IdentityCharacteristics(message)]
            history?.actionKind = ActionKind.RETRACTED
            history?.actionReason = reason.uppercase()
        }
        chatGUI.refreshChat()
    }
}