Merge pull request #3116 from evazion/fix-artstation-project-urls

Fix #3108: support https://$artist.artstation.com/projects/$id urls

app/logical/downloads/rewrite_strategies/art_station.rb

@@ -8,7 +8,7 @@ module Downloads
           if test_original(original_url)
             url = original_url
           end
-        elsif url =~ %r!https?://\w+\.artstation\.com/artwork/!
+        else
           url, headers = rewrite_html_url(url, headers)
         end
 
@@ -22,6 +22,8 @@ module Downloads
       end
 
       def rewrite_html_url(url, headers)
+        return [url, headers] unless Sources::Strategies::ArtStation.url_match?(url)
+
         source = Sources::Site.new(url)
         source.get
         [source.image_url, headers]

app/logical/sources/strategies/art_station.rb

@@ -3,11 +3,19 @@ module Sources::Strategies
     attr_reader :json, :image_urls
 
     def self.url_match?(url)
-      url =~ %r!^https?://\w+\.artstation\.com/artwork/[a-z0-9]+!i
+      self.project_id(url).present?
+    end
+
+    def self.project_id(url)
+      if url =~ %r!\Ahttps?://\w+\.artstation\.com/(?:artwork|projects)/(?<project_id>[a-z0-9]+)\z!i
+        $~[:project_id]
+      else
+        nil
+      end
     end
 
     def referer_url
-      if @referer_url =~ %r!^https?://\w+\.artstation\.com/artwork/[a-z0-9]+!i
+      if self.class.url_match?(@referer_url)
         @referer_url
       else
         @url
@@ -18,8 +26,16 @@ module Sources::Strategies
       "ArtStation"
     end
 
+    def project_id
+      self.class.project_id(referer_url)
+    end
+
+    def page_url
+      "https://www.artstation.com/artwork/#{project_id}"
+    end
+
     def api_url
-      url.sub(%r!^https?://\w+\.!, "https://www.").sub(%r!/artwork/!, "/projects/") + ".json"
+      "https://www.artstation.com/projects/#{project_id}.json"
     end
 
     def image_url