From 68032840d6ac0e7e3d41ada4de8ab1cea5f7df3e Mon Sep 17 00:00:00 2001
From: joe petrowski <25483142+joepetrowski@users.noreply.github.com>
Date: Sat, 8 Jun 2024 20:14:31 +0200
Subject: [PATCH] Use `authorize_upgrade` (#27)

* kusama

* fmt

* polkadot

* people too
---
 src/build_upgrade.rs | 40 ++++++++++++++++------------------------
 1 file changed, 16 insertions(+), 24 deletions(-)

diff --git a/src/build_upgrade.rs b/src/build_upgrade.rs
index 845a39e..8204412 100644
--- a/src/build_upgrade.rs
+++ b/src/build_upgrade.rs
@@ -281,7 +281,7 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 		match chain.network {
 			Network::Kusama | Network::Polkadot => continue, // do relay chain separately
 			Network::KusamaAssetHub => {
-				use kusama_asset_hub::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use kusama_asset_hub::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}asset-hub-kusama_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -291,15 +291,14 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Kusama Asset Hub Runtime Hash:   0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::KusamaAssetHub(
-					KusamaAssetHubRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					KusamaAssetHubRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);
 			},
 			Network::KusamaBridgeHub => {
-				use kusama_bridge_hub::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use kusama_bridge_hub::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}bridge-hub-kusama_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -309,15 +308,14 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Kusama Bridge Hub Runtime Hash:  0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::KusamaBridgeHub(
-					KusamaBridgeHubRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					KusamaBridgeHubRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);
 			},
 			Network::KusamaPeople => {
-				use kusama_people::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use kusama_people::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}people-kusama_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -327,15 +325,14 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Kusama People Runtime Hash:      0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::KusamaPeople(
-					KusamaPeopleRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					KusamaPeopleRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);
 			},
 			Network::KusamaCoretime => {
-				use kusama_coretime::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use kusama_coretime::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}coretime-kusama_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -345,15 +342,14 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Kusama Coretime Runtime Hash:    0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::KusamaCoretime(
-					KusamaCoretimeRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					KusamaCoretimeRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);
 			},
 			Network::KusamaEncointer => {
-				use kusama_encointer::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use kusama_encointer::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}encointer-kusama_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -363,15 +359,14 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Kusama Encointer Runtime Hash:   0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::KusamaEncointer(
-					KusamaEncointerRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					KusamaEncointerRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);
 			},
 			Network::PolkadotAssetHub => {
-				use polkadot_asset_hub::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use polkadot_asset_hub::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}asset-hub-polkadot_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -381,15 +376,14 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Polkadot Asset Hub Runtime Hash:   0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::PolkadotAssetHub(
-					PolkadotAssetHubRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					PolkadotAssetHubRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);
 			},
 			Network::PolkadotCollectives => {
-				use polkadot_collectives::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use polkadot_collectives::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}collectives-polkadot_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -399,15 +393,14 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Polkadot Collectives Runtime Hash: 0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::PolkadotCollectives(
-					CollectivesRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					CollectivesRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);
 			},
 			Network::PolkadotBridgeHub => {
-				use polkadot_bridge_hub::runtime_types::cumulus_pallet_parachain_system::pallet::Call;
+				use polkadot_bridge_hub::runtime_types::frame_system::pallet::Call;
 				let path = format!(
 					"{}bridge-hub-polkadot_runtime-v{}.compact.compressed.wasm",
 					upgrade_details.directory, runtime_version
@@ -417,9 +410,8 @@ fn generate_authorize_upgrade_calls(upgrade_details: &UpgradeDetails) -> Vec<Cal
 				println!("Polkadot Bridge Hub Runtime Hash:  0x{}", hex::encode(runtime_hash));
 
 				let call = CallInfo::from_runtime_call(NetworkRuntimeCall::PolkadotBridgeHub(
-					PolkadotBridgeHubRuntimeCall::ParachainSystem(Call::authorize_upgrade {
+					PolkadotBridgeHubRuntimeCall::System(Call::authorize_upgrade {
 						code_hash: H256(runtime_hash),
-						check_version: true,
 					}),
 				));
 				authorization_calls.push(call);