Arduino_NetworkConfigurator.h

/*
  Copyright (c) 2024 Arduino SA

  This Source Code Form is subject to the terms of the Mozilla Public
  License, v. 2.0. If a copy of the MPL was not distributed with this
  file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/

#pragma once
#include "ANetworkConfigurator_Config.h"
#if NETWORK_CONFIGURATOR_COMPATIBLE

#include "Arduino.h"
#include "Arduino_ConnectionHandler.h"
#include "configuratorAgents/AgentsManager.h"
#include <settings/settings.h>
#include <Arduino_TimedAttempt.h>
#include <Arduino_KVStore.h>
#include "utility/ResetInput.h"
#include "utility/LEDFeedback.h"

/**
 * @enum NetworkConfiguratorStates
 * @brief Represents the various states of the NetworkConfiguratorClass.
 *
 * - ZERO_TOUCH_CONFIG: Automatically configures the network settings without user intervention.
 * - READ_STORED_CONFIG: Reads the stored network configuration from persistent storage and loads to the ConnectionHandler.
 * - WAITING_FOR_CONFIG: Waits for the user to provide network configuration settings.
 * - CONNECTING: Attempts to connect to the network using the provided configuration.
 * - CONFIGURED: Indicates that the ConnectionHanlder has been successfully configured.
 * - UPDATING_CONFIG: Updates the network configuration.
 * - ERROR: Represents an error state during the configuration process.
 * - END: Marks the end of the network configuration process.
 */
enum class NetworkConfiguratorStates { ZERO_TOUCH_CONFIG,
                                       READ_STORED_CONFIG,
                                       WAITING_FOR_CONFIG,
                                       CONNECTING,
                                       CONFIGURED,
                                       UPDATING_CONFIG,
                                       ERROR,
                                       END };

/**
 * @class NetworkConfiguratorClass
 * @brief Class responsible for managing the network configuration of an Arduino_ConnectionHandler instance.
 * The library provides a way to configure and update
 * the network settings using different interfaces like BLE, Serial, etc.
 * The NetworkConfigurator library stores the provided network settings in a persistent
 * key-value storage (KVStore) and uses the ConnectionHandler library to connect to the network.
 * At startup the NetworkConfigurator library reads the stored network settings from the storage
 * and loads them into the ConnectionHandler object.
 *
 * The NetworkConfigurator library provides a way for wiping out the stored network settings and forcing
 * the restart of the BLE interface if turned off.
 * Reconfiguration procedure:
 * - Arduino Opta: press and hold the user button (BTN_USER) until the led (LED_USER) turns off
 * - Arduino MKR WiFi 1010: short the pin 7 to GND until the led turns off
 * - Arduino GIGA R1 WiFi: short the pin 7 to GND until the led turns off
 * - Arduino Nano RP2040 Connect: short the pin 2 to 3.3V until the led turns off
 * - Other boards: short the pin 2 to GND until the led turns off
 *
 */
class NetworkConfiguratorClass {
public:
  /**
   * @brief Constructor for the NetworkConfiguratorClass.
   * @param connectionHandler Reference to a ConnectionHandler object.
   */
  NetworkConfiguratorClass(ConnectionHandler &connectionHandler);

  /**
   * @brief Initializes the network configurator and starts the interfaces
   * for configuration.
   * @return True if initialization is successful, false otherwise.
   */
  bool begin();

  /**
   * @brief Polls the current state of the network configurator.
   * @return The current state as a NetworkConfiguratorStates enum.
   */
  NetworkConfiguratorStates update();

  /**
   * @brief Resets the stored network configuration.
   * This method is an alternative to the reconfiguration procedure
   * without forcing the restart of the BLE interface if turned off.
   * @return True if the reset is successful, false otherwise.
   */
  bool resetStoredConfiguration();

  /**
   * @brief Ends the network configurator session and stops the
   * interfaces for configuration.
   * @return True if the session ends successfully, false otherwise.
   */
  bool end();

  /**
   * @brief Scans for available network options.
   * @return True if the scan is successful, false otherwise.
   */
  bool scanNetworkOptions();

  /**
   * @brief Sets the storage backend for network configuration.
   * @param kvstore Reference to a KVStore object.
   */
  void setStorage(KVStore &kvstore);

