d5981754c4
Automatically tag animated_gif and animated_png when a post is edited. Add them back if the user tries to remove them from an animated post, or remove them if the user tries to add them to a non-animated post. Before we added these tags at upload time, but it was possible for users to remove them after upload, or to incorrectly add them to non-animated posts. They were added at upload time because we couldn't afford to open the file and parse the metadata on every tag edit. Now that we save the metadata in the database, we can do this. This also makes it so you can't tag ugoira on non-ugoira files. Known bug: it's possible to have an animated GIF where every frame is identical. Post #3770975 is an example. This will be detected as an animated GIF even though visually it doesn't appear to be animated. Fixes #4041: Animated_gif tag not added to preprocessed uploads
97 lines
3.3 KiB
Ruby
97 lines
3.3 KiB
Ruby
class UploadService
|
|
module Utils
|
|
module_function
|
|
|
|
def distribute_files(file, record, type, original_post_id: nil)
|
|
# need to do this for hybrid storage manager
|
|
post = Post.new
|
|
post.id = original_post_id if original_post_id.present?
|
|
post.md5 = record.md5
|
|
post.file_ext = record.file_ext
|
|
[Danbooru.config.storage_manager, Danbooru.config.backup_storage_manager].each do |sm|
|
|
sm.store_file(file, post, type)
|
|
end
|
|
end
|
|
|
|
def is_downloadable?(source)
|
|
source =~ %r{\Ahttps?://}
|
|
end
|
|
|
|
def generate_resizes(media_file)
|
|
preview_file = media_file.preview(Danbooru.config.small_image_width, Danbooru.config.small_image_width)
|
|
crop_file = media_file.crop(Danbooru.config.small_image_width, Danbooru.config.small_image_width)
|
|
|
|
if media_file.is_ugoira?
|
|
sample_file = media_file.convert
|
|
elsif media_file.is_image? && media_file.width > Danbooru.config.large_image_width
|
|
sample_file = media_file.preview(Danbooru.config.large_image_width, media_file.height)
|
|
else
|
|
sample_file = nil
|
|
end
|
|
|
|
[preview_file, crop_file, sample_file]
|
|
end
|
|
|
|
def process_resizes(upload, file, original_post_id, media_file: nil)
|
|
media_file ||= upload.media_file
|
|
preview_file, crop_file, sample_file = Utils.generate_resizes(media_file)
|
|
|
|
begin
|
|
Utils.distribute_files(file, upload, :original, original_post_id: original_post_id) if file.present?
|
|
Utils.distribute_files(sample_file, upload, :large, original_post_id: original_post_id) if sample_file.present?
|
|
Utils.distribute_files(preview_file, upload, :preview, original_post_id: original_post_id) if preview_file.present?
|
|
Utils.distribute_files(crop_file, upload, :crop, original_post_id: original_post_id) if crop_file.present?
|
|
ensure
|
|
preview_file.try(:close!)
|
|
crop_file.try(:close!)
|
|
sample_file.try(:close!)
|
|
end
|
|
end
|
|
|
|
def process_file(upload, file, original_post_id: nil)
|
|
upload.file = file
|
|
media_file = upload.media_file
|
|
|
|
upload.file_ext = media_file.file_ext.to_s
|
|
upload.file_size = media_file.file_size
|
|
upload.md5 = media_file.md5
|
|
upload.image_width = media_file.width
|
|
upload.image_height = media_file.height
|
|
|
|
upload.validate!(:file)
|
|
upload.tag_string = "#{upload.tag_string} #{Utils.automatic_tags(media_file)}"
|
|
|
|
process_resizes(upload, file, original_post_id)
|
|
|
|
MediaAsset.create!(file: media_file)
|
|
end
|
|
|
|
def automatic_tags(media_file)
|
|
tags = []
|
|
tags << "sound" if media_file.has_audio?
|
|
tags.join(" ")
|
|
end
|
|
|
|
def get_file_for_upload(upload, file: nil)
|
|
return file if file.present?
|
|
raise "No file or source URL provided" if upload.source_url.blank?
|
|
|
|
strategy = Sources::Strategies.find(upload.source_url, upload.referer_url)
|
|
raise NotImplementedError, "No login credentials configured for #{strategy.site_name}." unless strategy.class.enabled?
|
|
|
|
file = strategy.download_file!
|
|
|
|
if strategy.data[:ugoira_frame_data].present?
|
|
upload.context = {
|
|
"ugoira" => {
|
|
"frame_data" => strategy.data[:ugoira_frame_data],
|
|
"content_type" => "image/jpeg"
|
|
}
|
|
}
|
|
end
|
|
|
|
file
|
|
end
|
|
end
|
|
end
|