Fix translate

nayaabkhan · Sep 30, 2019 · 8a92584 · 8a92584
1 parent c5c6fac
commit 8a92584
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/example/greeter.js b/example/greeter.js
@@ -12,7 +12,7 @@ const Greeter = ({ name, nrCars, t }) => (
 Greeter.propTypes = {
   name: PropTypes.string.isRequired,
   nrCars: PropTypes.number.isRequired,
-  t: PropTypes.func
+  t: PropTypes.func,
 }
 
 export default translate()(Greeter)
diff --git a/src/translate.js b/src/translate.js
@@ -2,12 +2,15 @@ import React from 'react'
 import hoistNonReactStatics from 'hoist-non-react-statics'
 import I18nContext from './i18n-context'
 
-export default function translate()(WrappedComponent) {
-  const _translate = props => (
-    <I18nContext.Consumer>
-      {t => <WrappedComponent {...props} t={t} />}
-    </I18nContext.Consumer>
-  )
+// higher order decorator for components that need `t`
+export default function translate() {
+  return WrappedComponent => {
+    const _translate = props => (
+      <I18nContext.Consumer>
+        {t => <WrappedComponent {...props} t={t} />}
+      </I18nContext.Consumer>
+    )
 
-  return hoistNonReactStatics(_translate, WrappedComponent)
+    return hoistNonReactStatics(_translate, WrappedComponent)
+  }
 }