Jaco
4913cf65fd
* Move api-derive to correct location * Adjust (linting ok) * No composite * ApiDerive split * Adjust * Still WIP, getting closer * It lints * derive build issues sorted * It builds :) * Add system_health to mock * api-augment usage * Cleanups * Adjust paths * linting * Some <'s * Revert <>'s * linting * ordering * Augmentation tweaks * yarn.lock * export
54 lines
2.2 KiB
TypeScript
54 lines
2.2 KiB
TypeScript
// Copyright 2017-2021 @polkadot/api-derive authors & contributors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
import type { Observable } from 'rxjs';
|
|
import type { Option, Vec } from '@polkadot/types';
|
|
import type { AccountId } from '@polkadot/types/interfaces';
|
|
import type { NodeRuntimeSessionKeys } from '@polkadot/types/lookup';
|
|
import type { ITuple } from '@polkadot/types/types';
|
|
import type { DeriveApi } from '../types';
|
|
import type { DeriveStakingKeys } from './types';
|
|
|
|
import { combineLatest, map, of, switchMap } from 'rxjs';
|
|
|
|
import { firstMemo, memo } from '../util';
|
|
|
|
function extractsIds (stashId: Uint8Array | string, queuedKeys: [AccountId, NodeRuntimeSessionKeys | AccountId[]][], nextKeys: Option<NodeRuntimeSessionKeys>): DeriveStakingKeys {
|
|
const sessionIds = (queuedKeys.find(([currentId]) => currentId.eq(stashId)) || [undefined, [] as AccountId[]])[1];
|
|
const nextSessionIds = nextKeys.unwrapOr([] as AccountId[]);
|
|
|
|
return {
|
|
nextSessionIds: Array.isArray(nextSessionIds)
|
|
? nextSessionIds
|
|
: [...nextSessionIds.values()] as AccountId[],
|
|
sessionIds: Array.isArray(sessionIds)
|
|
? sessionIds
|
|
: [...sessionIds.values()] as AccountId[]
|
|
};
|
|
}
|
|
|
|
export const keys = firstMemo(
|
|
(api: DeriveApi, stashId: Uint8Array | string) =>
|
|
api.derive.staking.keysMulti([stashId])
|
|
);
|
|
|
|
export function keysMulti (instanceId: string, api: DeriveApi): (stashIds: (Uint8Array | string)[]) => Observable<DeriveStakingKeys[]> {
|
|
return memo(instanceId, (stashIds: (Uint8Array | string)[]): Observable<DeriveStakingKeys[]> =>
|
|
stashIds.length
|
|
? api.query.session.queuedKeys().pipe(
|
|
switchMap((queuedKeys): Observable<[Vec<ITuple<[AccountId, NodeRuntimeSessionKeys]>>, Option<NodeRuntimeSessionKeys>[]]> =>
|
|
combineLatest([
|
|
of(queuedKeys),
|
|
api.consts.session?.dedupKeyPrefix
|
|
? api.query.session.nextKeys.multi(stashIds.map((stashId) => [api.consts.session.dedupKeyPrefix, stashId]))
|
|
: api.query.session.nextKeys.multi(stashIds)
|
|
])
|
|
),
|
|
map(([queuedKeys, nextKeys]) =>
|
|
stashIds.map((stashId, index) => extractsIds(stashId, queuedKeys, nextKeys[index]))
|
|
)
|
|
)
|
|
: of([])
|
|
);
|
|
}
|