Merge pull request #1076 from matiu/bug/02-open-wallet

Bug/02 open wallet

css/main.css

@@ -1178,7 +1178,7 @@ a.text-warning:hover {color: #FD7262;}
 
 .wide-page {
   background-color: #2C3E50;
-  margin: 10% 0;
+  margin: 5% 0;
   padding: 50px;
 }
 

index.html

@@ -17,7 +17,8 @@
     <div class="off-canvas-wrap">
       <div class="inner-wrap">
 
-        <nav class="tab-bar" ng-class="{'hide-tab-bar' : !$root.wallet || !$root.wallet.isReady()}">
+        <nav class="tab-bar" ng-class="{'hide-tab-bar' : !$root.wallet ||
+          !$root.wallet.isReady() || $root.wallet.isLocked}">
           <section class="left-small">
             <a class="left-off-canvas-toggle menu-icon" ><span></span></a>
           </section>
@@ -39,12 +40,15 @@
         <div notifications="right top"></div>
 
         <div 
-          ng-class="{'sidebar' : $root.wallet && $root.wallet.isReady()}"
+          ng-class="{'sidebar' : $root.wallet && $root.wallet.isReady() &&
+          !$root.wallet.isLocked}"
           ng-include="'views/includes/sidebar.html'" 
           role='navigation'
-          ng-if="$root.wallet && $root.wallet.isReady()"></div>
+          ng-if="$root.wallet && $root.wallet.isReady() &&
+          !$root.wallet.isLocked"></div>
 
-        <section ng-class="{'main' : $root.wallet && $root.wallet.isReady()}" ng-view></section>
+        <section ng-class="{'main' : $root.wallet && $root.wallet.isReady() &&
+          !$root.wallet.isLocked}" ng-view></section>
 
         <a class="exit-off-canvas"></a>
 
@@ -108,6 +112,7 @@
   <script src="js/controllers/settings.js"></script>
   <script src="js/controllers/uriPayment.js"></script>
   <script src="js/controllers/version.js"></script>
+  <script src="js/controllers/warning.js"></script>
 
   <!-- PLACEHOLDER: CORDOVA SRIPT -->
   <script src="js/mobile.js"></script>

js/controllers/sidebar.js

@@ -1,7 +1,6 @@
 'use strict';
 
