From 168fce09da2d5b26f24187970a425ae875a49527 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Tue, 25 Jun 2019 10:24:19 +0200
Subject: [PATCH] Implement HTML.unparse and Markdown.unparse

---
 pyrogram/client/style/html.py     | 44 +++++++++++++++++++
 pyrogram/client/style/markdown.py | 72 +++++++++++++++++++++++++------
 2 files changed, 103 insertions(+), 13 deletions(-)

diff --git a/pyrogram/client/style/html.py b/pyrogram/client/style/html.py
index 579ed7ec..17a5daa6 100644
--- a/pyrogram/client/style/html.py
+++ b/pyrogram/client/style/html.py
@@ -134,3 +134,47 @@ class HTML:
             ("message", utils.remove_surrogates(parser.text)),
             ("entities", parser.entities)
         ])
+
+    @staticmethod
+    def unparse(text: str, entities: list):
+        text = utils.add_surrogates(text)
+        copy = text
+
+        for entity in entities:
+            start = entity.offset
+            end = start + entity.length
+
+            type = entity.type
+
+            url = entity.url
+            user = entity.user
+
+            sub = copy[start:end]
+
+            if type == "bold":
+                style = "b"
+            elif type == "italic":
+                style = "i"
+            elif type == "underline":
+                style = "u"
+            elif type == "strike":
+                style = "s"
+            elif type == "code":
+                style = "code"
+            elif type == "pre":
+                style = "pre"
+            elif type == "blockquote":
+                style = "blockquote"
+            elif type == "text_link":
+                text = text[:start] + text[start:].replace(sub, '<a href="{}">{}</a>'.format(url, sub), 1)
+                continue
+            elif type == "text_mention":
+                text = text[:start] + text[start:].replace(
+                    sub, '<a href="tg://user?id={}">{}</a>'.format(user.id, sub), 1)
+                continue
+            else:
+                continue
+
+            text = text[:start] + text[start:].replace(sub, "<{0}>{1}</{0}>".format(style, sub), 1)
+
+        return utils.remove_surrogates(text)
diff --git a/pyrogram/client/style/markdown.py b/pyrogram/client/style/markdown.py
index 520008a8..93d8fc9a 100644
--- a/pyrogram/client/style/markdown.py
+++ b/pyrogram/client/style/markdown.py
@@ -20,17 +20,18 @@ import html
 import re
 
 import pyrogram
+from . import utils
 from .html import HTML
 
+BOLD_DELIM = "**"
+ITALIC_DELIM = "__"
+UNDERLINE_DELIM = "--"
+STRIKE_DELIM = "~~"
+CODE_DELIM = "`"
+PRE_DELIM = "```"
+
 
 class Markdown:
-    BOLD_DELIM = "**"
-    ITALIC_DELIM = "__"
-    UNDERLINE_DELIM = "--"
-    STRIKE_DELIM = "~~"
-    CODE_DELIM = "`"
-    PRE_DELIM = "```"
-
     MARKDOWN_RE = re.compile(r"({d})".format(
         d="|".join(
             ["".join(i) for i in [
@@ -66,17 +67,17 @@ class Markdown:
             start, stop = match.span()
             delim = match.group(1)
 
-            if delim == Markdown.BOLD_DELIM:
+            if delim == BOLD_DELIM:
                 tag = "b"
-            elif delim == Markdown.ITALIC_DELIM:
+            elif delim == ITALIC_DELIM:
                 tag = "i"
-            elif delim == Markdown.UNDERLINE_DELIM:
+            elif delim == UNDERLINE_DELIM:
                 tag = "u"
-            elif delim == Markdown.STRIKE_DELIM:
+            elif delim == STRIKE_DELIM:
                 tag = "s"
-            elif delim == Markdown.CODE_DELIM:
+            elif delim == CODE_DELIM:
                 tag = "code"
-            elif delim == Markdown.PRE_DELIM:
+            elif delim == PRE_DELIM:
                 tag = "pre"
             else:
                 continue
@@ -109,3 +110,48 @@ class Markdown:
             offset += len(replace) - len(full)
 
         return self.html.parse(text)
+
+    @staticmethod
+    def unparse(text: str, entities: list):
+        text = utils.add_surrogates(text)
+        copy = text
+
+        for entity in entities:
+            start = entity.offset
+            end = start + entity.length
+
+            type = entity.type
+
+            url = entity.url
+            user = entity.user
+
+            sub = copy[start:end]
+
+            if type == "bold":
+                style = BOLD_DELIM
+            elif type == "italic":
+                style = ITALIC_DELIM
+            elif type == "underline":
+                style = UNDERLINE_DELIM
+            elif type == "strike":
+                style = STRIKE_DELIM
+            elif type == "code":
+                style = CODE_DELIM
+            elif type == "pre":
+                style = PRE_DELIM
+            # TODO: Blockquote for MD
+            # elif type == "blockquote":
+            #     style = ...
+            elif type == "text_link":
+                text = text[:start] + text[start:].replace(sub, '[{1}]({0})'.format(url, sub), 1)
+                continue
+            elif type == "text_mention":
+                text = text[:start] + text[start:].replace(
+                    sub, '[{1}](tg://user?id={0})'.format(user.id, sub), 1)
+                continue
+            else:
+                continue
+
+            text = text[:start] + text[start:].replace(sub, "{0}{1}{0}".format(style, sub), 1)
+
+        return utils.remove_surrogates(text)