From 21a779455f691235690fa5bb1aba892d09748326 Mon Sep 17 00:00:00 2001
From: evazion <noizave@gmail.com>
Date: Tue, 15 Nov 2022 15:20:30 -0600
Subject: [PATCH] discord: fix /tagme command to work with html page URLs.

---
 app/logical/discord_slash_command/tagme_command.rb | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/app/logical/discord_slash_command/tagme_command.rb b/app/logical/discord_slash_command/tagme_command.rb
index 6883deb7f..2f610ed9c 100644
--- a/app/logical/discord_slash_command/tagme_command.rb
+++ b/app/logical/discord_slash_command/tagme_command.rb
@@ -31,7 +31,10 @@ class DiscordSlashCommand
     end
 
     def tagme(url, confidence, limit: 50, size: 500)
-      response, file = http.download_media(url)
+      extractor = Source::Extractor.find(url)
+      image_url = extractor.image_urls.first
+      file = extractor.download_file!(image_url)
+
       preview = file.preview(size, size)
       tags = autotagger.evaluate(preview, limit: limit, confidence: confidence).to_a
       tags = tags.sort_by { |tag, confidence| [TagCategory.split_header_list.index(tag.category_name.downcase), -confidence] }.to_h
@@ -45,7 +48,7 @@ class DiscordSlashCommand
             icon_url: "https://danbooru.donmai.us/images/danbooru-logo-96x96.png",
           },
           image: {
-            url: url,
+            url: image_url,
           },
         }]
       }