memos/web/src/components/MemoContent/MermaidBlock.tsx

import { useColorScheme } from "@mui/joy";
import mermaid from "mermaid";
import { useEffect, useRef } from "react";

interface Props {
  content: string;
}

const MermaidBlock: React.FC<Props> = ({ content }: Props) => {
  const { mode } = useColorScheme();
  const mermaidDockBlock = useRef<null>(null);
  mermaid.initialize({ startOnLoad: false, theme: mode });

  useEffect(() => {
    if (!mermaidDockBlock.current) {
      return;
    }

    // Render mermaid when mounted.
    mermaid.run({
      nodes: [mermaidDockBlock.current],
    });
  });

  return (
    <pre ref={mermaidDockBlock} className="w-full p-2 whitespace-pre-wrap relative">
      {content}
    </pre>
  );
};

export default MermaidBlock;