fix(docz-plugin-babel6): props parsing when typescript is enable

doczjs · Jul 11, 2018 · 0564bdf · 0564bdf
1 parent 21bd8c5
commit 0564bdf
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/packages/docz-plugin-babel6/package.json b/packages/docz-plugin-babel6/package.json
@@ -26,6 +26,7 @@
     "babel-preset-react-app": "^3.1.2",
     "docz-core": "^0.5.6",
     "happypack": "^5.0.0",
+    "react-docgen-typescript-loader": "^2.1.1",
     "ts-loader": "^4.4.2"
   }
 }
diff --git a/packages/docz-plugin-babel6/src/index.ts b/packages/docz-plugin-babel6/src/index.ts
@@ -27,7 +27,7 @@ export const babel = () =>
 
       return babelrc
     },
-    modifyBundlerConfig: config => {
+    modifyBundlerConfig: (config, dev, args) => {
       const happypackJsx = config.plugins.findIndex(findHappypackJsx)
       const happypackMdx = config.plugins.findIndex(findHappypackMdx)
       const tsxRule = config.module.rules.find(findTsxRule)
@@ -36,6 +36,8 @@ export const babel = () =>
       modifyHappypackLoader(config.plugins[happypackJsx])
       modifyHappypackLoader(config.plugins[happypackMdx])
 
+      console.log(args)
+
       if (tsxRule) {
         config.module.rules[tsxIdx] = {
           ...tsxRule,
@@ -52,6 +54,13 @@ export const babel = () =>
                   happyPackMode: true,
                 },
               },
+              ...(args.propsParser
+                ? [
+                    {
+                      loader: require.resolve('react-docgen-typescript-loader'),
+                    },
+                  ]
+                : []),
             ],
           })
         )