feat: Solana Vault Swaps Elections

state-chain/pallets/cf-elections/src/lib.rs

@@ -124,7 +124,7 @@ use frame_system::pallet_prelude::*;
 
 pub use pallet::*;
 
-pub const PALLET_VERSION: StorageVersion = StorageVersion::new(1);
+pub const PALLET_VERSION: StorageVersion = StorageVersion::new(2);
 
 pub use pallet::UniqueMonotonicIdentifier;
 
@@ -492,7 +492,7 @@ pub mod pallet {
 
 	/// Stores persistent state the electoral system needs.
 	#[pallet::storage]
-	pub(crate) type ElectoralUnsynchronisedState<T: Config<I>, I: 'static = ()> = StorageValue<
+	pub type ElectoralUnsynchronisedState<T: Config<I>, I: 'static = ()> = StorageValue<
 		_,
 		<T::ElectoralSystemRunner as ElectoralSystemRunner>::ElectoralUnsynchronisedState,
 		OptionQuery,

state-chain/runtime/src/chainflip/solana_elections.rs

@@ -84,7 +84,7 @@ pub fn initial_state(
 			(),
 			(),
 			(),
-			0,
+			0u32,
 		),
 		unsynchronised_settings: (
 			(),

state-chain/runtime/src/lib.rs

@@ -67,7 +67,7 @@ use cf_traits::{
 };
 use codec::{alloc::string::ToString, Decode, Encode};
 use core::ops::Range;
-use frame_support::{derive_impl, instances::*};
+use frame_support::{derive_impl, instances::*, migrations::VersionedMigration};
 pub use frame_system::Call as SystemCall;
 use pallet_cf_governance::GovCallHash;
 use pallet_cf_ingress_egress::{
@@ -1291,7 +1291,13 @@ type PalletMigrations = (
 	pallet_cf_cfe_interface::migrations::PalletMigration<Runtime>,
 );
 
-type MigrationsForV1_8 = ();
+type MigrationsForV1_8 = VersionedMigration<
+	1,
+	2,
+	migrations::solana_vault_swaps_migration::SolanaVaultSwapsMigration,
+	pallet_cf_elections::Pallet<Runtime, SolanaInstance>,
+	DbWeight,
+>;
 
 #[cfg(feature = "runtime-benchmarks")]
 #[macro_use]

state-chain/runtime/src/migrations.rs

@@ -2,3 +2,4 @@
 
 pub mod housekeeping;
 pub mod reap_old_accounts;
+pub mod solana_vault_swaps_migration;

state-chain/runtime/src/migrations/solana_vault_swaps_migration.rs

@@ -0,0 +1,106 @@
+use crate::*;
+use chainflip::solana_elections::SolanaVaultSwapsSettings;
+use frame_support::{pallet_prelude::Weight, storage::unhashed, traits::UncheckedOnRuntimeUpgrade};
+
+use pallet_cf_elections::{
+	Config, ElectoralSettings, ElectoralSystemRunner, ElectoralUnsynchronisedState,
+};
+#[cfg(feature = "try-runtime")]
+use sp_runtime::DispatchError;
+
+use cf_utilities::bs58_array;
+use codec::{Decode, Encode};
+
+pub struct SolanaVaultSwapsMigration;
+
+impl UncheckedOnRuntimeUpgrade for SolanaVaultSwapsMigration {
+	fn on_runtime_upgrade() -> Weight {
+		let mut raw_unsynchronised_state = unhashed::get_raw(&ElectoralUnsynchronisedState::<
+			Runtime,
+			SolanaInstance,
+		>::hashed_key())
+		.unwrap();
+		raw_unsynchronised_state.extend(0u32.encode());
+		ElectoralUnsynchronisedState::<Runtime, SolanaInstance>::put(<<Runtime as Config<SolanaInstance>>::ElectoralSystemRunner as ElectoralSystemRunner>::ElectoralUnsynchronisedState::decode(&mut &raw_unsynchronised_state[..]).unwrap());
+
+		let (usdc_token_mint_pubkey, swap_endpoint_data_account_address) =
+			match cf_runtime_utilities::genesis_hashes::genesis_hash::<Runtime>() {
+				cf_runtime_utilities::genesis_hashes::BERGHAIN => (
+					SolAddress(bs58_array("EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v")),
+					SolAddress(bs58_array("5mFsKrqCH5v9Q9uF5o6qrsUi1GV2myuhc23NAi5YFs4M")),
+				),
+				cf_runtime_utilities::genesis_hashes::PERSEVERANCE => (
+					SolAddress(bs58_array("4zMMC9srt5Ri5X14GAgXhaHii3GnPAEERYPJgZJDncDU")),
+					SolAddress(bs58_array("4hD7UM6rQtcqQWtzELvrafpmBYReVXvCpssB6qjY1Sg5")),
+				),
+				cf_runtime_utilities::genesis_hashes::SISYPHOS => (
+					SolAddress(bs58_array("4zMMC9srt5Ri5X14GAgXhaHii3GnPAEERYPJgZJDncDU")),
+					SolAddress(bs58_array("mYabVW1uMXpGqwgHUBQu4Fg6GT9EMYUzYaGYbi3zgT7")),
+				),
+				_ => (
+					SolAddress(bs58_array("24PNhTaNtomHhoy3fTRaMhAFCRj4uHqhZEEoWrKDbR5p")),
+					SolAddress(bs58_array("2tmtGLQcBd11BMiE9B1tAkQXwmPNgR79Meki2Eme4Ec9")),
+				),
+			};
+
+		for key in ElectoralSettings::<Runtime, SolanaInstance>::iter_keys() {
+			let mut raw_storage_at_key = unhashed::get_raw(&ElectoralSettings::<
+				Runtime,
+				SolanaInstance,
+			>::hashed_key_for(key))
+			.expect("We just got the keys directly from the storage");
+			raw_storage_at_key.extend(
+				SolanaVaultSwapsSettings {
+					usdc_token_mint_pubkey,
+					swap_endpoint_data_account_address,
+				}
+				.encode(),
+			);
+			ElectoralSettings::<Runtime, SolanaInstance>::insert(key, <<Runtime as Config<SolanaInstance>>::ElectoralSystemRunner as ElectoralSystemRunner>::ElectoralSettings::decode(&mut &raw_storage_at_key[..]).unwrap());
+		}
+
+		Weight::zero()
+	}
+
+	#[cfg(feature = "try-runtime")]
+	fn pre_upgrade() -> Result<Vec<u8>, sp_runtime::TryRuntimeError> {
+		assert!(ElectoralUnsynchronisedState::<Runtime, SolanaInstance>::exists());
+		assert!(ElectoralSettings::<Runtime, SolanaInstance>::iter_keys().next().is_some());
+		Ok(Default::default())
+	}
+
+	#[cfg(feature = "try-runtime")]
+	fn post_upgrade(_state: Vec<u8>) -> Result<(), DispatchError> {
+		let (.., last_block_number) =
+			ElectoralUnsynchronisedState::<Runtime, SolanaInstance>::get().unwrap();
+		assert_eq!(last_block_number, 0u32);
+		for (
+			..,
+			SolanaVaultSwapsSettings { usdc_token_mint_pubkey, swap_endpoint_data_account_address },
+		) in ElectoralSettings::<Runtime, SolanaInstance>::iter_values()
+		{
+			assert_eq!(
+				(usdc_token_mint_pubkey, swap_endpoint_data_account_address),
+				match cf_runtime_utilities::genesis_hashes::genesis_hash::<Runtime>() {
+					cf_runtime_utilities::genesis_hashes::BERGHAIN => (
+						SolAddress(bs58_array("EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v")),
+						SolAddress(bs58_array("5mFsKrqCH5v9Q9uF5o6qrsUi1GV2myuhc23NAi5YFs4M")),
+					),
+					cf_runtime_utilities::genesis_hashes::PERSEVERANCE => (
+						SolAddress(bs58_array("4zMMC9srt5Ri5X14GAgXhaHii3GnPAEERYPJgZJDncDU")),
+						SolAddress(bs58_array("4hD7UM6rQtcqQWtzELvrafpmBYReVXvCpssB6qjY1Sg5")),
+					),
+					cf_runtime_utilities::genesis_hashes::SISYPHOS => (
+						SolAddress(bs58_array("4zMMC9srt5Ri5X14GAgXhaHii3GnPAEERYPJgZJDncDU")),
+						SolAddress(bs58_array("mYabVW1uMXpGqwgHUBQu4Fg6GT9EMYUzYaGYbi3zgT7")),
+					),
+					_ => (
+						SolAddress(bs58_array("24PNhTaNtomHhoy3fTRaMhAFCRj4uHqhZEEoWrKDbR5p")),
+						SolAddress(bs58_array("2tmtGLQcBd11BMiE9B1tAkQXwmPNgR79Meki2Eme4Ec9")),
+					),
+				}
+			);
+		}
+		Ok(())
+	}
+}