ascii escape

Author: Simon Berger

gisi/cogs/ascii.py

@@ -4,6 +4,7 @@
 from discord.ext.commands import command
 
 from gisi import SetDefaults
+from gisi.utils import text_utils
 
 log = logging.getLogger(__name__)
 
@@ -21,7 +22,7 @@ def get_asciimoji(self, key):
 
     def asciiemojify(self, text, require_wrapping=True):
         prog = r"(?<!\\)-(\w+)-" if require_wrapping else r"(\w+)"
-        return re.sub(prog, lambda m: self.get_asciimoji(m.group(1).lower()) or m.group(0),
+        return re.sub(prog, lambda m: text_utils.escape(self.get_asciimoji(m.group(1).lower())) or m.group(0),
                       text)
 
     @command()

gisi/utils/text_utils.py

@@ -0,0 +1,43 @@
+"""Text utilities."""
+
+from functools import partial
+
+DISCORD_FORMATTING_CHARS = {"*", "_", "~", "`", "\\"}
+BOLD_SEQ = "**"
+ITALIC_SEQ = "*"
+STRIKETHROUGH_SEQ = "~~"
+CODE_SEQ = "```"
+URL_ESCAPE_SEQ = ("<", ">")
+
+QUOTE_CHAR = "\""
+ESCAPE_CHAR = "\\"
+
+
+def wrap(s, wrap):
+    """Wrap around text."""
+    if isinstance(wrap, (tuple, list, set)):
+        start, end = wrap
+        return start + s + end
+    return wrap + s + wrap
+
+
+bold = partial(wrap, wrap=BOLD_SEQ)
+italic = partial(wrap, wrap=ITALIC_SEQ)
+strikethrough = partial(wrap, wrap=STRIKETHROUGH_SEQ)
+quote = partial(wrap, wrap=QUOTE_CHAR)
+escape_url = partial(wrap, wrap=URL_ESCAPE_SEQ)
+
+
+def code(s, lang):
+    """Put it in a code block."""
+    return f"{CODE_SEQ}{lang}\n{s}{CODE_SEQ}"
+
+
+def escape(s):
+    """Escape discord formatting in string."""
+    res = ""
+    for c in s:
+        if c in DISCORD_FORMATTING_CHARS:
+            res += ESCAPE_CHAR
+        res += c
+    return res