media_attachment.rb 5.04 KB
Newer Older
1
# frozen_string_literal: true
yhirano's avatar
yhirano committed
2 3 4 5
# == Schema Information
#
# Table name: media_attachments
#
6 7
#  id                :bigint           not null, primary key
#  status_id         :bigint
yhirano's avatar
yhirano committed
8 9 10 11 12
#  file_file_name    :string
#  file_content_type :string
#  file_file_size    :integer
#  file_updated_at   :datetime
#  remote_url        :string           default(""), not null
13
#  account_id        :bigint
yhirano's avatar
yhirano committed
14 15 16 17 18
#  created_at        :datetime         not null
#  updated_at        :datetime         not null
#  shortcode         :string
#  type              :integer          default("image"), not null
#  file_meta         :json
19
#  description       :text
yhirano's avatar
yhirano committed
20
#
21

22 23
require 'mime/types'

24
class MediaAttachment < ApplicationRecord
25 26
  self.inheritance_column = nil

27
  enum type: [:image, :gifv, :video, :unknown]
28

29 30 31
  IMAGE_FILE_EXTENSIONS = ['.jpg', '.jpeg', '.png', '.gif'].freeze
  VIDEO_FILE_EXTENSIONS = ['.webm', '.mp4', '.m4v'].freeze

Eugen Rochko's avatar
Eugen Rochko committed
32
  IMAGE_MIME_TYPES = ['image/jpeg', 'image/png', 'image/gif'].freeze
33
  VIDEO_MIME_TYPES = ['video/webm', 'video/mp4'].freeze
34

35 36 37 38 39 40 41 42 43 44 45 46 47
  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

48 49 50
  belongs_to :account, inverse_of: :media_attachments
  belongs_to :status,  inverse_of: :media_attachments

51
  has_attached_file :file,
52 53
                    styles: ->(f) { file_styles f },
                    processors: ->(f) { file_processors f },
54
                    convert_options: { all: '-quality 90 -strip' }
55 56 57

  include Remotable

58
  validates_attachment_content_type :file, content_type: IMAGE_MIME_TYPES + VIDEO_MIME_TYPES
59
  validates_attachment_size :file, less_than: 8.megabytes
60 61

  validates :account, presence: true
62
  validates :description, length: { maximum: 420 }, if: :local?
63

64
  scope :attached,   -> { where.not(status_id: nil) }
65
  scope :unattached, -> { where(status_id: nil) }
66 67 68
  scope :local,      -> { where(remote_url: '') }
  scope :remote,     -> { where.not(remote_url: '') }

69
  default_scope { order(id: :asc) }
70

71
  def local?
Eugen Rochko's avatar
Eugen Rochko committed
72
    remote_url.blank?
73
  end
74

75 76 77 78
  def needs_redownload?
    file.blank? && remote_url.present?
  end

79 80 81 82
  def to_param
    shortcode
  end

83
  before_create :prepare_description, unless: :local?
84
  before_create :set_shortcode
85
  before_post_process :set_type_and_extension
86
  before_save :set_meta
87

Eugen Rochko's avatar
Eugen Rochko committed
88 89 90 91
  class << self
    private

    def file_styles(f)
92
      if f.instance.file_content_type == 'image/gif'
Eugen Rochko's avatar
Eugen Rochko committed
93
        {
94 95
          small: IMAGE_STYLES[:small],
          original: {
96
            format: 'mp4',
Eugen Rochko's avatar
Eugen Rochko committed
97 98
            convert_options: {
              output: {
99 100 101 102 103
                'movflags' => 'faststart',
                'pix_fmt'  => 'yuv420p',
                'vf'       => 'scale=\'trunc(iw/2)*2:trunc(ih/2)*2\'',
                'vsync'    => 'cfr',
                'b:v'      => '1300K',
104
                'maxrate'  => '500K',
105 106
                'bufsize'  => '1300K',
                'crf'      => 18,
107
              },
Eugen Rochko's avatar
Eugen Rochko committed
108
            },
109
          },
Eugen Rochko's avatar
Eugen Rochko committed
110
        }
111 112 113 114 115 116 117 118 119 120 121
      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
122
        [:video_transcoder]
123 124
      else
        [:thumbnail]
Eugen Rochko's avatar
Eugen Rochko committed
125
      end
126 127
    end
  end
128 129 130 131

  private

  def set_shortcode
132
    self.type = :unknown if file.blank? && !type_changed?
133

134 135 136 137 138 139 140
    return unless local?

    loop do
      self.shortcode = SecureRandom.urlsafe_base64(14)
      break if MediaAttachment.find_by(shortcode: shortcode).nil?
    end
  end
141

142
  def prepare_description
143
    self.description = description.strip[0...420] unless description.nil?
144 145
  end

146
  def set_type_and_extension
147 148 149
    self.type = VIDEO_MIME_TYPES.include?(file_content_type) ? :video : :image
    extension = appropriate_extension
    basename  = Paperclip::Interpolations.basename(file, :original)
150
    file.instance_write :file_name, [basename, extension].delete_if(&:blank?).join('.')
151 152
  end

153 154 155 156 157 158 159 160
  def set_meta
    meta = populate_meta
    return if meta == {}
    file.instance_write :meta, meta
  end

  def populate_meta
    meta = {}
161

162 163 164
    file.queued_for_write.each do |style, file|
      begin
        geo = Paperclip::Geometry.from_file file
165

166 167 168 169 170 171 172 173 174 175
        meta[style] = {
          width: geo.width.to_i,
          height: geo.height.to_i,
          size: "#{geo.width.to_i}x#{geo.height.to_i}",
          aspect: geo.width.to_f / geo.height.to_f,
        }
      rescue Paperclip::Errors::NotIdentifiedByImageMagickError
        meta[style] = {}
      end
    end
176

177 178 179
    meta
  end

180 181 182 183 184 185 186
  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
187
  end
188
end