Remove debt from total deposit. (#351)

* Remove debt from total deposit.

* Add migration and use proper version.

Author: piobab

Cargo.lock

@@ -1,7 +1,7 @@
 [package]
 name          = "mars-params"
 description   = "Contract storing the asset params for Credit Manager and Red Bank."
-version       = { workspace = true }
+version       = "2.0.1"
 authors       = { workspace = true }
 license       = { workspace = true }
 edition       = { workspace = true }

contracts/params/Cargo.toml

@@ -1,6 +1,6 @@
 #[cfg(not(feature = "library"))]
 use cosmwasm_std::entry_point;
-use cosmwasm_std::{to_binary, Binary, Deps, DepsMut, Env, MessageInfo, Response};
+use cosmwasm_std::{to_binary, Binary, Deps, DepsMut, Empty, Env, MessageInfo, Response};
 use cw2::set_contract_version;
 use mars_owner::OwnerInit::SetInitialOwner;
 use mars_types::params::{
@@ -15,15 +15,16 @@ use crate::{
         assert_thf, update_asset_params, update_config, update_target_health_factor,
         update_vault_config,
     },
+    migrations,
     query::{
         query_all_asset_params, query_all_vault_configs, query_config, query_total_deposit,
         query_vault_config,
     },
     state::{ADDRESS_PROVIDER, ASSET_PARAMS, OWNER, TARGET_HEALTH_FACTOR},
 };
 
-const CONTRACT_NAME: &str = env!("CARGO_PKG_NAME");
-const CONTRACT_VERSION: &str = env!("CARGO_PKG_VERSION");
+pub const CONTRACT_NAME: &str = env!("CARGO_PKG_NAME");
+pub const CONTRACT_VERSION: &str = env!("CARGO_PKG_VERSION");
 
 #[cfg_attr(not(feature = "library"), entry_point)]
 pub fn instantiate(
@@ -109,3 +110,8 @@ pub fn query(deps: Deps, env: Env, msg: QueryMsg) -> ContractResult<Binary> {
     };
     res.map_err(Into::into)
 }
+
+#[cfg_attr(not(feature = "library"), entry_point)]
+pub fn migrate(deps: DepsMut, _env: Env, _msg: Empty) -> ContractResult<Response> {
+    migrations::v2_0_1::migrate(deps)
+}

contracts/params/src/contract.rs

@@ -1,4 +1,5 @@
 use cosmwasm_std::{DecimalRangeExceeded, StdError};
+use cw2::VersionError;
 use mars_owner::OwnerError;
 use mars_types::error::MarsError;
 use mars_utils::error::ValidationError;
@@ -22,4 +23,7 @@ pub enum ContractError {
 
     #[error("{0}")]
     Mars(#[from] MarsError),
+
+    #[error("{0}")]
+    Version(#[from] VersionError),
 }

contracts/params/src/error.rs

@@ -2,5 +2,6 @@ pub mod contract;
 pub mod emergency_powers;
 pub mod error;
 pub mod execute;
+pub mod migrations;
 pub mod query;
 pub mod state;

contracts/params/src/lib.rs

@@ -0,0 +1 @@
+pub mod v2_0_1;

contracts/params/src/migrations/mod.rs

@@ -0,0 +1,21 @@
+use cosmwasm_std::{DepsMut, Response};
+use cw2::{assert_contract_version, set_contract_version};
+
+use crate::{
+    contract::{CONTRACT_NAME, CONTRACT_VERSION},
+    error::ContractError,
+};
+
+const FROM_VERSION: &str = "2.0.0";
+
+pub fn migrate(deps: DepsMut) -> Result<Response, ContractError> {
+    // make sure we're migrating the correct contract and from the correct version
+    assert_contract_version(deps.storage, &format!("crates.io:{CONTRACT_NAME}"), FROM_VERSION)?;
+
+    set_contract_version(deps.storage, format!("crates.io:{CONTRACT_NAME}"), CONTRACT_VERSION)?;
+
+    Ok(Response::new()
+        .add_attribute("action", "migrate")
+        .add_attribute("from_version", FROM_VERSION)
+        .add_attribute("to_version", CONTRACT_VERSION))
+}

contracts/params/src/migrations/v2_0_1.rs

@@ -4,7 +4,7 @@ use mars_interest_rate::get_underlying_liquidity_amount;
 use mars_types::{
     address_provider::{self, MarsAddressType},
     params::{AssetParams, ConfigResponse, TotalDepositResponse, VaultConfig},
-    red_bank::{self, Market, UserDebtResponse},
+    red_bank::{self, Market},
 };
 
 use crate::state::{ADDRESS_PROVIDER, ASSET_PARAMS, VAULT_CONFIGS};
@@ -115,26 +115,13 @@ pub fn query_total_deposit(
         .transpose()?
         .unwrap_or_else(Uint128::zero);
 
-    // amount of debt in this asset the Credit Manager owes to Red Bank
-    // this query returns zero if no debt is owed
-    let cm_debt = deps
-        .querier
-        .query_wasm_smart::<UserDebtResponse>(
-            red_bank_addr,
-            &red_bank::QueryMsg::UserDebt {
-                user: credit_manager_addr.into(),
-                denom: denom.clone(),
-            },
-        )?
-        .amount;
-
     // amount of this asset deposited into Credit Manager
     // this is simply the coin balance of the CM contract
     // note that this way, we don't include LP tokens or vault positions
     let cm_deposit = deps.querier.query_balance(credit_manager_addr, &denom)?.amount;
 
     // total deposited amount
-    let amount = rb_deposit.checked_add(cm_deposit)?.checked_sub(cm_debt)?;
+    let amount = rb_deposit.checked_add(cm_deposit)?;
 
     // additionally, we include the deposit cap in the response
     let asset_params = ASSET_PARAMS.load(deps.storage, &denom)?;

contracts/params/src/query.rs

@@ -3,6 +3,7 @@ mod helpers;
 mod test_asset_validation;
 mod test_deposit_cap;
 mod test_emergency_powers;
+mod test_migration;
 mod test_owner;
 mod test_target_health_factor;
 mod test_update_asset_params;

contracts/params/tests/tests/mod.rs

@@ -62,7 +62,7 @@ fn querying_total_deposit(rb_market: Market, rb_debt: UserDebtResponse, cm_balan
 
     // setup
     deps.querier.set_redbank_market(rb_market.clone());
-    deps.querier.set_red_bank_user_debt(CREDIT_MANAGER, rb_debt.clone());
+    deps.querier.set_red_bank_user_debt(CREDIT_MANAGER, rb_debt);
     deps.querier.update_balances(CREDIT_MANAGER, coins(cm_balance.u128(), MOCK_DENOM));
     ADDRESS_PROVIDER.save(deps.as_mut().storage, &Addr::unchecked("address_provider")).unwrap();
     ASSET_PARAMS.save(deps.as_mut().storage, MOCK_DENOM, &params).unwrap();
@@ -71,7 +71,7 @@ fn querying_total_deposit(rb_market: Market, rb_debt: UserDebtResponse, cm_balan
     let rb_deposit =
         get_underlying_liquidity_amount(rb_market.collateral_total_scaled, &rb_market, TIMESTAMP)
             .unwrap();
-    let exp_total_deposit = rb_deposit + cm_balance - rb_debt.amount;
+    let exp_total_deposit = rb_deposit + cm_balance;
 
     // query total deposit
     let res = query_total_deposit(deps.as_ref(), &env, MOCK_DENOM.into()).unwrap();