@polkadot/api-provider/mock/mocks.js

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

var _values = _interopRequireDefault(require("@babel/runtime/core-js/object/values"));

require("core-js/modules/web.dom.iterable");

// Copyright 2017-2018 Jaco Greeff
// This software may be modified and distributed under the terms
// of the ISC license. See the LICENSE file for details.
// FIME: This file is way too long and way too messy
const BN = require('bn.js');

const headerEncode = require('@polkadot/primitives-json/header/encode');

const createKey = require('@polkadot/storage/key');

const state = require('@polkadot/storage');

const bnToU8a = require('@polkadot/util/bn/toU8a');

const u8aToHex = require('@polkadot/util/u8a/toHex');

const randomAsU8a = require('@polkadot/util-crypto/random/asU8a');

const testKeyring = require('@polkadot/util-keyring/testing');

const keyring = testKeyring();
const emitEvents = ['connected', 'disconnected'];
let emitIndex = 0;

function makeBlockHeader(prevNumber) {
  const blockNumber = prevNumber.addn(1);
  return {
    digest: {
      logs: []
    },
    extrinsicsRoot: randomAsU8a(),
    number: blockNumber,
    parentHash: prevNumber.eqn(-1) ? new Uint8Array(32) : bnToU8a(prevNumber, 256),
    stateRoot: bnToU8a(blockNumber, 256)
  };
}

function updateSubs(subscriptions, method, value) {
  subscriptions[method].lastValue = value;
  (0, _values.default)(subscriptions[method].callbacks).forEach(cb => {
    try {
      // flowlint-next-line unclear-type:off
      cb(null, value);
    } catch (error) {
      console.error(`Error on '${method}' subscription`, error);
    }
  });
}

function setStorageBn(storage, key, value, ...keyParams) {
  const keyValue = u8aToHex(createKey(key).apply(null, keyParams));
  storage[keyValue] = bnToU8a(value, 64, true);
}

module.exports = function mocks({
  emitter,
  storage,
  subscriptions
}) {
  let newHead = makeBlockHeader(new BN(-1));
  setInterval(() => {
    if (++emitIndex === emitEvents.length) {
      emitIndex = 0;
    }

    emitter.emit(emitEvents[emitIndex]);
    newHead = makeBlockHeader(newHead.number);
    keyring.getPairs().forEach(({
      publicKey
    }, index) => {
      setStorageBn(storage, state.staking.public.freeBalanceOf, newHead.number.muln(3).iaddn(index), publicKey());
      setStorageBn(storage, state.system.public.accountIndexOf, newHead.number.addn(index), publicKey());
    });
    setStorageBn(storage, state.timestamp.public.current, Math.floor(Date.now() / 1000));
    updateSubs(subscriptions, 'subscribe_newHead', headerEncode(newHead));
  }, 5000);
};