feat: contract and account hooks transactions

Signed-off-by: Ivaylo Nikolov <[email protected]>

Author: ivaylonikolov7

src/account/account_create_transaction.rs

@@ -5,6 +5,10 @@ use hedera_proto::services::crypto_service_client::CryptoServiceClient;
 use time::Duration;
 use tonic::transport::Channel;
 
+use crate::hooks::{
+    HookCreationDetails,
+    LambdaEvmHook,
+};
 use crate::ledger_id::RefLedgerId;
 use crate::protobuf::{
     FromProtobuf,
@@ -76,6 +80,9 @@ pub struct AccountCreateTransactionData {
 
     /// If true, the account declines receiving a staking reward. The default value is false.
     decline_staking_reward: bool,
+
+    /// Hooks to add immediately after creating this account.
+    hooks: Vec<HookCreationDetails>,
 }
 
 impl Default for AccountCreateTransactionData {
@@ -91,6 +98,7 @@ impl Default for AccountCreateTransactionData {
             alias: None,
             staked_id: None,
             decline_staking_reward: false,
+            hooks: Vec::new(),
         }
     }
 }
@@ -293,6 +301,29 @@ impl AccountCreateTransaction {
         self.data_mut().decline_staking_reward = decline;
         self
     }
+
+    pub fn add_hook(&mut self, hook: HookCreationDetails) -> &mut Self {
+        self.data_mut().hooks.push(hook);
+        self
+    }
+
+    pub fn add_lambda_evm_hook(&mut self, hook: LambdaEvmHook) -> &mut Self {
+        // Helper to add a Lambda EVM hook with default extension point and hook ID
+        use crate::hooks::HookExtensionPoint;
+        let details =
+            HookCreationDetails::new(HookExtensionPoint::AccountAllowanceHook, 1, Some(hook));
+        self.data_mut().hooks.push(details);
+        self
+    }
+
+    pub fn set_hooks(&mut self, hooks: Vec<HookCreationDetails>) -> &mut Self {
+        self.data_mut().hooks = hooks;
+        self
+    }
+
+    pub fn get_hooks(&self) -> &[HookCreationDetails] {
+        &self.data().hooks
+    }
 }
 
 impl TransactionData for AccountCreateTransactionData {}
@@ -353,6 +384,11 @@ impl FromProtobuf<services::CryptoCreateTransactionBody> for AccountCreateTransa
             alias,
             staked_id: Option::from_protobuf(pb.staked_id)?,
             decline_staking_reward: pb.decline_reward,
+            hooks: pb
+                .hook_creation_details
+                .into_iter()
+                .map(HookCreationDetails::from_protobuf)
+                .collect::<Result<Vec<_>, _>>()?,
         })
     }
 }
@@ -391,6 +427,7 @@ impl ToProtobuf for AccountCreateTransactionData {
             alias: self.alias.map_or(vec![], |it| it.to_bytes().to_vec()),
             decline_reward: self.decline_staking_reward,
             staked_id,
+            hook_creation_details: self.hooks.iter().map(|h| h.to_protobuf()).collect(),
         }
     }
 }
@@ -417,8 +454,13 @@ mod tests {
         AccountCreateTransaction,
         AccountId,
         AnyTransaction,
+        ContractId,
         EvmAddress,
+        EvmHookSpec,
         Hbar,
+        HookCreationDetails,
+        HookExtensionPoint,
+        LambdaEvmHook,
         PublicKey,
     };
 
@@ -560,6 +602,7 @@ mod tests {
                         20,
                         23,
                     ],
