|
| 1 | +/////////////////////////////////////////////////////////////////////////////// |
| 2 | +// |
| 3 | +// Copyright 2018-2024 Robonomics Network <[email protected]> |
| 4 | +// |
| 5 | +// Licensed under the Apache License, Version 2.0 (the "License"); |
| 6 | +// you may not use this file except in compliance with the License. |
| 7 | +// You may obtain a copy of the License at |
| 8 | +// |
| 9 | +// http://www.apache.org/licenses/LICENSE-2.0 |
| 10 | +// |
| 11 | +// Unless required by applicable law or agreed to in writing, software |
| 12 | +// distributed under the License is distributed on an "AS IS" BASIS, |
| 13 | +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 14 | +// See the License for the specific language governing permissions and |
| 15 | +// limitations under the License. |
| 16 | +// |
| 17 | +/////////////////////////////////////////////////////////////////////////////// |
| 18 | +//! On-chain XCM setup & information. |
| 19 | +#![cfg_attr(not(feature = "std"), no_std)] |
| 20 | + |
| 21 | +pub use pallet::*; |
| 22 | + |
| 23 | +#[frame_support::pallet] |
| 24 | +pub mod pallet { |
| 25 | + use super::*; |
| 26 | + use frame_support::pallet_prelude::*; |
| 27 | + use frame_system::{ensure_root, pallet_prelude::*}; |
| 28 | + use sp_runtime::traits::MaybeEquivalence; |
| 29 | + use xcm::latest::prelude::*; |
| 30 | + |
| 31 | + #[pallet::config] |
| 32 | + pub trait Config: frame_system::Config { |
| 33 | + /// AssetId type for asset<>location linkage setup. |
| 34 | + type AssetId: Parameter + Copy + Default + MaxEncodedLen; |
| 35 | + /// The overarching event type. |
| 36 | + type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>; |
| 37 | + } |
| 38 | + |
| 39 | + #[pallet::error] |
| 40 | + pub enum Error<T> {} |
| 41 | + |
| 42 | + #[pallet::event] |
| 43 | + #[pallet::generate_deposit(pub(super) fn deposit_event)] |
| 44 | + pub enum Event<T: Config> { |
| 45 | + /// Updated Relay XCM identifier. |
| 46 | + RelayNetworkChanged(NetworkId), |
| 47 | + /// Added new asset XCM location. |
| 48 | + AssetLinkAdded(T::AssetId, MultiLocation), |
| 49 | + } |
| 50 | + |
| 51 | + #[pallet::pallet] |
| 52 | + pub struct Pallet<T>(PhantomData<T>); |
| 53 | + |
| 54 | + /// Relay network identifier. |
| 55 | + #[pallet::storage] |
| 56 | + #[pallet::getter(fn relay_network)] |
| 57 | + pub(super) type RelayNetwork<T> = StorageValue<_, NetworkId>; |
| 58 | + |
| 59 | + /// AssetId to location mapping. |
| 60 | + #[pallet::storage] |
| 61 | + #[pallet::getter(fn location_of)] |
| 62 | + pub(super) type LocationOf<T: Config> = |
| 63 | + StorageMap<_, Blake2_128Concat, T::AssetId, MultiLocation>; |
| 64 | + |
| 65 | + /// Location to AssetId mapping. |
| 66 | + #[pallet::storage] |
| 67 | + #[pallet::getter(fn assetid_of)] |
| 68 | + pub(super) type AssetIdOf<T: Config> = |
| 69 | + StorageMap<_, Blake2_128Concat, MultiLocation, T::AssetId>; |
| 70 | + |
| 71 | + #[pallet::call] |
| 72 | + impl<T: Config> Pallet<T> { |
| 73 | + #[pallet::call_index(0)] |
| 74 | + #[pallet::weight(10_000)] |
| 75 | + pub fn set_relay_network(origin: OriginFor<T>, network_id: NetworkId) -> DispatchResult { |
| 76 | + ensure_root(origin)?; |
| 77 | + |
| 78 | + <RelayNetwork<T>>::put(network_id); |
| 79 | + Self::deposit_event(Event::<T>::RelayNetworkChanged(network_id)); |
| 80 | + |
| 81 | + Ok(()) |
| 82 | + } |
| 83 | + |
| 84 | + #[pallet::call_index(1)] |
| 85 | + #[pallet::weight(10_000)] |
| 86 | + pub fn set_asset_link( |
| 87 | + origin: OriginFor<T>, |
| 88 | + asset_id: T::AssetId, |
| 89 | + location: MultiLocation, |
| 90 | + ) -> DispatchResult { |
| 91 | + ensure_root(origin)?; |
| 92 | + |
| 93 | + <LocationOf<T>>::insert(asset_id, location); |
| 94 | + <AssetIdOf<T>>::insert(location, asset_id); |
| 95 | + Self::deposit_event(Event::<T>::AssetLinkAdded(asset_id, location)); |
| 96 | + |
| 97 | + Ok(()) |
| 98 | + } |
| 99 | + } |
| 100 | + |
| 101 | + impl<T: Config> MaybeEquivalence<MultiLocation, T::AssetId> for Pallet<T> { |
| 102 | + fn convert(id: &MultiLocation) -> Option<T::AssetId> { |
| 103 | + <AssetIdOf<T>>::get(id) |
| 104 | + } |
| 105 | + fn convert_back(what: &T::AssetId) -> Option<MultiLocation> { |
| 106 | + <LocationOf<T>>::get(what) |
| 107 | + } |
| 108 | + } |
| 109 | +} |
0 commit comments