/*!
 * walletdb.js - storage for wallets
 * Copyright (c) 2014-2015, Fedor Indutny (MIT License)
 * Copyright (c) 2014-2017, Christopher Jeffrey (MIT License).
 * https://github.com/bcoin-org/bcoin
 */

'use strict';

const assert = require('assert');
const path = require('path');
const EventEmitter = require('events');
const bio = require('bufio');
const {BloomFilter} = require('bfilter');
const {Lock, MapLock} = require('bmutex');
const BDB = require('bdb');
const Logger = require('blgr');
const ccmp = require('bcrypto/lib/ccmp');
const aes = require('bcrypto/lib/aes');
const Network = require('../protocol/network');
const Path = require('./path');
const common = require('./common');
const Wallet = require('./wallet');
const Account = require('./account');
const Outpoint = require('../primitives/outpoint');
const layouts = require('./layout');
const records = require('./records');
const NullClient = require('./nullclient');
const {encoding} = bio;
const {u32} = encoding;
const layout = layouts.walletdb;

const {
  ChainState,
  BlockMeta,
  TXRecord,
  MapRecord
} = records;

/**
 * WalletDB
 * @alias module:wallet.WalletDB
 * @constructor
 * @param {Object} options
 */

function WalletDB(options) {
  if (!(this instanceof WalletDB))
    return new WalletDB(options);

  EventEmitter.call(this);

  this.options = new WalletOptions(options);

  this.network = this.options.network;
  this.logger = this.options.logger.context('wallet');
  this.workers = this.options.workers;
  this.client = this.options.client || new NullClient(this);
  this.feeRate = this.options.feeRate;
  this.db = new BDB(this.options);

  this.primary = null;
  this.state = new ChainState();
  this.height = 0;
  this.wallets = new Map();
  this.depth = 0;
  this.rescanning = false;

  // Wallet read lock.
  this.readLock = new MapLock();

  // Wallet write lock (creation and rename).
  this.writeLock = new Lock();

  // Lock for handling anything tx related.
  this.txLock = new Lock();

  // Address and outpoint filter.
  this.filter = new BloomFilter();

  this.init();
}

Object.setPrototypeOf(WalletDB.prototype, EventEmitter.prototype);

/**
 * Database layout.
 * @type {Object}
 */

WalletDB.layout = layout;

/**
 * Initialize walletdb.
 * @private
 */

WalletDB.prototype.init = function init() {
  let items = 3000000;
  let flag = -1;

  // Highest number of items with an
  // FPR of 0.001. We have to do this
  // by hand because BloomFilter.fromRate's
  // policy limit enforcing is fairly
  // naive.
  if (this.options.spv) {
    items = 20000;
    flag = BloomFilter.flags.ALL;
  }

  this.filter = BloomFilter.fromRate(items, 0.001, flag);
  this._bind();
};

/**
 * Bind to node events.
 * @private
 */

WalletDB.prototype._bind = function _bind() {
  this.client.on('error', (err) => {
    this.emit('error', err);
  });

  this.client.on('connect', async () => {
    try {
      await this.syncNode();
    } catch (e) {
      this.emit('error', e);
    }
  });

  this.client.bind('block connect', async (entry, txs) => {
    try {
      await this.addBlock(entry, txs);
    } catch (e) {
      this.emit('error', e);
    }
  });

  this.client.bind('block disconnect', async (entry) => {
    try {
      await this.removeBlock(entry);
    } catch (e) {
      this.emit('error', e);
    }
  });

  this.client.hook('block rescan', async (entry, txs) => {
    try {
      await this.rescanBlock(entry, txs);
    } catch (e) {
      this.emit('error', e);
    }
  });

  this.client.bind('tx', async (tx) => {
    try {
      await this.addTX(tx);
    } catch (e) {
      this.emit('error', e);
    }
  });

  this.client.bind('chain reset', async (tip) => {
    try {
      await this.resetChain(tip);
    } catch (e) {
      this.emit('error', e);
    }
  });
};

/**
 * Open the walletdb, wait for the database to load.
 * @returns {Promise}
 */

WalletDB.prototype.open = async function open() {
  await this.db.open();
  await this.db.checkVersion('V', 7);

  this.depth = await this.getDepth();

  if (this.options.wipeNoReally)
    await this.wipe();

  await this.watch();
  await this.connect();

  this.logger.info(
    'WalletDB loaded (depth=%d, height=%d, start=%d).',
    this.depth,
    this.state.height,
    this.state.startHeight);

  const wallet = await this.ensure({
    id: 'primary'
  });

  const addr = await wallet.receiveAddress();

  this.logger.info(
    'Loaded primary wallet (id=%s, wid=%d, address=%s)',
    wallet.id, wallet.wid, addr.toString(this.network));

  this.primary = wallet;
};

/**
 * Close the walletdb, wait for the database to close.
 * @returns {Promise}
 */

WalletDB.prototype.close = async function close() {
  await this.disconnect();

  for (const wallet of this.wallets.values()) {
    await wallet.destroy();
    this.unregister(wallet);
  }

  await this.db.close();
};

/**
 * Watch addresses and outpoints.
 * @private
 * @returns {Promise}
 */

WalletDB.prototype.watch = async function watch() {
  const piter = this.db.iterator({
    gte: layout.p(encoding.NULL_HASH),
    lte: layout.p(encoding.HIGH_HASH)
  });

  let hashes = 0;

  await piter.each((key) => {
    const data = layout.pp(key);

    this.filter.add(data, 'hex');

    hashes += 1;
  });

  this.logger.info('Added %d hashes to WalletDB filter.', hashes);

  const oiter = this.db.iterator({
    gte: layout.o(encoding.NULL_HASH, 0),
    lte: layout.o(encoding.HIGH_HASH, 0xffffffff)
  });

  let outpoints = 0;

  await oiter.each((key) => {
    const [hash, index] = layout.oo(key);
    const outpoint = new Outpoint(hash, index);
    const data = outpoint.toRaw();

    this.filter.add(data);

    outpoints += 1;
  });

  this.logger.info('Added %d outpoints to WalletDB filter.', outpoints);
};

