Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions contracts/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ members = [
"contrib",
"multisig-wallet",
"multisig_transfer",
"./opsce",
]

[workspace.dependencies]
Expand Down
2 changes: 1 addition & 1 deletion contracts/assetsup/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ pub(crate) mod dividends;
pub(crate) mod error;
pub(crate) mod insurance;
pub(crate) mod lease;
pub(crate) mod tokenization;
pub mod tokenization;
pub(crate) mod transfer_restrictions;
pub(crate) mod types;
pub(crate) mod voting;
Expand Down
15 changes: 15 additions & 0 deletions contracts/opsce/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
[package]
name = "opsce"
version = "0.1.0"
edition = "2021"

t[lib]
crate-type = ["lib", "cdylib"]
doctest = false

[dependencies]
assetsup = { path = "../assetsup" }
soroban-sdk = { workspace = true }

[dev-dependencies]
soroban-sdk = { workspace = true, features = ["testutils"] }
58 changes: 58 additions & 0 deletions contracts/opsce/src/batch_transfer.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
use assetsup::tokenization::{get_token_balance, transfer_tokens};
use soroban_sdk::{contract, contracterror, contractimpl, contracttype, symbol_short, Address, Env, Vec};

#[contracterror]
#[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
#[repr(u32)]
pub enum ContractError {
InsufficientBalance = 1,
BatchLimitExceeded = 2,
InvalidTransferAmount = 3,
TransferFailed = 4,
}

#[contract]
pub struct BatchTokenTransferContract;

#[contractimpl]
impl BatchTokenTransferContract {
pub fn batch_transfer_tokens(
env: Env,
asset_id: u64,
transfers: Vec<(Address, i128)>,
) -> Result<(), ContractError> {
if transfers.len() > 50 {
return Err(ContractError::BatchLimitExceeded);
}

let sender = env.invoker();
sender.require_auth();

let mut total_amount: i128 = 0;
for (_, amount) in transfers.iter() {
if *amount <= 0 {
return Err(ContractError::InvalidTransferAmount);
}
total_amount = total_amount
.checked_add(*amount)
.ok_or(ContractError::InvalidTransferAmount)?;
}

let balance = get_token_balance(&env, asset_id, sender.clone())
.map_err(|_| ContractError::InsufficientBalance)?;
if balance < total_amount {
return Err(ContractError::InsufficientBalance);
}

for (recipient, amount) in transfers.iter() {
transfer_tokens(&env, asset_id, sender.clone(), recipient.clone(), *amount)
.map_err(|_| ContractError::TransferFailed)?;
env.events().publish(
("token", "token_transferred"),
(asset_id, sender.clone(), recipient.clone(), *amount),
);
}

Ok(())
}
}
174 changes: 174 additions & 0 deletions contracts/opsce/src/batch_transfer_test.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,174 @@
#![cfg(test)]
extern crate std;

use super::*;
use soroban_sdk::testutils::Address as _;
use soroban_sdk::{vec, Address, Env, String, Vec};

#[test]
fn test_batch_transfer_successful() {
let env = Env::default();
env.mock_all_auths();

let contract_id = env.register(BatchTokenTransferContract, ());
let client = BatchTokenTransferContractClient::new(&env, &contract_id);

let sender = Address::generate(&env);
let asset_id = 1u64;
let metadata = assetsup::TokenMetadata {
name: String::from_str(&env, "Batch Asset"),
description: String::from_str(&env, "Batch transfer test asset"),
asset_type: assetsup::AssetType::Digital,
ipfs_uri: None,
legal_docs_hash: None,
valuation_report_hash: None,
accredited_investor_required: false,
geographic_restrictions: Vec::new(&env),
};

assetsup::tokenization::tokenize_asset(
&env,
asset_id,
String::from_str(&env, "BATCH"),
1000,
2,
1,
sender.clone(),
metadata,
)
.unwrap();

let recipient1 = Address::generate(&env);
let recipient2 = Address::generate(&env);
let transfers: Vec<(Address, i128)> = vec![&env, (recipient1.clone(), 100_i128), (recipient2.clone(), 200_i128)];

client.batch_transfer_tokens(&asset_id, &transfers).unwrap();

let sender_balance = assetsup::tokenization::get_token_balance(&env, asset_id, sender.clone()).unwrap();
assert_eq!(sender_balance, 700);

let recipient1_balance = assetsup::tokenization::get_token_balance(&env, asset_id, recipient1).unwrap();
assert_eq!(recipient1_balance, 100);

let recipient2_balance = assetsup::tokenization::get_token_balance(&env, asset_id, recipient2).unwrap();
assert_eq!(recipient2_balance, 200);
}

