Add getCommand() api for OTA update type (#2259)

esp8266 · Jul 15, 2016 · c4f9f10 · c4f9f10
1 parent 1f7989b
commit c4f9f10
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/libraries/ArduinoOTA/ArduinoOTA.cpp b/libraries/ArduinoOTA/ArduinoOTA.cpp
@@ -180,7 +180,7 @@ void ArduinoOTAClass::_onRx(){
       nonce_md5.add(String(micros()));
       nonce_md5.calculate();
       _nonce = nonce_md5.toString();
-      
+
       char auth_req[38];
       sprintf(auth_req, "AUTH %s", _nonce.c_str());
       _udp_ota->append((const char *)auth_req, strlen(auth_req));
@@ -326,4 +326,8 @@ void ArduinoOTAClass::handle() {
   }
 }
 
+int ArduinoOTAClass::getCommand() {
+  return _cmd;
+}
+
 ArduinoOTAClass ArduinoOTA;
diff --git a/libraries/ArduinoOTA/ArduinoOTA.h b/libraries/ArduinoOTA/ArduinoOTA.h
@@ -40,6 +40,7 @@ class ArduinoOTAClass
     void onProgress(THandlerFunction_Progress fn);
     void begin();
     void handle();
+    int getCommand(); // get update command type after OTA started- either U_FLASH or U_SPIFFS
 
   private:
     int _port;

diff --git a/libraries/ArduinoOTA/examples/BasicOTA/BasicOTA.ino b/libraries/ArduinoOTA/examples/BasicOTA/BasicOTA.ino
@@ -27,7 +27,14 @@ void setup() {
   // ArduinoOTA.setPassword((const char *)"123");
 
   ArduinoOTA.onStart([]() {
-    Serial.println("Start");
+    String type;
+    if (ArduinoOTA.getCommand() == U_FLASH)
+      type = "sketch";
+    else // U_SPIFFS
+      type = "filesystem";
+
+    // NOTE: if updating SPIFFS this would be the place to unmount SPIFFS using SPIFFS.end()
+    Serial.println("Start updating " + type);
   });
   ArduinoOTA.onEnd([]() {
     Serial.println("\nEnd");