have fun

components/m5stack_4relay/m5stack_4relay.h

@@ -0,0 +1,49 @@
+#pragma once
+
+#include "esphome/components/i2c/i2c.h"
+#include "esphome/components/switch/switch.h"
+#include "esphome/core/component.h"
+#include "esphome/core/log.h"
+
+namespace esphome {
+namespace m5stack_4relay {
+
+class M5STACK4RELAYSwitchComponent;
+class M5STACK4RELAYChannel : public switch_::Switch {
+ public:
+  M5STACK4RELAYChannel(M5STACK4RELAYSwitchComponent *parent, uint8_t channel) : parent_(parent), channel_(channel) {}
+
+ protected:
+  void write_state(bool state) override;
+
+  M5STACK4RELAYSwitchComponent *parent_;
+  uint8_t channel_;
+};
+
+class M5STACK4RELAYSwitchComponent : public Component, public i2c::I2CDevice {
+ public:
+  const uint8_t MODE_CONTROL_REG = 0x10;
+  const uint8_t RELAY_CONTROL_REG = 0x11;
+  uint8_t component_status;
+  M5STACK4RELAYSwitchComponent() {}
+  M5STACK4RELAYChannel *create_channel(uint8_t channel);
+
+  void setup() override;
+  float get_setup_priority() const override { return setup_priority::HARDWARE; }
+
+ protected:
+  friend M5STACK4RELAYChannel;
+
+  void get_status(uint8_t *state);
+
+  bool set_channel_value_(uint8_t channel, bool state);
+
+  bool read_bytes_(uint8_t a_register, uint8_t *data, uint8_t len, uint32_t conversion) {
+    return this->raw_receive(this->address_, data, len);
+  }
+
+  float value_;
+};
+
+}  // namespace m5stack_4relay
+}  // namespace esphome