Wallet/js/models/core/TxProposals.js

'use strict';


var imports = require('soop').imports();
var bitcore = require('bitcore');
var util = bitcore.util;
var Transaction = bitcore.Transaction;
var Builder = bitcore.TransactionBuilder;
var Script = bitcore.Script;
var buffertools = bitcore.buffertools;

function TxProposal(opts) {
  this.creator = opts.creator;
  this.createdTs = opts.createdTs;
  this.seenBy = opts.seenBy || {};
  this.signedBy = opts.signedBy || {};
  this.rejectedBy = opts.rejectedBy || {};
  this.builder = opts.builder;
  this.sentTs = opts.sentTs || null;
  this.sentTxid = opts.sentTxid || null;
  this.inputChainPaths = opts.inputChainPaths || [];
  this.comment = opts.comment || null;
}

TxProposal.prototype.getID = function() {
  var ntxid = this.builder.build().getNormalizedHash().toString('hex');
  return ntxid;
};

TxProposal.prototype.toObj = function() {
  var o = JSON.parse(JSON.stringify(this));
  delete o['builder'];
  o.builderObj = this.builder.toObj();
  return o;
};


TxProposal.prototype.setSent = function(sentTxid) {
  this.sentTxid = sentTxid;
  this.sentTs = Date.now();
};

TxProposal.fromObj = function(o) {
  var t = new TxProposal(o);
  var b = new Builder.fromObj(o.builderObj);
  t.builder = b;
  return t;
};

TxProposal.getSentTs = function() {
  return this.sentTs;
};

TxProposal.prototype.merge = function(other) {
  var ret = {};
  ret.events = this.mergeMetadata(other);
  ret.hasChanged = this.mergeBuilder(other);
  return ret;
};

TxProposal.prototype.mergeBuilder = function(other) {
  var b0 = this.builder;
  var b1 = other.builder;

  // TODO: improve this comparison
  var before = JSON.stringify(b0.toObj());
  b0.merge(b1);
  var after = JSON.stringify(b0.toObj());
  return after !== before;
};

TxProposal.prototype.mergeMetadata = function(v1) {
  var events = [];
  var v0 = this;

  var ntxid = this.getID();

  Object.keys(v1.seenBy).forEach(function(k) {
    if (!v0.seenBy[k]) {
      v0.seenBy[k] = v1.seenBy[k];
      events.push({
        type: 'seen',
        cId: k,
        txId: ntxid
      });
    }
  });

  Object.keys(v1.signedBy).forEach(function(k) {
    if (!v0.signedBy[k]) {
      v0.signedBy[k] = v1.signedBy[k];
      events.push({
        type: 'signed',
        cId: k,
        txId: ntxid
      });
    }
  });

  Object.keys(v1.rejectedBy).forEach(function(k) {
    if (!v0.rejectedBy[k]) {
      v0.rejectedBy[k] = v1.rejectedBy[k];
      events.push({
        type: 'rejected',
        cId: k,
        txId: ntxid
      });
    }
  });

  if (!v0.sentTxid && v1.sentTxid) {
    v0.sentTs = v1.sentTs;
    v0.sentTxid = v1.sentTxid;
    events.push({
      type: 'broadcast',
      txId: ntxid
    });
  }

  return events;

};

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


function TxProposals(opts) {
  opts = opts || {};
  this.walletId = opts.walletId;
  this.network = opts.networkName === 'livenet' ?
    bitcore.networks.livenet : bitcore.networks.testnet;
  this.txps = {};
}

TxProposals.fromObj = function(o) {
  var ret = new TxProposals({
    networkName: o.networkName,
    walletId: o.walletId,
  });
  o.txps.forEach(function(o2) {
    var t = TxProposal.fromObj(o2);
    var id = t.builder.build().getNormalizedHash().toString('hex');
    ret.txps[id] = t;
  });
  return ret;
};

TxProposals.prototype.getNtxids = function() {
  return Object.keys(this.txps);
};

TxProposals.prototype.toObj = function(onlyThisNtxid) {
  if (onlyThisNtxid) throw new Error();
  var ret = [];
  for (var id in this.txps) {

    if (onlyThisNtxid && id != onlyThisNtxid)
      continue;

    var t = this.txps[id];
    if (!t.sent)
      ret.push(t.toObj());
  }
  return {
    txps: ret,
    walletId: this.walletId,
    networkName: this.network.name,
  };
};

TxProposals.prototype.merge = function(inTxp) {
  var myTxps = this.txps;

  var ntxid = inTxp.getID();
  var ret = {};
  ret.events = [];
  ret.events.hasChanged = false;

  if (myTxps[ntxid]) {
    var v0 = myTxps[ntxid];
    var v1 = inTxp;
    ret = v0.merge(v1);
  } else {
    ret.hasChanged = true;
    ret.events.push({
      type: 'new',
      cid: inTxp.creator,
      tx: ntxid
    });
  }
  return ret;
};

TxProposals.prototype.add = function(data) {
  var txp = new TxProposal(data);
  var ntxid = txp.getID();
  this.txps[ntxid] = txp;
  return ntxid;
};

TxProposals.prototype.setSent = function(ntxid, txid) {
  //sent TxProposals are local an not broadcasted.
  this.txps[ntxid].setSent(txid);
};


TxProposals.prototype.getTxProposal = function(ntxid, copayers) {
  var txp = this.txps[ntxid];
  var i = JSON.parse(JSON.stringify(txp));
  i.builder = txp.builder;
  i.ntxid = ntxid;
  i.peerActions = {};

  if (copayers) {
    for (var j = 0; j < copayers.length; j++) {
      var p = copayers[j];
      i.peerActions[p] = {};
    }
  }

  for (var p in txp.seenBy) {
    i.peerActions[p] = {
      seen: txp.seenBy[p]
    };
  }
  for (var p in txp.signedBy) {
    i.peerActions[p] = i.peerActions[p] || {};
    i.peerActions[p].sign = txp.signedBy[p];
  }
  var r = 0;
  for (var p in txp.rejectedBy) {
    i.peerActions[p] = i.peerActions[p] || {};
    i.peerActions[p].rejected = txp.rejectedBy[p];
    r++;
  }
  i.rejectCount = r;

  var c = txp.creator;
  i.peerActions[c] = i.peerActions[c] || {};
  i.peerActions[c].create = txp.createdTs;
  return i;
};

//returns the unspent txid-vout used in PENDING Txs
TxProposals.prototype.getUsedUnspent = function(maxRejectCount) {
  var ret = {};
  for (var i in this.txps) {
    var u = this.txps[i].builder.getSelectedUnspent();
    var p = this.getTxProposal(i);
    if (p.rejectCount > maxRejectCount || p.sentTxid)
      continue;

    for (var j in u) {
      ret[u[j].txid + ',' + u[j].vout] = 1;
    }
  }
  return ret;
};

TxProposals.TxProposal = TxProposal;
module.exports = require('soop')(TxProposals);