feat: new macros

Author: Evalir

src/lib.rs

@@ -380,6 +380,8 @@ pub use fill::{Block, Cfg, NoopBlock, NoopCfg, Tx};
 mod lifecycle;
 pub use lifecycle::{ethereum_receipt, BlockOutput, PostTx, PostflightResult};
 
+mod macros;
+
 mod states;
 pub(crate) use states::sealed::*;
 pub use states::{

src/macros.rs

@@ -0,0 +1,28 @@
+/// Unwraps a Result, returning the value if successful, or returning an errored `Trevm` if not.
+#[macro_export]
+macro_rules! unwrap_or_trevm_err {
+    ($e:expr, $trevm:expr) => {
+        match $e {
+            Ok(val) => val,
+            Err(e) => return Err($trevm.errored(e.into())),
+        }
+    };
+}
+
+/// Executes a condition, returning an errored `Trevm` if not successful.
+#[macro_export]
+macro_rules! trevm_ensure {
+    ($cond:expr, $trevm:expr, $err:expr) => {
+        if !$cond {
+            trevm_bail!($trevm, $err);
+        }
+    };
+}
+
+/// Returns an errored `Trevm` with the provided error.
+#[macro_export]
+macro_rules! trevm_bail {
+    ($trevm:expr, $err:expr) => {
+        return Err($trevm.errored($err))
+    };
+}