Implement replying to issues

jsmnbom · jsmnbom · commit e2046a11b8f6 · 2018-10-25T13:16:26.000+02:00
diff --git a/bot/github.py b/bot/github.py
@@ -9,7 +9,7 @@
 from bot.utils import github_cleaner, link, truncate, encode_data_link
 
 TRUNCATED_MESSAGE = '\n<b>[Truncated message, open on GitHub to read more]</b>'
-REPLY_MESSAGE = '\n\n<i>Reply to this message to post a comment on GitHub.</i>'
+REPLY_MESSAGE = '\n\n<i>Reply to this message to post a comment on GitHub (use ! to suppress).</i>'
 
 
 def render_github_markdown(markdown, context: str):
diff --git a/bot/githubapi.py b/bot/githubapi.py
@@ -11,6 +11,8 @@
                        GITHUB_OAUTH_CLIENT_SECRET, GITHUB_OAUTH_REDIRECT_URI)
 from bot.utils import secure_encode_64
 
+GITHUB_API_ACCEPT = {'Accept': 'application/vnd.github.machine-man-preview+json'}
+
 
 class JWTAuth(AuthBase):
     def __init__(self, app_id):
@@ -150,5 +152,14 @@ def markdown(self, markdown, context):
 
         return r.text
 
+    def add_issue_comment(self, repo, issue_number, body, access_token):
+        r = self.post(f'https://api.github.com/repos/{repo}/issues/{issue_number}/comments', json={
+            'body': body
+        }, access_token=access_token)
+
+        r.raise_for_status()
+
+        return r.text
+
 
 github_api = GithubAPI()
diff --git a/bot/main.py b/bot/main.py
@@ -1,16 +1,18 @@
 import http.client
 import logging
 
-from telegram import Update, ParseMode
-from telegram.ext import TypeHandler, CallbackContext, CommandHandler
+from telegram import Update, ParseMode, InlineKeyboardMarkup, InlineKeyboardButton
+from telegram.ext import TypeHandler, CallbackContext, CommandHandler, MessageHandler, Filters
 
 from bot import settings
 from bot.const import TELEGRAM_BOT_TOKEN, DATABASE_FILE
 from bot.github import GithubHandler
+from bot.githubapi import github_api
 from bot.githubupdates import GithubUpdate, GithubAuthUpdate
 from bot.menu import reply_menu
 from bot.persistence import Persistence
 from bot.text import HELP_ADD_REPO
+from bot.utils import decode_first_data_entity, deep_link, reply_data_link_filter
 from bot.webhookupdater import WebhookUpdater
 
 http.client.HTTPConnection.debuglevel = 5
@@ -61,6 +63,37 @@ def test_handler(update: Update, context: CallbackContext):
     pass
 
 
+def reply_handler(update: Update, context: CallbackContext):
+    msg = update.effective_message
+
+    if msg.text[0] == '!':
+        return
+
+    data = decode_first_data_entity(msg.reply_to_message.entities)
+
+    if not data:
+        return
+
+    issue_type, repo, number, author = data
+
+    access_token = context.user_data.get('access_token')
+
+    if not access_token:
+        msg.reply_text(f'Cannot reply to {issue_type}, since you are not logged in. '
+                       f'Press button below to go to a private chat with me and login.\n\n'
+                       f'<i>This message will self destruct in 30 sec.</i>',
+                       reply_markup=InlineKeyboardMarkup([[
+                           InlineKeyboardButton('Login', url=deep_link(context.bot, 'login'))
+                       ]]),
+                       parse_mode=ParseMode.HTML)
+        return
+
+    if issue_type == 'issue':
+        text = f'@{author} {msg.text_markdown}'
+
+        github_api.add_issue_comment(repo, number, text, access_token=access_token)
+
+
 if __name__ == '__main__':
     persistence = Persistence(DATABASE_FILE)
     updater = WebhookUpdater(TELEGRAM_BOT_TOKEN,
@@ -78,6 +111,9 @@ def test_handler(update: Update, context: CallbackContext):
     dp.add_handler(CommandHandler('login', login_handler))
     dp.add_handler(CommandHandler('test', test_handler))
 
+    dp.add_handler(MessageHandler(Filters.reply & reply_data_link_filter, reply_handler,
+                                  channel_post_updates=False, edited_updates=False))
+
     settings.add_handlers(dp)
 
     github_handler = GithubHandler(dp)
diff --git a/bot/utils.py b/bot/utils.py
@@ -12,10 +12,11 @@
 from html5lib.filters.base import Filter
 from html5lib.serializer import HTMLSerializer
 from telegram import MessageEntity
+from telegram.ext import BaseFilter
 
 from bot.const import HMAC_SECRET
 
-URL_BASE = 'https://a.test/'
+URL_BASE = 'https://ghbot.test/'
 
 
 class HMACException(Exception):
@@ -75,6 +76,21 @@ def decode_first_data_entity(entities):
             return decode_data_entity(entity)
 
 
+def deep_link(bot, data):
+    return f'https://telegram.me/{bot.username}?start={data}'
+
+
+class _ReplyDataLinkFilter(BaseFilter):
+    def filter(self, message):
+        for entity in message.reply_to_message.entities:
+            if entity.type == MessageEntity.TEXT_LINK:
+                return entity.url.startswith(URL_BASE)
+            break
+
+
+reply_data_link_filter = _ReplyDataLinkFilter()
+
+
 class _GithubFilter(Filter):
     def __iter__(self):
         in_quote = False
