IntenseWebs/discourse
3
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2025-02-25 18:55:32 -06:00
Code Issues Packages Projects Releases Wiki Activity

properly fix missing uploads by redownloading them

Browse Source
This commit is contained in:
Régis Hanol 2015-05-11 19:07:39 +02:00
parent b6357e4d48
commit 95a57ce3e4
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lib/tasks/uploads.rake
View File

@ -217,7 +217,9 @@ def regenerate_missing_optimized_all_sites
end end
def regenerate_missing_optimized def regenerate_missing_optimized
puts "Regenerating missing optimized images for '#{RailsMultisite::ConnectionManagement.current_db}'..." db = RailsMultisite::ConnectionManagement.current_db
puts "Regenerating missing optimized images for '#{db}'..."
if Discourse.store.external? if Discourse.store.external?
puts "This task only works for internal storages." puts "This task only works for internal storages."
@ -233,30 +235,35 @@ def regenerate_missing_optimized
.order(:id) .order(:id)
.find_each do |optimized_image| .find_each do |optimized_image|
upload = optimized_image.upload
next unless optimized_image.url =~ /^\/uploads\// next unless optimized_image.url =~ /^\/uploads\//
next unless optimized_image.upload.url =~ /^\/uploads\// next unless upload.url =~ /^\/uploads\//
thumbnail = "#{public_directory}#{optimized_image.url}" thumbnail = "#{public_directory}#{optimized_image.url}"
upload = "#{public_directory}#{optimized_image.upload.url}" original = "#{public_directory}#{upload.url}"
if !File.exists?(thumbnail) || File.size(thumbnail) <= 0 if !File.exists?(thumbnail) || File.size(thumbnail) <= 0
# make sure the original image exists locally # make sure the original image exists locally
if (!File.exists?(upload) || File.size(upload) <= 0) && optimized_image.upload.origin.present? if (!File.exists?(original) || File.size(original) <= 0) && upload.origin.present?
# try to fix it by redownloading it # try to fix it by redownloading it
begin begin
downloaded = FileHelper.download(optimized_image.upload.origin, SiteSetting.max_image_size_kb.kilobytes, "discourse-missing", true) rescue nil downloaded = FileHelper.download(upload.origin, SiteSetting.max_image_size_kb.kilobytes, "discourse-missing", true) rescue nil
Discourse.store.store_upload(downloaded, optimized_image.upload) if downloaded && downloaded.size > 0
FileUtils.mkdir_p(File.dirname(original))
File.open(original, "wb") { |f| f.write(downloaded.read) }
end
ensure ensure
downloaded.try(:close!) if downloaded.respond_to?(:close!) downloaded.try(:close!) if downloaded.respond_to?(:close!)
end end
end end
if File.exists?(upload) && File.size(upload) > 0 if File.exists?(original) && File.size(original) > 0
FileUtils.mkdir_p(File.dirname(thumbnail)) FileUtils.mkdir_p(File.dirname(thumbnail))
OptimizedImage.resize(upload, thumbnail, optimized_image.width, optimized_image.height) OptimizedImage.resize(original, thumbnail, optimized_image.width, optimized_image.height)
putc "#" putc "#"
else else
missing_uploads << upload missing_uploads << original
putc "X" putc "X"
end end
else else