mastodon_neko/app/workers/post_process_media_worker.rb

# frozen_string_literal: true

class PostProcessMediaWorker
  include Sidekiq::Worker

  sidekiq_options retry: 1, dead: false

  sidekiq_retries_exhausted do |msg|
    media_attachment_id = msg['args'].first

    ActiveRecord::Base.connection_pool.with_connection do
      begin
        media_attachment = MediaAttachment.find(media_attachment_id)
        media_attachment.processing = :failed
        media_attachment.save
      rescue ActiveRecord::RecordNotFound
        true
      end
    end

    Sidekiq.logger.error("Processing media attachment #{media_attachment_id} failed with #{msg['error_message']}")
  end

  def perform(media_attachment_id)
    media_attachment = MediaAttachment.find(media_attachment_id)
    media_attachment.processing = :in_progress
    media_attachment.save
    media_attachment.file.reprocess!(:original)
    media_attachment.processing = :complete
    media_attachment.save
  rescue ActiveRecord::RecordNotFound
    true
  end
end