From 0dd8bd79cbbb6abc8df9e6e3d740e3ddfc79c618 Mon Sep 17 00:00:00 2001
From: Shadowfacts <me@shadowfacts.net>
Date: Thu, 17 Oct 2019 17:39:59 -0400
Subject: [PATCH] End the game when new pieces can't be placed

---
 Tetris Mac/ContentView.swift   |  4 ++++
 Tetris/ContentView.swift       |  4 ++++
 TetrisKit/GameController.swift | 26 ++++++++++++++++----------
 3 files changed, 24 insertions(+), 10 deletions(-)

diff --git a/Tetris Mac/ContentView.swift b/Tetris Mac/ContentView.swift
index 391251f..b151a17 100644
--- a/Tetris Mac/ContentView.swift	
+++ b/Tetris Mac/ContentView.swift	
@@ -60,6 +60,10 @@ struct ContentView: View {
     
     func startTimer() {
         self.timer = Timer.scheduledTimer(withTimeInterval: 0.5, repeats: true, block: { (_) in
+            guard case .playing(_) = self.controller.state else {
+                self.stopTimer()
+                return
+            }
             self.controller.step()
         })
     }
diff --git a/Tetris/ContentView.swift b/Tetris/ContentView.swift
index f122425..496fe1d 100644
--- a/Tetris/ContentView.swift
+++ b/Tetris/ContentView.swift
@@ -123,6 +123,10 @@ struct ContentView: View {
 
     func startTimer() {
         self.timer = Timer.scheduledTimer(withTimeInterval: 0.5, repeats: true) { (_) in
+            guard case .playing(_) = self.controller.state else {
+                self.stopTimer()
+                return
+            }
             self.controller.step()
         }
     }
diff --git a/TetrisKit/GameController.swift b/TetrisKit/GameController.swift
index 00b104f..004f6c6 100644
--- a/TetrisKit/GameController.swift
+++ b/TetrisKit/GameController.swift
@@ -32,10 +32,6 @@ public class GameController: ObservableObject {
     @Published public var score = 0
     var previousPieceWasTetris = false
 
-    public var ended: Bool {
-        return (0..<width).first(where: { board[$0, 0] }) != nil
-    }
-
     public init() {
         self.board = GameBoard(width: width, height: height)
         self.currentPiece = nil
@@ -48,13 +44,20 @@ public class GameController: ObservableObject {
     }
     
     func nextPiece() {
-        let tetromino = nextTetrominoes.removeFirst()
-        currentPiece = GamePiece(tetromino: tetromino, topLeft: ((width - tetromino.shape.count) / 2, 0))
+        let tetromino = nextTetrominoes.first!
+        let nextPiece = GamePiece(tetromino: tetromino, topLeft: ((width - tetromino.shape.count) / 2, 0))
         
-        if currentBag.isEmpty {
-            generateBag()
+        if overlapsAny(nextPiece) {
+            state = .ended
+        } else {
+            nextTetrominoes.removeFirst()
+            if currentBag.isEmpty {
+                generateBag()
+            }
+            nextTetrominoes.append(currentBag.removeFirst())
+            
+            currentPiece = nextPiece
         }
-        nextTetrominoes.append(currentBag.removeFirst())
     }
     
     func generateBag() {
@@ -70,7 +73,9 @@ public class GameController: ObservableObject {
         clearLines()
         nextPiece()
         
-        state = .playing(.normal)
+        if case .playing(.dropped) = state {
+            state = .playing(.normal)
+        }
     }
     
     func clearLines() {
@@ -194,6 +199,7 @@ public class GameController: ObservableObject {
 public enum GameState {
     case waitingForStart
     case playing(PlayState)
+    case ended
 }
 
 public enum PlayState {