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.

321 lines
7.2 KiB

  1. /*
  2. ESPurna
  3. Copyright (C) 2016-2019 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 "espurna.h"
  16. #include "alexa.h"
  17. #include "api.h"
  18. #include "broker.h"
  19. #include "button.h"
  20. #include "crash.h"
  21. #include "curtain_kingart.h"
  22. #include "debug.h"
  23. #include "domoticz.h"
  24. #include "encoder.h"
  25. #include "homeassistant.h"
  26. #include "i2c.h"
  27. #include "influxdb.h"
  28. #include "ir.h"
  29. #include "led.h"
  30. #include "light.h"
  31. #include "lightfox.h"
  32. #include "llmnr.h"
  33. #include "mdns.h"
  34. #include "mqtt.h"
  35. #include "netbios.h"
  36. #include "nofuss.h"
  37. #include "ntp.h"
  38. #include "ota.h"
  39. #include "relay.h"
  40. #include "rfbridge.h"
  41. #include "rfm69.h"
  42. #include "rpc.h"
  43. #include "rpnrules.h"
  44. #include "rtcmem.h"
  45. #include "scheduler.h"
  46. #include "sensor.h"
  47. #include "ssdp.h"
  48. #include "telnet.h"
  49. #include "thermostat.h"
  50. #include "thingspeak.h"
  51. #include "tuya.h"
  52. #include "uartmqtt.h"
  53. #include "web.h"
  54. #include "ws.h"
  55. std::vector<void_callback_f> _loop_callbacks;
  56. std::vector<void_callback_f> _reload_callbacks;
  57. bool _reload_config = false;
  58. unsigned long _loop_delay = 0;
  59. // -----------------------------------------------------------------------------
  60. // GENERAL CALLBACKS
  61. // -----------------------------------------------------------------------------
  62. void espurnaRegisterLoop(void_callback_f callback) {
  63. _loop_callbacks.push_back(callback);
  64. }
  65. void espurnaRegisterReload(void_callback_f callback) {
  66. _reload_callbacks.push_back(callback);
  67. }
  68. void espurnaReload() {
  69. _reload_config = true;
  70. }
  71. void _espurnaReload() {
  72. for (const auto& callback : _reload_callbacks) {
  73. callback();
  74. }
  75. }
  76. unsigned long espurnaLoopDelay() {
  77. return _loop_delay;
  78. }
  79. // -----------------------------------------------------------------------------
  80. // BOOTING
  81. // -----------------------------------------------------------------------------
  82. void setup() {
  83. // -------------------------------------------------------------------------
  84. // Basic modules, will always run
  85. // -------------------------------------------------------------------------
  86. // Cache initial free heap value
  87. setInitialFreeHeap();
  88. // Init logging module
  89. #if DEBUG_SUPPORT
  90. debugSetup();
  91. #endif
  92. // Init GPIO functions
  93. gpioSetup();
  94. // Init RTCMEM
  95. rtcmemSetup();
  96. // Init EEPROM
  97. eepromSetup();
  98. // Init persistance
  99. settingsSetup();
  100. // Configure logger and crash recorder
  101. #if DEBUG_SUPPORT
  102. debugConfigureBoot();
  103. crashSetup();
  104. #endif
  105. // Return bogus free heap value for broken devices
  106. // XXX: device is likely to trigger other bugs! tread carefuly
  107. wtfHeap(getSetting<int>("wtfHeap", 0));
  108. // Init Serial, SPIFFS and system check
  109. systemSetup();
  110. // Init terminal features
  111. #if TERMINAL_SUPPORT
  112. terminalSetup();
  113. #endif
  114. // Hostname & board name initialization
  115. if (getSetting("hostname").length() == 0) {
  116. setDefaultHostname();
  117. }
  118. setBoardName();
  119. // Show welcome message and system configuration
  120. info(true);
  121. wifiSetup();
  122. #if OTA_ARDUINOOTA_SUPPORT
  123. arduinoOtaSetup();
  124. #endif
  125. #if TELNET_SUPPORT
  126. telnetSetup();
  127. #endif
  128. #if OTA_CLIENT != OTA_CLIENT_NONE
  129. otaClientSetup();
  130. #endif
  131. // -------------------------------------------------------------------------
  132. // Check if system is stable
  133. // -------------------------------------------------------------------------
  134. #if SYSTEM_CHECK_ENABLED
  135. if (!systemCheck()) return;
  136. #endif
  137. // -------------------------------------------------------------------------
  138. // Next modules will be only loaded if system is flagged as stable
  139. // -------------------------------------------------------------------------
  140. // Init webserver required before any module that uses API
  141. #if WEB_SUPPORT
  142. webSetup();
  143. wsSetup();
  144. #if DEBUG_WEB_SUPPORT
  145. debugWebSetup();
  146. #endif
  147. #if OTA_WEB_SUPPORT
  148. otaWebSetup();
  149. #endif
  150. #endif
  151. #if API_SUPPORT
  152. apiSetup();
  153. #endif
  154. // lightSetup must be called before relaySetup
  155. #if LIGHT_PROVIDER != LIGHT_PROVIDER_NONE
  156. lightSetup();
  157. #endif
  158. #if RELAY_SUPPORT
  159. relaySetup();
  160. #endif
  161. #if BUTTON_SUPPORT
  162. buttonSetup();
  163. #endif
  164. #if ENCODER_SUPPORT && (LIGHT_PROVIDER != LIGHT_PROVIDER_NONE)
  165. encoderSetup();
  166. #endif
  167. #if LED_SUPPORT
  168. ledSetup();
  169. #endif
  170. #if MQTT_SUPPORT
  171. mqttSetup();
  172. #endif
  173. #if MDNS_SERVER_SUPPORT
  174. mdnsServerSetup();
  175. #endif
  176. #if MDNS_CLIENT_SUPPORT
  177. mdnsClientSetup();
  178. #endif
  179. #if LLMNR_SUPPORT
  180. llmnrSetup();
  181. #endif
  182. #if NETBIOS_SUPPORT
  183. netbiosSetup();
  184. #endif
  185. #if SSDP_SUPPORT
  186. ssdpSetup();
  187. #endif
  188. #if NTP_SUPPORT
  189. ntpSetup();
  190. #endif
  191. #if I2C_SUPPORT
  192. i2cSetup();
  193. #endif
  194. #if RF_SUPPORT
  195. rfbSetup();
  196. #endif
  197. #if ALEXA_SUPPORT
  198. alexaSetup();
  199. #endif
  200. #if NOFUSS_SUPPORT
  201. nofussSetup();
  202. #endif
  203. #if SENSOR_SUPPORT
  204. sensorSetup();
  205. #endif
  206. #if INFLUXDB_SUPPORT
  207. idbSetup();
  208. #endif
  209. #if THINGSPEAK_SUPPORT
  210. tspkSetup();
  211. #endif
  212. #if RFM69_SUPPORT
  213. rfm69Setup();
  214. #endif
  215. #if IR_SUPPORT
  216. irSetup();
  217. #endif
  218. #if DOMOTICZ_SUPPORT
  219. domoticzSetup();
  220. #endif
  221. #if HOMEASSISTANT_SUPPORT
  222. haSetup();
  223. #endif
  224. #if SCHEDULER_SUPPORT
  225. schSetup();
  226. #endif
  227. #if RPN_RULES_SUPPORT
  228. rpnSetup();
  229. #endif
  230. #if UART_MQTT_SUPPORT
  231. uartmqttSetup();
  232. #endif
  233. #ifdef FOXEL_LIGHTFOX_DUAL
  234. lightfoxSetup();
  235. #endif
  236. #if THERMOSTAT_SUPPORT
  237. thermostatSetup();
  238. #endif
  239. #if THERMOSTAT_DISPLAY_SUPPORT
  240. displaySetup();
  241. #endif
  242. #if TUYA_SUPPORT
  243. Tuya::tuyaSetup();
  244. #endif
  245. #if KINGART_CURTAIN_SUPPORT
  246. kingartCurtainSetup();
  247. #endif
  248. // 3rd party code hook
  249. #if USE_EXTRA
  250. extraSetup();
  251. #endif
  252. // Prepare configuration for version 2.0
  253. migrate();
  254. // Set up delay() after loop callbacks are finished
  255. // Note: should be after settingsSetup()
  256. _loop_delay = constrain(
  257. getSetting("loopDelay", LOOP_DELAY_TIME), 0, 300
  258. );
  259. saveSettings();
  260. }
  261. void loop() {
  262. // Reload config before running any callbacks
  263. if (_reload_config) {
  264. _espurnaReload();
  265. _reload_config = false;
  266. }
  267. // Call registered loop callbacks
  268. for (unsigned char i = 0; i < _loop_callbacks.size(); i++) {
  269. (_loop_callbacks[i])();
  270. }
  271. // Power saving delay
  272. if (_loop_delay) delay(_loop_delay);
  273. }