  /**
   * @brief Sets the pin used for the reconfiguration procedure.
   * This must be set before calling the begin() method.
   * Use the value DISABLE_PIN to disable the reset pin and
   * the reset procedure.
   * @param pin The pin number to be used for reconfiguration,
   * internally it's mapped to an interrupt with INPUT_PULLUP mode.
   * The pin must be in the list of digital pins usable for interrupts.
   * Please refer to the Arduino documentation for more details:
   * https://docs.arduino.cc/language-reference/en/functions/external-interrupts/attachInterrupt/
   */
  void setReconfigurePin(int pin);

  /**
   * @brief Adds a callback function to be triggered every time the
   * interrupt on the reconfiguration pin is fired.
   * @param callback Pointer to the callback function.
   */
  void addReconfigurePinCallback(ResetInput::ResetInputCallback callback);

  /**
   * @brief Checks if a specific configuration agent is enabled.
   * @param type The type of the agent to check (e.g., BLE, Serial).
   * @return True if the agent is enabled, false otherwise.
   */
  bool isAgentEnabled(ConfiguratorAgent::AgentTypes type);

  /**
   * @brief Enables or disables a specific configuration agent.
   * @param type The type of the agent to enable or disable (e.g., BLE, Serial).
   * @param enable True to enable the agent, false to disable it.
   */
  void enableAgent(ConfiguratorAgent::AgentTypes type, bool enable);

  /**
   * @brief Disconnects the current configuration agent from the peer.
   */
  void disconnectAgent();

  /**
   * @brief Adds a new configurator agent.
   * Configurator agents handle the configuration interfaces such as BLE, Serial, etc.
   * This method should be called before the begin() method.
   * @param agent Reference to a ConfiguratorAgent object.
   * @return True if the agent is added successfully, false otherwise.
   */
  bool addAgent(ConfiguratorAgent &agent);

private:
  NetworkConfiguratorStates _state;
  ConnectionHandler *_connectionHandler;
  static inline models::NetworkSetting _networkSetting;
  bool _connectionHandlerIstantiated;
  ResetInput *_resetInput;
  LEDFeedbackClass *_ledFeedback;
  /* Timeout instances */
  // Timeout for connection attempt
  TimedAttempt _connectionTimeout;
  // Timeout for retrying to connect using the provided credentials
  TimedAttempt _connectionRetryTimer;
  // Timeout for updating the network options ex. periodically scanning for new WiFi networks
  TimedAttempt _optionUpdateTimer;
  /* List of events the NetworkConfigurator can handle from the AgentsManager */
  enum class NetworkConfiguratorEvents { NONE,
                                         SCAN_REQ,
                                         CONNECT_REQ,
                                         NEW_NETWORK_SETTINGS,
                                         GET_WIFI_FW_VERSION,
                                         GET_NET_CONF_LIB_VERSION };
  static inline NetworkConfiguratorEvents _receivedEvent;

  enum class ConnectionResult { SUCCESS,
                                FAILED,
                                IN_PROGRESS };

  KVStore *_kvstore;
  AgentsManagerClass *_agentsManager;

  /* FSM handler functions */
#if ZERO_TOUCH_ENABLED
  NetworkConfiguratorStates handleZeroTouchConfig();
#endif
  NetworkConfiguratorStates handleReadStorage();
  NetworkConfiguratorStates handleWaitingForConf();
  NetworkConfiguratorStates handleConnecting();
  NetworkConfiguratorStates handleConfigured();
  NetworkConfiguratorStates handleUpdatingConfig();
  NetworkConfiguratorStates handleErrorState();
  bool handleConnectRequest();
  void handleGetWiFiFWVersion();
  void handleGetNetConfLibVersion();

  void startReconfigureProcedure();

  String decodeConnectionErrorMessage(NetworkConnectionState err, StatusMessage *errorCode);
  ConnectionResult connectToNetwork(StatusMessage *err);
  ConnectionResult disconnectFromNetwork();
  bool sendStatus(StatusMessage msg);
  static void printNetworkSettings();
#ifdef BOARD_HAS_ETHERNET
  void defaultEthernetSettings();
#endif
#ifdef BOARD_HAS_WIFI
  bool scanWiFiNetworks(WiFiOption &wifiOptObj);
  bool insertWiFiAP(WiFiOption &wifiOptObj, char *ssid, int rssi);
#endif
  /* Callback for agentsManager */
  static void scanReqHandler();
  static void connectReqHandler();
  static void setNetworkSettingsHandler(models::NetworkSetting *netSetting);
  static void getWiFiFWVersionHandler();
  static void getNetConfLibVersionHandler();
};

#endif // NETWORK_CONFIGURATOR_COMPATIBLE