From c8f59134776565c8ba95df61271a2d912b58a82e Mon Sep 17 00:00:00 2001
From: Christopher Jeffrey <chjjeffrey@gmail.com>
Date: Wed, 2 Mar 2016 03:55:57 -0800
Subject: [PATCH] retroactively save wallets.

---
 lib/bcoin/walletdb.js | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/lib/bcoin/walletdb.js b/lib/bcoin/walletdb.js
index 84393f90..0252b18b 100644
--- a/lib/bcoin/walletdb.js
+++ b/lib/bcoin/walletdb.js
@@ -134,9 +134,6 @@ WalletDB.prototype.saveJSON = function saveJSON(id, json, callback) {
 
   callback = utils.ensure(callback);
 
-  if (json instanceof bcoin.wallet)
-    json = json.toJSON();
-
   function cb(err, json) {
     var batch;
 
@@ -289,6 +286,15 @@ WalletDB.prototype.save = function save(options, callback) {
 
   callback = utils.ensure(callback);
 
+  if (options instanceof bcoin.wallet) {
+    if (!options.provider) {
+      options.on('add address', self._onAddress(options, options.id));
+      options.provider = self;
+    }
+    if (options instanceof bcoin.wallet)
+      options = options.toJSON();
+  }
+
   return this.saveJSON(options.id, options, callback);
 };