Refactor code.js to simplify getCustomTokenStyle function

jpg619 · jpg619 · commit c69b0681853f · 2024-10-09T11:33:32.000-04:00
diff --git a/src/mdx/code.js b/src/mdx/code.js
@@ -91,15 +91,14 @@ const CodeBlock = ({children, code, className, style}) => (
   </Box>
 )
 
-function getCustomTokenStyle(tokenProps) {
-  const {className, style} = tokenProps
+function getCustomTokenStyle({className, style}) {
   const colorMap = {
     'token comment': '#747458',
     'token parameter variable': '#277d7b',
     'token function': '#cf3846',
   }
 
-  return className in colorMap ? {...style, color: colorMap[className]} : style
+  return colorMap[className] ? {...style, color: colorMap[className]} : style
 }
 
 function Code({className = '', prompt, children}) {

Original file line number	Diff line number	Diff line change
`@@ -91,15 +91,14 @@ const CodeBlock = ({children, code, className, style}) => (`
`91`	`91`	`</Box>`
`92`	`92`	`)`
`93`	`93`
`94`		`-function getCustomTokenStyle(tokenProps) {`
`95`		`- const {className, style} = tokenProps`
	`94`	`+function getCustomTokenStyle({className, style}) {`
`96`	`95`	`const colorMap = {`
`97`	`96`	`'token comment': '#747458',`
`98`	`97`	`'token parameter variable': '#277d7b',`
`99`	`98`	`'token function': '#cf3846',`
`100`	`99`	`}`
`101`	`100`
`102`		`- return className in colorMap ? {...style, color: colorMap[className]} : style`
	`101`	`+ return colorMap[className] ? {...style, color: colorMap[className]} : style`
`103`	`102`	`}`
`104`	`103`
`105`	`104`	`function Code({className = '', prompt, children}) {`