improve search button display and screenshotting, and fix tooltips (#2)

7 files changed, 280 insertions, 34 deletions

diff --git a/src/main/kotlin/com/raeids/stratus/Stratus.kt b/src/main/kotlin/com/raeids/stratus/Stratus.kt
index d771915..7588749 100644
--- a/src/main/kotlin/com/raeids/stratus/Stratus.kt
+++ b/src/main/kotlin/com/raeids/stratus/Stratus.kt
@@ -4,17 +4,18 @@ import com.raeids.stratus.command.StratusCommand
 import com.raeids.stratus.config.StratusConfig
 import com.raeids.stratus.hook.ChatShortcuts
 import com.raeids.stratus.hook.ChatTabs
+import com.raeids.stratus.hook.GuiNewChatHook
+import com.raeids.stratus.hook.filterMessages
 import com.raeids.stratus.mixin.GuiNewChatAccessor
 import com.raeids.stratus.updater.Updater
 import com.raeids.stratus.utils.RenderHelper
 import gg.essential.api.EssentialAPI
 import gg.essential.universal.UDesktop
 import net.minecraft.client.Minecraft
-import net.minecraft.client.gui.FontRenderer
-import net.minecraft.client.gui.GuiChat
-import net.minecraft.client.renderer.GlStateManager
+import net.minecraft.client.gui.*
 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
@@ -25,6 +26,7 @@ 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.*
@@ -81,38 +83,51 @@ object Stratus {
     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) {
-                screenshot()
+                screenshotChat()
                 doTheThing = false
             }
         }
     }
 
-    private fun screenshot() {
+    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>()
+        filterMessages((chat as GuiNewChatHook).prevText, (chat as GuiNewChatAccessor).drawnChatLines)?.let { drawnLines ->
+            for (i in scrollPos until drawnLines.size.coerceAtMost(scrollPos+GuiNewChat.calculateChatboxHeight(Minecraft.getMinecraft().gameSettings.chatHeightFocused)/9)){
+                chatLines.add(drawnLines[i].chatComponent.formattedText)
+            }
 
-        /* Render chat fully. */
-        var w = chat.chatWidth
-        var h = chat.chatHeight
-        if ((chat as GuiNewChatAccessor).drawnChatLines.size < 20) {
-            h = (chat as GuiNewChatAccessor).drawnChatLines
-                .size * Minecraft.getMinecraft().fontRendererObj.FONT_HEIGHT
+            screenshot(chatLines, chat.chatWidth)?.let {
+                RenderHelper.copyBufferedImageToClipboard(it)
+            }
         }
-        if (w <= 0 || h <= 0 || (chat as GuiNewChatAccessor).drawnChatLines.isEmpty()) {
+    }
+
+    private fun screenshot(messages: List<String>, width: Int): BufferedImage? {
+        if (messages.isEmpty()) {
             EssentialAPI.getNotifications().push("Stratus", "Chat window is empty.")
-            return
+            return null
         }
-        val chatLines: MutableList<String> = ArrayList()
+
         val fr: FontRenderer = Minecraft.getMinecraft().fontRendererObj
-        for (chatLine in (chat as GuiNewChatAccessor).drawnChatLines) chatLines.add(chatLine.chatComponent.formattedText)
-        if (chatLines.isNotEmpty()) {
-            w = fr.getStringWidth(chatLines.stream().max(Comparator.comparingInt { obj: String -> obj.length }).get())
-        }
-        val fb: Framebuffer = RenderHelper.createBindFramebuffer(w, h)
-        GlStateManager.translate(-2f, (160 - (180 - h)).toFloat(), 0f)
-        chat.drawChat(hud.updateCounter)
+        val fb: Framebuffer = RenderHelper.createBindFramebuffer(width, messages.size*9)
         val file = File(Minecraft.getMinecraft().mcDataDir, "screenshots/chat/" + fileFormatter.format(Date()))
-        RenderHelper.screenshotFramebuffer(fb, file)
+
+        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()
@@ -121,5 +136,6 @@ object Stratus {
                     EssentialAPI.getNotifications().push("Stratus", "Could not browse!")
                 }
             }
+        return image
     }
 }
