From e075c24306c8921dc4b23fa572d8412471166ffc Mon Sep 17 00:00:00 2001
From: Ceda EI <ceda_ei@protonmail.com>
Date: Mon, 28 Jan 2019 16:32:31 +0530
Subject: [PATCH] Deal with all data types and replies.

---
 bot/bot.py | 83 ++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 74 insertions(+), 9 deletions(-)

diff --git a/bot/bot.py b/bot/bot.py
index 9f0d0ad..9bd184e 100644
--- a/bot/bot.py
+++ b/bot/bot.py
@@ -3,25 +3,90 @@
 import config
 import logging
 import requests
+import re
 from telegram.ext import Updater, MessageHandler, Filters
 
 logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \
                     %(message)s', level=logging.INFO)
 
 
+def get_text_or_type(msg, bot, truncate=False):
+    message = ""
+
+    # Set message contents
+    if msg.text:
+        if msg.from_user.username == bot.username:
+            x = re.compile('(<.*?>)(.*)')
+            y = x.search(msg.text)
+            z = y.group(2)
+            message += (z if len(z) <= 10 or not truncate else z[0:9] + "...")
+        else:
+            message += (msg.text if len(msg.text) <= 10 or not truncate else
+                        msg.text[0:9] + "...")
+    elif msg.audio:
+        message += "[Audio]"
+    elif msg.document:
+        message += "[Document]"
+    elif msg.animation:
+        message += "[GIF]"
+    elif msg.game:
+        message += "[Game]"
+    elif msg.photo:
+        message += "[Photo]"
+    elif msg.sticker:
+        message += "[Sticker]"
+    elif msg.video:
+        message += "[Video]"
+    elif msg.voice:
+        message += "[Voice Note]"
+    elif msg.video_note:
+        message += "[Video Note]"
+    elif msg.contact:
+        message += "[Contact]"
+    elif msg.location:
+        message += "[Location]"
+    elif msg.venue:
+        message += "[Venue]"
+    else:
+        message += "[Unknown type]"
+
+    # If caption exists
+    if msg.caption:
+        message += (msg.caption if len(msg.caption) <= 10 or not truncate else
+                    msg.caption[0:9] + "...")
+
+    # Set sender
+    return_message = ""
+    for i in message.strip().split("\n"):
+        if msg.from_user.username != bot.username:
+            if msg.from_user.username:
+                return_message += f"@{msg.from_user.username}: "
+            else:
+                return_message += f"{msg.from_user.first_name}: "
+        else:
+            x = re.compile('(<.*?>)(.*)')
+            y = x.search(msg.text)
+            return_message = y.group(1) + ": " + i + "\n"
+        # Set the replied to message's contents
+        if msg.reply_to_message:
+            return_message += ("[Reply to " +
+                               get_text_or_type(msg.reply_to_message, bot,
+                                                True).split("\n")[0] +
+                               f"] ")
+        return_message += i + "\n"
+
+    return return_message.strip()
+
+
 def message_handling(bot, update):
-    print("in here")
     if update.message.chat.id in config.groups:
-        print("in here")
-        for i in update.message.text.split("\n"):
-            params = {"message": (f"TG - {update.message.from_user.first_name}"
-                                  f": {i}")}
-            requests.post("http://localhost:" + str(config.port) + "/post",
-                          data=params)
-            print(params)
+        message = get_text_or_type(update.message, bot)
+        params = {"message": message}
+        requests.post("http://localhost:" + str(config.port) + "/post",
+                      data=params)
 
 
 updater = Updater(token=config.api_key)
 dispatcher = updater.dispatcher
-dispatcher.add_handler(MessageHandler(Filters.text, message_handling))
+dispatcher.add_handler(MessageHandler(Filters.all, message_handling))
 updater.start_polling()