Merge pull request #16 from Terrain123456/main

ANSI Formatted Github Code-Blocks

Author: AbstractUmbra

modules/github.py

@@ -22,9 +22,11 @@
 """
 from __future__ import annotations
 
+import asyncio
 import re
 from typing import TYPE_CHECKING
 
+import aiohttp
 import discord
 
 import core
@@ -35,6 +37,9 @@
 GITHUB_ISSUE_URL = "https://github.com/{}/issues/{}"
 LIB_ISSUE_REGEX = re.compile(r"(?P<lib>[a-z]+)?##(?P<number>[0-9]+)", flags=re.IGNORECASE)
 
+GITHUB_BASE_URL = "https://github.com/"
+GITHUB_RAW_CONTENT_URL = "https://raw.githubusercontent.com/"
+
 aliases = [
     (("wavelink", "wave", "wl"), "PythonistaGuild/Wavelink"),
     (("discordpy", "discord", "dpy"), "Rapptz/discord.py"),
@@ -46,6 +51,76 @@
 class GitHub(core.Cog):
     def __init__(self, bot: Bot) -> None:
         self.bot: Bot = bot
+        self.code_highlight_emoji = "📃"
+        self.highlight_timeout = 10
+
+    def _strip_content_path(self, url: str) -> str:
+        file_path = url[len(GITHUB_BASE_URL):]
+        return file_path
+
+    async def format_highlight_block(self, url: str, line_adjustment: int = 10):
+        try:
+            highlighted_line = int(url.split("#L")[1])  # seperate the #L{n} highlight
+        except IndexError:
+            return
+
+        file_path = self._strip_content_path(url)
+        raw_url = GITHUB_RAW_CONTENT_URL + file_path.replace("blob/", "")  # Convert it to a raw user content URL
+
+        code = ""
+        async with self.bot.session.get(raw_url) as resp:
+            if resp.status == 404:
+                return
+
+            code += await resp.text()
+
+        code = code.splitlines()
+
+        code_block_dict = {"lines": {}}
+        j = 0
+        for i in code:
+            # populate the dict
+            code_block_dict["lines"][j] = i
+            j += 1
+
+        code_block_dict["lines"][j] = "\n"
+
+        line_list = code_block_dict["lines"]
+
+        if highlighted_line - 1 not in line_list:
+            return None
+
+        bound_adj = line_adjustment  # adjustment for upper and lower bound display
+        _minBoundary = (highlighted_line - 1 - bound_adj)
+        _maxBoundary = (highlighted_line - 1 + bound_adj)
+
+        # loop through all the lines, and adjust the formatting
+        msg = "```ansi\n"
+        key = _minBoundary
+        while key <= _maxBoundary:
+            currLineNum = str(key + 1)
+            # insert a space if there is no following char before the first character...
+            if key + 1 == highlighted_line:
+                highlighted_msg_format = "\u001b[0;37m\u001b[4;31m{}  {}\u001b[0;0m\n".format(
+                    currLineNum, line_list[key]
+                )
+
+                msg += highlighted_msg_format
+            else:
+                display_str = "{}  {}\n" if line_list.get(
+                    key) is not None else ""  # if we hit the end of the file, just write an empty string
+                msg += display_str.format(currLineNum, line_list.get(key))
+            key += 1
+
+        msg += "\n```"
+
+        github_dict = {
+            "path": file_path,
+            "min": _minBoundary if _minBoundary > 0 else highlighted_line,  # Do not display negative numbers if <0
+            "max": _maxBoundary,
+            "msg": msg
+        }
+        return github_dict
 
     @core.Cog.listener()
     async def on_message(self, message: discord.Message) -> None:
@@ -60,6 +135,31 @@ async def on_message(self, message: discord.Message) -> None:
 
             await message.channel.send(GITHUB_ISSUE_URL.format(lib, issue))
 
+        code_segment = await self.format_highlight_block(message.content)
+
+        if code_segment is None:
+            return
+
+        await message.add_reaction(self.code_highlight_emoji)
+
+        path = code_segment['path']
+        _min = code_segment['min']
+        _max = code_segment['max']
+        code_fmt = code_segment['msg']
+
+        def check(reaction, user):
+            return reaction.emoji == self.code_highlight_emoji and user != self.bot.user \
+                and message.id == reaction.message.id
+
+        try:
+            await self.bot.wait_for("reaction_add", check=check, timeout=self.highlight_timeout)
+            await message.channel.send(
+                content="Showing lines `{}` - `{}` in: `{}`...\n{}".format(_min, _max, path, code_fmt),
+                suppress_embeds=True
+            )
+        except asyncio.TimeoutError:
+            return
+
 
 async def setup(bot: Bot) -> None:
     await bot.add_cog(GitHub(bot))