#!/usr/bin/env ruby

require 'timeout'

def run(no, count)
  Dir.glob("./codes/*#{no}.java").each do |f|
    code = File.read(f)
    code.sub!(/public\s+class\s+\S+/, 'public class Main')

    java_file = f.sub('codes', 'temp')

    File.write(java_file, code)

    count.times do |i|
      in_file = "./generator/#{no}-#{i + 1}.in"
      out_file = "#{f.sub('codes', 'out').sub('.java', '')}-#{i + 1}.out"
      pid = Process.spawn "/usr/local/opt/openjdk/bin/java #{java_file} < #{in_file} > #{out_file}"
      Timeout.timeout(5) do
        Process.wait pid
      end
    rescue Timeout::Error
      puts "Timeout for #{f} when tests #{i + 1} case"
      File.delete out_file
      Process.kill 'KILL', pid
    end
  rescue StandardError
    puts "Error for #{f}"
  end
end

run 'A', 3
run 'B', 3
run 'C', 4