Loom_Wifi.h

#pragma once
 
#include <WiFi101.h>
#include <WiFiUdp.h>
#include <FlashStorage.h>
 
 
#include "Module.h"
#include "Loom_Manager.h"
#include "../../../Hardware/Loom_BatchSD/Loom_BatchSD.h"
 
enum CommunicationMode{
    CLIENT,         // Connect to a remote router to handle traffic
    AP              // Set the feather itself as an access point
};
 
/* WiFi info struct that will be used to store the WiFi data on flash */
typedef struct {
    bool is_valid;
    char name[100];
    char password[100];
} WifiInfo;
 
class Loom_WIFI : public Module{
    protected:
 
        /* These aren't used with the Wifi manager */
        void measure() override {};                                
 
        // Initialize the device and connect to the network
        void initialize() override;
        void package() override;   
 
        // Reconnect to the network
        void power_up() override;
 
        // Disconnect from the network
        void power_down() override;
    
    public:
 
        Loom_WIFI(Manager& man, CommunicationMode mode, const char* name = "", const char* password = "", int connectionRetries = 5);
 
        Loom_WIFI(Manager& man);
 
        void loadConfigFromJSON(char* json);
 
        WiFiClient& getClient() { return wifiClient; };
 
        WiFiUDP* getUDP() { return new WiFiUDP(); };
 
        bool verifyConnection();
 
        void connect_to_network();
 
        void start_ap();
 
        /* Take in new WiFi information store it to flash and restart the WiFi chip to put the new credentials into effect*/
        void storeNewWiFiCreds(const char* name, const char* password);
 
        IPAddress getIPAddress();
 
        IPAddress getSubnetMask();
 
        IPAddress getGateway();
 
        IPAddress getBroadcast();
 
 
        bool isConnected();
 
        void useMax() {usingMax = true; };
 
        void setBatchSD(Loom_BatchSD& batch) { batchSD = &batch; };
 
        void setMaxRetries(int retries) { connectionRetries = retries; };
 
        void ipToString(IPAddress ip, char array[16]) { 
            snprintf(array, 16, "%u.%u.%u.%u", ip[0], ip[1], ip[2], ip[3]);
        };
 
    private:
        Manager* manInst;                   // Pointer to the manager
 
        WiFiClient wifiClient;              // Wifi client that can be used with the MQTT client or other additional objects
        Loom_BatchSD* batchSD = nullptr;
 
        bool powerUp = true;                // Whether or not the WiFi should power up (used with batch uploads)
 
        char wifi_name[100];                // Access point to connect to
        char wifi_password[100];          // Password to connect to the access point
        int connectionRetries;
 
        bool usingMax = false;              // If we are using max
        bool firstInit = true;
        CommunicationMode mode;             // Current WiFi mode we are in
 
        IPAddress remoteIP;                 // IP address to send the UDP requests to 
 
        
};