diff --git a/src/main/kotlin/com/raeids/stratus/config/StratusConfig.kt b/src/main/kotlin/com/raeids/stratus/config/StratusConfig.kt
index e9d2401..ecc0c09 100644
--- a/src/main/kotlin/com/raeids/stratus/config/StratusConfig.kt
+++ b/src/main/kotlin/com/raeids/stratus/config/StratusConfig.kt
@@ -24,6 +24,19 @@ object StratusConfig : Vigilant(File(Stratus.modDir, "${Stratus.ID}.toml"), Stra
     var chatSearch = true
 
     @Property(
+        type = PropertyType.SELECTOR,
+        name = "Screenshot Mode",
+        description = "The mode in which screenshotting will work.",
+        category = "Screenshotting",
+        options = [
+            "Save To System",
+            "Add To Clipboard",
+            "Both"
+        ]
+    )
+    var copyMode = 0
+
+    @Property(
         type = PropertyType.SWITCH,
         name = "Chat Tabs",
         description = "Add chat tabs.",
diff --git a/src/main/kotlin/com/raeids/stratus/hook/ChatSearching.kt b/src/main/kotlin/com/raeids/stratus/hook/ChatSearching.kt
index 57cfc93..6dd4904 100644
--- a/src/main/kotlin/com/raeids/stratus/hook/ChatSearching.kt
+++ b/src/main/kotlin/com/raeids/stratus/hook/ChatSearching.kt
@@ -22,14 +22,14 @@ private var POOL: ThreadPoolExecutor = ThreadPoolExecutor(
     )
 }
 
-val cache: Cache<String, List<ChatLine?>> = Caffeine.newBuilder().executor(POOL).maximumSize(5000).build()
+val cache: Cache<String, List<ChatLine>> = Caffeine.newBuilder().executor(POOL).maximumSize(5000).build()
 
-fun filterMessages(text: String, list: List<ChatLine?>?): List<ChatLine?>? {
-    if (list.isNullOrEmpty() || text.isBlank()) return list
+fun filterMessages(text: String, list: List<ChatLine>): List<ChatLine>? {
+    if (text.isBlank()) return list
     val cached = cache.getIfPresent(text)
     return cached ?: run {
         cache.put(text, list.filter {
-            it != null && UTextComponent.stripFormatting(it.chatComponent.unformattedText).lowercase()
+            UTextComponent.stripFormatting(it.chatComponent.unformattedText).lowercase()
                 .contains(text.lowercase())
         })
         cache.getIfPresent(text)
diff --git a/src/main/kotlin/com/raeids/stratus/hook/CleanSearchButton.kt b/src/main/kotlin/com/raeids/stratus/hook/CleanSearchButton.kt
index 4b7ba5a..7a68c75 100644
--- a/src/main/kotlin/com/raeids/stratus/hook/CleanSearchButton.kt
+++ b/src/main/kotlin/com/raeids/stratus/hook/CleanSearchButton.kt
@@ -1,10 +1,14 @@
 package com.raeids.stratus.hook
 
+import com.raeids.stratus.Stratus
 import gg.essential.universal.UResolution
 import net.minecraft.client.Minecraft
+import net.minecraft.client.gui.Gui
 import net.minecraft.client.gui.GuiTextField
+import net.minecraft.client.renderer.GlStateManager
+import net.minecraft.util.ResourceLocation
 
-class CleanSearchButton: CleanButton(3993935, {UResolution.scaledWidth - 42}, {UResolution.scaledHeight - 27}, 40, 12, "Search") {
+class CleanSearchButton: CleanButton(3993935, {UResolution.scaledWidth - 14}, {UResolution.scaledHeight - 27}, 12, 12, "") {
     val inputField = SearchTextField()
     private var chatBox = false
 
@@ -13,7 +17,6 @@ class CleanSearchButton: CleanButton(3993935, {UResolution.scaledWidth - 42}, {U
     }
 
     override fun onMousePress() {
-        println("hi")
         chatBox = !chatBox
         inputField.setEnabled(chatBox)
         inputField.isFocused = chatBox
@@ -24,6 +27,17 @@ class CleanSearchButton: CleanButton(3993935, {UResolution.scaledWidth - 42}, {U
     override fun drawButton(mc: Minecraft, mouseX: Int, mouseY: Int) {
         inputField.drawTextBox()
         super.drawButton(mc, mouseX, mouseY)
+        if (visible) {
+            mc.textureManager.bindTexture(ResourceLocation(Stratus.ID, "search.png"))
+            if (isEnabled()) {
+                GlStateManager.color(224f/255f, 224f/255f, 224f/255f)
+            } else if (mouseX >= xPosition && mouseX <= xPosition + 10 && mouseY >= yPosition && mouseY <= yPosition + 10) {
+                GlStateManager.color(1f, 1f, 160f/255f)
+            } else {
+                GlStateManager.color(1f, 1f, 1f)
+            }
+            Gui.drawModalRectWithCustomSizedTexture(xPosition+1, yPosition+1, 0f, 0f, 10, 10, 10f, 10f)
+        }
     }
 
     inner class SearchTextField: GuiTextField(69420, Minecraft.getMinecraft().fontRendererObj, UResolution.scaledWidth * 4 / 5 - 60, UResolution.scaledHeight - 27, UResolution.scaledWidth / 5, 12) {
diff --git a/src/main/kotlin/com/raeids/stratus/hook/ScreenshotButton.kt b/src/main/kotlin/com/raeids/stratus/hook/ScreenshotButton.kt
new file mode 100644
index 0000000..70c64c2
--- /dev/null
+++ b/src/main/kotlin/com/raeids/stratus/hook/ScreenshotButton.kt
@@ -0,0 +1,34 @@
+package com.raeids.stratus.hook
+
+import com.raeids.stratus.Stratus
+import com.raeids.stratus.mixin.GuiNewChatAccessor
+import gg.essential.api.utils.GuiUtil
+import gg.essential.universal.UResolution
+import net.minecraft.client.Minecraft
+import net.minecraft.client.gui.Gui
+import net.minecraft.client.gui.GuiChat
+import net.minecraft.client.renderer.GlStateManager
+import net.minecraft.util.ResourceLocation
+
+class ScreenshotButton: CleanButton(448318, { UResolution.scaledWidth - 28}, { UResolution.scaledHeight - 27}, 12, 12, "") {
+
+    override fun onMousePress() {
+        val chat = Minecraft.getMinecraft().ingameGUI.chatGUI
+        if (GuiUtil.getOpenedScreen() is GuiChat) {
+            Stratus.screenshotChat((chat as GuiNewChatAccessor).scrollPos)
+        }
+    }
+
+    override fun drawButton(mc: Minecraft, mouseX: Int, mouseY: Int) {
+        super.drawButton(mc, mouseX, mouseY)
+        if (visible) {
+            if (hovered) {
+                GlStateManager.color(1f, 1f, 160f / 255f)
+            } else {
+                GlStateManager.color(1f, 1f, 1f)
+            }
+            mc.textureManager.bindTexture(ResourceLocation(Stratus.ID, "screenshot.png"))
+            Gui.drawModalRectWithCustomSizedTexture(xPosition + 1, yPosition + 1, 0f, 0f, 10, 10, 10f, 10f)
+        }
+    }
+}
\ No newline at end of file
diff --git a/src/main/kotlin/com/raeids/stratus/utils/ImageTransferable.kt b/src/main/kotlin/com/raeids/stratus/utils/ImageTransferable.kt
new file mode 100644
index 0000000..81571a5
--- /dev/null
+++ b/src/main/kotlin/com/raeids/stratus/utils/ImageTransferable.kt
@@ -0,0 +1,22 @@
+package com.raeids.stratus.utils
+
+import java.awt.Image
+import java.awt.datatransfer.DataFlavor
+import java.awt.datatransfer.Transferable
+import java.awt.datatransfer.UnsupportedFlavorException
+
+class ImageTransferable(private val image: Image): Transferable {
+
+    override fun getTransferDataFlavors(): Array<DataFlavor> {
+        return arrayOf(DataFlavor.imageFlavor)
+    }
+
+    override fun isDataFlavorSupported(flavor: DataFlavor?): Boolean {
+        return DataFlavor.imageFlavor.equals(flavor)
+    }
+
+    override fun getTransferData(flavor: DataFlavor?): Any {
+        if (isDataFlavorSupported(flavor)) return image
+        throw UnsupportedFlavorException(flavor)
+    }
+}
\ No newline at end of file
diff --git a/src/main/kotlin/com/raeids/stratus/utils/RenderHelper.kt b/src/main/kotlin/com/raeids/stratus/utils/RenderHelper.kt
index d2aac4d..664f5de 100644
--- a/src/main/kotlin/com/raeids/stratus/utils/RenderHelper.kt
+++ b/src/main/kotlin/com/raeids/stratus/utils/RenderHelper.kt
@@ -1,21 +1,165 @@
 package com.raeids.stratus.utils
 
+import com.raeids.stratus.config.StratusConfig
 import net.minecraft.client.renderer.GlStateManager
 import net.minecraft.client.renderer.texture.TextureUtil
 import net.minecraft.client.shader.Framebuffer
+import org.apache.commons.lang3.SystemUtils
 import org.lwjgl.BufferUtils
 import org.lwjgl.opengl.GL11
 import org.lwjgl.opengl.GL12
+import sun.awt.datatransfer.DataTransferer
+import sun.awt.datatransfer.SunClipboard
+import java.awt.Toolkit
 import java.awt.image.BufferedImage
 import java.io.File
+import java.lang.reflect.Field
+import java.lang.reflect.Method
+import java.nio.ByteBuffer
+import java.nio.ByteOrder
 import javax.imageio.ImageIO
 
 
 object RenderHelper {
     /**
      * Taken from https://github.com/Moulberry/HyChat
+     * Modified so if not on Windows just in case it will switch it to RGB and remove the transparent background.
      */
-    fun screenshotFramebuffer(framebuffer: Framebuffer, file: File) {
+    fun copyBufferedImageToClipboard(bufferedImage: BufferedImage) {
+        if (SystemUtils.IS_OS_WINDOWS) {
+            try {
+                val width = bufferedImage.width
+                val height = bufferedImage.height
+                val hdrSize = 0x28
+                val buffer: ByteBuffer = ByteBuffer.allocate(hdrSize + width * height * 4)
+                buffer.order(ByteOrder.LITTLE_ENDIAN)
+                //Header size
+                buffer.putInt(hdrSize)
+                //Width
+                buffer.putInt(width)
+                //Int32 biHeight;
+                buffer.putInt(height)
+                //Int16 biPlanes;
+                buffer.put(1.toByte())
+                buffer.put(0.toByte())
+                //Int16 biBitCount;
+                buffer.put(32.toByte())
+                buffer.put(0.toByte())
+                //Compression
+                buffer.putInt(0)
+                //Int32 biSizeImage;
+                buffer.putInt(width * height * 4)
+                buffer.putInt(0)
+                buffer.putInt(0)
+                buffer.putInt(0)
+                buffer.putInt(0)
+
+                //Image data
+                for (y in 0 until height) {
+                    for (x in 0 until width) {
+                        val argb: Int = bufferedImage.getRGB(x, height - y - 1)
+                        if (argb shr 24 and 0xFF == 0) {
+                            buffer.putInt(0x00000000)
+                        } else {
+                            buffer.putInt(argb)
+                        }
+                    }
+                }
+                buffer.flip()
+                val hdrSizev5 = 0x7C
+                val bufferv5: ByteBuffer = ByteBuffer.allocate(hdrSizev5 + width * height * 4)
+                bufferv5.order(ByteOrder.LITTLE_ENDIAN)
+                //Header size
+                bufferv5.putInt(hdrSizev5)
+                //Width
+                bufferv5.putInt(width)
+                //Int32 biHeight;
+                bufferv5.putInt(height)
+                //Int16 biPlanes;
+                bufferv5.put(1.toByte())
+                bufferv5.put(0.toByte())
+                //Int16 biBitCount;
+                bufferv5.put(32.toByte())
+                bufferv5.put(0.toByte())
+                //Compression
+                bufferv5.putInt(0)
+                //Int32 biSizeImage;
+                bufferv5.putInt(width * height * 4)
+                bufferv5.putInt(0)
+                bufferv5.putInt(0)
+                bufferv5.putInt(0)
+                bufferv5.putInt(0)
+                bufferv5.order(ByteOrder.BIG_ENDIAN)
+                bufferv5.putInt(-0x1000000)
+                bufferv5.putInt(0x00FF0000)
+                bufferv5.putInt(0x0000FF00)
+                bufferv5.putInt(0x000000FF)
+                bufferv5.order(ByteOrder.LITTLE_ENDIAN)
+
+                //BGRs
+                bufferv5.put(0x42.toByte())
+                bufferv5.put(0x47.toByte())
+                bufferv5.put(0x52.toByte())
+                bufferv5.put(0x73.toByte())
+                for (i in bufferv5.position() until hdrSizev5) {
+                    bufferv5.put(0.toByte())
+                }
+
+                //Image data
+                for (y in 0 until height) {
+                    for (x in 0 until width) {
+                        val argb: Int = bufferedImage.getRGB(x, height - y - 1)
+                        val a = argb shr 24 and 0xFF
+                        var r = argb shr 16 and 0xFF
+                        var g = argb shr 8 and 0xFF
+                        var b = argb and 0xFF
+                        r = r * a / 0xFF
+                        g = g * a / 0xFF
+                        b = b * a / 0xFF
+                        bufferv5.putInt(a shl 24 or (r shl 16) or (g shl 8) or b)
+                    }
+                }
+                bufferv5.flip()
+                val clip = Toolkit.getDefaultToolkit().systemClipboard
+                val dt = DataTransferer.getInstance()
+                val f: Field = dt.javaClass.getDeclaredField("CF_DIB")
+                f.isAccessible = true
+                val format: Long = f.getLong(null)
+                val openClipboard: Method = clip.javaClass.getDeclaredMethod("openClipboard", SunClipboard::class.java)
+                openClipboard.isAccessible = true
+                openClipboard.invoke(clip, clip)
+                val publishClipboardData: Method = clip.javaClass.getDeclaredMethod(
+                    "publishClipboardData",
+                    Long::class.javaPrimitiveType,
+                    ByteArray::class.java
+                )
+                publishClipboardData.isAccessible = true
+                val arr: ByteArray = buffer.array()
+                publishClipboardData.invoke(clip, format, arr)
+                val closeClipboard: Method = clip.javaClass.getDeclaredMethod("closeClipboard")
+                closeClipboard.isAccessible = true
+                closeClipboard.invoke(clip)
+                return
+            } catch (e: Exception) {
+                e.printStackTrace()
+            }
+        }
+        val pixels: IntArray = bufferedImage.getRGB(0, 0, bufferedImage.width, bufferedImage.height, null, 0, bufferedImage.width)
+        val newImage = BufferedImage(bufferedImage.width, bufferedImage.height, BufferedImage.TYPE_INT_RGB)
+        newImage.setRGB(0, 0, newImage.width, newImage.height, pixels, 0, newImage.width)
+
+        try {
+            Toolkit.getDefaultToolkit().systemClipboard.setContents(ImageTransferable(bufferedImage), null)
+        } catch (e: Exception) {
+            e.printStackTrace()
+        }
+    }
+
+
+    /**
+     * Taken from https://github.com/Moulberry/HyChat
+     */
+    fun screenshotFramebuffer(framebuffer: Framebuffer, file: File): BufferedImage {
         val w = framebuffer.framebufferWidth
         val h = framebuffer.framebufferHeight
         val i = w * h
@@ -34,12 +178,15 @@ object RenderHelper {
                 bufferedimage.setRGB(l, k - j, pixelValues[k * framebuffer.framebufferTextureWidth + l])
             }
         }
-        try {
-            file.parentFile.mkdirs()
-            ImageIO.write(bufferedimage, "png", file)
-        } catch (e: Exception) {
-            e.printStackTrace()
+        if (StratusConfig.copyMode != 1) {
+            try {
+                file.parentFile.mkdirs()
+                ImageIO.write(bufferedimage, "png", file)
+            } catch (e: Exception) {
+                e.printStackTrace()
+            }
         }
+        return bufferedimage
     }
 
     /**