require "yaml"
require 'erb'
 
problem = YAML::load_file(ARGV[0] + "/desc.yml")
output = ERB.new(IO.read('template.rt'))
puts output.result()

if(ARGV[1] == 'table') 
	require 'terminal-table'
	files = Dir[ARGV[0]+"/**/*/desc.yml"]
	files.sort
	entry = []
	files.each do |file| 
		yaml = YAML::load_file(file)
		nested = file.split('/')
		count = 0
		if( yaml['solution'] and yaml['solutions']) 
			yaml['solutions'].each do |sol|
				count += 1
			end
		end
		entry << {'title'=>yaml['title'],'code'=>nested[-2].to_s,'solved'=>count}
	end

	rows = []
	entry.each do |e|
		row = [e['code'],e['title'],e['solved']]
		rows << row
	end
	table = Terminal::Table.new(:headings => ['Code','Title','Solved'],:rows => rows)
	puts table
end