require 'digest/sha1' require 'image_sizer' require 's3' require 'local_store' require 'tempfile' require 'pathname' class Upload < ActiveRecord::Base belongs_to :user has_many :post_uploads has_many :posts, through: :post_uploads has_many :optimized_images, dependent: :destroy validates_presence_of :filesize validates_presence_of :original_filename def thumbnail @thumbnail ||= optimized_images.where(width: width, height: height).first end def thumbnail_url thumbnail.url if has_thumbnail? end def has_thumbnail? thumbnail.present? end def create_thumbnail! return unless SiteSetting.create_thumbnails? return if SiteSetting.enable_s3_uploads? return if width < SiteSetting.max_image_width return if has_thumbnail? thumbnail = OptimizedImage.create_for(self, width, height) optimized_images << thumbnail if thumbnail end def destroy Upload.transaction do Upload.remove_file url super end end def self.create_for(user_id, file) # compute the sha sha1 = Digest::SHA1.file(file.tempfile).hexdigest # check if the file has already been uploaded upload = Upload.where(sha1: sha1).first # otherwise, create it if upload.blank? # retrieve image info image_info = FastImage.new(file.tempfile, raise_on_failure: true) # compute image aspect ratio width, height = ImageSizer.resize(*image_info.size) # create a db record (so we can use the id) upload = Upload.create!({ user_id: user_id, original_filename: file.original_filename, filesize: File.size(file.tempfile), sha1: sha1, width: width, height: height, url: "" }) # make sure we're at the beginning of the file (FastImage is moving the pointer) file.rewind # store the file and update its url upload.url = Upload.store_file(file, sha1, image_info, upload.id) # save the url upload.save end # return the uploaded file upload end def self.store_file(file, sha1, image_info, upload_id) return S3.store_file(file, sha1, image_info, upload_id) if SiteSetting.enable_s3_uploads? return LocalStore.store_file(file, sha1, image_info, upload_id) end def self.remove_file(url) S3.remove_file(url) if SiteSetting.enable_s3_uploads? LocalStore.remove_file(url) end def self.has_been_uploaded?(url) is_relative?(url) || is_local?(url) || is_on_s3?(url) end def self.is_relative?(url) (url =~ /^\/[^\/]/) == 0 end def self.is_local?(url) url.start_with?(base_url) end def self.is_on_s3?(url) SiteSetting.enable_s3_uploads? && url.start_with?(S3.base_url) end def self.base_url asset_host.present? ? asset_host : Discourse.base_url_no_prefix end def self.asset_host ActionController::Base.asset_host end end # == Schema Information # # Table name: uploads # # id :integer not null, primary key # user_id :integer not null # original_filename :string(255) not null # filesize :integer not null # width :integer # height :integer # url :string(255) not null # created_at :datetime not null # updated_at :datetime not null # sha1 :string(40) # # Indexes # # index_uploads_on_sha1 (sha1) UNIQUE # index_uploads_on_url (url) # index_uploads_on_user_id (user_id) #