-angular.module('copayApp.controllers').controller('SidebarController', 
-    function($scope, $rootScope, $sce, $location, $http, notification, controllerUtils) {
+angular.module('copayApp.controllers').controller('SidebarController', function($scope, $rootScope, $sce, $location, $http, notification, controllerUtils) {
 
   $scope.menu = [{
     'title': 'Receive',
@@ -26,7 +25,9 @@ angular.module('copayApp.controllers').controller('SidebarController',
   };
 
   // Ensures a graceful disconnect
-    window.onbeforeunload = logout;
+  window.onbeforeunload = function() {
+    controllerUtils.logout();
+  };
 
   $scope.$on('$destroy', function() {
     window.onbeforeunload = undefined;
@@ -73,5 +74,4 @@ angular.module('copayApp.controllers').controller('SidebarController',
       notification.warning('Session closed', 'Session closed because a long time of inactivity');
     });
   }
-
-  });
+});

js/controllers/warning.js

@@ -0,0 +1,28 @@
+'use strict';
+angular.module('copayApp.controllers').controller('WarningController', function($scope, $rootScope, $location, controllerUtils) {
+
+
+  $scope.checkLock = function() {
+    if (!$rootScope.tmp || !$rootScope.tmp.getLock()) {
+      controllerUtils.redirIfLogged();
+    }
+  };
+
+
+  $scope.signout = function() {
+    controllerUtils.logout();
+  };
+
+  $scope.ignoreLock = function() {
+    var w = $rootScope.tmp;
+    delete $rootScope['tmp'];
+
+    if (!w) {
+      $location.path('/');
+    } else {
+      w.ignoreLock = 1;
+      $scope.loading = true;
+      controllerUtils.startNetwork(w, $scope);
+    }
+  };
+});

js/models/core/Wallet.js

@@ -44,6 +44,7 @@ function Wallet(opts) {
   this.id = opts.id || Wallet.getRandomId();
   this.name = opts.name;
 
+  this.ignoreLock = opts.ignoreLock;
   this.verbose = opts.verbose;
   this.publicKeyRing.walletId = this.id;
   this.txProposals.walletId = this.id;
@@ -92,6 +93,27 @@ Wallet.prototype.connectToAll = function() {
   }
 };
 
+Wallet.prototype.getLock = function() {
+  return this.storage.getLock(this.id);
+};
+
+Wallet.prototype.setLock = function() {
+  return this.storage.setLock(this.id);
+};
+
+Wallet.prototype.unlock = function() {
+  this.storage.removeLock(this.id);
+};
+
+Wallet.prototype.checkAndLock = function() {
+  if (this.getLock()) {
+    return true;
+  } 
+
+  this.setLock();
+  return false;
+};
+
 Wallet.prototype._handleIndexes = function(senderId, data, isInbound) {
   this.log('RECV INDEXES:', data);
   var inIndexes = HDParams.fromList(data.indexes);
@@ -112,7 +134,7 @@ Wallet.prototype._handlePublicKeyRing = function(senderId, data, isInbound) {
   try {
     hasChanged = this.publicKeyRing.merge(inPKR, true);
   } catch (e) {
-    this.log('## WALLET ERROR', e); //TODO
+    this.log('## WALLET ERROR', e);
     this.emit('connectionError', e.message);
     return;
   }
@@ -306,9 +328,10 @@ Wallet.prototype._handleData = function(senderId, data, isInbound) {
 
   if (data.type !== 'walletId' && this.id !== data.walletId) {
     this.emit('badMessage', senderId);
-    this.log('badMessage FROM:', senderId); //TODO
+    this.log('badMessage FROM:', senderId);
     return;
   }
+
   switch (data.type) {
     // This handler is repeaded on WalletFactory (#join). TODO
     case 'walletId':
@@ -409,6 +432,12 @@ Wallet.prototype._lockIncomming = function() {
 Wallet.prototype.netStart = function(callback) {
   var self = this;
   var net = this.network;
+
+  if (this.checkAndLock() && !this.ignoreLock) {
+    this.emit('locked');
+    return;
+  }
+
   net.removeAllListeners();
   net.on('connect', self._handleConnect.bind(self));
   net.on('disconnect', self._handleDisconnect.bind(self));
@@ -722,7 +751,6 @@ Wallet.prototype.sendTx = function(ntxid, cb) {
     } else {
       self.log('Sent failed. Checking is the TX was sent already');
       self._checkSentTx(ntxid, function(txid) {
-        console.log('[Wallet.js.730:txid:]', txid); //TODO
         if (txid)
           self.store();
 
@@ -993,6 +1021,7 @@ Wallet.prototype.indexDiscovery = function(start, change, cosigner, gap, cb) {
 
 Wallet.prototype.disconnect = function() {
   this.log('## DISCONNECTING');
+  this.unlock();
   this.network.disconnect();
 };
 

js/models/storage/LocalEncrypted.js

@@ -15,7 +15,7 @@ function Storage(opts) {
     this.localStorage = opts.localStorage;
   } else if (localStorage) {
     this.localStorage = localStorage;
-}
+  }
 }
 
 var pps = {};
@@ -180,6 +180,18 @@ Storage.prototype.getLastOpened = function() {
   return this.getGlobal('lastOpened');
 }
 
+Storage.prototype.setLock = function(walletId) {
+  this.setGlobal(this._key(walletId, 'Lock'), true);
+}
+
+Storage.prototype.getLock = function(walletId) {
+  return this.getGlobal(this._key(walletId, 'Lock'));
+}
+
+Storage.prototype.removeLock = function(walletId) {
+  this.removeGlobal(this._key(walletId, 'Lock'));
+}
+
 //obj contains keys to be set
 Storage.prototype.setFromObj = function(walletId, obj) {
   for (var k in obj) {

js/routes.js

@@ -2,8 +2,8 @@
 
 //Setting up route
 angular
-  .module('copayApp')
-  .config(function($routeProvider) {
+.module('copayApp')
+.config(function($routeProvider) {
 
   $routeProvider
   .when('/', {
@@ -56,39 +56,59 @@ angular
   .when('/uri-payment/:data', {
     templateUrl: 'views/uri-payment.html'
   })
+  .when('/warning', {
+    templateUrl: 'views/warning.html',
+    validate: true
+  })
   .otherwise({
     templateUrl: 'views/errors/404.html',
     title: 'Error'
   });
-  });
+});
 
 //Setting HTML5 Location Mode
 angular
-  .module('copayApp')
-  .config(function($locationProvider, $idleProvider) {
+.module('copayApp')
+.config(function($locationProvider, $idleProvider) {
   $locationProvider
   .html5Mode(false)
   .hashPrefix('!');
   // IDLE timeout
   $idleProvider.idleDuration(15 * 60); // in seconds
   $idleProvider.warningDuration(10); // in seconds
-  })
-  .run(function($rootScope, $location, $idle) {
+})
+.run(function($rootScope, $location, $idle) {
   $idle.watch();
   $rootScope.$on('$routeChangeStart', function(event, next, current) {
     if (!util.supports.data) {
       $location.path('unsupported');
     } else {
+
+      // Locked?
+      if ($rootScope.showLockWarning) {
+        if ($rootScope.tmp) {
+          if ($location.path() !== '/warning') {
+            $location.path('/warning');
+          }
+          else {
+            delete $rootScope['showLockWarning'];
+          }
+        }
+        return;
+      }
+
       if ((!$rootScope.wallet || !$rootScope.wallet.id) && next.validate) {
         $idle.unwatch();
         $location.path('/');
       }
+
+      // In creation?
       if ($rootScope.wallet && !$rootScope.wallet.isReady()) {
         $location.path('/copayers');
       }
     }
   });
-  })
-  .config(function($compileProvider) {
+})
+.config(function($compileProvider) {
   $compileProvider.aHrefSanitizationWhitelist(/^\s*(https?|ftp|mailto|file|tel|chrome-extension|resource):/);
-  });
+});

js/services/controllerUtils.js

@@ -2,7 +2,7 @@
 var bitcore = require('bitcore');
 
 angular.module('copayApp.services')
-  .factory('controllerUtils', function($rootScope, $sce, $location, notification, $timeout, Socket, video, uriHandler) {
+.factory('controllerUtils', function($rootScope, $sce, $location, notification, $timeout, Socket, video, uriHandler) {
   var root = {};
   root.getVideoMutedStatus = function(copayer) {
     if (!$rootScope.videoInfo) return;
@@ -15,17 +15,20 @@ angular.module('copayApp.services')
   };
 
   root.redirIfLogged = function() {
-      var w = $rootScope.wallet;
-      if (w) {
-        $location.path('addresses');
+    if ($rootScope.wallet) {
+      $rootScope.wallet.path('receive');
     }
   };
 
   root.logout = function() { 
+    if ($rootScope.wallet)
+      $rootScope.wallet.disconnect();
+
     Socket.removeAllListeners();
 
-      $rootScope.wallet = null;
+    $rootScope.wallet = $rootScope.tmp = null;
     delete $rootScope['wallet'];
+
     video.close();
     // Clear rootScope
     for (var i in $rootScope) {
@@ -69,6 +72,7 @@ angular.module('copayApp.services')
   root.setupRootVariables = function() {
     uriHandler.register();
     $rootScope.unitName = config.unitName;
+    $rootScope.showLockWarning = false;
     $rootScope.txAlertCount = 0;
     $rootScope.insightError = 0;
     $rootScope.isCollapsed = true;
@@ -121,6 +125,12 @@ angular.module('copayApp.services')
     };
 
     notification.enableHtml5Mode(); // for chrome: if support, enable it
+    w.on('locked', function() {
+      $rootScope.tmp = w;
+      $rootScope.showLockWarning=true;
+      $location.path('/warning');
+      $rootScope.$digest();
+    });
 
     w.on('badMessage', function(peerId) {
       notification.error('Error', 'Received wrong message from peer ' + peerId);
@@ -369,4 +379,4 @@ angular.module('copayApp.services')
     }
   };
   return root;
-  });
+});

test/mocks/FakeStorage.js

@@ -27,6 +27,18 @@ FakeStorage.prototype.getLastOpened = function() {
   return this.storage['lastOpened'];
 };
 
+FakeStorage.prototype.setLock = function(id) {
+  this.storage[id + '::lock'] = true;
+}
+
+FakeStorage.prototype.getLock = function(id) {
+  return this.storage[id + '::lock'];
+}
+
+FakeStorage.prototype.removeLock = function(id) {
+  delete this.storage[id + '::lock'];
+}
+
 FakeStorage.prototype.removeGlobal = function(id) {
   delete this.storage[id];
 };

test/mocks/FakeWallet.js

@@ -4,6 +4,7 @@ var FakeWallet = function() {
   this.safeBalance = 1000;
   this.totalCopayers = 2;
   this.requiredCopayers = 2;
+  this.isLocked = false;
   this.balanceByAddr = {
     '1CjPR7Z5ZSyWk6WtXvSFgkptmpoi4UM9BC': 1000
   };

test/test.Wallet.js

@@ -181,6 +181,7 @@ describe('Wallet model', function() {
       cachedW2obj.opts.reconnectDelay = 100;
     }
     var w = Wallet.fromObj(cachedW2obj, cachedW2.storage, cachedW2.network, cachedW2.blockchain);
+    w.unlock();
     return w;
   };
 
@@ -1022,6 +1023,37 @@ describe('Wallet model', function() {
       w.netStart();
       w.network.start.getCall(0).args[0].privkey.length.should.equal(64);
     });
+
+    it('should check if wallet is already opened', function() {
+      var w = cachedCreateW2();
+      should.not.exist(w.getLock());
+      w.checkAndLock().should.equal(false);
+      w.getLock().should.equal(true);
+    });
+    it('should check if wallet is already opened', function() {
+      var w = cachedCreateW2();
+      should.not.exist(w.getLock());
+      w.checkAndLock().should.equal(false);
+      w.getLock().should.equal(true);
+    });
+
+
+    it('should not start if locked', function() {
+      var w = cachedCreateW2();
+      w.netStart();
+      w.emit = sinon.spy();
+      w.netStart();
+      w.emit.getCall(0).args[0].should.equal('locked');
+    });
+
+    it('should accept ignoreLocked', function() {
+      var w = cachedCreateW2();
+      w.netStart();
+      w.network.start = sinon.spy();
+      w.ignoreLock=1;
+      w.netStart();
+      w.network.start.getCall(0).args[0].privkey.length.should.equal(64);
+    });
   });
 
   describe('#forceNetwork in config', function() {

test/test.storage.LocalEncrypted.js

@@ -160,6 +160,19 @@ describe('Storage/LocalEncrypted model', function() {
     });
   });
 
+  describe('#WalletLock', function() {
+    it('should get/set/remove opened', function() {
+      var s = new LocalEncrypted({
+        localStorage: localMock,
+        password: 'password'
+      });
+      s.setLock('walletId');
+      s.getLock('walletId').should.equal(true);
+      s.removeLock('walletId');
+      should.not.exist(s.getLock('walletId'));
+    });
+  });
+
   describe('#getWallets', function() {
     it('should retreive wallets from storage', function() {
       var s = new LocalEncrypted({

test/unit/controllers/controllersSpec.js

@@ -288,8 +288,10 @@ describe("Unit: Controllers", function() {
   describe("Unit: Sidebar Controller", function() {
     var rootScope;
     beforeEach(inject(function($controller, $rootScope) {
-      rootScope = $rootScope;
       scope = $rootScope.$new();
+      rootScope = $rootScope;
+      rootScope.wallet = new FakeWallet(config);
+
       headerCtrl = $controller('SidebarController', {
         $scope: scope,
       });
@@ -437,4 +439,18 @@ describe("Unit: Controllers", function() {
     });
   });
 
+  describe('Warning Controller', function() {
+    var what;
+    beforeEach(inject(function($controller, $rootScope) {
+      scope = $rootScope.$new();
+      what = $controller('WarningController', {
+        $scope: scope,
+      });
+    }));
+
+    it('should exist', function() {
+      should.exist(what);
+    });
+  });
+
 });

util/build.js

@@ -97,14 +97,14 @@ var createBundle = function(opts) {
     expose: '../js/models/core/HDPath'
   }); 
 
-  if (opts.dontminify) {
+  if (opts.debug) {
     //include dev dependencies
     b.require('sinon');
     b.require('blanket');
     b.require('soop');
   }
 
-  if (!opts.dontminify) {
+  if (!opts.debug) {
     b.transform({
       global: true
     }, 'uglifyify');
@@ -120,7 +120,7 @@ if (require.main === module) {
   var program = require('commander');
   program
     .version('0.0.1')
-    .option('-d, --dontminify', 'Development. Don\'t minify the code.')
+    .option('-d, --debug', 'Development. Don\'t minify the codem and include debug packages.')
     .option('-o, --stdout', 'Specify output as stdout')
     .parse(process.argv);
 

views/warning.html

@@ -0,0 +1,26 @@
+<div class="wide-page" ng-controller="WarningController"
+  ng-init="checkLock()">
+  <div class="text-center">
+    <img src="img/logo-negative-beta.svg" alt="Copay">
+    <div class="text-white" ng-include="'views/includes/version.html'"></div>
+  </div>
+  <h1 class="text-center text-warning">Warning!</h1>
+  <h3 class="text-center text-white">
+    This wallet appears to be currently open.
+    <br>
+    Opening the wallet in multiple browser tabs could lead to unexpected results
+  </h3>
+  <div class="text-center m30v large-12 columns">
+    <div class="row">
+      <div class="large-12 columns medium-12 small-12 text-center">
+        <a href class="button sucess" ng-click="signout()">Go back</a> 
+        <br>
+        <br>
+        <a href ng-click="ignoreLock()">Continue anyways</a> 
+      </div>
+    </div>
+  </div>
+  <div class="text-center text-gray small cb">
+  </div>
+</div>
+