# frozen_string_literal: true class MediaAttachment < ApplicationRecord self.inheritance_column = nil enum type: [:image, :gifv, :video, :unknown] 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 :attached, -> { where.not(status_id: nil) } 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_and_extension 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 self.type = :unknown if file.blank? && type.blank? return unless local? loop do self.shortcode = SecureRandom.urlsafe_base64(14) break if MediaAttachment.find_by(shortcode: shortcode).nil? end end def set_type_and_extension self.type = VIDEO_MIME_TYPES.include?(file_content_type) ? :video : :image extension = appropriate_extension basename = Paperclip::Interpolations.basename(file, :original) file.instance_write :file_name, [basename, extension].delete_if(&:empty?).join('.') end def appropriate_extension mime_type = MIME::Types[file.content_type] extensions_for_mime_type = mime_type.empty? ? [] : mime_type.first.extensions original_extension = Paperclip::Interpolations.extension(file, :original) extensions_for_mime_type.include?(original_extension) ? original_extension : extensions_for_mime_type.first end end