discourse/app/models/upload.rb

116 lines
3.1 KiB
Ruby
Raw Normal View History

2013-02-05 14:16:51 -05:00
require 'digest/sha1'
2013-06-04 18:34:53 -04:00
require 'image_sizer'
require 's3'
require 'local_store'
2013-06-16 19:00:25 -04:00
require 'tempfile'
require 'pathname'
2013-02-05 14:16:51 -05:00
class Upload < ActiveRecord::Base
belongs_to :user
2013-06-13 17:44:24 -04:00
has_many :post_uploads
has_many :posts, through: :post_uploads
2013-06-16 04:39:48 -04:00
has_many :optimized_images
2013-02-05 14:16:51 -05:00
validates_presence_of :filesize
validates_presence_of :original_filename
2013-06-16 19:00:25 -04:00
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?
thumbnail = OptimizedImage.create_for(self, width, height)
optimized_images << thumbnail if thumbnail
2013-06-16 19:00:25 -04:00
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
2013-06-15 05:52:40 -04:00
# otherwise, create it
if upload.blank?
2013-06-15 05:52:40 -04:00
# 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)
2013-06-15 05:52:40 -04:00
# save the url
upload.save
end
2013-06-15 05:52:40 -04:00
# return the uploaded file
2013-04-07 11:52:46 -04:00
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)
2013-06-04 18:34:53 -04:00
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
2013-02-05 14:16:51 -05:00
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
2013-06-16 20:48:58 -04:00
# index_uploads_on_user_id (user_id)
#
2013-06-16 20:48:58 -04:00