Bitbot can do Telegram now, I guess

1 files changed, 59 insertions, 0 deletions

diff --git a/modules/telegram.py b/modules/telegram.py
new file mode 100644
index 00000000..1766f964
--- /dev/null
+++ b/modules/telegram.py
@@ -0,0 +1,59 @@
+import telegram
+from telegram.ext import CommandHandler, MessageHandler, Updater, Filters
+
+import json
+from datetime import datetime
+from threading import Thread
+
+class Module(Thread):
+    _name = "telegram"
+
+    def __init__(self, dolphin):
+        self.dolphin = dolphin
+
+        self.updater = Updater(dolphin.config["telegram-api-key"])
+        self.dispatcher = self.updater.dispatcher
+
+        start_handler = CommandHandler("start", self.start)
+        command_handler = MessageHandler(Filters.command, self.handle)
+        self.dispatcher.add_handler(start_handler)
+        self.dispatcher.add_handler(command_handler)
+
+        self.updater.start_polling()
+        dolphin.events.on("signal").on("interrupt").hook(self.sigint)
+
+    def start(self, bot, update):
+        bot.send_message(chat_id=update.message.chat_id, text="`Dolphin, but Telegram`", parse_mode="Markdown")
+
+    def handle(self, bot, update):
+        message, text = update.message, update.message.text
+        text = text.replace("\r", '').replace("\n", " ")
+        command = text.split(" ")[0][1:]
+        command = command.split("@")[0]
+        args = text.split(" ", 1)[1:][0] if " " in text else ""
+        data = {
+            "chat_id": message.chat_id,
+            "message_id": message.message_id,
+            "line": text,
+            "command": command,
+            "args": args,
+            "args_split": text.split(" ")[1:],
+            "stdout": IOWrapper(bot, message.chat_id, message.message_id),
+            "stderr": IOWrapper(bot, message.chat_id, message.message_id),
+            "external": True,
+            }
+        self.dolphin.events.on("telegram").on("command").on(command).call(**data)
+
+    def sigint(self, event):
+        self.updater.stop()
+
+class IOWrapper:
+    def __init__(self, bot, chat_id, message_id):
+        self.bot = bot
+        self.chat_id = chat_id
+        self.message_id = message_id
+    def write(self, text):
+        if len(text)>4096-10:
+            text = text[:4086] + "…"
+        self.bot.send_message(chat_id=self.chat_id, text="```\n" + text + "\n```",
+            reply_to_message_id=self.message_id, parse_mode="Markdown")