From a20473d71812183758ed0081e2faad15c53743f5 Mon Sep 17 00:00:00 2001
From: Samuel Elliott <samuel@fancy.org.uk>
Date: Wed, 1 Aug 2018 16:08:47 +0100
Subject: [PATCH] Add a function to force update all instances of a component

---
 client/src/modules/reactcomponents.js | 40 +++++++++++++++++++--------
 client/src/ui/reflection.js           |  2 +-
 2 files changed, 30 insertions(+), 12 deletions(-)

diff --git a/client/src/modules/reactcomponents.js b/client/src/modules/reactcomponents.js
index 022e708d..55f80141 100644
--- a/client/src/modules/reactcomponents.js
+++ b/client/src/modules/reactcomponents.js
@@ -166,10 +166,18 @@ class Helpers {
 export { Helpers as ReactHelpers };
 
 class ReactComponent {
-    constructor(id, component, retVal) {
+    constructor(id, component, retVal, important) {
         this.id = id;
         this.component = component;
         this.retVal = retVal;
+        this.important = important;
+    }
+
+    forceUpdateAll() {
+        if (!this.important || !this.important.selector) return;
+        for (let e of document.querySelectorAll(this.important.selector)) {
+            Reflection(e).forceUpdate(this);
+        }
     }
 }
 
@@ -181,22 +189,28 @@ export class ReactComponents {
 
     static get ReactComponent() { return ReactComponent }
 
-    static push(component, retVal) {
+    static push(component, retVal, important) {
         if (!(component instanceof Function)) return null;
         const { displayName } = component;
         if (!displayName) {
             return this.processUnknown(component, retVal);
         }
+
         const have = this.components.find(comp => comp.id === displayName);
-        if (have) return component;
-        const c = new ReactComponent(displayName, component, retVal);
-        this.components.push(c);
-        const listener = this.listeners.find(listener => listener.id === displayName);
-        if (!listener) return c;
-        for (const l of listener.listeners) {
-            l(c);
+        if (have) {
+            if (!have.important) have.important = important;
+            return component;
         }
-        this.listeners.splice(this.listeners.findIndex(listener => listener.id === displayName), 1);
+
+        const c = new ReactComponent(displayName, component, retVal, important);
+        this.components.push(c);
+
+        const listener = this.listeners.find(listener => listener.id === displayName);
+        if (listener) {
+            for (const l of listener.listeners) l(c);
+            Utils.removeFromArray(this.listeners, listener);
+        }
+
         return c;
     }
 
@@ -210,6 +224,7 @@ export class ReactComponents {
     static async getComponent(name, important, filter) {
         const have = this.components.find(c => c.id === name);
         if (have) return have;
+
         if (important) {
             const importantInterval = setInterval(() => {
                 if (this.components.find(c => c.id === name)) {
@@ -231,14 +246,17 @@ export class ReactComponents {
 
                 if (!component.displayName) component.displayName = name;
                 Logger.info('ReactComponents', [`Found important component ${name} with reflection`, reflect]);
-                this.push(component);
+                important.filter = filter;
+                this.push(component, undefined, important);
             }, 50);
         }
+
         let listener = this.listeners.find(l => l.id === name);
         if (!listener) this.listeners.push(listener = {
             id: name,
             listeners: []
         });
+
         return new Promise(resolve => {
             listener.listeners.push(resolve);
         });
diff --git a/client/src/ui/reflection.js b/client/src/ui/reflection.js
index 9e96ee68..18e33c75 100644
--- a/client/src/ui/reflection.js
+++ b/client/src/ui/reflection.js
@@ -107,7 +107,7 @@ class Reflection {
         return stateNodes;
     }
 
-    static findComponentStateNode(node, component) {
+    static getComponentStateNode(node, component) {
         if (component instanceof ReactComponents.ReactComponent) component = component.component;
 
         for (let stateNode of this.getStateNodes(node)) {