Skip to content
Open
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
99 changes: 95 additions & 4 deletions contracts/escrow/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,8 @@ pub enum DataKey {
Locked,
MultisigConfig(u64), // Per-job multisig configuration
UpgradeAdmin,
Treasury,
FeeBps,
}

#[contracttype]
Expand Down Expand Up @@ -150,6 +152,8 @@ pub enum EscrowError {
UpgradeAdminNotSet = 18,
ArithmeticOverflow = 19,
DisputeResolutionExpired = 20,
FeeTooHigh = 21,
NothingToSweep = 22,
}

/// Maximum platform fee, in basis points (100% = 10_000 bps).
Expand Down Expand Up @@ -257,6 +261,41 @@ pub struct DisputeExpiredEvent {
pub expired_at: u64,
}

#[contracttype]
#[derive(Clone)]
pub struct FeeConfigUpdatedEvent {
pub treasury: Address,
pub fee_bps: u32,
pub updated_at: u64,
}

#[contracttype]
#[derive(Clone)]
pub struct LockupUpdatedEvent {
pub job_id: u64,
pub expires_at: u64,
pub updated_at: u64,
}

#[contracttype]
#[derive(Clone)]
pub struct EmergencySweepEvent {
pub job_id: u64,
pub admin: Address,
pub rescue_address: Address,
pub amount: i128,
pub swept_at: u64,
}

#[contracttype]
#[derive(Clone)]
pub struct MilestonesAmendedEvent {
pub job_id: u64,
pub milestone_count: u32,
pub remaining_amount: i128,
pub amended_at: u64,
}

fn enter_reentrancy_guard(env: &Env) {
if env.storage().instance().has(&DataKey::Locked) {
panic_with_error!(env, EscrowError::ReentrancyDetected);
Expand Down Expand Up @@ -1450,11 +1489,12 @@ impl EscrowContract {
treasury: Address,
fee_bps: u32,
) -> Result<(), EscrowError> {
let admin: Address = env
let config: ContractConfig = env
.storage()
.instance()
.get(&DataKey::Admin)
.get(&DataKey::Config)
.ok_or(EscrowError::NotInitialized)?;
let admin = config.admin;
admin.require_auth();

if fee_bps > MAX_FEE_BPS {
Expand Down Expand Up @@ -1560,11 +1600,12 @@ impl EscrowContract {
job_id: u64,
rescue_address: Address,
) -> Result<(), EscrowError> {
let admin: Address = env
let config: ContractConfig = env
.storage()
.instance()
.get(&DataKey::Admin)
.get(&DataKey::Config)
.ok_or(EscrowError::NotInitialized)?;
let admin = config.admin;
admin.require_auth();

let key = DataKey::Job(job_id);
Expand Down Expand Up @@ -1691,6 +1732,56 @@ impl EscrowContract {

Ok(())
}

fn fee_bps(env: &Env) -> u32 {
env.storage().instance().get(&DataKey::FeeBps).unwrap_or(0)
}

fn payout_with_fee(
env: &Env,
_job_id: u64,
job: &EscrowJob,
amount: i128,
) {
let treasury_opt: Option<Address> = env.storage().instance().get(&DataKey::Treasury);
let fee_bps = Self::fee_bps(env);
let token_client = token::Client::new(env, &job.token);

if let Some(treasury) = treasury_opt {
if fee_bps > 0 && fee_bps <= 10_000 {
// fee_amount = amount * fee_bps / 10_000
let fee_amount = amount
.checked_mul(fee_bps as i128)
.unwrap()
.checked_div(10_000)
.unwrap();
let freelancer_amount = amount.checked_sub(fee_amount).unwrap();

if fee_amount > 0 {
token_client.transfer(
&env.current_contract_address(),
&treasury,
&fee_amount,
);
}
if freelancer_amount > 0 {
token_client.transfer(
&env.current_contract_address(),
&job.freelancer,
&freelancer_amount,
);
}
return;
}
}

// Default: no fee or fee is 0 or treasury not configured
token_client.transfer(
&env.current_contract_address(),
&job.freelancer,
&amount,
);
}
}

#[cfg(test)]
Expand Down
Loading
Loading