#[test]
fn test_batch_transfer_insufficient_balance() {
let env = Env::default();
env.mock_all_auths();

let contract_id = env.register(BatchTokenTransferContract, ());
let client = BatchTokenTransferContractClient::new(&env, &contract_id);

let sender = Address::generate(&env);
let asset_id = 2u64;
let metadata = assetsup::TokenMetadata {
name: String::from_str(&env, "Batch Asset"),
description: String::from_str(&env, "Insufficient balance asset"),
asset_type: assetsup::AssetType::Digital,
ipfs_uri: None,
legal_docs_hash: None,
valuation_report_hash: None,
accredited_investor_required: false,
geographic_restrictions: Vec::new(&env),
};

assetsup::tokenization::tokenize_asset(
&env,
asset_id,
String::from_str(&env, "BATCH2"),
100,
2,
1,
sender.clone(),
metadata,
)
.unwrap();

let recipient = Address::generate(&env);
let transfers: Vec<(Address, i128)> = vec![&env, (recipient.clone(), 60_i128), (recipient.clone(), 50_i128)];

let result = client.batch_transfer_tokens(&asset_id, &transfers);
assert_eq!(result, Err(ContractError::InsufficientBalance));
}

#[test]
fn test_batch_transfer_limit_exceeded() {
let env = Env::default();
env.mock_all_auths();

let contract_id = env.register(BatchTokenTransferContract, ());
let client = BatchTokenTransferContractClient::new(&env, &contract_id);

let sender = Address::generate(&env);
let asset_id = 3u64;
let metadata = assetsup::TokenMetadata {
name: String::from_str(&env, "Batch Limit Asset"),
description: String::from_str(&env, "Limit exceeded asset"),
asset_type: assetsup::AssetType::Digital,
ipfs_uri: None,
legal_docs_hash: None,
valuation_report_hash: None,
accredited_investor_required: false,
geographic_restrictions: Vec::new(&env),
};

assetsup::tokenization::tokenize_asset(
&env,
asset_id,
String::from_str(&env, "BATCH3"),
10000,
2,
1,
sender.clone(),
metadata,
)
.unwrap();

let mut transfers: Vec<(Address, i128)> = Vec::new(&env);
for _ in 0..51 {
transfers.push_back((Address::generate(&env), 1_i128));
}

let result = client.batch_transfer_tokens(&asset_id, &transfers);
assert_eq!(result, Err(ContractError::BatchLimitExceeded));
}

#[test]
fn test_batch_transfer_empty_batch() {
let env = Env::default();
env.mock_all_auths();

let contract_id = env.register(BatchTokenTransferContract, ());
let client = BatchTokenTransferContractClient::new(&env, &contract_id);

let asset_id = 4u64;
let sender = Address::generate(&env);
let metadata = assetsup::TokenMetadata {
name: String::from_str(&env, "Batch Empty Asset"),
description: String::from_str(&env, "Empty transfer asset"),
asset_type: assetsup::AssetType::Digital,
ipfs_uri: None,
legal_docs_hash: None,
valuation_report_hash: None,
accredited_investor_required: false,
geographic_restrictions: Vec::new(&env),
};

assetsup::tokenization::tokenize_asset(
&env,
asset_id,
String::from_str(&env, "BATCH4"),
500,
2,
1,
sender,
metadata,
)
.unwrap();

let transfers: Vec<(Address, i128)> = Vec::new(&env);
client.batch_transfer_tokens(&asset_id, &transfers).unwrap();
}
Loading
Loading