/**
 * Connect to the node server (client required).
 * @returns {Promise}
 */

WalletDB.prototype.connect = async function connect() {
  return this.client.open();
};

/**
 * Disconnect from node server (client required).
 * @returns {Promise}
 */

WalletDB.prototype.disconnect = async function disconnect() {
  return this.client.close();
};

/**
 * Sync state with server on every connect.
 * @returns {Promise}
 */

WalletDB.prototype.syncNode = async function syncNode() {
  const unlock = await this.txLock.lock();
  try {
    this.logger.info('Resyncing from server...');
    await this.syncState();
    await this.syncFilter();
    await this.syncChain();
    await this.resend();
  } finally {
    unlock();
  }
};

/**
 * Initialize and write initial sync state.
 * @returns {Promise}
 */

WalletDB.prototype.syncState = async function syncState() {
  const cache = await this.getState();

  if (cache) {
    this.state = cache;
    this.height = cache.height;
    return;
  }

  this.logger.info('Initializing database state from server.');

  const b = this.db.batch();

  let tip = null;

  const hashes = await this.client.getHashes();

  for (let height = 0; height < hashes.length; height++) {
    const hash = hashes[height];
    const meta = new BlockMeta(hash, height);
    b.put(layout.h(height), meta.toHash());
    tip = meta;
  }

  assert(tip);

  const state = this.state.clone();
  state.startHeight = tip.height;
  state.startHash = tip.hash;
  state.height = tip.height;
  state.marked = false;

  b.put(layout.R, state.toRaw());

  await b.write();

  this.state = state;
  this.height = state.height;
};

/**
 * Connect and sync with the chain server.
 * @private
 * @returns {Promise}
 */

WalletDB.prototype.syncChain = async function syncChain() {
  let height = this.state.height;

  this.logger.info('Syncing state from height %d.', height);

  for (;;) {
    const tip = await this.getBlock(height);
    assert(tip);

    if (await this.client.getEntry(tip.hash))
      break;

    assert(height !== 0);
    height -= 1;
  }

  await this.scan(height);
};

/**
 * Rescan blockchain from a given height.
 * @private
 * @param {Number?} height
 * @returns {Promise}
 */

WalletDB.prototype.scan = async function scan(height) {
  if (height == null)
    height = this.state.startHeight;

  assert((height >>> 0) === height, 'WDB: Must pass in a height.');

  await this.rollback(height);

  this.logger.info(
    'WalletDB is scanning %d blocks.',
    this.state.height - height + 1);

  const tip = await this.getTip();

  try {
    this.rescanning = true;
    await this.client.rescan(tip.hash);
  } finally {
    this.rescanning = false;
  }
};

/**
 * Force a rescan.
 * @param {Number} height
 * @returns {Promise}
 */

WalletDB.prototype.rescan = async function rescan(height) {
  const unlock = await this.txLock.lock();
  try {
    return await this._rescan(height);
  } finally {
    unlock();
  }
};

/**
 * Force a rescan (without a lock).
 * @private
 * @param {Number} height
 * @returns {Promise}
 */

WalletDB.prototype._rescan = async function _rescan(height) {
  return this.scan(height);
};

/**
 * Broadcast a transaction via chain server.
 * @param {TX} tx
 * @returns {Promise}
 */

WalletDB.prototype.send = async function send(tx) {
  return this.client.send(tx);
};

/**
 * Estimate smart fee from chain server.
 * @param {Number} blocks
 * @returns {Promise}
 */

WalletDB.prototype.estimateFee = async function estimateFee(blocks) {
  if (this.feeRate > 0)
    return this.feeRate;

  const rate = await this.client.estimateFee(blocks);

  if (rate < this.network.feeRate)
    return this.network.feeRate;

  if (rate > this.network.maxFeeRate)
    return this.network.maxFeeRate;

  return rate;
};

/**
 * Send filter to the remote node.
 * @private
 * @returns {Promise}
 */

WalletDB.prototype.syncFilter = function syncFilter() {
  this.logger.info('Sending filter to server (%dmb).',
    this.filter.size / 8 / (1 << 20));

  return this.client.setFilter(this.filter);
};

/**
 * Add data to remote filter.
 * @private
 * @param {Buffer} data
 * @returns {Promise}
 */

WalletDB.prototype.addFilter = function addFilter(data) {
  return this.client.addFilter(data);
};

/**
 * Reset remote filter.
 * @private
 * @returns {Promise}
 */

WalletDB.prototype.resetFilter = function resetFilter() {
  return this.client.resetFilter();
};

/**
 * Backup the wallet db.
 * @param {String} path
 * @returns {Promise}
 */

WalletDB.prototype.backup = function backup(path) {
  return this.db.backup(path);
};

/**
 * Wipe the txdb - NEVER USE.
 * @returns {Promise}
 */

WalletDB.prototype.wipe = async function wipe() {
  this.logger.warning('Wiping WalletDB TXDB...');
  this.logger.warning('I hope you know what you\'re doing.');

  const iter = this.db.iterator({
    gte: Buffer.from([0x00]),
    lte: Buffer.from([0xff])
  });

  const b = this.db.batch();

  let total = 0;

  await iter.each((key) => {
    switch (key[0]) {
      case 0x62: // b
      case 0x63: // c
      case 0x65: // e
      case 0x74: // t
      case 0x6f: // o
      case 0x68: // h
      case 0x52: // R
        b.del(key);
        total += 1;
        break;
    }
  });

  this.logger.warning('Wiped %d txdb records.', total);

  await b.write();
};

