feat: added get_range method

m62624 · Aug 27, 2023 · 4eae649 · 4eae649
1 parent 4adefb3
commit 4eae649
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/flexible_inspect_rs/src/rules/rule_bytes/base.rs b/flexible_inspect_rs/src/rules/rule_bytes/base.rs
@@ -1,3 +1,4 @@
+use super::range::Range;
 use super::SimpleRulesBytes;
 use super::*;
 use crate::rules::traits::RuleBase;
@@ -42,4 +43,8 @@ impl RuleBase for RuleBytes {
     fn get_complex_rules(&self) -> Option<&IndexSet<Self::RuleType>> {
         None
     }
+
+    fn get_range(&self) -> Option<&Range> {
+        self.0.general_modifiers.range.as_ref()
+    }
 }
diff --git a/flexible_inspect_rs/src/rules/rule_str/base.rs b/flexible_inspect_rs/src/rules/rule_str/base.rs
@@ -1,3 +1,4 @@
+use super::range::Range;
 use super::*;
 use crate::rules::traits::RuleBase;
 
@@ -47,4 +48,8 @@ impl RuleBase for Rule {
         }
         None
     }
+
+    fn get_range(&self) -> Option<&Range> {
+        self.0.general_modifiers.range.as_ref()
+    }
 }
diff --git a/flexible_inspect_rs/src/rules/traits.rs b/flexible_inspect_rs/src/rules/traits.rs
@@ -4,7 +4,7 @@ They are necessary to avoid code duplicates. Especially in context_match, where
 */
 
 // =======================================================
-use super::common_elements::range::{RangeBoundaries, RangeMode};
+use super::common_elements::range::*;
 use super::{CaptureData, Counter, ModeMatch};
 use crate::prelude::MatchRequirement;
 use indexmap::IndexSet;
@@ -26,6 +26,7 @@ pub trait RuleBase {
     fn get_counter(&self) -> Option<Counter>;
     fn get_mode_match(&self) -> &ModeMatch;
     fn get_str(&self) -> &str;
+    fn get_range(&self) -> Option<&Range>;
 }
 
 /// The main trait for `context_match`, that is,