adding wallet factory

js/models/core/Wallet.js

@@ -1,13 +1,15 @@
 'use strict';
 
-var imports     = require('soop').imports();
+var imports = require('soop').imports();
 
-var bitcore     = require('bitcore');
-var http        = require('http');
+var bitcore = require('bitcore');
+var coinUtil = bitcore.util;
+var buffertools = bitcore.buffertools;
+var http = require('http');
 
-var Storage     = imports.Storage;
-var Network     = imports.Network;
-var Blockchain  = imports.Blockchain;
+var Storage = imports.Storage;
+var Network = imports.Network;
+var Blockchain = imports.Blockchain;
 
 var copay = copay || require('../../../copay');
 
@@ -18,31 +20,33 @@ function Wallet(config) {
 
 
 Wallet.prototype._startInterface = function(config) {
-  this.storage    = new Storage(config.storage);
-  this.network    = new Network(config.network);
+  this.storage = new Storage(config.storage);
+  this.network = new Network(config.network);
   this.blockchain = new Blockchain(config.blockchain);
 };
 
- 
+
 Wallet.prototype._createNew = function(config, opts) {
 
-  console.log('### CREATING NEW WALLET.' 
-    + (opts.walletId ? ' USING ID: ' +opts.walletId  : ' NEW ID') );
+  this.id = opts.id || Wallet.getRandomId();
+  console.log('### CREATING NEW WALLET.' + (opts.id ? ' USING ID: ' + opts.id : ' NEW ID'));
 
-  this.privateKey = new copay.PrivateKey({networkName: config.networkName});
+  this.privateKey = new copay.PrivateKey({
+    networkName: config.networkName
+  });
   console.log('\t### PrivateKey Initialized');
 
   this.publicKeyRing = opts.publicKeyRing || new copay.PublicKeyRing({
-    id: opts.walletId,
+    id: this.id,
     requiredCopayers: opts.requiredCopayers || config.wallet.requiredCopayers,
-    totalCopayers: opts.totalCopayers       || config.wallet.totalCopayers,
+    totalCopayers: opts.totalCopayers || config.wallet.totalCopayers,
     networkName: config.networkName,
   });
   this.publicKeyRing.addCopayer(this.privateKey.getBIP32().extendedPublicKeyString());
-  console.log('\t### PublicKeyRing Initialized WalletID: ' +  this.publicKeyRing.id);
+  console.log('\t### PublicKeyRing Initialized WalletID: ' + this.publicKeyRing.id);
 
   this.txProposals = opts.txProposals || new copay.TxProposals({
-    walletId: this.publicKeyRing.id,
+    walletId: this.id,
     publicKeyRing: this.publicKeyRing,
     networkName: config.networkName,
   });
@@ -51,14 +55,14 @@ Wallet.prototype._createNew = function(config, opts) {
 
 
 Wallet.prototype._load = function(config, walletId) {
-  this.id             = walletId; 
-  this.publicKeyRing  = new copay.PublicKeyRing.fromObj(
+  this.id = walletId;
+  this.publicKeyRing = new copay.PublicKeyRing.fromObj(
     this.storage.get(this.id, 'publicKeyRing')
   );
-  this.txProposals    = new copay.TxProposals.fromObj(
+  this.txProposals = new copay.TxProposals.fromObj(
     this.storage.get(this.id, 'txProposals')
   );
-  this.privateKey     = new copay.PrivateKey.fromObj(
+  this.privateKey = new copay.PrivateKey.fromObj(
     this.storage.get(this.id, 'privateKey')
   ); //TODO secure
 
@@ -76,7 +80,7 @@ Wallet.prototype._load = function(config, walletId) {
 // CONSTRUCTORS
 Wallet.read = function(config, walletId) {
   var w = new Wallet(config);
-  w.load(walletId);
+  w._load(walletId);
 
   return w;
 };
@@ -87,8 +91,56 @@ Wallet.create = function(config, opts) {
 
   return w;
 };
-  
 
+Wallet.getRandomId = function() {
+  var r = buffertools.toHex(coinUtil.generateNonce());
+  return r;
+};
+
+Wallet.prototype.store = function() {
+  // TODO store each variable
+};
+
+var WalletFactory = function() {
+  this.storage = Storage.
+  default ();
+};
+
+WalletFactory.prototype.create = function(config, opts) {
+  var w = new Wallet.create(config, opts);
+  w.store();
+  this._addWalletId(w.id);
+  return w;
+};
+
+WalletFactory.prototype.get = function(config, walletId) {
+  return Wallet.read(config, walletId);
+};
+
+WalletFactory.prototype.remove = function(walletId) {
+  // TODO remove wallet contents, not only the id (Wallet.remove?)
+  this._delWalletId(walletId);
+};
+
+WalletFactory.prototype._addWalletId = function(walletId) {
+  var ids = this._getWalletIds();
+  if (ids.indexOf(walletId) == -1) return;
+  storage.set('walletIds', (ids ? ids + ',' : '') + walletId);
+};
+
+WalletFactory.prototype._delWalletId = function(walletId) {
+  var ids = this._getWalletIds();
+  var index = ids.indexOf(walletId);
+  if (index == -1) return;
+  ids.splice(index, 1); // removes walletId
+  this.storage.set('walletIds', ids.join(','));
+};
+
+WalletFactory.prototype._getWalletIds = function() {
+  var ids = this.storage.get('walletIds');
+  return ids ? ids.split(',') : [];
+};
+
+Wallet.factory = new WalletFactory();
 
 module.exports = require('soop')(Wallet);
-