+                    hook_creation_details: [],
                     staked_id: Some(
                         StakedAccountId(
                             AccountId {
@@ -683,6 +726,7 @@ mod tests {
                         20,
                         23,
                     ],
+                    hook_creation_details: [],
                     staked_id: Some(
                         StakedNodeId(
                             4,
@@ -710,6 +754,13 @@ mod tests {
     #[test]
     fn from_proto_body() {
         #[allow(deprecated)]
+        let contract_id = ContractId::new(0, 0, 1);
+        let hooks = vec![HookCreationDetails::new(
+            HookExtensionPoint::AccountAllowanceHook,
+            0,
+            Some(LambdaEvmHook::new(EvmHookSpec::new(Some(contract_id)), vec![])),
+        )];
+
         let tx = services::CryptoCreateTransactionBody {
             key: Some(key().to_protobuf()),
             initial_balance: INITIAL_BALANCE.to_tinybars() as u64,
@@ -728,6 +779,7 @@ mod tests {
             staked_id: Some(services::crypto_create_transaction_body::StakedId::StakedAccountId(
                 STAKED_ACCOUNT_ID.to_protobuf(),
             )),
+            hook_creation_details: hooks.iter().map(|h| h.to_protobuf()).collect(),
         };
 
         let tx = AccountCreateTransactionData::from_protobuf(tx).unwrap();

src/account/account_update_transaction.rs

@@ -8,6 +8,7 @@ use time::{
 };
 use tonic::transport::Channel;
 
+use crate::hooks::HookCreationDetails;
 use crate::ledger_id::RefLedgerId;
 use crate::protobuf::{
     FromProtobuf,
@@ -89,6 +90,10 @@ pub struct AccountUpdateTransactionData {
 
     /// If true, the account declines receiving a staking reward. The default value is false.
     decline_staking_reward: Option<bool>,
+
+    /// Hooks to add immediately after updating this account.
+    hooks: Vec<HookCreationDetails>,
+    hook_ids_to_delete: Vec<i64>,
 }
 
 impl AccountUpdateTransaction {
@@ -271,6 +276,42 @@ impl AccountUpdateTransaction {
         self.data_mut().decline_staking_reward = Some(decline);
         self
     }
+
+    /// Returns the hooks to be created.
+    #[must_use]
+    pub fn get_hooks_to_create(&self) -> &[HookCreationDetails] {
+        &self.data().hooks
+    }
+
+    /// Adds a hook to be created.
+    pub fn add_hook(&mut self, hook: HookCreationDetails) -> &mut Self {
+        self.data_mut().hooks.push(hook);
+        self
+    }
+
+    /// Sets the hooks to be created.
+    pub fn set_hooks(&mut self, hooks: Vec<HookCreationDetails>) -> &mut Self {
+        self.data_mut().hooks = hooks;
+        self
+    }
+
+    /// Returns the hook IDs to be deleted.
+    #[must_use]
+    pub fn get_hooks_to_delete(&self) -> &[i64] {
+        &self.data().hook_ids_to_delete
+    }
+
+    /// Adds a hook ID to be deleted.
+    pub fn delete_hook(&mut self, hook_id: i64) -> &mut Self {
+        self.data_mut().hook_ids_to_delete.push(hook_id);
+        self
+    }
+
+    /// Sets the hook IDs to be deleted.
+    pub fn delete_hooks(&mut self, hook_ids: Vec<i64>) -> &mut Self {
+        self.data_mut().hook_ids_to_delete = hook_ids;
+        self
+    }
 }
 
 impl TransactionData for AccountUpdateTransactionData {}
@@ -339,6 +380,12 @@ impl FromProtobuf<services::CryptoUpdateTransactionBody> for AccountUpdateTransa
             max_automatic_token_associations: pb.max_automatic_token_associations,
             staked_id: Option::from_protobuf(pb.staked_id)?,
             decline_staking_reward: pb.decline_reward,
+            hooks: pb
+                .hook_creation_details
+                .into_iter()
+                .map(HookCreationDetails::from_protobuf)
+                .collect::<Result<Vec<_>, _>>()?,
+            hook_ids_to_delete: pb.hook_ids_to_delete,
         })
     }
 }
@@ -382,6 +429,8 @@ impl ToProtobuf for AccountUpdateTransactionData {
             receive_record_threshold_field: None,
             receiver_sig_required_field: receiver_signature_required,
             staked_id,
+            hook_creation_details: self.hooks.iter().map(|hook| hook.to_protobuf()).collect(),
+            hook_ids_to_delete: self.hook_ids_to_delete.clone(),
         }
     }
 }
@@ -410,6 +459,11 @@ mod tests {
         AccountId,
         AccountUpdateTransaction,
         AnyTransaction,
+        ContractId,
+        EvmHookSpec,
+        HookCreationDetails,
+        HookExtensionPoint,
+        LambdaEvmHook,
         PublicKey,
     };
 
@@ -563,6 +617,8 @@ mod tests {
                         100,
                     ),
                     decline_reward: None,
+                    hook_ids_to_delete: [],
+                    hook_creation_details: [],
                     send_record_threshold_field: None,
                     receive_record_threshold_field: None,
                     receiver_sig_required_field: Some(
@@ -696,6 +752,8 @@ mod tests {
                         100,
                     ),
                     decline_reward: None,
+                    hook_ids_to_delete: [],
+                    hook_creation_details: [],
                     send_record_threshold_field: None,
                     receive_record_threshold_field: None,
                     receiver_sig_required_field: Some(
@@ -730,6 +788,14 @@ mod tests {
     #[test]
     fn from_proto_body() {
         #[allow(deprecated)]
+        let contract_id = ContractId::new(0, 0, 1);
+        let hooks = vec![HookCreationDetails::new(
+            HookExtensionPoint::AccountAllowanceHook,
+            0,
+            Some(LambdaEvmHook::new(EvmHookSpec::new(Some(contract_id)), vec![])),
+        )];
+        let hook_ids_to_delete = vec![1, 2, 3];
+
         let tx = services::CryptoUpdateTransactionBody {
             account_id_to_update: Some(ACCOUNT_ID.to_protobuf()),
             key: Some(key().to_protobuf()),
@@ -746,6 +812,8 @@ mod tests {
             )),
             proxy_fraction: 0,
             expiration_time: Some(EXPIRATION_TIME.to_protobuf()),
+            hook_creation_details: hooks.iter().map(|h| h.to_protobuf()).collect(),
+            hook_ids_to_delete,
         };
 
         let tx = AccountUpdateTransactionData::from_protobuf(tx).unwrap();

src/contract/contract_create_transaction.rs

@@ -5,6 +5,7 @@ use hedera_proto::services::smart_contract_service_client::SmartContractServiceC
 use time::Duration;
 use tonic::transport::Channel;
 
+use crate::hooks::HookCreationDetails;
 use crate::ledger_id::RefLedgerId;
 use crate::protobuf::FromProtobuf;
 use crate::staked_id::StakedId;
@@ -57,6 +58,9 @@ pub struct ContractCreateTransactionData {
     staked_id: Option<StakedId>,
 
     decline_staking_reward: bool,
+
+    /// Hooks to add immediately after creating this contract.
+    hooks: Vec<HookCreationDetails>,
 }
 
 impl Default for ContractCreateTransactionData {
@@ -74,6 +78,7 @@ impl Default for ContractCreateTransactionData {
             auto_renew_account_id: None,
             staked_id: None,
             decline_staking_reward: false,
+            hooks: Vec::new(),
         }
     }
 }
@@ -240,6 +245,24 @@ impl ContractCreateTransaction {
         self.data_mut().decline_staking_reward = decline;
         self
     }
+
+    /// Returns the hooks to be created.
+    #[must_use]
+    pub fn get_hooks_to_create(&self) -> &[HookCreationDetails] {
+        &self.data().hooks
+    }
+
+    /// Adds a hook to be created.
+    pub fn add_hook(&mut self, hook: HookCreationDetails) -> &mut Self {
+        self.data_mut().hooks.push(hook);
+        self
+    }
+
+    /// Sets the hooks to be created.
+    pub fn set_hooks(&mut self, hooks: Vec<HookCreationDetails>) -> &mut Self {
+        self.data_mut().hooks = hooks;
+        self
+    }
 }
 
 impl TransactionData for ContractCreateTransactionData {
@@ -313,6 +336,11 @@ impl FromProtobuf<services::ContractCreateTransactionBody> for ContractCreateTra
             auto_renew_account_id: Option::from_protobuf(pb.auto_renew_account_id)?,
             staked_id: Option::from_protobuf(pb.staked_id)?,
             decline_staking_reward: pb.decline_reward,
+            hooks: pb
+                .hook_creation_details
+                .into_iter()
+                .map(HookCreationDetails::from_protobuf)
+                .collect::<Result<Vec<_>, _>>()?,
         })
     }
 }
@@ -372,6 +400,7 @@ impl ToProtobuf for ContractCreateTransactionData {
             decline_reward: self.decline_staking_reward,
             initcode_source,
             staked_id,
+            hook_creation_details: self.hooks.iter().map(|hook| hook.to_protobuf()).collect(),
         }
     }
 }
