feat(integration): Handle nodered:loaded event from when HA loads cus…

…tom component
zachowj · Feb 12, 2020 · 072cec7 · 072cec7
1 parent aba9eab
commit 072cec7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/ha-websocket.js b/lib/ha-websocket.js
@@ -207,10 +207,12 @@ class HaWebsocket extends EventEmitter {
 
         if (msg) {
             const eventType = msg.event_type;
-            const entityId = msg.data ? msg.data.entity_id : null;
+            const entityId = msg.data && msg.data.entity_id;
 
             if (eventType === 'nodered') {
-                if (msg.data.type === 'unloaded') {
+                if (msg.data.type === 'loaded') {
+                    this.integrationVersion = msg.data.version;
+                } else if (msg.data.type === 'unloaded') {
                     this.integrationVersion = 0;
                 }
                 this.emit(`integration`, msg.data.type);