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.

1025 lines
30 KiB

7 years ago
7 years ago
7 years ago
7 years ago
  1. /*
  2. RELAY MODULE
  3. Copyright (C) 2016-2018 by Xose Pérez <xose dot perez at gmail dot com>
  4. */
  5. #include <EEPROM_Rotate.h>
  6. #include <Ticker.h>
  7. #include <ArduinoJson.h>
  8. #include <vector>
  9. #include <functional>
  10. typedef struct {
  11. // Configuration variables
  12. unsigned char pin; // GPIO pin for the relay
  13. unsigned char type; // RELAY_TYPE_NORMAL, RELAY_TYPE_INVERSE, RELAY_TYPE_LATCHED or RELAY_TYPE_LATCHED_INVERSE
  14. unsigned char reset_pin; // GPIO to reset the relay if RELAY_TYPE_LATCHED
  15. unsigned long delay_on; // Delay to turn relay ON
  16. unsigned long delay_off; // Delay to turn relay OFF
  17. unsigned char pulse; // RELAY_PULSE_NONE, RELAY_PULSE_OFF or RELAY_PULSE_ON
  18. unsigned long pulse_ms; // Pulse length in millis
  19. // Status variables
  20. bool current_status; // Holds the current (physical) status of the relay
  21. bool target_status; // Holds the target status
  22. unsigned long fw_start; // Flood window start time
  23. unsigned char fw_count; // Number of changes within the current flood window
  24. unsigned long change_time; // Scheduled time to change
  25. bool report; // Whether to report to own topic
  26. bool group_report; // Whether to report to group topic
  27. // Helping objects
  28. Ticker pulseTicker; // Holds the pulse back timer
  29. } relay_t;
  30. std::vector<relay_t> _relays;
  31. bool _relayRecursive = false;
  32. Ticker _relaySaveTicker;
  33. // -----------------------------------------------------------------------------
  34. // RELAY PROVIDERS
  35. // -----------------------------------------------------------------------------
  36. void _relayProviderStatus(unsigned char id, bool status) {
  37. // Check relay ID
  38. if (id >= _relays.size()) return;
  39. // Store new current status
  40. _relays[id].current_status = status;
  41. #if RELAY_PROVIDER == RELAY_PROVIDER_RFBRIDGE
  42. rfbStatus(id, status);
  43. #endif
  44. #if RELAY_PROVIDER == RELAY_PROVIDER_DUAL
  45. // Calculate mask
  46. unsigned char mask=0;
  47. for (unsigned char i=0; i<_relays.size(); i++) {
  48. if (_relays[i].current_status) mask = mask + (1 << i);
  49. }
  50. // Send it to F330
  51. Serial.flush();
  52. Serial.write(0xA0);
  53. Serial.write(0x04);
  54. Serial.write(mask);
  55. Serial.write(0xA1);
  56. Serial.flush();
  57. #endif
  58. #if RELAY_PROVIDER == RELAY_PROVIDER_STM
  59. Serial.flush();
  60. Serial.write(0xA0);
  61. Serial.write(id + 1);
  62. Serial.write(status);
  63. Serial.write(0xA1 + status + id);
  64. Serial.flush();
  65. #endif
  66. #if RELAY_PROVIDER == RELAY_PROVIDER_LIGHT
  67. // If the number of relays matches the number of light channels
  68. // assume each relay controls one channel.
  69. // If the number of relays is the number of channels plus 1
  70. // assume the first one controls all the channels and
  71. // the rest one channel each.
  72. // Otherwise every relay controls all channels.
  73. // TODO: this won't work with a mixed of dummy and real relays
  74. // but this option is not allowed atm (YANGNI)
  75. if (_relays.size() == lightChannels()) {
  76. lightState(id, status);
  77. lightState(true);
  78. } else if (_relays.size() == (lightChannels() + 1u)) {
  79. if (id == 0) {
  80. lightState(status);
  81. } else {
  82. lightState(id-1, status);
  83. }
  84. } else {
  85. lightState(status);
  86. }
  87. lightUpdate(true, true);
  88. #endif
  89. #if RELAY_PROVIDER == RELAY_PROVIDER_RELAY
  90. if (_relays[id].type == RELAY_TYPE_NORMAL) {
  91. digitalWrite(_relays[id].pin, status);
  92. } else if (_relays[id].type == RELAY_TYPE_INVERSE) {
  93. digitalWrite(_relays[id].pin, !status);
  94. } else if (_relays[id].type == RELAY_TYPE_LATCHED || _relays[id].type == RELAY_TYPE_LATCHED_INVERSE) {
  95. bool pulse = RELAY_TYPE_LATCHED ? HIGH : LOW;
  96. digitalWrite(_relays[id].pin, !pulse);
  97. if (GPIO_NONE != _relays[id].reset_pin) digitalWrite(_relays[id].reset_pin, !pulse);
  98. if (status || (GPIO_NONE == _relays[id].reset_pin)) {
  99. digitalWrite(_relays[id].pin, pulse);
  100. } else {
  101. digitalWrite(_relays[id].reset_pin, pulse);
  102. }
  103. nice_delay(RELAY_LATCHING_PULSE);
  104. digitalWrite(_relays[id].pin, !pulse);
  105. if (GPIO_NONE != _relays[id].reset_pin) digitalWrite(_relays[id].reset_pin, !pulse);
  106. }
  107. #endif
  108. }
  109. /**
  110. * Walks the relay vector processing only those relays
  111. * that have to change to the requested mode
  112. * @bool mode Requested mode
  113. */
  114. void _relayProcess(bool mode) {
  115. unsigned long current_time = millis();
  116. for (unsigned char id = 0; id < _relays.size(); id++) {
  117. bool target = _relays[id].target_status;
  118. // Only process the relays we have to change
  119. if (target == _relays[id].current_status) continue;
  120. // Only process the relays we have change to the requested mode
  121. if (target != mode) continue;
  122. // Only process if the change_time has arrived
  123. if (current_time < _relays[id].change_time) continue;
  124. DEBUG_MSG_P(PSTR("[RELAY] #%d set to %s\n"), id, target ? "ON" : "OFF");
  125. // Call the provider to perform the action
  126. _relayProviderStatus(id, target);
  127. // Send to Broker
  128. #if BROKER_SUPPORT
  129. brokerPublish(MQTT_TOPIC_RELAY, id, target ? "1" : "0");
  130. #endif
  131. // Send MQTT
  132. #if MQTT_SUPPORT
  133. relayMQTT(id);
  134. #endif
  135. if (!_relayRecursive) {
  136. relayPulse(id);
  137. _relaySaveTicker.once_ms(RELAY_SAVE_DELAY, relaySave);
  138. #if WEB_SUPPORT
  139. wsSend(_relayWebSocketUpdate);
  140. #endif
  141. }
  142. #if DOMOTICZ_SUPPORT
  143. domoticzSendRelay(id);
  144. #endif
  145. #if INFLUXDB_SUPPORT
  146. relayInfluxDB(id);
  147. #endif
  148. #if THINGSPEAK_SUPPORT
  149. tspkEnqueueRelay(id, target);
  150. tspkFlush();
  151. #endif
  152. // Flag relay-based LEDs to update status
  153. #if LED_SUPPORT
  154. ledUpdate(true);
  155. #endif
  156. _relays[id].report = false;
  157. _relays[id].group_report = false;
  158. }
  159. }
  160. #if defined(ITEAD_SONOFF_IFAN02)
  161. unsigned char _relay_ifan02_speeds[] = {0, 1, 3, 5};
  162. unsigned char getSpeed() {
  163. unsigned char speed =
  164. (_relays[1].target_status ? 1 : 0) +
  165. (_relays[2].target_status ? 2 : 0) +
  166. (_relays[3].target_status ? 4 : 0);
  167. for (unsigned char i=0; i<4; i++) {
  168. if (_relay_ifan02_speeds[i] == speed) return i;
  169. }
  170. return 0;
  171. }
  172. void setSpeed(unsigned char speed) {
  173. if ((0 <= speed) & (speed <= 3)) {
  174. if (getSpeed() == speed) return;
  175. unsigned char states = _relay_ifan02_speeds[speed];
  176. for (unsigned char i=0; i<3; i++) {
  177. relayStatus(i+1, states & 1 == 1);
  178. states >>= 1;
  179. }
  180. }
  181. }
  182. #endif
  183. // -----------------------------------------------------------------------------
  184. // RELAY
  185. // -----------------------------------------------------------------------------
  186. void relayPulse(unsigned char id) {
  187. _relays[id].pulseTicker.detach();
  188. byte mode = _relays[id].pulse;
  189. if (mode == RELAY_PULSE_NONE) return;
  190. unsigned long ms = _relays[id].pulse_ms;
  191. if (ms == 0) return;
  192. bool status = relayStatus(id);
  193. bool pulseStatus = (mode == RELAY_PULSE_ON);
  194. if (pulseStatus != status) {
  195. DEBUG_MSG_P(PSTR("[RELAY] Scheduling relay #%d back in %lums (pulse)\n"), id, ms);
  196. _relays[id].pulseTicker.once_ms(ms, relayToggle, id);
  197. // Reconfigure after dynamic pulse
  198. _relays[id].pulse = getSetting("relayPulse", id, RELAY_PULSE_MODE).toInt();
  199. _relays[id].pulse_ms = 1000 * getSetting("relayTime", id, RELAY_PULSE_MODE).toFloat();
  200. }
  201. }
  202. bool relayStatus(unsigned char id, bool status, bool report, bool group_report) {
  203. if (id >= _relays.size()) return false;
  204. bool changed = false;
  205. if (_relays[id].current_status == status) {
  206. if (_relays[id].target_status != status) {
  207. DEBUG_MSG_P(PSTR("[RELAY] #%d scheduled change cancelled\n"), id);
  208. _relays[id].target_status = status;
  209. _relays[id].report = false;
  210. _relays[id].group_report = false;
  211. changed = true;
  212. }
  213. // For RFBridge, keep sending the message even if the status is already the required
  214. #if RELAY_PROVIDER == RELAY_PROVIDER_RFBRIDGE
  215. rfbStatus(id, status);
  216. #endif
  217. // Update the pulse counter if the relay is already in the non-normal state (#454)
  218. relayPulse(id);
  219. } else {
  220. unsigned long current_time = millis();
  221. unsigned long fw_end = _relays[id].fw_start + 1000 * RELAY_FLOOD_WINDOW;
  222. unsigned long delay = status ? _relays[id].delay_on : _relays[id].delay_off;
  223. _relays[id].fw_count++;
  224. _relays[id].change_time = current_time + delay;
  225. // If current_time is off-limits the floodWindow...
  226. if (current_time < _relays[id].fw_start || fw_end <= current_time) {
  227. // We reset the floodWindow
  228. _relays[id].fw_start = current_time;
  229. _relays[id].fw_count = 1;
  230. // If current_time is in the floodWindow and there have been too many requests...
  231. } else if (_relays[id].fw_count >= RELAY_FLOOD_CHANGES) {
  232. // We schedule the changes to the end of the floodWindow
  233. // unless it's already delayed beyond that point
  234. if (fw_end - delay > current_time) {
  235. _relays[id].change_time = fw_end;
  236. }
  237. }
  238. _relays[id].target_status = status;
  239. if (report) _relays[id].report = true;
  240. if (group_report) _relays[id].group_report = true;
  241. relaySync(id);
  242. DEBUG_MSG_P(PSTR("[RELAY] #%d scheduled %s in %u ms\n"),
  243. id, status ? "ON" : "OFF",
  244. (_relays[id].change_time - current_time));
  245. changed = true;
  246. }
  247. return changed;
  248. }
  249. bool relayStatus(unsigned char id, bool status) {
  250. return relayStatus(id, status, true, true);
  251. }
  252. bool relayStatus(unsigned char id) {
  253. // Check relay ID
  254. if (id >= _relays.size()) return false;
  255. // Get status from storage
  256. return _relays[id].current_status;
  257. }
  258. void relaySync(unsigned char id) {
  259. // No sync if none or only one relay
  260. if (_relays.size() < 2) return;
  261. // Do not go on if we are comming from a previous sync
  262. if (_relayRecursive) return;
  263. // Flag sync mode
  264. _relayRecursive = true;
  265. byte relaySync = getSetting("relaySync", RELAY_SYNC).toInt();
  266. bool status = _relays[id].target_status;
  267. // If RELAY_SYNC_SAME all relays should have the same state
  268. if (relaySync == RELAY_SYNC_SAME) {
  269. for (unsigned short i=0; i<_relays.size(); i++) {
  270. if (i != id) relayStatus(i, status);
  271. }
  272. // If NONE_OR_ONE or ONE and setting ON we should set OFF all the others
  273. } else if (status) {
  274. if (relaySync != RELAY_SYNC_ANY) {
  275. for (unsigned short i=0; i<_relays.size(); i++) {
  276. if (i != id) relayStatus(i, false);
  277. }
  278. }
  279. // If ONLY_ONE and setting OFF we should set ON the other one
  280. } else {
  281. if (relaySync == RELAY_SYNC_ONE) {
  282. unsigned char i = (id + 1) % _relays.size();
  283. relayStatus(i, true);
  284. }
  285. }
  286. // Unflag sync mode
  287. _relayRecursive = false;
  288. }
  289. void relaySave() {
  290. unsigned char bit = 1;
  291. unsigned char mask = 0;
  292. for (unsigned int i=0; i < _relays.size(); i++) {
  293. if (relayStatus(i)) mask += bit;
  294. bit += bit;
  295. }
  296. EEPROMr.write(EEPROM_RELAY_STATUS, mask);
  297. DEBUG_MSG_P(PSTR("[RELAY] Saving mask: %d\n"), mask);
  298. EEPROMr.commit();
  299. }
  300. void relayToggle(unsigned char id, bool report, bool group_report) {
  301. if (id >= _relays.size()) return;
  302. relayStatus(id, !relayStatus(id), report, group_report);
  303. }
  304. void relayToggle(unsigned char id) {
  305. relayToggle(id, true, true);
  306. }
  307. unsigned char relayCount() {
  308. return _relays.size();
  309. }
  310. unsigned char relayParsePayload(const char * payload) {
  311. // Payload could be "OFF", "ON", "TOGGLE"
  312. // or its number equivalents: 0, 1 or 2
  313. if (payload[0] == '0') return 0;
  314. if (payload[0] == '1') return 1;
  315. if (payload[0] == '2') return 2;
  316. // trim payload
  317. char * p = ltrim((char *)payload);
  318. // to lower
  319. unsigned int l = strlen(p);
  320. if (l>6) l=6;
  321. for (unsigned char i=0; i<l; i++) {
  322. p[i] = tolower(p[i]);
  323. }
  324. unsigned int value = 0xFF;
  325. if (strcmp(p, "off") == 0) {
  326. value = 0;
  327. } else if (strcmp(p, "on") == 0) {
  328. value = 1;
  329. } else if (strcmp(p, "toggle") == 0) {
  330. value = 2;
  331. } else if (strcmp(p, "query") == 0) {
  332. value = 3;
  333. }
  334. return value;
  335. }
  336. // BACKWARDS COMPATIBILITY
  337. void _relayBackwards() {
  338. byte relayMode = getSetting("relayMode", RELAY_BOOT_MODE).toInt();
  339. byte relayPulseMode = getSetting("relayPulseMode", RELAY_PULSE_MODE).toInt();
  340. float relayPulseTime = getSetting("relayPulseTime", RELAY_PULSE_TIME).toFloat();
  341. if (relayPulseMode == RELAY_PULSE_NONE) relayPulseTime = 0;
  342. for (unsigned int i=0; i<_relays.size(); i++) {
  343. if (!hasSetting("relayBoot", i)) setSetting("relayBoot", i, relayMode);
  344. if (!hasSetting("relayPulse", i)) setSetting("relayPulse", i, relayPulseMode);
  345. if (!hasSetting("relayTime", i)) setSetting("relayTime", i, relayPulseTime);
  346. }
  347. delSetting("relayMode");
  348. delSetting("relayPulseMode");
  349. delSetting("relayPulseTime");
  350. }
  351. void _relayBoot() {
  352. _relayRecursive = true;
  353. unsigned char bit = 1;
  354. bool trigger_save = false;
  355. // Get last statuses from EEPROM
  356. unsigned char mask = EEPROMr.read(EEPROM_RELAY_STATUS);
  357. DEBUG_MSG_P(PSTR("[RELAY] Retrieving mask: %d\n"), mask);
  358. // Walk the relays
  359. bool status = false;
  360. for (unsigned int i=0; i<_relays.size(); i++) {
  361. unsigned char boot_mode = getSetting("relayBoot", i, RELAY_BOOT_MODE).toInt();
  362. DEBUG_MSG_P(PSTR("[RELAY] Relay #%d boot mode %d\n"), i, boot_mode);
  363. switch (boot_mode) {
  364. case RELAY_BOOT_SAME:
  365. status = ((mask & bit) == bit);
  366. break;
  367. case RELAY_BOOT_TOGGLE:
  368. status = ((mask & bit) != bit);
  369. mask ^= bit;
  370. trigger_save = true;
  371. break;
  372. case RELAY_BOOT_ON:
  373. status = true;
  374. break;
  375. case RELAY_BOOT_OFF:
  376. default:
  377. status = false;
  378. break;
  379. }
  380. _relays[i].current_status = !status;
  381. _relays[i].target_status = status;
  382. #if RELAY_PROVIDER == RELAY_PROVIDER_STM
  383. _relays[i].change_time = millis() + 3000 + 1000 * i;
  384. #else
  385. _relays[i].change_time = millis();
  386. #endif
  387. bit <<= 1;
  388. }
  389. // Save if there is any relay in the RELAY_BOOT_TOGGLE mode
  390. if (trigger_save) {
  391. EEPROMr.write(EEPROM_RELAY_STATUS, mask);
  392. EEPROMr.commit();
  393. }
  394. _relayRecursive = false;
  395. }
  396. void _relayConfigure() {
  397. for (unsigned int i=0; i<_relays.size(); i++) {
  398. pinMode(_relays[i].pin, OUTPUT);
  399. if (GPIO_NONE != _relays[i].reset_pin) {
  400. pinMode(_relays[i].reset_pin, OUTPUT);
  401. }
  402. if (_relays[i].type == RELAY_TYPE_INVERSE) {
  403. //set to high to block short opening of relay
  404. digitalWrite(_relays[i].pin, HIGH);
  405. }
  406. _relays[i].pulse = getSetting("relayPulse", i, RELAY_PULSE_MODE).toInt();
  407. _relays[i].pulse_ms = 1000 * getSetting("relayTime", i, RELAY_PULSE_MODE).toFloat();
  408. }
  409. }
  410. //------------------------------------------------------------------------------
  411. // WEBSOCKETS
  412. //------------------------------------------------------------------------------
  413. #if WEB_SUPPORT
  414. bool _relayWebSocketOnReceive(const char * key, JsonVariant& value) {
  415. return (strncmp(key, "relay", 5) == 0);
  416. }
  417. void _relayWebSocketUpdate(JsonObject& root) {
  418. JsonArray& relay = root.createNestedArray("relayStatus");
  419. for (unsigned char i=0; i<relayCount(); i++) {
  420. relay.add(_relays[i].target_status);
  421. }
  422. }
  423. void _relayWebSocketOnStart(JsonObject& root) {
  424. if (relayCount() == 0) return;
  425. // Statuses
  426. _relayWebSocketUpdate(root);
  427. // Configuration
  428. JsonArray& config = root.createNestedArray("relayConfig");
  429. for (unsigned char i=0; i<relayCount(); i++) {
  430. JsonObject& line = config.createNestedObject();
  431. line["gpio"] = _relays[i].pin;
  432. line["type"] = _relays[i].type;
  433. line["reset"] = _relays[i].reset_pin;
  434. line["boot"] = getSetting("relayBoot", i, RELAY_BOOT_MODE).toInt();
  435. line["pulse"] = _relays[i].pulse;
  436. line["pulse_ms"] = _relays[i].pulse_ms / 1000.0;
  437. #if MQTT_SUPPORT
  438. line["group"] = getSetting("mqttGroup", i, "");
  439. line["group_inv"] = getSetting("mqttGroupInv", i, 0).toInt();
  440. line["on_disc"] = getSetting("relayOnDisc", i, 0).toInt();
  441. #endif
  442. }
  443. if (relayCount() > 1) {
  444. root["multirelayVisible"] = 1;
  445. root["relaySync"] = getSetting("relaySync", RELAY_SYNC);
  446. }
  447. root["relayVisible"] = 1;
  448. }
  449. void _relayWebSocketOnAction(uint32_t client_id, const char * action, JsonObject& data) {
  450. if (strcmp(action, "relay") != 0) return;
  451. if (data.containsKey("status")) {
  452. unsigned char value = relayParsePayload(data["status"]);
  453. if (value == 3) {
  454. wsSend(_relayWebSocketUpdate);
  455. } else if (value < 3) {
  456. unsigned int relayID = 0;
  457. if (data.containsKey("id")) {
  458. String value = data["id"];
  459. relayID = value.toInt();
  460. }
  461. // Action to perform
  462. if (value == 0) {
  463. relayStatus(relayID, false);
  464. } else if (value == 1) {
  465. relayStatus(relayID, true);
  466. } else if (value == 2) {
  467. relayToggle(relayID);
  468. }
  469. }
  470. }
  471. }
  472. void relaySetupWS() {
  473. wsOnSendRegister(_relayWebSocketOnStart);
  474. wsOnActionRegister(_relayWebSocketOnAction);
  475. wsOnReceiveRegister(_relayWebSocketOnReceive);
  476. }
  477. #endif // WEB_SUPPORT
  478. //------------------------------------------------------------------------------
  479. // REST API
  480. //------------------------------------------------------------------------------
  481. #if API_SUPPORT
  482. void relaySetupAPI() {
  483. char key[20];
  484. // API entry points (protected with apikey)
  485. for (unsigned int relayID=0; relayID<relayCount(); relayID++) {
  486. snprintf_P(key, sizeof(key), PSTR("%s/%d"), MQTT_TOPIC_RELAY, relayID);
  487. apiRegister(key,
  488. [relayID](char * buffer, size_t len) {
  489. snprintf_P(buffer, len, PSTR("%d"), _relays[relayID].target_status ? 1 : 0);
  490. },
  491. [relayID](const char * payload) {
  492. unsigned char value = relayParsePayload(payload);
  493. if (value == 0xFF) {
  494. DEBUG_MSG_P(PSTR("[RELAY] Wrong payload (%s)\n"), payload);
  495. return;
  496. }
  497. if (value == 0) {
  498. relayStatus(relayID, false);
  499. } else if (value == 1) {
  500. relayStatus(relayID, true);
  501. } else if (value == 2) {
  502. relayToggle(relayID);
  503. }
  504. }
  505. );
  506. snprintf_P(key, sizeof(key), PSTR("%s/%d"), MQTT_TOPIC_PULSE, relayID);
  507. apiRegister(key,
  508. [relayID](char * buffer, size_t len) {
  509. dtostrf((double) _relays[relayID].pulse_ms / 1000, 1-len, 3, buffer);
  510. },
  511. [relayID](const char * payload) {
  512. unsigned long pulse = 1000 * String(payload).toFloat();
  513. if (0 == pulse) return;
  514. if (RELAY_PULSE_NONE != _relays[relayID].pulse) {
  515. DEBUG_MSG_P(PSTR("[RELAY] Overriding relay #%d pulse settings\n"), relayID);
  516. }
  517. _relays[relayID].pulse_ms = pulse;
  518. _relays[relayID].pulse = relayStatus(relayID) ? RELAY_PULSE_ON : RELAY_PULSE_OFF;
  519. relayToggle(relayID, true, false);
  520. }
  521. );
  522. #if defined(ITEAD_SONOFF_IFAN02)
  523. apiRegister(MQTT_TOPIC_SPEED,
  524. [relayID](char * buffer, size_t len) {
  525. snprintf(buffer, len, "%u", getSpeed());
  526. },
  527. [relayID](const char * payload) {
  528. setSpeed(atoi(payload));
  529. }
  530. );
  531. #endif
  532. }
  533. }
  534. #endif // API_SUPPORT
  535. //------------------------------------------------------------------------------
  536. // MQTT
  537. //------------------------------------------------------------------------------
  538. #if MQTT_SUPPORT
  539. void relayMQTT(unsigned char id) {
  540. if (id >= _relays.size()) return;
  541. // Send state topic
  542. if (_relays[id].report) {
  543. _relays[id].report = false;
  544. mqttSend(MQTT_TOPIC_RELAY, id, _relays[id].current_status ? RELAY_MQTT_ON : RELAY_MQTT_OFF);
  545. }
  546. // Check group topic
  547. if (_relays[id].group_report) {
  548. _relays[id].group_report = false;
  549. String t = getSetting("mqttGroup", id, "");
  550. if (t.length() > 0) {
  551. bool status = relayStatus(id);
  552. if (getSetting("mqttGroupInv", id, 0).toInt() == 1) status = !status;
  553. mqttSendRaw(t.c_str(), status ? RELAY_MQTT_ON : RELAY_MQTT_OFF);
  554. }
  555. }
  556. // Send speed for IFAN02
  557. #if defined (ITEAD_SONOFF_IFAN02)
  558. char buffer[5];
  559. snprintf(buffer, sizeof(buffer), "%u", getSpeed());
  560. mqttSend(MQTT_TOPIC_SPEED, buffer);
  561. #endif
  562. }
  563. void relayMQTT() {
  564. for (unsigned int id=0; id < _relays.size(); id++) {
  565. mqttSend(MQTT_TOPIC_RELAY, id, _relays[id].current_status ? RELAY_MQTT_ON : RELAY_MQTT_OFF);
  566. }
  567. }
  568. void relayStatusWrap(unsigned char id, unsigned char value, bool is_group_topic) {
  569. switch (value) {
  570. case 0:
  571. relayStatus(id, false, mqttForward(), !is_group_topic);
  572. break;
  573. case 1:
  574. relayStatus(id, true, mqttForward(), !is_group_topic);
  575. break;
  576. case 2:
  577. relayToggle(id, true, true);
  578. break;
  579. default:
  580. _relays[id].report = true;
  581. relayMQTT(id);
  582. break;
  583. }
  584. }
  585. void relayMQTTCallback(unsigned int type, const char * topic, const char * payload) {
  586. if (type == MQTT_CONNECT_EVENT) {
  587. // Send status on connect
  588. #if (HEARTBEAT_MODE == HEARTBEAT_NONE) or (not HEARTBEAT_REPORT_RELAY)
  589. relayMQTT();
  590. #endif
  591. // Subscribe to own /set topic
  592. char relay_topic[strlen(MQTT_TOPIC_RELAY) + 3];
  593. snprintf_P(relay_topic, sizeof(relay_topic), PSTR("%s/+"), MQTT_TOPIC_RELAY);
  594. mqttSubscribe(relay_topic);
  595. // Subscribe to pulse topic
  596. char pulse_topic[strlen(MQTT_TOPIC_PULSE) + 3];
  597. snprintf_P(pulse_topic, sizeof(pulse_topic), PSTR("%s/+"), MQTT_TOPIC_PULSE);
  598. mqttSubscribe(pulse_topic);
  599. #if defined(ITEAD_SONOFF_IFAN02)
  600. mqttSubscribe(MQTT_TOPIC_SPEED);
  601. #endif
  602. // Subscribe to group topics
  603. for (unsigned int i=0; i < _relays.size(); i++) {
  604. String t = getSetting("mqttGroup", i, "");
  605. if (t.length() > 0) mqttSubscribeRaw(t.c_str());
  606. }
  607. }
  608. if (type == MQTT_MESSAGE_EVENT) {
  609. String t = mqttMagnitude((char *) topic);
  610. // magnitude is relay/#/pulse
  611. if (t.startsWith(MQTT_TOPIC_PULSE)) {
  612. unsigned int id = t.substring(strlen(MQTT_TOPIC_PULSE)+1).toInt();
  613. if (id >= relayCount()) {
  614. DEBUG_MSG_P(PSTR("[RELAY] Wrong relayID (%d)\n"), id);
  615. return;
  616. }
  617. unsigned long pulse = 1000 * String(payload).toFloat();
  618. if (0 == pulse) return;
  619. if (RELAY_PULSE_NONE != _relays[id].pulse) {
  620. DEBUG_MSG_P(PSTR("[RELAY] Overriding relay #%d pulse settings\n"), id);
  621. }
  622. _relays[id].pulse_ms = pulse;
  623. _relays[id].pulse = relayStatus(id) ? RELAY_PULSE_ON : RELAY_PULSE_OFF;
  624. relayToggle(id, true, false);
  625. return;
  626. }
  627. // magnitude is relay/#
  628. if (t.startsWith(MQTT_TOPIC_RELAY)) {
  629. // Get relay ID
  630. unsigned int id = t.substring(strlen(MQTT_TOPIC_RELAY)+1).toInt();
  631. if (id >= relayCount()) {
  632. DEBUG_MSG_P(PSTR("[RELAY] Wrong relayID (%d)\n"), id);
  633. return;
  634. }
  635. // Get value
  636. unsigned char value = relayParsePayload(payload);
  637. if (value == 0xFF) return;
  638. relayStatusWrap(id, value, false);
  639. return;
  640. }
  641. // Check group topics
  642. for (unsigned int i=0; i < _relays.size(); i++) {
  643. String t = getSetting("mqttGroup", i, "");
  644. if ((t.length() > 0) && t.equals(topic)) {
  645. unsigned char value = relayParsePayload(payload);
  646. if (value == 0xFF) return;
  647. if (value < 2) {
  648. if (getSetting("mqttGroupInv", i, 0).toInt() == 1) {
  649. value = 1 - value;
  650. }
  651. }
  652. DEBUG_MSG_P(PSTR("[RELAY] Matched group topic for relayID %d\n"), i);
  653. relayStatusWrap(i, value, true);
  654. }
  655. }
  656. // Itead Sonoff IFAN02
  657. #if defined (ITEAD_SONOFF_IFAN02)
  658. if (t.startsWith(MQTT_TOPIC_SPEED)) {
  659. setSpeed(atoi(payload));
  660. }
  661. #endif
  662. }
  663. if (type == MQTT_DISCONNECT_EVENT) {
  664. for (unsigned int i=0; i < _relays.size(); i++){
  665. int reaction = getSetting("relayOnDisc", i, 0).toInt();
  666. if (1 == reaction) { // switch relay OFF
  667. DEBUG_MSG_P(PSTR("[RELAY] Reset relay (%d) due to MQTT disconnection\n"), i);
  668. relayStatusWrap(i, false, false);
  669. } else if(2 == reaction) { // switch relay ON
  670. DEBUG_MSG_P(PSTR("[RELAY] Set relay (%d) due to MQTT disconnection\n"), i);
  671. relayStatusWrap(i, true, false);
  672. }
  673. }
  674. }
  675. }
  676. void relaySetupMQTT() {
  677. mqttRegister(relayMQTTCallback);
  678. }
  679. #endif
  680. //------------------------------------------------------------------------------
  681. // InfluxDB
  682. //------------------------------------------------------------------------------
  683. #if INFLUXDB_SUPPORT
  684. void relayInfluxDB(unsigned char id) {
  685. if (id >= _relays.size()) return;
  686. idbSend(MQTT_TOPIC_RELAY, id, relayStatus(id) ? "1" : "0");
  687. }
  688. #endif
  689. //------------------------------------------------------------------------------
  690. // Settings
  691. //------------------------------------------------------------------------------
  692. #if TERMINAL_SUPPORT
  693. void _relayInitCommands() {
  694. settingsRegisterCommand(F("RELAY"), [](Embedis* e) {
  695. if (e->argc < 2) {
  696. DEBUG_MSG_P(PSTR("-ERROR: Wrong arguments\n"));
  697. return;
  698. }
  699. int id = String(e->argv[1]).toInt();
  700. if (id >= relayCount()) {
  701. DEBUG_MSG_P(PSTR("-ERROR: Wrong relayID (%d)\n"), id);
  702. return;
  703. }
  704. if (e->argc > 2) {
  705. int value = String(e->argv[2]).toInt();
  706. if (value == 2) {
  707. relayToggle(id);
  708. } else {
  709. relayStatus(id, value == 1);
  710. }
  711. }
  712. DEBUG_MSG_P(PSTR("Status: %s\n"), _relays[id].target_status ? "true" : "false");
  713. if (_relays[id].pulse != RELAY_PULSE_NONE) {
  714. DEBUG_MSG_P(PSTR("Pulse: %s\n"), (_relays[id].pulse == RELAY_PULSE_ON) ? "ON" : "OFF");
  715. DEBUG_MSG_P(PSTR("Pulse time: %d\n"), _relays[id].pulse_ms);
  716. }
  717. DEBUG_MSG_P(PSTR("+OK\n"));
  718. });
  719. }
  720. #endif // TERMINAL_SUPPORT
  721. //------------------------------------------------------------------------------
  722. // Setup
  723. //------------------------------------------------------------------------------
  724. void _relayLoop() {
  725. _relayProcess(false);
  726. _relayProcess(true);
  727. }
  728. void relaySetup() {
  729. // Dummy relays for AI Light, Magic Home LED Controller, H801,
  730. // Sonoff Dual and Sonoff RF Bridge
  731. #if DUMMY_RELAY_COUNT > 0
  732. // No delay_on or off for these devices to easily allow having more than
  733. // 8 channels. This behaviour will be recovered with v2.
  734. for (unsigned char i=0; i < DUMMY_RELAY_COUNT; i++) {
  735. _relays.push_back((relay_t) {0, RELAY_TYPE_NORMAL, 0, 0, 0});
  736. }
  737. #else
  738. #if RELAY1_PIN != GPIO_NONE
  739. _relays.push_back((relay_t) { RELAY1_PIN, RELAY1_TYPE, RELAY1_RESET_PIN, RELAY1_DELAY_ON, RELAY1_DELAY_OFF });
  740. #endif
  741. #if RELAY2_PIN != GPIO_NONE
  742. _relays.push_back((relay_t) { RELAY2_PIN, RELAY2_TYPE, RELAY2_RESET_PIN, RELAY2_DELAY_ON, RELAY2_DELAY_OFF });
  743. #endif
  744. #if RELAY3_PIN != GPIO_NONE
  745. _relays.push_back((relay_t) { RELAY3_PIN, RELAY3_TYPE, RELAY3_RESET_PIN, RELAY3_DELAY_ON, RELAY3_DELAY_OFF });
  746. #endif
  747. #if RELAY4_PIN != GPIO_NONE
  748. _relays.push_back((relay_t) { RELAY4_PIN, RELAY4_TYPE, RELAY4_RESET_PIN, RELAY4_DELAY_ON, RELAY4_DELAY_OFF });
  749. #endif
  750. #if RELAY5_PIN != GPIO_NONE
  751. _relays.push_back((relay_t) { RELAY5_PIN, RELAY5_TYPE, RELAY5_RESET_PIN, RELAY5_DELAY_ON, RELAY5_DELAY_OFF });
  752. #endif
  753. #if RELAY6_PIN != GPIO_NONE
  754. _relays.push_back((relay_t) { RELAY6_PIN, RELAY6_TYPE, RELAY6_RESET_PIN, RELAY6_DELAY_ON, RELAY6_DELAY_OFF });
  755. #endif
  756. #if RELAY7_PIN != GPIO_NONE
  757. _relays.push_back((relay_t) { RELAY7_PIN, RELAY7_TYPE, RELAY7_RESET_PIN, RELAY7_DELAY_ON, RELAY7_DELAY_OFF });
  758. #endif
  759. #if RELAY8_PIN != GPIO_NONE
  760. _relays.push_back((relay_t) { RELAY8_PIN, RELAY8_TYPE, RELAY8_RESET_PIN, RELAY8_DELAY_ON, RELAY8_DELAY_OFF });
  761. #endif
  762. #endif
  763. _relayBackwards();
  764. _relayConfigure();
  765. _relayBoot();
  766. _relayLoop();
  767. #if WEB_SUPPORT
  768. relaySetupWS();
  769. #endif
  770. #if API_SUPPORT
  771. relaySetupAPI();
  772. #endif
  773. #if MQTT_SUPPORT
  774. relaySetupMQTT();
  775. #endif
  776. #if TERMINAL_SUPPORT
  777. _relayInitCommands();
  778. #endif
  779. // Main callbacks
  780. espurnaRegisterLoop(_relayLoop);
  781. espurnaRegisterReload(_relayConfigure);
  782. DEBUG_MSG_P(PSTR("[RELAY] Number of relays: %d\n"), _relays.size());
  783. }