bug fix in game diagonal logic

sn99 · Oct 12, 2018 · 06d44a0 · 06d44a0
1 parent f466221
commit 06d44a0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -134,6 +134,20 @@ impl MainState {
             check = false;
             self.winner_declare(2);
         }
+        if grid[0][2].option == Some(true)
+            && grid[1][1].option == Some(true)
+            && grid[2][0].option == Some(true)
+            {
+                check = false;
+                self.winner_declare(1);
+            }
+        if grid[0][2].option == Some(false)
+            && grid[1][1].option == Some(false)
+            && grid[2][0].option == Some(false)
+            {
+                check = false;
+                self.winner_declare(2);
+            }
 
         // check for draw
         if check {