# module: source
 
class Source < Thor
  class Clean < Thor
 
    desc "whitespace", "Clean trailing whitespace."
    method_options :extension => :optional
    def whitespace
      Dir.glob("./**/*#{options[:extension]}").each do |file|
        next unless File.file?(file)
        cleaned = []
        File.open(file) do |f|
          f.readlines.inject(cleaned) do |cleaned, line|
            cleaned << if line =~ /^\s+$/ then "\n" else line.sub(/[ ]+$/, '') end
          end
        end
        File.open(file, 'w') {|f| f.write(cleaned) }
      end
    end # def whitespace
 
  end # class Clean
end # class Source