114 lines
4.3 KiB
TypeScript
114 lines
4.3 KiB
TypeScript
// Copyright 2017-2020 @polkadot/metadata authors & contributors
|
|
// This software may be modified and distributed under the terms
|
|
// of the Apache-2.0 license. See the LICENSE file for details.
|
|
|
|
import { EventMetadataV12, EventMetadataLatest, FunctionMetadataV12, FunctionMetadataLatest, MetadataV12, MetadataLatest, ModuleMetadataLatest, StorageMetadataV12, StorageMetadataLatest, StorageEntryMetadataLatest } from '@polkadot/types/interfaces/metadata';
|
|
import { Registry, OverrideModuleType } from '@polkadot/types/types';
|
|
|
|
import { getModuleTypes } from '@polkadot/types-known';
|
|
import { Type } from '@polkadot/types/primitive';
|
|
import { stringCamelCase } from '@polkadot/util';
|
|
|
|
// TODO Handle consts as well
|
|
|
|
/**
|
|
* Find and apply the correct type override
|
|
* @internal
|
|
**/
|
|
function setTypeOverride (sectionTypes: OverrideModuleType, type: Type): void {
|
|
const override = Object.keys(sectionTypes).find((aliased) => type.eq(aliased));
|
|
|
|
if (override) {
|
|
type.setOverride(sectionTypes[override]);
|
|
} else {
|
|
// FIXME: NOT happy with this approach, but gets over the initial hump cased by (Vec<Announcement>,BalanceOf)
|
|
const orig = type.toString();
|
|
const alias = Object.entries(sectionTypes).reduce((result: string, [from, to]): string =>
|
|
[['<', '>'], ['<', ','], [',', '>'], ['(', ')'], ['(', ','], [',', ','], [',', ')']].reduce((result, [one, two]): string =>
|
|
result.replace(`${one}${from}${two}`, `${one}${to}${two}`), result), orig);
|
|
|
|
if (orig !== alias) {
|
|
type.setOverride(alias);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Apply module-specific type overrides (always be done as part of toLatest)
|
|
* @internal
|
|
**/
|
|
function convertCalls (registry: Registry, calls: FunctionMetadataV12[], sectionTypes: OverrideModuleType): FunctionMetadataLatest[] {
|
|
return calls.map(({ args, documentation, name }): FunctionMetadataLatest => {
|
|
args.forEach(({ type }) => setTypeOverride(sectionTypes, type));
|
|
|
|
return registry.createType('FunctionMetadataLatest', { args, documentation, name });
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Apply module-specific type overrides (always be done as part of toLatest)
|
|
* @internal
|
|
**/
|
|
function convertEvents (registry: Registry, events: EventMetadataV12[], sectionTypes: OverrideModuleType): EventMetadataLatest[] {
|
|
return events.map(({ args, documentation, name }): EventMetadataLatest => {
|
|
args.forEach((type) => setTypeOverride(sectionTypes, type));
|
|
|
|
return registry.createType('EventMetadataLatest', { args, documentation, name });
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Apply module-specific storage type overrides (always part of toLatest)
|
|
* @internal
|
|
**/
|
|
function convertStorage (registry: Registry, { items, prefix }: StorageMetadataV12, sectionTypes: OverrideModuleType): StorageMetadataLatest {
|
|
return registry.createType('StorageMetadataLatest', {
|
|
items: items.map(({ documentation, fallback, modifier, name, type }): StorageEntryMetadataLatest => {
|
|
let resultType: Type;
|
|
|
|
if (type.isMap) {
|
|
resultType = type.asMap.value;
|
|
} else if (type.isDoubleMap) {
|
|
resultType = type.asDoubleMap.value;
|
|
} else {
|
|
resultType = type.asPlain;
|
|
}
|
|
|
|
setTypeOverride(sectionTypes, resultType);
|
|
|
|
return registry.createType('StorageEntryMetadataLatest', { documentation, fallback, modifier, name, type });
|
|
}),
|
|
prefix
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Convert the Metadata (which is an alias) to latest - effectively this _always_ get applied to the top-level &
|
|
* most-recent metadata, since it allows us a chance to actually apply call and storage specific type aliasses
|
|
* @internal
|
|
**/
|
|
export default function toLatest (registry: Registry, { extrinsic, modules }: MetadataV12): MetadataLatest {
|
|
return registry.createType('MetadataLatest', {
|
|
extrinsic,
|
|
modules: modules.map((mod): ModuleMetadataLatest => {
|
|
const calls = mod.calls.unwrapOr(null);
|
|
const events = mod.events.unwrapOr(null);
|
|
const storage = mod.storage.unwrapOr(null);
|
|
const sectionTypes = getModuleTypes(registry, stringCamelCase(mod.name.toString()));
|
|
|
|
return registry.createType('ModuleMetadataLatest', {
|
|
...mod,
|
|
calls: calls
|
|
? convertCalls(registry, calls, sectionTypes)
|
|
: null,
|
|
events: events
|
|
? convertEvents(registry, events, sectionTypes)
|
|
: null,
|
|
storage: storage
|
|
? convertStorage(registry, storage, sectionTypes)
|
|
: null
|
|
});
|
|
})
|
|
});
|
|
}
|