/**
 * Get current wallet wid depth.
 * @private
 * @returns {Promise}
 */

WalletDB.prototype.getDepth = async function getDepth() {
  // This may seem like a strange way to do
  // this, but updating a global state when
  // creating a new wallet is actually pretty
  // damn tricky. There would be major atomicity
  // issues if updating a global state inside
  // a "scoped" state. So, we avoid all the
  // nonsense of adding a global lock to
  // walletdb.create by simply seeking to the
  // highest wallet wid.
  const iter = this.db.iterator({
    gte: layout.w(0x00000000),
    lte: layout.w(0xffffffff),
    reverse: true,
    limit: 1
  });

  if (!await iter.next())
    return 1;

  const {key} = iter;

  await iter.end();

  const depth = layout.ww(key);

  return depth + 1;
};

/**
 * Test the bloom filter against a tx or address hash.
 * @private
 * @param {Hash} hash
 * @returns {Boolean}
 */

WalletDB.prototype.testFilter = function testFilter(data) {
  return this.filter.test(data, 'hex');
};

/**
 * Add hash to local and remote filters.
 * @private
 * @param {Hash} hash
 */

WalletDB.prototype.addHash = function addHash(hash) {
  this.filter.add(hash, 'hex');
  return this.addFilter(hash);
};

/**
 * Add outpoint to local filter.
 * @private
 * @param {Hash} hash
 * @param {Number} index
 */

WalletDB.prototype.addOutpoint = function addOutpoint(hash, index) {
  const outpoint = new Outpoint(hash, index);
  this.filter.add(outpoint.toRaw());
};

/**
 * Dump database (for debugging).
 * @returns {Promise} - Returns Object.
 */

WalletDB.prototype.dump = function dump() {
  return this.db.dump();
};

/**
 * Register an object with the walletdb.
 * @param {Object} object
 */

WalletDB.prototype.register = function register(wallet) {
  assert(!this.wallets.has(wallet.wid));
  this.wallets.set(wallet.wid, wallet);
};

/**
 * Unregister a object with the walletdb.
 * @param {Object} object
 * @returns {Boolean}
 */

WalletDB.prototype.unregister = function unregister(wallet) {
  assert(this.wallets.has(wallet.wid));
  this.wallets.delete(wallet.wid);
};

/**
 * Map wallet id to wid.
 * @param {String} id
 * @returns {Promise} - Returns {WalletID}.
 */

WalletDB.prototype.getWID = async function getWID(id) {
  if (!id)
    return null;

  if (typeof id === 'number')
    return id;

  const data = await this.db.get(layout.l(id));

  if (!data)
    return null;

  assert(data.length === 4);

  return data.readUInt32LE(0, true);
};

/**
 * Get a wallet from the database, setup watcher.
 * @param {WalletID} wid
 * @returns {Promise} - Returns {@link Wallet}.
 */

WalletDB.prototype.get = async function get(id) {
  const wid = await this.getWID(id);

  if (!wid)
    return null;

  const unlock = await this.readLock.lock(wid);

  try {
    return await this._get(wid);
  } finally {
    unlock();
  }
};

/**
 * Get a wallet from the database without a lock.
 * @private
 * @param {WalletID} wid
 * @returns {Promise} - Returns {@link Wallet}.
 */

WalletDB.prototype._get = async function _get(wid) {
  const cache = this.wallets.get(wid);

  if (cache)
    return cache;

  const data = await this.db.get(layout.w(wid));

  if (!data)
    return null;

  const wallet = Wallet.fromRaw(this, data);

  await wallet.open();

  this.register(wallet);

  return wallet;
};

/**
 * Save a wallet to the database.
 * @param {Wallet} wallet
 */

WalletDB.prototype.save = function save(b, wallet) {
  const wid = wallet.wid;
  const id = wallet.id;

  b.put(layout.w(wid), wallet.toRaw());
  b.put(layout.l(id), u32(wid));
};

/**
 * Rename a wallet.
 * @param {Wallet} wallet
 * @param {String} id
 * @returns {Promise}
 */

WalletDB.prototype.rename = async function rename(wallet, id) {
  const unlock = await this.writeLock.lock();
  try {
    return await this._rename(wallet, id);
  } finally {
    unlock();
  }
};

/**
 * Rename a wallet without a lock.
 * @private
 * @param {Wallet} wallet
 * @param {String} id
 * @returns {Promise}
 */

WalletDB.prototype._rename = async function _rename(wallet, id) {
  if (!common.isName(id))
    throw new Error('WDB: Bad wallet ID.');

  if (await this.has(id))
    throw new Error('WDB: ID not available.');

  const old = wallet.id;
  const b = this.db.batch();

  b.del(layout.l(old));

  wallet.id = id;

  this.save(b, wallet);

  await b.write();
};

/**
 * Rename an account.
 * @param {Account} account
 * @param {String} name
 */

WalletDB.prototype.renameAccount = function renameAccount(b, account, name) {
  // Remove old wid/name->account index.
  b.del(layout.i(account.wid, account.name));

  account.name = name;

  this.saveAccount(b, account);
};

/**
 * Get a wallet with token auth first.
 * @param {WalletID} wid
 * @param {Buffer} token
 * @returns {Promise} - Returns {@link Wallet}.
 */

WalletDB.prototype.auth = async function auth(wid, token) {
  const wallet = await this.get(wid);

  if (!wallet)
    return null;

  // Compare in constant time:
  if (!ccmp(token, wallet.token))
    throw new Error('WDB: Authentication error.');

  return wallet;
};

