potential fix for #2328

This commit is contained in:
r888888888
2014-12-17 14:18:25 -08:00
parent 116b4ea178
commit 43ee2875bf
2 changed files with 12 additions and 2 deletions

View File

@@ -3,12 +3,13 @@ module Downloads
class Error < Exception ; end
attr_reader :data, :options
attr_accessor :source, :content_type, :file_path
attr_accessor :source, :original_source, :content_type, :file_path
def initialize(source, file_path, options = {})
# source can potentially get rewritten in the course
# of downloading a file, so check it again
@source = source
@original_source = source
# where to save the download
@file_path = file_path
@@ -41,6 +42,7 @@ module Downloads
def after_download(src)
fix_image_board_sources(src)
fix_twitter_sources(src)
end
def http_get_streaming(src, datums = {}, options = {})
@@ -106,5 +108,13 @@ module Downloads
src
end
end
def fix_twitter_sources(src)
if src =~ %r!^https?://pbs\.twimg\.com/! && original_source =~ %r!^https?://twitter\.com/!
original_source
else
src
end
end
end
end