Switch to extension based Cogs

sanic.py

@@ -1,14 +1,22 @@
 from discord.ext import commands
 
-from sanicbot.core.cogs import HelpCog, GitCog
 from sanicbot.core.config import config
 
-bot = commands.Bot(help_command=None, command_prefix=None)
-git_cog = GitCog(bot)
 
+EXTENSIONS = (
+    'sanicbot.extensions.git',
+    'sanicbot.extensions.help',
+)
+
+
+def main():
+    bot = commands.Bot(help_command=None, command_prefix='!')
+
+    for ext in EXTENSIONS:
+        bot.load_extension(ext)
 
-if __name__ == '__main__':
-    bot.command_prefix = '!'
-    bot.add_cog(git_cog)
-    bot.add_cog(HelpCog(bot))
     bot.run(config['SANIC']['token'])
+
+
+if __name__ == '__main__':
+    main()

sanicbot/core/cogs.py → sanicbot/extensions/git.py

@@ -5,13 +5,14 @@
 from discord.ext import commands
 from discord.ext.commands import Context
 
+from sanicbot.core.config import config
 from sanicbot.core.utils import failure_message, success_message
 
 
-class GitCog(commands.Cog):
+class Git(commands.Cog):
     issue_pattern = re.compile(r"#(?P<issue_id>[1,2]\d{3})")
 
-    def __init__(self, bot):
+    def __init__(self, bot: commands.Bot) -> None:
         self.bot = bot
 
     async def lookup(self, ctx: Context, number: int, repo: str):
@@ -44,19 +45,34 @@ async def retrieve_github_issue(
             repo = f"sanic-{repo}"
         await self.lookup(ctx, number, repo)
 
+    @commands.command()
+    async def issue(self, ctx: Context, repo: str, title: str, *, content: str) -> None:
+        repo = repo if repo.startswith("sanic") else "sanic-" + repo
+        url = f"https://api.github.com/repos/sanic-org/{repo}/issues"
+        body = f"Forwarded from discord (Author: {ctx.author}, ID: {ctx.author.id})\n" + content
+        data = {
+            "title": title,
+            "body": body,
+        }
+        headers = {
+            "Accept": "application/vnd.github.v3+json",
+            "Authorization": "token " + config["SANIC"]["git_token"]
+        }
+        async with httpx.AsyncClient() as client:
+            response = await client.post(url, data=data, headers=headers)
+            if response.status_code == 200:
+                await success_message(ctx, "Issue created\n" + response.json()["url"])
+            else:
+                await failure_message(ctx, "An error occured: `" + response.json()["message"] + "`")
+
     @commands.Cog.listener('on_message')
     async def github_issue_message_listener(self, message: Message):
         if not message.author.bot:
             if match := self.issue_pattern.search(message.content):
                 await self.lookup(
                     message.channel, int(match.group("issue_id")), "sanic"
                 )
-            else:
-                await self.bot.process_commands(message)
 
 
-class HelpCog(commands.Cog):
-    @commands.command()
-    async def help(self, ctx):
-        with open("./resources/help.txt") as f:
-            await ctx.send(f.read())
+def setup(bot: commands.Bot):
+    bot.add_cog(Git(bot))

sanicbot/extensions/help.py

@@ -0,0 +1,13 @@
+from discord.ext import commands
+
+
+class Help(commands.Cog):
+    @commands.command()
+    async def help(self, ctx):
+        with open("./resources/help.txt") as f:
+            await ctx.send(f.read())
+
+
+def setup(bot: commands.Bot):
+    bot.add_cog(Help())
+