add mouse keycodes

Author: xiongyihui

keyboard/__init__.py

@@ -110,7 +110,7 @@ def __init__(self, keymap=(), pairs=(), verbose=True):
         self.adv_timeout = None
 
         size = 4 + self.matrix.keys
-        self.data = array.array("L", microcontroller.nvm[:size*4])
+        self.data = array.array("L", microcontroller.nvm[: size * 4])
         if self.data[0] != 0x424B5950:
             self.data[0] = 0x424B5950
             self.data[1] = 1
@@ -415,6 +415,33 @@ def send_consumer(self, keycode):
         except Exception as e:
             print(e)
 
+    def press_mouse(self, buttons):
+        try:
+            if self.usb_status == 0x3 and usb_is_connected():
+                self.usb_hid.press_mouse(buttons)
+            elif self.ble.connected:
+                self.ble_hid.press_mouse(buttons)
+        except Exception as e:
+            print(e)
+
+    def release_mouse(self, buttons):
+        try:
+            if self.usb_status == 0x3 and usb_is_connected():
+                self.usb_hid.release_mouse(buttons)
+            elif self.ble.connected:
+                self.ble_hid.release_mouse(buttons)
+        except Exception as e:
+            print(e)
+
+    def move_mouse(self, x=0, y=0, wheel=0):
+        try:
+            if self.usb_status == 0x3 and usb_is_connected():
+                self.usb_hid.move_mouse(x, y, wheel)
+            elif self.ble.connected:
+                self.ble_hid.move_mouse(x, y, wheel)
+        except Exception as e:
+            print(e)
+
     def get(self):
         event = self.matrix.get()
         key = event & 0x7F
@@ -432,10 +459,19 @@ def run(self):
         keys = [0] * matrix.keys
         ms = matrix.ms
         last_time = 0
+        mouse_action = 0
+        mouse_time = 0
         while True:
-            t = 20 if self.backlight.check() else 1000
+            t = 20 if self.backlight.check() or mouse_action else 1000
             n = matrix.wait(t)
             self.check()
+
+            if mouse_action:
+                x, y, wheel = MS_MOVEMENT[mouse_action]
+                dt = 1 + (time.monotonic_ns() - mouse_time) // 8000000
+                mouse_time = time.monotonic_ns()
+                self.move_mouse(x * dt, y * dt, wheel * dt)
+
             if n == 0:
                 continue
 
@@ -495,8 +531,11 @@ def run(self):
                             if action_code & 0x400:
                                 self.send_consumer(action_code & 0x3FF)
                         elif kind == ACT_MOUSEKEY:
-                            # todo
-                            pass
+                            if action_code & 0xF00 == 0:
+                                self.press_mouse(action_code & 0xF)
+                            else:
+                                mouse_action = (action_code >> 8) & 0xF
+                                mouse_time = time.monotonic_ns()
                         elif kind == ACT_LAYER_TAP or kind == ACT_LAYER_TAP_EXT:
                             layer = (action_code >> 8) & 0x1F
                             mask = 1 << layer
@@ -599,7 +638,11 @@ def run(self):
                             if action_code & 0x400:
                                 self.send_consumer(0)
                         elif kind == ACT_MOUSEKEY:
-                            pass
+                            if action_code & 0xF00 == 0:
+                                self.release_mouse(action_code & 0xF)
+                            elif (action_code >> 8) & 0xF == mouse_action:
+                                mouse_action = 0
+                                self.move_mouse(0, 0, 0)
                         elif kind == ACT_LAYER_TAP or kind == ACT_LAYER_TAP_EXT:
                             layer = (action_code >> 8) & 0x1F
                             keycode = action_code & 0xFF

keyboard/action_code.py

@@ -545,6 +545,30 @@ def mods_to_keycodes(mods):
 ACTION_USAGE_CONSUMER = lambda n: ACTION(ACT_USAGE, 1 << 10 | (n))
 ACTION_MOUSEKEY = lambda key: ACTION(ACT_MOUSEKEY, key)
 
+
+MS_BTN1 = MOUSEKEY(1 << 0)
+MS_BTN2 = MOUSEKEY(1 << 1)
+MS_BTN3 = MOUSEKEY(1 << 2)
+MS_BTN4 = MOUSEKEY(1 << 3)
+MS_BTN5 = MOUSEKEY(1 << 4)
+MS_UP   = MOUSEKEY(1 << 8)
+MS_DN = MOUSEKEY(2 << 8)
+MS_LT = MOUSEKEY(3 << 8)
+MS_RT = MOUSEKEY(4 << 8)
+MS_UL = MOUSEKEY(5 << 8)
+MS_UR = MOUSEKEY(6 << 8)
+MS_DL = MOUSEKEY(7 << 8)
+MS_DR = MOUSEKEY(8 << 8)
+MS_W_UP = MOUSEKEY(9 << 8)
+MS_W_DN = MOUSEKEY(10 << 8)
+
+MS_MOVEMENT = (
+    (0, 0, 0),
+    (0, -2, 0), (0, 2, 0), (-2, 0, 0), (2, 0, 0),
+    (-1, -1, 0), (1, -1, 0), (-1, 1, 0), (1, 1, 0),
+    (0, 0, 1), (0, 0, -1)
+)
+
 MACRO = lambda n: ACTION(ACT_MACRO, n)
 BACKLIGHT = lambda n: ACTION(ACT_BACKLIGHT, n)
 

keyboard/hid.py

@@ -16,7 +16,7 @@ class HID:
     def __init__(self, devices):
         self.keyboard = find_device(devices, usage_page=0x1, usage=0x06)
         self.consumer_control = find_device(devices, usage_page=0x0C, usage=0x01)
-        # self.mouse = find_device(devices, usage_page=0x1, usage=0x02)
+        self.mouse = find_device(devices, usage_page=0x1, usage=0x02)
 
         self.consumer_report = bytearray(2)
 
@@ -26,6 +26,8 @@ def __init__(self, devices):
         self.report = bytearray(8)
         self.report_keys = memoryview(self.report)[2:]
 
+        self.mouse_report = bytearray(4)
+
         for device in devices:
             if (
                 device.usage_page == 0x1
@@ -75,11 +77,34 @@ def send_consumer(self, keycode):
         struct.pack_into("<H", self.consumer_report, 0, keycode)
         self.consumer_control.send_report(self.consumer_report)
 
+    def press_mouse(self, buttons):
+        self.mouse_report[0] |= buttons
+        self.mouse_report[1] = 0
+        self.mouse_report[2] = 0
+        self.mouse_report[3] = 0
+        self.mouse.send_report(self.mouse_report)
+
+    def release_mouse(self, buttons):
+        self.mouse_report[0] &= ~buttons
+        self.mouse_report[1] = 0
+        self.mouse_report[2] = 0
+        self.mouse_report[3] = 0
+        self.mouse.send_report(self.mouse_report)
+
+    def move_mouse(self, x=0, y=0, wheel=0):
+        self.mouse_report[1] = x & 0xFF
+        self.mouse_report[2] = y & 0xFF
+        self.mouse_report[3] = wheel & 0xFF
+        self.mouse.send_report(self.mouse_report)
+
     def release_all(self):
         self.send_consumer(0)
         for i in range(8):
             self.report[i] = 0
         self.keyboard.send_report(self.report)
+        for i in range(4):
+            self.mouse_report[i] = 0
+        self.mouse.send_report(self.mouse_report)
 
     @property
     def leds(self):