rubyfmt

#!/usr/bin/env ruby

require 'filemagic'
require 'pathname'
require 'parser/current'
require 'unparser'

RUBY_TYPE = 'ruby script'.freeze

# TODO: handle stdin

ARGV.flat_map(&Pathname.method(:glob)).select(&:file?).each do |path|
  type = FileMagic.open(FileMagic::MAGIC_NONE) { |fm| fm.file(path.to_path) }
  next unless type.include?(RUBY_TYPE)
  path.write(
    Unparser.unparse(*Parser::CurrentRuby.parse_file_with_comments(path))
  )
end