From cde058c72ac5944a3d6909cf82f10942d0627ee3 Mon Sep 17 00:00:00 2001
From: johnnyjoy <yourselfhosted@gmail.com>
Date: Mon, 13 Jan 2025 19:46:06 +0800
Subject: [PATCH] chore: tweak theme code

---
 web/src/components/MemoContent/MermaidBlock.tsx | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/web/src/components/MemoContent/MermaidBlock.tsx b/web/src/components/MemoContent/MermaidBlock.tsx
index 064d4608..ac5f377c 100644
--- a/web/src/components/MemoContent/MermaidBlock.tsx
+++ b/web/src/components/MemoContent/MermaidBlock.tsx
@@ -6,16 +6,14 @@ interface Props {
 }
 
 const MermaidBlock: React.FC<Props> = ({ content }: Props) => {
+  const { mode: colorMode } = useColorScheme();
   const mermaidDockBlock = useRef<null>(null);
-  const { mode } = useColorScheme();
-
-  const mermaidTheme = mode == "dark" ? "dark" : "default";
 
   useEffect(() => {
     // Dynamically import mermaid to ensure compatibility with Vite
     const initializeMermaid = async () => {
       const mermaid = (await import("mermaid")).default;
-      mermaid.initialize({ startOnLoad: false, theme: mermaidTheme });
+      mermaid.initialize({ startOnLoad: false, theme: colorMode == "dark" ? "dark" : "default" });
       if (mermaidDockBlock.current) {
         mermaid.run({
           nodes: [mermaidDockBlock.current],