Fix the displayName of HOC components

i18next · Mar 22, 2019 · d9fc0fb · d9fc0fb
1 parent 61c4645
commit d9fc0fb
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -25,7 +25,8 @@
   },
   "dependencies": {
     "@babel/runtime": "^7.3.1",
-    "html-parse-stringify2": "2.0.1"
+    "html-parse-stringify2": "2.0.1",
+    "react-display-name": "^0.2.4"
   },
   "devDependencies": {
     "@babel/cli": "^7.2.3",

diff --git a/src/withSSR.js b/src/withSSR.js
@@ -13,6 +13,7 @@ export function withSSR() {
     }
 
     I18nextWithSSR.getInitialProps = composeInitialProps(WrappedComponent);
+    I18nextWithSSR.displayName = `withI18nextSSR(${getDisplayName(WrappedComponent)})`;
 
     return I18nextWithSSR;
   };

diff --git a/src/withTranslation.js b/src/withTranslation.js
@@ -1,4 +1,5 @@
 import React from 'react';
+import getDisplayName from 'react-display-name';
 import { useTranslation } from './useTranslation';
 
 export function withTranslation(ns) {
@@ -14,6 +15,10 @@ export function withTranslation(ns) {
       });
     }
 
+    I18nextWithTranslation.displayName = `withI18nextTranslation(${getDisplayName(
+      WrappedComponent,
+    )})`;
+
     return I18nextWithTranslation;
   };
 }