// 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;

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(())

	}

}

/// Migration to move `DelegationScheduledRequests` from a single `StorageMap` keyed by collator

/// into a `StorageDoubleMap` keyed by (collator, delegator) and to initialize the per-collator

/// counter `DelegationScheduledRequestsPerCollator`.

///

/// This assumes the on-chain data was written with the old layout where:

/// - Storage key: ParachainStaking::DelegationScheduledRequests

/// - Value type: BoundedVec<ScheduledRequest<..>, AddGet<MaxTop, MaxBottom>>

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

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

		use frame_support::storage::migration::storage_key_iter;

		type OldScheduledRequests<T> = frame_support::BoundedVec<

			ScheduledRequest<<T as frame_system::Config>::AccountId, BalanceOf<T>>,

			AddGet<

				<T as pallet::Config>::MaxTopDelegationsPerCandidate,

				<T as pallet::Config>::MaxBottomDelegationsPerCandidate,

			>,

		>;

		// Collect all existing entries under the old layout.

			<T as frame_system::Config>::AccountId,

			OldScheduledRequests<T>,

			frame_support::Blake2_128Concat,

		// Clear all existing keys for DelegationScheduledRequests to avoid mixing

		// old layout entries with the new double-map layout.

		//

		// We use the low-level `sp_io::storage::clear_prefix` on the full storage prefix so

		// that *all* existing keys under `ParachainStaking::DelegationScheduledRequests`

		// (from the legacy single-map layout) are removed before we insert the new keys for

		// the double-map layout.

		// Rebuild storage using the new layout and initialize the per-collator counters.

					// This should not happen since we only ever push a single element.

			}

		}

}