Pre-fight/EvilFirmwareAPI.cpp at main · stackzac22/Pre-fight · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
/*
   Evil Firmware Fleet Management API - Implementation

   Core implementation of the device API library.
*/

#include "EvilFirmwareAPI.h"

#ifdef ARDUINO_ARCH_ESP32
#include <esp_adc_cal.h>
#endif

EvilFirmwareAPI::EvilFirmwareAPI() {
  device_id = "";
  device_type = "UNKNOWN";
  firmware_version = "0.0.0";
}

void EvilFirmwareAPI::begin(String id, String type, String version) {
  device_id = id;
  device_type = type;
  firmware_version = version;
}

void EvilFirmwareAPI::updateStatus() {
#ifdef ARDUINO_ARCH_ESP32
  current_status.device_id = device_id;
  current_status.device_type = device_type;
  current_status.firmware_version = firmware_version;
  current_status.api_version = EVIL_API_VERSION;
  current_status.wifi_connected = WiFi.status() == WL_CONNECTED;
  current_status.wifi_ssid = WiFi.SSID();
  current_status.wifi_rssi = WiFi.RSSI();
  current_status.uptime_ms = millis();
  current_status.free_heap = ESP.getFreeHeap();
  current_status.total_heap = ESP.getHeapSize();

  // Battery voltage (ADC pin 4 on most ESP32 boards)
  int raw = analogRead(4);
  current_status.battery_voltage = (raw / 4095.0) * 3.3 * 2.0;  // Assuming voltage divider
  current_status.battery_percent = map(constrain(current_status.battery_voltage * 100, 300, 420), 300, 420, 0, 100);

  // CPU temperature
  current_status.cpu_temp = (temprature_sens_read() - 32) / 1.8;
#endif
}

DeviceStatus EvilFirmwareAPI::getStatus() {
  updateStatus();
  return current_status;
}

String EvilFirmwareAPI::statusToJSON() {
  DynamicJsonDocument doc(512);
  DeviceStatus status = getStatus();

  doc["device_id"] = status.device_id;
  doc["device_type"] = status.device_type;
  doc["firmware_version"] = status.firmware_version;
  doc["api_version"] = status.api_version;
  doc["wifi_connected"] = status.wifi_connected;
  doc["wifi_ssid"] = status.wifi_ssid;
  doc["wifi_rssi"] = status.wifi_rssi;
  doc["uptime_ms"] = status.uptime_ms;
  doc["battery_voltage"] = status.battery_voltage;
  doc["battery_percent"] = status.battery_percent;
  doc["cpu_temp"] = status.cpu_temp;
  doc["free_heap"] = status.free_heap;
  doc["total_heap"] = status.total_heap;

  String output;
  serializeJson(doc, output);
  return output;
}

String EvilFirmwareAPI::sensorsToJSON() {
  DynamicJsonDocument doc(256);

  doc["temperature"] = sensor_data.temperature;
  doc["humidity"] = sensor_data.humidity;
  doc["pressure"] = sensor_data.pressure;
  doc["gps_valid"] = sensor_data.gps_valid;
  doc["latitude"] = sensor_data.latitude;
  doc["longitude"] = sensor_data.longitude;
  doc["gps_accuracy"] = sensor_data.gps_accuracy;

  String output;
  serializeJson(doc, output);
  return output;
}

String EvilFirmwareAPI::batteryToJSON() {
  DynamicJsonDocument doc(128);
  updateStatus();

  doc["voltage"] = current_status.battery_voltage;
  doc["percent"] = current_status.battery_percent;
  doc["timestamp"] = millis();

  String output;
  serializeJson(doc, output);
  return output;
}

void EvilFirmwareAPI::setSensorData(SensorData data) {
  sensor_data = data;
}

SensorData EvilFirmwareAPI::getSensorData() {
  return sensor_data;
}

uint32_t EvilFirmwareAPI::getUptime() {
  return millis();
}

float EvilFirmwareAPI::getBatteryVoltage() {
  updateStatus();
  return current_status.battery_voltage;
}

int EvilFirmwareAPI::getBatteryPercent() {
  updateStatus();
  return current_status.battery_percent;
}

String EvilFirmwareAPI::getDeviceID() {
  return device_id;
}

String EvilFirmwareAPI::getFirmwareVersion() {
  return firmware_version;
}