1 files changed, 160 insertions, 0 deletions
diff --git a/src/main/kotlin/cc/woverflow/chattils/Chattils.kt b/src/main/kotlin/cc/woverflow/chattils/Chattils.kt
new file mode 100644
index 0000000..c9714a2
--- /dev/null
+++ b/src/main/kotlin/cc/woverflow/chattils/Chattils.kt
@@ -0,0 +1,160 @@
+package cc.woverflow.chattils
+
+import cc.woverflow.chattils.chat.ChatSearchingManager
+import cc.woverflow.chattils.chat.ChatShortcuts
+import cc.woverflow.chattils.chat.ChatTabs
+import cc.woverflow.chattils.command.ChattilsCommand
+import cc.woverflow.chattils.config.ChattilsConfig
+import cc.woverflow.chattils.hook.GuiNewChatHook
+import cc.woverflow.chattils.mixin.GuiNewChatAccessor
+import cc.woverflow.chattils.updater.Updater
+import cc.woverflow.chattils.utils.ModCompatHooks
+import cc.woverflow.chattils.utils.RenderHelper
+import gg.essential.api.EssentialAPI
+import gg.essential.universal.UDesktop
+import gg.essential.universal.UResolution
+import net.minecraft.client.Minecraft
+import net.minecraft.client.gui.*
+import net.minecraft.client.renderer.GlStateManager
+import net.minecraft.client.settings.KeyBinding
+import net.minecraft.client.shader.Framebuffer
+import net.minecraft.util.MathHelper
+import net.minecraftforge.common.MinecraftForge.EVENT_BUS
+import net.minecraftforge.fml.client.registry.ClientRegistry
+import net.minecraftforge.fml.common.Loader
+import net.minecraftforge.fml.common.Mod
+import net.minecraftforge.fml.common.event.FMLInitializationEvent
+import net.minecraftforge.fml.common.event.FMLPostInitializationEvent
+import net.minecraftforge.fml.common.event.FMLPreInitializationEvent
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
+import net.minecraftforge.fml.common.gameevent.TickEvent
+import org.lwjgl.input.Keyboard
+import java.awt.image.BufferedImage
+import java.io.File
+import java.text.SimpleDateFormat
+import java.util.*
+
+
+@Mod(
+    modid = Chattils.ID,
+    name = Chattils.NAME,
+    version = Chattils.VER,
+    modLanguageAdapter = "gg.essential.api.utils.KotlinAdapter"
+)
+object Chattils {
+
+    val keybind = KeyBinding("Screenshot Chat", Keyboard.KEY_NONE, "Chattils")
+    const val NAME = "@NAME@"
+    const val VER = "@VER@"
+    const val ID = "@ID@"
+    var doTheThing = false
+    lateinit var jarFile: File
+        private set
+    var isPatcher = false
+        private set
+    var isBetterChat = false
+        private set
+
+    private val fileFormatter: SimpleDateFormat = SimpleDateFormat("yyyy-MM-dd_HH.mm.ss'.png'")
+
+    val modDir = File(File(Minecraft.getMinecraft().mcDataDir, "W-OVERFLOW"), NAME)
+
+    @Mod.EventHandler
+    fun onFMLPreInitialization(event: FMLPreInitializationEvent) {
+        if (!modDir.exists()) modDir.mkdirs()
+        jarFile = event.sourceFile
+    }
+
+    @Mod.EventHandler
+    fun onInitialization(event: FMLInitializationEvent) {
+        ChattilsConfig.preload()
+        ChattilsCommand.register()
+        ClientRegistry.registerKeyBinding(keybind)
+        EVENT_BUS.register(this)
+        ChatTabs.initialize()
+        ChatShortcuts.initialize()
+        Updater.update()
+    }
+
+    @Mod.EventHandler
+    fun onPostInitialization(event: FMLPostInitializationEvent) {
+        isPatcher = Loader.isModLoaded("patcher")
+        isBetterChat = Loader.isModLoaded("betterchat")
+    }
+
+    @SubscribeEvent
+    fun onTickEvent(event: TickEvent.ClientTickEvent) {
+        if (event.phase == TickEvent.Phase.START && Minecraft.getMinecraft().theWorld != null && Minecraft.getMinecraft().thePlayer != null && (Minecraft.getMinecraft().currentScreen == null || Minecraft.getMinecraft().currentScreen is GuiChat)) {
+            if (doTheThing) {
+                screenshotChat()
+                doTheThing = false
+            }
+        }
+    }
+
+    fun getChatHeight(opened: Boolean): Int {
+        var height = if (opened) ChattilsConfig.focusedHeight else ChattilsConfig.unfocusedHeight
+        height = (height * Minecraft.getMinecraft().gameSettings.chatScale).toInt()
+        val chatY = ModCompatHooks.yOffset + ModCompatHooks.chatPosition
+        if (height + chatY + 27 > (UResolution.scaledHeight / Minecraft.getMinecraft().gameSettings.chatScale).toInt() - 27 - chatY) {
+            height = (UResolution.scaledHeight / Minecraft.getMinecraft().gameSettings.chatScale).toInt() - 27 - chatY
+        }
+        return height
+    }
+
+    fun screenshotLine(line: ChatLine): BufferedImage? {
+        val hud = Minecraft.getMinecraft().ingameGUI
+        val chat = hud.chatGUI
+        val i = MathHelper.floor_float(chat.chatWidth / chat.chatScale)
+        return screenshot(GuiUtilRenderComponents.splitText(line.chatComponent, i, Minecraft.getMinecraft().fontRendererObj, false, false).map { it.formattedText }.reversed(), chat.chatWidth)
+    }
+
+    private fun screenshotChat() {
+        screenshotChat(0)
+    }
+
+    fun screenshotChat(scrollPos: Int) {
+        val hud = Minecraft.getMinecraft().ingameGUI
+        val chat = hud.chatGUI
+        val chatLines = ArrayList<String>()
+        ChatSearchingManager.filterMessages((chat as GuiNewChatHook).prevText, (chat as GuiNewChatAccessor).drawnChatLines)?.let { drawnLines ->
+            val chatHeight = if (ChattilsConfig.customChatHeight) getChatHeight(true) / 9 else GuiNewChat.calculateChatboxHeight(Minecraft.getMinecraft().gameSettings.chatHeightFocused / 9)
+            for (i in scrollPos until drawnLines.size.coerceAtMost(scrollPos + chatHeight)) {
+                chatLines.add(drawnLines[i].chatComponent.formattedText)
+            }
+
+            screenshot(chatLines, chat.chatWidth)?.let {
+                RenderHelper.copyBufferedImageToClipboard(it)
+            }
+        }
+    }
+
+    private fun screenshot(messages: List<String>, width: Int): BufferedImage? {
+        if (messages.isEmpty()) {
+            EssentialAPI.getNotifications().push("Chattils", "Chat window is empty.")
+            return null
+        }
+
+        val fr: FontRenderer = ModCompatHooks.fontRenderer
+        val fb: Framebuffer = RenderHelper.createBindFramebuffer(width * 3, (messages.size * 9) * 3)
+        val file = File(Minecraft.getMinecraft().mcDataDir, "screenshots/chat/" + fileFormatter.format(Date()))
+
+        GlStateManager.scale(3f, 3f, 1f)
+        val scale = Minecraft.getMinecraft().gameSettings.chatScale
+        GlStateManager.scale(scale, scale, 1f)
+        for (i in messages.indices) {
+            fr.drawStringWithShadow(messages[i], 0f, (messages.size - 1 - i) * 9f, 0xffffff)
+        }
+
+        val image = RenderHelper.screenshotFramebuffer(fb, file)
+        Minecraft.getMinecraft().entityRenderer.setupOverlayRendering()
+        Minecraft.getMinecraft().framebuffer.bindFramebuffer(true)
+        EssentialAPI.getNotifications()
+            .push("Chattils", "Chat screenshotted successfully.\nClick to open.") {
+                if (!UDesktop.browse(file.toURI())) {
+                    EssentialAPI.getNotifications().push("Chattils", "Could not browse!")
+                }
+            }
+        return image
+    }
+}