module Pro2

# File lib/pro2-runner.rb, line 75
def command_build(filter = nil)
  javajava = parse_project!
  javajava = filter_javajava(javajava, filter) if filter
  javajava.each do |java|
    filename = java.fetch('filename')
    build_cmd = java.fetch('build')

    puts "Build #{filename} ..."
    puts %x(#{build_cmd})
    puts ""
  end
end

# File lib/pro2-runner.rb, line 88
def command_execute(filter = nil)
  javajava = parse_project!
  javajava = filter_javajava(javajava, filter) if filter
  javajava.each do |java|
    filename = java.fetch('filename')
    run_cmd = java.fetch('run')

    puts "Execute #{filename} ..."
    puts %x(#{run_cmd})
    puts ""
  end
end

# File lib/pro2-runner.rb, line 67
def command_init(prefix, num_file)
  project_file = 'project.yaml'

  javajava = initialize_project(prefix, num_file)
  file_write!(project_file, javajava.to_yaml)
  generate_empty_javajava(javajava)
end

# File lib/pro2-runner.rb, line 101
def command_run(filter = nil)
  command_build(filter)
  command_execute(filter)
end

# File lib/pro2-runner.rb, line 61
def file_write!(filename, content)
  raise "#{filename} already exist!" if File.exists? filename
  File.write(filename, content)
end

# File lib/pro2-runner.rb, line 36
def filter_javajava(javajava, filter)
  javajava.select {|java| java.fetch('filename') == filter}
end

# File lib/pro2-runner.rb, line 53
def generate_empty_javajava(javajava)
  javajava.each do |java|
    filename = java.fetch('filename')
    file_write!(filename, generate_java(filename))
  end
end

# File lib/pro2-runner.rb, line 41
def generate_java(filename)
  class_name = File.basename(filename, '.java')

  content =
%(public class #{class_name} {
  public static void main(String[] args) {
    System.out.println("Hello World!");
  }
})
end

# File lib/pro2-runner.rb, line 14
def initialize_project(prefix, num_file)
  javajava = []
  1.upto(num_file) do |i|
    filename = "Prob#{prefix}#{i}"
    javajava.push({
      'filename' => "#{filename}.java",
      'build' => "javac #{filename}.java",
      'run' => "java #{filename}",
    })
  end
  javajava
end

# File lib/pro2-runner.rb, line 28
def parse_project!
  project_file = 'project.yaml'
  raise "#{project_file} not found!! Run `pro2 init`" unless File.exists? project_file
  yaml = File.read(project_file)
  YAML.load yaml
end