Add safebooru.org support.

Refactor the Gelbooru source extractor to support Safebooru.org as well.

test/functional/uploads_controller_test.rb

@@ -441,6 +441,7 @@ class UploadsControllerTest < ActionDispatch::IntegrationTest
         should_upload_successfully("https://konachan.com/image/ca12cdb79a66d242e95a6f958341bf05/Konachan.com%20-%20270916.png")
 
         should_upload_successfully("https://gelbooru.com/index.php?page=post&s=view&id=7798121")
+        should_upload_successfully("https://safebooru.org/index.php?page=post&s=view&id=4189916")
 
         should_upload_successfully("http://lohas.nicoseiga.jp/o/910aecf08e542285862954017f8a33a8c32a8aec/1433298801/4937663")
         should_upload_successfully("http://seiga.nicovideo.jp/seiga/im4937663")

test/unit/sources/gelbooru_test.rb

@@ -101,5 +101,90 @@ module Sources
       assert_equal("https://gelbooru.com/index.php?page=post&s=list&md5=edd1d2b3881cf70c3acf540780507531", Source::URL.page_url(source2))
       assert_equal("https://gelbooru.com/index.php?page=post&s=list&md5=0b3ae5e225072b8e391c827cb470d29c", Source::URL.page_url(source3))
     end
