Fabo/migrate

contracts/staking/Cargo.toml

@@ -42,6 +42,7 @@ regex = "1.10.2"
 prost = {version = "0.11.2", default-features = false, features = ["prost-derive"]}
 enum-repr = "0.2.6"
 prost-derive = "0.12.3"
+semver = "1.0.20"
 
 [dev-dependencies]
 cw-multi-test = "0.17.0"

contracts/staking/src/contract.rs

@@ -22,14 +22,15 @@ use crate::{
     msg::{ExecuteMsg, IBCLifecycleComplete, InstantiateMsg, MigrateMsg, QueryMsg, SudoMsg},
 };
 use cosmwasm_std::{
-    entry_point, to_binary, Binary, Deps, DepsMut, Env, MessageInfo, Reply, Response, StdResult,
-    Uint128,
+    entry_point, to_binary, Binary, Deps, DepsMut, Env, MessageInfo, Reply, Response, StdError,
+    StdResult, Uint128,
 };
 use cosmwasm_std::{CosmosMsg, Timestamp};
 use cw2::set_contract_version;
 use cw_utils::must_pay;
 use milky_way::staking::Batch;
 use osmosis_std::types::osmosis::tokenfactory::v1beta1::MsgCreateDenom;
+use semver::Version;
 
 // Version information for migration
 const CONTRACT_NAME: &str = env!("CARGO_PKG_NAME");
@@ -232,9 +233,35 @@ pub fn query(deps: Deps, _env: Env, msg: QueryMsg) -> StdResult<Binary> {
 /// MIGRATE ///
 ///////////////
 
-#[entry_point]
-pub fn migrate(_deps: DepsMut, _env: Env, _msg: MigrateMsg) -> StdResult<Response> {
-    // TODO: note implement yet
+#[cfg_attr(not(feature = "library"), entry_point)]
+pub fn migrate(deps: DepsMut, _env: Env, _msg: MigrateMsg) -> Result<Response, ContractError> {
+    let current_version = cw2::get_contract_version(deps.storage)?;
+    if &CONTRACT_NAME != &current_version.contract.as_str() {
+        return Err(StdError::generic_err("Cannot upgrade to a different contract").into());
+    }
+
+    let version: Version = current_version
+        .version
+        .parse()
+        .map_err(|_| StdError::generic_err("Invalid contract version"))?;
+    let new_version: Version = CONTRACT_VERSION
+        .parse()
+        .map_err(|_| StdError::generic_err("Invalid contract version"))?;
+
+    // current version not launchpad v2
+    if version > new_version {
+        return Err(StdError::generic_err("Cannot upgrade to a previous contract version").into());
+    }
+    // if same version return
+    if version == new_version {
+        return Ok(Response::new());
+    }
+
+    // migrate data
+    // NONE currently
+
+    // set new contract version
+    set_contract_version(deps.storage, CONTRACT_NAME, CONTRACT_VERSION)?;
     Ok(Response::new())
 }
 

scripts/deploy/migrate_stake_contract_testnet.sh

@@ -0,0 +1,14 @@
+docker run --rm -v "$(pwd)":/code \
+  --mount type=volume,source="$(basename "$(pwd)")_cache",target=/code/target \
+  --mount type=volume,source=registry_cache,target=/usr/local/cargo/registry \
+  cosmwasm/workspace-optimizer-arm64:0.14.0
+
+# double check contract address https://decento-labs.slack.com/canvas/C04R92P76L9
+CONTRACT="osmo1h3s40qhqztppucdfggzwykz7xgwhcnps952y86xc73eupqfu9jmqcaghze"
+RES=$(osmosisd tx wasm store ./artifacts/staking-aarch64.wasm --from test_master --output json -y -b block --gas-prices 0.025stake --gas-adjustment 1.7 --gas auto --chain-id osmosis-dev-1)
+CODE_ID=$(echo $RES | jq -r '.logs[0].events[-1].attributes[-1].value')
+osmosisd tx wasm migrate $CONTRACT $CODE_ID '{}' \
+    --from test_master -y \
+    -b block --node https://rpc.testnet.osmosis.zone:443 \
+    --gas-prices 0.025uosmo --gas-adjustment 1.7 --gas auto  \
+    --chain-id osmo-test-5