From 3c8469a0aa654742688a210ac0a39df1d8600298 Mon Sep 17 00:00:00 2001
From: Divyansh Singh <40380293+brc-dd@users.noreply.github.com>
Date: Thu, 20 Jan 2022 02:35:22 +0530
Subject: [PATCH] fix: append base to links

---
 src/node/markdown/markdown.ts     | 17 +++++++++++------
 src/node/markdown/plugins/link.ts |  5 ++++-
 src/node/markdownToVue.ts         |  5 +++--
 src/node/plugin.ts                |  3 ++-
 4 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/src/node/markdown/markdown.ts b/src/node/markdown/markdown.ts
index 1fdb9268..c9d68b9e 100644
--- a/src/node/markdown/markdown.ts
+++ b/src/node/markdown/markdown.ts
@@ -49,7 +49,8 @@ export type { Header }
 
 export const createMarkdownRenderer = (
   srcDir: string,
-  options: MarkdownOptions = {}
+  options: MarkdownOptions = {},
+  base: string
 ): MarkdownRenderer => {
   const md = MarkdownIt({
     html: true,
@@ -66,11 +67,15 @@ export const createMarkdownRenderer = (
     .use(hoistPlugin)
     .use(containerPlugin)
     .use(headingPlugin)
-    .use(linkPlugin, {
-      target: '_blank',
-      rel: 'noopener noreferrer',
-      ...options.externalLinks
-    })
+    .use(
+      linkPlugin,
+      {
+        target: '_blank',
+        rel: 'noopener noreferrer',
+        ...options.externalLinks
+      },
+      base
+    )
     // 3rd party plugins
     .use(attrs, options.attrs)
     .use(anchor, {
diff --git a/src/node/markdown/plugins/link.ts b/src/node/markdown/plugins/link.ts
index e44c043a..f897ecd6 100644
--- a/src/node/markdown/plugins/link.ts
+++ b/src/node/markdown/plugins/link.ts
@@ -11,7 +11,8 @@ const indexRE = /(^|.*\/)index.md(#?.*)$/i
 
 export const linkPlugin = (
   md: MarkdownIt,
-  externalAttrs: Record<string, string>
+  externalAttrs: Record<string, string>,
+  base: string
 ) => {
   md.renderer.rules.link_open = (tokens, idx, options, env, self) => {
     const token = tokens[idx]
@@ -76,6 +77,8 @@ export const linkPlugin = (
     // export it for existence check
     pushLink(url.replace(/\.html$/, ''))
 
+    url = EXTERNAL_URL_RE.test(url) ? url : `${base}${url}`.replace(/\/+/g, '/')
+
     // markdown-it encodes the uri
     hrefAttr[1] = decodeURI(url)
   }
diff --git a/src/node/markdownToVue.ts b/src/node/markdownToVue.ts
index a596e22b..c969c152 100644
--- a/src/node/markdownToVue.ts
+++ b/src/node/markdownToVue.ts
@@ -25,9 +25,10 @@ export function createMarkdownToVueRenderFn(
   options: MarkdownOptions = {},
   pages: string[],
   userDefines: Record<string, any> | undefined,
-  isBuild = false
+  isBuild = false,
+  base: string
 ) {
-  const md = createMarkdownRenderer(srcDir, options)
+  const md = createMarkdownRenderer(srcDir, options, base)
   pages = pages.map((p) => slash(p.replace(/\.md$/, '')))
 
   const userDefineRegex = userDefines
diff --git a/src/node/plugin.ts b/src/node/plugin.ts
index fc6d8e6f..7f253d06 100644
--- a/src/node/plugin.ts
+++ b/src/node/plugin.ts
@@ -84,7 +84,8 @@ export function createVitePressPlugin(
         markdown,
         pages,
         config.define,
-        config.command === 'build'
+        config.command === 'build',
+        config.base
       )
     },