From 7cfd369ffb8cb5ddfe2e88d10d6d648f8a9db60a Mon Sep 17 00:00:00 2001
From: nanahi <130121847+na-na-hi@users.noreply.github.com>
Date: Sun, 31 Mar 2024 15:55:33 -0400
Subject: [PATCH] terminal-win: support mouse input events

The mouse input information is available from the INPUT_RECORD
with MOUSE_EVENT event type.
---
 osdep/terminal-win.c | 97 ++++++++++++++++++++++++++++++++------------
 1 file changed, 71 insertions(+), 26 deletions(-)

diff --git a/osdep/terminal-win.c b/osdep/terminal-win.c
index 2846463038..435354155a 100644
--- a/osdep/terminal-win.c
+++ b/osdep/terminal-win.c
@@ -159,35 +159,80 @@ static void read_input(HANDLE in)
             break;
 
         // Only key-down events are interesting to us
-        if (event.EventType != KEY_EVENT)
-            continue;
-        KEY_EVENT_RECORD *record = &event.Event.KeyEvent;
-        if (!record->bKeyDown)
-            continue;
+        switch (event.EventType)
+        {
+        case KEY_EVENT: {
+            KEY_EVENT_RECORD *record = &event.Event.KeyEvent;
+            if (!record->bKeyDown)
+                continue;
 
-        UINT vkey = record->wVirtualKeyCode;
-        bool ext = record->dwControlKeyState & ENHANCED_KEY;
+            UINT vkey = record->wVirtualKeyCode;
+            bool ext = record->dwControlKeyState & ENHANCED_KEY;
 
-        int mods = 0;
-        if (record->dwControlKeyState & (LEFT_ALT_PRESSED | RIGHT_ALT_PRESSED))
-            mods |= MP_KEY_MODIFIER_ALT;
-        if (record->dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
-            mods |= MP_KEY_MODIFIER_CTRL;
-        if (record->dwControlKeyState & SHIFT_PRESSED)
-            mods |= MP_KEY_MODIFIER_SHIFT;
+            int mods = 0;
+            if (record->dwControlKeyState & (LEFT_ALT_PRESSED | RIGHT_ALT_PRESSED))
+                mods |= MP_KEY_MODIFIER_ALT;
+            if (record->dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
+                mods |= MP_KEY_MODIFIER_CTRL;
+            if (record->dwControlKeyState & SHIFT_PRESSED)
+                mods |= MP_KEY_MODIFIER_SHIFT;
 
-        int mpkey = mp_w32_vkey_to_mpkey(vkey, ext);
-        if (mpkey) {
-            mp_input_put_key(input_ctx, mpkey | mods);
-        } else {
-            // Only characters should be remaining
-            int c = record->uChar.UnicodeChar;
-            // The ctrl key always produces control characters in the console.
-            // Shift them back up to regular characters.
-            if (c > 0 && c < 0x20 && (mods & MP_KEY_MODIFIER_CTRL))
-                c += (mods & MP_KEY_MODIFIER_SHIFT) ? 0x40 : 0x60;
-            if (c >= 0x20)
-                mp_input_put_key(input_ctx, c | mods);
+            int mpkey = mp_w32_vkey_to_mpkey(vkey, ext);
+            if (mpkey) {
+                mp_input_put_key(input_ctx, mpkey | mods);
+            } else {
+                // Only characters should be remaining
+                int c = record->uChar.UnicodeChar;
+                // The ctrl key always produces control characters in the console.
+                // Shift them back up to regular characters.
+                if (c > 0 && c < 0x20 && (mods & MP_KEY_MODIFIER_CTRL))
+                    c += (mods & MP_KEY_MODIFIER_SHIFT) ? 0x40 : 0x60;
+                if (c >= 0x20)
+                    mp_input_put_key(input_ctx, c | mods);
+            }
+            break;
+        }
+        case MOUSE_EVENT: {
+            MOUSE_EVENT_RECORD *record = &event.Event.MouseEvent;
+            int mods = 0;
+            if (record->dwControlKeyState & (LEFT_ALT_PRESSED | RIGHT_ALT_PRESSED))
+                mods |= MP_KEY_MODIFIER_ALT;
+            if (record->dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
+                mods |= MP_KEY_MODIFIER_CTRL;
+            if (record->dwControlKeyState & SHIFT_PRESSED)
+                mods |= MP_KEY_MODIFIER_SHIFT;
+
+            switch (record->dwEventFlags) {
+            case MOUSE_MOVED: {
+                int w = 0, h = 0;
+                if (get_font_size(&w, &h)) {
+                    mp_input_set_mouse_pos(input_ctx, w * (record->dwMousePosition.X + 0.5),
+                                                      h * (record->dwMousePosition.Y + 0.5));
+                }
+                break;
+            }
+            case MOUSE_HWHEELED: {
+                int button = (int16_t)HIWORD(record->dwButtonState) > 0 ? MP_WHEEL_RIGHT : MP_WHEEL_LEFT;
+                mp_input_put_key(input_ctx, button | mods);
+                break;
+            }
+            case MOUSE_WHEELED: {
+                int button = (int16_t)HIWORD(record->dwButtonState) > 0 ? MP_WHEEL_UP : MP_WHEEL_DOWN;
+                mp_input_put_key(input_ctx, button | mods);
+                break;
+            }
+            default: {
+                int left_button_state = record->dwButtonState & FROM_LEFT_1ST_BUTTON_PRESSED ?
+                                        MP_KEY_STATE_DOWN : MP_KEY_STATE_UP;
+                mp_input_put_key(input_ctx, MP_MBTN_LEFT | mods | left_button_state);
+                int right_button_state = record->dwButtonState & RIGHTMOST_BUTTON_PRESSED ?
+                                        MP_KEY_STATE_DOWN : MP_KEY_STATE_UP;
+                mp_input_put_key(input_ctx, MP_MBTN_RIGHT | mods | right_button_state);
+                break;
+            }
+            }
+            break;
+        }
         }
     }
 }