/**
 * Create a new wallet, save to database, setup watcher.
 * @param {Object} options - See {@link Wallet}.
 * @returns {Promise} - Returns {@link Wallet}.
 */

WalletDB.prototype.create = async function create(options) {
  const unlock = await this.writeLock.lock();

  if (!options)
    options = {};

  try {
    return await this._create(options);
  } finally {
    unlock();
  }
};

/**
 * Create a new wallet, save to database without a lock.
 * @private
 * @param {Object} options - See {@link Wallet}.
 * @returns {Promise} - Returns {@link Wallet}.
 */

WalletDB.prototype._create = async function _create(options) {
  if (await this.has(options.id))
    throw new Error('WDB: Wallet already exists.');

  const wallet = Wallet.fromOptions(this, options);

  wallet.wid = this.depth;

  await wallet.init(options);

  this.depth += 1;

  this.register(wallet);

  this.logger.info('Created wallet %s in WalletDB.', wallet.id);

  return wallet;
};

/**
 * Test for the existence of a wallet.
 * @param {WalletID} id
 * @returns {Promise}
 */

WalletDB.prototype.has = async function has(id) {
  const wid = await this.getWID(id);
  return wid != null;
};

/**
 * Attempt to create wallet, return wallet if already exists.
 * @param {Object} options - See {@link Wallet}.
 * @returns {Promise}
 */

WalletDB.prototype.ensure = async function ensure(options) {
  const wallet = await this.get(options.id);

  if (wallet)
    return wallet;

  return this.create(options);
};

/**
 * Get an account from the database by wid.
 * @private
 * @param {WalletID} wid
 * @param {Number} index - Account index.
 * @returns {Promise} - Returns {@link Wallet}.
 */

WalletDB.prototype.getAccount = async function getAccount(wid, index) {
  const data = await this.db.get(layout.a(wid, index));

  if (!data)
    return null;

  return Account.fromRaw(this, data);
};

/**
 * List account names and indexes from the db.
 * @param {WalletID} wid
 * @returns {Promise} - Returns Array.
 */

WalletDB.prototype.getAccounts = function getAccounts(wid) {
  return this.db.values({
    gte: layout.n(wid, 0x00000000),
    lte: layout.n(wid, 0xffffffff),
    parse: data => data.toString('ascii')
  });
};

/**
 * Lookup the corresponding account name's index.
 * @param {WalletID} wid
 * @param {String} name - Account name/index.
 * @returns {Promise} - Returns Number.
 */

WalletDB.prototype.getAccountIndex = async function getAccountIndex(wid, name) {
  const index = await this.db.get(layout.i(wid, name));

  if (!index)
    return -1;

  return index.readUInt32LE(0, true);
};

/**
 * Lookup the corresponding account index's name.
 * @param {WalletID} wid
 * @param {Number} index
 * @returns {Promise} - Returns Number.
 */

WalletDB.prototype.getAccountName = async function getAccountName(wid, index) {
  const name = await this.db.get(layout.n(wid, index));

  if (!name)
    return null;

  return name.toString('ascii');
};

/**
 * Save an account to the database.
 * @param {Account} account
 * @returns {Promise}
 */

WalletDB.prototype.saveAccount = function saveAccount(b, account) {
  const wid = account.wid;
  const index = account.accountIndex;
  const name = account.name;

  // Account data
  b.put(layout.a(wid, index), account.toRaw());

  // Name->Index lookups
  b.put(layout.i(wid, name), u32(index));

  // Index->Name lookups
  b.put(layout.n(wid, index), Buffer.from(name, 'ascii'));
};

/**
 * Test for the existence of an account.
 * @param {WalletID} wid
 * @param {String|Number} acct
 * @returns {Promise} - Returns Boolean.
 */

WalletDB.prototype.hasAccount = function hasAccount(wid, index) {
  return this.db.has(layout.a(wid, index));
};

/**
 * Save an address to the path map.
 * @param {Wallet} wallet
 * @param {WalletKey} ring
 * @returns {Promise}
 */

WalletDB.prototype.saveKey = function saveKey(b, wid, ring) {
  return this.savePath(b, wid, ring.toPath());
};

/**
 * Save a path to the path map.
 *
 * The path map exists in the form of:
 *   - `p[address-hash] -> wid map`
 *   - `P[wid][address-hash] -> path data`
 *   - `r[wid][account-index][address-hash] -> dummy`
 *
 * @param {Wallet} wallet
 * @param {Path} path
 * @returns {Promise}
 */

WalletDB.prototype.savePath = async function savePath(b, wid, path) {
  // Address Hash -> Wallet Map
  await this.addPathMap(b, path.hash, wid);

  // Wallet ID + Address Hash -> Path Data
  b.put(layout.P(wid, path.hash), path.toRaw());

  // Wallet ID + Account Index + Address Hash -> Dummy
  b.put(layout.r(wid, path.account, path.hash), null);
};

/**
 * Retrieve path by hash.
 * @param {WalletID} wid
 * @param {Hash} hash
 * @returns {Promise}
 */

WalletDB.prototype.getPath = async function getPath(wid, hash) {
  const path = await this.readPath(wid, hash);

  if (!path)
    return null;

  path.name = await this.getAccountName(wid, path.account);
  assert(path.name);

  return path;
};

/**
 * Retrieve path by hash.
 * @param {WalletID} wid
 * @param {Hash} hash
 * @returns {Promise}
 */

WalletDB.prototype.readPath = async function readPath(wid, hash) {
  const data = await this.db.get(layout.P(wid, hash));

  if (!data)
    return null;

  const path = Path.fromRaw(data);
  path.hash = hash;

  return path;
};

/**
 * Test whether a wallet contains a path.
 * @param {WalletID} wid
 * @param {Hash} hash
 * @returns {Promise}
 */

