Make code.tsx simpler

Author: timolins

src/components/code.tsx

@@ -1,22 +1,20 @@
 import * as React from "react";
-import Prism from "prismjs";
+import { highlight, languages } from "prismjs";
 import "prismjs/components/prism-jsx";
 
 const Code: React.FC<{ code: string; language: string }> = ({
   code,
   language = "javascript"
 }) => {
+  const prismLanguage =
+    languages[language.toLowerCase()] || languages.javascript;
+
   return (
     <pre>
       <code
         className="notion-code"
         dangerouslySetInnerHTML={{
-          __html: Prism.highlight(
-            code,
-            Prism.languages[language.toLowerCase()] ||
-              Prism.languages.javascript,
-            language
-          )
+          __html: highlight(code, prismLanguage, language)
         }}
       />
     </pre>