const inherits = require('util').inherits const EventEmitter = require('events').EventEmitter const async = require('async') const KeyStore = require('eth-lightwallet').keystore const createPayload = require('web3-provider-engine/util/create-payload') var selectedAddress = null var identities = {} module.exports = IdentityManager var provider = null var pubsub = new EventEmitter() inherits(IdentityManager, EventEmitter) function IdentityManager(opts){ const self = this self.on('block', function(){ self.updateIdentities() }) } // plugin popup IdentityManager.prototype.getState = getState IdentityManager.prototype.submitPassword = submitPassword IdentityManager.prototype.setSelectedAddress = setSelectedAddress IdentityManager.prototype.signTransaction = signTransaction IdentityManager.prototype.setLocked = setLocked // eth rpc IdentityManager.prototype.getAccounts = getAccounts IdentityManager.prototype.confirmTransaction = confirmTransaction // etc IdentityManager.prototype.newBlock = newBlock IdentityManager.prototype.setProvider = setProvider function setProvider(_provider){ provider = _provider } function newBlock(block){ var self = this self.emit('block', block) } function getState(cb){ var result = _getState() cb(null, result) } function _getState(cb){ var unlocked = isUnlocked() var result = { isUnlocked: unlocked, identities: unlocked ? getIdentities() : {}, selectedAddress: selectedAddress, } return result } function isUnlocked(){ var password = window.sessionStorage['password'] var result = Boolean(password) return result } function setLocked(){ delete window.sessionStorage['password'] } function setSelectedAddress(address, cb){ selectedAddress = address cb(null, _getState()) } function submitPassword(password, cb){ const self = this console.log('submitPassword:', password) tryPassword(password, function(err){ if (err) console.log('bad password:', password, err) if (err) return cb(err) console.log('good password:', password) window.sessionStorage['password'] = password // load identities before returning... self.loadIdentities() var state = _getState() cb(null, state) // trigger an update but dont wait for it console.log(self) self.updateIdentities() }) } // get the current selected address function getAccounts(cb){ var result = selectedAddress ? [selectedAddress] : [] console.log('getAccounts:', result) cb(null, result) } function getIdentities(){ return identities } // load identities from keyStore IdentityManager.prototype.loadIdentities = function(){ const self = this if (!isUnlocked()) throw new Error('not unlocked') var keyStore = getKeyStore() var addresses = keyStore.getAddresses().map(function(address){ return '0x'+address }) addresses.forEach(function(address){ var identity = { name: 'Wally', img: 'QmW6hcwYzXrNkuHrpvo58YeZvbZxUddv69ATSHY3BHpPdd', address: address, balance: null, txCount: null, } identities[address] = identity }) self.emit('update', _getState()) } // foreach in identities, update balance + nonce IdentityManager.prototype.updateIdentities = function(cb){ var self = this cb = cb || function(){} if (!isUnlocked()) return cb(new Error('Not unlocked.')) var addresses = Object.keys(identities) async.map(addresses, self.updateIdentity.bind(self), cb) } // gets latest info from the network for the identity IdentityManager.prototype.updateIdentity = function(address, cb){ var self = this async.parallel([ getAccountBalance.bind(null, address), getTxCount.bind(null, address), ], function(err, result){ if (err) return cb(err) var identity = identities[address] identity.balance = result[0] identity.txCount = result[1] console.log('updated!') self.emit('update', _getState()) cb() }) } function getTxCount(address, cb){ provider.sendAsync(createPayload({ method: 'eth_getTransactionCount', // we actually want the pending txCount // but pending is broken in provider-engine // https://github.com/MetaMask/provider-engine/issues/11 // params: [address, 'pending'], params: [address, 'latest'], }), function(err, res){ if (err) return cb(err) if (res.error) return cb(res.error) cb(null, res.result) }) } function getAccountBalance(address, cb){ provider.sendAsync(createPayload({ method: 'eth_getBalance', params: [address, 'latest'], }), function(err, res){ if (err) return cb(err) if (res.error) return cb(res.error) cb(null, res.result) }) } function tryPassword(password, cb){ var keyStore = getKeyStore(password) var address = keyStore.getAddresses()[0] if (!address) return cb(new Error('KeyStore - No address to check.')) var hdPathString = keyStore.defaultHdPathString try { var encKey = keyStore.generateEncKey(password) var encPrivKey = keyStore.ksData[hdPathString].encPrivKeys[address] var privKey = KeyStore._decryptKey(encPrivKey, encKey) var addrFromPrivKey = KeyStore._computeAddressFromPrivKey(privKey) } catch (err) { return cb(err) } if (addrFromPrivKey !== address) return cb(new Error('KeyStore - Decrypting private key failed!')) cb() } function confirmTransaction(txParams, cb){ console.log('confirmTransaction:', txParams) } function signTransaction(txParams, cb){ console.log('signTransaction:', txParams) } var keyStore = null function getKeyStore(password){ if (keyStore) return keyStore password = password || getPassword() var serializedKeystore = window.localStorage['lightwallet'] // returning user if (serializedKeystore) { keyStore = KeyStore.deserialize(serializedKeystore) // first time here } else { var defaultPassword = 'test' console.log('creating new keystore with default password:', defaultPassword) var secretSeed = KeyStore.generateRandomSeed() keyStore = new KeyStore(secretSeed, defaultPassword) keyStore.generateNewAddress(defaultPassword, 3) saveKeystore() } keyStore.passwordProvider = unlockKeystore return keyStore } function saveKeystore(){ window.localStorage['lightwallet'] = keyStore.serialize() } function getPassword(){ var password = window.sessionStorage['password'] if (!password) throw new Error('No password found...') } function unlockKeystore(cb){ var password = getPassword() console.warn('unlocking keystore...') cb(null, password) }