Permacache attachments when logging them

core/clients.py

@@ -13,6 +13,10 @@
 
 from core.models import InvalidConfigError, getLogger
 
+import asyncio
+import aiohttp
+import os
+
 logger = getLogger(__name__)
 
 
@@ -434,6 +438,8 @@ async def get_user_info(self) -> Optional[dict]:
 
 
 class MongoDBClient(ApiClient):
+    background_tasks = set()
+
     def __init__(self, bot):
         mongo_uri = bot.config["connection_uri"]
         if mongo_uri is None:
@@ -643,6 +649,11 @@ async def edit_message(self, message_id: Union[int, str], new_content: str) -> N
             {"$set": {"messages.$.content": new_content, "messages.$.edited": True}},
         )
 
+    async def warm_permacache(self, url):
+        async with asyncio.timeout(10):
+            async with aiohttp.ClientSession() as session:
+                await session.head(url)
+
     async def append_log(
         self,
         message: Message,
@@ -660,6 +671,12 @@ async def append_log(
         else:
             avatar_url = message.author.display_avatar.url
 
+        if 'PERMACACHE_LOCATION' in os.environ:
+            for a in message.attachments:
+                task = asyncio.create_task(self.warm_permacache(a.url.replace('cdn.discordapp.com/', os.environ['PERMACACHE_LOCATION'])))
+                self.background_tasks.add(task)
+                task.add_done_callback(self.background_tasks.discard)
+
         data = {
             "timestamp": str(message.created_at),
             "message_id": message_id,