Add WiFi/MQTT/API connected condition (#536)

* Add WiFi/MQTT/API connected condition

* Lint

src/esphome/api/api_server.cpp

@@ -75,7 +75,7 @@ void APIServer::loop() {
 
   if (this->reboot_timeout_ != 0) {
     const uint32_t now = millis();
-    if (this->clients_.empty()) {
+    if (!this->is_connected()) {
       if (now - this->last_connected_ > this->reboot_timeout_) {
         ESP_LOGE(TAG, "No client connected to API. Rebooting...");
         reboot("api");
@@ -211,6 +211,7 @@ void APIServer::request_time() {
   }
 }
 #endif
+bool APIServer::is_connected() const { return !this->clients_.empty(); }
 
 // APIConnection
 APIConnection::APIConnection(AsyncClient *client, APIServer *parent)

src/esphome/api/api_server.h

@@ -182,6 +182,8 @@ class APIServer : public Component, public StoringUpdateListenerController {
   void request_time();
 #endif
 
+  bool is_connected() const;
+
   struct HomeAssistantStateSubscription {
     std::string entity_id;
     std::function<void(std::string)> callback;
@@ -218,6 +220,11 @@ template<typename... Ts> class HomeAssistantServiceCallAction : public Action<Ts
   ServiceCallResponse resp_;
 };
 
+template<typename... Ts> class APIConnectedCondition : public Condition<Ts...> {
+ public:
+  bool check(Ts... x) override;
+};
+
 template<typename... Ts> HomeAssistantServiceCallAction<Ts...> *APIServer::make_home_assistant_service_call_action() {
   return new HomeAssistantServiceCallAction<Ts...>(this);
 }
@@ -248,6 +255,8 @@ UserService<Ts...> *APIServer::make_user_service_trigger(const std::string &name
   return service;
 }
 
+template<typename... Ts> bool APIConnectedCondition<Ts...>::check(Ts... x) { return global_api_server->is_connected(); }
+
 }  // namespace api
 
 ESPHOME_NAMESPACE_END

src/esphome/mqtt/mqtt_client_component.h

@@ -345,6 +345,11 @@ template<typename... Ts> class MQTTPublishJsonAction : public Action<Ts...> {
   bool retain_{false};
 };
 
+template<typename... Ts> class MQTTConnectedCondition : public Condition<Ts...> {
+ public:
+  bool check(Ts... x) override;
+};
+
 // =============== TEMPLATE DEFINITIONS ===============
 
 template<typename... Ts>
@@ -378,6 +383,10 @@ template<typename... Ts> MQTTPublishAction<Ts...> *MQTTClientComponent::make_pub
   return new MQTTPublishAction<Ts...>();
 }
 
+template<typename... Ts> bool MQTTConnectedCondition<Ts...>::check(Ts... x) {
+  return global_mqtt_client->is_connected();
+}
+
 }  // namespace mqtt
 
 ESPHOME_NAMESPACE_END

src/esphome/wifi_component.h

@@ -14,6 +14,7 @@
 #include <ESP8266WiFi.h>
 #endif
 
+#include "esphome/automation.h"
 #include "esphome/component.h"
 #include "esphome/helpers.h"
 #include "esphome/defines.h"
@@ -224,8 +225,17 @@ class WiFiComponent : public Component {
   bool ap_setup_{false};
 };
 
+template<typename... Ts> class WiFiConnectedCondition : public Condition<Ts...> {
+ public:
+  bool check(Ts... x) override;
+};
+
 extern WiFiComponent *global_wifi_component;
 
+template<typename... Ts> bool WiFiConnectedCondition<Ts...>::check(Ts... x) {
+  return global_wifi_component->is_connected();
+}
+
 ESPHOME_NAMESPACE_END
 
 #endif  // ESPHOME_WIFI_COMPONENT_H