Refactor to generate JS AST

packages/babel-plugin-apply-mdx-type-props/index.js

@@ -16,6 +16,27 @@ class BabelPluginApplyMdxTypeProp {
           JSXOpeningElement(path) {
             const jsxName = path.node.name.name
 
+            let parentPath = path.parentPath.parentPath
+            let parentName
+
+            while (parentPath) {
+              if (parentPath.node.type === 'JSXElement') {
+                parentName = parentPath.node.openingElement.name.name
+                break
+              }
+
+              parentPath = parentPath.parentPath
+            }
+
+            if (typeof parentName === 'string' && parentName !== 'MDXLayout') {
+              path.node.attributes.push(
+                t.jSXAttribute(
+                  t.jSXIdentifier(`parentName`),
+                  t.stringLiteral(parentName)
+                )
+              )
+            }
+
             if (startsWithCapitalLetter(jsxName)) {
               names.push(jsxName)
 

packages/babel-plugin-apply-mdx-type-props/test/index.test.js

@@ -19,7 +19,7 @@ const transform = value => {
 describe('babel-plugin-add-mdx-type-prop', () => {
   test('should add `mdxType` to components', () => {
     expect(transform('var d = <div><Button /></div>').code).toEqual(
-      'var d = <div><Button mdxType="Button" /></div>;'
+      'var d = <div><Button parentName="div" mdxType="Button" /></div>;'
     )
   })
 

packages/mdx/index.js

@@ -2,6 +2,7 @@ const unified = require('unified')
 const remarkParse = require('remark-parse')
 const remarkMdx = require('remark-mdx')
 const squeeze = require('remark-squeeze-paragraphs')
+const minifyWhitespace = require('rehype-minify-whitespace')
 const mdxAstToMdxHast = require('./mdx-ast-to-mdx-hast')
 const mdxHastToJsx = require('./mdx-hast-to-jsx')
 
@@ -20,6 +21,7 @@ function createMdxAstCompiler(options = {}) {
 function createCompiler(options = {}) {
   return createMdxAstCompiler(options)
     .use(options.rehypePlugins)
+    .use(minifyWhitespace, {newlines: true})
     .use(mdxHastToJsx, options)
 }