1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

// Copyright 2025 Moonbeam foundation
// 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/>.

use core::marker::PhantomData;

use frame_support::traits::{
	fungible::{self, NativeOrWithId},
	tokens::ConversionFromAssetBalance,
};
use moonbeam_core_primitives::{AssetId, Balance};
use pallet_xcm_weight_trader::RELATIVE_PRICE_DECIMALS;
use sp_runtime::traits::MaybeEquivalence;

pub struct AssetRateConverter<T, NativeAsset>(PhantomData<(T, NativeAsset)>);
impl<
		T: frame_system::Config
			+ pallet_xcm_weight_trader::Config
			+ pallet_moonbeam_foreign_assets::Config,
		NativeAsset: fungible::Mutate<T::AccountId> + fungible::Inspect<T::AccountId>,
	> ConversionFromAssetBalance<Balance, NativeOrWithId<AssetId>, Balance>
	for AssetRateConverter<T, NativeAsset>
{
	type Error = pallet_xcm_weight_trader::Error<T>;

	fn from_asset_balance(
		balance: Balance,
		asset_kind: NativeOrWithId<AssetId>,
	) -> Result<Balance, Self::Error> {
		match asset_kind {
			NativeOrWithId::Native => Ok(balance),
			NativeOrWithId::WithId(asset_id) => {
				let location = pallet_moonbeam_foreign_assets::Pallet::<T>::convert_back(&asset_id)
					.ok_or(pallet_xcm_weight_trader::Error::<T>::AssetNotFound)?;
				let relative_price =
					pallet_xcm_weight_trader::Pallet::<T>::get_asset_relative_price(&location)
						.ok_or(pallet_xcm_weight_trader::Error::<T>::AssetNotFound)?;
				Ok(balance
					.checked_mul(relative_price)
					.ok_or(pallet_xcm_weight_trader::Error::<T>::PriceOverflow)?
					.checked_div(10u128.pow(RELATIVE_PRICE_DECIMALS))
					.ok_or(pallet_xcm_weight_trader::Error::<T>::PriceOverflow)?)
			}
		}
	}

	/// Set a conversion rate to `1` for the `asset_id`.
	#[cfg(feature = "runtime-benchmarks")]
	fn ensure_successful(asset_id: NativeOrWithId<AssetId>) {
		use frame_support::traits::OriginTrait;
		use xcm::latest::{Junction::Parachain, Location};
		match asset_id {
			NativeOrWithId::Native => (),
			NativeOrWithId::WithId(asset_id) => {
				if let None = pallet_moonbeam_foreign_assets::Pallet::<T>::convert_back(&asset_id) {
					let location = Location::new(1, [Parachain(1000)]);
					let root = <T as frame_system::Config>::RuntimeOrigin::root();

					use sp_std::vec;
					pallet_moonbeam_foreign_assets::Pallet::<T>::do_create_asset(
						asset_id,
						location.clone(),
						12,
						vec![b'M', b'T'].try_into().expect("invalid ticker"),
						vec![b'M', b'y', b'T', b'o', b'k']
							.try_into()
							.expect("invalid name"),
						None,
					)
					.expect("Failed to create foreign asset");

					pallet_xcm_weight_trader::Pallet::<T>::add_asset(root, location.clone(), 1u128)
						.expect("Could not add asset");
				}
			}
		}
	}
}