have fun

components/m5stack_4relay/m5stack_4relay.cpp

@@ -0,0 +1,49 @@
+#include "m5stack_4relay.h"
+#include <bitset>
+#include "esphome/core/log.h"
+namespace esphome {
+namespace m5stack_4relay {
+static const char *const TAG = "m5stack_4relay.switch";
+
+void M5STACK4RELAYSwitchComponent::setup() {
+  ESP_LOGCONFIG(TAG, "Setting up M5STACK4RELAY (0x%02X)...", this->address_);
+  component_status = 0;
+  get_status(&component_status);
+  ESP_LOGD(TAG, "Setup Status 0x%02X", component_status);
+}
+
+M5STACK4RELAYChannel *M5STACK4RELAYSwitchComponent::create_channel(uint8_t channel) {
+  return new M5STACK4RELAYChannel(this, channel);
+}
+
+void M5STACK4RELAYSwitchComponent::get_status(uint8_t *state) { this->read_byte(RELAY_CONTROL_REG); }
+
+bool M5STACK4RELAYSwitchComponent::set_channel_value_(uint8_t channel, bool state) {
+  get_status(&component_status);
+
+  ESP_LOGD(TAG, "Current status 0x%02X", component_status);
+  ESP_LOGD(TAG, "Desired channel: %1u", channel);
+  ESP_LOGD(TAG, "Desired state: %1u", state);
+
+  channel = 3 - channel;
+  if (state) {
+    component_status |= (1u << channel);
+  } else {
+    component_status &= ~(1u << channel);
+  }
+
+  ESP_LOGD(TAG, "New status 0x%02X", component_status);
+
+  return this->write_byte(this->address_, RELAY_CONTROL_REG, component_status);
+}
+
+void M5STACK4RELAYChannel::write_state(bool state) {
+  if (!this->set_channel_value_(this->channel_, state)) {
+    publish_state(false);
+  } else {
+    publish_state(state);
+  }
+}
+
+}  // namespace m5stack_4relay
+}  // namespace esphome