fix(gatsby-theme-docz): fix theme components customization bug

rakannimer · rakannimer · commit a9384b03cbd3 · 2020-04-05T11:28:46.000+03:00
diff --git a/core/gatsby-theme-docz/src/base/Layout.js b/core/gatsby-theme-docz/src/base/Layout.js
@@ -2,6 +2,7 @@ import React, { Fragment } from 'react'
 import PropTypes from 'prop-types'
 import { useComponents } from 'docz'
 import { propEq, get } from 'lodash/fp'
+import { MDXProvider } from '@mdx-js/react'
 
 import { useDbQuery } from '../hooks/useDbQuery'
 import Wrapper from '../wrapper'
@@ -17,9 +18,11 @@ const Route = ({ children, entry, isTransclusion, ...defaultProps }) => {
   return isTransclusion ? (
     children
   ) : (
-    <Wrapper>
-      <Layout {...props}>{children}</Layout>
-    </Wrapper>
+    <MDXProvider components={components}>
+      <Wrapper>
+        <Layout {...props}>{children}</Layout>
+      </Wrapper>
+    </MDXProvider>
   )
 }
 
diff --git a/core/gatsby-theme-docz/src/index.js b/core/gatsby-theme-docz/src/index.js
@@ -9,7 +9,7 @@ import components from '~components'
 const Theme = ({ children }) => {
   const config = useConfig()
   return (
-    <ThemeProvider theme={config.themeConfig} components={components}>
+    <ThemeProvider theme={config.themeConfig}>
       <ComponentsProvider components={components}>
         <Styled.root>{children}</Styled.root>
       </ComponentsProvider>
