// Copyright 2019-2025 PureStake Inc.

// This file is part of Moonbeam.

// Moonbeam is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

// Moonbeam is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License

// along with Moonbeam.  If not, see <http://www.gnu.org/licenses/>.

extern crate alloc;

#[cfg(feature = "try-runtime")]

use alloc::vec::Vec;

use frame_support::{traits::OnRuntimeUpgrade, weights::Weight};

use crate::*;

#[derive(

	Clone,

	PartialEq,

	Eq,

	parity_scale_codec::Decode,

	parity_scale_codec::Encode,

)]

/// Reserve information { account, percent_of_inflation }

pub struct OldParachainBondConfig<AccountId> {

	/// Account which receives funds intended for parachain bond

	pub account: AccountId,

	/// Percent of inflation set aside for parachain bond account

	pub percent: sp_runtime::Percent,

}

pub struct MigrateParachainBondConfig<T>(sp_std::marker::PhantomData<T>);

impl<T: Config> OnRuntimeUpgrade for MigrateParachainBondConfig<T> {

		{

		} else {

		};

	#[cfg(feature = "try-runtime")]

	fn pre_upgrade() -> Result<Vec<u8>, sp_runtime::DispatchError> {

		use frame_support::ensure;

		use parity_scale_codec::Encode;

		let state = frame_support::storage::migration::get_storage_value::<

			OldParachainBondConfig<T::AccountId>,

		>(b"ParachainStaking", b"ParachainBondInfo", &[]);

		ensure!(state.is_some(), "State not found");

		Ok(state

			.expect("should be Some(_) due to former call to ensure!")

			.encode())

	}

	#[cfg(feature = "try-runtime")]

	fn post_upgrade(state: Vec<u8>) -> Result<(), sp_runtime::DispatchError> {

		use frame_support::ensure;

		let old_state: OldParachainBondConfig<T::AccountId> =

			parity_scale_codec::Decode::decode(&mut &state[..])

				.map_err(|_| sp_runtime::DispatchError::Other("Failed to decode old state"))?;

		let new_state = InflationDistributionInfo::<T>::get();

		let pbr = InflationDistributionAccount {

			account: old_state.account,

			percent: old_state.percent,

		};

		let treasury = InflationDistributionAccount::<T::AccountId>::default();

		let expected_new_state: InflationDistributionConfig<T::AccountId> = [pbr, treasury].into();

		ensure!(new_state == expected_new_state, "State migration failed");

		Ok(())

	}

}