aboutsummaryrefslogtreecommitdiffstats
path: root/app/scripts/lib/pending-balance-calculator.js
blob: 4e1189a651afbe4345f15635772ce85cb0f3b8d3 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
const BN = require('ethereumjs-util').BN
const EthQuery = require('ethjs-query')
const normalize = require('eth-sig-util').normalize

class PendingBalanceCalculator {

  // Must be initialized with two functions:
  // getBalance => Returns a promise of a BN of the current balance in Wei
  // getPendingTransactions => Returns an array of TxMeta Objects,
  // which have txParams properties, which include value, gasPrice, and gas,
  // all in a base=16 hex format.
  constructor ({ getBalance, getPendingTransactions }) {
    this.getPendingTransactions = getPendingTransactions
    this.getNetworkBalance = getBalance
  }

  async getBalance() {
    console.log('getting balance')
    const results = await Promise.all([
      this.getNetworkBalance(),
      this.getPendingTransactions(),
    ])

    const balance = results[0]
    const pending = results[1]

    console.dir(pending)
    console.dir(balance.toString())
    console.trace('but why')

    const pendingValue = pending.reduce((total, tx) => {
      return total.add(this.valueFor(tx))
    }, new BN(0))

    console.log(`subtracting ${pendingValue.toString()} from ${balance.toString()}`)

    return `0x${ balance.sub(pendingValue).toString(16) }`
  }

  valueFor (tx) {
    const txValue = tx.txParams.value
    const normalized = normalize(txValue).substring(2)
    console.log({ txValue, normalized })
    const value = this.hexToBn(txValue)
    return value
  }

  hexToBn (hex) {
    return new BN(normalize(hex).substring(2), 16)
  }

}

module.exports = PendingBalanceCalculator