accept doc comments when deriving ShaderType (#42)

teoxoy · May 9, 2023 · 0b6d6b4 · 0b6d6b4
1 parent 2dfa542
commit 0b6d6b4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/derive/impl/src/lib.rs b/derive/impl/src/lib.rs
@@ -197,6 +197,9 @@ pub fn derive_shader_type(input: DeriveInput, root: &Path) -> TokenStream {
                 align: None,
             };
             for attr in &field.attrs {
+                if !(attr.meta.path().is_ident("size") || attr.meta.path().is_ident("align")) {
+                    continue;
+                }
                 match attr.meta.require_list() {
                     Ok(meta_list) => {
                         let span = meta_list.tokens.span();

diff --git a/tests/pass/attributes.rs b/tests/pass/attributes.rs
@@ -18,3 +18,9 @@ struct TestRtArray {
     #[size(runtime)]
     b: Vec<u32>,
 }
+
+#[derive(ShaderType)]
+struct TestDocComment {
+    /// This is an unsigned integer
+    a: u32,
+}