#![cfg_attr(not(feature = "std"), no_std)]
#![deny(missing_docs)]
extern crate alloc;
mod from;
mod utils;
use alloc::borrow::Cow;
use alloc::collections::BTreeMap;
use alloc::string::String;
use alloc::sync::Arc;
use alloc::vec::Vec;
use frame_decode::extrinsics::{
ExtrinsicCallInfo, ExtrinsicExtensionInfo, ExtrinsicInfoArg, ExtrinsicInfoError,
ExtrinsicSignatureInfo,
};
use hashbrown::HashMap;
use scale_info::{PortableRegistry, Variant, form::PortableForm};
use utils::{
ordered_map::OrderedMap,
validation::{HASH_LEN, get_custom_value_hash},
variant_index::VariantIndex,
};
type ArcStr = Arc<str>;
pub use from::SUPPORTED_METADATA_VERSIONS;
pub use from::TryFromError;
pub use utils::validation::MetadataHasher;
type CustomMetadataInner = frame_metadata::v15::CustomMetadata<PortableForm>;
#[derive(Debug, Clone)]
pub struct Metadata {
types: PortableRegistry,
pallets: OrderedMap<ArcStr, PalletMetadataInner>,
pallets_by_index: HashMap<u8, usize>,
extrinsic: ExtrinsicMetadata,
outer_enums: OuterEnumsMetadata,
dispatch_error_ty: Option<u32>,
apis: OrderedMap<ArcStr, RuntimeApiMetadataInner>,
custom: CustomMetadataInner,
}
impl frame_decode::extrinsics::ExtrinsicTypeInfo for Metadata {
type TypeId = u32;
fn get_call_info(
&self,
pallet_index: u8,
call_index: u8,
) -> Result<ExtrinsicCallInfo<'_, Self::TypeId>, ExtrinsicInfoError<'_>> {
let pallet = self.pallet_by_index(pallet_index).ok_or({
ExtrinsicInfoError::PalletNotFound {
index: pallet_index,
}
})?;
let call = pallet.call_variant_by_index(call_index).ok_or_else(|| {
ExtrinsicInfoError::CallNotFound {
index: call_index,
pallet_index,
pallet_name: Cow::Borrowed(pallet.name()),
}
})?;
Ok(ExtrinsicCallInfo {
pallet_name: Cow::Borrowed(pallet.name()),
call_name: Cow::Borrowed(&call.name),
args: call
.fields
.iter()
.map(|f| ExtrinsicInfoArg {
name: Cow::Borrowed(f.name.as_deref().unwrap_or("")),
id: f.ty.id,
})
.collect(),
})
}
fn get_signature_info(
&self,
) -> Result<ExtrinsicSignatureInfo<Self::TypeId>, ExtrinsicInfoError<'_>> {
Ok(ExtrinsicSignatureInfo {
address_id: self.extrinsic().address_ty,
signature_id: self.extrinsic().signature_ty,
})
}
fn get_extension_info(
&self,
extension_version: Option<u8>,
) -> Result<ExtrinsicExtensionInfo<'_, Self::TypeId>, ExtrinsicInfoError<'_>> {
let extension_version = extension_version.unwrap_or_else(|| {
self.extrinsic()
.transaction_extension_version_to_use_for_decoding()
});
let extension_ids = self
.extrinsic()
.transaction_extensions_by_version(extension_version)
.ok_or(ExtrinsicInfoError::ExtrinsicExtensionVersionNotFound { extension_version })?
.map(|f| ExtrinsicInfoArg {
name: Cow::Borrowed(f.identifier()),
id: f.extra_ty(),
})
.collect();
Ok(ExtrinsicExtensionInfo { extension_ids })
}
}
impl Metadata {
pub fn types(&self) -> &PortableRegistry {
&self.types
}
pub fn types_mut(&mut self) -> &mut PortableRegistry {
&mut self.types
}
pub fn dispatch_error_ty(&self) -> Option<u32> {
self.dispatch_error_ty
}
pub fn extrinsic(&self) -> &ExtrinsicMetadata {
&self.extrinsic
}
pub fn outer_enums(&self) -> OuterEnumsMetadata {
self.outer_enums
}
pub fn pallets(&self) -> impl ExactSizeIterator<Item = PalletMetadata<'_>> {
self.pallets.values().iter().map(|inner| PalletMetadata {
inner,
types: self.types(),
})
}
pub fn pallet_by_index(&self, variant_index: u8) -> Option<PalletMetadata<'_>> {
let inner = self
.pallets_by_index
.get(&variant_index)
.and_then(|i| self.pallets.get_by_index(*i))?;
Some(PalletMetadata {
inner,
types: self.types(),
})
}
pub fn pallet_by_name(&self, pallet_name: &str) -> Option<PalletMetadata<'_>> {
let inner = self.pallets.get_by_key(pallet_name)?;
Some(PalletMetadata {
inner,
types: self.types(),
})
}
pub fn runtime_api_traits(&self) -> impl ExactSizeIterator<Item = RuntimeApiMetadata<'_>> {
self.apis.values().iter().map(|inner| RuntimeApiMetadata {
inner,
types: self.types(),
})
}
pub fn runtime_api_trait_by_name(&'_ self, name: &str) -> Option<RuntimeApiMetadata<'_>> {
let inner = self.apis.get_by_key(name)?;
Some(RuntimeApiMetadata {
inner,
types: self.types(),
})
}
pub fn view_function_by_query_id(
&'_ self,
query_id: &[u8; 32],
) -> Option<ViewFunctionMetadata<'_>> {
self.pallets()
.flat_map(|p| p.view_functions())
.find(|vf| vf.query_id() == query_id)
}
pub fn custom(&self) -> CustomMetadata<'_> {
CustomMetadata {
types: self.types(),
inner: &self.custom,
}
}
pub fn hasher(&self) -> MetadataHasher<'_> {
MetadataHasher::new(self)
}
pub fn type_hash(&self, id: u32) -> Option<[u8; HASH_LEN]> {
self.types.resolve(id)?;
Some(crate::utils::validation::get_type_hash(&self.types, id))
}
}
#[derive(Debug, Clone, Copy)]
pub struct PalletMetadata<'a> {
inner: &'a PalletMetadataInner,
types: &'a PortableRegistry,
}
impl<'a> PalletMetadata<'a> {
pub fn name(&self) -> &'a str {
&self.inner.name
}
pub fn index(&self) -> u8 {
self.inner.index
}
pub fn docs(&self) -> &'a [String] {
&self.inner.docs
}
pub fn call_ty_id(&self) -> Option<u32> {
self.inner.call_ty
}
pub fn event_ty_id(&self) -> Option<u32> {
self.inner.event_ty
}
pub fn error_ty_id(&self) -> Option<u32> {
self.inner.error_ty
}
pub fn storage(&self) -> Option<&'a StorageMetadata> {
self.inner.storage.as_ref()
}
pub fn event_variants(&self) -> Option<&'a [Variant<PortableForm>]> {
VariantIndex::get(self.inner.event_ty, self.types)
}
pub fn event_variant_by_index(&self, variant_index: u8) -> Option<&'a Variant<PortableForm>> {
self.inner.event_variant_index.lookup_by_index(
variant_index,
self.inner.event_ty,
self.types,
)
}
pub fn has_view_functions(&self) -> bool {
!self.inner.view_functions.is_empty()
}
pub fn view_functions(
&self,
) -> impl ExactSizeIterator<Item = ViewFunctionMetadata<'a>> + use<'a> {
self.inner
.view_functions
.values()
.iter()
.map(|vf: &'a _| ViewFunctionMetadata {
inner: vf,
types: self.types,
})
}
pub fn view_function_by_name(&self, name: &str) -> Option<ViewFunctionMetadata<'a>> {
self.inner
.view_functions
.get_by_key(name)
.map(|vf: &'a _| ViewFunctionMetadata {
inner: vf,
types: self.types,
})
}
pub fn associated_types(&self) -> impl ExactSizeIterator<Item = (&'a str, u32)> + use<'a> {
self.inner
.associated_types
.iter()
.map(|(name, ty)| (&**name, *ty))
}
pub fn associated_type_id(&self, name: &str) -> Option<u32> {
self.inner.associated_types.get(name).copied()
}
pub fn call_variants(&self) -> Option<&'a [Variant<PortableForm>]> {
VariantIndex::get(self.inner.call_ty, self.types)
}
pub fn call_variant_by_index(&self, variant_index: u8) -> Option<&'a Variant<PortableForm>> {
self.inner
.call_variant_index
.lookup_by_index(variant_index, self.inner.call_ty, self.types)
}
pub fn call_variant_by_name(&self, call_name: &str) -> Option<&'a Variant<PortableForm>> {
self.inner
.call_variant_index
.lookup_by_name(call_name, self.inner.call_ty, self.types)
}
pub fn error_variants(&self) -> Option<&'a [Variant<PortableForm>]> {
VariantIndex::get(self.inner.error_ty, self.types)
}
pub fn error_variant_by_index(&self, variant_index: u8) -> Option<&'a Variant<PortableForm>> {
self.inner.error_variant_index.lookup_by_index(
variant_index,
self.inner.error_ty,
self.types,
)
}
pub fn constant_by_name(&self, name: &str) -> Option<&'a ConstantMetadata> {
self.inner.constants.get_by_key(name)
}
pub fn constants(&self) -> impl ExactSizeIterator<Item = &'a ConstantMetadata> + use<'a> {
self.inner.constants.values().iter()
}
pub fn storage_hash(&self, entry_name: &str) -> Option<[u8; HASH_LEN]> {
crate::utils::validation::get_storage_hash(self, entry_name)
}
pub fn constant_hash(&self, constant_name: &str) -> Option<[u8; HASH_LEN]> {
crate::utils::validation::get_constant_hash(self, constant_name)
}
pub fn call_hash(&self, call_name: &str) -> Option<[u8; HASH_LEN]> {
crate::utils::validation::get_call_hash(self, call_name)
}
pub fn hash(&self) -> [u8; HASH_LEN] {
crate::utils::validation::get_pallet_hash(*self)
}
}
#[derive(Debug, Clone)]
struct PalletMetadataInner {
name: ArcStr,
index: u8,
storage: Option<StorageMetadata>,
call_ty: Option<u32>,
call_variant_index: VariantIndex,
event_ty: Option<u32>,
event_variant_index: VariantIndex,
error_ty: Option<u32>,
error_variant_index: VariantIndex,
constants: OrderedMap<ArcStr, ConstantMetadata>,
view_functions: OrderedMap<ArcStr, ViewFunctionMetadataInner>,
associated_types: BTreeMap<String, u32>,
docs: Vec<String>,
}
#[derive(Debug, Clone)]
pub struct StorageMetadata {
prefix: String,
entries: OrderedMap<ArcStr, StorageEntryMetadata>,
}
impl StorageMetadata {
pub fn prefix(&self) -> &str {
&self.prefix
}
pub fn entries(&self) -> &[StorageEntryMetadata] {
self.entries.values()
}
pub fn entry_by_name(&self, name: &str) -> Option<&StorageEntryMetadata> {
self.entries.get_by_key(name)
}
}
#[derive(Debug, Clone)]
pub struct StorageEntryMetadata {
name: ArcStr,
modifier: StorageEntryModifier,
entry_type: StorageEntryType,
default: Vec<u8>,
docs: Vec<String>,
}
impl StorageEntryMetadata {
pub fn name(&self) -> &str {
&self.name
}
pub fn modifier(&self) -> StorageEntryModifier {
self.modifier
}
pub fn entry_type(&self) -> &StorageEntryType {
&self.entry_type
}
pub fn default_bytes(&self) -> &[u8] {
&self.default
}
pub fn docs(&self) -> &[String] {
&self.docs
}
}
#[derive(Debug, Clone)]
pub enum StorageEntryType {
Plain(u32),
Map {
hashers: Vec<StorageHasher>,
key_ty: u32,
value_ty: u32,
},
}
impl StorageEntryType {
pub fn value_ty(&self) -> u32 {
match self {
StorageEntryType::Map { value_ty, .. } | StorageEntryType::Plain(value_ty) => *value_ty,
}
}
pub fn key_ty(&self) -> Option<u32> {
match self {
StorageEntryType::Map { key_ty, .. } => Some(*key_ty),
StorageEntryType::Plain(_) => None,
}
}
}
#[derive(Debug, Clone, Copy)]
pub enum StorageHasher {
Blake2_128,
Blake2_256,
Blake2_128Concat,
Twox128,
Twox256,
Twox64Concat,
Identity,
}
impl StorageHasher {
pub fn len_excluding_key(&self) -> usize {
match self {
StorageHasher::Blake2_128Concat => 16,
StorageHasher::Twox64Concat => 8,
StorageHasher::Blake2_128 => 16,
StorageHasher::Blake2_256 => 32,
StorageHasher::Twox128 => 16,
StorageHasher::Twox256 => 32,
StorageHasher::Identity => 0,
}
}
pub fn ends_with_key(&self) -> bool {
matches!(
self,
StorageHasher::Blake2_128Concat | StorageHasher::Twox64Concat | StorageHasher::Identity
)
}
}
#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub enum StorageEntryModifier {
Optional,
Default,
}
#[derive(Debug, Clone)]
pub struct ConstantMetadata {
name: ArcStr,
ty: u32,
value: Vec<u8>,
docs: Vec<String>,
}
impl ConstantMetadata {
pub fn name(&self) -> &str {
&self.name
}
pub fn ty(&self) -> u32 {
self.ty
}
pub fn value(&self) -> &[u8] {
&self.value
}
pub fn docs(&self) -> &[String] {
&self.docs
}
}
#[derive(Debug, Clone)]
pub struct ExtrinsicMetadata {
address_ty: u32,
signature_ty: u32,
supported_versions: Vec<u8>,
transaction_extensions: Vec<TransactionExtensionMetadataInner>,
transaction_extensions_by_version: BTreeMap<u8, Vec<u32>>,
}
impl ExtrinsicMetadata {
pub fn supported_versions(&self) -> &[u8] {
&self.supported_versions
}
pub fn transaction_extensions_by_version(
&self,
version: u8,
) -> Option<impl Iterator<Item = TransactionExtensionMetadata<'_>>> {
let extension_indexes = self.transaction_extensions_by_version.get(&version)?;
let iter = extension_indexes.iter().map(|index| {
let tx_metadata = self
.transaction_extensions
.get(*index as usize)
.expect("transaction extension should exist if index is in transaction_extensions_by_version");
TransactionExtensionMetadata {
identifier: &tx_metadata.identifier,
extra_ty: tx_metadata.extra_ty,
additional_ty: tx_metadata.additional_ty,
}
});
Some(iter)
}
pub fn transaction_extension_version_to_use_for_encoding(&self) -> u8 {
*self
.transaction_extensions_by_version
.keys()
.max()
.expect("At least one version of transaction extensions is expected")
}
pub fn transaction_extensions_to_use_for_encoding(
&self,
) -> impl Iterator<Item = TransactionExtensionMetadata<'_>> {
let encoding_version = self.transaction_extension_version_to_use_for_encoding();
self.transaction_extensions_by_version(encoding_version)
.unwrap()
}
pub fn transaction_extension_version_to_use_for_decoding(&self) -> u8 {
*self
.transaction_extensions_by_version
.keys()
.max()
.expect("At least one version of transaction extensions is expected")
}
}
#[derive(Debug, Clone)]
pub struct TransactionExtensionMetadata<'a> {
identifier: &'a str,
extra_ty: u32,
additional_ty: u32,
}
#[derive(Debug, Clone)]
struct TransactionExtensionMetadataInner {
identifier: String,
extra_ty: u32,
additional_ty: u32,
}
impl<'a> TransactionExtensionMetadata<'a> {
pub fn identifier(&self) -> &'a str {
self.identifier
}
pub fn extra_ty(&self) -> u32 {
self.extra_ty
}
pub fn additional_ty(&self) -> u32 {
self.additional_ty
}
}
#[derive(Debug, Clone, Copy)]
pub struct OuterEnumsMetadata {
call_enum_ty: u32,
event_enum_ty: u32,
error_enum_ty: u32,
}
impl OuterEnumsMetadata {
pub fn call_enum_ty(&self) -> u32 {
self.call_enum_ty
}
pub fn event_enum_ty(&self) -> u32 {
self.event_enum_ty
}
pub fn error_enum_ty(&self) -> u32 {
self.error_enum_ty
}
}
#[derive(Debug, Clone, Copy)]
pub struct RuntimeApiMetadata<'a> {
inner: &'a RuntimeApiMetadataInner,
types: &'a PortableRegistry,
}
impl<'a> RuntimeApiMetadata<'a> {
pub fn name(&self) -> &'a str {
&self.inner.name
}
pub fn docs(&self) -> &[String] {
&self.inner.docs
}
pub fn methods(&self) -> impl ExactSizeIterator<Item = RuntimeApiMethodMetadata<'a>> + use<'a> {
self.inner
.methods
.values()
.iter()
.map(|item| RuntimeApiMethodMetadata {
trait_name: &self.inner.name,
inner: item,
types: self.types,
})
}
pub fn method_by_name(&self, name: &str) -> Option<RuntimeApiMethodMetadata<'a>> {
self.inner
.methods
.get_by_key(name)
.map(|item| RuntimeApiMethodMetadata {
trait_name: &self.inner.name,
inner: item,
types: self.types,
})
}
pub fn hash(&self) -> [u8; HASH_LEN] {
crate::utils::validation::get_runtime_apis_hash(*self)
}
}
#[derive(Debug, Clone)]
struct RuntimeApiMetadataInner {
name: ArcStr,
methods: OrderedMap<ArcStr, RuntimeApiMethodMetadataInner>,
docs: Vec<String>,
}
#[derive(Debug, Clone)]
pub struct RuntimeApiMethodMetadata<'a> {
trait_name: &'a str,
inner: &'a RuntimeApiMethodMetadataInner,
types: &'a PortableRegistry,
}
impl<'a> RuntimeApiMethodMetadata<'a> {
pub fn name(&self) -> &'a str {
&self.inner.name
}
pub fn docs(&self) -> &[String] {
&self.inner.docs
}
pub fn inputs(&self) -> impl ExactSizeIterator<Item = &'a MethodParamMetadata> + use<'a> {
self.inner.inputs.iter()
}
pub fn output_ty(&self) -> u32 {
self.inner.output_ty
}
pub fn hash(&self) -> [u8; HASH_LEN] {
crate::utils::validation::get_runtime_api_hash(self)
}
}
#[derive(Debug, Clone)]
struct RuntimeApiMethodMetadataInner {
name: ArcStr,
inputs: Vec<MethodParamMetadata>,
output_ty: u32,
docs: Vec<String>,
}
#[derive(Debug, Clone, Copy)]
pub struct ViewFunctionMetadata<'a> {
inner: &'a ViewFunctionMetadataInner,
types: &'a PortableRegistry,
}
impl<'a> ViewFunctionMetadata<'a> {
pub fn name(&self) -> &'a str {
&self.inner.name
}
pub fn query_id(&self) -> &'a [u8; 32] {
&self.inner.query_id
}
pub fn docs(&self) -> &'a [String] {
&self.inner.docs
}
pub fn inputs(&self) -> impl ExactSizeIterator<Item = &'a MethodParamMetadata> + use<'a> {
self.inner.inputs.iter()
}
pub fn output_ty(&self) -> u32 {
self.inner.output_ty
}
pub fn hash(&self) -> [u8; HASH_LEN] {
crate::utils::validation::get_view_function_hash(self)
}
}
#[derive(Debug, Clone)]
struct ViewFunctionMetadataInner {
name: ArcStr,
query_id: [u8; 32],
inputs: Vec<MethodParamMetadata>,
output_ty: u32,
docs: Vec<String>,
}
#[derive(Debug, Clone)]
pub struct MethodParamMetadata {
pub name: String,
pub ty: u32,
}
#[derive(Debug, Clone)]
pub struct CustomMetadata<'a> {
types: &'a PortableRegistry,
inner: &'a CustomMetadataInner,
}
impl<'a> CustomMetadata<'a> {
pub fn get(&self, name: &str) -> Option<CustomValueMetadata<'a>> {
self.inner
.map
.get_key_value(name)
.map(|(name, e)| CustomValueMetadata {
types: self.types,
type_id: e.ty.id,
data: &e.value,
name,
})
}
pub fn iter(&self) -> impl Iterator<Item = CustomValueMetadata<'a>> + use<'a> {
self.inner.map.iter().map(|(name, e)| CustomValueMetadata {
types: self.types,
type_id: e.ty.id,
data: &e.value,
name: name.as_ref(),
})
}
pub fn types(&self) -> &PortableRegistry {
self.types
}
}
pub struct CustomValueMetadata<'a> {
types: &'a PortableRegistry,
type_id: u32,
data: &'a [u8],
name: &'a str,
}
impl<'a> CustomValueMetadata<'a> {
pub fn types(&self) -> &PortableRegistry {
self.types
}
pub fn bytes(&self) -> &'a [u8] {
self.data
}
pub fn type_id(&self) -> u32 {
self.type_id
}
pub fn name(&self) -> &str {
self.name
}
pub fn hash(&self) -> [u8; HASH_LEN] {
get_custom_value_hash(self)
}
}
impl codec::Decode for Metadata {
fn decode<I: codec::Input>(input: &mut I) -> Result<Self, codec::Error> {
let metadata = frame_metadata::RuntimeMetadataPrefixed::decode(input)?;
let metadata = match metadata.1 {
frame_metadata::RuntimeMetadata::V14(md) => md.try_into(),
frame_metadata::RuntimeMetadata::V15(md) => md.try_into(),
frame_metadata::RuntimeMetadata::V16(md) => md.try_into(),
_ => return Err("Cannot try_into() to Metadata: unsupported metadata version".into()),
};
metadata.map_err(|_e| "Cannot try_into() to Metadata.".into())
}
}