Mouse bindings

src/main/kotlin/com/lambda/client/event/ForgeEventProcessor.kt

@@ -22,6 +22,7 @@ import net.minecraftforge.fml.common.gameevent.InputEvent
 import net.minecraftforge.fml.common.gameevent.TickEvent
 import net.minecraftforge.fml.common.network.FMLNetworkEvent
 import org.lwjgl.input.Keyboard
+import org.lwjgl.input.Mouse
 
 internal object ForgeEventProcessor {
     private val mc = Wrapper.minecraft
@@ -82,6 +83,13 @@ internal object ForgeEventProcessor {
         ModuleManager.onBind(Keyboard.getEventKey())
     }
 
+    @SubscribeEvent
+    fun onEventMouse(event: InputEvent.MouseInputEvent) {
+        LambdaEventBus.post(event)
+        if (!Mouse.getEventButtonState()) return
+        ModuleManager.onMouseBind(Mouse.getEventButton() + 1)
+    }
+
     @SubscribeEvent(priority = EventPriority.HIGHEST)
     fun onChatSent(event: ClientChatEvent) {
         MessageDetection.Command.BARITONE.removedOrNull(event.message)?.let {
@@ -94,11 +102,6 @@ internal object ForgeEventProcessor {
         }
     }
 
-    @SubscribeEvent
-    fun onEventMouse(event: InputEvent.MouseInputEvent) {
-        LambdaEventBus.post(event)
-    }
-
     /**
      * Includes events of subclasses like ChunkEvent and GetCollisionBoxesEvent
      */

src/main/kotlin/com/lambda/client/gui/rgui/component/BindButton.kt

@@ -4,6 +4,7 @@ import com.lambda.client.module.modules.client.ClickGUI
 import com.lambda.client.module.modules.client.CustomFont
 import com.lambda.client.module.modules.client.GuiColors
 import com.lambda.client.setting.settings.impl.other.BindSetting
+import com.lambda.client.util.Bind.Companion.minMouseIndex
 import com.lambda.client.util.graphics.VertexHelper
 import com.lambda.client.util.graphics.font.FontRenderAdapter
 import com.lambda.client.util.math.Vec2f
@@ -20,6 +21,11 @@ class BindButton(
 
     override fun onRelease(mousePos: Vec2f, buttonId: Int) {
         super.onRelease(mousePos, buttonId)
+        if (listening && buttonId >= minMouseIndex) {
+            setting.value.apply {
+                setMouseBind(buttonId + 1)
+            }
+        }
         listening = !listening
     }
 

src/main/kotlin/com/lambda/client/module/ModuleManager.kt

@@ -64,9 +64,15 @@ object ModuleManager : AsyncLoader<List<Class<out AbstractModule>>> {
 
     internal fun onBind(eventKey: Int) {
         if (Keyboard.isKeyDown(Keyboard.KEY_F3)) return  // if key is the 'none' key (stuff like mod key in i3 might return 0)
-        for (module in modules) {
-            if (module.bind.value.isDown(eventKey)) module.toggle()
-        }
+        modules.filter {
+            it.bind.value.isDown(eventKey)
+        }.forEach { it.toggle() }
+    }
+
+    internal fun onMouseBind(eventMouse: Int) {
+        modules.filter {
+            it.bind.value.isMouseDown(eventMouse)
+        }.forEach { it.toggle() }
     }
 
     fun getModuleOrNull(moduleName: String?) = moduleName?.let { moduleSet[it] }

src/main/kotlin/com/lambda/client/setting/settings/impl/other/BindSetting.kt

@@ -14,7 +14,7 @@ class BindSetting(
     description: String = ""
 ) : ImmutableSetting<Bind>(name, value, visibility, { _, input -> input }, description, unit = "") {
 
-    override val defaultValue: Bind = Bind(TreeSet(value.modifierKeys), value.key)
+    override val defaultValue: Bind = Bind(TreeSet(value.modifierKeys), value.key, null)
 
     override fun resetValue() {
         value.setBind(defaultValue.modifierKeys, defaultValue.key)
@@ -25,6 +25,12 @@ class BindSetting(
             value.clear()
             return
         }
+        if (valueIn.startsWith("Mouse", ignoreCase = true)) {
+            valueIn.split("Mouse").lastOrNull()?.toIntOrNull()?.let {
+                value.setMouseBind(it)
+            }
+            return
+        }
 
         val splitNames = valueIn.split('+')
         val lastKey = KeyboardUtils.getKey(splitNames.last())

src/main/kotlin/com/lambda/client/util/Bind.kt

@@ -5,20 +5,23 @@ import java.util.*
 
 class Bind(
     modifierKeysIn: TreeSet<Int>,
-    keyIn: Int
+    keyIn: Int,
+    mouseIn: Int?
 ) {
 
     constructor() : this(0)
 
-    constructor(key: Int) : this(TreeSet(keyComparator), key)
+    constructor(key: Int) : this(TreeSet(keyComparator), key, null)
 
-    constructor(vararg modifierKeys: Int, key: Int) : this(TreeSet(keyComparator).apply { modifierKeys.forEach { add(it) } }, key)
+    constructor(vararg modifierKeys: Int, key: Int) : this(TreeSet(keyComparator).apply { modifierKeys.forEach { add(it) } }, key, null)
 
     val modifierKeys = modifierKeysIn
     var key = keyIn; private set
+    var mouseKey = mouseIn; private set
+
     private var cachedName = getName()
 
-    val isEmpty get() = key !in 1..255
+    val isEmpty get() = key !in 1..255 && compareValues(mouseKey, minMouseIndex) < 0
 
     fun isDown(eventKey: Int): Boolean {
         return eventKey != 0
@@ -27,6 +30,12 @@ class Bind(
             && synchronized(this) { modifierKeys.all { isModifierKeyDown(eventKey, it) } }
     }
 
+    fun isMouseDown(eventKey: Int): Boolean {
+        return eventKey > minMouseIndex
+            && !isEmpty
+            && mouseKey == (eventKey)
+    }
+
     private fun isModifierKeyDown(eventKey: Int, modifierKey: Int) =
         eventKey != modifierKey
             && when (modifierKey) {
@@ -62,11 +71,21 @@ class Bind(
         setBind(cache, keyIn)
     }
 
+    fun setMouseBind(mouseIn: Int) {
+        synchronized(this) {
+            modifierKeys.clear()
+            key = 0
+            mouseKey = mouseIn
+            cachedName = getName()
+        }
+    }
+
     fun setBind(modifierKeysIn: Collection<Int>, keyIn: Int) {
         synchronized(this) {
             modifierKeys.clear()
             modifierKeys.addAll(modifierKeysIn)
             key = keyIn
+            mouseKey = null
             cachedName = getName()
         }
     }
@@ -75,6 +94,7 @@ class Bind(
         synchronized(this) {
             modifierKeys.clear()
             key = 0
+            mouseKey = null
             cachedName = getName()
         }
     }
@@ -88,18 +108,25 @@ class Bind(
             "None"
         } else {
             StringBuilder().run {
-                for (key in modifierKeys) {
-                    val name = modifierName[key] ?: KeyboardUtils.getDisplayName(key) ?: continue
-                    append(name)
-                    append('+')
+                mouseKey?.let {
+                    if (it > minMouseIndex) append("Mouse$mouseKey")
+                } ?: run {
+                    for (key in modifierKeys) {
+                        val name = modifierName[key] ?: KeyboardUtils.getDisplayName(key) ?: continue
+                        append(name)
+                        append('+')
+                    }
+                    append(KeyboardUtils.getDisplayName(key))
                 }
-                append(KeyboardUtils.getDisplayName(key))
+
                 toString()
             }
         }
     }
 
     companion object {
+        const val minMouseIndex: Int = 2 // middle click button index. Button number = index + 1.
+
         private val modifierName: Map<Int, String> = hashMapOf(
             Keyboard.KEY_LCONTROL to "Ctrl",
             Keyboard.KEY_RCONTROL to "Ctrl",

src/main/kotlin/com/lambda/client/util/KeyboardUtils.kt

@@ -36,7 +36,7 @@ object KeyboardUtils {
     fun sendUnknownKeyError(bind: String) {
         MessageSendHelper.sendErrorMessage("Unknown key [${formatValue(bind)}]! " +
             "Right shift is ${formatValue("rshift")}, " +
-            "left Control is ${formatValue("lcontrol")}, " +
+            "left control is ${formatValue("lcontrol")}, " +
             "and ` is ${formatValue("grave")}. " +
             "You cannot bind the ${formatValue("meta")} key."
         )