Make the LED update API a bit more flexible

KeyboardioFirmware.ino

@@ -46,7 +46,7 @@ void loop() {
     x = 0;
     y = 0;
     scan_matrix();
-    LEDs.update_leds(temporary_keymap == NUMPAD_KEYMAP);
+    LEDs.update_leds(temporary_keymap);
     Keyboard.sendReport();
     Keyboard.releaseAll();
     handle_mouse_movement(x, y);

led_control.cpp

@@ -55,14 +55,14 @@ void LEDControl::set_led_mode(uint8_t mode) {
 
 
 
-void LEDControl::update_leds(uint8_t numlock_enabled) {
-    if (numlock_enabled) {
+void LEDControl::update_leds(uint8_t current_keymap) {
+    if (current_keymap == NUMPAD_KEYMAP) {
         if (led_mode != LED_SPECIAL_MODE_NUMLOCK) {
             stored_led_mode = led_mode;
         }
         led_mode = LED_SPECIAL_MODE_NUMLOCK;
     }
-    if (!numlock_enabled &&
+    if (current_keymap != NUMPAD_KEYMAP &&
             led_mode == LED_SPECIAL_MODE_NUMLOCK
        ) {
         led_mode = stored_led_mode;

led_control.h

@@ -20,7 +20,7 @@ class LEDControl {
   public:
     void next_led_mode();
     void led_bootup();
-    void update_leds(uint8_t numlock_enabled);
+    void update_leds(uint8_t current_keymap);
     void led_type_letter(uint8_t letter);
     void set_led_mode(uint8_t mode);