Contracts: Use RuntimeUpgrade hooks instead of Hooks::on_runtime_upgrade

pallets/dmp-queue/src/migration.rs

@@ -19,12 +19,12 @@
 use crate::{Config, Configuration, Overweight, Pallet, DEFAULT_POV_SIZE};
 use frame_support::{
 	pallet_prelude::*,
-	traits::StorageVersion,
+	traits::{OnRuntimeUpgrade, StorageVersion},
 	weights::{constants::WEIGHT_REF_TIME_PER_MILLIS, Weight},
 };
 
 /// The current storage version.
-pub const STORAGE_VERSION: StorageVersion = StorageVersion::new(1);
+pub const STORAGE_VERSION: StorageVersion = StorageVersion::new(2);
 
 /// Migrates the pallet storage to the most recent version, checking and setting the
 /// `StorageVersion`.
@@ -46,6 +46,13 @@ pub fn migrate_to_latest<T: Config>() -> Weight {
 	weight
 }
 
+pub struct Migration<T: Config>(PhantomData<T>);
+impl<T: Config> OnRuntimeUpgrade for Migration<T> {
+	fn on_runtime_upgrade() -> Weight {
+		migrate_to_latest::<T>()
+	}
+}
+
 mod v0 {
 	use super::*;
 	use codec::{Decode, Encode};

pallets/parachain-system/src/lib.rs

@@ -57,7 +57,7 @@ use sp_runtime::{
 use sp_std::{cmp, collections::btree_map::BTreeMap, prelude::*};
 use xcm::latest::XcmHash;
 
-mod migration;
+pub mod migration;
 mod relay_state_snapshot;
 #[macro_use]
 pub mod validate_block;

pallets/parachain-system/src/migration.rs

@@ -16,7 +16,8 @@
 
 use crate::{Config, Pallet, ReservedDmpWeightOverride, ReservedXcmpWeightOverride};
 use frame_support::{
-	traits::{Get, StorageVersion},
+	pallet_prelude::*,
+	traits::{Get, OnRuntimeUpgrade, StorageVersion},
 	weights::Weight,
 };
 
@@ -44,6 +45,13 @@ pub fn on_runtime_upgrade<T: Config>() -> Weight {
 	weight
 }
 
+pub struct Migration<T: Config>(PhantomData<T>);
+impl<T: Config> OnRuntimeUpgrade for Migration<T> {
+	fn on_runtime_upgrade() -> Weight {
+		on_runtime_upgrade::<T>()
+	}
+}
+
 /// V2: Migrate to 2D weights for ReservedXcmpWeightOverride and ReservedDmpWeightOverride.
 mod v2 {
 	use super::*;

pallets/xcmp-queue/src/migration.rs

@@ -19,12 +19,12 @@
 use crate::{Config, Overweight, Pallet, QueueConfig, DEFAULT_POV_SIZE};
 use frame_support::{
 	pallet_prelude::*,
-	traits::StorageVersion,
+	traits::{OnRuntimeUpgrade, StorageVersion},
 	weights::{constants::WEIGHT_REF_TIME_PER_MILLIS, Weight},
 };
 
 /// The current storage version.
-pub const STORAGE_VERSION: StorageVersion = StorageVersion::new(2);
+pub const STORAGE_VERSION: StorageVersion = StorageVersion::new(3);
 
 /// Migrates the pallet storage to the most recent version, checking and setting the
 /// `StorageVersion`.
@@ -46,6 +46,13 @@ pub fn migrate_to_latest<T: Config>() -> Weight {
 	weight
 }
 
+pub struct Migration<T: Config>(PhantomData<T>);
+impl<T: Config> OnRuntimeUpgrade for Migration<T> {
+	fn on_runtime_upgrade() -> Weight {
+		migrate_to_latest::<T>()
+	}
+}
+
 mod v1 {
 	use super::*;
 	use codec::{Decode, Encode};

parachains/runtimes/contracts/contracts-rococo/Cargo.toml

@@ -86,6 +86,7 @@ std = [
 	"frame-support/std",
 	"frame-system-rpc-runtime-api/std",
 	"frame-system/std",
+    "frame-try-runtime/std",
 	"kusama-runtime-constants/std",
 	"pallet-aura/std",
 	"pallet-authorship/std",

parachains/runtimes/contracts/contracts-rococo/src/lib.rs

@@ -97,7 +97,15 @@ pub type UncheckedExtrinsic =
 /// Extrinsic type that has already been checked.
 pub type CheckedExtrinsic = generic::CheckedExtrinsic<AccountId, RuntimeCall, SignedExtra>;
 
-pub type Migrations = (pallet_contracts::Migration<Runtime>,);
+pub type Migrations = (
+	cumulus_pallet_parachain_system::migration::Migration<Runtime>,
+	cumulus_pallet_xcmp_queue::migration::Migration<Runtime>,
+	pallet_balances::migration::MigrateToTrackInactive<Runtime, xcm_config::CheckingAccount>,
+	pallet_multisig::migrations::v1::MigrateToV1<Runtime>,
+	pallet_contracts::Migration<Runtime>,
+	pallet_xcm::migration::v1::MigrateToV1<Runtime>,
+	cumulus_pallet_dmp_queue::migration::Migration<Runtime>,
+);
 
 type EventRecord = frame_system::EventRecord<
 	<Runtime as frame_system::Config>::RuntimeEvent,

parachains/runtimes/contracts/contracts-rococo/src/xcm_config.rs

@@ -43,6 +43,7 @@ parameter_types! {
 	pub RelayChainOrigin: RuntimeOrigin = cumulus_pallet_xcm::Origin::Relay.into();
 	pub UniversalLocation: InteriorMultiLocation = Parachain(ParachainInfo::parachain_id().into()).into();
 	pub const ExecutiveBody: BodyId = BodyId::Executive;
+	pub CheckingAccount: AccountId = PolkadotXcm::check_account();
 }
 
 /// We allow root and the Relay Chain council to execute privileged collator selection operations.