thedrummeraki/tanoshimu

# frozen_string_literal: true
class TrackableJob < ApplicationJob
  # How to use this:
  # 1) Extend the job to keep track of to 'TrackableJob'
  # 2) Call perform_[now/later](model?) on that job

  around_perform do |job, block|
    event = before_perform(job)
    begin
      block&.call

      after_perform(event)
    rescue => e
      event.mark_as_failed!(e)
      puts e
      puts e.backtrace
      raise e
    end
  end

  private

  def before_perform(job)
    model = model_from_args!(job)
    task = task_name_from_args(job)

    JobEvent.create!(
      job_id: job.job_id,
      user: Users::Admin.system,
      model_id: model&.id,
      used_by_model: model&.class&.table_name,
      job_name: self.class.name,
      started_at: Time.now.utc,
      task: task,
    )
  end

  def after_perform(event)
    event.mark_as_complete!
  end

  def staff_from_args(job)
    fetch_arg_from_args(job, Users::Admin, :staff)
  end

  def model_from_args!(job)
    model = job.arguments.first
    return unless model.is_a?(ApplicationRecord)

    model
  end

  def task_name_from_args(job)
    fetch_arg_from_args(job, String, :task)
  end

  def fetch_arg_from_args(job, klass, type)
    job.arguments.each do |arg|
      return arg if arg.is_a?(klass)

      if arg.is_a?(Hash)
        return arg[type] if arg.keys.include?(type)
      end
    end

    nil
  end
end