From 1e370817560c905a0a3520e245c317f308b6a5e7 Mon Sep 17 00:00:00 2001
From: Armand Philippot <git@armandphilippot.com>
Date: Tue, 1 Feb 2022 15:15:55 +0100
Subject: chore: add a new settings to handle prism theme from toolbar

---
 src/components/MDX/CodeBlock/CodeBlock.tsx | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/components/MDX/CodeBlock')

diff --git a/src/components/MDX/CodeBlock/CodeBlock.tsx b/src/components/MDX/CodeBlock/CodeBlock.tsx
index a822744..45a6176 100644
--- a/src/components/MDX/CodeBlock/CodeBlock.tsx
+++ b/src/components/MDX/CodeBlock/CodeBlock.tsx
@@ -5,6 +5,7 @@ import Prism from 'prismjs';
 import { ReactChildren, useEffect } from 'react';
 import { useIntl } from 'react-intl';
 import '@utils/plugins/prism-color-scheme';
+import { usePrismTheme } from '@utils/providers/prism';
 
 const CodeBlock = ({
   className,
@@ -29,6 +30,15 @@ const CodeBlock = ({
     translateCopyButton(locale, intl);
   }, [intl, locale]);
 
+  const { setCodeBlocks } = usePrismTheme();
+
+  useEffect(() => {
+    const allPre: NodeListOf<HTMLPreElement> = document.querySelectorAll(
+      'pre[data-prismjs-color-scheme'
+    );
+    setCodeBlocks(allPre);
+  }, [setCodeBlocks, router.asPath]);
+
   return (
     <div>
       <pre className={classNames.join(' ')}>
-- 
cgit v1.2.3