Use cw2::set_contract_version in tfi-factory and tfi-pair

confio · Dec 14, 2021 · ba5f250 · ba5f250
1 parent 4501321
commit ba5f250
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/contracts/tfi-factory/Cargo.toml b/contracts/tfi-factory/Cargo.toml
@@ -25,6 +25,7 @@ backtraces = ["cosmwasm-std/backtraces"]
 [dependencies]
 tfi = { path = "../../packages/tfi", default-features = false, version = "0.2.2"}
 cosmwasm-std = "=1.0.0-beta"
+cw2 = "=0.10"
 cw-storage-plus = "=0.10"
 schemars = "0.8.1"
 serde = { version = "1.0.103", default-features = false, features = ["derive"] }

diff --git a/contracts/tfi-factory/src/contract.rs b/contracts/tfi-factory/src/contract.rs
@@ -4,6 +4,7 @@ use cosmwasm_std::{
     to_binary, Binary, Decimal, Deps, DepsMut, Env, MessageInfo, Reply, Response, StdError,
     StdResult, SubMsg, WasmMsg,
 };
+use cw2::set_contract_version;
 
 use crate::error::ContractError;
 use crate::querier::query_liquidity_token;
@@ -17,13 +18,18 @@ use tfi::factory::{
 };
 use tfi::pair::InstantiateMsg as PairInstantiateMsg;
 
+// version info for migration info
+const CONTRACT_NAME: &str = "crates.io:tfi-factory";
+const CONTRACT_VERSION: &str = env!("CARGO_PKG_VERSION");
+
 #[cfg_attr(not(feature = "library"), entry_point)]
 pub fn instantiate(
     deps: DepsMut,
     _env: Env,
     info: MessageInfo,
     msg: InstantiateMsg,
 ) -> Result<Response, ContractError> {
+    set_contract_version(deps.storage, CONTRACT_NAME, CONTRACT_VERSION)?;
     if !(Decimal::zero()..=Decimal::one()).contains(&msg.default_commission) {
         return Err(ContractError::InvalidCommission(msg.default_commission));
     }

diff --git a/contracts/tfi-pair/src/contract.rs b/contracts/tfi-pair/src/contract.rs
@@ -10,6 +10,7 @@ use cosmwasm_std::{
     Response, StdError, StdResult, SubMsg, Uint128, WasmMsg,
 };
 
+use cw2::set_contract_version;
 use cw20::{Cw20ExecuteMsg, Cw20ReceiveMsg, MinterResponse};
 use integer_sqrt::IntegerSquareRoot;
 use tfi::asset::{Asset, AssetInfo, PairInfo};
@@ -20,13 +21,18 @@ use tfi::pair::{
 use tfi::querier::query_supply;
 use tfi::token::InstantiateMsg as TokenInstantiateMsg;
 
+// version info for migration info
+const CONTRACT_NAME: &str = "crates.io:tfi-pair";
+const CONTRACT_VERSION: &str = env!("CARGO_PKG_VERSION");
+
 #[cfg_attr(not(feature = "library"), entry_point)]
 pub fn instantiate(
     deps: DepsMut,
     env: Env,
     _info: MessageInfo,
     msg: InstantiateMsg,
 ) -> Result<Response, ContractError> {
+    set_contract_version(deps.storage, CONTRACT_NAME, CONTRACT_VERSION)?;
     if !(Decimal::zero()..=Decimal::one()).contains(&msg.commission) {
         return Err(ContractError::InvalidCommission(msg.commission));
     }