cucumber-ruby/lib/cucumber/runtime/step_hooks.rb

# frozen_string_literal: true

module Cucumber
  class Runtime
    class StepHooks
      def initialize(id_generator, hooks, event_bus)
        @hooks = hooks
        @id_generator = id_generator
        @event_bus = event_bus
      end

      def apply(test_steps)
        test_steps.flat_map do |test_step|
          [test_step] + after_step_hooks(test_step)
        end
      end

      private

      def after_step_hooks(test_step)
        @hooks.map do |hook|
          action = ->(*args) { hook.invoke('AfterStep', [args, test_step]) }
          hook_step = Hooks.after_step_hook(@id_generator.new_id, test_step, hook.location, &action)
          @event_bus.hook_test_step_created(hook_step, hook)
          hook_step
        end
      end
    end
  end
end