Fix macro_metavar_expr_concat behavior with nested repetitions

Author: Delta17920

compiler/rustc_expand/src/mbe/transcribe.rs

@@ -558,26 +558,29 @@ fn metavar_expr_concat<'tx>(
             MetaVarExprConcatElem::Ident(elem) => elem.name,
             MetaVarExprConcatElem::Literal(elem) => *elem,
             MetaVarExprConcatElem::Var(ident) => {
-                match matched_from_ident(dcx, *ident, tscx.interp)? {
-                    NamedMatch::MatchedSeq(named_matches) => {
-                        let Some((curr_idx, _)) = tscx.repeats.last() else {
-                            return Err(dcx.struct_span_err(dspan.entire(), "invalid syntax"));
-                        };
-                        match &named_matches[*curr_idx] {
-                            // FIXME(c410-f3r) Nested repetitions are unimplemented
-                            MatchedSeq(_) => {
+                let mut matched = matched_from_ident(dcx, *ident, tscx.interp)?;
+                let mut repeats_iter = tscx.repeats.iter();
+
+                loop {
+                    match matched {
+                        NamedMatch::MatchedSingle(pnr) => {
+                            let symbol = extract_symbol_from_pnr(dcx, pnr, ident.span)?;
+                            concatenated.push_str(symbol.as_str());
+                            break;
+                        }
+                        NamedMatch::MatchedSeq(named_matches) => {
+                            if let Some((idx, _)) = repeats_iter.next() {
+                                matched = &named_matches[*idx];
+                            } else {
                                 return Err(dcx.struct_span_err(
                                     ident.span,
-                                    "nested repetitions with `${concat(...)}` metavariable expressions are not yet supported",
+                                    "`${concat(...)}` variable is still repeating at this depth",
                                 ));
                             }
-                            MatchedSingle(pnr) => extract_symbol_from_pnr(dcx, pnr, ident.span)?,
                         }
                     }
-                    NamedMatch::MatchedSingle(pnr) => {
-                        extract_symbol_from_pnr(dcx, pnr, ident.span)?
-                    }
                 }
+                continue;
             }
         };
         concatenated.push_str(symbol.as_str());

tests/ui/macros/concat-nested-repetition.rs

@@ -0,0 +1,35 @@
+//@ check-pass
+#![feature(macro_metavar_expr_concat)]
+
+struct A;
+struct B;
+const AA: A = A;
+const BB: B = B;
+
+macro_rules! define_ioctl_data {
+    (struct $s:ident {
+        $($field:ident: $ty:ident $([$opt:ident])?,)*
+    }) => {
+        pub struct $s {
+            $($field: $ty,)*
+        }
+
+        impl $s {
+            $($(
+                fn ${concat(get_, $field)}(&self) -> $ty {
+                    let _ = $opt;
+                    todo!()
+                }
+            )?)*
+        }
+    };
+}
+
+define_ioctl_data! {
+    struct Foo {
+        a: A [AA],
+        b: B [BB],
+    }
+}
+
+fn main() {}