You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
108 lines
2.6 KiB
108 lines
2.6 KiB
# frozen_string_literal: true |
|
|
|
class MediaAttachment < ApplicationRecord |
|
self.inheritance_column = nil |
|
|
|
enum type: [:image, :gifv, :video] |
|
|
|
IMAGE_MIME_TYPES = ['image/jpeg', 'image/png', 'image/gif'].freeze |
|
VIDEO_MIME_TYPES = ['video/webm', 'video/mp4'].freeze |
|
|
|
IMAGE_STYLES = { original: '1280x1280>', small: '400x400>' }.freeze |
|
VIDEO_STYLES = { |
|
small: { |
|
convert_options: { |
|
output: { |
|
vf: 'scale=\'min(400\, iw):min(400\, ih)\':force_original_aspect_ratio=decrease', |
|
}, |
|
}, |
|
format: 'png', |
|
time: 0, |
|
}, |
|
}.freeze |
|
|
|
belongs_to :account, inverse_of: :media_attachments |
|
belongs_to :status, inverse_of: :media_attachments |
|
|
|
has_attached_file :file, |
|
styles: ->(f) { file_styles f }, |
|
processors: ->(f) { file_processors f }, |
|
convert_options: { all: '-quality 90 -strip' } |
|
validates_attachment_content_type :file, content_type: IMAGE_MIME_TYPES + VIDEO_MIME_TYPES |
|
validates_attachment_size :file, less_than: 8.megabytes |
|
|
|
validates :account, presence: true |
|
|
|
scope :local, -> { where(remote_url: '') } |
|
default_scope { order('id asc') } |
|
|
|
def local? |
|
remote_url.blank? |
|
end |
|
|
|
def file_remote_url=(url) |
|
self.file = URI.parse(url) |
|
end |
|
|
|
def to_param |
|
shortcode |
|
end |
|
|
|
before_create :set_shortcode |
|
before_post_process :set_type |
|
|
|
class << self |
|
private |
|
|
|
def file_styles(f) |
|
if f.instance.file_content_type == 'image/gif' |
|
{ |
|
small: IMAGE_STYLES[:small], |
|
original: { |
|
format: 'mp4', |
|
convert_options: { |
|
output: { |
|
'movflags' => 'faststart', |
|
'pix_fmt' => 'yuv420p', |
|
'vf' => 'scale=\'trunc(iw/2)*2:trunc(ih/2)*2\'', |
|
'vsync' => 'cfr', |
|
'b:v' => '1300K', |
|
'maxrate' => '500K', |
|
'crf' => 6, |
|
}, |
|
}, |
|
}, |
|
} |
|
elsif IMAGE_MIME_TYPES.include? f.instance.file_content_type |
|
IMAGE_STYLES |
|
else |
|
VIDEO_STYLES |
|
end |
|
end |
|
|
|
def file_processors(f) |
|
if f.file_content_type == 'image/gif' |
|
[:gif_transcoder] |
|
elsif VIDEO_MIME_TYPES.include? f.file_content_type |
|
[:video_transcoder] |
|
else |
|
[:thumbnail] |
|
end |
|
end |
|
end |
|
|
|
private |
|
|
|
def set_shortcode |
|
return unless local? |
|
|
|
loop do |
|
self.shortcode = SecureRandom.urlsafe_base64(14) |
|
break if MediaAttachment.find_by(shortcode: shortcode).nil? |
|
end |
|
end |
|
|
|
def set_type |
|
self.type = VIDEO_MIME_TYPES.include?(file_content_type) ? :video : :image |
|
end |
|
end
|
|
|