mirror of
https://github.com/usememos/memos.git
synced 2025-12-17 22:28:52 +08:00
feat(web): add theme-aware Mermaid diagram rendering
Update MermaidBlock component to automatically adapt to light/dark theme changes, ensuring diagrams remain visually consistent with the app theme. Changes: - Add MutationObserver to watch for data-theme attribute changes - Map app themes to Mermaid themes (default-dark → dark, others → default) - Re-render diagrams automatically when theme switches - Maintain proper TypeScript typing for Mermaid theme values The component now responds instantly to theme changes, providing a seamless user experience when switching between light and dark modes or using "Sync with system" option.
This commit is contained in:
parent
8259757800
commit
6fb7520518
1 changed files with 56 additions and 8 deletions
|
|
@ -7,23 +7,60 @@ interface MermaidBlockProps {
|
||||||
className?: string;
|
className?: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Initialize mermaid with default configuration
|
/**
|
||||||
mermaid.initialize({
|
* Maps app theme to Mermaid theme
|
||||||
startOnLoad: false,
|
* @param appTheme - The app's theme value from data-theme attribute
|
||||||
theme: "default",
|
* @returns Mermaid theme name
|
||||||
securityLevel: "strict",
|
*/
|
||||||
fontFamily: "inherit",
|
const getMermaidTheme = (appTheme: string | null): "default" | "dark" => {
|
||||||
});
|
switch (appTheme) {
|
||||||
|
case "default-dark":
|
||||||
|
return "dark";
|
||||||
|
case "default":
|
||||||
|
case "paper":
|
||||||
|
case "whitewall":
|
||||||
|
default:
|
||||||
|
return "default";
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Gets the current theme from the document
|
||||||
|
*/
|
||||||
|
const getCurrentTheme = (): string => {
|
||||||
|
return document.documentElement.getAttribute("data-theme") || "default";
|
||||||
|
};
|
||||||
|
|
||||||
export const MermaidBlock = ({ children, className }: MermaidBlockProps) => {
|
export const MermaidBlock = ({ children, className }: MermaidBlockProps) => {
|
||||||
const containerRef = useRef<HTMLDivElement>(null);
|
const containerRef = useRef<HTMLDivElement>(null);
|
||||||
const [svg, setSvg] = useState<string>("");
|
const [svg, setSvg] = useState<string>("");
|
||||||
const [error, setError] = useState<string>("");
|
const [error, setError] = useState<string>("");
|
||||||
|
const [currentTheme, setCurrentTheme] = useState<string>(getCurrentTheme());
|
||||||
|
|
||||||
// Extract the code element and its content
|
// Extract the code element and its content
|
||||||
const codeElement = children as React.ReactElement;
|
const codeElement = children as React.ReactElement;
|
||||||
const codeContent = String(codeElement?.props?.children || "").replace(/\n$/, "");
|
const codeContent = String(codeElement?.props?.children || "").replace(/\n$/, "");
|
||||||
|
|
||||||
|
// Watch for theme changes
|
||||||
|
useEffect(() => {
|
||||||
|
const observer = new MutationObserver((mutations) => {
|
||||||
|
mutations.forEach((mutation) => {
|
||||||
|
if (mutation.type === "attributes" && mutation.attributeName === "data-theme") {
|
||||||
|
const newTheme = getCurrentTheme();
|
||||||
|
setCurrentTheme(newTheme);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
observer.observe(document.documentElement, {
|
||||||
|
attributes: true,
|
||||||
|
attributeFilter: ["data-theme"],
|
||||||
|
});
|
||||||
|
|
||||||
|
return () => observer.disconnect();
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
// Render diagram when content or theme changes
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const renderDiagram = async () => {
|
const renderDiagram = async () => {
|
||||||
if (!codeContent || !containerRef.current) {
|
if (!codeContent || !containerRef.current) {
|
||||||
|
|
@ -34,6 +71,17 @@ export const MermaidBlock = ({ children, className }: MermaidBlockProps) => {
|
||||||
// Generate a unique ID for this diagram
|
// Generate a unique ID for this diagram
|
||||||
const id = `mermaid-${Math.random().toString(36).substring(7)}`;
|
const id = `mermaid-${Math.random().toString(36).substring(7)}`;
|
||||||
|
|
||||||
|
// Get the appropriate Mermaid theme for current app theme
|
||||||
|
const mermaidTheme = getMermaidTheme(currentTheme);
|
||||||
|
|
||||||
|
// Initialize mermaid with current theme
|
||||||
|
mermaid.initialize({
|
||||||
|
startOnLoad: false,
|
||||||
|
theme: mermaidTheme,
|
||||||
|
securityLevel: "strict",
|
||||||
|
fontFamily: "inherit",
|
||||||
|
});
|
||||||
|
|
||||||
// Render the mermaid diagram
|
// Render the mermaid diagram
|
||||||
const { svg: renderedSvg } = await mermaid.render(id, codeContent);
|
const { svg: renderedSvg } = await mermaid.render(id, codeContent);
|
||||||
setSvg(renderedSvg);
|
setSvg(renderedSvg);
|
||||||
|
|
@ -45,7 +93,7 @@ export const MermaidBlock = ({ children, className }: MermaidBlockProps) => {
|
||||||
};
|
};
|
||||||
|
|
||||||
renderDiagram();
|
renderDiagram();
|
||||||
}, [codeContent]);
|
}, [codeContent, currentTheme]);
|
||||||
|
|
||||||
// If there's an error, fall back to showing the code
|
// If there's an error, fall back to showing the code
|
||||||
if (error) {
|
if (error) {
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue