From dff933b16b6d52f1b135a721e97c09e5b1d93dab Mon Sep 17 00:00:00 2001
From: Spacehuhn <github@spacehuhn.de>
Date: Sun, 17 Jul 2022 01:43:15 +0200
Subject: [PATCH] Initial Delay setting

---
 USBNova.ino                     |  2 +-
 src/preferences/preferences.cpp | 12 ++++++++++++
 src/preferences/preferences.h   |  2 ++
 3 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/USBNova.ino b/USBNova.ino
index 3ac8fac..b3d1c43 100644
--- a/USBNova.ino
+++ b/USBNova.ino
@@ -74,7 +74,7 @@ void setup() {
 
     // Start attack
     if (selector::mode() == ATTACK && !preferences::getRunOnIndicator()) {
-        delay(1000); // Wait 1s to give the computer time to initialize the keyboard
+        delay(preferences::getInitialDelay()); // Wait to give computer time to init keyboard
         attack::start(); // Start keystroke injection attack
     }
 
diff --git a/src/preferences/preferences.cpp b/src/preferences/preferences.cpp
index 6a6b3ba..e36762f 100644
--- a/src/preferences/preferences.cpp
+++ b/src/preferences/preferences.cpp
@@ -47,6 +47,8 @@ namespace preferences {
     bool disable_capslock { true };
     bool run_on_indicator { false };
 
+    int initial_delay { 1000 };
+
     // Array help functions
     void add_array(JsonDocument& doc, const char* name, int* array, int size) {
         JsonArray jarr = doc.createNestedArray(name);
@@ -86,6 +88,8 @@ namespace preferences {
 
         root["disable_capslock"] = disable_capslock;
         root["run_on_indicator"]  = run_on_indicator;
+
+        root["initial_delay"] = initial_delay;
     }
 
     // ======== PUBLIC ======== //
@@ -133,6 +137,8 @@ namespace preferences {
 
         if (!config_doc.containsKey("disable_capslock")) config_doc["disable_capslock"] = disable_capslock;
         if (!config_doc.containsKey("run_on_indicator")) config_doc["run_on_indicator"] = run_on_indicator;
+        
+        if (!config_doc.containsKey("initial_delay")) config_doc["initial_delay"] = initial_delay;
 
         // === Fetch values === //
         enable_msc = config_doc["enable_msc"].as<bool>();
@@ -163,6 +169,8 @@ namespace preferences {
 
         disable_capslock = config_doc["disable_capslock"].as<bool>();
         run_on_indicator  = config_doc["run_on_indicator"].as<bool>();
+        
+        initial_delay  = config_doc["initial_delay"].as<int>();
     }
 
     void save() {
@@ -257,4 +265,8 @@ namespace preferences {
     bool getRunOnIndicator() {
         return run_on_indicator;
     }
+    
+    int getInitialDelay() {
+        return initial_delay;
+    }
 }
\ No newline at end of file
diff --git a/src/preferences/preferences.h b/src/preferences/preferences.h
index 686e367..3c4d1eb 100644
--- a/src/preferences/preferences.h
+++ b/src/preferences/preferences.h
@@ -33,4 +33,6 @@ namespace preferences {
     
     bool getDisableCapslock();
     bool getRunOnIndicator();
+
+    int getInitialDelay();
 }
\ No newline at end of file