metamaps--metamaps/app/models/concerns/attachable.rb

# frozen_string_literal: true

module Attachable
  extend ActiveSupport::Concern

  included do
    has_many :attachments, as: :attachable, dependent: :destroy
  end

  def images
    attachments.where(file_content_type: image_types)
  end

  def audios
    attachments.where(file_content_type: audio_types)
  end

  def texts
    attachments.where(file_content_type: text_types)
  end

  def pdfs
    attachments.where(file_content_type: pdf_types)
  end

  def documents
    attachments.where(file_content_type: text_types + pdf_types)
  end

  class << self
    def image_types
      ['image/png', 'image/gif', 'image/jpeg']
    end

    def audio_types
      # .ogg files might be labelled as video
      ['audio/ogg', 'video/ogg', 'audio/mpeg', 'audio/wav', 'video/webm']
    end

    def text_types
      ['text/plain']
    end

    def pdf_types
      ['application/pdf']
    end

    def allowed_types
      image_types + audio_types + text_types + pdf_types
    end
  end
end