183 lines
5.4 KiB
C++
183 lines
5.4 KiB
C++
#include <Arduino.h>
|
|
#include <functional>
|
|
#include "Globals.h"
|
|
#include "NodeContext.h"
|
|
#include "NetworkManager.h"
|
|
#include "ClusterManager.h"
|
|
#include "ApiServer.h"
|
|
#include "TaskManager.h"
|
|
|
|
using namespace std;
|
|
|
|
class WiFiScannerService {
|
|
public:
|
|
WiFiScannerService(NodeContext& ctx, TaskManager& taskMgr, NetworkManager& networkMgr)
|
|
: ctx(ctx), taskManager(taskMgr), network(networkMgr) {
|
|
registerTasks();
|
|
}
|
|
|
|
void registerApi(ApiServer& api) {
|
|
api.addEndpoint("/api/wifi/scan", HTTP_POST, [this](AsyncWebServerRequest* request) {
|
|
if (network.isWiFiScanning()) {
|
|
JsonDocument resp;
|
|
resp["success"] = false;
|
|
resp["message"] = "WiFi scan already in progress";
|
|
String json;
|
|
serializeJson(resp, json);
|
|
request->send(409, "application/json", json);
|
|
return;
|
|
}
|
|
|
|
network.startWiFiScan();
|
|
|
|
JsonDocument resp;
|
|
resp["success"] = true;
|
|
resp["message"] = "WiFi scan started";
|
|
String json;
|
|
serializeJson(resp, json);
|
|
request->send(200, "application/json", json);
|
|
});
|
|
|
|
api.addEndpoint("/api/wifi/status", HTTP_GET, [this](AsyncWebServerRequest* request) {
|
|
JsonDocument doc;
|
|
doc["scanning"] = network.isWiFiScanning();
|
|
doc["networks_found"] = ctx.wifiAccessPoints->size();
|
|
|
|
JsonArray networks = doc["networks"].to<JsonArray>();
|
|
for (const auto& ap : *ctx.wifiAccessPoints) {
|
|
JsonObject network = networks.add<JsonObject>();
|
|
network["ssid"] = ap.ssid;
|
|
network["rssi"] = ap.rssi;
|
|
network["channel"] = ap.channel;
|
|
network["encryption"] = ap.encryption;
|
|
network["hidden"] = ap.isHidden;
|
|
}
|
|
|
|
String json;
|
|
serializeJson(doc, json);
|
|
request->send(200, "application/json", json);
|
|
});
|
|
}
|
|
|
|
private:
|
|
void registerTasks() {
|
|
// Register task to start WiFi scan every minute (60000ms)
|
|
taskManager.registerTask("wifi_scan_periodic", 60000, [this]() {
|
|
if (!network.isWiFiScanning()) {
|
|
Serial.println("[WiFiScannerService] Starting periodic WiFi scan...");
|
|
ctx.fire("wifi/scan/start", nullptr);
|
|
network.startWiFiScan();
|
|
} else {
|
|
Serial.println("[WiFiScannerService] Skipping scan - already in progress");
|
|
}
|
|
});
|
|
}
|
|
|
|
NodeContext& ctx;
|
|
TaskManager& taskManager;
|
|
NetworkManager& network;
|
|
};
|
|
|
|
NodeContext ctx({
|
|
{"app", "wifiscan"},
|
|
{"role", "demo"}
|
|
});
|
|
NetworkManager network(ctx);
|
|
TaskManager taskManager(ctx);
|
|
ClusterManager cluster(ctx, taskManager);
|
|
ApiServer apiServer(ctx, taskManager, ctx.config.api_server_port);
|
|
WiFiScannerService wifiScanner(ctx, taskManager, network);
|
|
|
|
// WiFi scan start callback
|
|
void onWiFiScanStart(void* data) {
|
|
Serial.println("[WiFi] Scan started...");
|
|
}
|
|
|
|
// WiFi scan complete callback
|
|
void onWiFiScanComplete(void* data) {
|
|
Serial.println("\n=== WiFi Scan Results ===");
|
|
|
|
std::vector<WiFiAccessPoint>* accessPoints = static_cast<std::vector<WiFiAccessPoint>*>(data);
|
|
|
|
if (accessPoints->empty()) {
|
|
Serial.println("No networks found");
|
|
return;
|
|
}
|
|
|
|
Serial.printf("Found %d networks:\n", accessPoints->size());
|
|
Serial.println("SSID\t\t\tRSSI\tChannel\tEncryption");
|
|
Serial.println("----------------------------------------");
|
|
|
|
for (const auto& ap : *accessPoints) {
|
|
String encryptionStr = "Unknown";
|
|
switch (ap.encryption) {
|
|
case ENC_TYPE_NONE:
|
|
encryptionStr = "Open";
|
|
break;
|
|
case ENC_TYPE_WEP:
|
|
encryptionStr = "WEP";
|
|
break;
|
|
case ENC_TYPE_TKIP:
|
|
encryptionStr = "WPA";
|
|
break;
|
|
case ENC_TYPE_CCMP:
|
|
encryptionStr = "WPA2";
|
|
break;
|
|
case ENC_TYPE_AUTO:
|
|
encryptionStr = "Auto";
|
|
break;
|
|
}
|
|
|
|
Serial.printf("%-20s\t%d\t%d\t%s\n",
|
|
ap.ssid.c_str(), ap.rssi, ap.channel, encryptionStr.c_str());
|
|
}
|
|
Serial.println("========================\n");
|
|
}
|
|
|
|
// WiFi scan error callback
|
|
void onWiFiScanError(void* data) {
|
|
Serial.println("[WiFi] Scan failed - error occurred");
|
|
}
|
|
|
|
// WiFi scan timeout callback
|
|
void onWiFiScanTimeout(void* data) {
|
|
Serial.println("[WiFi] Scan failed - timeout");
|
|
}
|
|
|
|
void setup() {
|
|
Serial.begin(115200);
|
|
Serial.println("\n=== WiFi Scanner Example ===");
|
|
|
|
// Initialize task manager first
|
|
taskManager.initialize();
|
|
ctx.taskManager = &taskManager;
|
|
|
|
// Initialize network tasks
|
|
network.initTasks();
|
|
|
|
// Setup WiFi
|
|
network.setupWiFi();
|
|
|
|
// Start the API server
|
|
apiServer.begin();
|
|
wifiScanner.registerApi(apiServer);
|
|
|
|
// Register WiFi scan event callbacks
|
|
ctx.on("wifi/scan/start", onWiFiScanStart);
|
|
ctx.on("wifi/scan/complete", onWiFiScanComplete);
|
|
ctx.on("wifi/scan/error", onWiFiScanError);
|
|
ctx.on("wifi/scan/timeout", onWiFiScanTimeout);
|
|
|
|
// Start an initial WiFi scan
|
|
Serial.println("Starting initial WiFi scan...");
|
|
ctx.fire("wifi/scan/start", nullptr);
|
|
network.startWiFiScan();
|
|
|
|
// Print initial task status
|
|
taskManager.printTaskStatus();
|
|
}
|
|
|
|
void loop() {
|
|
taskManager.execute();
|
|
yield();
|
|
} |