RPGnotes/utils/markdown.py

import re

import bleach
import markdown
from bleach_allowlist import markdown_tags, markdown_attrs

custom_allowed_tags = ["del", "ins"]


def md_to_html(md: str) -> str:
    md = autolink(md)
    html = markdown.markdown(
        md,
        output_format="html",
        extensions=[
            "nl2br",
        ]
    )
    html = bleach.clean(
        html,
        tags=markdown_tags + custom_allowed_tags,
        attributes=markdown_attrs
    )
    return html


def autolink(md: str) -> str:
    from utils.urls import name2url
    for name, url in name2url().items():
        regex = r"\bWORD\b".replace("WORD", name)
        md = re.sub(regex, f"[{name}]({url})", md)
    return md