require "digest/sha1" require_dependency "image_sizer" require_dependency "file_helper" require_dependency "validators/upload_validator" class Upload < ActiveRecord::Base belongs_to :user has_many :post_uploads, dependent: :destroy has_many :posts, through: :post_uploads has_many :optimized_images, dependent: :destroy validates_presence_of :filesize validates_presence_of :original_filename validates_with ::Validators::UploadValidator def thumbnail(width = self.width, height = self.height) optimized_images.find_by(width: width, height: height) end def has_thumbnail?(width, height) thumbnail(width, height).present? end def create_thumbnail!(width, height) return unless SiteSetting.create_thumbnails? thumbnail = OptimizedImage.create_for(self, width, height) if thumbnail optimized_images << thumbnail self.width = width self.height = height save! end end def destroy Upload.transaction do Discourse.store.remove_upload(self) super end end def extension File.extname(original_filename) end # options # - content_type # - origin def self.create_for(user_id, file, filename, filesize, options = {}) # compute the sha sha1 = Digest::SHA1.file(file).hexdigest # check if the file has already been uploaded upload = Upload.find_by(sha1: sha1) # delete the previously uploaded file if there's been an error if upload && upload.url.blank? upload.destroy upload = nil end # create the upload unless upload # initialize a new upload upload = Upload.new( user_id: user_id, original_filename: filename, filesize: filesize, sha1: sha1, url: "" ) # trim the origin if any upload.origin = options[:origin][0...1000] if options[:origin] # deal with width & height for images if FileHelper.is_image?(filename) begin # fix orientation first Upload.fix_image_orientation(file.path) # retrieve image info image_info = FastImage.new(file, raise_on_failure: true) # compute image aspect ratio upload.width, upload.height = ImageSizer.resize(*image_info.size) # make sure we're at the beginning of the file (FastImage moves the pointer) file.rewind rescue FastImage::ImageFetchFailure upload.errors.add(:base, I18n.t("upload.images.fetch_failure")) rescue FastImage::UnknownImageType upload.errors.add(:base, I18n.t("upload.images.unknown_image_type")) rescue FastImage::SizeNotFound upload.errors.add(:base, I18n.t("upload.images.size_not_found")) end return upload unless upload.errors.empty? end # create a db record (so we can use the id) return upload unless upload.save # store the file and update its url url = Discourse.store.store_upload(file, upload, options[:content_type]) if url.present? upload.url = url upload.save else upload.errors.add(:url, I18n.t("upload.store_failure", { upload_id: upload.id, user_id: user_id })) end end # return the uploaded file upload end def self.get_from_url(url) return if url.blank? # we store relative urls, so we need to remove any host/cdn url = url.gsub(/^#{Discourse.asset_host}/i, "") if Discourse.asset_host.present? Upload.find_by(url: url) if Discourse.store.has_been_uploaded?(url) end def self.fix_image_orientation(path) `convert #{path} -auto-orient #{path}` 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 # updated_at :datetime # sha1 :string(40) # origin :string(1000) # # Indexes # # index_uploads_on_id_and_url (id,url) # index_uploads_on_sha1 (sha1) UNIQUE # index_uploads_on_url (url) # index_uploads_on_user_id (user_id) #