feat: refine HTML transformation to handle root-relative links for im…

…g and link tags
gkd-kit · Nov 7, 2024 · 33ae124 · 33ae124
1 parent cc14590
commit 33ae124
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/docs/.vitepress/plugins/mirror.ts b/docs/.vitepress/plugins/mirror.ts
@@ -87,9 +87,15 @@ export const transformHtml = (code: string) => {
       e.setAttribute(attr, mirrorBaseUrl + e.getAttribute(attr));
     });
   });
-  doc.querySelectorAll('link[href^="/"]').forEach((e) => {
+
+  doc.querySelectorAll('[href^="/"]').forEach((e) => {
+    const tag = e.tagName.toLowerCase();
     const href = e.getAttribute('href');
-    if (href && href.lastIndexOf('/') === 0) {
+    if (
+      (tag === 'img' || tag === 'link') &&
+      href &&
+      href.lastIndexOf('/') === 0
+    ) {
       e.setAttribute('href', mirrorBaseUrl + href);
     }
   });