injectInComponent only first time

Author: mbasso

src/injector.js

@@ -25,26 +25,30 @@ function injectInElement(component, componentName) {
 }
 
 function injectInComponent(Component, componentName) {
-  const componentPrototype = Component.prototype;
-  const componentConstructor = Component;
+  let newComponent = Component;
+  if (!Component.hasReactCSSOM) {
+    const componentPrototype = Component.prototype;
+    const componentConstructor = Component;
 
-  //eslint-disable-next-line
-  const newComponent = function ReactComponent(...params) {
-    componentConstructor.apply(this, [...params]);
-    const originalRender = this.render;
-    this.render = () => (
-      injectInElement(
-        originalRender.apply(this),
-        componentName || this.constructor.displayName || this.constructor.name
-      )
-    );
-  };
+    //eslint-disable-next-line
+    newComponent = function ReactComponent(...params) {
+      componentConstructor.apply(this, [...params]);
+      const originalRender = this.render;
+      this.render = () => (
+        injectInElement(
+          originalRender.apply(this),
+          componentName || this.constructor.displayName || this.constructor.name
+        )
+      );
+    };
 
-  //eslint-disable-next-line
-  newComponent.prototype = componentPrototype;
-  Object.keys(Component).forEach((x) => {
-    newComponent[x] = Component[x];
-  });
+    newComponent.prototype = componentPrototype;
+    Object.keys(Component).forEach((x) => {
+      newComponent[x] = Component[x];
+    });
+
+    newComponent.hasReactCSSOM = true;
+  }
 
   return newComponent;
 }

test/injector.js

@@ -258,5 +258,43 @@ describe('injector', () => {
         className: '⚛Foo ',
       });
     });
+
+    it('should inject only the first time', () => {
+      //eslint-disable-next-line
+      class Foo extends React.Component {
+        render() {
+          return (
+            <div />
+          );
+        }
+      }
+
+      let oldClass = Foo;
+      let Bar = inject(Foo)();
+      expect(oldClass).toNotEqual(Bar);
+      expect(Foo.hasReactCSSOM).toBeFalsy();
+      expect(Bar.hasReactCSSOM).toBeTruthy();
+      oldClass = Bar;
+      Bar = inject(Foo)();
+      expect(oldClass).toEqual(Bar);
+
+      oldClass = React.Component;
+      React.Component = inject(React.Component)();
+      expect(oldClass).toNotEqual(React.Component);
+      oldClass = React.Component;
+      React.Component = inject(React.Component)();
+      expect(oldClass).toEqual(React.Component);
+
+      //eslint-disable-next-line
+      class Baz extends React.Component {
+        render() {
+          return (
+            <div />
+          );
+        }
+      }
+
+      expect(Baz.hasReactCSSOM).toBeTruthy();
+    });
   });
 });