[fixed] Use bound ref functions

Widen · Jun 20, 2017 · cb6504c · cb6504c
1 parent 7da6ec8
commit cb6504c
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/components/ModalPortal.js b/src/components/ModalPortal.js
@@ -91,6 +91,14 @@ export default class ModalPortal extends Component {
     this.focusAfterRender = focus;
   }
 
+  setOverlayRef = (overlay) => {
+    this.overlay = overlay;
+  }
+
+  setContentRef = (content) => {
+    this.content = content;
+  }
+
   afterClose = () => {
     focusManager.returnFocus();
     focusManager.teardownScopedFocus();
@@ -209,12 +217,12 @@ export default class ModalPortal extends Component {
 
     return this.shouldBeClosed() ? <div /> : (
       <div
-        ref={overlay => { this.overlay = overlay; }}
+        ref={this.setOverlayRef}
         className={this.buildClassName('overlay', overlayClassName)}
         style={{ ...overlayStyles, ...this.props.style.overlay }}
         onClick={this.handleOverlayOnClick}>
         <div
-          ref={content => { this.content = content; }}
+          ref={this.setContentRef}
           style={{ ...contentStyles, ...this.props.style.content }}
           className={this.buildClassName('content', className)}
           tabIndex="-1"