Extensible embed replacement system

src/bot.py

@@ -12,7 +12,7 @@
 import minecraft
 import excuse
 import markov
-import twitter
+import embeds
 import autogen_buildtime
 
 logging.basicConfig(level=logging.INFO)
@@ -52,7 +52,7 @@ async def on_message(message: discord.Message):
     await excuse.on_message(message)
     await markov.on_message(message)
     await faces.on_message(message, client, config)
-    await twitter.on_message(message, config)
+    await embeds.on_message(message, config)
     if await roll.on_message(message):
       return
 

src/config.py.example

@@ -33,4 +33,11 @@ face_dims = [
 # Max total pixels for a face image to have, set to -1 to disable
 face_max_pixels = 2_000_000
 
-enforce_vxtwitter = True
+enforce_embeds = True
+embeds = {
+  "twitter.com": "vxtwitter.com",
+  "tiktok.com": "vxtiktok.com",
+  "www.tiktok.com": "www.vxtiktok.com",
+  "instagram.com": "ddinstagram.com",
+  "www.instagram.com": "www.ddinstagram.com",
+}

src/embeds.py

@@ -0,0 +1,23 @@
+import discord
+
+async def on_message(message: discord.Message, conf):
+  if conf.enforce_embeds:
+    for base, replacement in conf.embeds.items():
+      test = f'https://{base}'
+      test_len = len(test)
+      i = message.content.find(test)
+      if i == -1:
+        test = f'http://{base}'
+        test_len = len(test)
+        i = message.content.find(test)
+      if i != -1:
+        space = message.content.find(" ", i)
+        if space == -1:
+          space = len(message.content)
+        url = f'https://{replacement}{message.content[i + test_len: space]}'
+        if i == 0 and space == len(message.content):
+          await message.channel.send(f'{message.author.mention} (🤡) sent: {url}')
+          await message.delete()
+        else:
+          await message.edit(suppress=True)
+          await message.reply(url)