Add toggle for live training

2 files changed, 38 insertions, 6 deletions

diff --git a/quantum/oryx.c b/quantum/oryx.c
index 93ee50e935..7f0b60a797 100644
--- a/quantum/oryx.c
+++ b/quantum/oryx.c
@@ -1,18 +1,41 @@
 #include "oryx.h"
 
-bool webusb_state_live_training_enabled;
+bool oryx_state_live_training_enabled;
 
 bool webusb_receive_oryx(uint8_t *data, uint8_t length) {
     uint8_t command = data[0];
+    uint8_t param = data[1];
 
     switch (command) {
         case ORYX_GET_LAYER:
             oryx_layer_event();
             return true; break;
+        case ORYX_CMD_LIVE_TRAINING: {
+            uint8_t event[4];
+            switch (param) { // 0 for state, 1 for off, 2 for on
+                case 0:
+                    break;
+                case 1:
+                    oryx_state_live_training_enabled = false;
+                    break;
+                case 2:
+                    oryx_state_live_training_enabled = true;
+                    break;
+                default:
+                    webusb_error(WEBUSB_STATUS_UNKNOWN_COMMAND);
+                    return true;
+            }
+            event[0] = WEBUSB_STATUS_OK;
+            event[1] = WEBUSB_EVT_PAIRED;
+            event[2] = oryx_state_live_training_enabled;
+            event[3] = WEBUSB_STOP_BIT;
+            webusb_send(event, sizeof(event));
+            return true;
+            }
+        default:
+            return webusb_receive_kb(data, length);
 
     }
-
-    return false;
 }
 
 
@@ -28,3 +51,7 @@ void oryx_layer_event(void) {
     webusb_send(event, sizeof(event));
 #endif
 }
+
+bool is_oryx_live_training_enabled(void) {
+    return oryx_state_live_training_enabled;
+}
diff --git a/quantum/oryx.h b/quantum/oryx.h
index eaef5bd335..13ef72d548 100644
--- a/quantum/oryx.h
+++ b/quantum/oryx.h
@@ -8,12 +8,16 @@
 #endif
 
 #ifndef DYNAMIC_KEYMAP_ENABLE
-#    error "Dynamic Keymaps are not enabled. It must be aenbled "
+#    error "Dynamic Keymaps are not enabled. It must be enabled"
 #endif
 
+// enum Oryx_Status_code {
+//     PLACEHOLDER = WEBUSB_STATUS_SAFE_RANGE,
+// }
 
 enum Oryx_Command_Code {
     ORYX_GET_LAYER = WEBUSB_CMD_SAFE_RANGE,
+    ORYX_CMD_LIVE_TRAINING,
 
 };
 
@@ -21,14 +25,15 @@ enum Oryx_Event_Code {
     ORYX_EVT_KEYDOWN = WEBUSB_EVT_SAFE_RANGE,
     ORYX_EVT_KEYUP,
     ORYX_EVT_LAYER,
+    ORYX_EVT_LIVE_TRAINING
 };
 
 
 
-extern bool webusb_state_live_training_enabled;
+extern bool oryx_state_live_training_enabled;
 
 bool webusb_receive_oryx(uint8_t *data, uint8_t length);
 void oryx_layer_event(void);
-
+bool is_oryx_live_training_enabled(void);
 
 void keyboard_pre_init_oryx(void);