feat(smart-contracts): add a query for simulate swap operations

White-Whale-Defi-Platform · Apr 23, 2024 · 61f9235 · 61f9235
1 parent 4abb16e
commit 61f9235
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 14 deletions.
diff --git a/contracts/liquidity_hub/pool-manager/src/contract.rs b/contracts/liquidity_hub/pool-manager/src/contract.rs
@@ -1,4 +1,5 @@
 use crate::error::ContractError;
+use crate::helpers::simulate_swap_operations;
 use crate::queries::{get_swap_route, get_swap_route_creator, get_swap_routes};
 use crate::router::commands::{add_swap_routes, remove_swap_routes};
 use crate::state::{Config, MANAGER_CONFIG, PAIRS, PAIR_COUNTER};
@@ -214,15 +215,14 @@ pub fn query(deps: Deps, env: Env, msg: QueryMsg) -> Result<Binary, ContractErro
             offer_asset,
             pair_identifier,
         )?)?),
-        // QueryMsg::SimulateSwapOperations {
-        //     offer_amount,
-        //     operations,
-        // } => Ok(to_binary(&queries::simulate_swap_operations(
-        //     deps,
-        //     env,
-        //     offer_amount,
-        //     operations,
-        // )?)?),
+        QueryMsg::SimulateSwapOperations {
+            offer_amount,
+            operations,
+        } => Ok(to_json_binary(&simulate_swap_operations(
+            deps,
+            offer_amount,
+            operations,
+        )?)?),
         // QueryMsg::ReverseSimulateSwapOperations {
         //     ask_amount,
         //     operations,

diff --git a/contracts/liquidity_hub/pool-manager/src/helpers.rs b/contracts/liquidity_hub/pool-manager/src/helpers.rs
@@ -599,6 +599,8 @@ pub fn assert_admin(deps: Deps, env: &Env, sender: &Addr) -> Result<(), Contract
     Ok(())
 }
 
+/// This function iterates over the swap operations, simulates each swap
+/// to get the final amount after all the swaps.
 pub fn simulate_swap_operations(
     deps: Deps,
     offer_amount: Uint128,

diff --git a/packages/white-whale-std/src/pool_manager.rs b/packages/white-whale-std/src/pool_manager.rs
@@ -240,11 +240,11 @@ pub enum QueryMsg {
     SwapRoutes {},
 
     // /// Simulates swap operations.
-    // #[returns(SimulateSwapOperationsResponse)]
-    // SimulateSwapOperations {
-    //     offer_amount: Uint128,
-    //     operations: Vec<SwapOperation>,
-    // },
+    #[returns(SimulateSwapOperationsResponse)]
+    SimulateSwapOperations {
+        offer_amount: Uint128,
+        operations: Vec<SwapOperation>,
+    },
     // /// Simulates a reverse swap operations, i.e. given the ask asset, how much of the offer asset
     // /// is needed to perform the swap.
     // #[returns(SimulateSwapOperationsResponse)]