+
+    context "Safebooru:" do
+      # source: https://i.pximg.net/img-original/img/2021/10/24/09/53/44/93646177_p0.jpg
+      context "A https://safebooru.org/images/$dir/$hash.jpg?$post_id URL without a referer" do
+        strategy_should_work(
+          "https://safebooru.org//images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg?4189916",
+          image_urls: ["https://safebooru.org//images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg?4189916"],
+          artist_name: "チー之介",
+          profile_url: "https://www.pixiv.net/users/57673194",
+          tags: %w[1girl ^^^ animal_ears black_bow black_bowtie black_gloves black_hair black_hairband black_skirt black_wings blue_flower blue_rose bow bowtie brooch center_frills changing_room cheesecake_(artist) collared_shirt commentary dated demon_wings fangs flower flying_sweatdrops frilled_hairband frilled_sleeves frills frown gloves hair_flower hair_ornament hair_over_one_eye hairband halloween halloween_costume high-waist_skirt highres horse_ears horse_girl indoors jack-o'-lantern_ornament jewelry lace-trimmed_gloves lace_trim long_hair looking_at_viewer make_up_in_halloween!_(umamusume) official_alternate_costume open_mouth orange_bow puffy_short_sleeves puffy_sleeves rice_shower_(make_up_vampire!)_(umamusume) rice_shower_(umamusume) rose shirt short_sleeves skirt skirt_bow solo spider_web_print standing star_ornament twitter_username umamusume violet_eyes white_shirt wings rating:q ウマ娘 ライスシャワー ハロウィンイラスト 二次創作 ウマ娘プリティーダービー ライスシャワー(ウマ娘) 更衣室 Make_up_Vampire! ドラキュライス 困り顔],
+          artist_commentary_title: "ハロウィンライス",
+          artist_commentary_desc: "更衣室でハロウィン衣装に着替えたあと「がおーっ！」のポーズを鏡の前で密かに練習してたら、見つかっちゃってあわてるライスシャワーを描きました。",
+          download_size: 771_175,
+        )
+      end
+
+      # source: https://i.pximg.net/img-original/img/2021/10/24/09/53/44/93646177_p0.jpg
+      context "A https://safebooru.org/images/$dir/$hash.jpg URL without a referer" do
+        strategy_should_work(
+          "https://safebooru.org//images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg",
+          image_urls: ["https://safebooru.org//images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg"],
+          artist_name: nil,
+          profile_url: nil,
+          tags: [],
+          artist_commentary_title: nil,
+          artist_commentary_desc: nil,
+        )
+      end
+
+      # source: https://i.pximg.net/img-original/img/2021/10/24/09/53/44/93646177_p0.jpg
+      context "A https://safebooru.org/images/$dir/$hash.jpg URL with a referer" do
+        strategy_should_work(
+          "https://safebooru.org//images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg",
+          referer: "https://safebooru.org/index.php?page=post&s=view&id=4189916",
+          image_urls: ["https://safebooru.org//images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg"],
+          artist_name: "チー之介",
+          profile_url: "https://www.pixiv.net/users/57673194",
+          tags: %w[1girl ^^^ animal_ears black_bow black_bowtie black_gloves black_hair black_hairband black_skirt black_wings blue_flower blue_rose bow bowtie brooch center_frills changing_room cheesecake_(artist) collared_shirt commentary dated demon_wings fangs flower flying_sweatdrops frilled_hairband frilled_sleeves frills frown gloves hair_flower hair_ornament hair_over_one_eye hairband halloween halloween_costume high-waist_skirt highres horse_ears horse_girl indoors jack-o'-lantern_ornament jewelry lace-trimmed_gloves lace_trim long_hair looking_at_viewer make_up_in_halloween!_(umamusume) official_alternate_costume open_mouth orange_bow puffy_short_sleeves puffy_sleeves rice_shower_(make_up_vampire!)_(umamusume) rice_shower_(umamusume) rose shirt short_sleeves skirt skirt_bow solo spider_web_print standing star_ornament twitter_username umamusume violet_eyes white_shirt wings rating:q ウマ娘 ライスシャワー ハロウィンイラスト 二次創作 ウマ娘プリティーダービー ライスシャワー(ウマ娘) 更衣室 Make_up_Vampire! ドラキュライス 困り顔],
+          artist_commentary_title: "ハロウィンライス",
+          artist_commentary_desc: "更衣室でハロウィン衣装に着替えたあと「がおーっ！」のポーズを鏡の前で密かに練習してたら、見つかっちゃってあわてるライスシャワーを描きました。",
+          download_size: 771_175,
+        )
+      end
+
+      # source: https://i.pximg.net/img-original/img/2021/10/24/09/53/44/93646177_p0.jpg
+      context "A https://safebooru.org/index.php?page=post&s=view&id=$post_id URL" do
+        strategy_should_work(
+          "https://safebooru.org/index.php?page=post&s=view&id=4189916",
+          image_urls: ["https://safebooru.org/images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg"],
+          artist_name: "チー之介",
+          profile_url: "https://www.pixiv.net/users/57673194",
+          tags: %w[1girl ^^^ animal_ears black_bow black_bowtie black_gloves black_hair black_hairband black_skirt black_wings blue_flower blue_rose bow bowtie brooch center_frills changing_room cheesecake_(artist) collared_shirt commentary dated demon_wings fangs flower flying_sweatdrops frilled_hairband frilled_sleeves frills frown gloves hair_flower hair_ornament hair_over_one_eye hairband halloween halloween_costume high-waist_skirt highres horse_ears horse_girl indoors jack-o'-lantern_ornament jewelry lace-trimmed_gloves lace_trim long_hair looking_at_viewer make_up_in_halloween!_(umamusume) official_alternate_costume open_mouth orange_bow puffy_short_sleeves puffy_sleeves rice_shower_(make_up_vampire!)_(umamusume) rice_shower_(umamusume) rose shirt short_sleeves skirt skirt_bow solo spider_web_print standing star_ornament twitter_username umamusume violet_eyes white_shirt wings rating:q ウマ娘 ライスシャワー ハロウィンイラスト 二次創作 ウマ娘プリティーダービー ライスシャワー(ウマ娘) 更衣室 Make_up_Vampire! ドラキュライス 困り顔],
+          artist_commentary_title: "ハロウィンライス",
+          artist_commentary_desc: "更衣室でハロウィン衣装に着替えたあと「がおーっ！」のポーズを鏡の前で密かに練習してたら、見つかっちゃってあわてるライスシャワーを描きました。",
+          download_size: 771_175,
+        )
+      end
+
+      # source: https://i.pximg.net/img-original/img/2021/10/24/09/53/44/93646177_p0.jpg
+      context "A https://safebooru.org/index.php?page=post&s=list&md5=$md5 URL" do
+        strategy_should_work(
+          "https://safebooru.org/index.php?page=post&s=list&md5=8ca0f76e014175f11085d64932d980a5",
+          image_urls: ["https://safebooru.org/images/4010/febe33d5f6d46e21c073289bb9884d4e0630761c.jpg"],
+          artist_name: "チー之介",
+          profile_url: "https://www.pixiv.net/users/57673194",
+          tags: %w[1girl ^^^ animal_ears black_bow black_bowtie black_gloves black_hair black_hairband black_skirt black_wings blue_flower blue_rose bow bowtie brooch center_frills changing_room cheesecake_(artist) collared_shirt commentary dated demon_wings fangs flower flying_sweatdrops frilled_hairband frilled_sleeves frills frown gloves hair_flower hair_ornament hair_over_one_eye hairband halloween halloween_costume high-waist_skirt highres horse_ears horse_girl indoors jack-o'-lantern_ornament jewelry lace-trimmed_gloves lace_trim long_hair looking_at_viewer make_up_in_halloween!_(umamusume) official_alternate_costume open_mouth orange_bow puffy_short_sleeves puffy_sleeves rice_shower_(make_up_vampire!)_(umamusume) rice_shower_(umamusume) rose shirt short_sleeves skirt skirt_bow solo spider_web_print standing star_ornament twitter_username umamusume violet_eyes white_shirt wings rating:q ウマ娘 ライスシャワー ハロウィンイラスト 二次創作 ウマ娘プリティーダービー ライスシャワー(ウマ娘) 更衣室 Make_up_Vampire! ドラキュライス 困り顔],
+          artist_commentary_title: "ハロウィンライス",
+          artist_commentary_desc: "更衣室でハロウィン衣装に着替えたあと「がおーっ！」のポーズを鏡の前で密かに練習してたら、見つかっちゃってあわてるライスシャワーを描きました。",
+          download_size: 771_175,
+        )
+      end
+
+      # source: https://i.pximg.net/img-original/img/2021/10/24/09/53/44/93646177_p0.jpg
+      context "A https://safebooru.org/images/$dir/$md5.jpg URL without a referer" do
+        strategy_should_work(
+          "https://safebooru.org//images/4016/64779fbfc87020ed5fd94854fe973bc0.jpeg",
+          image_urls: ["https://safebooru.org//images/4016/64779fbfc87020ed5fd94854fe973bc0.jpeg"],
+          artist_name: nil,
+          profile_url: nil,
+          tags: %w[brown_eyes d4dj dress long_hair pink_hair sword yano_hiiro yorha_no._2_type_b rating:s],
+          artist_commentary_title: nil,
+          artist_commentary_desc: nil,
+        )
+      end
+    end
   end
 end