Refactor ref as Component's instance function (#808)

text-mask · Jul 24, 2018 · f2260f3 · f2260f3
1 parent 55edd7a
commit f2260f3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/react/src/reactTextMask.js b/react/src/reactTextMask.js
@@ -8,10 +8,15 @@ export default class MaskedInput extends React.PureComponent {
   constructor(...args) {
     super(...args)
 
+    this.setRef = this.setRef.bind(this)
     this.onBlur = this.onBlur.bind(this)
     this.onChange = this.onChange.bind(this)
   }
 
+  setRef(inputElement) {
+    this.inputElement = inputElement
+  }
+
   initTextMask() {
     const {props, props: {value}} = this
 
@@ -66,9 +71,7 @@ export default class MaskedInput extends React.PureComponent {
     delete props.onChange
     delete props.showMask
 
-    const ref = (inputElement) => (this.inputElement = inputElement)
-
-    return render(ref, {
+    return render(this.setRef, {
       onBlur: this.onBlur,
       onChange: this.onChange,
       defaultValue: this.props.value,