OCMock to Kiwi

convert.rb

files = [ARGV[0] || Dir["*.m"]].flatten

files.each do |file|
  content = File.read(file)         
  mapped_content = content.lines.map do |line| 
    [
      {from: /\[\[\[(.+) (?:stub|expect)\] andReturn:(.+)\] (.+)/, to: "[[%s stubAndReturn:%s] %s"},
      {from: /\[\[\[(.+) (?:stub|expect)\] andReturn\w+:(.+)\] (.+)/, to: "[[%s stubAndReturn:theValue(%s)] %s"},
      {from: /\[(?:\(\w+\s*\*\)){,1}\[(.+) expect\] (.+)/, to: "[[[%s should] receive] %s"},
      {from: /\[OCMockObject observerMock\]/, to: "[Kiwi observerMock]"},
      {from: /\[OCMockObject mockFor(Class|Protocol):(.+)/, to: "[Kiwi mockFor%s: %s"},
      {from: /\[OCMockObject niceMockFor(Class|Protocol):(.+)/, to: "[Kiwi nullMockFor%s: %s"}
    ].each do |x|
      match = line.match(x[:from]) 
      if match
        line = line.gsub(x[:from], x[:to] % match[1..-1])
       end
    end                    
    line
  end
  File.open(file, "w") { |f| f.print mapped_content.join("") }
end