From 0a5f1b53bab2be02536d5895ec098c8086330ac6 Mon Sep 17 00:00:00 2001
From: Toks <tttoksss@gmail.com>
Date: Sun, 5 Jul 2015 11:02:01 -0400
Subject: [PATCH] #2426 refactor to use predetermined referer

---
 app/controllers/uploads_controller.rb   | 20 ++-------------
 app/logical/image_proxy.rb              | 34 +++++++++++++++++++++++++
 app/logical/sources/strategies/base.rb  |  5 ++++
 app/logical/sources/strategies/pixiv.rb |  6 ++++-
 app/views/uploads/_image.html.erb       |  6 ++++-
 5 files changed, 51 insertions(+), 20 deletions(-)
 create mode 100644 app/logical/image_proxy.rb

diff --git a/app/controllers/uploads_controller.rb b/app/controllers/uploads_controller.rb
index 5449c35e3..6b4c2ef00 100644
--- a/app/controllers/uploads_controller.rb
+++ b/app/controllers/uploads_controller.rb
@@ -34,24 +34,8 @@ class UploadsController < ApplicationController
   end
 
   def image_proxy
-    if params[:url].blank? || params[:ref].blank?
-      raise "Must specify url and referer"
-    end
-
-    url = URI.parse(params[:url])
-    headers = {
-      "Referer" => params[:ref],
-      "User-Agent" => "#{Danbooru.config.safe_app_name}/#{Danbooru.config.version}"
-    }
-
-    Net::HTTP.start(url.host, url.port) do |http|
-      resp = http.request_get(url.request_uri, headers)
-      if resp.is_a?(Net::HTTPSuccess)
-        send_data resp.body, type: resp.content_type, disposition: "inline"
-      else
-        raise "HTTP error code: #{resp.code} #{resp.message}"
-      end
-    end
+    resp = ImageProxy.get_image(params[:url])
+    send_data resp.body, :type => resp.content_type, :disposition => "inline"
   end
 
   def index
diff --git a/app/logical/image_proxy.rb b/app/logical/image_proxy.rb
new file mode 100644
index 000000000..3f0a32c9f
--- /dev/null
+++ b/app/logical/image_proxy.rb
@@ -0,0 +1,34 @@
+class ImageProxy
+  def self.needs_proxy?(url)
+    fake_referer_for(url).present?
+  end
+
+  def self.fake_referer_for(url)
+    Sources::Site.new(url).strategy.fake_referer
+  end
+
+  def self.get_image(url)
+    if url.blank?
+      raise "Must specify url"
+    end
+
+    if !needs_proxy?(url)
+      raise "Proxy not allowed for this site"
+    end
+
+    uri = URI.parse(url)
+    headers = {
+      "Referer" => fake_referer_for(url),
+      "User-Agent" => "#{Danbooru.config.safe_app_name}/#{Danbooru.config.version}"
+    }
+
+    Net::HTTP.start(uri.host, uri.port) do |http|
+      resp = http.request_get(uri.request_uri, headers)
+      if resp.is_a?(Net::HTTPSuccess)
+        return resp
+      else
+        raise "HTTP error code: #{resp.code} #{resp.message}"
+      end
+    end
+  end
+end
\ No newline at end of file
diff --git a/app/logical/sources/strategies/base.rb b/app/logical/sources/strategies/base.rb
index 4d1dcda3a..54a95a5cd 100644
--- a/app/logical/sources/strategies/base.rb
+++ b/app/logical/sources/strategies/base.rb
@@ -74,6 +74,11 @@ module Sources
         [image_url]
       end
 
+      # Should be set to a url for sites that prevent hotlinking, or left nil for sites that don't.
+      def fake_referer
+        nil
+      end
+
     protected
       def agent
         raise NotImplementedError
diff --git a/app/logical/sources/strategies/pixiv.rb b/app/logical/sources/strategies/pixiv.rb
index e5c19c1b6..a39e1b476 100644
--- a/app/logical/sources/strategies/pixiv.rb
+++ b/app/logical/sources/strategies/pixiv.rb
@@ -6,7 +6,7 @@ module Sources
   module Strategies
     class Pixiv < Base
       attr_reader :zip_url, :ugoira_frame_data, :ugoira_content_type
-      
+
       MONIKER   = '(?:[a-zA-Z0-9_-]+)'
       TIMESTAMP = '(?:[0-9]{4}/[0-9]{2}/[0-9]{2}/[0-9]{2}/[0-9]{2}/[0-9]{2})'
       EXT = "(?:jpg|jpeg|png|gif)"
@@ -35,6 +35,10 @@ module Sources
         @pixiv_moniker
       end
 
+      def fake_referer
+        "http://www.pixiv.net"
+      end
+
       def has_artist_commentary?
         @artist_commentary_desc.present?
       end
diff --git a/app/views/uploads/_image.html.erb b/app/views/uploads/_image.html.erb
index aacf2d887..6af397405 100644
--- a/app/views/uploads/_image.html.erb
+++ b/app/views/uploads/_image.html.erb
@@ -1,5 +1,9 @@
 <% if params[:url] %>
-  <%= image_tag(image_proxy_uploads_path(:url => @normalized_url, :ref => params[:ref]), :title => "Preview", :id => "image") %>
+  <% if ImageProxy.needs_proxy?(@normalized_url) %>
+    <%= image_tag(image_proxy_uploads_path(:url => @normalized_url), :title => "Preview", :id => "image") %>
+  <% else %>
+    <%= image_tag(@normalized_url, :title => "Preview", :id => "image") %>
+  <% end %>
 
   <ul>
     <% if @remote_size %>