@@ -397,8 +426,13 @@ mod tests {
         AccountId,
         AnyTransaction,
         ContractCreateTransaction,
+        ContractId,
+        EvmHookSpec,
         FileId,
         Hbar,
+        HookCreationDetails,
+        HookExtensionPoint,
+        LambdaEvmHook,
         PublicKey,
     };
 
@@ -539,6 +573,7 @@ mod tests {
                         },
                     ),
                     decline_reward: false,
+                    hook_creation_details: [],
                     initcode_source: Some(
                         FileId(
                             FileId {
@@ -665,6 +700,7 @@ mod tests {
                         },
                     ),
                     decline_reward: false,
+                    hook_creation_details: [],
                     initcode_source: Some(
                         Initcode(
                             [
@@ -702,6 +738,13 @@ mod tests {
     #[test]
     fn from_proto_body() {
         #[allow(deprecated)]
+        let contract_id = ContractId::new(0, 0, 1);
+        let hooks = vec![HookCreationDetails::new(
+            HookExtensionPoint::AccountAllowanceHook,
+            0,
+            Some(LambdaEvmHook::new(EvmHookSpec::new(Some(contract_id)), vec![])),
+        )];
+
         let tx = services::ContractCreateTransactionBody {
             admin_key: Some(admin_key().to_protobuf()),
             initial_balance: INITIAL_BALANCE.to_tinybars(),
@@ -724,6 +767,7 @@ mod tests {
                     BYTECODE_FILE_ID.to_protobuf(),
                 ),
             ),
+            hook_creation_details: hooks.iter().map(|h| h.to_protobuf()).collect(),
         };
         let tx = ContractCreateTransactionData::from_protobuf(tx).unwrap();