feat(bones_ecs): make insert_stage_before/after() chainable. (#90)

fishfolk · Feb 9, 2023 · 7c578b4 · 7c578b4
1 parent b912295
commit 7c578b4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/crates/bones_ecs/src/stage.rs b/crates/bones_ecs/src/stage.rs
@@ -76,13 +76,15 @@ impl SystemStages {
         &mut self,
         label: L,
         stage: S,
-    ) {
+    ) -> &mut Self {
         let stage_idx = self
             .stages
             .iter()
             .position(|x| x.id() == label.id())
             .unwrap_or_else(|| panic!("Could not find stage with label `{}`", label.name()));
         self.stages.insert(stage_idx, Box::new(stage));
+
+        self
     }
 
     /// Insert a new stage, after another existing stage
@@ -91,13 +93,15 @@ impl SystemStages {
         &mut self,
         label: L,
         stage: S,
-    ) {
+    ) -> &mut Self {
         let stage_idx = self
             .stages
             .iter()
             .position(|x| x.id() == label.id())
             .unwrap_or_else(|| panic!("Could not find stage with label `{}`", label.name()));
         self.stages.insert(stage_idx + 1, Box::new(stage));
+
+        self
     }
 }