@@ -11,7 +11,7 @@ use crate::rwa::{
1111/// Storage keys for the modular compliance contract.
1212#[ contracttype]
1313#[ derive( Clone , Debug , Eq , PartialEq , PartialOrd , Ord ) ]
14- pub enum DataKey {
14+ pub enum ComplianceDataKey {
1515 /// Maps ComplianceHook -> `Vec<Address>` for registered modules
1616 HookModules ( ComplianceHook ) ,
1717}
@@ -33,7 +33,7 @@ pub enum DataKey {
3333/// A vector of module addresses registered for the specified hook.
3434/// Returns an empty vector if no modules are registered.
3535pub fn get_modules_for_hook ( e : & Env , hook : ComplianceHook ) -> Vec < Address > {
36- let key = DataKey :: HookModules ( hook) ;
36+ let key = ComplianceDataKey :: HookModules ( hook) ;
3737 if let Some ( existing_modules) = e. storage ( ) . persistent ( ) . get ( & key) {
3838 e. storage ( ) . persistent ( ) . extend_ttl (
3939 & key,
@@ -110,7 +110,7 @@ pub fn add_module_to(e: &Env, hook: ComplianceHook, module: Address) {
110110 }
111111
112112 // Add the module
113- let key = DataKey :: HookModules ( hook. clone ( ) ) ;
113+ let key = ComplianceDataKey :: HookModules ( hook. clone ( ) ) ;
114114 modules. push_back ( module. clone ( ) ) ;
115115 e. storage ( ) . persistent ( ) . set ( & key, & modules) ;
116116
@@ -159,7 +159,7 @@ pub fn remove_module_from(e: &Env, hook: ComplianceHook, module: Address) {
159159 modules. remove ( index) ;
160160
161161 // Update storage
162- let key = DataKey :: HookModules ( hook. clone ( ) ) ;
162+ let key = ComplianceDataKey :: HookModules ( hook. clone ( ) ) ;
163163 e. storage ( ) . persistent ( ) . set ( & key, & modules) ;
164164
165165 // Emit event
0 commit comments