Add Safe Mode Restart Switch (#2437)

esphome/components/safe_mode/__init__.py

@@ -0,0 +1,5 @@
+import esphome.codegen as cg
+
+CODEOWNERS = ["@paulmonigatti"]
+
+safe_mode_ns = cg.esphome_ns.namespace("safe_mode")

esphome/components/safe_mode/switch/__init__.py

@@ -0,0 +1,36 @@
+import esphome.codegen as cg
+import esphome.config_validation as cv
+from esphome.components import switch
+from esphome.components.ota import OTAComponent
+from esphome.const import (
+    CONF_ID,
+    CONF_INVERTED,
+    CONF_ICON,
+    CONF_OTA,
+    ICON_RESTART_ALERT,
+)
+from .. import safe_mode_ns
+
+DEPENDENCIES = ["ota"]
+
+SafeModeSwitch = safe_mode_ns.class_("SafeModeSwitch", switch.Switch, cg.Component)
+
+CONFIG_SCHEMA = switch.SWITCH_SCHEMA.extend(
+    {
+        cv.GenerateID(): cv.declare_id(SafeModeSwitch),
+        cv.GenerateID(CONF_OTA): cv.use_id(OTAComponent),
+        cv.Optional(CONF_INVERTED): cv.invalid(
+            "Safe Mode Restart switches do not support inverted mode!"
+        ),
+        cv.Optional(CONF_ICON, default=ICON_RESTART_ALERT): switch.icon,
+    }
+).extend(cv.COMPONENT_SCHEMA)
+
+
+async def to_code(config):
+    var = cg.new_Pvariable(config[CONF_ID])
+    await cg.register_component(var, config)
+    await switch.register_switch(var, config)
+
+    ota = await cg.get_variable(config[CONF_OTA])
+    cg.add(var.set_ota(ota))

esphome/components/safe_mode/switch/safe_mode_switch.cpp

@@ -0,0 +1,29 @@
+#include "safe_mode_switch.h"
+#include "esphome/core/hal.h"
+#include "esphome/core/log.h"
+#include "esphome/core/application.h"
+
+namespace esphome {
+namespace safe_mode {
+
+static const char *const TAG = "safe_mode_switch";
+
+void SafeModeSwitch::set_ota(ota::OTAComponent *ota) { this->ota_ = ota; }
+
+void SafeModeSwitch::write_state(bool state) {
+  // Acknowledge
+  this->publish_state(false);
+
+  if (state) {
+    ESP_LOGI(TAG, "Restarting device in safe mode...");
+    this->ota_->set_safe_mode_pending(true);
+
+    // Let MQTT settle a bit
+    delay(100);  // NOLINT
+    App.safe_reboot();
+  }
+}
+void SafeModeSwitch::dump_config() { LOG_SWITCH("", "Safe Mode Switch", this); }
+
+}  // namespace safe_mode
+}  // namespace esphome

esphome/components/safe_mode/switch/safe_mode_switch.h

@@ -0,0 +1,21 @@
+#pragma once
+
+#include "esphome/core/component.h"
+#include "esphome/components/ota/ota_component.h"
+#include "esphome/components/switch/switch.h"
+
+namespace esphome {
+namespace safe_mode {
+
+class SafeModeSwitch : public switch_::Switch, public Component {
+ public:
+  void dump_config() override;
+  void set_ota(ota::OTAComponent *ota);
+
+ protected:
+  ota::OTAComponent *ota_;
+  void write_state(bool state) override;
+};
+
+}  // namespace safe_mode
+}  // namespace esphome