refactor(Mermaid): add container

diplodoc-platform · Jan 30, 2025 · dddf36f · dddf36f
1 parent 570e82d
commit dddf36f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/plugin/transform.ts b/src/plugin/transform.ts
@@ -98,7 +98,14 @@ export function transform(options: Partial<PluginOptions> = {}) {
             const token = tokens[idx];
             const code = encodeURIComponent(token.content.trimStart());
 
-            return `<div class="mermaid" data-content="${code}"></div>`;
+            const containerClassName = 'marmaid-container';
+
+            return [
+                `<div class="${containerClassName}">`,
+                `<div class="mermaid" data-content="${code}">`,
+                '</div>',
+                '</div>',
+            ].join('');
         };
     };