You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
161 lines
3.6 KiB
161 lines
3.6 KiB
#!/usr/bin/ruby -w
|
|
|
|
# Ruby beautifier, version 2.1, 09/11/2006
|
|
# Copyright (c) 2006, P. Lutus
|
|
# Released under the GPL
|
|
|
|
$tabSize = 2
|
|
$tabStr = " "
|
|
|
|
# indent regexp tests
|
|
|
|
$indentExp = [
|
|
/^module\b/,
|
|
/^if\b/,
|
|
/(=\s*|^)until\b/,
|
|
/(=\s*|^)for\b/,
|
|
/^unless\b/,
|
|
/(=\s*|^)while\b/,
|
|
/(=\s*|^)begin\b/,
|
|
/(^| )case\b/,
|
|
/\bthen\b/,
|
|
/^class\b/,
|
|
/^rescue\b/,
|
|
/^def\b/,
|
|
/\bdo\b/,
|
|
/^else\b/,
|
|
/^elsif\b/,
|
|
/^ensure\b/,
|
|
/\bwhen\b/,
|
|
/\{[^\}]*$/,
|
|
/\[[^\]]*$/
|
|
]
|
|
|
|
# outdent regexp tests
|
|
|
|
$outdentExp = [
|
|
/^rescue\b/,
|
|
/^ensure\b/,
|
|
/^elsif\b/,
|
|
/^end\b/,
|
|
/^else\b/,
|
|
/\bwhen\b/,
|
|
/^[^\{]*\}/,
|
|
/^[^\[]*\]/
|
|
]
|
|
|
|
def makeTab(tab)
|
|
return (tab < 0)?"":$tabStr * $tabSize * tab
|
|
end
|
|
|
|
def addLine(line,tab)
|
|
line.strip!
|
|
line = makeTab(tab)+line if line.length > 0
|
|
return line + "\n"
|
|
end
|
|
|
|
def beautifyRuby(path)
|
|
commentBlock = false
|
|
programEnd = false
|
|
multiLineArray = Array.new
|
|
multiLineStr = ""
|
|
tab = 0
|
|
source = File.read(path)
|
|
dest = ""
|
|
source.split("\n").each do |line|
|
|
if(!programEnd)
|
|
# detect program end mark
|
|
if(line =~ /^__END__$/)
|
|
programEnd = true
|
|
else
|
|
# combine continuing lines
|
|
if(!(line =~ /^\s*#/) && line =~ /[^\\]\\\s*$/)
|
|
multiLineArray.push line
|
|
multiLineStr += line.sub(/^(.*)\\\s*$/,"\\1")
|
|
next
|
|
end
|
|
|
|
# add final line
|
|
if(multiLineStr.length > 0)
|
|
multiLineArray.push line
|
|
multiLineStr += line.sub(/^(.*)\\\s*$/,"\\1")
|
|
end
|
|
|
|
tline = ((multiLineStr.length > 0)?multiLineStr:line).strip
|
|
if(tline =~ /^=begin/)
|
|
commentBlock = true
|
|
end
|
|
end
|
|
end
|
|
if(commentBlock || programEnd)
|
|
# add the line unchanged
|
|
dest += line + "\n"
|
|
else
|
|
commentLine = (tline =~ /^#/)
|
|
if(!commentLine)
|
|
# throw out sequences that will
|
|
# only sow confusion
|
|
while tline.gsub!(/'.*?'/,"")
|
|
end
|
|
while tline.gsub!(/".*?"/,"")
|
|
end
|
|
while tline.gsub!(/\`.*?\`/,"")
|
|
end
|
|
while tline.gsub!(/\{[^\{]*?\}/,"")
|
|
end
|
|
while tline.gsub!(/\([^\(]*?\)/,"")
|
|
end
|
|
while tline.gsub!(/\/.*?\//,"")
|
|
end
|
|
while tline.gsub!(/%r(.).*?\1/,"")
|
|
end
|
|
tline.gsub!(/\\\"/,"'")
|
|
$outdentExp.each do |re|
|
|
if(tline =~ re)
|
|
tab -= 1
|
|
break
|
|
end
|
|
end
|
|
end
|
|
if (multiLineArray.length > 0)
|
|
multiLineArray.each do |ml|
|
|
dest += addLine(ml,tab)
|
|
end
|
|
multiLineArray.clear
|
|
multiLineStr = ""
|
|
else
|
|
dest += addLine(line,tab)
|
|
end
|
|
if(!commentLine)
|
|
$indentExp.each do |re|
|
|
if(tline =~ re && !(tline =~ /\s+end\s*$/))
|
|
tab += 1
|
|
break
|
|
end
|
|
end
|
|
end
|
|
end
|
|
if(tline =~ /^=end/)
|
|
commentBlock = false
|
|
end
|
|
end
|
|
if(source != dest)
|
|
# make a backup copy
|
|
File.open(path + "~","w") { |f| f.write(source) }
|
|
# overwrite the original
|
|
File.open(path,"w") { |f| f.write(dest) }
|
|
end
|
|
if(tab != 0)
|
|
STDERR.puts "#{path}: Indentation error: #{tab}"
|
|
end
|
|
end
|
|
|
|
if(!ARGV[0])
|
|
STDERR.puts "usage: Ruby filenames to beautify."
|
|
exit 0
|
|
end
|
|
|
|
ARGV.each do |path|
|
|
beautifyRuby(path)
|
|
end
|