class Cucumber::Formatter::AstLookup::TestStepLookupBuilder

def initialize(gherkin_document)
  @lookup_hash = {}
  process_scenario_container(gherkin_document.feature)
end

def process_scenario_container(container)
  container.children.each do |child|
    if child.respond_to?(:rule) && child.rule
      process_scenario_container(child.rule)
    elsif child.respond_to?(:scenario) && child.scenario
      child.scenario.steps.each do |step|
        @lookup_hash[step.location.line] = StepSource.new(:Step, step)
      end
    elsif !child.background.nil?
      child.background.steps.each do |step|
        @lookup_hash[step.location.line] = StepSource.new(:Step, step)
      end
    end
  end
end