OTA.ino

Go to the documentation of this file.

 
#include <Arduino.h>
#include <FastLED.h>
#include "fl/net/ota.h"
#include "fl/stl/sstream.h"
 
// ====== Configure your transport here ======
#define USE_WIFI         1
 
// ====== Your identifiers ======
static const char* HOSTNAME   = "my-device";
static const char* OTA_PASS   = "supersecret";     // used for both ArduinoIDE OTA (hashed) and Web UI Basic Auth
 
// ====== Wi-Fi creds (only used if USE_WIFI=1 and you call beginWiFi) ======
static const char* WIFI_SSID  = "MySSID";
static const char* WIFI_PASS  = "MyPass";
 
// ====== LED Configuration ======
#define NUM_LEDS 60
#define DATA_PIN 2
CRGB leds[NUM_LEDS];
 
fl::net::OTA ota;
 
void setup() {
  Serial.begin(115200);
  delay(100);
 
  Serial.println("FastLED OTA Example");
  Serial.println("===================");
 
  // Initialize FastLED
  FastLED.addLeds<WS2812, DATA_PIN, GRB>(leds, NUM_LEDS);
  FastLED.setBrightness(50);
 
  // Optional: Enable AP fallback when Wi-Fi STA fails (must be called before beginWiFi)
  // ota.enableApFallback("MyDevice-Setup", "setup-only");
 
#if USE_WIFI
  // Start Wi-Fi (async mode - returns immediately)
  Serial.println("Starting Wi-Fi (async mode)...");
  ota.beginWiFi(HOSTNAME, OTA_PASS, WIFI_SSID, WIFI_PASS);
  Serial.println("Wi-Fi connecting in background...");
  Serial.println("LEDs will show connection status:");
  Serial.println("  - Pulsing blue: Connecting...");
  Serial.println("  - Solid green: Connected!");
 
  // Check for service initialization failures
  uint8_t failed = ota.getFailedServices();
  if (failed != 0) {
    Serial.println("\nWARNING: Some OTA services failed to initialize:");
    if (failed & (uint8_t)fl::net::ota::Service::MDNS_FAILED) {
      Serial.println("  - mDNS failed: Device won't be discoverable at hostname.local");
      Serial.println("    Try accessing via IP address instead");
    }
    if (failed & (uint8_t)fl::net::ota::Service::HTTP_FAILED) {
      Serial.println("  - HTTP server failed: Web OTA unavailable");
      Serial.println("    You can still use Arduino IDE OTA (port 3232)");
    }
    if (failed & (uint8_t)fl::net::ota::Service::ARDUINO_OTA_FAILED) {
      Serial.println("  - ArduinoOTA failed: IDE OTA unavailable");
      Serial.println("    You can still use Web OTA interface");
    }
  }
#else
  // If your networking is already up (Wi-Fi or ETH), just start OTA services:
  // For Ethernet users: Call ETH.begin() first, then ota.begin()
  Serial.println("Starting OTA services (network assumed configured)...");
  if (ota.begin(HOSTNAME, OTA_PASS)) {
    Serial.println("OTA services started successfully");
  } else {
    Serial.println("OTA services failed to start");
  }
 
  // Check for service initialization failures
  uint8_t failed = ota.getFailedServices();
  if (failed != 0) {
    Serial.println("\nWARNING: Some OTA services failed to initialize:");
    if (failed & (uint8_t)fl::net::ota::Service::MDNS_FAILED) {
      Serial.println("  - mDNS failed: Device won't be discoverable at hostname.local");
      Serial.println("    Try accessing via IP address instead");
    }
    if (failed & (uint8_t)fl::net::ota::Service::HTTP_FAILED) {
      Serial.println("  - HTTP server failed: Web OTA unavailable");
      Serial.println("    You can still use Arduino IDE OTA (port 3232)");
    }
    if (failed & (uint8_t)fl::net::ota::Service::ARDUINO_OTA_FAILED) {
      Serial.println("  - ArduinoOTA failed: IDE OTA unavailable");
      Serial.println("    You can still use Web OTA interface");
    }
  }
#endif
 
  // Optional callbacks for monitoring OTA progress:
  ota.onProgress([](size_t written, size_t total){
    //Serial.printf("OTA Progress: %u/%u (%d%%)\n", (unsigned)written, (unsigned)total, (int)((written * 100) / total));
    fl::sstream msg;
    msg << "OTA Progress: " << written << "/" << total << " (" << (int)((written * 100) / total) << "%)";
    Serial.println(msg.c_str());
  });
  ota.onError([](const char* error_msg){
    // Serial.printf("OTA error: %s\n", error_msg);
    fl::sstream msg;
    msg << "OTA error: " << error_msg;
    Serial.println(msg.c_str());
  });
  ota.onState([](uint8_t state){
    //Serial.printf("OTA state: %u\n", state);
    fl::sstream msg;
    msg << "OTA state: " << (int)state;
    Serial.println(msg.c_str());
  });
 
  // Callback invoked before device reboots after successful OTA update
  // Use this to save state, turn off LEDs, or perform cleanup
  ota.onBeforeReboot([](){
    Serial.println("OTA complete! Preparing to reboot...");
    Serial.println("Saving state and turning off LEDs...");
 
    // Turn off all LEDs before reboot
    fill_solid(leds, NUM_LEDS, CRGB::Black);
    FastLED.show();
 
    // In a real application, you might:
    // - Save LED configuration to EEPROM/preferences
    // - Save animation state
    // - Close file handles
    // - Notify other systems of impending reboot
  });
 
  Serial.println("Setup complete. Running LED animation...");
}
 
uint8_t hue = 0;
bool wifi_connected_notified = false;
 
void loop() {
  // Service OTA (low overhead: ~73µs when idle)
  ota.poll();
 
#if USE_WIFI
  // Async WiFi mode: Show connection status via LEDs
  if (!ota.isConnected()) {
    // Connecting: Show pulsing blue animation
    uint8_t brightness = beatsin8(60, 30, 255);  // Pulse at 60 BPM
    fill_solid(leds, NUM_LEDS, CRGB::Blue);
    FastLED.setBrightness(brightness);
    FastLED.show();
    delay(20);
    return;  // Skip normal animation while connecting
  } else if (!wifi_connected_notified) {
    // Just connected: Show success and print info
    wifi_connected_notified = true;
    Serial.println("\nWi-Fi connected!");
    Serial.print("Access web interface at: http://");
    Serial.print(HOSTNAME);
    Serial.println(".local/");
 
    // Brief celebration: Green flash
    fill_solid(leds, NUM_LEDS, CRGB::Green);
    FastLED.setBrightness(255);
    FastLED.show();
    delay(500);
 
    // Restore normal brightness for animation
    FastLED.setBrightness(50);
  }
#endif
 
  // Normal LED animation (rainbow)
  fill_rainbow(leds, NUM_LEDS, hue, 7);
  FastLED.show();
 
  hue++;
 
  // Small delay for animation speed
  delay(20);
}