discourse/app/models/optimized_image.rb

86 lines
2.0 KiB
Ruby

require "digest/sha1"
class OptimizedImage < ActiveRecord::Base
belongs_to :upload
def self.create_for(upload, width, height)
return unless width && height
@image_sorcery_loaded ||= require "image_sorcery"
original_path = "#{Rails.root}/public#{upload.url}"
# create a temp file with the same extension as the original
temp_file = Tempfile.new(["discourse", File.extname(original_path)])
temp_path = temp_file.path
if ImageSorcery.new(original_path).convert(temp_path, resize: "#{width}x#{height}")
thumbnail = OptimizedImage.new({
upload_id: upload.id,
sha1: Digest::SHA1.file(temp_path).hexdigest,
extension: File.extname(temp_path),
width: width,
height: height
})
# make sure the directory exists
FileUtils.mkdir_p Pathname.new(thumbnail.path).dirname
# move the temp file to the right location
File.open(thumbnail.path, "wb") do |f|
f.write temp_file.read
end
end
# close && remove temp file
temp_file.close
temp_file.unlink
thumbnail
end
def destroy
OptimizedImage.transaction do
remove_file
super
end
end
def remove_file
File.delete path
rescue Errno::ENOENT
end
def url
"#{LocalStore.base_url}/#{optimized_path}/#{filename}"
end
def path
"#{LocalStore.base_path}/#{optimized_path}/#{filename}"
end
def optimized_path
"_optimized/#{sha1[0..2]}/#{sha1[3..5]}"
end
def filename
"#{sha1[6..16]}_#{width}x#{height}#{extension}"
end
end
# == Schema Information
#
# Table name: optimized_images
#
# id :integer not null, primary key
# sha1 :string(40) not null
# extension :string(10) not null
# width :integer not null
# height :integer not null
# upload_id :integer not null
#
# Indexes
#
# index_optimized_images_on_upload_id (upload_id)
# index_optimized_images_on_upload_id_and_width_and_height (upload_id,width,height) UNIQUE
#