Finish 3.1 Colored boxes

src/components.rs

@@ -2,6 +2,7 @@ use specs::{Component, NullStorage, VecStorage, World, WorldExt};
 use std::fmt;
 use std::fmt::Display;
 
+#[derive(PartialEq)]
 pub enum BoxColor {
     Red,
     Blue,

src/systems/gameplay_state_sytem.rs

@@ -30,10 +30,17 @@ impl<'a> System<'a> for GameplayStateSystem {
             .collect();
 
         // loop through all box spots and check if there is a corresponding
-        // box at that position
+        // box at that position. since we now have different types of boxes
-        for (_box_spot, position) in (&box_spots, &positions).join() {
+        // we need to make sure the right type of box is on the right
-            if boxes_by_position.contains_key(&(position.x, position.y)) {
+        // type of spot.
+        for (box_spot, position) in (&box_spots, &positions).join() {
+            if let Some(the_box) = boxes_by_position.get(&(position.x, position.y)) {
+                if the_box.color == box_spot.color {
                     // continue
+                } else {
+                    // return, haven't won yet
+                    return;
+                }
             } else {
                 gameplay_state.state = GameplayState::Playing;
                 return;