@polkadot/types
Version:
Implementation of the Parity codec
109 lines (108 loc) • 4.24 kB
JavaScript
import { AbstractBase } from '@polkadot/types-codec';
import { isBigInt, isBn, isHex, isNumber, isU8a, u8aConcat, u8aToBn, u8aToHex, u8aToU8a } from '@polkadot/util';
import { decodeAddress } from '@polkadot/util-crypto';
import { GenericAccountId } from './AccountId.js';
import { GenericAccountIndex } from './AccountIndex.js';
export const ACCOUNT_ID_PREFIX = new Uint8Array([0xff]);
/** @internal */
function decodeString(registry, value) {
const decoded = decodeAddress(value);
return decoded.length === 32
? registry.createTypeUnsafe('AccountId', [decoded])
: registry.createTypeUnsafe('AccountIndex', [u8aToBn(decoded)]);
}
/** @internal */
function decodeU8a(registry, value) {
// This allows us to instantiate an address with a raw publicKey. Do this first before
// we checking the first byte, otherwise we may split an already-existent valid address
if (value.length === 32) {
return registry.createTypeUnsafe('AccountId', [value]);
}
else if (value[0] === 0xff) {
return registry.createTypeUnsafe('AccountId', [value.subarray(1)]);
}
const [offset, length] = GenericAccountIndex.readLength(value);
return registry.createTypeUnsafe('AccountIndex', [u8aToBn(value.subarray(offset, offset + length))]);
}
/** @internal */
function decodeAddressOrIndex(registry, value) {
return value instanceof GenericLookupSource
? value.inner
: value instanceof GenericAccountId || value instanceof GenericAccountIndex
? value
: isBn(value) || isNumber(value) || isBigInt(value)
? registry.createTypeUnsafe('AccountIndex', [value])
: Array.isArray(value) || isHex(value) || isU8a(value)
? decodeU8a(registry, u8aToU8a(value))
: decodeString(registry, value);
}
/**
* @name LookupSource
* @description
* A wrapper around an AccountId and/or AccountIndex that is encoded with a prefix.
* Since we are dealing with underlying publicKeys (or shorter encoded addresses),
* we extend from Base with an AccountId/AccountIndex wrapper. Basically the Address
* is encoded as `[ <prefix-byte>, ...publicKey/...bytes ]` as per spec
*/
export class GenericLookupSource extends AbstractBase {
constructor(registry, value = new Uint8Array()) {
super(registry, decodeAddressOrIndex(registry, value));
}
/**
* @description The length of the value when encoded as a Uint8Array
*/
get encodedLength() {
const rawLength = this._rawLength;
return rawLength + (
// for 1 byte AccountIndexes, we are not adding a specific prefix
rawLength > 1
? 1
: 0);
}
/**
* @description The length of the raw value, either AccountIndex or AccountId
*/
get _rawLength() {
return this.inner instanceof GenericAccountIndex
? GenericAccountIndex.calcLength(this.inner)
: this.inner.encodedLength;
}
/**
* @description Returns a breakdown of the hex encoding for this Codec
*/
inspect() {
const value = this.inner.toU8a().subarray(0, this._rawLength);
return {
outer: [
new Uint8Array(this.inner instanceof GenericAccountIndex
? GenericAccountIndex.writeLength(value)
: ACCOUNT_ID_PREFIX),
value
]
};
}
/**
* @description Returns a hex string representation of the value
*/
toHex() {
return u8aToHex(this.toU8a());
}
/**
* @description Returns the base runtime type name for this instance
*/
toRawType() {
return 'Address';
}
/**
* @description Encodes the value as a Uint8Array as per the SCALE specifications
* @param isBare true when the value has none of the type-specific prefixes (internal)
*/
toU8a(isBare) {
const encoded = this.inner.toU8a().subarray(0, this._rawLength);
return isBare
? encoded
: u8aConcat(this.inner instanceof GenericAccountIndex
? GenericAccountIndex.writeLength(encoded)
: ACCOUNT_ID_PREFIX, encoded);
}
}