diff --git a/app/controllers/uploads_controller.rb b/app/controllers/uploads_controller.rb index 94a94a7ff..9c69e7381 100644 --- a/app/controllers/uploads_controller.rb +++ b/app/controllers/uploads_controller.rb @@ -8,6 +8,15 @@ class UploadsController < ApplicationController @upload = Upload.new(:rating => "q") if params[:url] @post = Post.find_by_source(params[:url]) + + @normalized_url = params[:url] + headers = { + "User-Agent" => "#{Danbooru.config.safe_app_name}/#{Danbooru.config.version}" + } + Downloads::Strategies::Base.strategies.each do |strategy| + @normalized_url, headers = strategy.new.rewrite(@normalized_url, headers) + end + begin @source = Sources::Site.new(params[:url]) rescue Exception diff --git a/app/views/uploads/_image.html.erb b/app/views/uploads/_image.html.erb index c9989e77a..d37bb2c20 100644 --- a/app/views/uploads/_image.html.erb +++ b/app/views/uploads/_image.html.erb @@ -1,5 +1,5 @@ <% if params[:url] %> - <%= image_tag(params[:url], :title => "Preview", :id => "image") %> + <%= image_tag(@normalized_url, :title => "Preview", :id => "image") %> <% if params[:ref] %>

Referrer: <%= params[:ref] %>

<% end %>