prompt before restarting

1 files changed, 5 insertions, 5 deletions

diff --git a/firmware/buttons.cpp b/firmware/buttons.cpp
index d28c4cc..27ce40b 100644
--- a/firmware/buttons.cpp
+++ b/firmware/buttons.cpp
@@ -1,10 +1,6 @@
 #include "buttons.h"
 #include <Arduino.h>
-
-#define BTN_UP 4
-#define BTN_DOWN 5
-#define BTN_SELECT 6
-#define BTN_BACK 7
+#include "config.h"
 
 void buttonsInit()
 {
@@ -12,9 +8,13 @@ void buttonsInit()
     pinMode(BTN_DOWN, INPUT_PULLUP);
     pinMode(BTN_SELECT, INPUT_PULLUP);
     pinMode(BTN_BACK, INPUT_PULLUP);
+    pinMode(BTN_RIGHT, INPUT_PULLUP);
+    pinMode(BTN_LEFT, INPUT_PULLUP);
 }
 
 bool btnUp() { return !digitalRead(BTN_UP); }
 bool btnDown() { return !digitalRead(BTN_DOWN); }
 bool btnSelect() { return !digitalRead(BTN_SELECT); }
 bool btnBack() { return !digitalRead(BTN_BACK); }
+bool btnRight() { return !digitalRead(BTN_RIGHT); }
+bool btnLeft() { return !digitalRead(BTN_LEFT); }