hhkb: Fix for USB NKRO for RN42

keyboard/hhkb/Makefile.rn42

@@ -1,5 +1,5 @@
 TARGET ?= hhkb_rn42
 CONFIG_H ?= config_rn42.h
 RN42_ENABLE ?= yes
-NKRO_ENABLE ?= no
+NKRO_ENABLE ?= yes
 include Makefile

keyboard/hhkb/rn42/main.c

@@ -72,12 +72,6 @@ int main(void)
     /* init modules */
     keyboard_init();
 
-    if (!rn42_rts()) {
-        host_set_driver(&rn42_driver);
-    } else {
-        host_set_driver(&lufa_driver);
-    }
-
 #ifdef SLEEP_LED_ENABLE
     sleep_led_init();
 #endif

keyboard/hhkb/rn42/rn42_task.c

@@ -33,6 +33,9 @@ static void status_led(bool on)
 void rn42_task_init(void)
 {
     battery_init();
+#ifdef NKRO_ENABLE
+    rn42_nkro_last = keyboard_nkro;
+#endif
 }
 
 void rn42_task(void)
@@ -69,16 +72,27 @@ void rn42_task(void)
         }
     }
 
-    /* Bluetooth mode when ready */
-    if (!config_mode && !force_usb) {
-        if (!rn42_rts() && host_get_driver() != &rn42_driver) {
+    /* Switch between USB and Bluetooth */
+    if (!config_mode) { // not switch while config mode
+        if (!force_usb && !rn42_rts()) {
+            if (host_get_driver() != &rn42_driver) {
                 clear_keyboard();
+#ifdef NKRO_ENABLE
+                rn42_nkro_last = keyboard_nkro;
+                keyboard_nkro = false;
+#endif
                 host_set_driver(&rn42_driver);
-        } else if (rn42_rts() && host_get_driver() != &lufa_driver) {
+            }
+        } else {
+            if (host_get_driver() != &lufa_driver) {
                 clear_keyboard();
+#ifdef NKRO_ENABLE
+                keyboard_nkro = rn42_nkro_last;
+#endif
                 host_set_driver(&lufa_driver);
             }
         }
+    }
 
 
     static uint16_t prev_timer = 0;
@@ -339,8 +353,6 @@ bool command_extra(uint8_t code)
             } else {
                 print("USB mode\n");
                 force_usb = true;
-                clear_keyboard();
-                host_set_driver(&lufa_driver);
             }
             return true;
         case KC_DELETE:
@@ -360,6 +372,14 @@ bool command_extra(uint8_t code)
         case KC_SCROLLLOCK:
             init_rn42();
             return true;
+#ifdef NKRO_ENABLE
+        case KC_N:
+            if (host_get_driver() != &lufa_driver) {
+                // ignored unless USB mode
+                return true;
+            }
+            return false;
+#endif
         default:
             if (config_mode)
                 return true;

keyboard/hhkb/rn42/rn42_task.h

@@ -4,6 +4,10 @@
 #include <stdbool.h>
 #include "rn42.h"
 
+#ifdef NKRO_ENABLE
+bool rn42_nkro_last;
+#endif
+
 void rn42_task_init(void);
 void rn42_task(void);