feat: added not any_of and all_of filters

txpipe · Aug 29, 2023 · 8df5c10 · 8df5c10
1 parent cb88b44
commit 8df5c10
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/filters/match_pattern.rs b/src/filters/match_pattern.rs
@@ -128,6 +128,9 @@ pub enum Predicate {
     OutputAddress(AddressPattern),
     WithdrawalAddress(AddressPattern),
     CollateralAddress(AddressPattern),
+    Not(Box<Predicate>),
+    AnyOf(Vec<Predicate>),
+    AllOf(Vec<Predicate>),
 }
 
 impl Predicate {
@@ -141,6 +144,23 @@ impl Predicate {
             Predicate::CollateralAddress(address_pattern) => {
                 Ok(collateral_match(tx, address_pattern)?)
             }
+            Predicate::Not(x) => Ok(!x.tx_match(point, tx)?),
+            Predicate::AnyOf(x) => {
+                for p in x {
+                    if p.tx_match(point, tx)? {
+                        return Ok(true);
+                    };
+                }
+                Ok(false)
+            }
+            Predicate::AllOf(x) => {
+                for p in x {
+                    if !p.tx_match(point, tx)? {
+                        return Ok(false);
+                    };
+                }
+                Ok(true)
+            }
         }
     }
 }