discourse/app/models/upload.rb

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

  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 unless width > SiteSetting.auto_link_images_wider_than
    return if has_thumbnail?
    @image_sorcery_loaded ||= require "image_sorcery"
    original_path = "#{Rails.root}/public#{url}"
    temp_file = Tempfile.new(["discourse", File.extname(original_path)])
    if ImageSorcery.new(original_path).convert(temp_file.path, resize: "#{width}x#{height}")
      thumbnail = OptimizedImage.create_for(id, temp_file.path)
      optimized_images << thumbnail
      # 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 if it exists
    temp_file.close
    temp_file.unlink
  end

  def self.create_for(user_id, file)
    # compute the sha
    sha = Digest::SHA1.file(file.tempfile).hexdigest
    # check if the file has already been uploaded
    upload = Upload.where(sha: sha).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),
        sha: sha,
        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, sha, image_info, upload.id)
      # save the url
      upload.save
    end
    # return the uploaded file
    upload
  end

  def self.store_file(file, sha, image_info, upload_id)
    return S3.store_file(file, sha, image_info, upload_id) if SiteSetting.enable_s3_uploads?
    return LocalStore.store_file(file, sha, image_info, upload_id)
  end

  def self.uploaded_regex
    /\/uploads\/#{RailsMultisite::ConnectionManagement.current_db}\/(?<upload_id>\d+)\/[0-9a-f]{16}\.(png|jpg|jpeg|gif|tif|tiff|bmp)/
  end

  def self.has_been_uploaded?(url)
    (url =~ /^\/[^\/]/) == 0 || url.start_with?(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
#
# Indexes
#
#  index_uploads_on_forum_thread_id  (topic_id)
#  index_uploads_on_user_id          (user_id)
#