bitcore-node-zcash/app/models/Address.js

'use strict';

var imports            = require('soop').imports();
var async              = require('async');
var BitcoreAddress     = require('bitcore/Address');
var BitcoreTransaction = require('bitcore/Transaction');
var BitcoreUtil        = require('bitcore/util/util');
var Parser             = require('bitcore/util/BinaryParser');
var Buffer             = require('buffer').Buffer;
var TransactionDb      = imports.TransactionDb || require('../../lib/TransactionDb').default();
var CONCURRENCY        = 5;

function Address(addrStr) {
  this.balanceSat        = 0;
  this.totalReceivedSat  = 0;
  this.totalSentSat      = 0;

  this.unconfirmedBalanceSat  = 0;

  this.txApperances           = 0;
  this.unconfirmedTxApperances= 0;

  // TODO store only txids? +index? +all?
  this.transactions   = [];

  var a = new BitcoreAddress(addrStr);
  a.validate();
  this.addrStr        = addrStr;
  
  Object.defineProperty(this, 'totalSent', {
    get: function() {
      return parseFloat(this.totalSentSat) / parseFloat(BitcoreUtil.COIN);
    },
    set:  function(i) {
      this.totalSentSat =  i * BitcoreUtil.COIN;
    },
    enumerable: 1,
  });

  Object.defineProperty(this, 'balance', {
    get: function() {
      return parseFloat(this.balanceSat) / parseFloat(BitcoreUtil.COIN);
    },
    set:  function(i) {
      this.balance =   i * BitcoreUtil.COIN;
    },
    enumerable: 1,
  });

  Object.defineProperty(this, 'totalReceived', {
    get: function() {
      return parseFloat(this.totalReceivedSat) / parseFloat(BitcoreUtil.COIN);
    },
    set:  function(i) {
      this.totalReceived =  i * BitcoreUtil.COIN;
    },
    enumerable: 1,
  });


  Object.defineProperty(this, 'unconfirmedBalance', {
    get: function() {
      return parseFloat(this.unconfirmedBalanceSat) / parseFloat(BitcoreUtil.COIN);
    },
    set:  function(i) {
      this.unconfirmedBalanceSat =  i * BitcoreUtil.COIN;
    },
    enumerable: 1,
  });

}

Address.prototype._getScriptPubKey = function(hex,n) {
  // ScriptPubKey is not provided by bitcoind RPC, so we parse it from tx hex.

  var parser = new Parser(new Buffer(hex,'hex'));
  var tx = new BitcoreTransaction();
  tx.parse(parser);
  return (tx.outs[n].s.toString('hex'));
};

Address.prototype.getUtxo = function(next) {
  var self = this;
  if (!self.addrStr) return next();

  var ret  = [];
  var db   = TransactionDb;

  db.fromAddr(self.addrStr, function(err,txOut){
    if (err) return next(err);

    // Complete utxo info
    async.eachLimit(txOut,CONCURRENCY,function (txItem, a_c) {
      db.fromIdInfoSimple(txItem.txid, function(err, info) {

        var scriptPubKey = self._getScriptPubKey(info.hex, txItem.index);

        // we are filtering out even unconfirmed spents!
        // add || !txItem.spentIsConfirmed 
        if (!txItem.spentTxId) {
          ret.push({
            address: self.addrStr,
            txid: txItem.txid,
            vout: txItem.index,
            ts: txItem.ts,
            scriptPubKey: scriptPubKey,
            amount: txItem.value_sat / BitcoreUtil.COIN,
            confirmations: txItem.isConfirmed ? info.confirmations : 0,
          });
        }
        return a_c(err);
      });
    }, function(err) {
      return next(err,ret);
    });
  });
};

Address.prototype.update = function(next) {
  var self = this;
  if (!self.addrStr) return next();

  var txs  = [];
  var db   = TransactionDb;
  async.series([
    function (cb) {
      var seen={};
      db.fromAddr(self.addrStr, function(err,txOut){
        if (err) return cb(err);
        txOut.forEach(function(txItem){
          var add=0, addSpend=0;
          var v = txItem.value_sat;

          if ( !seen[txItem.txid] ) {
            txs.push({txid: txItem.txid, ts: txItem.ts});
            seen[txItem.txid]=1;
            add=1;
          }

          if (txItem.spentTxId && !seen[txItem.spentTxId]  ) {
            txs.push({txid: txItem.spentTxId, ts: txItem.spentTs});
            seen[txItem.spentTxId]=1;
            addSpend=1;
          }

          if (txItem.isConfirmed) {
            self.txApperances += add;
            self.totalReceivedSat += v;
            if (! txItem.spentTxId ) {
              //unspent
              self.balanceSat   += v;
            }
            else if(!txItem.spentIsConfirmed) {
              // unspent
              self.balanceSat   += v;
              self.unconfirmedBalanceSat -= v;
              self.unconfirmedTxApperances += addSpend;
            }
            else {
              // spent
              self.totalSentSat += v;
              self.txApperances += addSpend;
            }
          }
          else {
            self.unconfirmedBalanceSat += v;
            self.unconfirmedTxApperances += add;
          }
        });
        return cb();
      });
    },
  ], function (err) {

    // sort input and outputs togheter
    txs.sort(
      function compare(a,b) {
        if (a.ts < b.ts) return 1;
        if (a.ts > b.ts) return -1;
        return 0;
      });

    self.transactions = txs.map(function(i) { return i.txid; } );
    return next(err);
  });
};

module.exports = require('soop')(Address);