fix: goldenframe handle not static image case

leomotors · Sep 10, 2023 · 60a732b · 60a732b
1 parent b565992
commit 60a732b
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/apps/bot/src/commands/main.slash.ts b/apps/bot/src/commands/main.slash.ts
@@ -81,10 +81,10 @@ export class Main extends CogSlashClass {
 
     let url: string | null;
     if (who) {
-      url = who.avatarURL({ size: 4096 });
+      url = who.avatarURL({ size: 4096, forceStatic: true });
       if (!url) {
         await ctx.followUp(
-          "Cannot G O L D E N F R A M E: Target user has no profile picture!",
+          "Golden Frame Failed: Target user has no profile picture!",
         );
         return;
       }
@@ -93,8 +93,15 @@ export class Main extends CogSlashClass {
     }
 
     const res = await fetch(url);
-    if (!res.body) {
-      await ctx.followUp("Fetch Error: Cannot get Image");
+
+    const contentType = res.headers.get("Content-Type");
+    // Allow only static images
+    const allowedContentTypes = ["image/jpeg", "image/png", "image/webp"];
+
+    if (!allowedContentTypes.includes(contentType!)) {
+      await ctx.followUp(
+        `Fetch Error: Content-Type of ${contentType} is invalid!`,
+      );
       return;
     }