Now it works correct with adding the segment

Author: BlindDev

SnakeGame/GameFieldView.swift

@@ -39,7 +39,7 @@ class GameFieldView: UIView {
 
             segmentView = segmentViews[i]
 
-            segmentView.frame = segment.rect()
+            segmentView.frame = segment.rect
             segmentView.backgroundColor = segment.color
         }
     }
@@ -50,7 +50,7 @@ class GameFieldView: UIView {
 
         for border in borders{
 
-            let borderView = UIView(frame: border.rect())
+            let borderView = UIView(frame: border.rect)
 
             borderView.backgroundColor = border.color
             self.addSubview(borderView)

SnakeGame/GameModel.swift

@@ -66,6 +66,34 @@ class GameBrain {
         }
     }
 
+    func setDefaults() {
+        
+        //prepareView
+        borders = []
+        createBorders()
+        
+        //prepare segments
+        snake = []
+        //main default point for the head
+        headPoint = centerPoint()
+        
+        //setting default direction
+        setDirection("Up")
+        
+        //creating the head
+        snake.append(createHead())
+        snake.append(createHead())
+        snake.append(createHead())
+        
+        //create segment to eat
+        food = newSegment()
+    }
+    
+    init(viewSize: CGSize){
+        screen = viewSize
+        setDefaults()
+    }
+    
     private func newSegment() -> SnakeSegment {
         return SnakeSegment(point: randomPoint(), side: side)
     }
@@ -144,49 +172,19 @@ class GameBrain {
     private func createHead() -> SnakeSegment {
         let headSegment = SnakeSegment(point: headPoint, side: side)
         headSegment.isEaten = true
-        headSegment.isHead = true
         return headSegment
     }
 
-    func setDefaults() {
-        
-        //prepareView
-        borders = []
-        createBorders()
-        
-        //prepare segments
-        snake = []
-        //main default point for the head
-        headPoint = centerPoint()
-        
-        //setting default direction
-        setDirection("Up")
-        
-        //creating the head
-        snake.append(createHead())
-        snake.append(createHead())
-        snake.append(createHead())
-
-        
-        //create segment to eat
-        food = newSegment()
-    }
-    
-    init(viewSize: CGSize){
-        screen = viewSize
-        setDefaults()
-    }
-    
     func updateHead(){
 
         headPoint.x += direction.x
         headPoint.y += direction.y
 
         for border in borders {
 
-            if border.rect().contains(headPoint) {
-//                border.isEaten = true
+            if border.rect.contains(headPoint) {
 
+                print("Dead")
                 //add the death
             }
         }
@@ -196,22 +194,10 @@ class GameBrain {
 
     private func moveHead() {
 
-        guard let head = snake.first else{
-            return
-        }
-        
-        head.isHead = false
-        
         snake.insert(createHead(), atIndex: 0)
 
         snake.removeLast()
 
-        guard let tail = snake.last else{
-            return
-        }
-        
-        tail.isTail = true
-        
         checkFood()
     }
 
@@ -220,13 +206,11 @@ class GameBrain {
             return
         }
 
-        let isEaten = CGRectIntersectsRect(head.rect(), food.rect())
+        let isEaten = CGRectIntersectsRect(head.rect, food.rect)
 
         if isEaten {
 
-            food.isHead = true
-            head.isHead = false
-            snake.append(food)
+            snake.insert(food, atIndex: 0)
 
             food = newSegment()
         }

SnakeGame/Snake.swift

@@ -9,68 +9,43 @@
 import UIKit
 
 class GameSegment {
-    var isEaten: Bool!{
-        didSet{
-            isEatenAction()
-        }
-    }
 
     var color: UIColor!
     var point: CGPoint!
     var side: CGFloat!
 
+    var rect: CGRect {
+        get{
+            return CGRect(x: point.x, y: point.y, width: side, height: side)
+        }
+    }
+    
     init(point: CGPoint, side: CGFloat){
 
         self.point = point
         self.side = side
-        self.isEaten = false
-        isEatenAction()
-    }
-    
-    private func isEatenAction(){}
-    
-    func rect() -> CGRect {
-        
-        return CGRect(x: point.x, y: point.y, width: side, height: side)
     }
 }
 
 class SnakeSegment: GameSegment {
-    var isHead: Bool!{
-        didSet{
-            head()
-        }
-    }
 
-    var isTail: Bool!{
+    var isEaten: Bool! {
         didSet{
-            tail()
+            color = isEaten == true ? UIColor.greenColor().colorWithAlphaComponent(0.8) : UIColor.redColor()
         }
     }
 
-    private func head(){
-        color = isHead == true ? UIColor.blueColor() : UIColor.greenColor()
-    }
-    
-    private func tail(){
-        color = isTail == true ? UIColor.grayColor() : UIColor.greenColor()
-    }
-    
     override init(point: CGPoint, side: CGFloat) {
         super.init(point: point, side: side)
-    }
-    
-    override func isEatenAction() {
-        color = isEaten == true ? UIColor.greenColor() : UIColor.redColor()
+        
+        color = UIColor.redColor()
     }
 }
 
 class Border: GameSegment {
     override init(point: CGPoint, side: CGFloat) {
         super.init(point: point, side: side)
-    }
-    
-    private override func isEatenAction() {
-        color = isEaten == true ? UIColor.redColor() : UIColor.blackColor()
+        
+        color = UIColor.blackColor()
     }
 }