class Object

# File lib/main.rb, line 67
def display_saves
  Dir.children('./saves').each_with_index { |name, idx| puts "#{idx}. #{name}" }
end

# File lib/main.rb, line 51
def load_game
  begin
    display_saves
    choice = gets.chomp
    raise StandardError, 'Please choose an option from the above list.' unless choice.to_i.between?(0,
                                                                                                    Dir.children('./saves').size - 1)
  rescue StandardError => e
    puts e.message
    retry
  end

  "./saves/#{Dir.children('./saves')[choice.to_i]}"
end

# File lib/main.rb, line 6
def main
  loop do
    puts <<~MAIN_MENU
      Welcome to RubyChess! Please select one of the following options:

      1. Start a new game of Human vs. Human
      2. Start a new game of Human vs. Computer
      3. Start a new game of Computer vs. Human
      4. Start a new game of Computer vs. Computer
      5. Load a previously saved game
      6. Exit or quit
    MAIN_MENU

    player_input = gets.chomp
    player_input&.downcase

    case player_input&.to_i
    when 1
      game = ChessGame.new('human', 'human')
    when 2
      game = ChessGame.new('human', 'computer')
    when 3
      game = ChessGame.new('computer', 'human')
    when 4
      game = ChessGame.new('computer', 'computer')
    when 5
      puts 'Load saved game'
      game_file = File.new(load_game, 'r')
      game = YAML.safe_load(game_file)
      game_file.close
    else
      puts 'Thank you! Have a nice day.'
      break
    end
    winner = game.play
    puts "#{winner} wins! Play again?" unless winner.nil? || winner == 'Stalemate'
    puts 'Looks like it\'s a stalemate! Play again?' if winner == 'Stalemate'
    next
  end
end

# File spec/chess_game_spec.rb, line 12
def setup_board(loc_to_piece_hash = {})
  board = Array.new(8) { Array.new(8, ChessGame::BLANK_SQUARE) }

  loc_to_piece_hash.each do |loc, piece|
    row, col = loc
    board[row][col] = piece
  end

  board
end