Merge pull request #162 from curbengh/fix-autolink

fix: autolink option should not apply on markdown syntax
hexojs · Aug 31, 2020 · 0abcf7d · 0abcf7d
2 parents 29b49aa + 40d8204
commit 0abcf7d
Show file tree

Hide file tree

Showing 2 changed files with 73 additions and 13 deletions.
diff --git a/lib/renderer.js b/lib/renderer.js
@@ -1,8 +1,10 @@
 'use strict';
 
 const marked = require('marked');
+const { escape } = require('marked/src/helpers');
 const { encodeURL, slugize, stripHTML, url_for, isExternalLink } = require('hexo-util');
 const MarkedRenderer = marked.Renderer;
+const MarkedTokenizer = marked.Tokenizer;
 
 const anchorId = (str, transformOption) => {
   return slugize(str.trim(), {transform: transformOption});
@@ -39,19 +41,15 @@ class Renderer extends MarkedRenderer {
     return `<h${level} id="${id}"><a href="#${id}" class="headerlink" title="${stripHTML(text)}"></a>${text}</h${level}>`;
   }
 
-  // Support AutoLink option
   link(href, title, text) {
-    const { autolink, external_link, sanitizeUrl } = this.options;
+    const { external_link, sanitizeUrl } = this.options;
     const { url: urlCfg } = this.hexo.config;
 
     if (sanitizeUrl) {
       if (href.startsWith('javascript:') || href.startsWith('vbscript:') || href.startsWith('data:')) {
         href = '';
       }
     }
-    if (!autolink && href === text && title == null) {
-      return href;
-    }
 
     let out = `<a href="${encodeURL(href)}"`;
     if (title) {
@@ -117,6 +115,53 @@ marked.setOptions({
   langPrefix: ''
 });
 
+class Tokenizer extends MarkedTokenizer {
+  // Support AutoLink option
+  // https://github.com/markedjs/marked/blob/b6773fca412c339e0cedd56b63f9fa1583cfd372/src/Tokenizer.js#L606-L641
+  url(src, mangle) {
+    const { options, rules } = this;
+    const { mangle: isMangle, autolink } = options;
+
+    if (!autolink) return;
+
+    const cap = rules.inline.url.exec(src);
+    if (cap) {
+      let text, href;
+      if (cap[2] === '@') {
+        text = escape(isMangle ? mangle(cap[0]) : cap[0]);
+        href = 'mailto:' + text;
+      } else {
+        // do extended autolink path validation
+        let prevCapZero;
+        do {
+          prevCapZero = cap[0];
+          cap[0] = rules.inline._backpedal.exec(cap[0])[0];
+        } while (prevCapZero !== cap[0]);
+        text = escape(cap[0]);
+        if (cap[1] === 'www.') {
+          href = 'http://' + text;
+        } else {
+          href = text;
+        }
+      }
+
+      return {
+        type: 'link',
+        raw: cap[0],
+        text,
+        href,
+        tokens: [
+          {
+            type: 'text',
+            raw: text,
+            text
+          }
+        ]
+      };
+    }
+  }
+}
+
 module.exports = function(data, options) {
   const { post_asset_folder, marked: markedCfg, source_dir } = this.config;
   const { prependRoot, postAsset } = markedCfg;
@@ -126,6 +171,8 @@ module.exports = function(data, options) {
   const renderer = new Renderer(this);
   this.execFilterSync('marked:renderer', renderer, {context: this});
 
+  const tokenizer = new Tokenizer();
+
   let postPath = '';
   if (path && post_asset_folder && prependRoot && postAsset) {
     const Post = this.model('Post');
@@ -136,6 +183,7 @@ module.exports = function(data, options) {
   }
 
   return marked(text, Object.assign({
-    renderer
+    renderer,
+    tokenizer
   }, markedCfg, options, { postPath }));
 };
diff --git a/test/index.js b/test/index.js
@@ -165,26 +165,38 @@ describe('Marked renderer', () => {
     const body = [
       'Great website http://hexo.io',
       '',
-      '[Hexo](http://hexo.io)'
+      'A webpage www.example.com',
+      '',
+      '[Hexo](http://hexo.io)',
+      '',
+      '[http://lorem.com/foo/](http://lorem.com/foo/)',
+      '',
+      '<http://dolor.com>'
     ].join('\n');
 
     it('autolink enabled', () => {
       const result = r({text: body});
 
       result.should.eql([
-        '<p>Great website <a href="http://hexo.io/">http://hexo.io</a></p>\n',
-        '<p><a href="http://hexo.io/">Hexo</a></p>\n'
-      ].join(''));
+        '<p>Great website <a href="http://hexo.io/">http://hexo.io</a></p>',
+        '<p>A webpage <a href="http://www.example.com/">www.example.com</a></p>',
+        '<p><a href="http://hexo.io/">Hexo</a></p>',
+        '<p><a href="http://lorem.com/foo/">http://lorem.com/foo/</a></p>',
+        '<p><a href="http://dolor.com/">http://dolor.com</a></p>'
+      ].join('\n') + '\n');
     });
 
     it('autolink disabled', () => {
       ctx.config.marked.autolink = false;
       const result = r({text: body});
 
       result.should.eql([
-        '<p>Great website http://hexo.io</p>\n',
-        '<p><a href="http://hexo.io/">Hexo</a></p>\n'
-      ].join(''));
+        '<p>Great website http://hexo.io</p>',
+        '<p>A webpage www.example.com</p>',
+        '<p><a href="http://hexo.io/">Hexo</a></p>',
+        '<p><a href="http://lorem.com/foo/">http://lorem.com/foo/</a></p>',
+        '<p><a href="http://dolor.com/">http://dolor.com</a></p>'
+      ].join('\n') + '\n');
     });
   });