refactor(transaction): add common getters to tramsaction

starkware-libs · Aug 21, 2024 · 110a23f · 110a23f
1 parent a57ad02
commit 110a23f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/crates/blockifier/src/transaction/account_transaction.rs b/crates/blockifier/src/transaction/account_transaction.rs
@@ -193,6 +193,14 @@ impl AccountTransaction {
         Ok(())
     }
 
+    pub fn transaction_hash(&self) -> &TransactionHash {
+        match self {
+            AccountTransaction::Invoke(tx) => &tx.tx_hash,
+            AccountTransaction::Declare(tx) => &tx.tx_hash,
+            AccountTransaction::DeployAccount(tx) => &tx.tx_hash,
+        }
+    }
+
     fn check_fee_bounds(
         &self,
         tx_context: &TransactionContext,

diff --git a/crates/blockifier/src/transaction/transaction_execution.rs b/crates/blockifier/src/transaction/transaction_execution.rs
@@ -97,6 +97,13 @@ impl Transaction {
             _ => unimplemented!(),
         }
     }
+
+    pub fn transaction_hash(tx: &Transaction) -> &TransactionHash {
+        match tx {
+            Transaction::AccountTransaction(tx) => tx.transaction_hash(),
+            Transaction::L1HandlerTransaction(tx) => &tx.tx_hash,
+        }
+    }
 }
 
 impl TransactionInfoCreator for Transaction {