1 files changed, 18 insertions, 4 deletions

diff --git a/recipes/home-assistant/webview.js b/recipes/home-assistant/webview.js
index 7a0ab47..aee42c0 100644
--- a/recipes/home-assistant/webview.js
+++ b/recipes/home-assistant/webview.js
@@ -1,12 +1,26 @@
+function _interopRequireDefault(obj) {
+  return obj && obj.__esModule ? obj : { default: obj };
+}
+
+const _path = _interopRequireDefault(require('path'));
+
 module.exports = Ferdium => {
   const getMessages = () => {
-    const badges = document.querySelector("home-assistant").shadowRoot.querySelector("home-assistant-main").shadowRoot.querySelector("ha-sidebar").shadowRoot.querySelectorAll(".notification-badge");
+    const badges = document
+      .querySelector('home-assistant')
+      .shadowRoot.querySelector('home-assistant-main')
+      .shadowRoot.querySelector('ha-sidebar')
+      .shadowRoot.querySelectorAll('.notification-badge');
     if (badges.length > 0) {
-      var count = Ferdium.safeParseInt(badges[0].textContent.replace(/[^\p{N}]/gu, ''));
-      Ferdium.setBadge(count)
+      const count = Ferdium.safeParseInt(
+        badges[0].textContent.replaceAll(/[^\p{N}]/gu, ''),
+      );
+      Ferdium.setBadge(count);
     } else {
-      Ferdium.setBadge(0)
+      Ferdium.setBadge(0);
     }
   };
   Ferdium.loop(getMessages);
+
+  Ferdium.injectCSS(_path.default.join(__dirname, 'service.css'));
 };