From d7bb9013d4a33aa1e54ffd77c1b981e73f3bba3d Mon Sep 17 00:00:00 2001
From: Pascal Vizeli <pascal.vizeli@syshack.ch>
Date: Sun, 7 Apr 2019 14:42:07 +0200
Subject: [PATCH] Improve add-on rebuild (#1011)

* Check version

* Use image instead next_image
---
 hassio/addons/addon.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/hassio/addons/addon.py b/hassio/addons/addon.py
index 63293eb87..d451b9eb2 100644
--- a/hassio/addons/addon.py
+++ b/hassio/addons/addon.py
@@ -977,8 +977,11 @@ class Addon(CoreSysAttributes):
         last_state = await self.state()
 
         if not self.need_build:
-            _LOGGER.error("Can't rebuild a none local build add-on!")
+            _LOGGER.error("Can't rebuild a image based add-on")
             raise AddonsNotSupportedError()
+        if self.version_installed != self.last_version:
+            _LOGGER.error("Version changed, use Update instead Rebuild")
+            raise AddonsError()
 
         # remove docker container but not addon config
         try:
@@ -986,6 +989,8 @@ class Addon(CoreSysAttributes):
             await self.instance.install(self.version_installed)
         except DockerAPIError:
             raise AddonsError() from None
+        else:
+            self._set_update(self.image, self.version_installed)
 
         # restore state
         if last_state == STATE_STARTED: