From dbb4152cb28e27d5d26d3cad01124cd04c9eb093 Mon Sep 17 00:00:00 2001
From: Gergely Nagy <algernon@keyboard.io>
Date: Mon, 14 May 2018 22:43:50 +0200
Subject: [PATCH] Updated to use the new plugin API

Signed-off-by: Gergely Nagy <algernon@keyboard.io>
---
 src/Kaleidoscope-Model01-TestMode.cpp | 16 ++++++++++------
 src/Kaleidoscope-Model01-TestMode.h   | 12 +++++++++---
 2 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/src/Kaleidoscope-Model01-TestMode.cpp b/src/Kaleidoscope-Model01-TestMode.cpp
index 7846fa39..fe27c90a 100644
--- a/src/Kaleidoscope-Model01-TestMode.cpp
+++ b/src/Kaleidoscope-Model01-TestMode.cpp
@@ -13,22 +13,26 @@
 
 
 
-TestMode_::TestMode_(void) {
+#if KALEIDOSCOPE_ENABLE_V1_PLUGIN_API
+void TestMode_::begin(void) {
+  Kaleidoscope.useLoopHook(this->legacyLoopHook);
 }
 
+void TestMode_::legacyLoopHook(bool is_post_clear) {
+  if (is_post_clear)
+    return;
 
-void TestMode_::begin(void) {
-  Kaleidoscope.useLoopHook(this->loopHook);
+  TestMode.beforeReportingState();
 }
+#endif
 
-void TestMode_::loopHook(bool postClear) {
-  if (postClear)
-    return;
+kaleidoscope::EventHandlerResult TestMode_::beforeReportingState() {
   if (KeyboardHardware.leftHandState.all == TEST_MODE_KEY_COMBO
 //  && KeyboardHardware.rightHandState.all == combo.rightHand
      ) {
     run_tests();
   }
+  return kaleidoscope::EventHandlerResult::OK;
 }
 
 void TestMode_::waitForKeypress() {
diff --git a/src/Kaleidoscope-Model01-TestMode.h b/src/Kaleidoscope-Model01-TestMode.h
index 1fa7a1ff..5be46c75 100644
--- a/src/Kaleidoscope-Model01-TestMode.h
+++ b/src/Kaleidoscope-Model01-TestMode.h
@@ -12,10 +12,17 @@ typedef struct {
   uint32_t badKeys;
 
 } side_data_t;
-class TestMode_ : public KaleidoscopePlugin {
+class TestMode_ : public kaleidoscope::Plugin {
  public:
-  TestMode_(void);
+  TestMode_(void) {};
+
+  kaleidoscope::EventHandlerResult beforeReportingState();
+
+#if KALEIDOSCOPE_ENABLE_V1_PLUGIN_API
+ protected:
   void begin();
+  static void legacyLoopHook(bool is_post_clear);
+#endif
 
  private:
   static void run_tests();
@@ -24,7 +31,6 @@ class TestMode_ : public KaleidoscopePlugin {
   static void toggle_programming_leds_on();
   static void handleKeyEvent(side_data_t *side, keydata_t *oldState, keydata_t *newState, uint8_t row, uint8_t col, uint8_t col_offset);
   static void waitForKeypress();
-  static void loopHook(bool postClear);
   static void set_leds(cRGB color);
 };