fixed nested annotations

1 files changed, 9 insertions, 6 deletions

diff --git a/main.py b/main.py
index ebb03aa..33dda13 100755
--- a/main.py
+++ b/main.py
@@ -35,22 +35,25 @@ def register_signald(number):
 
 def handle_replies(replies):
     answer = ""
+    if type(replies) is not list:
+        replies = [replies]
     for reply in replies:
-        answer = reply['text']
+        l_answer = reply['text']
         for annotation in reply["annotations"]:
             a_type = annotation["type"]
             a_extra = handle_replies(annotation["extra"])
 
             if a_type == "command":
-                answer = f"/{answer}: {a_extra}\n"
+                l_answer = f"/{l_answer}: {a_extra}\n"
             elif a_type == "link":
-                answer = f"{a_extra} ({answer})"
+                l_answer = f"{a_extra} ({l_answer})"
             elif a_type == "bold":
-                answer = f"*{answer}*"
+                l_answer = f"*{l_answer}*"
             elif a_type == "italic":
-                answer = f"_{answer}_"
+                l_answer = f"_{l_answer}_"
             elif a_type == "strikethrough":
-                answer = f"~{answer}~"
+                l_answer = f"~{l_answer}~"
+        answer += l_answer
     return answer