Merge pull request #75 from bifrost-finance/add-control-origin

Add ControlOrigin
zenlinkpro · Oct 13, 2023 · 6094173 · 6094173
2 parents 6c546a9 + 1b8da47
commit 6094173
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/example/runtime/src/zenlink.rs b/example/runtime/src/zenlink.rs
@@ -33,6 +33,7 @@ impl zenlink_protocol::Config for Runtime {
 	type TargetChains = ();
 	type SelfParaId = SelfParaId;
 	type WeightInfo = ();
+	type ControlOrigin = EnsureRoot<AccountId>;
 }
 
 type MultiAssets = ZenlinkMultiAssets<ZenlinkProtocol, Balances, LocalAssetAdaptor<Tokens>>;

diff --git a/zenlink-protocol/src/lib.rs b/zenlink-protocol/src/lib.rs
@@ -80,6 +80,8 @@ pub mod pallet {
 	pub trait Config: frame_system::Config {
 		/// Because this pallet emits events, it depends on the runtime's definition of an event.
 		type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;
+		/// The only origin that can create pair.
+		type ControlOrigin: EnsureOrigin<<Self as frame_system::Config>::RuntimeOrigin>;
 		/// The assets interface beyond native currency and other assets.
 		type MultiAssetsHandler: MultiAssetsHandler<Self::AccountId, Self::AssetId>;
 		/// This pallet id.
@@ -521,7 +523,7 @@ pub mod pallet {
 			asset_0: T::AssetId,
 			asset_1: T::AssetId,
 		) -> DispatchResult {
-			ensure_root(origin)?;
+			T::ControlOrigin::ensure_origin(origin)?;
 			ensure!(asset_0.is_support() && asset_1.is_support(), Error::<T>::UnsupportedAssetType);
 
 			ensure!(asset_0 != asset_1, Error::<T>::DeniedCreatePair);