find_step.rb

require 'rubygems'
gem "ruby_parser", "~> 2.0"
require 'ruby_parser'
require 'yaml'

class Step
  attr_reader :file, :line, :regexp
  def initialize(regexp, file, line)
    @file, @line = file, line
    self.regexp = regexp
  end

  def regexp=(value)
    @regexp =
      case value
      when String
        pieces, regexp = [], value.dup
        regexp.gsub!(/\$\w+/) { |match| pieces << match; "TOKEN" }
        regexp = Regexp.escape(regexp)
        regexp.gsub!(/TOKEN/) { |match| "(.*)" }
        Regexp.new("^#{regexp}$")
      when Regexp
        value
      else
        STDERR.puts "Warning: invalid parameter to Given/When/Then on #{file}:#{line}.  Expected Regexp or String, got #{value.class} #{value.inspect}"
        Regexp.new(/^INVALID PARAM$/)
      end
  end

  def match?(text)
    @regexp.match(text)
  end
end

class StepParser
  def self.parser
    @parser ||= RubyParser.new
  end

  attr_accessor :steps, :file

  def initialize(file, keywords)
    @file = file
    @steps = []
    @keywords = keywords
    extract_steps(self.class.parser.parse(File.read(file)))
  end

  def extract_steps(sexp)
    return unless sexp.is_a?(Sexp)
    case sexp.first
    when :block
      sexp[1..-1].each do |child_sexp|
        extract_steps(child_sexp)
      end
    when :iter
      child_sexp = sexp[1]
      return unless child_sexp[0] == :call && @keywords.include?(child_sexp[2])
      regexp = child_sexp[3][1] && child_sexp[3][1][1]
      @steps << Step.new(regexp, file, child_sexp.line)
    else
      sexp.each do |child_sexp|
        extract_steps(child_sexp)
      end
    end
  end
end

i18n_key = ARGV[0]
i18n_file = ARGV[1]
i18n_map = YAML::load(File.open(i18n_file))
i18n = i18n_map[i18n_key]
keywords = (i18n["when"] + i18n["then"] + i18n["given"] + i18n["and"]).gsub("*","").split("|").last(4).map {|k| k.to_sym }

input_text = ARGV[2].strip.gsub(/(#{keywords.join("|")}) */, "")

files = Dir["features/**/*steps.rb"]
steps = []
files.each do |file|
  steps.concat(StepParser.new(file, keywords).steps)
end

steps.each do |step|
  if step.match?(input_text)
    puts "#{step.file}:#{step.line}"
  end
end