-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
482 additions
and
125 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,130 @@ | ||
| use cosmwasm_otc_pkg::otc::definitions::{OtcItem, OtcItemInfo, OtcPosition, OtcPositionStatus}; | ||
| use cosmwasm_std::{ | ||
| attr, Addr, Attribute, Coin, CosmosMsg, DepsMut, Env, StdError, StdResult, Uint128, | ||
| }; | ||
|
|
||
| use crate::state::{active_positions, execute_positions}; | ||
|
|
||
| pub fn collect_otc_items( | ||
| env: &Env, | ||
| items: &Vec<OtcItem>, | ||
| sender: Addr, | ||
| funds: Vec<Coin>, | ||
| ) -> StdResult<(Vec<CosmosMsg>, Vec<Coin>)> { | ||
| let coins = assert_received_funds( | ||
| &items.iter().map(|val| val.item_info.clone()).collect(), | ||
| funds, | ||
| )?; | ||
| let mut msgs: Vec<CosmosMsg> = vec![]; | ||
| for item in items { | ||
| match &item.item_info { | ||
| OtcItemInfo::Cw20 { .. } | &OtcItemInfo::Cw721 { .. } => msgs.push( | ||
| item.item_info | ||
| .build_send_msg(env, &sender, &env.contract.address, None)?, | ||
| ), | ||
| _ => {} | ||
| } | ||
| } | ||
|
|
||
| Ok((msgs, coins)) | ||
| } | ||
|
|
||
| pub fn send_otc_items( | ||
| env: &Env, | ||
| items: &mut Vec<OtcItem>, | ||
| position_status: &OtcPositionStatus, | ||
| to: &Addr, | ||
| ) -> StdResult<Vec<CosmosMsg>> { | ||
| let mut msgs: Vec<CosmosMsg> = vec![]; | ||
| for item in items { | ||
| let amount = item.sendable_amount_and_update_claimed_amount(env, position_status)?; | ||
|
|
||
| if amount > Uint128::zero() { | ||
| msgs.push(item.item_info.build_send_msg( | ||
| env, | ||
| &env.contract.address, | ||
| to, | ||
| Some(amount), | ||
| )?) | ||
| } | ||
| } | ||
| Ok(msgs) | ||
| } | ||
|
|
||
| pub fn send_fee( | ||
| env: &Env, | ||
| items_info: &Vec<OtcItemInfo>, | ||
| fee_collector: &Addr, | ||
| funds: Vec<Coin>, | ||
| ) -> StdResult<Vec<CosmosMsg>> { | ||
| assert_received_funds(items_info, funds)?; | ||
| build_send_otc_info_items(env, items_info, fee_collector) | ||
| } | ||
|
|
||
| pub fn cancel_otc(env: &Env, position: &OtcPosition) -> StdResult<Vec<CosmosMsg>> { | ||
| build_send_otc_info_items( | ||
| env, | ||
| &position | ||
| .offer | ||
| .iter() | ||
| .map(|val| val.item_info.clone()) | ||
| .collect(), | ||
| &position.owner, | ||
| ) | ||
| } | ||
|
|
||
| pub fn build_send_otc_info_items( | ||
| env: &Env, | ||
| items_info: &Vec<OtcItemInfo>, | ||
| to: &Addr, | ||
| ) -> StdResult<Vec<CosmosMsg>> { | ||
| let mut msgs: Vec<CosmosMsg> = vec![]; | ||
| for item_info in items_info { | ||
| msgs.push(item_info.build_send_msg(env, &env.contract.address, to, None)?) | ||
| } | ||
| Ok(msgs) | ||
| } | ||
|
|
||
| pub fn assert_received_funds(items: &Vec<OtcItemInfo>, funds: Vec<Coin>) -> StdResult<Vec<Coin>> { | ||
| let mut coins = rhaki_cw_plus::coin::vec_coins_to_hashmap(funds)?; | ||
|
|
||
| for item in items { | ||
| if let OtcItemInfo::Token { denom, amount } = &item { | ||
| let available_amount = coins | ||
| .get(denom) | ||
| .ok_or(StdError::generic_err(format!("Coin not received {denom}")))?; | ||
|
|
||
| if amount > available_amount { | ||
| return Err(StdError::generic_err(format!( | ||
| "Amount received for {denom} is to low: expected: {amount}, received: {amount}" | ||
| ))); | ||
| } | ||
|
|
||
| coins.insert(denom.clone(), available_amount - amount); | ||
| } | ||
| } | ||
|
|
||
| Ok(coins | ||
| .into_iter() | ||
| .map(|(denom, amount)| Coin::new(amount.u128(), denom)) | ||
| .collect()) | ||
| } | ||
|
|
||
| pub fn try_close_position( | ||
| deps: DepsMut, | ||
| env: &Env, | ||
| position: &mut OtcPosition, | ||
| ) -> StdResult<Vec<Attribute>> { | ||
| position.try_close(env)?; | ||
|
|
||
| if let OtcPositionStatus::Executed(..) = position.status { | ||
| active_positions().remove(deps.storage, position.id)?; | ||
| execute_positions().save(deps.storage, position.id, position)?; | ||
| return Ok(vec![ | ||
| attr("action", "executed_position"), | ||
| attr("id", position.id.to_string()), | ||
| ]); | ||
| } | ||
|
|
||
| Ok(vec![]) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,6 +1,7 @@ | ||
| #[cfg(not(feature = "library"))] | ||
| pub mod contract; | ||
| mod execute; | ||
| mod functions; | ||
| mod query; | ||
| mod response; | ||
| mod state; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.