mastodon-docker-playground/app/models/media_attachment.rb

109 lines
2.6 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
2016-09-05 15:46:36 +00:00
class MediaAttachment < ApplicationRecord
self.inheritance_column = nil
enum type: [:image, :gifv, :video]
2016-09-29 19:28:21 +00:00
IMAGE_MIME_TYPES = ['image/jpeg', 'image/png', 'image/gif'].freeze
2016-10-12 12:29:10 +00:00
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
2016-09-05 15:46:36 +00:00
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
2016-09-05 15:46:36 +00:00
validates :account, presence: true
scope :local, -> { where(remote_url: '') }
2016-11-28 12:49:42 +00:00
default_scope { order('id asc') }
2016-09-05 15:46:36 +00:00
def local?
2016-09-29 19:28:21 +00:00
remote_url.blank?
2016-09-05 15:46:36 +00:00
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
2016-10-23 09:56:04 +00:00
class << self
private
def file_styles(f)
if f.instance.file_content_type == 'image/gif'
2016-10-23 09:56:04 +00:00
{
small: IMAGE_STYLES[:small],
original: {
format: 'mp4',
2016-10-23 09:56:04 +00:00
convert_options: {
output: {
'movflags' => 'faststart',
'pix_fmt' => 'yuv420p',
'vf' => 'scale=\'trunc(iw/2)*2:trunc(ih/2)*2\'',
'vsync' => 'cfr',
'b:v' => '1300K',
2017-03-06 00:50:35 +00:00
'maxrate' => '500K',
'crf' => 6,
},
2016-10-23 09:56:04 +00:00
},
},
2016-10-23 09:56:04 +00:00
}
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]
2016-10-23 09:56:04 +00:00
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
2016-09-05 15:46:36 +00:00
end