From 1eff41a6f9aa3a0b93425640d0bdaaf4d0148dab Mon Sep 17 00:00:00 2001
From: evazion <noizave@gmail.com>
Date: Sat, 17 Feb 2018 11:02:48 -0600
Subject: [PATCH] downloads: rewrite url in `download!`, not
 `http_get_streaming`.

Refactor Downloads::File#http_get_streaming to just download the given
url, not rewrite it.

Don't clobber @source or @data in `#size` either.
---
 app/logical/downloads/file.rb | 22 ++++++++++------------
 1 file changed, 10 insertions(+), 12 deletions(-)

diff --git a/app/logical/downloads/file.rb b/app/logical/downloads/file.rb
index 9699b0942..28372e8be 100644
--- a/app/logical/downloads/file.rb
+++ b/app/logical/downloads/file.rb
@@ -26,20 +26,23 @@ module Downloads
     end
 
     def size
-      @source, headers, @data = before_download(@source, @data)
+      url, headers, _ = before_download(@source, @data)
       options = { timeout: 3, headers: headers }.deep_merge(Danbooru.config.httparty_options)
-      res = HTTParty.head(@source, options)
+      res = HTTParty.head(url, options)
       res.content_length
     end
 
     def download!
+      url, headers, @data = before_download(@source, @data)
+
       ::File.open(@file_path, "wb") do |out|
-        @source, @data = http_get_streaming(@source, @data) do |response|
+        http_get_streaming(url, headers) do |response|
           out.write(response)
         end
       end
-      @downloaded_source = @source
-      @source = after_download(@source)
+
+      @downloaded_source = url
+      @source = after_download(url)
     end
 
     def before_download(url, datums)
@@ -67,7 +70,7 @@ module Downloads
       end
     end
 
-    def http_get_streaming(src, datums = {}, options = {}, &block)
+    def http_get_streaming(src, headers = {}, options = {}, &block)
       max_size = options[:max_size] || Danbooru.config.max_file_size
       max_size = nil if max_size == 0 # unlimited
       limit = 4
@@ -79,9 +82,6 @@ module Downloads
           raise Error.new("URL must be HTTP or HTTPS")
         end
 
-        src, headers, datums = before_download(src, datums)
-        url = URI.parse(src)
-
         validate_local_hosts(url)
 
         begin
@@ -96,7 +96,7 @@ module Downloads
 
             @content_type = res["Content-Type"]
 
-            return [src, datums]
+            return
           else
             raise Error.new("HTTP error code: #{res.code} #{res.message}")
           end
@@ -109,8 +109,6 @@ module Downloads
           end
         end
       end # while
-
-      [src, datums]
     end # def
 
     def fix_twitter_sources(src)