class Rufo::Command

def self.parse_options(argv)
  want_check = false
  filename_for_dot_rufo = nil
  OptionParser.new do |opts|
    opts.version = Rufo::VERSION
    opts.banner = "Usage: rufo files or dirs [options]"
    opts.on("-c", "--check", "Only check formating changes") do
      want_check = true
    end
    opts.on("--filename=value", "Filename to use to lookup .rufo (useful for STDIN formatting)") do |value|
      filename_for_dot_rufo = value
    end
    opts.on("-h", "--help", "Show this help") do
      puts opts
      exit
    end
  end.parse!(argv)
  [want_check, filename_for_dot_rufo]
end

def self.run(argv)
  want_check, filename_for_dot_rufo = parse_options(argv)
  new(want_check, filename_for_dot_rufo).run(argv)
end

def format(code, dir)
  formatter = Rufo::Formatter.new(code)
  dot_rufo = @dot_file.find_in(dir)
  if dot_rufo
    begin
      formatter.instance_eval(dot_rufo)
    rescue => ex
      STDERR.puts "Error evaluating #{dot_rufo}"
      raise ex
    end
  end
  formatter.format
  formatter.result
end

def format_args(args)
  files = []
  args.each do |arg|
    if Dir.exist?(arg)
      files.concat Dir[File.join(arg, '**', '*.rb')].select(&File.method(:file?))
    elsif File.exist?(arg)
      files << arg
    else
      STDERR.puts "Error: file or directory not found: #{arg}"
    end
  end
  changed = false
  files.each do |file|
    changed |= format_file file
  end
  exit 1 if changed
end

def format_file(filename)
  code = File.read(filename)
  begin
    result = format(code, File.dirname(filename))
  rescue Rufo::SyntaxError
    # We ignore syntax errors as these might be template files
    # with .rb extension
    STDERR.puts "Error: #{filename} has syntax errors"
    return true
  end
  if code.force_encoding(result.encoding) != result
    if @want_check
      STDERR.puts "Error: formatting #{filename} produced changes"
    else
      File.write(filename, result)
      puts "Format: #{filename}"
    end
    return true
  end
  false
rescue Rufo::SyntaxError
  STDERR.puts "Error: the given text in #{filename} is not a valid ruby program (it has syntax errors)"
  exit 1
rescue => ex
  STDERR.puts "You've found a bug!"
  STDERR.puts "It happened while trying to format the file #{filename}"
  STDERR.puts "Please report it to https://github.com/asterite/rufo/issues with code that triggers it"
  STDERR.puts
  raise ex
end

def format_stdin
  code = STDIN.read
  result = format(code, @filename_for_dot_rufo || Dir.getwd)
  if @want_check
    exit 1 if result != code
  else
    print result
  end
rescue Rufo::SyntaxError
  STDERR.puts "Error: the given text is not a valid ruby program (it has syntax errors)"
  exit 1
rescue => ex
  STDERR.puts "You've found a bug!"
  STDERR.puts "Please report it to https://github.com/asterite/rufo/issues with code that triggers it"
  STDERR.puts
  raise ex
end

def initialize(want_check, filename_for_dot_rufo)
  @want_check = want_check
  @filename_for_dot_rufo = filename_for_dot_rufo
  @dot_file = Rufo::DotFile.new
end

def run(argv)
  if argv.empty?
    format_stdin
  else
    format_args argv
  end
end