From f82f30dc6247b21a4db304140d4a603d018e09b2 Mon Sep 17 00:00:00 2001
From: jjlawren <jjlawren@users.noreply.github.com>
Date: Sun, 22 Sep 2019 16:47:41 -0500
Subject: [PATCH] Unload Plex config entries (#26771)

* Unload config entries

* Await coroutines

* Unnecessary ensure
---
 homeassistant/components/plex/__init__.py     | 23 ++++++++++++++++++-
 homeassistant/components/plex/const.py        |  1 +
 homeassistant/components/plex/media_player.py |  5 +++-
 3 files changed, 27 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/plex/__init__.py b/homeassistant/components/plex/__init__.py
index 665091d69b9..dd458dda078 100644
--- a/homeassistant/components/plex/__init__.py
+++ b/homeassistant/components/plex/__init__.py
@@ -1,4 +1,5 @@
 """Support to embed Plex."""
+import asyncio
 import logging
 
 import plexapi.exceptions
@@ -21,6 +22,7 @@ from .const import (
     CONF_USE_EPISODE_ART,
     CONF_SHOW_ALL_CONTROLS,
     CONF_SERVER,
+    CONF_SERVER_IDENTIFIER,
     DEFAULT_PORT,
     DEFAULT_SSL,
     DEFAULT_VERIFY_SSL,
@@ -28,6 +30,7 @@ from .const import (
     PLATFORMS,
     PLEX_MEDIA_PLAYER_OPTIONS,
     PLEX_SERVER_CONFIG,
+    REFRESH_LISTENERS,
     SERVERS,
 )
 from .server import PlexServer
@@ -61,7 +64,7 @@ _LOGGER = logging.getLogger(__package__)
 
 def setup(hass, config):
     """Set up the Plex component."""
-    hass.data.setdefault(PLEX_DOMAIN, {SERVERS: {}})
+    hass.data.setdefault(PLEX_DOMAIN, {SERVERS: {}, REFRESH_LISTENERS: {}})
 
     plex_config = config.get(PLEX_DOMAIN, {})
     if plex_config:
@@ -129,3 +132,21 @@ async def async_setup_entry(hass, entry):
         )
 
     return True
+
+
+async def async_unload_entry(hass, entry):
+    """Unload a config entry."""
+    server_id = entry.data[CONF_SERVER_IDENTIFIER]
+
+    cancel = hass.data[PLEX_DOMAIN][REFRESH_LISTENERS].pop(server_id)
+    await hass.async_add_executor_job(cancel)
+
+    tasks = [
+        hass.config_entries.async_forward_entry_unload(entry, platform)
+        for platform in PLATFORMS
+    ]
+    await asyncio.gather(*tasks)
+
+    hass.data[PLEX_DOMAIN][SERVERS].pop(server_id)
+
+    return True
diff --git a/homeassistant/components/plex/const.py b/homeassistant/components/plex/const.py
index e77ac303bf1..478dd3754e7 100644
--- a/homeassistant/components/plex/const.py
+++ b/homeassistant/components/plex/const.py
@@ -7,6 +7,7 @@ DEFAULT_SSL = False
 DEFAULT_VERIFY_SSL = True
 
 PLATFORMS = ["media_player", "sensor"]
+REFRESH_LISTENERS = "refresh_listeners"
 SERVERS = "servers"
 
 PLEX_CONFIG_FILE = "plex.conf"
diff --git a/homeassistant/components/plex/media_player.py b/homeassistant/components/plex/media_player.py
index bc19ff41dfe..4d097253ea1 100644
--- a/homeassistant/components/plex/media_player.py
+++ b/homeassistant/components/plex/media_player.py
@@ -39,6 +39,7 @@ from .const import (
     DOMAIN as PLEX_DOMAIN,
     NAME_FORMAT,
     PLEX_MEDIA_PLAYER_OPTIONS,
+    REFRESH_LISTENERS,
     SERVERS,
 )
 
@@ -71,7 +72,9 @@ def _setup_platform(hass, config_entry, add_entities_callback):
     plexserver = hass.data[PLEX_DOMAIN][SERVERS][server_id]
     plex_clients = {}
     plex_sessions = {}
-    track_time_interval(hass, lambda now: update_devices(), timedelta(seconds=10))
+    hass.data[PLEX_DOMAIN][REFRESH_LISTENERS][server_id] = track_time_interval(
+        hass, lambda now: update_devices(), timedelta(seconds=10)
+    )
 
     def update_devices():
         """Update the devices objects."""