artstation: fix fetch source data for image urls.
Fixes 'Fetch source data' when uploading an image url and the referer url is available: * https://danbooru.donmai.us/uploads/new?url=https://cdna.artstation.com/p/assets/images/images/006/029/978/large/amama-l-z.jpg&ref=https://www.artstation.com/artwork/4BWW2
This commit is contained in:
@@ -14,12 +14,12 @@ module Sources
|
|||||||
[Strategies::PixivWhitecube, Strategies::Pixiv, Strategies::NicoSeiga, Strategies::DeviantArt, Strategies::ArtStation, Strategies::Nijie, Strategies::Twitter, Strategies::Tumblr, Strategies::Pawoo]
|
[Strategies::PixivWhitecube, Strategies::Pixiv, Strategies::NicoSeiga, Strategies::DeviantArt, Strategies::ArtStation, Strategies::Nijie, Strategies::Twitter, Strategies::Tumblr, Strategies::Pawoo]
|
||||||
end
|
end
|
||||||
|
|
||||||
def initialize(url, options = {})
|
def initialize(url, referer_url: nil)
|
||||||
@url = url
|
@url = url
|
||||||
|
|
||||||
Site.strategies.each do |strategy|
|
Site.strategies.each do |strategy|
|
||||||
if strategy.url_match?(url)
|
if strategy.url_match?(url) || strategy.url_match?(referer_url)
|
||||||
@strategy = strategy.new(url, options[:referer_url])
|
@strategy = strategy.new(url, referer_url)
|
||||||
break
|
break
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|||||||
@@ -71,6 +71,19 @@ module Sources
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
context "The source site for a http://cdna.artstation.com/p/assets/... url" do
|
||||||
|
setup do
|
||||||
|
@url = "https://cdna.artstation.com/p/assets/images/images/006/029/978/large/amama-l-z.jpg"
|
||||||
|
@ref = "https://www.artstation.com/artwork/4BWW2"
|
||||||
|
@site = Sources::Site.new(@url, referer_url: @ref)
|
||||||
|
@site.get
|
||||||
|
end
|
||||||
|
|
||||||
|
should "fetch the source data" do
|
||||||
|
assert_equal("amama", @site.artist_name)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
context "The source site for an ArtStation gallery" do
|
context "The source site for an ArtStation gallery" do
|
||||||
setup do
|
setup do
|
||||||
@site = Sources::Site.new("https://www.artstation.com/artwork/BDxrA")
|
@site = Sources::Site.new("https://www.artstation.com/artwork/BDxrA")
|
||||||
|
|||||||
Reference in New Issue
Block a user