module Appsignal
  class Hooks
    class ShoryukenMiddleware
      def call(_worker_instance, queue, sqs_msg, body)
        metadata = {
          :queue => queue
        }
        exclude_keys = [:job_class, :queue_name, :arguments]
        metadata.merge!(body.reject { |key| exclude_keys.member?(key.to_sym) })
        metadata.merge!(sqs_msg.attributes)

        options = {
          :class => body["job_class"],
          :method => "perform",
          :metadata => metadata
        }
        options[:params] = body["arguments"] if body.key?("arguments")
        options[:queue_start] = Time.at(sqs_msg.attributes["SentTimestamp"].to_i / 1000) if sqs_msg.attributes.key?("SentTimestamp")

        Appsignal.monitor_transaction("perform_job.shoryuken", options) do
          yield
        end
      end
    end

    class ShoryukenHook < Appsignal::Hooks::Hook
      register :shoryuken

      def dependencies_present?
        defined?(::Shoryuken)
      end

      def install
        ::Shoryuken.configure_server do |config|
          config.server_middleware do |chain|
            chain.add Appsignal::Hooks::ShoryukenMiddleware
          end
        end
      end
    end
  end
end