| from typing import TYPE_CHECKING, Match |
|
|
| if TYPE_CHECKING: |
| from ..block_parser import BlockParser |
| from ..core import BaseRenderer, BlockState, InlineState |
| from ..inline_parser import InlineParser |
| from ..markdown import Markdown |
|
|
| __all__ = ["math", "math_in_quote", "math_in_list"] |
|
|
| BLOCK_MATH_PATTERN = r"^ {0,3}\$\$[ \t]*\n(?P<math_text>[\s\S]+?)\n\$\$[ \t]*$" |
| INLINE_MATH_PATTERN = r"\$(?!\s)(?P<math_text>.+?)(?!\s)\$" |
|
|
|
|
| def parse_block_math(block: "BlockParser", m: Match[str], state: "BlockState") -> int: |
| text = m.group("math_text") |
| state.append_token({"type": "block_math", "raw": text}) |
| return m.end() + 1 |
|
|
|
|
| def parse_inline_math(inline: "InlineParser", m: Match[str], state: "InlineState") -> int: |
| text = m.group("math_text") |
| state.append_token({"type": "inline_math", "raw": text}) |
| return m.end() |
|
|
|
|
| def render_block_math(renderer: "BaseRenderer", text: str) -> str: |
| return '<div class="math">$$\n' + text + "\n$$</div>\n" |
|
|
|
|
| def render_inline_math(renderer: "BaseRenderer", text: str) -> str: |
| return r'<span class="math">\(' + text + r"\)</span>" |
|
|
|
|
| def math(md: "Markdown") -> None: |
| """A mistune plugin to support math. The syntax is used |
| by many markdown extensions: |
| |
| .. code-block:: text |
| |
| Block math is surrounded by $$: |
| |
| $$ |
| f(a)=f(b) |
| $$ |
| |
| Inline math is surrounded by `$`, such as $f(a)=f(b)$ |
| |
| :param md: Markdown instance |
| """ |
| md.block.register("block_math", BLOCK_MATH_PATTERN, parse_block_math, before="list") |
| md.inline.register("inline_math", INLINE_MATH_PATTERN, parse_inline_math, before="link") |
| if md.renderer and md.renderer.NAME == "html": |
| md.renderer.register("block_math", render_block_math) |
| md.renderer.register("inline_math", render_inline_math) |
|
|
|
|
| def math_in_quote(md: "Markdown") -> None: |
| """Enable block math plugin in block quote.""" |
| md.block.insert_rule(md.block.block_quote_rules, "block_math", before="list") |
|
|
|
|
| def math_in_list(md: "Markdown") -> None: |
| """Enable block math plugin in list.""" |
| md.block.insert_rule(md.block.list_rules, "block_math", before="list") |
|
|
