fix(components): 🐛 fix issue with setting up code block language

filipowm · Jul 3, 2020 · 98750c7 · 98750c7
1 parent 33f407b
commit 98750c7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/components/MdxComponents/codeBlock.js b/src/components/MdxComponents/codeBlock.js
@@ -25,12 +25,13 @@ const LoadableComponent = Loadable({
 });
 
 /* eslint-disable react/jsx-key */
-const CodeBlock = ({ children: exampleCode, ...props }) => {
+const CodeBlock = ({ children: code, ...props }) => {
  if (props['react-live']) {
- return <LoadableComponent code={exampleCode} />;
+ return <LoadableComponent code={code} />;
  } else {
+ const lang = props.className ? props.className.split('-')[1] : null;
  return (
- <Highlight {...defaultProps} code={exampleCode} language="javascript" theme={theme}>
+ <Highlight {...defaultProps} code={code} language={lang ? lang : "javascript"} theme={theme}>
  {({ className, style, tokens, getLineProps, getTokenProps }) => (
  <pre className={className + ' pre'} style={style} p={3}>
  {cleanTokens(tokens).map((line, i) => {