Start work on supporting images

rss/bot.py

@@ -20,6 +20,7 @@ from string import Template
 import asyncio
 
 import aiohttp
+import requests
 import hashlib
 import feedparser
 
@@ -116,9 +117,31 @@ class RSSBot(Plugin):
             **entry._asdict(),
         }), msgtype=MessageType.NOTICE if sub.send_notice else MessageType.TEXT, allow_html=True)
 
+    async def _send_image(self, feed: Feed, entry: Entry, sub: Subscription) -> Awaitable[EventID]:
+        # Download image from url feed.thumbnail.url
+        data = requests.get(feed.thumbnail.url, stream = True)
+        if data.status_code == 200:
+            data.raw_decode = True
+            try:
+                # Upload image to homeserver
+                mxc = self.client.upload_media(
+                    data, mime_type=feed.thumbnail.mime_type, size=feed.thumbnail.size)
+
+                # Send image
+                # TODO: Maybe set additional metadata
+                # TODO: Set filename
+                await self.client.send_image(sub.room_id, mxc)
+            except MatrixResponseError:
+                pass
+
+    async def _send_with_image(self, feed: Feed, entry: Entry, sub: Subscription) -> Awaitable[EventID]:
+        await self._send(feed, entry, sub)
+        if feed.thumbnail:
+            await self._send_image(feed, entry, sub)
+
     async def _broadcast(self, feed: Feed, entry: Entry, subscriptions: List[Subscription]) -> None:
         spam_sleep = self.config["spam_sleep"]
-        tasks = [self._send(feed, entry, sub) for sub in subscriptions]
+        tasks = [self._send_with_image(feed, entry, sub) for sub in subscriptions]
         for task in tasks:
             try:
                 await task