3RNN/Lib/site-packages/markdown_it/rules_inline/image.py
2024-05-26 19:49:15 +02:00

149 lines
4.0 KiB
Python

# Process ![image](<src> "title")
from __future__ import annotations
from ..common.utils import isStrSpace, normalizeReference
from ..token import Token
from .state_inline import StateInline
def image(state: StateInline, silent: bool) -> bool:
label = None
href = ""
oldPos = state.pos
max = state.posMax
if state.src[state.pos] != "!":
return False
if state.pos + 1 < state.posMax and state.src[state.pos + 1] != "[":
return False
labelStart = state.pos + 2
labelEnd = state.md.helpers.parseLinkLabel(state, state.pos + 1, False)
# parser failed to find ']', so it's not a valid link
if labelEnd < 0:
return False
pos = labelEnd + 1
if pos < max and state.src[pos] == "(":
#
# Inline link
#
# [link]( <href> "title" )
# ^^ skipping these spaces
pos += 1
while pos < max:
ch = state.src[pos]
if not isStrSpace(ch) and ch != "\n":
break
pos += 1
if pos >= max:
return False
# [link]( <href> "title" )
# ^^^^^^ parsing link destination
start = pos
res = state.md.helpers.parseLinkDestination(state.src, pos, state.posMax)
if res.ok:
href = state.md.normalizeLink(res.str)
if state.md.validateLink(href):
pos = res.pos
else:
href = ""
# [link]( <href> "title" )
# ^^ skipping these spaces
start = pos
while pos < max:
ch = state.src[pos]
if not isStrSpace(ch) and ch != "\n":
break
pos += 1
# [link]( <href> "title" )
# ^^^^^^^ parsing link title
res = state.md.helpers.parseLinkTitle(state.src, pos, state.posMax)
if pos < max and start != pos and res.ok:
title = res.str
pos = res.pos
# [link]( <href> "title" )
# ^^ skipping these spaces
while pos < max:
ch = state.src[pos]
if not isStrSpace(ch) and ch != "\n":
break
pos += 1
else:
title = ""
if pos >= max or state.src[pos] != ")":
state.pos = oldPos
return False
pos += 1
else:
#
# Link reference
#
if "references" not in state.env:
return False
# /* [ */
if pos < max and state.src[pos] == "[":
start = pos + 1
pos = state.md.helpers.parseLinkLabel(state, pos)
if pos >= 0:
label = state.src[start:pos]
pos += 1
else:
pos = labelEnd + 1
else:
pos = labelEnd + 1
# covers label == '' and label == undefined
# (collapsed reference link and shortcut reference link respectively)
if not label:
label = state.src[labelStart:labelEnd]
label = normalizeReference(label)
ref = state.env["references"].get(label, None)
if not ref:
state.pos = oldPos
return False
href = ref["href"]
title = ref["title"]
#
# We found the end of the link, and know for a fact it's a valid link
# so all that's left to do is to call tokenizer.
#
if not silent:
content = state.src[labelStart:labelEnd]
tokens: list[Token] = []
state.md.inline.parse(content, state.md, state.env, tokens)
token = state.push("image", "img", 0)
token.attrs = {"src": href, "alt": ""}
token.children = tokens or None
token.content = content
if title:
token.attrSet("title", title)
# note, this is not part of markdown-it JS, but is useful for renderers
if label and state.md.options.get("store_labels", False):
token.meta["label"] = label
state.pos = pos
state.posMax = max
return True