change diagramTheme and themes to return string

mermaid-js · Jun 26, 2023 · f2c4027 · f2c4027
1 parent 07f6358
commit f2c4027
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/packages/mermaid/src/styles.ts b/packages/mermaid/src/styles.ts
@@ -1,7 +1,7 @@
 import type { FlowChartStyleOptions } from './diagrams/flowchart/styles.js';
 import { log } from './logger.js';
 
-const themes: Record<string, any> = {};
+const themes: Record<string, (options?: any) => string> = {};
 
 const getStyles = (
   type: string,
@@ -73,9 +73,12 @@ const getStyles = (
 `;
 };
 
-export const addStylesForDiagram = (type: string, diagramTheme?: unknown): void => {
+export const addStylesForDiagram = (
+  type: string,
+  diagramTheme?: (options?: any) => string
+): void => {
   if (diagramTheme !== undefined) {
-    themes[type] = diagramTheme;
+    themes[type] = (options) => diagramTheme(options);
   }
 };