From 2e2031ed4c3e7ba39cdda4210551881a389da9d6 Mon Sep 17 00:00:00 2001
From: kamiazya <yuki@kamiazya.tech>
Date: Thu, 7 Apr 2022 01:18:42 +0900
Subject: [PATCH] feat: add mermaid version option

---
 README.md                              | 18 +++++++++++++++
 src/index.ts                           |  2 +-
 src/plugin.ts                          | 31 +++++++++++++++++++-------
 test/__snapshots__/plugin.test.ts.snap |  2 +-
 test/plugin.test.ts                    |  7 ++++--
 5 files changed, 48 insertions(+), 12 deletions(-)

diff --git a/README.md b/README.md
index a141f0e..69ead1a 100644
--- a/README.md
+++ b/README.md
@@ -60,6 +60,24 @@ export class Mermaid { }
 
 [![Example](./media/example.png)](https://kamiazya.github.io/typedoc-plugin-samples/classes/hoge.html)
 
+### Arguments
+
+The following arguments can be used in addition to the default TypeDoc arguments.
+
+```bash
+$ typedoc --help
+...
+Options:
+ --mermaidVersion            [Mermaid Plugin] The version of mermaid.js to use.
+...
+```
+
+- `--mermaidVersion`
+
+    Specify mermaid.js version to use.
+
+    Default: `latest`, Example: `8.14.0`
+
 ## Contributors
 
 Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/docs/en/emoji-key)):
diff --git a/src/index.ts b/src/index.ts
index fdd35ad..2899870 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -2,5 +2,5 @@ import { Application } from 'typedoc';
 import { MermaidPlugin } from './plugin';
 
 export function load(app: Application): void {
-  new MermaidPlugin().addToApplication(app);
+  new MermaidPlugin(app).initialize();
 }
diff --git a/src/plugin.ts b/src/plugin.ts
index 4f49a48..ae76815 100644
--- a/src/plugin.ts
+++ b/src/plugin.ts
@@ -1,5 +1,5 @@
 import * as html from 'html-escaper';
-import { Converter, Context, PageEvent, Application, ReflectionKind, MarkdownEvent } from 'typedoc';
+import { Converter, Context, PageEvent, Application, ReflectionKind, MarkdownEvent, ParameterType } from 'typedoc';
 
 const style = String.raw`
 <style>
@@ -49,8 +49,9 @@ body.dark, :root[data-theme="dark"] {
  * 2. Initialize mermaid.
  * 3. Add special attribute after SVG has been inserted.
  */
-const script = String.raw`
-<script src="https://unpkg.com/mermaid/dist/mermaid.min.js"></script>
+function createScript(version: string): string {
+  return String.raw`
+<script src="https://unpkg.com/mermaid@${version}/dist/mermaid.min.js"></script>
 <script>
 (function() {
   if (typeof mermaid === "undefined") {
@@ -77,24 +78,34 @@ const script = String.raw`
 })();
 </script>
 `;
+}
 
 const mermaidBlockStart = '<div class="mermaid-block">';
 const mermaidBlockEnd = '</div>';
 
 export class MermaidPlugin {
-  public addToApplication(app: Application): void {
-    app.converter.on(Converter.EVENT_RESOLVE_BEGIN, (context: Context) => {
+  constructor(private app: Application) {}
+
+  public initialize(): void {
+    this.app.options.addDeclaration({
+      help: '[Mermaid Plugin] The version of mermaid.js to use.',
+      name: 'mermaidVersion',
+      type: ParameterType.String,
+      defaultValue: 'latest',
+    });
+
+    this.app.converter.on(Converter.EVENT_RESOLVE_BEGIN, (context: Context) => {
       this.onConverterResolveBegin(context);
     });
 
-    app.renderer.on({
+    this.app.renderer.on({
       [PageEvent.END]: (event: PageEvent) => {
         this.onEndPage(event);
       },
     });
 
     // high priority markdown parser to catch blocks before the built-in parser
-    app.renderer.on(
+    this.app.renderer.on(
       MarkdownEvent.PARSE,
       (event: MarkdownEvent) => {
         this.onParseMarkdown(event);
@@ -174,6 +185,10 @@ export class MermaidPlugin {
 
     // find the closing </body> tag and insert our mermaid scripts
     const bodyEndIndex = html.lastIndexOf('</body>');
-    return html.slice(0, bodyEndIndex) + script + html.slice(bodyEndIndex);
+    return (
+      html.slice(0, bodyEndIndex) +
+      createScript(this.app.options.getValue('mermaidVersion') as string) +
+      html.slice(bodyEndIndex)
+    );
   }
 }
diff --git a/test/__snapshots__/plugin.test.ts.snap b/test/__snapshots__/plugin.test.ts.snap
index a9db9fc..06dce14 100644
--- a/test/__snapshots__/plugin.test.ts.snap
+++ b/test/__snapshots__/plugin.test.ts.snap
@@ -66,7 +66,7 @@ body.dark, :root[data-theme=\\"dark\\"] {
 }
 </style>
 </head><body><div class=\\"mermaid-block\\"></div>
-<script src=\\"https://unpkg.com/mermaid/dist/mermaid.min.js\\"></script>
+<script src=\\"https://unpkg.com/mermaid@latest/dist/mermaid.min.js\\"></script>
 <script>
 (function() {
   if (typeof mermaid === \\"undefined\\") {
diff --git a/test/plugin.test.ts b/test/plugin.test.ts
index 54fb1b9..4872d22 100644
--- a/test/plugin.test.ts
+++ b/test/plugin.test.ts
@@ -1,7 +1,10 @@
+import { Application } from 'typedoc';
 import { MermaidPlugin } from '../src/plugin';
 
 describe('MermaidPlugin', () => {
-  const plugin = new MermaidPlugin();
+  const app = new Application();
+  const plugin = new MermaidPlugin(app);
+  plugin.initialize();
 
   it('convert CommentTag', () => {
     const input = 'title\ngraph';
@@ -16,7 +19,7 @@ describe('MermaidPlugin', () => {
     const result = plugin.insertMermaidScript(input);
     expect(result).toMatch('</body>');
     expect(result).toMatch('mermaid.initialize({');
-    expect(result).toMatch(/src="https:\/\/unpkg.com\/mermaid\/dist\/mermaid.min.js"/);
+    expect(result).toMatch(/src="https:\/\/unpkg.com\/mermaid\@latest\/dist\/mermaid.min.js"/);
     expect(result).toMatchSnapshot();
   });