RWA: country compliance modules

Cargo.toml

@@ -21,6 +21,8 @@ members = [
     "examples/ownable",
     "examples/pausable",
     "examples/rwa/*",
+    "examples/rwa-country-allow",
+    "examples/rwa-country-restrict",
     "examples/sac-admin-generic",
     "examples/sac-admin-wrapper",
     "examples/multisig-smart-account/*",

examples/rwa-country-allow/Cargo.toml

@@ -0,0 +1,22 @@
+[package]
+name = "rwa-country-allow"
+edition.workspace = true
+license.workspace = true
+repository.workspace = true
+publish = false
+version.workspace = true
+authors.workspace = true
+
+[package.metadata.stellar]
+cargo_inherit = true
+
+[lib]
+crate-type = ["cdylib", "rlib"]
+doctest = false
+
+[dependencies]
+soroban-sdk = { workspace = true }
+stellar-tokens = { workspace = true }
+
+[dev-dependencies]
+soroban-sdk = { workspace = true, features = ["testutils"] }

examples/rwa-country-allow/src/contract.rs

@@ -0,0 +1,101 @@
+use soroban_sdk::{contract, contractimpl, contracttype, Address, Env, String, Vec};
+use stellar_tokens::rwa::compliance::modules::{
+    country_allow::storage as country_allow,
+    storage::{
+        get_compliance_address, module_name, set_compliance_address, set_irs_address,
+        ComplianceModuleStorageKey,
+    },
+    ComplianceModule,
+};
+
+#[contracttype]
+enum DataKey {
+    Admin,
+}
+
+#[contract]
+pub struct CountryAllowContract;
+
+fn set_admin(e: &Env, admin: &Address) {
+    e.storage().instance().set(&DataKey::Admin, admin);
+}
+
+fn get_admin(e: &Env) -> Address {
+    e.storage().instance().get(&DataKey::Admin).expect("admin must be set")
+}
+
+fn require_module_admin_or_compliance_auth(e: &Env) {
+    if let Some(compliance) =
+        e.storage().instance().get::<_, Address>(&ComplianceModuleStorageKey::Compliance)
+    {
+        compliance.require_auth();
+    } else {
+        get_admin(e).require_auth();
+    }
+}
+
+#[contractimpl]
+impl CountryAllowContract {
+    pub fn __constructor(e: &Env, admin: Address) {
+        set_admin(e, &admin);
+    }
+
+    pub fn set_identity_registry_storage(e: &Env, token: Address, irs: Address) {
+        require_module_admin_or_compliance_auth(e);
+        set_irs_address(e, &token, &irs);
+    }
+
+    pub fn add_allowed_country(e: &Env, token: Address, country: u32) {
+        require_module_admin_or_compliance_auth(e);
+        country_allow::add_allowed_country(e, &token, country);
+    }
+
+    pub fn remove_allowed_country(e: &Env, token: Address, country: u32) {
+        require_module_admin_or_compliance_auth(e);
+        country_allow::remove_allowed_country(e, &token, country);
+    }
+
+    pub fn batch_allow_countries(e: &Env, token: Address, countries: Vec<u32>) {
+        require_module_admin_or_compliance_auth(e);
+        country_allow::batch_allow_countries(e, &token, &countries);
+    }
+
+    pub fn batch_disallow_countries(e: &Env, token: Address, countries: Vec<u32>) {
+        require_module_admin_or_compliance_auth(e);
+        country_allow::batch_disallow_countries(e, &token, &countries);
+    }
+
+    pub fn is_country_allowed(e: &Env, token: Address, country: u32) -> bool {
+        country_allow::is_country_allowed(e, &token, country)
+    }
+}
+
+#[contractimpl(contracttrait)]
+impl ComplianceModule for CountryAllowContract {
+    fn on_transfer(_e: &Env, _from: Address, _to: Address, _amount: i128, _token: Address) {}
+
+    fn on_created(_e: &Env, _to: Address, _amount: i128, _token: Address) {}
+
+    fn on_destroyed(_e: &Env, _from: Address, _amount: i128, _token: Address) {}
+
+    fn can_transfer(e: &Env, _from: Address, to: Address, _amount: i128, token: Address) -> bool {
+        country_allow::can_transfer(e, &to, &token)
+    }
+
+    fn can_create(e: &Env, to: Address, _amount: i128, token: Address) -> bool {
+        country_allow::can_transfer(e, &to, &token)
+    }
+
+    fn name(e: &Env) -> String {
+        module_name(e, "CountryAllowModule")
+    }
+
+    fn get_compliance_address(e: &Env) -> Address {
+        get_compliance_address(e)
+    }
+
+    fn set_compliance_address(e: &Env, compliance: Address) {
+        get_admin(e).require_auth();
+        set_compliance_address(e, &compliance);
+    }
+}

examples/rwa-country-allow/src/lib.rs

@@ -0,0 +1,5 @@
+#![no_std]
+
+pub mod contract;
+#[cfg(test)]
+mod test;