Make entities unavailable on disconnect. Fix #407

custom_components/browser_mod/browser.py

@@ -172,15 +172,17 @@ def connection(self):
         """The current websocket connections for this Browser."""
         return self._connections
 
-    def open_connection(self, connection, cid):
+    def open_connection(self, hass, connection, cid):
         """Add a websocket connection."""
         self._connections.append((connection, cid))
+        self.update(hass, {"connected": True})
 
-    def close_connection(self, connection):
+    def close_connection(self, hass, connection):
         """Close a websocket connection."""
         self._connections = list(
             filter(lambda v: v[0] != connection, self._connections)
         )
+        self.update(hass, {"connected": False})
 
 
 def getBrowser(hass, browserID, *, create=True):

custom_components/browser_mod/connection.py

@@ -52,15 +52,15 @@ def close_connection():
             store_listener()
             dev = getBrowser(hass, browserID, create=False)
             if dev:
-                dev.close_connection(connection)
+                dev.close_connection(hass, connection)
 
         connection.subscriptions[msg["id"]] = close_connection
         connection.send_result(msg["id"])
 
         if store.get_browser(browserID).registered:
             dev = getBrowser(hass, browserID)
             dev.update_settings(hass, store.get_browser(browserID).asdict())
-            dev.open_connection(connection, msg["id"])
+            dev.open_connection(hass, connection, msg["id"])
             await store.set_browser(
                 browserID,
                 last_seen=datetime.now(tz=timezone.utc).isoformat(),

custom_components/browser_mod/entities.py

@@ -40,6 +40,10 @@ def extra_state_attributes(self):
             "browserID": self.browserID,
         }
 
+    @property
+    def available(self):
+        return self._data.get("connected", False)
+
     @property
     def name(self):
         return self._name