👌 IMPROVE: autolink parsing performance

Fix quadratic complexity in autolinks

implements: markdown-it/markdown-it@e729b90

Author: chrisjsewell

markdown_it/rules_inline/autolink.py

@@ -1,11 +1,12 @@
 # Process autolinks '<protocol:...>'
 import re
 from .state_inline import StateInline
+from ..common.normalize_url import normalizeLinkText, normalizeLink, validateLink
 
 EMAIL_RE = re.compile(
-    r"^<([a-zA-Z0-9.!#$%&\'*+\/=?^_`{|}~-]+@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*)>"  # noqa: E501
+    r"^([a-zA-Z0-9.!#$%&\'*+\/=?^_`{|}~-]+@[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?(?:\.[a-zA-Z0-9](?:[a-zA-Z0-9-]{0,61}[a-zA-Z0-9])?)*)$"  # noqa: E501
 )
-AUTOLINK_RE = re.compile(r"^<([a-zA-Z][a-zA-Z0-9+.\-]{1,31}):([^<>\x00-\x20]*)>")
+AUTOLINK_RE = re.compile(r"^([a-zA-Z][a-zA-Z0-9+.\-]{1,31}):([^<>\x00-\x20]*)$")
 
 
 def autolink(state: StateInline, silent: bool) -> bool:
@@ -15,17 +16,26 @@ def autolink(state: StateInline, silent: bool) -> bool:
     if state.srcCharCode[pos] != 0x3C:  # /* < */
         return False
 
-    tail = state.src[pos:]
+    start = state.pos
+    maximum = state.posMax
 
-    if ">" not in tail:
-        return False
+    while True:
+        pos += 1
+        if pos >= maximum:
+            return False
+
+        ch = state.srcCharCode[pos]
 
-    linkMatch = AUTOLINK_RE.search(tail)
-    if linkMatch is not None:
+        if ch == 0x3C:  # /* < */
+            return False
+        if ch == 0x3E:  # /* > */
+            break
+
+    url = state.src[start + 1 : pos]
 
-        url = linkMatch.group(0)[1:-1]
-        fullUrl = state.md.normalizeLink(url)
-        if not state.md.validateLink(fullUrl):
+    if AUTOLINK_RE.search(url) is not None:
+        fullUrl = normalizeLink(url)
+        if not validateLink(fullUrl):
             return False
 
         if not silent:
@@ -35,21 +45,18 @@ def autolink(state: StateInline, silent: bool) -> bool:
             token.info = "auto"
 
             token = state.push("text", "", 0)
-            token.content = state.md.normalizeLinkText(url)
+            token.content = normalizeLinkText(url)
 
             token = state.push("link_close", "a", -1)
             token.markup = "autolink"
             token.info = "auto"
 
-        state.pos += len(linkMatch.group(0))
+        state.pos += len(url) + 2
         return True
 
-    emailMatch = EMAIL_RE.search(tail)
-    if emailMatch is not None:
-
-        url = emailMatch.group(0)[1:-1]
-        fullUrl = state.md.normalizeLink("mailto:" + url)
-        if not state.md.validateLink(fullUrl):
+    if EMAIL_RE.search(url) is not None:
+        fullUrl = normalizeLink("mailto:" + url)
+        if not validateLink(fullUrl):
             return False
 
         if not silent:
@@ -59,13 +66,13 @@ def autolink(state: StateInline, silent: bool) -> bool:
             token.info = "auto"
 
             token = state.push("text", "", 0)
-            token.content = state.md.normalizeLinkText(url)
+            token.content = normalizeLinkText(url)
 
             token = state.push("link_close", "a", -1)
             token.markup = "autolink"
             token.info = "auto"
 
-        state.pos += len(emailMatch.group(0))
+        state.pos += len(url) + 2
         return True
 
     return False