Add wplink cog

3 files changed, 44 insertions, 0 deletions

diff --git a/wplink/__init__.py b/wplink/__init__.py
new file mode 100644
index 0000000..f82c406
--- /dev/null
+++ b/wplink/__init__.py
@@ -0,0 +1,5 @@
+from .wplink import WPLink
+
+
+async def setup(bot):
+    await bot.add_cog(WPLink(bot))
diff --git a/wplink/info.json b/wplink/info.json
new file mode 100644
index 0000000..bd066ba
--- /dev/null
+++ b/wplink/info.json
@@ -0,0 +1,4 @@
+{
+    "author": ["Arjun Satarkar"],
+    "requirements": ["aiohttp"]
+}
diff --git a/wplink/wplink.py b/wplink/wplink.py
new file mode 100644
index 0000000..e9781fc
--- /dev/null
+++ b/wplink/wplink.py
@@ -0,0 +1,35 @@
+import aiohttp
+import discord
+from redbot.core import commands
+import re
+import urllib.parse
+
+
+class WPLink(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.Cog.listener()
+    async def on_message(self, message: discord.Message):
+        wikilink_pattern = r"\[\[(.+)\]\]"
+        match = re.search(wikilink_pattern, message.content)
+        if match is not None:
+            title = match.group(1)
+            page_url = await self.look_up_page(title)
+            if page_url is not None:
+                await message.reply(
+                    page_url, allowed_mentions=discord.AllowedMentions.none()
+                )
+
+    async def look_up_page(self, title: str) -> str | None:
+        query_url = f"https://en.wikipedia.org/wiki/Special:Search?search={urllib.parse.quote(title)}&go=Go"
+        async with aiohttp.ClientSession() as session:
+            async with session.get(query_url) as response:
+                result_url = str(response.url)
+                return (
+                    result_url
+                    if not result_url.startswith(
+                        "https://en.wikipedia.org/wiki/Special:Search?"
+                    )
+                    else None
+                )