add for v2 as well

Author: endiliey

v1/lib/server/docs.js

@@ -50,14 +50,14 @@ function mdToHtmlify(oldContent, mdToHtml, metadata) {
   const mdLinks = [];
   const mdReferences = [];
 
-  // find any links to markdown files
+  // find any inline-style links to markdown files
   const linkRegex = /(?:\]\()(?:\.\/)?([^'")\]\s>]+\.md)/g;
   let linkMatch = linkRegex.exec(content);
   while (linkMatch !== null) {
     mdLinks.push(linkMatch[1]);
     linkMatch = linkRegex.exec(content);
   }
-  // find any reference links to markdown files
+  // find any reference-style links to markdown files
   const refRegex = /(?:\]:)(?:\s)?(?:\.\/|\.\.\/)?([^'")\]\s>]+\.md)/g;
   let refMatch = refRegex.exec(content);
   while (refMatch !== null) {

v2/lib/webpack/loader/markdown.js

@@ -52,14 +52,31 @@ module.exports = function(fileString) {
       if (fencedBlock) return line;
 
       let modifiedLine = line;
-      const mdLinks = [];
-      const mdRegex = /(?:\]\()(?:\.\/)?([^'")\]\s>]+\.md)/g;
-      let match = mdRegex.exec(content);
-      while (match !== null) {
-        mdLinks.push(match[1]);
-        match = mdRegex.exec(content);
+      const inlineLinks = [];
+      const refLinks = [];
+
+      /* Replace inline-style links e.g: 
+      This is [Document 1](doc1.md) -> we replace this doc1.md with correct link
+      */
+      const inlineRegex = /(?:\]\()(?:\.\/)?([^'")\]\s>]+\.md)/g;
+      let inlineMatch = inlineRegex.exec(content);
+      while (inlineMatch !== null) {
+        inlineLinks.push(inlineMatch[1]);
+        inlineMatch = inlineRegex.exec(content);
+      }
+
+      /* Replace reference-style links e.g:  
+        This is [Document 1][doc1].
+        [doc1]: doc1.md -> we replace this doc1.md with correct link
+      */
+      const refRegex = /(?:\]:)(?:\s)?(?:\.\/|\.\.\/)?([^'")\]\s>]+\.md)/g;
+      let refMatch = refRegex.exec(content);
+      while (refMatch !== null) {
+        refLinks.push(refMatch[1]);
+        refMatch = refRegex.exec(content);
       }
-      mdLinks.forEach(mdLink => {
+
+      [...refLinks, ...inlineLinks].forEach(mdLink => {
         const targetSource = `${sourceDir}/${mdLink}`;
         const {permalink} = sourceToMetadata[targetSource] || {};
         if (permalink) {