diff --git a/contracts/stream/src/lib.rs b/contracts/stream/src/lib.rs index 81d5d290..3f6be9b3 100644 --- a/contracts/stream/src/lib.rs +++ b/contracts/stream/src/lib.rs @@ -585,12 +585,12 @@ impl DripStream { } /// Read-only: current withdrawable balance for the recipient. - pub fn withdrawable(env: Env) -> i128 { + pub fn withdrawable(env: Env) -> Result { let info = state::load(&env); if info.is_cancelled() { - return 0; + return Ok(0); } - math::withdrawable(&env, &info).unwrap_or(0) + math::withdrawable(&env, &info) } /// Read-only: whether clawback is enabled for this stream. @@ -743,12 +743,12 @@ impl DripStream { /// /// Useful for UIs that want to show "X streamed, Y withdrawn, Z remaining" /// without the caller needing to reimplement the rate × elapsed math. - pub fn streamed_total(env: Env) -> i128 { + pub fn streamed_total(env: Env) -> Result { let info = state::load(&env); if info.is_cancelled() { - return 0; + return Ok(0); } - math::streamed_amount(&env, &info).unwrap_or(0) + math::streamed_amount(&env, &info) } /// Read-only: full stream state.