// Copyright (C) Parity Technologies (UK) Ltd. and Dijital Kurdistan Tech Institute
// This file is part of Pezkuwi.

// Pezkuwi 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.

// Pezkuwi 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 Pezkuwi.  If not, see <http://www.gnu.org/licenses/>.

pub use pezpallet::*;

#[cfg(feature = "runtime-benchmarks")]
pub mod benchmarking;
#[cfg(test)]
mod mock;

#[pezframe_support::pezpallet]
pub mod pezpallet {
	use pezframe_benchmarking::BenchmarkError;
	use pezframe_support::{dispatch::GetDispatchInfo, pezpallet_prelude::Encode};
	use pezsp_runtime::traits::Dispatchable;
	use xcm::latest::{
		Asset, Assets, InteriorLocation, Junction, Location, NetworkId, Response, WeightLimit,
	};

	#[pezpallet::config]
	pub trait Config<I: 'static = ()>: pezframe_system::Config + crate::Config {
		type RuntimeCall: Dispatchable<RuntimeOrigin = Self::RuntimeOrigin>
			+ GetDispatchInfo
			+ From<pezframe_system::Call<Self>>
			+ Encode;

		/// The type of `fungible` that is being used under the hood.
		///
		/// This is useful for testing and checking.
		type TransactAsset: pezframe_support::traits::fungible::Mutate<Self::AccountId>;

		///	The response which causes the most runtime weight.
		fn worst_case_response() -> (u64, Response);

		/// The pair of asset collections which causes the most runtime weight if demanded to be
		/// exchanged.
		///
		/// The first element in the returned tuple represents the assets that are being exchanged
		/// from, whereas the second element represents the assets that are being exchanged to.
		///
		/// If set to `Err`, benchmarks which rely on an `exchange_asset` will be skipped.
		fn worst_case_asset_exchange() -> Result<(Assets, Assets), BenchmarkError>;

		/// A `(Location, Junction)` that is one of the `UniversalAliases` configured by the
		/// XCM executor.
		///
		/// If set to `Err`, benchmarks which rely on a universal alias will be skipped.
		fn universal_alias() -> Result<(Location, Junction), BenchmarkError>;

		/// The `Location` and `RuntimeCall` used for successful transaction XCMs.
		///
		/// If set to `Err`, benchmarks which rely on a `transact_origin_and_runtime_call` will be
		/// skipped.
		fn transact_origin_and_runtime_call(
		) -> Result<(Location, <Self as crate::generic::Config<I>>::RuntimeCall), BenchmarkError>;

		/// A valid `Location` we can successfully subscribe to.
		///
		/// If set to `Err`, benchmarks which rely on a `subscribe_origin` will be skipped.
		fn subscribe_origin() -> Result<Location, BenchmarkError>;

		/// Return an origin, ticket, and assets that can be trapped and claimed.
		fn claimable_asset() -> Result<(Location, Location, Assets), BenchmarkError>;

		/// The worst case buy execution weight limit and
		/// asset to trigger the Trader::buy_execution in the XCM executor
		/// Used to buy weight in benchmarks, for example in
		/// `refund_surplus`.
		fn worst_case_for_trader() -> Result<(Asset, WeightLimit), BenchmarkError>;

		/// Return an unlocker, owner and assets that can be locked and unlocked.
		fn unlockable_asset() -> Result<(Location, Location, Asset), BenchmarkError>;

		/// A `(Location, NetworkId, InteriorLocation)` we can successfully export message
		/// to.
		///
		/// If set to `Err`, benchmarks which rely on `export_message` will be skipped.
		fn export_message_origin_and_destination(
		) -> Result<(Location, NetworkId, InteriorLocation), BenchmarkError>;

		/// A `(Location, Location)` that is one of the `Aliasers` configured by the XCM
		/// executor.
		///
		/// If set to `Err`, benchmarks which rely on a universal alias will be skipped.
		fn alias_origin() -> Result<(Location, Location), BenchmarkError>;

		/// Returns a valid pezpallet info for `ExpectPallet` or `QueryPallet` benchmark.
		///
		/// By default returns `pezframe_system::Pezpallet` info with expected pezpallet index `0`.
		fn valid_pallet() -> pezframe_support::traits::PalletInfoData {
			pezframe_support::traits::PalletInfoData {
				index: <pezframe_system::Pezpallet<Self> as pezframe_support::traits::PalletInfoAccess>::index(),
				name: <pezframe_system::Pezpallet<Self> as pezframe_support::traits::PalletInfoAccess>::name(),
				module_name: <pezframe_system::Pezpallet<Self> as pezframe_support::traits::PalletInfoAccess>::module_name(),
				crate_version: <pezframe_system::Pezpallet<Self> as pezframe_support::traits::PalletInfoAccess>::crate_version(),
			}
		}
	}

	#[pezpallet::pezpallet]
	pub struct Pezpallet<T, I = ()>(_);
}