From ebdec5c4aa764f5d01369e033f8004fe6e186cdb Mon Sep 17 00:00:00 2001
From: James O'Brien <james.obrien@drafly.net>
Date: Mon, 11 Sep 2023 19:32:47 -0700
Subject: [PATCH] Add dynamic reference methods

---
 crates/bevy_ecs/src/term_query/builder.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/crates/bevy_ecs/src/term_query/builder.rs b/crates/bevy_ecs/src/term_query/builder.rs
index 8612feef4d0bfe..509ac84d659d5a 100644
--- a/crates/bevy_ecs/src/term_query/builder.rs
+++ b/crates/bevy_ecs/src/term_query/builder.rs
@@ -54,6 +54,17 @@ impl<'w, Q: QueryTermGroup> QueryBuilder<'w, Q> {
         self
     }
 
+    pub fn ref_id(&mut self, id: ComponentId) -> &mut Self {
+        self.terms.push(Term::Component(ComponentTerm::read_id(id)));
+        self
+    }
+
+    pub fn mut_id(&mut self, id: ComponentId) -> &mut Self {
+        self.terms
+            .push(Term::Component(ComponentTerm::write_id(id)));
+        self
+    }
+
     pub fn set<T: Component>(&mut self) -> &mut Self {
         let id = self.world.init_component::<T>();
         self.set_id(id);