Wallet/src/js/controllers/import.js

'use strict';

angular.module('copayApp.controllers').controller('importController',
  function($scope, $rootScope, $ionicScrollDelegate, $timeout, $log, profileService, configService, notification, go, sjcl, gettext, lodash, ledger, trezor, derivationPathHelper, platformInfo) {

    var isChromeApp = platformInfo.isChromeApp;
    var isDevel = platformInfo.isDevel;

    var self = this;
    var reader = new FileReader();
    var defaults = configService.getDefaults();
    $scope.bwsurl = defaults.bws.url;
    $scope.derivationPath = derivationPathHelper.default;
    $scope.account = 1;

    window.ignoreMobilePause = true;
    $scope.$on('$destroy', function() {
      $timeout(function() {
        window.ignoreMobilePause = false;
      }, 100);
    });

    var updateSeedSourceSelect = function() {
      self.seedOptions = [];

      if (isChromeApp) {
        self.seedOptions.push({
          id: 'ledger',
          label: 'Ledger Hardware Wallet',
        });
      }

      if (isChromeApp || isDevel) {
        self.seedOptions.push({
          id: 'trezor',
          label: 'Trezor Hardware Wallet',
        });
        $scope.seedSource = self.seedOptions[0];
      }
    };

    this.setType = function(type) {
      $scope.type = type;
      this.error = null;
      $timeout(function() {
        $rootScope.$apply();
      });
    };

    var _importBlob = function(str, opts) {
      var str2, err;
      try {
        str2 = sjcl.decrypt(self.password, str);
      } catch (e) {
        err = gettext('Could not decrypt file, check your spending password');
        $log.warn(e);
      };

      if (err) {
        self.error = err;
        $ionicScrollDelegate.scrollTop();
        $timeout(function() {
          $rootScope.$apply();
        });
        return;
      }

      self.loading = true;
      opts.compressed = null;
      opts.password = null;

      $timeout(function() {
        profileService.importWallet(str2, opts, function(err, walletId) {
          self.loading = false;
          if (err) {
            self.error = err;
            $ionicScrollDelegate.scrollTop();
          } else {
            $rootScope.$emit('Local/WalletImported', walletId);
            notification.success(gettext('Success'), gettext('Your wallet has been imported correctly'));
            go.walletHome();
          }
        });
      }, 100);
    };

    var _importExtendedPrivateKey = function(xPrivKey, opts) {
      self.loading = true;

      $timeout(function() {
        profileService.importExtendedPrivateKey(xPrivKey, opts, function(err, walletId) {
          self.loading = false;
          if (err) {
            self.error = err;
            $ionicScrollDelegate.scrollTop();
            return $timeout(function() {
              $scope.$apply();
            });
          }
          $rootScope.$emit('Local/WalletImported', walletId);
          notification.success(gettext('Success'), gettext('Your wallet has been imported correctly'));
          go.walletHome();
        });
      }, 100);
    };

    var _importMnemonic = function(words, opts) {
      self.loading = true;

      $timeout(function() {
        profileService.importMnemonic(words, opts, function(err, walletId) {
          self.loading = false;
          if (err) {
            self.error = err;
            $ionicScrollDelegate.scrollTop();
            return $timeout(function() {
              $scope.$apply();
            });
          }
          $rootScope.$emit('Local/WalletImported', walletId);
          notification.success(gettext('Success'), gettext('Your wallet has been imported correctly'));
          go.walletHome();
        });
      }, 100);
    };

    $scope.getFile = function() {
      // If we use onloadend, we need to check the readyState.
      reader.onloadend = function(evt) {
        if (evt.target.readyState == FileReader.DONE) { // DONE == 2
          var opts = {};
          opts.bwsurl = $scope.bwsurl;
          _importBlob(evt.target.result, opts);
        }
      }
    };

    this.importBlob = function(form) {
      if (form.$invalid) {
        this.error = gettext('There is an error in the form');
        $ionicScrollDelegate.scrollTop();
        $timeout(function() {
          $scope.$apply();
        });
        return;
      }

      var backupFile = $scope.file;
      var backupText = form.backupText.$modelValue;
      var password = form.password.$modelValue;

      if (!backupFile && !backupText) {
        this.error = gettext('Please, select your backup file');
        $ionicScrollDelegate.scrollTop();
        $timeout(function() {
          $scope.$apply();
        });

        return;
      }

      if (backupFile) {
        reader.readAsBinaryString(backupFile);
      } else {
        var opts = {};
        opts.bwsurl = $scope.bwsurl;
        _importBlob(backupText, opts);
      }
    };

    this.importMnemonic = function(form) {
      if (form.$invalid) {
        this.error = gettext('There is an error in the form');
        $ionicScrollDelegate.scrollTop();
        $timeout(function() {
          $scope.$apply();
        });
        return;
      }

      var opts = {};
      if ($scope.bwsurl)
        opts.bwsurl = $scope.bwsurl;


      var pathData = derivationPathHelper.parse($scope.derivationPath);
      if (!pathData) {
        this.error = gettext('Invalid derivation path');
        $ionicScrollDelegate.scrollTop();
        return;
      }
      opts.account = pathData.account;
      opts.networkName = pathData.networkName;
      opts.derivationStrategy = pathData.derivationStrategy;

      var words = form.words.$modelValue;
      this.error = null;

      if (!words) {
        this.error = gettext('Please enter the recovery phrase');
        $ionicScrollDelegate.scrollTop();
      } else if (words.indexOf('xprv') == 0 || words.indexOf('tprv') == 0) {
        return _importExtendedPrivateKey(words, opts);
      } else {
        var wordList = words.split(/[\u3000\s]+/);

        if ((wordList.length % 3) != 0) {
          this.error = gettext('Wrong number of recovery words:') + wordList.length;
          $ionicScrollDelegate.scrollTop();
        }
      }

      if (this.error) {
        $timeout(function() {
          $scope.$apply();
        });
        return;
      }

      var passphrase = form.passphrase.$modelValue;
      opts.passphrase = form.passphrase.$modelValue || null;

      _importMnemonic(words, opts);
    };

    this.importTrezor = function(account, isMultisig) {
      var self = this;
      trezor.getInfoForNewWallet(isMultisig, account, function(err, lopts) {
        self.hwWallet = false;
        if (err) {
          self.error = err;
          $ionicScrollDelegate.scrollTop();
          $scope.$apply();
          return;
        }

        lopts.externalSource = 'trezor';
        lopts.bwsurl = $scope.bwsurl;
        self.loading = true;
        $log.debug('Import opts', lopts);

        profileService.importExtendedPublicKey(lopts, function(err, walletId) {
          self.loading = false;
          if (err) {
            self.error = err;
            $ionicScrollDelegate.scrollTop();
            return $timeout(function() {
              $scope.$apply();
            });
          }
          $rootScope.$emit('Local/WalletImported', walletId);
          notification.success(gettext('Success'), gettext('Your wallet has been imported correctly'));
          go.walletHome();
        });
      }, 100);
    };

    this.importHW = function(form) {
      if (form.$invalid || $scope.account < 0) {
        this.error = gettext('There is an error in the form');
        $ionicScrollDelegate.scrollTop();
        $timeout(function() {
          $scope.$apply();
        });
        return;
      }
      this.error = '';

      var account = +$scope.account;

      if (self.seedSourceId == 'trezor') {
        if (account < 1) {
          this.error = gettext('Invalid account number');
          $ionicScrollDelegate.scrollTop();
          return;
        }
        account = account - 1;
      }
      var isMultisig = form.isMultisig.$modelValue;

      switch (self.seedSourceId) {
        case ('ledger'):
          self.hwWallet = 'Ledger';
          self.importLedger(account);
          break;
        case ('trezor'):
          self.hwWallet = 'Trezor';
          self.importTrezor(account, isMultisig);
          break;
        default:
          throw ('Error: bad source id');
      };
    };

    this.setSeedSource = function() {
      if (!$scope.seedSource) return;
      self.seedSourceId = $scope.seedSource.id;

      $timeout(function() {
        $rootScope.$apply();
      });
    };

    this.importLedger = function(account) {
      var self = this;
      ledger.getInfoForNewWallet(true, account, function(err, lopts) {
        self.hwWallet = false;
        if (err) {
          self.error = err;
          $ionicScrollDelegate.scrollTop();
          $scope.$apply();
          return;
        }

        lopts.externalSource = 'ledger';
        lopts.bwsurl = $scope.bwsurl;
        self.loading = true;
        $log.debug('Import opts', lopts);

        profileService.importExtendedPublicKey(lopts, function(err, walletId) {
          self.loading = false;
          if (err) {
            self.error = err;
            $ionicScrollDelegate.scrollTop();
            return $timeout(function() {
              $scope.$apply();
            });
          }
          $rootScope.$emit('Local/WalletImported', walletId);
          notification.success(gettext('Success'), gettext('Your wallet has been imported correctly'));
          go.walletHome();
        });
      }, 100);
    };

    updateSeedSourceSelect();
    self.setSeedSource('new');
  });