tinami/pixa downloads now supported

app/logical/downloads/strategies/base.rb

@@ -0,0 +1,13 @@
+module Downloads
+  module Strategies
+    class Base
+      def self.strategies
+        [Pixiv, Tinami, Pixa]
+      end
+      
+      def rewrite(url, headers)
+        return [url, headers]
+      end
+    end
+  end
+end

app/logical/downloads/strategies/pixa.rb

@@ -0,0 +1,19 @@
+module Downloads
+  module Strategies
+    class Pixa < Base
+      def rewrite(url, headers)
+        if url =~ /https?:\/\/(?:\w+\.)?pixa\.cc/
+          url, headers = rewrite_headers(url, headers)
+        end
+        
+        return [url, headers]
+      end
+    
+    protected
+      def rewrite_headers(url, headers)
+        headers["Referer"] = "http://www.pixa.cc"
+        return [url, headers]
+      end
+    end
+  end
+end

app/logical/downloads/strategies/pixiv.rb

@@ -0,0 +1,58 @@
+module Downloads
+  module Strategies
+    class Pixiv < Base
+      def rewrite(url, headers)
+        if url =~ /https?:\/\/(?:\w+\.)?pixiv\.net/
+          url, headers = rewrite_headers(url, headers)
+          url, headers = rewrite_small_images(url, headers)
+          url, headers = rewrite_small_manga_pages(url, headers)
+        end
+        
+        return [url, headers]
+      end
+    
+    protected
+      def rewrite_headers(url, headers)
+        headers["Referer"] = "http://www.pixiv.net"
+        return [url, headers]
+      end
+      
+      def rewrite_small_images(url, headers)
+        if url =~ %r!(/img/.+?/.+?)_m.+$!
+          match = $1
+          url.sub!(match + "_m", match)
+        end
+        
+        return [url, headers]
+      end
+      
+      def rewrite_small_manga_pages(url, headers)
+        if url =~ %r!(\d+_p\d+)\.!
+          match = $1
+          repl = match.sub(/_p/, "_big_p")
+          big_url = url.sub(match, repl)
+          if http_exists?(big_url)
+            url = big_url
+          end
+        end
+        
+        return [url, headers]
+      end
+
+      def http_exists?(url)
+        # example: http://img01.pixiv.net/img/as-special/15649262_big_p2.jpg
+        exists = false
+        uri = URI.parse(url)
+        Net::HTTP.start(uri.host, uri.port) do |http|
+          headers = {"Referer" => "http://www.pixiv.net", "User-Agent" => "#{Danbooru.config.app_name}/#{Danbooru.config.version}"}
+          http.request_head(uri.request_uri, headers) do |res|
+            if res.is_a?(Net::HTTPSuccess)
+              exists = true
+            end
+          end
+        end
+        exists
+      end
+    end
+  end
+end

app/logical/downloads/strategies/tinami.rb

@@ -0,0 +1,19 @@
+module Downloads
+  module Strategies
+    class Tinami < Base
+      def rewrite(url, headers)
+        if url =~ /https?:\/\/(?:\w+\.)?tinami\.com/
+          url, headers = rewrite_headers(url, headers)
+        end
+        
+        return [url, headers]
+      end
+    
+    protected
+      def rewrite_headers(url, headers)
+        headers["Referer"] = "http://www.tinami.com/view"
+        return [url, headers]
+      end
+    end
+  end
+end