@layerzerolabs/protocol-stellar-v2 0.2.8 → 0.2.10

package/contracts/workers/worker-common/src/worker_common.rs

@@ -1,166 +0,0 @@
-use soroban_sdk::{Address, Bytes, Env, Vec};
-use stellar_access::access_control;
-use stellar_contract_utils::pausable::{self as pausable};
-use stellar_macros::{only_admin, only_role};
-
-use crate::{
-    constants::{ADMIN_ROLE, ALLOWLIST, DENYLIST, MESSAGE_LIB_ROLE},
-    events::{SetDefaultMultiplierBps, SetDepositAddress, SetPriceFeed, SetSupportedOptionTypes, SetWorkerFeeLib},
-    storage::WorkerData,
-};
-
-// ============================================================================
-// WorkerCommon - Encapsulates worker operations
-// ============================================================================
-
-/// WorkerCommon provides shared functionality for LayerZero workers (DVN and Executor)
-/// Uses OpenZeppelin AccessControl for role-based permissions
-pub struct WorkerCommon;
-
-impl WorkerCommon {
-    // ========================================================================
-    // Initialization (Constructor Helper)
-    // ========================================================================
-
-    /// Initialize worker with OpenZeppelin AccessControl
-    /// This should be called in the contract's constructor (__constructor)
-    pub fn initialize(
-        env: &Env,
-        role_admin: &Address,
-        message_libs: &Vec<Address>,
-        price_feed: &Address,
-        default_multiplier_bps: u32,
-        admins: &Vec<Address>,
-    ) {
-        // Set the DEFAULT_ADMIN_ROLE (the top-level admin)
-        access_control::set_admin(env, role_admin);
-
-        // Grant MESSAGE_LIB_ROLE to all message libraries
-        for i in 0..message_libs.len() {
-            let message_lib = message_libs.get(i).unwrap();
-            access_control::grant_role_no_auth(env, role_admin, &message_lib, &MESSAGE_LIB_ROLE);
-        }
-
-        // Grant ADMIN_ROLE to all admins
-        for i in 0..admins.len() {
-            let admin_addr = admins.get(i).unwrap();
-            access_control::grant_role_no_auth(env, role_admin, &admin_addr, &ADMIN_ROLE);
-        }
-
-        // Initialize worker-specific data
-        WorkerData::set_price_feed(env, price_feed);
-        WorkerData::set_default_multiplier_bps(env, &default_multiplier_bps);
-    }
-
-    // ========================================================================
-    // Mutable Functions
-    // ========================================================================
-
-    /// Set the paused state of the worker (requires DEFAULT_ADMIN_ROLE)
-    #[only_admin]
-    pub fn set_paused(env: &Env, paused: bool) {
-        if paused {
-            pausable::pause(env);
-        } else {
-            pausable::unpause(env);
-        }
-    }
-
-    /// Set the default multiplier in basis points (ADMIN_ROLE only)
-    #[only_role(admin, "ADMIN")]
-    pub fn set_default_multiplier_bps(env: &Env, admin: &Address, multiplier_bps: u32) {
-        WorkerData::set_default_multiplier_bps(env, &multiplier_bps);
-        SetDefaultMultiplierBps { multiplier_bps }.publish(env);
-    }
-
-    /// Set the deposit address where fees are collected (admin only)
-    #[only_role(admin, "ADMIN")]
-    pub fn set_deposit_address(env: &Env, admin: &Address, deposit_address: &Address) {
-        WorkerData::set_deposit_address(env, deposit_address);
-
-        SetDepositAddress { deposit_address: deposit_address.clone() }.publish(env);
-    }
-
-    /// Set the price feed address (admin only)
-    #[only_role(admin, "ADMIN")]
-    pub fn set_price_feed(env: &Env, admin: &Address, price_feed: &Address) {
-        WorkerData::set_price_feed(env, price_feed);
-
-        SetPriceFeed { price_feed: price_feed.clone() }.publish(env);
-    }
-
-    /// Set supported option types for a destination endpoint (ADMIN_ROLE only)
-    #[only_role(admin, "ADMIN")]
-    pub fn set_supported_option_types(env: &Env, admin: &Address, eid: u32, option_types: Bytes) {
-        WorkerData::set_supported_option_types(env, eid, &option_types);
-
-        SetSupportedOptionTypes { dst_eid: eid, option_types }.publish(env);
-    }
-
-    /// Set the worker fee library address (ADMIN_ROLE only)
-    #[only_role(admin, "ADMIN")]
-    pub fn set_worker_fee_lib(env: &Env, admin: &Address, worker_fee_lib: &Address) {
-        WorkerData::set_worker_fee_lib(env, worker_fee_lib);
-
-        SetWorkerFeeLib { fee_lib: worker_fee_lib.clone() }.publish(env);
-    }
-
-    // ========================================================================
-    // View Functions
-    // ========================================================================
-
-    /// Check if the worker is paused
-    pub fn paused(env: &Env) -> bool {
-        pausable::paused(env)
-    }
-
-    /// Get the default multiplier in basis points
-    pub fn default_multiplier_bps(env: &Env) -> u32 {
-        WorkerData::default_multiplier_bps(env).unwrap_or(0)
-    }
-
-    /// Get the deposit address
-    pub fn deposit_address(env: &Env) -> Address {
-        WorkerData::deposit_address(env).expect("deposit address not set")
-    }
-
-    /// Get the current price feed address
-    pub fn price_feed(env: &Env) -> Address {
-        WorkerData::price_feed(env).expect("price feed not set")
-    }
-
-    /// Get supported option types for a destination endpoint
-    pub fn get_supported_option_types(env: &Env, eid: u32) -> Bytes {
-        WorkerData::supported_option_types(env, eid).unwrap_or(Bytes::new(env))
-    }
-
-    /// Get the worker fee library address
-    pub fn worker_fee_lib(env: &Env) -> Address {
-        WorkerData::worker_fee_lib(env).expect("worker fee lib not set")
-    }
-
-    /// Check if an address has ACL permission
-    /// ACL logic (matching EVM Worker.sol):
-    /// 1) if address has DENYLIST role -> deny
-    /// 2) else if allowlist is empty (size == 0) OR address has ALLOWLIST role -> allow
-    /// 3) else deny
-    pub fn has_acl(env: &Env, sender: &Address) -> bool {
-        // Check denylist first
-        if access_control::has_role(env, sender, &DENYLIST).is_some() {
-            return false;
-        }
-
-        // Check allowlist size and membership
-        let allowlist_size = Self::allowlist_size(env);
-        if allowlist_size == 0 || access_control::has_role(env, sender, &ALLOWLIST).is_some() {
-            return true;
-        }
-
-        false
-    }
-
-    /// Get the size of the allowlist (number of accounts with ALLOWLIST role)
-    pub fn allowlist_size(env: &Env) -> u32 {
-        access_control::get_role_member_count(env, &ALLOWLIST)
-    }
-}