From 6af9ee68e2a314d8741ae16f544ef5755722df0c Mon Sep 17 00:00:00 2001
From: Manuel Araoz <manuelaraoz@gmail.com>
Date: Fri, 30 May 2014 15:07:52 -0300
Subject: [PATCH] include derivation path in tx proposal

---
 js/models/core/PrivateKey.js    | 12 ++++++++++--
 js/models/core/PublicKeyRing.js | 11 ++++++++++-
 js/models/core/TxProposals.js   |  1 +
 js/models/core/Wallet.js        | 16 ++++++++++++----
 4 files changed, 33 insertions(+), 7 deletions(-)

diff --git a/js/models/core/PrivateKey.js b/js/models/core/PrivateKey.js
index 086f2561b..cc2a4d715 100644
--- a/js/models/core/PrivateKey.js
+++ b/js/models/core/PrivateKey.js
@@ -61,8 +61,11 @@ PrivateKey.prototype._getHK = function(path) {
   return this.bip.derive(path);
 };
 
-PrivateKey.prototype.get = function(index,isChange) {
-  var path = Structure.FullBranch(index, isChange);
+PrivateKey.prototype.getForPaths = function(paths) {
+  return paths.map(this.getForPath.bind(this));
+};
+
+PrivateKey.prototype.getForPath = function(path) {
   var pk = this.privateKeyCache[path];
   if (!pk) {
     var derivedHK =  this._getHK(path);
@@ -73,6 +76,11 @@ PrivateKey.prototype.get = function(index,isChange) {
   return wk;
 };
 
+PrivateKey.prototype.get = function(index,isChange) {
+  var path = Structure.FullBranch(index, isChange);
+  return this.getForPath(path);
+};
+
 PrivateKey.prototype.getAll = function(addressIndex, changeAddressIndex) {
   var ret = [];
   for(var i=0;i<addressIndex; i++) {
diff --git a/js/models/core/PublicKeyRing.js b/js/models/core/PublicKeyRing.js
index 12a5317bd..52ad802b7 100644
--- a/js/models/core/PublicKeyRing.js
+++ b/js/models/core/PublicKeyRing.js
@@ -36,6 +36,7 @@ function PublicKeyRing(opts) {
   this.publicKeysCache = opts.publicKeysCache || {};
   this.nicknameFor = opts.nicknameFor || {};
   this.copayerIds = [];
+  this.addressToPath = {};
 }
 
 PublicKeyRing.fromObj = function (data) {
@@ -181,7 +182,15 @@ PublicKeyRing.prototype.getRedeemScript = function (index, isChange) {
 // TODO this could be cached
 PublicKeyRing.prototype.getAddress = function (index, isChange) {
   var script  = this.getRedeemScript(index,isChange);
-  return Address.fromScript(script, this.network.name);
+  var address = Address.fromScript(script, this.network.name);
+  this.addressToPath[address.toString()] = Structure.FullBranch(index, isChange);
+  return address;
+};
+
+PublicKeyRing.prototype.pathForAddress = function(address) {
+  var path = this.addressToPath[address];
+  if (!path) throw new Error('Couldn\'t find path for address '+address);
+  return path;
 };
 
 // TODO this could be cached
diff --git a/js/models/core/TxProposals.js b/js/models/core/TxProposals.js
index eaf9c75e4..12aab20e5 100644
--- a/js/models/core/TxProposals.js
+++ b/js/models/core/TxProposals.js
@@ -22,6 +22,7 @@ function TxProposal(opts) {
   this.builder  = opts.builder;
   this.sentTs = opts.sentTs || null;
   this.sentTxid = opts.sentTxid || null;
+  this.inputChainPaths = opts.inputChainPaths || [];
 }
 
 TxProposal.prototype.toObj = function() {
diff --git a/js/models/core/Wallet.js b/js/models/core/Wallet.js
index 7ffbfa58b..cb6a49057 100644
--- a/js/models/core/Wallet.js
+++ b/js/models/core/Wallet.js
@@ -432,7 +432,7 @@ Wallet.prototype.sign = function(ntxid, cb) {
     }
 
     var pkr = self.publicKeyRing;
-    var keys = self.privateKey.getAll(pkr.addressIndex, pkr.changeAddressIndex);
+    var keys = self.privateKey.getForPaths(txp.inputChainPaths);
 
     var b = txp.builder;
     var before = b.signaturesAdded;
@@ -626,9 +626,15 @@ Wallet.prototype.createTxSync = function(toAddress, amountSatStr, utxos, opts) {
       amountSat: amountSat
     }]);
 
-  var signRet;
+  var selectedUtxos = b.getSelectedUnspent();
+  
+  var inputChainPaths = selectedUtxos.map(function(utxo) {
+    return pkr.pathForAddress(utxo.address);
+  });
+
   if (priv) {
-    var signed = b.sign(priv.getAll(pkr.addressIndex, pkr.changeAddressIndex));
+    var keys = priv.getForPaths(inputChainPaths);
+    var signed = b.sign(keys);
   }
   var myId = this.getMyCopayerId();
   var now = Date.now();
@@ -640,6 +646,7 @@ Wallet.prototype.createTxSync = function(toAddress, amountSatStr, utxos, opts) {
   if (priv) meSeen[myId] = now;
 
   var data = {
+    inputChainPaths: inputChainPaths,
     signedBy: me,
     seenBy: meSeen,
     creator: myId,
@@ -647,7 +654,8 @@ Wallet.prototype.createTxSync = function(toAddress, amountSatStr, utxos, opts) {
     builder: b,
   };
 
-  return this.txProposals.add(data);
+  var ntxid = this.txProposals.add(data);
+  return ntxid;
 };
 
 Wallet.prototype.disconnect = function() {