From a40d4a5efb57e098e3dac73ce3d8a4e99abed331 Mon Sep 17 00:00:00 2001
From: Jason Hunter <hunterjm@gmail.com>
Date: Sun, 17 Mar 2019 23:52:21 -0400
Subject: [PATCH] Fix resetting access token on streams with keepalive

---
 homeassistant/components/stream/__init__.py |  6 ++++++
 homeassistant/components/stream/core.py     | 16 +++++++++++++---
 2 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/stream/__init__.py b/homeassistant/components/stream/__init__.py
index 1d04791a11ab..3f715af0e047 100644
--- a/homeassistant/components/stream/__init__.py
+++ b/homeassistant/components/stream/__init__.py
@@ -120,10 +120,16 @@ def remove_provider(self, provider):
         """Remove provider output stream."""
         if provider.format in self._outputs:
             del self._outputs[provider.format]
+            self.check_idle()
 
         if not self._outputs:
             self.stop()
 
+    def check_idle(self):
+        """Reset access token if all providers are idle."""
+        if all([p.idle for p in self._outputs.values()]):
+            self.access_token = None
+
     def start(self):
         """Start a stream."""
         if self._thread is None or not self._thread.isAlive():
diff --git a/homeassistant/components/stream/core.py b/homeassistant/components/stream/core.py
index 3d6ffa0e20c4..665803d38ebc 100644
--- a/homeassistant/components/stream/core.py
+++ b/homeassistant/components/stream/core.py
@@ -43,6 +43,7 @@ class StreamOutput:
 
     def __init__(self, stream) -> None:
         """Initialize a stream output."""
+        self.idle = False
         self._stream = stream
         self._cursor = None
         self._event = asyncio.Event()
@@ -77,10 +78,11 @@ def target_duration(self) -> int:
 
     def get_segment(self, sequence: int = None) -> Any:
         """Retrieve a specific segment, or the whole list."""
+        self.idle = False
         # Reset idle timeout
         if self._unsub is not None:
             self._unsub()
-        self._unsub = async_call_later(self._stream.hass, 300, self._cleanup)
+        self._unsub = async_call_later(self._stream.hass, 300, self._timeout)
 
         if not sequence:
             return self._segments
@@ -109,7 +111,7 @@ def put(self, segment: Segment) -> None:
         # Start idle timeout when we start recieving data
         if self._unsub is None:
             self._unsub = async_call_later(
-                self._stream.hass, 300, self._cleanup)
+                self._stream.hass, 300, self._timeout)
 
         if segment is None:
             self._event.set()
@@ -124,7 +126,15 @@ def put(self, segment: Segment) -> None:
         self._event.clear()
 
     @callback
-    def _cleanup(self, _now=None):
+    def _timeout(self, _now=None):
+        """Handle stream timeout."""
+        if self._stream.keepalive:
+            self.idle = True
+            self._stream.check_idle()
+        else:
+            self._cleanup()
+
+    def _cleanup(self):
         """Remove provider."""
         self._segments = []
         self._stream.remove_provider(self)