pezkuwichain
b6d35f6faf
Updated 4763 files with dual copyright: - Parity Technologies (UK) Ltd. - Dijital Kurdistan Tech Institute
134 lines
3.7 KiB
Rust
134 lines
3.7 KiB
Rust
// This file is part of Bizinikiwi.
|
|
|
|
// Copyright (C) Parity Technologies (UK) Ltd. and Dijital Kurdistan Tech Institute
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
//! Bizinikiwi core types around sessions.
|
|
|
|
#![cfg_attr(not(feature = "std"), no_std)]
|
|
|
|
extern crate alloc;
|
|
|
|
use codec::{Decode, DecodeWithMemTracking, Encode};
|
|
|
|
#[cfg(feature = "std")]
|
|
use pezsp_api::ProvideRuntimeApi;
|
|
#[cfg(feature = "std")]
|
|
use pezsp_runtime::traits::Block as BlockT;
|
|
|
|
use alloc::vec::Vec;
|
|
use pezsp_core::RuntimeDebug;
|
|
use pezsp_staking::SessionIndex;
|
|
|
|
pub mod runtime_api;
|
|
pub use runtime_api::*;
|
|
|
|
/// Number of validators in a given session.
|
|
pub type ValidatorCount = u32;
|
|
|
|
/// Proof of membership of a specific key in a given session.
|
|
#[derive(
|
|
Encode,
|
|
Decode,
|
|
DecodeWithMemTracking,
|
|
Clone,
|
|
Eq,
|
|
PartialEq,
|
|
Default,
|
|
RuntimeDebug,
|
|
scale_info::TypeInfo,
|
|
)]
|
|
pub struct MembershipProof {
|
|
/// The session index on which the specific key is a member.
|
|
pub session: SessionIndex,
|
|
/// Trie nodes of a merkle proof of session membership.
|
|
pub trie_nodes: Vec<Vec<u8>>,
|
|
/// The validator count of the session on which the specific key is a member.
|
|
pub validator_count: ValidatorCount,
|
|
}
|
|
|
|
/// A utility trait to get a session number. This is implemented for
|
|
/// `MembershipProof` below to fetch the session number the given session
|
|
/// membership proof is for. It is useful when we need to deal with key owner
|
|
/// proofs generically (i.e. just typing against the `KeyOwnerProofSystem`
|
|
/// trait) but still restrict their capabilities.
|
|
pub trait GetSessionNumber {
|
|
fn session(&self) -> SessionIndex;
|
|
}
|
|
|
|
/// A utility trait to get the validator count of a given session. This is
|
|
/// implemented for `MembershipProof` below and fetches the number of validators
|
|
/// in the session the membership proof is for. It is useful when we need to
|
|
/// deal with key owner proofs generically (i.e. just typing against the
|
|
/// `KeyOwnerProofSystem` trait) but still restrict their capabilities.
|
|
pub trait GetValidatorCount {
|
|
fn validator_count(&self) -> ValidatorCount;
|
|
}
|
|
|
|
impl GetSessionNumber for pezsp_core::Void {
|
|
fn session(&self) -> SessionIndex {
|
|
Default::default()
|
|
}
|
|
}
|
|
|
|
impl GetValidatorCount for pezsp_core::Void {
|
|
fn validator_count(&self) -> ValidatorCount {
|
|
Default::default()
|
|
}
|
|
}
|
|
|
|
impl GetSessionNumber for MembershipProof {
|
|
fn session(&self) -> SessionIndex {
|
|
self.session
|
|
}
|
|
}
|
|
|
|
impl GetValidatorCount for MembershipProof {
|
|
fn validator_count(&self) -> ValidatorCount {
|
|
self.validator_count
|
|
}
|
|
}
|
|
|
|
/// Generate the initial session keys with the given seeds, at the given block and store them in
|
|
/// the client's keystore.
|
|
#[cfg(feature = "std")]
|
|
pub fn generate_initial_session_keys<Block, T>(
|
|
client: std::sync::Arc<T>,
|
|
at: Block::Hash,
|
|
seeds: Vec<String>,
|
|
keystore: pezsp_keystore::KeystorePtr,
|
|
) -> Result<(), pezsp_api::ApiError>
|
|
where
|
|
Block: BlockT,
|
|
T: ProvideRuntimeApi<Block>,
|
|
T::Api: SessionKeys<Block>,
|
|
{
|
|
use pezsp_api::ApiExt;
|
|
|
|
if seeds.is_empty() {
|
|
return Ok(());
|
|
}
|
|
|
|
let mut runtime_api = client.runtime_api();
|
|
|
|
runtime_api.register_extension(pezsp_keystore::KeystoreExt::from(keystore));
|
|
|
|
for seed in seeds {
|
|
runtime_api.generate_session_keys(at, Some(seed.as_bytes().to_vec()))?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|