respect params with automatic and mutable access

JoJoJet · Dec 28, 2022 · fb9730a · fb9730a
1 parent 9daf39f
commit fb9730a
Showing 1 changed file with 23 additions and 11 deletions.
diff --git a/crates/bevy_ecs/macros/src/lib.rs b/crates/bevy_ecs/macros/src/lib.rs
@@ -496,19 +496,32 @@ pub fn derive_system_param(input: TokenStream) -> TokenStream {
         let end = Vec::from_iter(tuple_patterns.drain(..LIMIT));
         tuple_patterns.push(parse_quote!( (#(#end,)*) ));
     }
-    // Create a where clause for the `ReadOnlySystemParam` impl.
-    // Ensure that each field implements `ReadOnlySystemParam`.
-    let mut read_only_generics = generics.clone();
-    let read_only_where_clause = read_only_generics.make_where_clause();
-    for field_type in &field_types {
-        read_only_where_clause
-            .predicates
-            .push(syn::parse_quote!(#field_type: #path::system::ReadOnlySystemParam));
-    }
 
     let struct_name = &ast.ident;
     let state_struct_visibility = &ast.vis;
 
+    let read_only_impl = match access {
+        ParamAccess::Auto => {
+            // Create a where clause for the `ReadOnlySystemParam` impl.
+            // Ensure that each field implements `ReadOnlySystemParam`.
+            let mut read_only_generics = generics.clone();
+            let read_only_where_clause = read_only_generics.make_where_clause();
+            for field_type in &field_types {
+                read_only_where_clause
+                    .predicates
+                    .push(syn::parse_quote!(#field_type: #path::system::ReadOnlySystemParam));
+            }
+
+            quote! {
+                // Safety: Each field is `ReadOnlySystemParam`, so this can only read from the `World`
+                unsafe impl<'w, 's, #punctuated_generics> #path::system::ReadOnlySystemParam for #struct_name #ty_generics #read_only_where_clause
+                {}
+            }
+        }
+        ParamAccess::ReadOnly => todo!(),
+        ParamAccess::Mutable => quote! {},
+    };
+
     TokenStream::from(quote! {
         // We define the FetchState struct in an anonymous scope to avoid polluting the user namespace.
         // The struct can still be accessed via SystemParam::State, e.g. EventReaderState can be accessed via
@@ -566,8 +579,7 @@ pub fn derive_system_param(input: TokenStream) -> TokenStream {
                 }
             }
 
-            // Safety: Each field is `ReadOnlySystemParam`, so this can only read from the `World`
-            unsafe impl<'w, 's, #punctuated_generics> #path::system::ReadOnlySystemParam for #struct_name #ty_generics #read_only_where_clause {}
+            #read_only_impl
         };
     })
 }