Sha256: a3962baefb00bf66349fc47521d180a546ba890c13c2579d09ae4037df2af368
Contents?: true
Size: 1.91 KB
Versions: 14
Compression:
Stored size: 1.91 KB
Contents
require 'parser/runner'
require 'tempfile'
module Parser
class Runner::RubyRewrite < Runner
def initialize
super
@rewriters = []
end
private
def runner_name
'ruby-rewrite'
end
def setup_option_parsing
super
@slop.on 'l=', 'load=', 'Load a rewriter' do |file|
load_and_discover(file)
end
end
def load_and_discover(file)
load file
const_name = file.
sub(/\.rb$/, '').
gsub(/(^|_)([a-z])/) do |m|
"#{$2.upcase}"
end
@rewriters << Object.const_get(const_name)
end
def process(initial_buffer)
buffer = initial_buffer
@rewriters.each do |rewriter_class|
@parser.reset
ast = @parser.parse(buffer)
rewriter = rewriter_class.new
new_source = rewriter.rewrite(buffer, ast)
new_buffer = Source::Buffer.new(initial_buffer.name +
'|after ' + rewriter_class.name)
new_buffer.source = new_source
@parser.reset
new_ast = @parser.parse(new_buffer)
unless ast == new_ast
$stderr.puts "ASTs do not match:"
old = Tempfile.new('old')
old.write ast.inspect + "\n"; old.flush
new = Tempfile.new('new')
new.write new_ast.inspect + "\n"; new.flush
IO.popen("diff -u #{old.path} #{new.path}") do |io|
diff = io.read.
sub(/^---.*/, "--- #{buffer.name}").
sub(/^\+\+\+.*/, "+++ #{new_buffer.name}")
$stderr.write diff
end
exit 1
end
buffer = new_buffer
end
if File.exist?(buffer.name)
File.open(buffer.name, 'w') do |file|
file.write buffer.source
end
else
if input_size > 1
puts "Rewritten content of #{buffer.name}:"
end
puts buffer.source
end
end
end
end
Version data entries
14 entries across 14 versions & 1 rubygems