From 676dbf0a96f92ec48e8cc0175c20396c9bd7a921 Mon Sep 17 00:00:00 2001
From: Timothy Warren <twarren@nabancard.com>
Date: Thu, 5 Nov 2020 11:09:57 -0500
Subject: [PATCH] Add the tail

---
 src/main.rs | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 491723d..f1801f3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -43,10 +43,11 @@ impl Direction {
         }
     }
 }
+struct SegmentMaterial(Handle<ColorMaterial>);
 
 #[derive(Default)]
 struct SnakeSegment {
-    next_segment: Option<Entity>
+    next_segment: Option<Entity>,
 }
 
 struct SnakeHead {
@@ -56,6 +57,17 @@ struct SnakeHead {
 struct SnakeMoveTimer(Timer);
 struct HeadMaterial(Handle<ColorMaterial>);
 
+fn spawn_segment(commands: &mut Commands, material: Handle<ColorMaterial>, position: Position) {
+    commands
+        .spawn(SpriteComponents {
+            material,
+            ..Default::default()
+        })
+        .with(SnakeSegment { next_segment: None })
+        .with(position)
+        .with(Size::square(0.65));
+}
+
 fn snake_movement(
     time: Res<Time>,
     keyboard_input: Res<Input<KeyCode>>,
@@ -128,16 +140,27 @@ fn setup(mut commands: Commands, mut materials: ResMut<Assets<ColorMaterial>>) {
     commands.insert_resource(HeadMaterial(
         materials.add(Color::rgb(0.7, 0.7, 0.7).into()),
     ));
+    commands.insert_resource(SegmentMaterial(
+        materials.add(Color::rgb(0.3, 0.3, 0.3).into()),
+    ));
 }
 
-fn game_setup(mut commands: Commands, head_material: Res<HeadMaterial>) {
+fn game_setup(
+    mut commands: Commands,
+    head_material: Res<HeadMaterial>,
+    segment_material: Res<SegmentMaterial>,
+) {
+    spawn_segment(&mut commands, segment_material.0, Position { x: 10, y: 9 });
+    let first_segment = commands.current_entity().unwrap();
     commands
         .spawn(SpriteComponents {
             material: head_material.0,
-            sprite: Sprite::new(Vec2::new(10.0, 10.0)),
             ..Default::default()
         })
-        .with(SnakeHead { direction: Direction::Up })
+        .with(SnakeHead {
+            direction: Direction::Up,
+            next_segment: first_segment,
+        })
         .with(Position { x: 10, y: 10 })
         .with(Size::square(0.8));
 }