scinote-web/app/models/tiny_mce_asset.rb

# frozen_string_literal: true

class TinyMceAsset < ApplicationRecord
  attr_accessor :reference
  before_create :set_reference, optional: true
  after_create :update_estimated_size, :self_destruct
  after_destroy :release_team_space
  after_save :update_description

  belongs_to :team, inverse_of: :tiny_mce_assets, optional: true
  belongs_to :step, inverse_of: :tiny_mce_assets, touch: true, optional: true
  belongs_to :result_text,
             inverse_of: :tiny_mce_assets,
             touch: true,
             optional: true

  belongs_to :object, polymorphic: true,
                      optional: true,
                      inverse_of: :tiny_mce_assets
  has_attached_file :image,
                    styles: { large: [Constants::LARGE_PIC_FORMAT, :jpg] },
                    convert_options: { large: '-quality 100 -strip' }

  validates_attachment_content_type :image,
                                    content_type: %r{^image/#{ Regexp.union(
                                      Constants::WHITELISTED_IMAGE_TYPES
                                    ) }}
  validates_attachment :image,
                       presence: true,
                       size: {
                         less_than: Rails.configuration.x\
                                         .file_max_size_mb.megabytes
                       }
  validates :estimated_size, presence: true

  def self.update_images(object, images)
    images = JSON.parse(images)
    current_images = object.tiny_mce_assets.pluck(:id)
    images_to_delete = current_images.reject do |x|
      (images.include? Base62.encode(x))
    end
    images.each do |image|
      image_to_update = find_by_id(Base62.decode(image))
      image_to_update&.update(object: object, saved: true)
    end
    where(id: images_to_delete).destroy_all
  rescue StandardError => e
    Rails.logger.error e.message
  end

  def self.reload_images(images = [])
    images.each do |image|
      old_id = image.class == Array ? image[0] : image
      new_id = image.class == Array ? image[1] : image
      image_to_update = find_by_id(new_id)
      next unless image_to_update

      object_field = data_fields[image_to_update.object_type]
      next unless image_to_update.object

      old_description = Nokogiri::HTML(image_to_update.object[object_field])
      descirption_image = old_description.css(
        "img[data-token=\"#{Base62.encode(old_id)}\"]"
      )
      descirption_image.attr('src').value = image_to_update.url
      descirption_image.attr('data-token').value = Base62.encode(new_id)
      descirption_image[0]['class'] = 'img-responsive'
      new_descirption = old_description.css('body').inner_html.to_s
      image_to_update.object.update(object_field => new_descirption)
    end
  end

  def presigned_url(style = :large,
                    download: false,
                    timeout: Constants::URL_LONG_EXPIRE_TIME)
    if stored_on_s3?
      download_arg = if download
                       'attachment; filename=' + CGI.escape(image_file_name)
                     end

      signer = Aws::S3::Presigner.new(client: S3_BUCKET.client)
      signer.presigned_url(:get_object,
                           bucket: S3_BUCKET.name,
                           key: image.path(style)[1..-1],
                           expires_in: timeout,
                           response_content_disposition: download_arg)
    end
  end

  def stored_on_s3?
    image.options[:storage].to_sym == :s3
  end

  def url(style = :large, timeout: Constants::URL_LONG_EXPIRE_TIME)
    if image.is_stored_on_s3?
      presigned_url(style, timeout: timeout)
    else
      image.url(style)
    end
  end

  def open
    if image.is_stored_on_s3?
      Kernel.open(presigned_url, 'rb')
    else
      File.open(image.path, 'rb')
    end
  end

  def self.delete_unsaved_image(id)
    asset = find_by_id(id)
    asset.destroy if asset && !asset.saved
  end

  def self.data_fields
    {
      'Step' => :description,
      'ResultText' => :text
    }
  end

  private

  def update_description
    TinyMceAsset.reload_images([id]) if object
  end

  def self_destruct
    delay(queue: :assets, run_at: 1.days.from_now).delete_unsaved_image(id)
  end

  def update_estimated_size
    return if image_file_size.blank?

    es = image_file_size * Constants::ASSET_ESTIMATED_SIZE_FACTOR
    update(estimated_size: es)
    Rails.logger.info "Asset #{id}: Estimated size successfully calculated"
    # update team space taken
    team.take_space(es)
    team.save
  end

  def release_team_space
    team.release_space(estimated_size)
    team.save
  end

  def set_reference
    obj_type = "#{@reference.class.to_s.underscore}=".to_sym
    public_send(obj_type, @reference) if @reference
  end
end