summaryrefslogtreecommitdiff
path: root/src/mistune/helpers.py
diff options
context:
space:
mode:
authorArturs Artamonovs <dos21h@gmail.com>2023-01-29 10:30:54 +0000
committerArturs Artamonovs <dos21h@gmail.com>2023-01-29 10:30:54 +0000
commit66fa71a8f11b6ce5e8471b533f67cc3a1fdb85a8 (patch)
tree7aed7f385826a3bd88c76a373e28c6cfae4f396e /src/mistune/helpers.py
parent129c1201ea5c4418f0f89ad932633c7cea2439b7 (diff)
downloadmd-site-66fa71a8f11b6ce5e8471b533f67cc3a1fdb85a8.tar.gz
md-site-66fa71a8f11b6ce5e8471b533f67cc3a1fdb85a8.zip
Update to new mistune, removed old mistune, rewrite to python3
Diffstat (limited to 'src/mistune/helpers.py')
-rw-r--r--src/mistune/helpers.py137
1 files changed, 137 insertions, 0 deletions
diff --git a/src/mistune/helpers.py b/src/mistune/helpers.py
new file mode 100644
index 0000000..04c1df1
--- /dev/null
+++ b/src/mistune/helpers.py
@@ -0,0 +1,137 @@
+import re
+import string
+from .util import escape_url
+
+PREVENT_BACKSLASH = r'(?<!\\)(?:\\\\)*'
+PUNCTUATION = r'[' + re.escape(string.punctuation) + r']'
+
+LINK_LABEL = r'(?:[^\\\[\]]|\\.){0,500}'
+
+LINK_BRACKET_START = re.compile(r'[ \t]*\n?[ \t]*<')
+LINK_BRACKET_RE = re.compile(r'<([^<>\n\\\x00]*)>')
+LINK_HREF_BLOCK_RE = re.compile(r'[ \t]*\n?[ \t]*([^\s]+)(?:\s|$)')
+LINK_HREF_INLINE_RE = re.compile(
+ r'[ \t]*\n?[ \t]*([^ \t\n]*?)(?:[ \t\n]|'
+ r'(?:' + PREVENT_BACKSLASH + r'\)))'
+)
+
+LINK_TITLE_RE = re.compile(
+ r'[ \t\n]+('
+ r'"(?:\\' + PUNCTUATION + r'|[^"\x00])*"|' # "title"
+ r"'(?:\\" + PUNCTUATION + r"|[^'\x00])*'" # 'title'
+ r')'
+)
+PAREN_END_RE = re.compile(r'\s*\)')
+
+HTML_TAGNAME = r'[A-Za-z][A-Za-z0-9-]*'
+HTML_ATTRIBUTES = (
+ r'(?:\s+[A-Za-z_:][A-Za-z0-9_.:-]*'
+ r'(?:\s*=\s*(?:[^ !"\'=<>`]+|\'[^\']*?\'|"[^\"]*?"))?)*'
+)
+
+BLOCK_TAGS = (
+ 'address', 'article', 'aside', 'base', 'basefont', 'blockquote',
+ 'body', 'caption', 'center', 'col', 'colgroup', 'dd', 'details',
+ 'dialog', 'dir', 'div', 'dl', 'dt', 'fieldset', 'figcaption',
+ 'figure', 'footer', 'form', 'frame', 'frameset', 'h1', 'h2', 'h3',
+ 'h4', 'h5', 'h6', 'head', 'header', 'hr', 'html', 'iframe',
+ 'legend', 'li', 'link', 'main', 'menu', 'menuitem', 'meta', 'nav',
+ 'noframes', 'ol', 'optgroup', 'option', 'p', 'param', 'section',
+ 'source', 'summary', 'table', 'tbody', 'td', 'tfoot', 'th', 'thead',
+ 'title', 'tr', 'track', 'ul'
+)
+PRE_TAGS = ('pre', 'script', 'style', 'textarea')
+
+_INLINE_LINK_LABEL_RE = re.compile(LINK_LABEL + r'\]')
+_INLINE_SQUARE_BRACKET_RE = re.compile(PREVENT_BACKSLASH + r'[\[\]]')
+_ESCAPE_CHAR_RE = re.compile(r'\\(' + PUNCTUATION + r')')
+
+
+def unescape_char(text):
+ return _ESCAPE_CHAR_RE.sub(r'\1', text)
+
+
+def parse_link_text(src, pos):
+ level = 1
+ found = False
+ start_pos = pos
+
+ while pos < len(src):
+ m = _INLINE_SQUARE_BRACKET_RE.search(src, pos)
+ if not m:
+ break
+
+ pos = m.end()
+ marker = m.group(0)
+ if marker == ']':
+ level -= 1
+ if level == 0:
+ found = True
+ break
+ else:
+ level += 1
+
+ if found:
+ text = src[start_pos:pos-1]
+ return text, pos
+ return None, None
+
+
+def parse_link_label(src, start_pos):
+ m = _INLINE_LINK_LABEL_RE.match(src, start_pos)
+ if m:
+ label = m.group(0)[:-1]
+ return label, m.end()
+ return None, None
+
+
+def parse_link_href(src, start_pos, block=False):
+ m = LINK_BRACKET_START.match(src, start_pos)
+ if m:
+ start_pos = m.end() - 1
+ m = LINK_BRACKET_RE.match(src, start_pos)
+ if m:
+ return m.group(1), m.end()
+ return None, None
+
+ if block:
+ m = LINK_HREF_BLOCK_RE.match(src, start_pos)
+ else:
+ m = LINK_HREF_INLINE_RE.match(src, start_pos)
+
+ if not m:
+ return None, None
+
+ end_pos = m.end()
+ href = m.group(1)
+
+ if block and src[end_pos - 1] == href[-1]:
+ return href, end_pos
+ return href, end_pos - 1
+
+
+def parse_link_title(src, start_pos, max_pos):
+ m = LINK_TITLE_RE.match(src, start_pos, max_pos)
+ if m:
+ title = m.group(1)[1:-1]
+ title = unescape_char(title)
+ return title, m.end()
+ return None, None
+
+
+def parse_link(src, pos):
+ href, href_pos = parse_link_href(src, pos)
+ if href is None:
+ return None, None
+
+ title, title_pos = parse_link_title(src, href_pos, len(src))
+ next_pos = title_pos or href_pos
+ m = PAREN_END_RE.match(src, next_pos)
+ if not m:
+ return None, None
+
+ href = unescape_char(href)
+ attrs = {'url': escape_url(href)}
+ if title:
+ attrs['title'] = title
+ return attrs, m.end()