From f95192beb0b5b49c4d321d2d609844c6cbdaae51 Mon Sep 17 00:00:00 2001
From: Paul Wackerow <54227730+wackerow@users.noreply.github.com>
Date: Sun, 27 Nov 2022 13:15:51 -0500
Subject: [PATCH] fix Code interface

---
 src/components/MDXComponents.tsx | 16 ++++++-------
 src/components/UI/docs/Code.tsx  | 39 ++++++++++++++------------------
 2 files changed, 24 insertions(+), 31 deletions(-)

diff --git a/src/components/MDXComponents.tsx b/src/components/MDXComponents.tsx
index 1694ad0889..76f2bd2c4c 100644
--- a/src/components/MDXComponents.tsx
+++ b/src/components/MDXComponents.tsx
@@ -1,7 +1,7 @@
 import { Flex, Heading, Link, Stack, Table, Text } from '@chakra-ui/react';
 import NextLink from 'next/link';
 
-import { Code } from './UI/docs'
+import { Code } from './UI/docs';
 import { textStyles } from '../theme/foundations';
 
 const { header1, header2, header3, header4 } = textStyles;
@@ -75,15 +75,13 @@ const MDXComponents = {
     </Flex>
   ),
   // pre
-  pre: ({ children }: any) => {
-    return (
-      <Stack mb={5}>
-        <pre>{children}</pre>
-      </Stack>
-    );
-  },
+  pre: ({ children }: any) => (
+    <Stack mb={5}>
+      <pre>{children}</pre>
+    </Stack>
+  ),
   // code
-  code: Code
+  code: ({ children, ...props }: any) => <Code {...props}>{children}</Code>
 };
 
 export default MDXComponents;
diff --git a/src/components/UI/docs/Code.tsx b/src/components/UI/docs/Code.tsx
index 1ccb3a671e..67768ab44a 100644
--- a/src/components/UI/docs/Code.tsx
+++ b/src/components/UI/docs/Code.tsx
@@ -32,17 +32,17 @@ SyntaxHighlighter.registerLanguage('sh', sh);
 SyntaxHighlighter.registerLanguage('solidity', solidity);
 SyntaxHighlighter.registerLanguage('swift', swift);
 
-
 interface Props {
-  className: string
-  children: React.ReactNode
-  [key: string]: any
+  className: string;
+  children: string[];
+  inline?: boolean;
 }
-export const Code: React.FC<Props> = ({ className, children, ...code }: any) => {
+export const Code: React.FC<Props> = ({ className, children, inline }) => {
   const { colorMode } = useColorMode();
   const isDark = colorMode === 'dark';
-  const isTerminal = className?.includes('terminal') ;
-  if (code.inline)
+  const isTerminal = className?.includes('terminal');
+  const [content] = children;
+  if (inline)
     return (
       <Text
         as='span'
@@ -52,22 +52,17 @@ export const Code: React.FC<Props> = ({ className, children, ...code }: any) =>
         borderRadius='0.25em'
         textStyle='inline-code-snippet'
       >
-        {children[0]}
+        {content}
       </Text>
     );
   if (isTerminal)
-      return (
-        <Stack>
-          <ChakraCode
-            overflow='auto'
-            p={6}
-            background='terminal-bg'
-            color='terminal-text'
-          >
-            {children[0]}
-          </ChakraCode>
-        </Stack>
-      );
+    return (
+      <Stack>
+        <ChakraCode overflow='auto' p={6} background='terminal-bg' color='terminal-text'>
+          {content}
+        </ChakraCode>
+      </Stack>
+    );
   if (className?.startsWith(CLASSNAME_PREFIX))
     return (
       <SyntaxHighlighter
@@ -75,8 +70,8 @@ export const Code: React.FC<Props> = ({ className, children, ...code }: any) =>
         style={isDark ? nightOwl : prism}
         customStyle={{ borderRadius: '0.5rem', padding: '1rem' }}
       >
-        {children}
+        {content}
       </SyntaxHighlighter>
     );
-  return <Stack>{children[0]}</Stack>;
+  return <Stack>{content}</Stack>;
 };