From 2a45758a6da72eda995023f592614581f3f4864c Mon Sep 17 00:00:00 2001
From: Yegor Vialov <vyalov.egor@gmail.com>
Date: Fri, 25 Jan 2019 22:55:41 +0200
Subject: [PATCH] Resolves #268 Badges for Lovelace UI

---
 lib/home_assistant.class.dart | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/lib/home_assistant.class.dart b/lib/home_assistant.class.dart
index 7bf5b33..53b01e1 100644
--- a/lib/home_assistant.class.dart
+++ b/lib/home_assistant.class.dart
@@ -417,6 +417,16 @@ class HomeAssistant {
             name: rawView['title'],
             iconName: rawView['icon']
         );
+
+        if (rawView['badges'] != null && rawView['badges'] is List) {
+          rawView['badges'].forEach((entity) {
+            if (entities.isExist(entity)) {
+              Entity e = entities.get(entity);
+              view.badges.add(e);
+            }
+          });
+        }
+
         view.cards.addAll(_createLovelaceCards(rawView["cards"] ?? []));
         ui.views.add(
             view