// Reference: https://github.com/mermaid-js/mermaid

import { PluginSimple } from "markdown-it/lib";
import { hash } from "@vuepress/utils";

const MermaidPlugin: PluginSimple = function (md) {
  const fence = md.renderer.rules.fence;
  md.renderer.rules.fence = (...args) => {
    const [tokens, idx] = args;
    const { info } = tokens[idx];
    if (info.trim() === "mermaid") {
      const token = tokens[idx];
      const key = `mermaid_${hash(idx)}`;
      let { content } = token;
      content = content.replaceAll(";\n", ";");
      content = content.replaceAll("\n\n", ";");
      content = content.replaceAll("\n", ";");
      content = content.replaceAll('"', "'");
      return `<Mermaid identifier="${key}" graph="${content}"></Mermaid>`;
    }
    const rawCode = fence(...args);
    return `${rawCode}`;
  };
};

export { MermaidPlugin };