From 30057a07224000cfc3571bedf7de441f2fbdc658 Mon Sep 17 00:00:00 2001
From: Michal Piechowiak <misiek.piechowiak@gmail.com>
Date: Thu, 6 May 2021 16:20:44 +0200
Subject: [PATCH] save MDXContent to different file

---
 packages/gatsby-plugin-mdx/constants.js     |  2 ++
 packages/gatsby-plugin-mdx/utils/gen-mdx.js | 30 +++++++++++++++++++--
 2 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/packages/gatsby-plugin-mdx/constants.js b/packages/gatsby-plugin-mdx/constants.js
index bb3f006a6aff8..9a1971f6502a6 100644
--- a/packages/gatsby-plugin-mdx/constants.js
+++ b/packages/gatsby-plugin-mdx/constants.js
@@ -1,7 +1,9 @@
 const MDX_WRAPPERS_LOCATION = `mdx-wrappers-dir`
 const MDX_SCOPES_LOCATION = `mdx-scopes-dir`
+const MDX_LOADER_PASSTHROUGH_LOCATION = `mdx-loader-passthrough-dir`
 
 module.exports = {
   MDX_WRAPPERS_LOCATION,
   MDX_SCOPES_LOCATION,
+  MDX_LOADER_PASSTHROUGH_LOCATION,
 }
diff --git a/packages/gatsby-plugin-mdx/utils/gen-mdx.js b/packages/gatsby-plugin-mdx/utils/gen-mdx.js
index a991745f032a1..3dcd3a3f24054 100644
--- a/packages/gatsby-plugin-mdx/utils/gen-mdx.js
+++ b/packages/gatsby-plugin-mdx/utils/gen-mdx.js
@@ -2,6 +2,8 @@ const babel = require(`@babel/core`)
 const grayMatter = require(`gray-matter`)
 const mdx = require(`@mdx-js/mdx`)
 const objRestSpread = require(`@babel/plugin-proposal-object-rest-spread`)
+const path = require(`path`)
+const fs = require(`fs-extra`)
 
 const debug = require(`debug`)(`gatsby-plugin-mdx:gen-mdx`)
 
@@ -10,6 +12,7 @@ const htmlAttrToJSXAttr = require(`./babel-plugin-html-attr-to-jsx-attr`)
 const removeExportKeywords = require(`./babel-plugin-remove-export-keywords`)
 const BabelPluginPluckImports = require(`./babel-plugin-pluck-imports`)
 const { parseImportBindings } = require(`./import-parser`)
+const { MDX_LOADER_PASSTHROUGH_LOCATION } = require(`../constants`)
 
 /*
  * function mutateNode({
@@ -89,7 +92,9 @@ async function genMDX(
   // pull classic style frontmatter off the raw MDX body
   debug(`processing classic frontmatter`)
   const { data, content: frontMatterCodeResult } = grayMatter(node.rawBody)
-  const content = `${frontMatterCodeResult}
+  const content = isLoader
+    ? frontMatterCodeResult
+    : `${frontMatterCodeResult}
 
 export const _frontmatter = ${JSON.stringify(data)}`
 
@@ -136,11 +141,13 @@ export const _frontmatter = ${JSON.stringify(data)}`
     ),
   })
 
-  results.rawMDXOutput = `/* @jsx mdx */
+  const rawMDXOutput = `/* @jsx mdx */
 import { mdx } from '@mdx-js/react';
 ${code}`
 
   if (!isLoader) {
+    results.rawMDXOutput = rawMDXOutput
+
     debug(`compiling scope`)
     const instance = new BabelPluginPluckImports()
     const result = babel.transform(code, {
@@ -183,6 +190,25 @@ ${code}`
         /export\s*{\s*MDXContent\s+as\s+default\s*};?/,
         `return MDXContent;`
       )
+  } else {
+    // code path for webpack loader
+    // actual react component is saved to different file so that _frontmatter export doesn't
+    // disable react-refresh (multiple exports are not handled)
+    const filePath = path.join(
+      cache.directory,
+      MDX_LOADER_PASSTHROUGH_LOCATION,
+      `${helpers.createContentDigest(node.fileAbsolutePath)}.js`
+    )
+
+    await fs.outputFile(filePath, rawMDXOutput)
+
+    results.rawMDXOutput = `
+      import MDXContent from "${filePath}";
+      
+      export default MDXContent;
+
+      export const _frontmatter = ${JSON.stringify(data)};
+    `
   }
   /* results.html = renderToStaticMarkup(
    *   React.createElement(MDXRenderer, null, results.body)