From 252f0692c870e3a81926feee88cf7fa8f7d18c98 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <paulus@home-assistant.io>
Date: Thu, 18 Oct 2018 13:21:06 +0200
Subject: [PATCH] Add hass on badge (#1802)

* Add hass on badge

* Add more hass everywhere
---
 src/panels/lovelace/elements/hui-state-badge-element.js | 7 +++++--
 src/panels/lovelace/hui-view.js                         | 5 ++++-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/panels/lovelace/elements/hui-state-badge-element.js b/src/panels/lovelace/elements/hui-state-badge-element.js
index c4b7d2ea14..224f55ae50 100644
--- a/src/panels/lovelace/elements/hui-state-badge-element.js
+++ b/src/panels/lovelace/elements/hui-state-badge-element.js
@@ -22,8 +22,11 @@ class HuiStateBadgeElement extends PolymerElement {
     }
 
     const stateObj = hass.states[config.entity];
-    this._badge.state = stateObj;
-    this._badge.setAttribute("title", computeStateName(stateObj));
+    this._badge.setProperties({
+      hass,
+      state: stateObj,
+      title: computeStateName(stateObj),
+    });
 
     if (!this.lastChild) {
       this.appendChild(this._badge);
diff --git a/src/panels/lovelace/hui-view.js b/src/panels/lovelace/hui-view.js
index cf27e5dce9..b13edb5a10 100644
--- a/src/panels/lovelace/hui-view.js
+++ b/src/panels/lovelace/hui-view.js
@@ -107,7 +107,10 @@ class HUIView extends PolymerElement {
       if (!(entityId in this.hass.states)) continue;
 
       const element = document.createElement("ha-state-label-badge");
-      element.state = this.hass.states[entityId];
+      element.setProperties({
+        hass: this.hass,
+        state: this.hass.states[entityId],
+      });
       elements.push({ element, entityId });
       root.appendChild(element);
     }