Wallet/src/js/services/trezor.js

'use strict';

angular.module('copayApp.services')
  .factory('trezor', function($log, $timeout, bwcService, gettext, lodash) {
    var root = {};

    root.ENTROPY_INDEX_PATH = "0xb11e/";
    root.callbacks = {};

    root.getEntropySource = function(account, callback) {
      var path = root.ENTROPY_INDEX_PATH + account + "'";
      var xpub = root.getXPubKey(path, function(data) {
        if (!data.success) {
          $log.warn(data.message);
          return callback(data);
        }

        var b = bwcService.getBitcore();

        var x = b.HDPublicKey(data.xpubkey);
        data.entropySource = x.publicKey.toString();
        return callback(data);
      });
    };

    root.getXPubKeyForAddresses = function(account, callback) {
      return root.getXPubKey(root._getPath(account), callback);
    };

    root.getXPubKey = function(path, callback) {
      $log.debug('TREZOR deriving xPub path:', path);
      TrezorConnect.getXPubKey(path, callback);
    };


    root.getInfoForNewWallet = function(account, callback) {
      var opts = {};
      root.getEntropySource(account, function(data) {
        if (!data.success) {
          $log.warn(data.message);
          return callback(data.message);
        }
        opts.entropySource = data.entropySource;
        $log.debug('Waiting TREZOR to settle...');
        $timeout(function() {
          root.getXPubKeyForAddresses(account, function(data) {
            if (!data.success) {
              $log.warn(data.message);
              return callback(data);
            }
            opts.extendedPublicKey = data.xpubkey;
            opts.externalSource = 'trezor';
            opts.externalIndex = account;
            return callback(null, opts);
          });
        }, 5000);
      });
    };


    root.signTx = function(xPubKeys, txp, account, callback) {
      console.log('[trezor.js.66:txp:]', xPubKeys, txp); //TODO

      var inputs = [],
        outputs = [];
      var tmpOutputs = [];

      if (txp.type != 'simple')
        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,
      });



      if (txp.addressType == 'P2PKH') {

        var inAmount = 0;
        inputs = lodash.map(txp.inputs, function(i) {
          var pathArr = i.path.split('/');
          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,
          };
        });

        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'
          });
        }

      } else {

        // P2SH Wallet
        var inAmount = 0;

        var sigs = xPubKeys.map(function(v) {
          return '';
        });


        inputs = lodash.map(txp.inputs, function(i) {
          var pathArr = i.path.split('/');
          var n = [44 | 0x80000000, 0 | 0x80000000, account | 0x80000000, parseInt(pathArr[1]), parseInt(pathArr[2])];
//          var np = [parseInt(pathArr[1]), parseInt(pathArr[2])];
          inAmount += i.satoshis;

          var pubkeys = lodash(xPubKeys.map(function(v) {
            return {
              node: v,
              address_n: n,
            };
          })).reverse().value();
console.log('[trezor.js.121:pubkeys:]',pubkeys); //TODO

          return {
            address_n: n,
            prev_index: i.vout,
            prev_hash: i.txid,
            script_type: 'SPENDMULTISIG',
            multisig: {
              pubkeys: pubkeys,
              signatures: sigs,
              m: txp.requiredSignatures,
            }
          };
        });

        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])];

          var pubkeys = lodash(xPubKeys.map(function(v) {
            return {
              node: v,
              address_n: n,
            };
          })).reverse().value();

          tmpOutputs.push({
            address_n: n,
            amount: change,
            script_type: 'PAYTOMULTISIG',
            multisig: {
              pubkeys: pubkeys,
              signatures: sigs,
              m: txp.requiredSignatures,
            }
          });
        }
      }

      // 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 + "'";
    }

    return root;
  });