Wallet/js/models/Storage.js

'use strict';

var CryptoJS = require('node-cryptojs-aes').CryptoJS;
var bitcore = require('bitcore');
var preconditions = require('preconditions').instance();
var id = 0;

function Storage(opts) {
  opts = opts || {};

  this.__uniqueid = ++id;
  if (opts.password)
    this._setPassphrase(opts.password);

  try {
    this.storage = opts.storage || localStorage;
    this.sessionStorage = opts.sessionStorage || sessionStorage;
  } catch (e) {
    console.log('Error in storage:', e); //TODO
  };

  preconditions.checkState(this.localStorage, 'No localstorage found');
  preconditions.checkState(this.sessionStorage, 'No sessionStorage found');
}

var pps = {};
Storage.prototype._getPassphrase = function() {
  if (!pps[this.__uniqueid])
    throw new Error('No passprase set');

  return pps[this.__uniqueid];
}

Storage.prototype._setPassphrase = function(password) {
  pps[this.__uniqueid] = password;
}

Storage.prototype._encrypt = function(string) {
  var encrypted = CryptoJS.AES.encrypt(string, this._getPassphrase());
  var encryptedBase64 = encrypted.toString();
  return encryptedBase64;
};

Storage.prototype._decrypt = function(base64) {
  var decryptedStr = null;
  try {
    var decrypted = CryptoJS.AES.decrypt(base64, this._getPassphrase());
    if (decrypted)
      decryptedStr = decrypted.toString(CryptoJS.enc.Utf8);
  } catch (e) {
    // Error while decrypting
    return null;
  }
  return decryptedStr;
};


Storage.prototype._read = function(k) {
  var ret;
  ret = this.storage.getItem(k);
  if (!ret) return null;
  ret = this._decrypt(ret);
  if (!ret) return null;
  ret = ret.toString(CryptoJS.enc.Utf8);
  ret = JSON.parse(ret);
  return ret;
};

Storage.prototype._write = function(k, v) {
  v = JSON.stringify(v);
  v = this._encrypt(v);

  this.storage.setItem(k, v);
};

// get value by key
Storage.prototype.getGlobal = function(k) {
  var item = this.storage.getItem(k);
  return item == 'undefined' ? undefined : item;
};

// set value for key
Storage.prototype.setGlobal = function(k, v) {
  this.storage.setItem(k, typeof v === 'object' ? JSON.stringify(v) : v);
};

// remove value for key
Storage.prototype.removeGlobal = function(k) {
  this.storage.removeItem(k);
};

Storage.prototype.getSessionId = function() {
  var sessionId = this.sessionStorage.getItem('sessionId');
  if (!sessionId) {
    sessionId = bitcore.SecureRandom.getRandomBuffer(8).toString('hex');
    this.sessionStorage.setItem('sessionId', sessionId);
  }
  return sessionId;
};

Storage.prototype._key = function(walletId, k) {
  return walletId + '::' + k;
};
// get value by key
Storage.prototype.get = function(walletId, k) {
  var ret = this._read(this._key(walletId, k));
  return ret;
};

// set value for key
Storage.prototype.set = function(walletId, k, v) {
  this._write(this._key(walletId, k), v);
};

// remove value for key
Storage.prototype.remove = function(walletId, k) {
  this.removeGlobal(this._key(walletId, k));
};

Storage.prototype.setName = function(walletId, name) {
  this.setGlobal('nameFor::' + walletId, name);
};

Storage.prototype.getName = function(walletId) {
  var ret = this.getGlobal('nameFor::' + walletId);
  return ret;
};

Storage.prototype.getWalletIds = function() {
  var walletIds = [];
  var uniq = {};

console.log('[Encrypted.js.144]', this.storage); //TODO
console.log('[Encrypted.js.144]', this.storage.length); //TODO
  for (var i = 0; i < this.storage.length; i++) {
    var key = this.storage.key(i);
    var split = key.split('::');
    if (split.length == 2) {
      var walletId = split[0];

      if (!walletId || walletId === 'nameFor' || walletId === 'lock')
        continue;

      if (typeof uniq[walletId] === 'undefined') {
        walletIds.push(walletId);
        uniq[walletId] = 1;
      }
    }
  }
  return walletIds;
};

Storage.prototype.getWallets = function() {
  var wallets = [];
  var ids = this.getWalletIds();

  for (var i in ids) {
    wallets.push({
      id: ids[i],
      name: this.getName(ids[i]),
    });
  }
  return wallets;
};

Storage.prototype.deleteWallet = function(walletId) {
  var toDelete = {};
  toDelete['nameFor::' + walletId] = 1;

  for (var i = 0; i < this.storage.length; i++) {
    var key = this.storage.key(i);
    var split = key.split('::');
    if (split.length == 2 && split[0] === walletId) {
      toDelete[key] = 1;
    }
  }
  for (var i in toDelete) {
    this.removeGlobal(i);
  }
};

Storage.prototype.setLastOpened = function(walletId) {
  this.setGlobal('lastOpened', walletId);
}

Storage.prototype.getLastOpened = function() {
  return this.getGlobal('lastOpened');
}

//obj contains keys to be set
Storage.prototype.setFromObj = function(walletId, obj) {
  for (var k in obj) {
    this.set(walletId, k, obj[k]);
  }
  this.setName(walletId, obj.opts.name);
};

// remove all values
Storage.prototype.clearAll = function() {
  this.storage.clear();
};

Storage.prototype.import = function(base64) {
  var decryptedStr = this._decrypt(base64);
  return JSON.parse(decryptedStr);
};

Storage.prototype.export = function(obj) {
  var string = JSON.stringify(obj);
  return this._encrypt(string);
};

module.exports = Storage;