Merge pull request #22 from matiu/feat/delete-wallet

Feat/delete wallet

lib/server.js

@@ -460,6 +460,15 @@ CopayServer.prototype.getTx = function(opts, cb) {
   });
 };
 
+
+/**
+ * Remove wallet
+ */
+CopayServer.prototype.removeWallet = function(opts, cb) {
+  this.storage.removeWallet(this.walletId, cb);
+};
+
+
 CopayServer.prototype._broadcastTx = function(txp, cb) {
   var raw = txp.getRawTx();
   var bc = this._getBlockExplorer('insight', txp.getNetworkName());

lib/storage.js

@@ -2,6 +2,7 @@
 
 var _ = require('lodash');
 var levelup = require('levelup');
+var async = require('async');
 var $ = require('preconditions').singleton();
 var log = require('npmlog');
 log.debug = log.verbose;
@@ -19,6 +20,10 @@ var Storage = function(opts) {
 };
 
 
+var walletPrefix = function(id) {
+  return 'w!' + id;
+};
+
 var opKey = function(key) {
   return key ? '!' + key : '';
 };
@@ -30,20 +35,20 @@ var opKeyTs = function(key) {
 
 
 var KEY = {
-  WALLET: function(id) {
-    return 'wallet!' + id;
+  WALLET: function(walletId) {
+    return walletPrefix(walletId) + '!main';
   },
   COPAYER: function(id) {
     return 'copayer!' + id;
   },
   TXP: function(walletId, txProposalId) {
-    return 'txp!' + walletId + opKey(txProposalId);
+    return walletPrefix(walletId) + '!txp' + opKey(txProposalId);
   },
   PENDING_TXP: function(walletId, txProposalId) {
-    return 'pending-txp-ts!' + walletId + opKey(txProposalId);
+    return walletPrefix(walletId) + '!ptxp' + opKey(txProposalId);
   },
   ADDRESS: function(walletId, address) {
-    return 'address!' + walletId + opKey(address);
+    return walletPrefix(walletId) + '!addr' + opKey(address);
   },
 };
 
@@ -182,6 +187,104 @@ Storage.prototype.storeTx = function(walletId, txp, cb) {
   this.db.batch(ops, cb);
 };
 
+Storage.prototype.removeTx = function(walletId, txProposalId, cb) {
+  var ops = [{
+    type: 'del',
+    key: KEY.TXP(walletId, txp.id),
+  }, {
+    type: 'del',
+    key: KEY.PENDING_TXP(walletId, txp.id),
+    value: txp,
+  }];
+
+  this.db.batch(ops, cb);
+};
+
+Storage.prototype._delByKey = function(key, cb) {
+  var self = this;
+  var keys = [];
+  this.db.createKeyStream({
+    gte: key,
+    lt: key + '~',
+  })
+    .on('data', function(key) {
+      keys.push(key);
+    })
+    .on('error', function(err) {
+      if (err.notFound) return cb();
+
+      console.log('[storage.js.252]'); //TODO
+      return cb(err);
+    })
+    .on('end', function(err) {
+      self.db.batch(_.map(keys, function(k) {
+        return {
+          key: k,
+          type: 'del'
+        };
+      }), function(err) {
+        return cb(err);
+      });
+    });
+};
+
+Storage.prototype.removeAllPendingTxs = function(walletId, cb) {
+  this._delByKey(KEY.PENDING_TXP(walletId), cb);
+};
+
+Storage.prototype.removeAllTxs = function(walletId, cb) {
+  this._delByKey(KEY.TXP(walletId), cb);
+};
+
+
+Storage.prototype._removeCopayers = function(walletId, cb) {
+  var self = this;
+
+  this.fetchWallet(walletId, function(err, w) {
+    if (err || !w) return cb(err);
+
+    self.db.batch(_.map(w.copayers, function(c) {
+      return {
+        type: 'del',
+        key: KEY.COPAYER(c.id),
+      };
+    }), cb);
+  });
+};
+
+
+
+Storage.prototype._removeAllAddresses = function(walletId, cb) {
+  this._delByKey(KEY.ADDRESS(walletId), cb);
+};
+
+Storage.prototype.removeWallet = function(walletId, cb) {
+  var self = this;
+
+  async.series([
+    function(next) {
+      self._removeCopayers(walletId, next);
+    },
+    function(next) {
+      self._removeAllAddresses(walletId, next);
+    },
+    function(next) {
+      self.removeAllPendingTxs(walletId, next);
+    },
+    function(next) {
+      self.removeAllTxs(walletId, next);
+    },
+    function(next) {
+      var ops = [{
+        type: 'del',
+        key: KEY.WALLET(walletId),
+      }];
+      self.db.batch(ops, next);
+    },
+  ], cb);
+};
+
+
 Storage.prototype.fetchAddresses = function(walletId, cb) {
   var addresses = [];
   var key = KEY.ADDRESS(walletId);
@@ -219,9 +322,11 @@ Storage.prototype.removeAddress = function(walletId, address, cb) {
 };
 
 
-Storage.prototype._dump = function(cb) {
+Storage.prototype._dump = function(cb, fn) {
+  fn = fn || console.log;
+
   this.db.readStream()
-    .on('data', console.log)
+    .on('data', fn)
     .on('end', function() {
       if (cb) return cb();
     });

test/integration.js

@@ -1123,4 +1123,89 @@ describe('Copay server', function() {
       });
     });
   });
+
+
+  describe('#removeWallet', function() {
+    var server, wallet, clock;
+
+    beforeEach(function(done) {
+      helpers.createAndJoinWallet(1, 1, function(s, w) {
+        server = s;
+        wallet = w;
+
+        server.createAddress({}, function(err, address) {
+          helpers.createUtxos(server, wallet, helpers.toSatoshi(_.range(2)), function(utxos) {
+            helpers.stubBlockExplorer(server, utxos);
+            var txOpts = {
+              toAddress: '18PzpUFkFZE8zKWUPvfykkTxmB9oMR8qP7',
+              amount: helpers.toSatoshi(0.1),
+            };
+            async.eachSeries(_.range(2), function(i, next) {
+              server.createTx(txOpts, function(err, tx) {
+                next();
+              });
+            }, done);
+          });
+        });
+      });
+    });
+    it('should delete a wallet', function(done) {
+      var i = 0;
+      var count = function() {
+        return ++i;
+      };
+      server.storage._dump(function() {
+        i.should.above(1);
+        server.removeWallet({}, function(err) {
+          i = 0;
+          server.storage._dump(function() {
+            i.should.equal(0);
+            done();
+          }, count);
+        });
+      }, count);
+    });
+
+    // creates 2 wallet, and deletes only 1.
+    it('should delete a wallet, and only that wallet', function(done) {
+      var i = 0;
+      var db = [];
+      var cat = function(data) {
+        db.push(data);
+      };
+      server.storage._dump(function() {
+        var before = _.clone(db);
+        db.length.should.above(1);
+
+        helpers.createAndJoinWallet(2, 3, function(s, w) {
+          server = s;
+          wallet = w;
+
+          server.createAddress({}, function(err, address) {
+            helpers.createUtxos(server, wallet, helpers.toSatoshi(_.range(2)), function(utxos) {
+              helpers.stubBlockExplorer(server, utxos);
+              var txOpts = {
+                toAddress: '18PzpUFkFZE8zKWUPvfykkTxmB9oMR8qP7',
+                amount: helpers.toSatoshi(0.1),
+              };
+              async.eachSeries(_.range(2), function(i, next) {
+                server.createTx(txOpts, function(err, tx) {
+                  next();
+                });
+              }, function() {
+                server.removeWallet({}, function(err) {
+                  db=[];
+                  server.storage._dump(function() {
+                    var after = _.clone(db);
+                    after.should.deep.equal(before);
+                    done();
+                  }, cat);
+                });
+              }, cat);
+            });
+          });
+        });
+      }, cat);
+    });
+  });
 });