From 34299e2cfcabacc36f6deed1b6d83a7fa7cc08ce Mon Sep 17 00:00:00 2001
From: Ceda EI <ceda_ei@protonmail.com>
Date: Wed, 3 Oct 2018 01:58:00 +0530
Subject: [PATCH] Add /screenshot

---
 bot.py        | 24 ++++++++++++++++++++++++
 screenshot.sh |  3 +--
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/bot.py b/bot.py
index 92a8955..b704b8d 100644
--- a/bot.py
+++ b/bot.py
@@ -2,6 +2,8 @@
 
 import logging
 import telegram
+import os
+import subprocess
 from telegram.ext import Updater, CommandHandler
 
 try:
@@ -29,9 +31,31 @@ def start(bot, update):
                      reply_markup=reply_markup)
 
 
+def get_screenshot():
+    try:
+        stdout = subprocess.check_output(["./screenshot.sh"])
+        return stdout.decode('UTF-8')
+    except subprocess.CalledProcessError:
+        return False
+
+
+def screenshot(bot, update):
+    screenshot = get_screenshot()
+    chat_id = update.message.chat_id
+    if not screenshot:
+        bot.send_message(chat_id=chat_id, text="Failed to take a screenshot.")
+        return
+    bot.send_document(chat_id=chat_id, document=open(screenshot, 'rb'))
+    os.remove(screenshot)
+
+
 updater = Updater(token=config.api_key)
 dispatcher = updater.dispatcher
 
 start_handler = CommandHandler('start', start)
 dispatcher.add_handler(start_handler)
+
+screenshot_handler = CommandHandler('screenshot', screenshot)
+dispatcher.add_handler(screenshot_handler)
+
 updater.start_polling()
diff --git a/screenshot.sh b/screenshot.sh
index e0a5674..9e2a691 100755
--- a/screenshot.sh
+++ b/screenshot.sh
@@ -4,7 +4,6 @@ var=$(mktemp /tmp/XXXXXXXX.png)
 scrot -zm $var
 exit_code=$?
 if [[ $exit_code -ne 0 ]]; then
-  echo False
   exit $exit_code
 fi
-echo $var
+echo -n $var