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

'use strict';

var _values = require('babel-runtime/core-js/object/values');

var _values2 = _interopRequireDefault(_values);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

// 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

var BN = require('bn.js');
var headerEncode = require('@polkadot/primitives-json/header/encode');
var bnToU8a = require('@polkadot/util/bn/toU8a');
var randomAsU8a = require('@polkadot/util-crypto/random/asU8a');
var testKeyring = require('@polkadot/util-keyring/testing');

var storageKey = require('./storageKey');

var keyring = testKeyring();

function makeBlockHeader(prevNumber) {
  var 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, _values2.default)(subscriptions[method].callbacks).forEach(function (cb) {
    try {
      // flowlint-next-line unclear-type:off
      cb(null, value);
    } catch (error) {
      console.error('Error on \'' + method + '\' subscription', error);
    }
  });
}

function setStorageBn(storage, prefix, key, value) {
  storage[storageKey(prefix, key)] = bnToU8a(value, 64, true);
}

module.exports = function mocks(_ref) {
  var storage = _ref.storage,
      subscriptions = _ref.subscriptions;

  var newHead = makeBlockHeader(new BN(-1));

  setInterval(function () {
    newHead = makeBlockHeader(newHead.number);

    (0, _values2.default)(keyring).forEach(function (pair, index) {
      // flowlint-next-line unclear-type:off
      var publicKey = pair.publicKey();
      var balance = newHead.number.muln(3).iaddn(index);

      setStorageBn(storage, 'sta:bal:', publicKey, balance);
    });

    updateSubs(subscriptions, 'subscribe_newHead', headerEncode(newHead));
  }, 5000);
};