feat: 添加 NitWikit 插件转换器以增强 Markdown 处理能力

8aka-Team · Dec 25, 2024 · 461d974 · 461d974
1 parent 56c9c20
commit 461d974
Show file tree

Hide file tree

Showing 7 changed files with 120 additions and 58 deletions.
diff --git a/.vitepress/config.mts b/.vitepress/config.mts
@@ -2,6 +2,7 @@ import { getSidebar } from "./utils/vitepress";
 import themeConfig from "./theme.config";
 import { resolve } from "path";
 import { withMermaid } from "vitepress-plugin-mermaid";
+import { NitWikitAssertsTransformer, NitWikitBilibiliTransformer, NitWikitContentTransformer, NitWikitUrlTransformer } from "../build/plugins";
 
 const markdownRegExp = {
   image: /!\[(?<name>.*)\]\((?<url>(.*))\)/g,
@@ -25,64 +26,10 @@ export default withMermaid({
     publicDir: "../public",
     assetsInclude: ["**/*.JPG", "**/*.PNG"],
     plugins: [
-      {
-        name: "nitwikit-transformer",
-        enforce: "pre",
-        transform(code, id) {
-          if (id.endsWith(".md")) {
-            return code
-              .replace(/import ([\s\S]+) from ['"](@theme\/[\s\S]+)['"];/g, "")
-              .replace(/import ([\s\S]+) from &#39;(@theme\/[\s\S]+)&#39;;/g, "")
-              .replace(/values={\[([\s\S]*)\]}/g, "");
-          }
-        },
-      },
-      {
-        name: "nitwikit-content-transformer",
-        enforce: "pre",
-        transform(code, id) {
-          if (id.endsWith(".md")) {
-            const result = code.match(markdownRegExp.title);
-            if (result !== null) {
-              console.info(id, result.groups?.title);
-              return code.replace(`${result.groups?.heading}${result.groups?.title}`, `# ${result.groups?.title}`);
-            }
-          }
-        },
-      },
-      {
-        name: "nitwikit-url-transformer",
-        enforce: "pre",
-        transform(code, id) {
-          if (id.endsWith(".md")) {
-            return code.replace(markdownRegExp.nitwikitUrl, (match, _name, url) => {
-              return match.replace(url, url.replace("https://nitwikit.yizhan.wiki/", "/"));
-            });
-          }
-        },
-      },
-      {
-        name: "nitwikit-assets-transformer",
-        enforce: "pre",
-        transform(code, id) {
-          if (id.endsWith(".md")) {
-            if (code.match(markdownRegExp.image)) {
-              return code + `\n<nw-image-viewer />`;
-            }
-          }
-        },
-      },
-      {
-        name: "nitwikit-bilitv-transformer",
-        enforce: "pre",
-        transform(code, id) {
-          if (id.endsWith(".md")) {
-            return code.replace(markdownRegExp.bilitv, (_match, name, url, bvId) => {
-              return `<bili-player title="${name}" src="${url}" bv-id="${bvId}" />`;
-            });
-          }
-        },
-      },
+      new NitWikitContentTransformer().build(),
+      new NitWikitAssertsTransformer().build(),
+      new NitWikitUrlTransformer().build(),
+      new NitWikitBilibiliTransformer().build(),
     ],
     css: {
       preprocessorOptions: {

diff --git a/build/plugins/base.ts b/build/plugins/base.ts
@@ -0,0 +1,23 @@
+import { Plugin } from "vite";
+
+export abstract class BasePlugin {
+  #name: Plugin["name"];
+  #enforce: Plugin["enforce"];
+
+  constructor(options: { name: string; enforce: "pre" | "post" }) {
+    this.#name = options.name;
+    this.#enforce = options.enforce;
+  }
+
+  transform(code: string, id: string): string | Promise<string> | void {
+    return code;
+  }
+
+  build(): Plugin {
+    return {
+      name: this.#name,
+      enforce: this.#enforce,
+      transform: this.transform,
+    };
+  }
+}
diff --git a/build/plugins/index.ts b/build/plugins/index.ts
@@ -0,0 +1,4 @@
+export * from "./transformer/assets";
+export * from "./transformer/url";
+export * from "./transformer/content";
+export * from "./transformer/bilibili";
diff --git a/build/plugins/transformer/assets.ts b/build/plugins/transformer/assets.ts
@@ -0,0 +1,20 @@
+import { BasePlugin } from "../base";
+
+export class NitWikitAssertsTransformer extends BasePlugin {
+  static #regExp: RegExp = /!\[([^\]]*)\]\(([^)]*)\)/g;
+
+  constructor() {
+    super({
+      name: "nitwikit-assets-transformer",
+      enforce: "pre",
+    });
+  }
+
+  override transform(code: string, id: string): string | Promise<string> | void {
+    if (id.endsWith(".md")) {
+      if (code.match(NitWikitAssertsTransformer.#regExp)) {
+        return code + `\n<nw-image-viewer />`;
+      }
+    }
+  }
+}
diff --git a/build/plugins/transformer/bilibili.ts b/build/plugins/transformer/bilibili.ts
@@ -0,0 +1,20 @@
+import { BasePlugin } from "../base";
+
+export class NitWikitBilibiliTransformer extends BasePlugin {
+  static #regExp: RegExp = /\[(?<name>.*)\]\((?<url>https:\/\/www.bilibili.com\/video\/(?<id>(.*))\/(.*))\)/g;
+
+  constructor() {
+    super({
+      name: "nitwikit-bilibili-transformer",
+      enforce: "pre",
+    });
+  }
+
+  override transform(code: string, id: string): string | Promise<string> | void {
+    if (id.endsWith(".md")) {
+      return code.replace(NitWikitBilibiliTransformer.#regExp, (_match, name, url, bvId) => {
+        return `<bili-player title="${name}" src="${url}" bv-id="${bvId}" />`;
+      });
+    }
+  }
+}
diff --git a/build/plugins/transformer/content.ts b/build/plugins/transformer/content.ts
@@ -0,0 +1,27 @@
+import { BasePlugin } from "../base";
+
+export class NitWikitContentTransformer extends BasePlugin {
+  static #regExp: RegExp = /^(---([\s\S]*)---)?([\r|\n]*)(?<heading>[#]{2,6} )(?<title>[\s\S].*)/;
+
+  constructor() {
+    super({
+      name: "nitwikit-content-transformer",
+      enforce: "pre",
+    });
+  }
+
+  override transform(code: string, id: string): string | Promise<string> | void {
+    if (id.endsWith(".md")) {
+      const result = code.match(NitWikitContentTransformer.#regExp);
+      let newCode = code;
+      if (result !== null) {
+        newCode = code.replace(`${result.groups?.heading}${result.groups?.title}`, `# ${result.groups?.title}`);
+      }
+
+      return newCode
+        .replace(/import ([\s\S]+) from ['"](@theme\/[\s\S]+)['"];/g, "")
+        .replace(/import ([\s\S]+) from &#39;(@theme\/[\s\S]+)&#39;;/g, "")
+        .replace(/values={\[([\s\S]*)\]}/g, "");
+    }
+  }
+}
diff --git a/build/plugins/transformer/url.ts b/build/plugins/transformer/url.ts
@@ -0,0 +1,21 @@
+import { BasePlugin } from "../base";
+
+export class NitWikitUrlTransformer extends BasePlugin {
+  static #regExp: RegExp = /\[(?<name>.*)\]\((?<url>https:\/\/nitwikit\.yizhan\.wiki\/(.*))\)/g;
+  static #url: string = "https://nitwikit.yizhan.wiki/";
+
+  constructor() {
+    super({
+      name: "nitwikit-url-transformer",
+      enforce: "pre",
+    });
+  }
+
+  override transform(code: string, id: string): string | Promise<string> | void {
+    if (id.endsWith(".md")) {
+      return code.replace(NitWikitUrlTransformer.#regExp, (match, _name, url) => {
+        return match.replace(url, url.replace(NitWikitUrlTransformer.#url, "/"));
+      });
+    }
+  }
+}