sources: factor out Source::URL::ArtStation.

app/logical/source/url.rb

@@ -19,6 +19,8 @@ module Source
   class URL < Danbooru::URL
     SUBCLASSES = [
       Source::URL::Twitter,
+      Source::URL::ArtStation,
+      Source::URL::Foundation,
       Source::URL::HentaiFoundry,
       Source::URL::Lofter,
       Source::URL::Mastodon,
@@ -27,7 +29,6 @@ module Source
       Source::URL::Plurk,
       Source::URL::Skeb,
       Source::URL::TwitPic,
-      Source::URL::Foundation,
     ]
 
     # Parse a URL into a subclass of Source::URL, or raise an exception if the URL is not a valid HTTP or HTTPS URL.

app/logical/source/url/art_station.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+# Page URLs:
+#
+# * https://www.artstation.com/artwork/04XA4
+# * https://www.artstation.com/artwork/cody-from-sf (old; redirects to https://www.artstation.com/artwork/3JJA)
+# * https://sa-dui.artstation.com/projects/DVERn
+# * https://dudeunderscore.artstation.com/projects/NoNmD?album_id=23041
+#
+# Profile URLs:
+#
+# * https://www.artstation.com/artist/sa-dui
+# * https://www.artstation.com/sa-dui
+# * https://sa-dui.artstation.com/
+# * https://hosi_na.artstation.com
+#
+# Image URLs
+#
+# * https://cdna.artstation.com/p/assets/images/images/005/804/224/large/titapa-khemakavat-sa-dui-srevere.jpg?1493887236
+# * https://cdnb.artstation.com/p/assets/images/images/014/410/217/smaller_square/bart-osz-bartosz1812041.jpg?1543866276
+# * https://cdna.artstation.com/p/assets/images/images/007/253/680/4k/ina-wong-demon-girl-done-ttd-comp.jpg?1504793833
+# * https://cdna.artstation.com/p/assets/covers/images/007/262/828/small/monica-kyrie-1.jpg?1504865060
+#
+# API URLs
+#
+# * https://www.artstation.com/projects/04XA4.json
+
+class Source::URL::ArtStation < Source::URL
+  RESERVED_SUBDOMAINS = %w[www cdn cdna cdnb]
+  IMAGE_SUBDOMAINS = %w[cdn cdna cdnb]
+
+  attr_reader :username, :work_id
+
+  def self.match?(url)
+    url.domain == "artstation.com"
+  end
+
+  def site_name
+    "ArtStation"
+  end
+
+  def parse
+    case [host, *path_segments]
+
+    # https://cdna.artstation.com/p/assets/images/images/005/804/224/large/titapa-khemakavat-sa-dui-srevere.jpg?1493887236
+    # https://cdnb.artstation.com/p/assets/images/images/014/410/217/smaller_square/bart-osz-bartosz1812041.jpg?1543866276
+    # https://cdna.artstation.com/p/assets/images/images/007/253/680/4k/ina-wong-demon-girl-done-ttd-comp.jpg?1504793833
+    # https://cdna.artstation.com/p/assets/covers/images/007/262/828/small/monica-kyrie-1.jpg?1504865060
+    in _, "p", "assets", ("images" | "covers") => asset_type, "images", *subdirs, size, filename
+      @asset_type = asset_type
+      @asset_subdir = subdirs.join("/")
+      @filename = filename
+      @timestamp = query if query&.match?(/^\d+$/)
+
+    # https://www.artstation.com/artwork/04XA4
+    # https://www.artstation.com/artwork/cody-from-sf (old; redirects to https://www.artstation.com/artwork/3JJA)
+    # https://sa-dui.artstation.com/projects/DVERn
+    # https://dudeunderscore.artstation.com/projects/NoNmD?album_id=23041
+    in _, ("artwork" | "projects"), work_id
+      @work_id = work_id
+      @username = subdomain unless subdomain.in?(RESERVED_SUBDOMAINS)
+
+    # https://www.artstation.com/artist/sa-dui
+    in "www.artstation.com", "artist", username
+      @username = username
+
+    # https://www.artstation.com/sa-dui
+    in "www.artstation.com", username
+      @username = username
+
+    # https://sa-dui.artstation.com
+    # https://hosi_na.artstation.com
+    in *rest unless subdomain.in?(RESERVED_SUBDOMAINS)
+      @username = subdomain
+
+    else
+    end
+  end
+
+  def image_url?
+    @filename.present?
+  end
+
+  def full_image_url(size = "original")
+    return nil unless image_url?
+
+    if @timestamp.present?
+      "https://cdn.artstation.com/p/assets/#{@asset_type}/images/#{@asset_subdir}/#{size}/#{@filename}?#{@timestamp}"
+    else
+      "https://cdn.artstation.com/p/assets/#{@asset_type}/images/#{@asset_subdir}/#{size}/#{@filename}"
+    end
+  end
+end