Jake Naviasky
efb38dbdf8
* Fixing array bug in approvalFlagToBool derive. * Replicating 'take until last approval' logic.
32 lines
1.3 KiB
TypeScript
32 lines
1.3 KiB
TypeScript
import BN from 'bn.js';
|
|
import { Observable } from 'rxjs';
|
|
import { switchMap, map } from 'rxjs/operators';
|
|
import { ApiInterfaceRx } from '@polkadot/api/types';
|
|
import { AccountId, Vector, SetIndex, ApprovalFlag } from '@polkadot/types';
|
|
import { drr } from '../util/drr';
|
|
|
|
export function approvalFlagToBool (flags: Vector<ApprovalFlag>): boolean[] {
|
|
const bools: boolean[] = [];
|
|
for (const flag of flags) {
|
|
for (const bit of [...Array(flag.bitLength())].map((_, i): number => i)) {
|
|
bools.push(!flag.toBn().uand((new BN(1)).shln(bit)).isZero());
|
|
}
|
|
}
|
|
const lastApproval: number = bools.lastIndexOf(true);
|
|
return lastApproval >= 0 ? bools.slice(0, lastApproval + 1) : [];
|
|
}
|
|
|
|
export function approvalsOf (api: ApiInterfaceRx): (who: AccountId) => Observable<Vector<ApprovalFlag>> {
|
|
return (who: AccountId): Observable<Vector<ApprovalFlag>> =>
|
|
api.query.elections.nextVoterSet<SetIndex>().pipe(
|
|
switchMap((nextVoterSet: SetIndex): Observable<Vector<ApprovalFlag>[]> =>
|
|
api.query.elections.approvalsOf.multi(
|
|
[...Array(+nextVoterSet + 1)].map((_, i): [AccountId, number] => [who, i])
|
|
)
|
|
),
|
|
map((votes: Vector<ApprovalFlag>[]): Vector<ApprovalFlag> =>
|
|
votes.find((v): boolean => v.length > 0) || new Vector(ApprovalFlag, [])),
|
|
drr()
|
|
);
|
|
}
|