1 files changed, 81 insertions, 0 deletions
diff --git a/engine-ecs/examples/optional_component.rs b/engine-ecs/examples/optional_component.rs
new file mode 100644
index 0000000..79650b9
--- /dev/null
+++ b/engine-ecs/examples/optional_component.rs
@@ -0,0 +1,81 @@
+use engine_ecs::phase::UPDATE as UPDATE_PHASE;
+use engine_ecs::{Component, Query, World};
+
+#[derive(Debug, Component)]
+struct PettingCapacity
+{
+    capacity_left: u32,
+}
+
+#[derive(Debug, Clone, Copy, Component)]
+enum Aggressivity
+{
+    High,
+    Medium,
+    Low,
+}
+
+#[derive(Debug, Component)]
+pub struct CatName
+{
+    name: String,
+}
+
+fn pet_cats(query: Query<(&CatName, &mut PettingCapacity, Option<&Aggressivity>)>)
+{
+    for (cat_name, mut petting_capacity, aggressivity) in &query {
+        let Some(aggressivity) = aggressivity else {
+            println!("Aggressivity of cat {} is unknown. Skipping", cat_name.name);
+            continue;
+        };
+
+        if let Aggressivity::High = *aggressivity {
+            println!("Cat {} is aggressive. Skipping", cat_name.name);
+            continue;
+        }
+
+        if petting_capacity.capacity_left == 0 {
+            println!(
+                "Cat {} have had enough of being petted. Skipping",
+                cat_name.name
+            );
+            continue;
+        }
+
+        println!("Petting cat {}", cat_name.name);
+
+        petting_capacity.capacity_left -= 1;
+    }
+}
+
+fn main()
+{
+    let mut world = World::new();
+
+    world.register_system(*UPDATE_PHASE, pet_cats);
+
+    world.create_entity((
+        CatName { name: "Jasper".to_string() },
+        Aggressivity::Medium,
+        PettingCapacity { capacity_left: 5 },
+    ));
+
+    world.create_entity((
+        CatName { name: "Otto".to_string() },
+        PettingCapacity { capacity_left: 9 },
+    ));
+
+    world.create_entity((
+        CatName { name: "Carrie".to_string() },
+        PettingCapacity { capacity_left: 2 },
+        Aggressivity::High,
+    ));
+
+    world.create_entity((
+        CatName { name: "Tommy".to_string() },
+        PettingCapacity { capacity_left: 1 },
+        Aggressivity::Low,
+    ));
+
+    world.step();
+}