Add insert and remove to InstanceCommands

Zeenobit · Jul 20, 2024 · 7716f06 · 7716f06
1 parent 4dcae9a
commit 7716f06
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -848,6 +848,16 @@ impl<'a, T: Kind> InstanceCommands<'a, T> {
     pub fn as_entity(&mut self) -> &mut EntityCommands<'a> {
         &mut self.0
     }
+
+    pub fn insert(&mut self, bundle: impl Bundle) -> &mut Self {
+        self.0.insert(bundle);
+        self
+    }
+
+    pub fn remove<U: Component>(&mut self) -> &mut Self {
+        self.0.remove::<U>();
+        self
+    }
 }
 
 impl<'a, T: Kind> From<InstanceCommands<'a, T>> for Instance<T> {