WalletDB.prototype.hasPath = function hasPath(wid, hash) {
  return this.db.has(layout.P(wid, hash));
};

/**
 * Get all address hashes.
 * @returns {Promise}
 */

WalletDB.prototype.getHashes = function getHashes() {
  return this.db.keys({
    gte: layout.p(encoding.NULL_HASH),
    lte: layout.p(encoding.HIGH_HASH),
    parse: layout.pp
  });
};

/**
 * Get all outpoints.
 * @returns {Promise}
 */

WalletDB.prototype.getOutpoints = function getOutpoints() {
  return this.db.keys({
    gte: layout.o(encoding.NULL_HASH, 0),
    lte: layout.o(encoding.HIGH_HASH, 0xffffffff),
    parse: (key) => {
      const [hash, index] = layout.oo(key);
      return new Outpoint(hash, index);
    }
  });
};

/**
 * Get all address hashes.
 * @param {WalletID} wid
 * @returns {Promise}
 */

WalletDB.prototype.getWalletHashes = function getWalletHashes(wid) {
  return this.db.keys({
    gte: layout.P(wid, encoding.NULL_HASH),
    lte: layout.P(wid, encoding.HIGH_HASH),
    parse: layout.Pp
  });
};

/**
 * Get all account address hashes.
 * @param {WalletID} wid
 * @param {Number} account
 * @returns {Promise}
 */

WalletDB.prototype.getAccountHashes = function getAccountHashes(wid, account) {
  return this.db.keys({
    gte: layout.r(wid, account, encoding.NULL_HASH),
    lte: layout.r(wid, account, encoding.HIGH_HASH),
    parse: layout.rr
  });
};

/**
 * Get all paths for a wallet.
 * @param {WalletID} wid
 * @returns {Promise}
 */

WalletDB.prototype.getWalletPaths = async function getWalletPaths(wid) {
  const items = await this.db.range({
    gte: layout.P(wid, encoding.NULL_HASH),
    lte: layout.P(wid, encoding.HIGH_HASH)
  });

  const paths = [];

  for (const item of items) {
    const hash = layout.Pp(item.key);
    const path = Path.fromRaw(item.value);

    path.hash = hash;
    path.name = await this.getAccountName(wid, path.account);
    assert(path.name);

    paths.push(path);
  }

  return paths;
};

/**
 * Get all wallet ids.
 * @returns {Promise}
 */

WalletDB.prototype.getWallets = function getWallets() {
  return this.db.keys({
    gte: layout.l('\x00'),
    lte: layout.l('\xff'),
    parse: layout.ll
  });
};

/**
 * Encrypt all imported keys for a wallet.
 * @param {WalletID} wid
 * @param {Buffer} key
 * @returns {Promise}
 */

WalletDB.prototype.encryptKeys = async function encryptKeys(b, wid, key) {
  const iter = this.db.iterator({
    gte: layout.P(wid, encoding.NULL_HASH),
    lte: layout.P(wid, encoding.HIGH_HASH),
    values: true
  });

  await iter.each((k, value) => {
    const hash = layout.Pp(k);
    const path = Path.fromRaw(value);

    if (!path.data)
      return;

    assert(!path.encrypted);

    const bhash = Buffer.from(hash, 'hex');
    const iv = bhash.slice(0, 16);

    path.data = aes.encipher(path.data, key, iv);
    path.encrypted = true;

    b.put(k, path.toRaw());
  });
};

/**
 * Decrypt all imported keys for a wallet.
 * @param {WalletID} wid
 * @param {Buffer} key
 * @returns {Promise}
 */

WalletDB.prototype.decryptKeys = async function decryptKeys(b, wid, key) {
  const iter = this.db.iterator({
    gte: layout.P(wid, encoding.NULL_HASH),
    lte: layout.P(wid, encoding.HIGH_HASH),
    values: true
  });

  await iter.each((k, value) => {
    const hash = layout.Pp(k);
    const path = Path.fromRaw(value);

    if (!path.data)
      return;

    assert(path.encrypted);

    const bhash = Buffer.from(hash, 'hex');
    const iv = bhash.slice(0, 16);

    path.data = aes.decipher(path.data, key, iv);
    path.encrypted = false;

    b.put(k, path.toRaw());
  });
};

/**
 * Resend all pending transactions.
 * @returns {Promise}
 */

WalletDB.prototype.resend = async function resend() {
  const wids = await this.db.keys({
    gte: layout.w(0x00000000),
    lte: layout.w(0xffffffff),
    parse: k => layout.ww(k)
  });

  this.logger.info('Resending from %d wallets.', wids.length);

  for (const wid of wids)
    await this.resendPending(wid);
};

/**
 * Resend all pending transactions for a specific wallet.
 * @private
 * @param {WalletID} wid
 * @returns {Promise}
 */

WalletDB.prototype.resendPending = async function resendPending(wid) {
  const layout = layouts.txdb;
  const prefix = layout.prefix(wid);
  const b = this.db.bucket(prefix);

  const hashes = await b.keys({
    gte: layout.p(encoding.NULL_HASH),
    lte: layout.p(encoding.HIGH_HASH),
    parse: k => layout.pp(k)
  });

  if (hashes.length === 0)
    return;

  this.logger.info(
    'Rebroadcasting %d transactions for %d.',
    hashes.length,
    wid);

  const txs = [];

  for (const hash of hashes) {
    const data = await b.get(layout.t(hash));

    if (!data)
      continue;

    const wtx = TXRecord.fromRaw(data);

    if (wtx.tx.isCoinbase())
      continue;

    txs.push(wtx.tx);
  }

  for (const tx of common.sortDeps(txs))
    await this.send(tx);
};

