From a918a039abdba7afab8c639cf7a791c293faaff6 Mon Sep 17 00:00:00 2001
From: Tomaae <tomaaecontact@gmail.com>
Date: Mon, 24 Jan 2022 12:47:19 +0100
Subject: [PATCH] Redid integration remove to unload platform

---
 custom_components/mikrotik_router/__init__.py | 15 ++++++++-------
 custom_components/mikrotik_router/const.py    |  8 ++++++++
 2 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/custom_components/mikrotik_router/__init__.py b/custom_components/mikrotik_router/__init__.py
index e5436b0..8bfdbbd 100644
--- a/custom_components/mikrotik_router/__init__.py
+++ b/custom_components/mikrotik_router/__init__.py
@@ -12,6 +12,7 @@ from homeassistant.const import (
 )
 
 from .const import (
+    PLATFORMS,
     DOMAIN,
     DATA_CLIENT,
     RUN_SCRIPT_COMMAND,
@@ -104,12 +105,12 @@ async def async_setup_entry(hass, config_entry):
 async def async_unload_entry(hass, config_entry):
     """Unload a config entry."""
     controller = hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id]
-    await hass.config_entries.async_forward_entry_unload(config_entry, "sensor")
-    await hass.config_entries.async_forward_entry_unload(config_entry, "binary_sensor")
-    await hass.config_entries.async_forward_entry_unload(config_entry, "device_tracker")
-    await hass.config_entries.async_forward_entry_unload(config_entry, "switch")
-    await hass.config_entries.async_forward_entry_unload(config_entry, "button")
-    hass.services.async_remove(DOMAIN, RUN_SCRIPT_COMMAND)
     await controller.async_reset()
+    hass.services.async_remove(DOMAIN, RUN_SCRIPT_COMMAND)
     hass.data[DOMAIN][DATA_CLIENT].pop(config_entry.entry_id)
-    return True
+
+    unload_ok = await hass.config_entries.async_unload_platforms(
+        config_entry, PLATFORMS
+    )
+
+    return unload_ok
diff --git a/custom_components/mikrotik_router/const.py b/custom_components/mikrotik_router/const.py
index 088e240..28bbd97 100644
--- a/custom_components/mikrotik_router/const.py
+++ b/custom_components/mikrotik_router/const.py
@@ -1,5 +1,13 @@
 """Constants used by the Mikrotik Router component and platforms."""
+from homeassistant.const import Platform
 
+PLATFORMS = [
+    Platform.SENSOR,
+    Platform.BINARY_SENSOR,
+    Platform.DEVICE_TRACKER,
+    Platform.SWITCH,
+    Platform.BUTTON,
+]
 DOMAIN = "mikrotik_router"
 DEFAULT_NAME = "Mikrotik Router"
 DATA_CLIENT = "client"