'use strict'; var _ = require('lodash'); var levelup = require('levelup'); var $ = require('preconditions').singleton(); var async = require('async'); var log = require('npmlog'); log.debug = log.verbose; var Wallet = require('./model/wallet'); var Copayer = require('./model/copayer'); var Address = require('./model/address'); var Storage = function (opts) { opts = opts || {}; this.db = opts.db || levelup(opts.dbPath || './db/copay.db', { valueEncoding: 'json' }); }; Storage.prototype.fetchWallet = function (id, cb) { this.db.get('wallet-' + id, function (err, data) { if (err) { if (err.notFound) return cb(); return cb(err); } return cb(null, Wallet.fromObj(data)); }); }; Storage.prototype.fetchCopayer = function (walletId, copayerId, cb) { this.db.get('wallet-' + walletId + '-copayer-' + copayerId, function (err, data) { if (err) { if (err.notFound) return cb(); return cb(err); } return cb(null, Copayer.fromObj(data)); }); }; Storage.prototype.fetchCopayers = function (walletId, cb) { var copayers = []; var key = 'wallet-' + walletId + '-copayer-'; this.db.createReadStream({ gte: key, lt: key + '~' }) .on('data', function (data) { copayers.push(Copayer.fromObj(data.value)); }) .on('error', function (err) { if (err.notFound) return cb(); return cb(err); }) .on('end', function () { return cb(null, copayers); }); }; Storage.prototype.storeWallet = function (wallet, cb) { this.db.put('wallet-' + wallet.id, wallet, cb); }; Storage.prototype.storeCopayer = function (walletId, copayer, cb) { this.db.put('wallet-' + walletId + '-copayer-' + copayer.id, copayer, cb); }; Storage.prototype.storeAddress = function (walletId, address, cb) { this.db.put('wallet-' + walletId + '-address-' + address.address, address, cb); }; Storage.prototype.storeTx = function (walletId, tx, cb) { this.db.put('wallet-' + walletId + '-tx-' + tx.ntxid, tx, cb); }; Storage.prototype.getAddresses = function (walletId, cb) { var addresses = []; var key = 'wallet-' + walletId + '-address-'; this.db.createReadStream({ gte: key, lt: key + '~' }) .on('data', function (data) { addresses.push(Address.fromObj(data.value)); }) .on('error', function (err) { if (err.notFound) return cb(); return cb(err); }) .on('end', function () { return cb(null, addresses); }); }; Storage.prototype._dump = function (cb) { this.db.readStream() .on('data', console.log) .on('end', function () { if (cb) return cb(); }); }; module.exports = Storage;