/**
 * Get all wallet ids by output addresses and outpoints.
 * @param {Hash[]} hashes
 * @returns {Promise}
 */

WalletDB.prototype.getWalletsByTX = async function getWalletsByTX(tx) {
  const result = new Set();

  if (!tx.isCoinbase()) {
    for (const {prevout} of tx.inputs) {
      const {hash, index} = prevout;

      if (!this.testFilter(prevout.toRaw()))
        continue;

      const map = await this.getOutpointMap(hash, index);

      if (!map)
        continue;

      for (const wid of map.wids)
        result.add(wid);
    }
  }

  const hashes = tx.getOutputHashes('hex');

  for (const hash of hashes) {
    if (!this.testFilter(hash))
      continue;

    const map = await this.getPathMap(hash);

    if (!map)
      continue;

    for (const wid of map.wids)
      result.add(wid);
  }

  if (result.size === 0)
    return null;

  return result;
};

/**
 * Get the best block hash.
 * @returns {Promise}
 */

WalletDB.prototype.getState = async function getState() {
  const data = await this.db.get(layout.R);

  if (!data)
    return null;

  return ChainState.fromRaw(data);
};

/**
 * Sync the current chain state to tip.
 * @param {BlockMeta} tip
 * @returns {Promise}
 */

WalletDB.prototype.setTip = async function setTip(tip) {
  const b = this.db.batch();
  const state = this.state.clone();

  if (tip.height < state.height) {
    // Hashes ahead of our new tip
    // that we need to delete.
    while (state.height !== tip.height) {
      b.del(layout.h(state.height));
      state.height -= 1;
    }
  } else if (tip.height > state.height) {
    assert(tip.height === state.height + 1, 'Bad chain sync.');
    state.height += 1;
  }

  if (tip.height < state.startHeight) {
    state.startHeight = tip.height;
    state.startHash = tip.hash;
    state.marked = false;
  }

  // Save tip and state.
  b.put(layout.h(tip.height), tip.toHash());
  b.put(layout.R, state.toRaw());

  await b.write();

  this.state = state;
  this.height = state.height;
};

/**
 * Mark current state.
 * @param {BlockMeta} block
 * @returns {Promise}
 */

WalletDB.prototype.markState = async function markState(block) {
  const state = this.state.clone();
  state.startHeight = block.height;
  state.startHash = block.hash;
  state.marked = true;

  const b = this.db.batch();
  b.put(layout.R, state.toRaw());
  await b.write();

  this.state = state;
  this.height = state.height;
};

/**
 * Get a wallet map.
 * @param {Buffer} key
 * @returns {Promise}
 */

WalletDB.prototype.getMap = async function getMap(key) {
  const data = await this.db.get(key);

  if (!data)
    return null;

  return MapRecord.fromRaw(data);
};

/**
 * Add wid to a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.addMap = async function addMap(b, key, wid) {
  const data = await this.db.get(key);

  if (!data) {
    const map = new MapRecord();
    map.add(wid);
    b.put(key, map.toRaw());
    return;
  }

  assert(data.length >= 4);

  const len = data.readUInt32LE(0, true);
  const bw = bio.write(data.length + 4);

  bw.writeU32(len + 1);
  bw.copy(data, 4, data.length);
  bw.writeU32(wid);

  b.put(key, bw.render());
};

/**
 * Remove wid from a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.removeMap = async function removeMap(b, key, wid) {
  const map = await this.getMap(key);

  if (!map)
    return;

  if (!map.remove(wid))
    return;

  if (map.size === 0) {
    b.del(key);
    return;
  }

  b.put(key, map.toRaw());
};

/**
 * Get a wallet map.
 * @param {Buffer} key
 * @returns {Promise}
 */

WalletDB.prototype.getPathMap = function getPathMap(hash) {
  return this.getMap(layout.p(hash));
};

/**
 * Add wid to a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.addPathMap = async function addPathMap(b, hash, wid) {
  await this.addHash(hash);
  return this.addMap(b, layout.p(hash), wid);
};

/**
 * Remove wid from a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.removePathMap = function removePathMap(b, hash, wid) {
  return this.removeMap(b, layout.p(hash), wid);
};

/**
 * Get a wallet map.
 * @param {Buffer} key
 * @returns {Promise}
 */

WalletDB.prototype.getBlockMap = async function getBlockMap(height) {
  return this.getMap(layout.b(height));
};

/**
 * Add wid to a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.addBlockMap = function addBlockMap(b, height, wid) {
  return this.addMap(b, layout.b(height), wid);
};

/**
 * Remove wid from a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.removeBlockMap = function removeBlockMap(b, height, wid) {
  return this.removeMap(b, layout.b(height), wid);
};

/**
 * Get a wallet map.
 * @param {Buffer} key
 * @returns {Promise}
 */

WalletDB.prototype.getTXMap = function getTXMap(hash) {
  return this.getMap(layout.T(hash));
};

/**
 * Add wid to a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.addTXMap = function addTXMap(b, hash, wid) {
  return this.addMap(b, layout.T(hash), wid);
};

/**
 * Remove wid from a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.removeTXMap = function removeTXMap(b, hash, wid) {
  return this.removeMap(b, layout.T(hash), wid);
};

/**
 * Get a wallet map.
 * @param {Buffer} key
 * @returns {Promise}
 */

WalletDB.prototype.getOutpointMap = function getOutpointMap(hash, index) {
  return this.getMap(layout.o(hash, index));
};

/**
 * Add wid to a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.addOutpointMap = async function addOutpointMap(b, hash, index, wid) {
  await this.addOutpoint(hash, index);
  return this.addMap(b, layout.o(hash, index), wid);
};

/**
 * Remove wid from a wallet map.
 * @param {Wallet} wallet
 * @param {Buffer} key
 * @param {Number} wid
 */

