require 'colored'
require 'yaml'

class Boy
  attr_accessor :name, :yellow, :red
end

def init
  puts "Options".bold
  puts "1. Config"
  print ">"
  option = gets.chomp

  case option
  when '1'
    config
  when 'one'
    config
  else init
  end
end

def config
  puts "Config:".bold
  puts "1. View users"
  puts "2. Create user"
  print ">"
  config_option = gets.chomp

  case config_option
  when '1'
    view_users
  when '2'
    create_user
  else config
  end
end

def view_users
  puts "Users:".bold
  puts "---"
  users = YAML::load( File.open('boys.yaml'))
  user_size = users.size
  num = 0
  user_size.times do
    puts users[num].name
    puts "Yellows: #{users[num].yellow} "
    puts "Reds: #{users[num].red} "
    puts "---"
    num += 1
  end
  num = 0
  config
end

def create_user
  puts "Enter credentials for New User:".bold

  puts "Name:"
  new_name = gets.chomp
  if new_name == ""
    create_user
  end

  puts "Yellows:"
  new_yellow = gets.chomp
  if new_yellow == ""
    create_user
  end
  begin
    new_yellow = new_yellow.to_i
  rescue
    create_user
  end

  puts "Reds:"
  new_red = gets.chomp
  if new_red == ""
    create_user
  end
  begin
    new_red = new_red.to_i
  rescue
    create_user
  end

  puts "Is this okay? [Y/N]"
  puts "Name: #{new_name}"
  puts "Yellows: #{new_yellow}"
  puts "Reds: #{new_red}"
  okay = gets.chomp
  if okay == "Y" or okay == "y" or okay == "yes" or okay == "Yes"
    new_profile = Boy.new
    new_profile.name = new_name
    new_profile.yellow = new_yellow
    new_profile.red = new_red

    File.open("boys.yaml", "a+") {|f| f.write(new_profile.to_yaml.gsub("---", "" + "-")) }

    puts "New user created: #{new_name}"

  else
    create_user
  end

  config

end

puts "system".blue_on_white
init