feat: Draft of pallet_session_keys_registration

Author: Klapeyron

Cargo.lock

@@ -57,6 +57,7 @@ members = [
 	"toolkit/governed-map/primitives",
 	"toolkit/governed-map/pallet",
 	"toolkit/committee-selection/selection-simulator",
+	"toolkit/session_keys_registration",
 ]
 resolver = "2"
 
@@ -330,5 +331,8 @@ partner-chains-mock-data-sources = { path = "toolkit/data-sources/mock", default
 sp-governed-map = { path = "toolkit/governed-map/primitives", default-features = false }
 pallet-governed-map = { path = "toolkit/governed-map/pallet", default-features = false }
 
+# Session Keys Registration
+pallet-session-keys-registration = { path = "toolkit/session_keys_registration", default-features = false }
+
 # demo node
 partner-chains-demo-runtime = { path = "demo/runtime" }

Cargo.toml

@@ -93,6 +93,7 @@ pallet-governed-map = { workspace = true }
 sp-governed-map = { workspace = true }
 sp-block-producer-fees = { workspace = true }
 pallet-block-producer-fees = { workspace = true }
+pallet-session-keys-registration = { workspace = true }
 
 [dev-dependencies]
 sp-io = { workspace = true }
@@ -139,6 +140,7 @@ std = [
 	"pallet-transaction-payment/std",
 	"pallet-address-associations/std",
 	"pallet-block-producer-metadata/std",
+	"pallet-session-keys-registration/std",
 	"sp-block-producer-metadata/std",
 	"sp-api/std",
 	"sp-block-builder/std",

demo/runtime/Cargo.toml

@@ -681,6 +681,10 @@ impl pallet_governed_map::Config for Runtime {
 	type BenchmarkHelper = ();
 }
 
+impl pallet_session_keys_registration::Config for Runtime {
+	type PalletsOrigin = OriginCaller;
+}
+
 impl crate::test_helper_pallet::Config for Runtime {}
 
 // Create the runtime by composing the FRAME pallets that were previously configured.
@@ -702,7 +706,8 @@ construct_runtime!(
 		BlockProducerMetadata: pallet_block_producer_metadata,
 		BlockProductionLog: pallet_block_production_log,
 		BlockParticipation: pallet_block_participation,
-		Session: pallet_session,
+		Session: pallet_session exclude_parts { Call },
+		SessionKeysRegistration: pallet_session_keys_registration,
 		// Historical: pallet_session::historical,
 		NativeTokenManagement: pallet_native_token_management,
 		GovernedMap: pallet_governed_map,

demo/runtime/src/lib.rs

@@ -0,0 +1,33 @@
+[package]
+name = "pallet-session-keys-registration"
+authors.workspace = true
+edition.workspace = true
+homepage.workspace = true
+repository.workspace = true
+version.workspace = true
+license = "Apache-2.0"
+description = "Pallet for registering session keys for the partner chains"
+
+[lints]
+workspace = true
+
+[dependencies]
+frame-support = { workspace = true, default-features = false }
+frame-system = { workspace = true, default-features = false }
+parity-scale-codec = { workspace = true }
+scale-info = { workspace = true, default-features = false, features = [
+    "derive",
+] }
+sp-runtime = { workspace = true, default-features = false }
+pallet-session = { workspace = true }
+
+[features]
+default = ["std"]
+std = [
+    "frame-support/std",
+    "frame-system/std",
+    "parity-scale-codec/std",
+    "scale-info/std",
+    "sp-runtime/std",
+    "pallet-session/std",
+]

toolkit/session_keys_registration/Cargo.toml

@@ -0,0 +1,41 @@
+#![cfg_attr(not(feature = "std"), no_std)]
+
+pub use pallet::*;
+
+#[frame_support::pallet]
+pub mod pallet {
+	use frame_support::{dispatch::DispatchResultWithPostInfo, pallet_prelude::*};
+	use scale_info::prelude::boxed::Box;
+	use sp_runtime::Vec;
+
+	#[pallet::pallet]
+	pub struct Pallet<T>(_);
+
+	#[pallet::config]
+	pub trait Config: frame_system::Config + pallet_session::Config {
+		/// The caller origin, overarching type of all pallets origins.
+		type PalletsOrigin: Parameter +
+			Into<<Self as frame_system::Config>::RuntimeOrigin> +
+			IsType<<<Self as frame_system::Config>::RuntimeOrigin as frame_support::traits::OriginTrait>::PalletsOrigin>;
+	}
+
+	impl<T: Config> Pallet<T> {
+		/// Invokes the `pallet_session::Call::set_keys` function to set the session keys.
+		///
+		/// Allows the caller to set the session keys for the next session for particular user.
+		///
+		/// ## Complexity
+		/// - `O(1)`. Actual cost depends on the number of length of `T::Keys::key_ids()` which is
+		///   fixed.
+		pub fn set_keys(
+			as_origin: Box<T::PalletsOrigin>,
+			keys: <T as pallet_session::Config>::Keys,
+			proof: Vec<u8>,
+		) -> DispatchResultWithPostInfo {
+			let call = pallet_session::Call::<T>::set_keys { keys, proof };
+
+			use frame_support::traits::UnfilteredDispatchable;
+			call.dispatch_bypass_filter((*as_origin).into())
+		}
+	}
+}