WalletDB.prototype.removeOutpointMap = function removeOutpointMap(b, hash, index, wid) {
  return this.removeMap(b, layout.o(hash, index), wid);
};

/**
 * Get a wallet block meta.
 * @param {Hash} hash
 * @returns {Promise}
 */

WalletDB.prototype.getBlock = async function getBlock(height) {
  const data = await this.db.get(layout.h(height));

  if (!data)
    return null;

  const block = new BlockMeta();
  block.hash = data.toString('hex');
  block.height = height;

  return block;
};

/**
 * Get wallet tip.
 * @param {Hash} hash
 * @returns {Promise}
 */

WalletDB.prototype.getTip = async function getTip() {
  const tip = await this.getBlock(this.state.height);

  if (!tip)
    throw new Error('WDB: Tip not found!');

  return tip;
};

/**
 * Sync with chain height.
 * @param {Number} height
 * @returns {Promise}
 */

WalletDB.prototype.rollback = async function rollback(height) {
  if (height > this.state.height)
    throw new Error('WDB: Cannot rollback to the future.');

  if (height === this.state.height) {
    this.logger.info('Rolled back to same height (%d).', height);
    return;
  }

  this.logger.info(
    'Rolling back %d WalletDB blocks to height %d.',
    this.state.height - height, height);

  const tip = await this.getBlock(height);
  assert(tip);

  await this.revert(tip.height);
  await this.setTip(tip);
};

/**
 * Revert TXDB to an older state.
 * @param {Number} target
 * @returns {Promise}
 */

WalletDB.prototype.revert = async function revert(target) {
  const iter = this.db.iterator({
    gte: layout.b(target + 1),
    lte: layout.b(0xffffffff),
    reverse: true,
    values: true
  });

  let total = 0;

  await iter.each(async (key, value) => {
    const height = layout.bb(key);
    const block = MapRecord.fromRaw(value);

    for (const wid of block.wids) {
      const wallet = await this.get(wid);
      assert(wallet);
      total += await wallet.revert(height);
    }
  });

  this.logger.info('Rolled back %d WalletDB transactions.', total);
};

/**
 * Add a block's transactions and write the new best hash.
 * @param {ChainEntry} entry
 * @returns {Promise}
 */

WalletDB.prototype.addBlock = async function addBlock(entry, txs) {
  const unlock = await this.txLock.lock();
  try {
    return await this._addBlock(entry, txs);
  } finally {
    unlock();
  }
};

/**
 * Add a block's transactions without a lock.
 * @private
 * @param {ChainEntry} entry
 * @param {TX[]} txs
 * @returns {Promise}
 */

WalletDB.prototype._addBlock = async function _addBlock(entry, txs) {
  const tip = BlockMeta.fromEntry(entry);

  if (tip.height < this.state.height) {
    this.logger.warning(
      'WalletDB is connecting low blocks (%d).',
      tip.height);
    return 0;
  }

  this.logger.debug('Adding block: %d.', entry.height);

  if (tip.height === this.state.height) {
    // We let blocks of the same height
    // through specifically for rescans:
    // we always want to rescan the last
    // block since the state may have
    // updated before the block was fully
    // processed (in the case of a crash).
    this.logger.warning('Already saw WalletDB block (%d).', tip.height);
  } else if (tip.height !== this.state.height + 1) {
    await this.scan(this.state.height);
    return 0;
  }

  // Sync the state to the new tip.
  await this.setTip(tip);

  if (this.options.checkpoints && !this.state.marked) {
    if (tip.height <= this.network.lastCheckpoint)
      return 0;
  }

  let total = 0;

  for (const tx of txs) {
    if (await this._addTX(tx, tip))
      total += 1;
  }

  if (total > 0) {
    this.logger.info('Connected WalletDB block %s (tx=%d).',
      encoding.revHex(tip.hash), total);
  }

  return total;
};

/**
 * Unconfirm a block's transactions
 * and write the new best hash (SPV version).
 * @param {ChainEntry} entry
 * @returns {Promise}
 */

WalletDB.prototype.removeBlock = async function removeBlock(entry) {
  const unlock = await this.txLock.lock();
  try {
    return await this._removeBlock(entry);
  } finally {
    unlock();
  }
};

/**
 * Unconfirm a block's transactions.
 * @private
 * @param {ChainEntry} entry
 * @returns {Promise}
 */

WalletDB.prototype._removeBlock = async function _removeBlock(entry) {
  const tip = BlockMeta.fromEntry(entry);

  if (tip.height === 0)
    throw new Error('WDB: Bad disconnection (genesis block).');

  if (tip.height > this.state.height) {
    this.logger.warning(
      'WalletDB is disconnecting high blocks (%d).',
      tip.height);
    return 0;
  }

  if (tip.height !== this.state.height)
    throw new Error('WDB: Bad disconnection (height mismatch).');

  const prev = await this.getBlock(tip.height - 1);
  assert(prev);

  // Get the map of block->wids.
  const map = await this.getBlockMap(tip.height);

  if (!map) {
    await this.setTip(prev);
    return 0;
  }

  let total = 0;

  for (const wid of map.wids) {
    const wallet = await this.get(wid);
    assert(wallet);
    total += await wallet.revert(tip.height);
  }

  // Sync the state to the previous tip.
  await this.setTip(prev);

  this.logger.warning('Disconnected wallet block %s (tx=%d).',
    encoding.revHex(tip.hash), total);

  return total;
};

/**
 * Rescan a block.
 * @private
 * @param {ChainEntry} entry
 * @param {TX[]} txs
 * @returns {Promise}
 */

