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

Author: andig

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;

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;

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");