From 8eecdd1e86be2c99745bd950938c3aa2e2642144 Mon Sep 17 00:00:00 2001
From: Matias Alejo Garcia <ematiu@gmail.com>
Date: Mon, 28 Sep 2015 20:25:28 -0300
Subject: [PATCH] trezor sign 1-1 working

---
 public/views/create.html          |  2 +-
 src/js/services/profileService.js | 11 ++--
 src/js/services/trezor.js         | 88 ++++++++++++++++++++-----------
 3 files changed, 62 insertions(+), 39 deletions(-)

diff --git a/public/views/create.html b/public/views/create.html
index c11944d16..6f26bf575 100644
--- a/public/views/create.html
+++ b/public/views/create.html
@@ -110,7 +110,7 @@
 
             <label ng-show="1" for="hw-trezor" class="oh">
               <span translate>Use TREZOR hardware wallet</span>
-              <switch id="hw-trezor" name="hwTrezor" ng-model="hwTrezor" ng-change="isTestnet=false" class="green right m5t m10b"></switch>
+              <switch id="hw-trezor" name="hwTrezor" ng-model="hwTrezor" class="green right m5t m10b"></switch>
             </label>
 <!-- TODO account
             <div ng-show="hwLedger">
diff --git a/src/js/services/profileService.js b/src/js/services/profileService.js
index c3dd68916..dbb677d84 100644
--- a/src/js/services/profileService.js
+++ b/src/js/services/profileService.js
@@ -552,8 +552,9 @@ console.log('[profileService.js.239:walletClient:]',walletClient); //TODO
       $log.info('Requesting Ledger Chrome app to sign the transaction');
 
       ledger.signTx(txp, 0, function(result) {
+        $log.debug('Ledger response',result);
         if (!result.success)
-          return cb(result);
+          return cb(result.message || result.error);
 
         txp.signatures = lodash.map(result.signatures, function(s) {
           return s.substring(0, s.length - 2);
@@ -568,13 +569,11 @@ console.log('[profileService.js.239:walletClient:]',walletClient); //TODO
       $log.info('Requesting Trezor  to sign the transaction');
 
       trezor.signTx(txp, 0, function(result) {
-console.log('[profileService.js.570:result:]',result); //TODO
+        $log.debug('Trezor response',result);
         if (!result.success)
-          return cb(result);
+          return cb(result.error || result);
 
-        txp.signatures = lodash.map(result.signatures, function(s) {
-          return s.substring(0, s.length - 2);
-        });
+        txp.signatures = result.signatures;
         return fc.signTxProposal(txp, cb);
       });
     };
diff --git a/src/js/services/trezor.js b/src/js/services/trezor.js
index 951b21c5c..385428042 100644
--- a/src/js/services/trezor.js
+++ b/src/js/services/trezor.js
@@ -51,7 +51,6 @@ angular.module('copayApp.services')
             opts.extendedPublicKey = data.xpubkey;
             opts.externalSource = 'trezor';
             opts.externalIndex = account;
-            console.log('[trezor.js.51:opts:]', opts); //TODO
             return callback(null, opts);
           });
         }, 5000);
@@ -60,52 +59,77 @@ angular.module('copayApp.services')
 
 
     root.signTx = function(txp, account, callback) {
+      console.log('[trezor.js.66:txp:]', txp); //TODO
+
+      var inputs = [],
+        outputs = [];
+      var tmpOutputs = [];
+
       if (txp.type != 'simple')
-        return callback('Only simple TXs are supported in TREZOR');
+        return callback('Only TXPs type SIMPLE are supported in TREZOR');
+
+      var toScriptType = 'PAYTOADDRESS';
+      if (txp.toAddress.charAt(0) == '2' || txp.toAddress.charAt(0) == '3')
+        toScriptType = 'PAYTOSCRIPTHASH';
+
+
+      // Add to
+      tmpOutputs.push({
+        address: txp.toAddress,
+        amount: txp.amount,
+        script_type: toScriptType,
+      });
+
+
 
-      console.log('[trezor.js.90:txp:]', txp); //TODO
       if (txp.addressType == 'P2PKH') {
-        var tx = bwcService.getUtils().buildTx(txp);
 
-        // spend one change output
-        var inputs = lodash.map(txp.inputs, function(i){
+        var inAmount = 0;
+        inputs = lodash.map(txp.inputs, function(i) {
           var pathArr = i.path.split('/');
-console.log('[trezor.js.72:pathArr:]',pathArr); //TODO
-          var n = [44 | 0x80000000, 0 | 0x80000000, account | 0x80000000, parseInt(pathArr[1]) , parseInt(pathArr[2])];
-console.log('[trezor.js.74]', n); //TODO
+          var n = [44 | 0x80000000, 0 | 0x80000000, account | 0x80000000, parseInt(pathArr[1]), parseInt(pathArr[2])];
+          inAmount += i.satoshis;
           return {
             address_n: n,
             prev_index: i.vout,
             prev_hash: i.txid,
           };
         });
-console.log('[trezor.js.68:inputs:]',inputs); //TODO
 
-        var pathArr = txp.changeAddress.path.split('/');
-console.log('[trezor.js.82:pathArr:]',pathArr); //TODO
-        var n = [44 | 0x80000000, 0 | 0x80000000, account | 0x80000000, parseInt(pathArr[1]) , parseInt(pathArr[2])];
-        // send to 1 address output and one change output
-        var outputs = [{
-          address_n: n,
-          amount: txp.amount - txp.fee,
-          script_type: 'PAYTOADDRESS'
-        }, {
-          address: txp.toAddress, 
-          amount: txp.amount,
-          script_type: 'PAYTOADDRESS'
-        }];
-console.log('[trezor.js.84:outputs:]',outputs); //TODO
+        var change = inAmount - txp.fee - txp.amount;
+        if (change > 0) {
+          var pathArr = txp.changeAddress.path.split('/');
+          var n = [44 | 0x80000000, 0 | 0x80000000, account | 0x80000000, parseInt(pathArr[1]), parseInt(pathArr[2])];
+
+          tmpOutputs.push({
+            address_n: n,
+            amount: change,
+            script_type: 'PAYTOADDRESS'
+          });
+        }
 
-        TrezorConnect.signTx(inputs, outputs, function(result) {
-          console.log('[trezor.js.78:result:]', result); //TODO
-          callback(result);
-        });
       } else {
-        var msg = 'P2SH wallets are not supported with TREZOR';
-        $log.error(msg);
-        return callback(msg);
+        $log.error('TODO: multisig');
       }
-    }
+
+      // Shuffle outputs for improved privacy
+      if (tmpOutputs.length > 1) {
+        lodash.each(txp.outputOrder, function(order) {
+          outputs[order] = tmpOutputs.shift();
+        });
+
+        if (tmpOutputs.length) 
+          return cb("Error creating transaction: tmpOutput order");
+      } else {
+        outputs = tmpOutputs;
+      }
+
+
+      $log.debug('Signing with TREZOR', inputs, outputs);
+      TrezorConnect.signTx(inputs, outputs, function(result) {
+        callback(result);
+      });
+    };
 
     root._getPath = function(account) {
       return "44'/0'/" + account + "'";