From 501eb2ffae75d380da40bae940343ca624bb817b Mon Sep 17 00:00:00 2001
From: Siwat Sirichai <siwat@siwatinc.com>
Date: Sun, 19 May 2024 13:07:16 +0700
Subject: [PATCH] factory reset function

---
 .../lib/ESPMegaPRO/ESPMegaCommon.hpp          |  2 +-
 .../lib/ESPMegaPRO/ESPMegaProOS.cpp           | 36 ++++++++++++++++++-
 2 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaCommon.hpp b/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaCommon.hpp
index 2a68286..a724426 100644
--- a/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaCommon.hpp
+++ b/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaCommon.hpp
@@ -1,3 +1,3 @@
 #pragma once
 
-#define SDK_VESRION "2.6.0"
\ No newline at end of file
+#define SDK_VESRION "2.6.1"
\ No newline at end of file
diff --git a/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaProOS.cpp b/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaProOS.cpp
index 6b47667..47d85b5 100644
--- a/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaProOS.cpp
+++ b/ESPMegaPRO-OS-SDK/lib/ESPMegaPRO/ESPMegaProOS.cpp
@@ -27,7 +27,7 @@ bool ESPMegaPRO::begin()
     fram.begin(FRAM_ADDRESS);
     Serial.begin(115200);
     this->installCard(1, &outputs);
-    outputs.bindFRAM(&fram, 0);    
+    outputs.bindFRAM(&fram, 0);
     uint8_t outputPinMap[16] = {8, 9, 10, 11, 12, 13, 14, 15, 0, 1, 2, 3, 4, 5, 6, 7};
     outputs.loadPinMap(outputPinMap);
     outputs.loadFromFRAM();
@@ -40,6 +40,40 @@ bool ESPMegaPRO::begin()
     }
     uint8_t inputPinMap[16] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 15, 14, 13, 12};
     inputs.loadPinMap(inputPinMap);
+    // Detect GPIO2 Reset
+    gpio_num_t buttonPin = GPIO_NUM_2;
+    gpio_pad_select_gpio(buttonPin);
+    gpio_set_direction(buttonPin, GPIO_MODE_INPUT);
+    gpio_set_pull_mode(buttonPin, GPIO_PULLUP_ONLY);
+    if (gpio_get_level(buttonPin) == 0)
+    {
+        ESP_LOGW("ESPMegaPRO", "GPIO2 is low, if this condition persists for 5 more seconds, the device will factory reset");
+        bool shouldReset = true;
+        for (int i = 0; i < 50; i++)
+        {
+            vTaskDelay(100 / portTICK_PERIOD_MS);
+            if (gpio_get_level(buttonPin) == 1)
+            {
+                ESP_LOGI("ESPMegaPRO", "Reset condition cleared, Continuing boot process");
+                shouldReset = false;
+                break;
+            }
+        }
+        if (shouldReset)
+        {
+            ESP_LOGW("ESPMegaPRO", "Reset condition met, Factory Resetting device");
+            for (int i = 0; i < fram.getSizeBytes(); i++)
+            {
+                if (i % 1024 == 0)
+                    ESP_LOGV("ESPMegaPRO", "Clearing FRAM Address %d", i);
+                fram.write8(i, 0);
+            }
+            ESP_LOGI("ESPMegaPRO", "Factory Reset Complete");
+            ESP_LOGI("ESPMegaPRO", "Rebooting device");
+            esp_restart();
+        }
+    }
+
     return true;
 }