From a0bb1a6c989f1f76d270ee1b3eb842f3532db1fc Mon Sep 17 00:00:00 2001
From: Zevaryx <zevaryx@gmail.com>
Date: Tue, 19 Apr 2022 14:51:36 -0600
Subject: [PATCH] Catch URL errors in resize

---
 jarvis/cogs/image.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/jarvis/cogs/image.py b/jarvis/cogs/image.py
index d15fdd8..bcd57db 100644
--- a/jarvis/cogs/image.py
+++ b/jarvis/cogs/image.py
@@ -94,9 +94,20 @@ class ImageCog(Scale):
             filename = url.split("/")[-1]
 
         data = None
-        async with self._session.get(url) as resp:
-            if resp.status == 200:
-                data = await resp.read()
+        try:
+            async with self._session.get(url) as resp:
+                resp.raise_for_status()
+                if resp.content_type in ["image/jpeg", "image/png"]:
+                    data = await resp.read()
+                else:
+                    await ctx.send(
+                        "Unsupported content type. Please send a URL to a JPEG or PNG",
+                        ephemeral=True,
+                    )
+                    return
+        except Exception:
+            await ctx.send("Failed to retrieve image. Please verify url", ephemeral=True)
+            return
 
         size = len(data)
         if size <= tgt_size: