Node repetition operator's reference analysis bug fixed

Eliah-Lakhin · May 25, 2023 · ef66348 · ef66348
1 parent 8a44e95
commit ef66348
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/work/crates/derive/src/node/regex/references.rs b/work/crates/derive/src/node/regex/references.rs
@@ -41,7 +41,7 @@ use syn::{Error, Result};
 use crate::{
     node::{
         builder::{kind::VariantKind, Builder},
-        regex::{operand::RegexOperand, Regex},
+        regex::{operand::RegexOperand, operator::RegexOperator, Regex},
     },
     utils::{debug_panic, PredictableCollection, Set, SetImpl},
 };
@@ -141,7 +141,21 @@ impl CheckReferences for Regex {
 
             Self::Operand(RegexOperand::Token { .. }) => Ok(Set::empty()),
 
-            Self::Unary { inner, .. } => inner.check_references(context, builder),
+            Self::Unary { operator, inner } => {
+                let inner = inner.check_references(context, builder)?;
+
+                match operator {
+                    RegexOperator::ZeroOrMore {
+                        separator: Some(separator),
+                    } => Ok(separator.check_references(context, builder)?.merge(inner)),
+
+                    RegexOperator::OneOrMore {
+                        separator: Some(separator),
+                    } => Ok(separator.check_references(context, builder)?.merge(inner)),
+
+                    _ => Ok(inner),
+                }
+            }
 
             Self::Binary { left, right, .. } => {
                 let left = left.check_references(context, builder)?;