import re from typing import Any, List, Match from markdown import Markdown from markdown.extensions import Extension from markdown.preprocessors import Preprocessor from zerver.lib.emoji import EMOTICON_CONVERSIONS, name_to_codepoint from zerver.lib.markdown.preprocessor_priorities import PREPROCESSOR_PRIORITES REGEXP = re.compile(r"\{emoticon_translations\}") TABLE_HTML = """\ {body}
Emoticon Emoji
""" ROW_HTML = """\ {emoticon} {name} """ class EmoticonTranslationsHelpExtension(Extension): def extendMarkdown(self, md: Markdown) -> None: """Add SettingHelpExtension to the Markdown instance.""" md.registerExtension(self) md.preprocessors.register( EmoticonTranslation(), "emoticon_translations", PREPROCESSOR_PRIORITES["emoticon_translations"], ) class EmoticonTranslation(Preprocessor): def run(self, lines: List[str]) -> List[str]: for loc, line in enumerate(lines): match = REGEXP.search(line) if match: text = self.handleMatch(match) lines = lines[:loc] + text + lines[loc + 1 :] break return lines def handleMatch(self, match: Match[str]) -> List[str]: rows = [ ROW_HTML.format( emoticon=emoticon, name=name.strip(":"), codepoint=name_to_codepoint[name.strip(":")], ) for emoticon, name in EMOTICON_CONVERSIONS.items() ] body = "".join(rows).strip() return TABLE_HTML.format(body=body).strip().splitlines() def makeExtension(*args: Any, **kwargs: Any) -> EmoticonTranslationsHelpExtension: return EmoticonTranslationsHelpExtension(*args, **kwargs)