Fork of the espurna firmware for `mhsw` switches
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

301 lines
7.9 KiB

8 years ago
7 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
8 years ago
  1. /*
  2. ESPurna
  3. Copyright (C) 2016-2017 by Xose Pérez <xose dot perez at gmail dot com>
  4. This program is free software: you can redistribute it and/or modify
  5. it under the terms of the GNU General Public License as published by
  6. the Free Software Foundation, either version 3 of the License, or
  7. (at your option) any later version.
  8. This program is distributed in the hope that it will be useful,
  9. but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. GNU General Public License for more details.
  12. You should have received a copy of the GNU General Public License
  13. along with this program. If not, see <http://www.gnu.org/licenses/>.
  14. */
  15. #include "config/all.h"
  16. #include <EEPROM.h>
  17. // -----------------------------------------------------------------------------
  18. // METHODS
  19. // -----------------------------------------------------------------------------
  20. String getIdentifier() {
  21. char identifier[20];
  22. sprintf(identifier, "%s_%06X", DEVICE, ESP.getChipId());
  23. return String(identifier);
  24. }
  25. void heartbeat() {
  26. static unsigned long last_uptime = 0;
  27. static unsigned char uptime_overflows = 0;
  28. if (millis() < last_uptime) ++uptime_overflows;
  29. last_uptime = millis();
  30. unsigned long uptime_seconds = uptime_overflows * (UPTIME_OVERFLOW / 1000) + (last_uptime / 1000);
  31. unsigned int free_heap = ESP.getFreeHeap();
  32. DEBUG_MSG_P(PSTR("[MAIN] Time: %s\n"), (char *) ntpDateTime().c_str());
  33. if (!mqttConnected()) {
  34. DEBUG_MSG_P(PSTR("[MAIN] Uptime: %ld seconds\n"), uptime_seconds);
  35. DEBUG_MSG_P(PSTR("[MAIN] Free heap: %d bytes\n"), free_heap);
  36. #if ENABLE_ADC_VCC
  37. DEBUG_MSG_P(PSTR("[MAIN] Power: %d mV\n"), ESP.getVcc());
  38. #endif
  39. }
  40. #if (HEARTBEAT_REPORT_INTERVAL)
  41. mqttSend(MQTT_TOPIC_INTERVAL, HEARTBEAT_INTERVAL / 1000);
  42. #endif
  43. #if (HEARTBEAT_REPORT_APP)
  44. mqttSend(MQTT_TOPIC_APP, APP_NAME);
  45. #endif
  46. #if (HEARTBEAT_REPORT_VERSION)
  47. mqttSend(MQTT_TOPIC_VERSION, APP_VERSION);
  48. #endif
  49. #if (HEARTBEAT_REPORT_HOSTNAME)
  50. mqttSend(MQTT_TOPIC_HOSTNAME, getSetting("hostname").c_str());
  51. #endif
  52. #if (HEARTBEAT_REPORT_IP)
  53. mqttSend(MQTT_TOPIC_IP, getIP().c_str());
  54. #endif
  55. #if (HEARTBEAT_REPORT_MAC)
  56. mqttSend(MQTT_TOPIC_MAC, WiFi.macAddress().c_str());
  57. #endif
  58. #if (HEARTBEAT_REPORT_RSSI)
  59. mqttSend(MQTT_TOPIC_RSSI, String(WiFi.RSSI()).c_str());
  60. #endif
  61. #if (HEARTBEAT_REPORT_UPTIME)
  62. mqttSend(MQTT_TOPIC_UPTIME, String(uptime_seconds).c_str());
  63. #if ENABLE_INFLUXDB
  64. influxDBSend(MQTT_TOPIC_UPTIME, String(uptime_seconds).c_str());
  65. #endif
  66. #endif
  67. #if (HEARTBEAT_REPORT_FREEHEAP)
  68. mqttSend(MQTT_TOPIC_FREEHEAP, String(free_heap).c_str());
  69. #if ENABLE_INFLUXDB
  70. influxDBSend(MQTT_TOPIC_FREEHEAP, String(free_heap).c_str());
  71. #endif
  72. #endif
  73. #if (HEARTBEAT_REPORT_RELAY)
  74. relayMQTT();
  75. #endif
  76. #if LIGHT_PROVIDER != LIGHT_PROVIDER_NONE
  77. #if (HEARTBEAT_REPORT_COLOR)
  78. mqttSend(MQTT_TOPIC_COLOR, lightColor().c_str());
  79. #endif
  80. #endif
  81. #if (HEARTBEAT_REPORT_VCC)
  82. #if ENABLE_ADC_VCC
  83. mqttSend(MQTT_TOPIC_VCC, String(ESP.getVcc()).c_str());
  84. #endif
  85. #endif
  86. #if (HEARTBEAT_REPORT_STATUS)
  87. mqttSend(MQTT_TOPIC_STATUS, MQTT_STATUS_ONLINE, true);
  88. #endif
  89. }
  90. void customReset(unsigned char status) {
  91. EEPROM.write(EEPROM_CUSTOM_RESET, status);
  92. EEPROM.commit();
  93. }
  94. unsigned char customReset() {
  95. static unsigned char status = 255;
  96. if (status == 255) {
  97. status = EEPROM.read(EEPROM_CUSTOM_RESET);
  98. if (status > 0) customReset(0);
  99. if (status > CUSTOM_RESET_MAX) status = 0;
  100. }
  101. return status;
  102. }
  103. void hardwareSetup() {
  104. EEPROM.begin(4096);
  105. #ifdef DEBUG_PORT
  106. DEBUG_PORT.begin(SERIAL_BAUDRATE);
  107. if (customReset() == CUSTOM_RESET_HARDWARE) {
  108. DEBUG_PORT.setDebugOutput(true);
  109. }
  110. #elif defined(SERIAL_BAUDRATE)
  111. Serial.begin(SERIAL_BAUDRATE);
  112. #endif
  113. #if not EMBEDDED_WEB
  114. SPIFFS.begin();
  115. #endif
  116. }
  117. void hardwareLoop() {
  118. // Heartbeat
  119. static unsigned long last_uptime = 0;
  120. if ((millis() - last_uptime > HEARTBEAT_INTERVAL) || (last_uptime == 0)) {
  121. last_uptime = millis();
  122. heartbeat();
  123. }
  124. }
  125. // -----------------------------------------------------------------------------
  126. // BOOTING
  127. // -----------------------------------------------------------------------------
  128. void welcome() {
  129. DEBUG_MSG_P(PSTR("%s %s\n"), (char *) APP_NAME, (char *) APP_VERSION);
  130. DEBUG_MSG_P(PSTR("%s\n%s\n\n"), (char *) APP_AUTHOR, (char *) APP_WEBSITE);
  131. DEBUG_MSG_P(PSTR("ChipID: %06X\n"), ESP.getChipId());
  132. DEBUG_MSG_P(PSTR("CPU frequency: %d MHz\n"), ESP.getCpuFreqMHz());
  133. unsigned char custom_reset = customReset();
  134. if (custom_reset > 0) {
  135. char buffer[32];
  136. strcpy_P(buffer, custom_reset_string[custom_reset-1]);
  137. DEBUG_MSG_P(PSTR("Last reset reason: %s\n"), buffer);
  138. } else {
  139. DEBUG_MSG_P(PSTR("Last reset reason: %s\n"), (char *) ESP.getResetReason().c_str());
  140. }
  141. DEBUG_MSG_P(PSTR("Memory size (SDK): %d bytes\n"), ESP.getFlashChipSize());
  142. DEBUG_MSG_P(PSTR("Memory size (CHIP): %d bytes\n"), ESP.getFlashChipRealSize());
  143. DEBUG_MSG_P(PSTR("Free heap: %d bytes\n"), ESP.getFreeHeap());
  144. DEBUG_MSG_P(PSTR("Firmware size: %d bytes\n"), ESP.getSketchSize());
  145. DEBUG_MSG_P(PSTR("Free firmware space: %d bytes\n"), ESP.getFreeSketchSpace());
  146. #if not EMBEDDED_WEB
  147. FSInfo fs_info;
  148. if (SPIFFS.info(fs_info)) {
  149. DEBUG_MSG_P(PSTR("File system total size: %d bytes\n"), fs_info.totalBytes);
  150. DEBUG_MSG_P(PSTR(" used size : %d bytes\n"), fs_info.usedBytes);
  151. DEBUG_MSG_P(PSTR(" block size: %d bytes\n"), fs_info.blockSize);
  152. DEBUG_MSG_P(PSTR(" page size : %d bytes\n"), fs_info.pageSize);
  153. DEBUG_MSG_P(PSTR(" max files : %d\n"), fs_info.maxOpenFiles);
  154. DEBUG_MSG_P(PSTR(" max length: %d\n"), fs_info.maxPathLength);
  155. }
  156. #endif
  157. DEBUG_MSG_P(PSTR("\n\n"));
  158. }
  159. void setup() {
  160. hardwareSetup();
  161. welcome();
  162. settingsSetup();
  163. if (getSetting("hostname").length() == 0) {
  164. setSetting("hostname", getIdentifier());
  165. saveSettings();
  166. }
  167. webSetup();
  168. #if LIGHT_PROVIDER != LIGHT_PROVIDER_NONE
  169. lightSetup();
  170. #endif
  171. relaySetup();
  172. buttonSetup();
  173. ledSetup();
  174. delay(500);
  175. wifiSetup();
  176. otaSetup();
  177. mqttSetup();
  178. ntpSetup();
  179. #ifdef SONOFF_RFBRIDGE
  180. rfbSetup();
  181. #endif
  182. #if ENABLE_I2C
  183. i2cSetup();
  184. #endif
  185. #if ENABLE_FAUXMO
  186. fauxmoSetup();
  187. #endif
  188. #if ENABLE_NOFUSS
  189. nofussSetup();
  190. #endif
  191. #if ENABLE_INFLUXDB
  192. influxDBSetup();
  193. #endif
  194. #if ENABLE_HLW8012
  195. hlw8012Setup();
  196. #endif
  197. #if ENABLE_DS18B20
  198. dsSetup();
  199. #endif
  200. #if ENABLE_ANALOG
  201. analogSetup();
  202. #endif
  203. #if ENABLE_DHT
  204. dhtSetup();
  205. #endif
  206. #if ENABLE_RF
  207. rfSetup();
  208. #endif
  209. #if ENABLE_EMON
  210. powerMonitorSetup();
  211. #endif
  212. // Prepare configuration for version 2.0
  213. hwUpwardsCompatibility();
  214. }
  215. void loop() {
  216. hardwareLoop();
  217. buttonLoop();
  218. relayLoop();
  219. ledLoop();
  220. wifiLoop();
  221. otaLoop();
  222. mqttLoop();
  223. ntpLoop();
  224. #if ENABLE_FAUXMO
  225. fauxmoLoop();
  226. #endif
  227. #if !defined(SONOFF_DUAL) & !defined(SONOFF_RFBRIDGE)
  228. settingsLoop();
  229. #endif
  230. #ifdef SONOFF_RFBRIDGE
  231. rfbLoop();
  232. #endif
  233. #if ENABLE_NOFUSS
  234. nofussLoop();
  235. #endif
  236. #if ENABLE_HLW8012
  237. hlw8012Loop();
  238. #endif
  239. #if ENABLE_DS18B20
  240. dsLoop();
  241. #endif
  242. #if ENABLE_ANALOG
  243. analogLoop();
  244. #endif
  245. #if ENABLE_DHT
  246. dhtLoop();
  247. #endif
  248. #if ENABLE_RF
  249. rfLoop();
  250. #endif
  251. #if ENABLE_EMON
  252. powerMonitorLoop();
  253. #endif
  254. }