WalletDB.prototype.rescanBlock = async function rescanBlock(entry, txs) {
  if (!this.rescanning) {
    this.logger.warning('Unsolicited rescan block: %d.', entry.height);
    return;
  }

  if (entry.height > this.state.height + 1) {
    this.logger.warning('Rescan block too high: %d.', entry.height);
    return;
  }

  try {
    await this._addBlock(entry, txs);
  } catch (e) {
    this.emit('error', e);
    throw e;
  }
};

/**
 * Add a transaction to the database, map addresses
 * to wallet IDs, potentially store orphans, resolve
 * orphans, or confirm a transaction.
 * @param {TX} tx
 * @returns {Promise}
 */

WalletDB.prototype.addTX = async function addTX(tx) {
  const unlock = await this.txLock.lock();
  try {
    return await this._addTX(tx);
  } finally {
    unlock();
  }
};

/**
 * Add a transaction to the database without a lock.
 * @private
 * @param {TX} tx
 * @param {BlockMeta} block
 * @returns {Promise}
 */

WalletDB.prototype._addTX = async function _addTX(tx, block) {
  const wids = await this.getWalletsByTX(tx);

  assert(!tx.mutable, 'WDB: Cannot add mutable TX.');

  if (!wids)
    return null;

  if (block && !this.state.marked)
    await this.markState(block);

  this.logger.info(
    'Incoming transaction for %d wallets in WalletDB (%s).',
    wids.size, tx.txid());

  let result = false;

  // Insert the transaction
  // into every matching wallet.
  for (const wid of wids) {
    const wallet = await this.get(wid);

    assert(wallet);

    if (await wallet.add(tx, block)) {
      this.logger.info(
        'Added transaction to wallet in WalletDB: %s (%d).',
        wallet.id, wid);
      result = true;
    }
  }

  if (!result)
    return null;

  return wids;
};

/**
 * Handle a chain reset.
 * @param {ChainEntry} entry
 * @returns {Promise}
 */

WalletDB.prototype.resetChain = async function resetChain(entry) {
  const unlock = await this.txLock.lock();
  try {
    return await this._resetChain(entry);
  } finally {
    unlock();
  }
};

/**
 * Handle a chain reset without a lock.
 * @private
 * @param {ChainEntry} entry
 * @returns {Promise}
 */

WalletDB.prototype._resetChain = async function _resetChain(entry) {
  if (entry.height > this.state.height)
    throw new Error('WDB: Bad reset height.');

  await this.rollback(entry.height);
};

/**
 * WalletOptions
 * @alias module:wallet.WalletOptions
 * @constructor
 * @param {Object} options
 */

function WalletOptions(options) {
  if (!(this instanceof WalletOptions))
    return new WalletOptions(options);

  this.network = Network.primary;
  this.logger = Logger.global;
  this.workers = null;
  this.client = null;
  this.feeRate = 0;

  this.prefix = null;
  this.location = null;
  this.db = 'memory';
  this.maxFiles = 64;
  this.cacheSize = 16 << 20;
  this.compression = true;
  this.bufferKeys = layout.binary;

  this.spv = false;
  this.witness = false;
  this.checkpoints = false;
  this.startHeight = 0;
  this.wipeNoReally = false;

  if (options)
    this.fromOptions(options);
}

/**
 * Inject properties from object.
 * @private
 * @param {Object} options
 * @returns {WalletOptions}
 */

WalletOptions.prototype.fromOptions = function fromOptions(options) {
  if (options.network != null)
    this.network = Network.get(options.network);

  if (options.logger != null) {
    assert(typeof options.logger === 'object');
    this.logger = options.logger;
  }

  if (options.workers != null) {
    assert(typeof options.workers === 'object');
    this.workers = options.workers;
  }

  if (options.client != null) {
    assert(typeof options.client === 'object');
    this.client = options.client;
  }

  if (options.feeRate != null) {
    assert((options.feeRate >>> 0) === options.feeRate);
    this.feeRate = options.feeRate;
  }

  if (options.prefix != null) {
    assert(typeof options.prefix === 'string');
    this.prefix = options.prefix;
    this.location = path.join(this.prefix, 'walletdb');
  }

  if (options.location != null) {
    assert(typeof options.location === 'string');
    this.location = options.location;
  }

  if (options.db != null) {
    assert(typeof options.db === 'string');
    this.db = options.db;
  }

  if (options.maxFiles != null) {
    assert((options.maxFiles >>> 0) === options.maxFiles);
    this.maxFiles = options.maxFiles;
  }

  if (options.cacheSize != null) {
    assert(Number.isSafeInteger(options.cacheSize) && options.cacheSize >= 0);
    this.cacheSize = options.cacheSize;
  }

  if (options.compression != null) {
    assert(typeof options.compression === 'boolean');
    this.compression = options.compression;
  }

  if (options.spv != null) {
    assert(typeof options.spv === 'boolean');
    this.spv = options.spv;
  }

  if (options.witness != null) {
    assert(typeof options.witness === 'boolean');
    this.witness = options.witness;
  }

  if (options.checkpoints != null) {
    assert(typeof options.checkpoints === 'boolean');
    this.checkpoints = options.checkpoints;
  }

  if (options.startHeight != null) {
    assert(typeof options.startHeight === 'number');
    assert(options.startHeight >= 0);
    this.startHeight = options.startHeight;
  }

  if (options.wipeNoReally != null) {
    assert(typeof options.wipeNoReally === 'boolean');
    this.wipeNoReally = options.wipeNoReally;
  }

  return this;
};

/**
 * Instantiate chain options from object.
 * @param {Object} options
 * @returns {WalletOptions}
 */

WalletOptions.fromOptions = function fromOptions(options) {
  return new WalletOptions().fromOptions(options);
};

/*
 * Expose
 */

module.exports = WalletDB;