🔧 MAINTAIN: Add more type annotations (#96)

Author: hukkinj1

markdown_it/common/normalize_url.py

@@ -38,22 +38,22 @@
 )
 
 
-def unescape_char(s):
+def unescape_char(s: str) -> str:
     if s[0] == "\\":
         return s[1]
     else:
         return html.unescape(s)
 
 
-def unescape_string(s):
+def unescape_string(s: str) -> str:
     """Replace entities and backslash escapes with literal characters."""
     if re.search(reBackslashOrAmp, s):
         return re.sub(reEntityOrEscapedChar, lambda m: unescape_char(m.group()), s)
     else:
         return s
 
 
-def normalize_uri(uri):
+def normalize_uri(uri: str) -> str:
     return quote(uri.encode("utf-8"), safe=str("/@:+?=&()%#*,"))
 
 
@@ -63,7 +63,7 @@ def normalize_uri(uri):
 RECODE_HOSTNAME_FOR = ("http", "https", "mailto")
 
 
-def unescape_normalize_uri(x):
+def unescape_normalize_uri(x: str) -> str:
     return normalize_uri(unescape_string(x))
 
 
@@ -166,7 +166,7 @@ def normalizeLinkText(link):
 GOOD_DATA_RE = re.compile(r"^data:image\/(gif|png|jpeg|webp);")
 
 
-def validateLink(url: str, validator: Optional[Callable] = None):
+def validateLink(url: str, validator: Optional[Callable] = None) -> bool:
     """Validate URL link is allowed in output.
 
     This validator can prohibit more than really needed to prevent XSS.

markdown_it/common/utils.py

@@ -197,7 +197,7 @@ def escapeRE(string: str) -> str:
 # //////////////////////////////////////////////////////////////////////////////
 
 
-def isSpace(code):
+def isSpace(code) -> bool:
     return code in {0x09, 0x20}
 
 
@@ -216,7 +216,7 @@ def isSpace(code):
 }
 
 
-def isWhiteSpace(code):
+def isWhiteSpace(code: int) -> bool:
     r"""Zs (unicode class) || [\t\f\v\r\n]"""
     if code >= 0x2000 and code <= 0x200A:
         return True
@@ -231,7 +231,7 @@ def isWhiteSpace(code):
 
 
 # Currently without astral characters support.
-def isPunctChar(ch):
+def isPunctChar(ch: str) -> bool:
     return UNICODE_PUNCT_RE.search(ch) is not None
 
 

markdown_it/helpers/parse_link_destination.py

@@ -15,7 +15,7 @@ def __init__(self):
         self.str = ""
 
 
-def parseLinkDestination(string, pos, maximum):
+def parseLinkDestination(string: str, pos: int, maximum: int) -> _Result:
     lines = 0
     start = pos
     result = _Result()

markdown_it/helpers/parse_link_label.py

@@ -5,9 +5,10 @@
 returns the end of the label
 
 """
+from markdown_it.rules_inline import StateInline
 
 
-def parseLinkLabel(state, start, disableNested=False):
+def parseLinkLabel(state: StateInline, start: int, disableNested: bool = False) -> int:
 
     labelEnd = -1
     oldPos = state.pos

markdown_it/helpers/parse_link_title.py

@@ -16,7 +16,7 @@ def __str__(self):
         return self.str
 
 
-def parseLinkTitle(string, pos, maximum):
+def parseLinkTitle(string: str, pos: int, maximum: int) -> _Result:
     lines = 0
     start = pos
     result = _Result()

markdown_it/rules_block/state_block.py

@@ -117,7 +117,7 @@ def __repr__(self):
             f"(line={self.line},level={self.level},tokens={len(self.tokens)})"
         )
 
-    def push(self, ttype, tag, nesting):
+    def push(self, ttype: str, tag: str, nesting: int) -> Token:
         """Push new token to "stream"."""
         token = Token(ttype, tag, nesting)
         token.block = True
@@ -129,7 +129,7 @@ def push(self, ttype, tag, nesting):
         self.tokens.append(token)
         return token
 
-    def isEmpty(self, line):
+    def isEmpty(self, line: int) -> bool:
         """."""
         return (self.bMarks[line] + self.tShift[line]) >= self.eMarks[line]
 
@@ -183,7 +183,7 @@ def skipCharsBack(self, pos: int, code: int, minimum: int) -> int:
                 return pos + 1
         return pos
 
-    def getLines(self, begin: int, end: int, indent, keepLastLF):
+    def getLines(self, begin: int, end: int, indent: int, keepLastLF: bool) -> str:
         """Cut lines range from source."""
         line = begin
         if begin >= end:

markdown_it/rules_core/block.py

@@ -2,7 +2,7 @@
 from .state_core import StateCore
 
 
-def block(state: StateCore):
+def block(state: StateCore) -> None:
 
     if state.inlineMode:
         token = Token("inline", "", 0)

markdown_it/rules_core/inline.py

@@ -1,7 +1,7 @@
 from .state_core import StateCore
 
 
-def inline(state: StateCore):
+def inline(state: StateCore) -> None:
     """Parse inlines"""
     for token in state.tokens:
         if token.type == "inline":

markdown_it/rules_core/linkify.py

@@ -14,15 +14,15 @@
 TEST_MAILTO_RE = re.compile(r"^mailto:", flags=re.IGNORECASE)
 
 
-def isLinkOpen(string: str):
-    return LINK_OPEN_RE.search(string)
+def isLinkOpen(string: str) -> bool:
+    return bool(LINK_OPEN_RE.search(string))
 
 
-def isLinkClose(string: str):
-    return LINK_CLOSE_RE.search(string)
+def isLinkClose(string: str) -> bool:
+    return bool(LINK_CLOSE_RE.search(string))
 
 
-def linkify(state: StateCore):
+def linkify(state: StateCore) -> None:
     blockTokens = state.tokens
 
     if not state.md.options.linkify:

markdown_it/rules_core/normalize.py

@@ -9,7 +9,7 @@
 NULL_RE = re.compile(r"\0")
 
 
-def normalize(state: StateCore):
+def normalize(state: StateCore) -> None:
 
     # Normalize newlines
     string = NEWLINES_RE.sub("\n", state.src)