Complete the main loop

Author: odilson-dev

lib/game.rb

@@ -9,7 +9,7 @@ def initialize
         @chessboard = ChessBoard.new
 
     end
-    def play_with_two_humans
+    def human_vs_human
         puts "Player-1 What is your name?"
         name1 = gets.chomp
         player_1 = Player.new(name1, "white")
@@ -26,6 +26,7 @@ def play_with_two_humans
 
 
         current_player = player_1
+        do_break = false
         loop do
             @chessboard.display
             current_player.show_captured_piece
@@ -70,10 +71,12 @@ def play_with_two_humans
                                 the_attacked_piece = @chessboard.data.dig(array_indexes_of_new_positions[0], array_indexes_of_new_positions[1])
 
                                 if the_attacked_piece.class.name == "King"
+                                    do_break = true 
                                     puts "CHECKMATE".light_green
                                     puts "Congratulations #{current_player.name}, you win!!!".light_green
-                                    break 2
+                                    break
                                 end
+            
 
 
                                 # Capture the enemy's piece
@@ -117,12 +120,13 @@ def play_with_two_humans
                 redo
 
             end
+            break if do_break
             current_player == player_1 ? current_player = player_2 : current_player = player_1
         end
     end
 
 
-    def play_between_human_and_ai
+    def human_vs_ai
         puts "Player-1 What is your name?"
         name1 = gets.chomp
         player_1 = Player.new(name1, "white")
@@ -139,6 +143,7 @@ def play_between_human_and_ai
 
 
         current_player = player_1
+        do_break = false
         loop do
             @chessboard.display
             current_player.show_captured_piece
@@ -191,11 +196,12 @@ def play_between_human_and_ai
                                 the_attacked_piece = @chessboard.data.dig(array_indexes_of_new_positions[0], array_indexes_of_new_positions[1])
 
                                 if the_attacked_piece.class.name == "King"
+                                    do_break = true 
                                     puts "CHECKMATE".light_green
                                     puts "Congratulations #{current_player.name}, you win!!!".light_green
-                                    break 2
+                                    break
                                 end
-
+            
 
                                 # Capture the enemy's piece
                                 @chessboard.remove(the_attacked_piece)
@@ -238,12 +244,13 @@ def play_between_human_and_ai
                 redo
 
             end
+            break if do_break
             current_player == player_1 ? current_player = player_2 : current_player = player_1
         end
     end
 
 
-    def play_with_two_ai
+    def ai_vs_ai
 
         player_1 = AI.new("Siri", "white")
 
@@ -257,6 +264,7 @@ def play_with_two_ai
 
         puts "Let's start playing chess right now".light_blue
         current_player = player_1
+        do_break = false
         loop do
             @chessboard.display
             current_player.show_captured_piece
@@ -270,7 +278,6 @@ def play_with_two_ai
 
             @chessboard.display
             current_player.show_captured_piece
-            do_break = false
             loop do
                 new_position = current_player.choose_a_position_to_move(@chessboard.active_piece)
                 sleep(1)

main.rb

@@ -1,2 +1,37 @@
+require_relative 'lib/game.rb'
+loop do
+    puts "Please choose from the following: 1) Human vs Human
+    2) Human vs AI
+    3) AI vs AI
+    4) Load an old game
+    5) View a game from the collection library
+    "
+    answer = gets.chomp
+    game = ChessGame.new
+    if answer == "1"
+        game.human_vs_human
+        
+    elsif answer == "2"
+        game.human_vs_ai
+    elsif answer == "3"
+        game.ai_vs_ai
+    elsif answer == "4"
+        
+    elsif answer == "5"
+        
+    else
+        puts "This option doesn't exist"
+        redo
+    end
+    puts "Do you want to play again[y/n]"
+    play_again = gets.chomp
+    if play_again == "y"
+        redo
+    else
+        puts "Thank you for playing!"
+        puts "Don't hesitate to follow me on GitHub: odilsonjs"
+        break
+    end
+end