Fix clearing a spend.

darkwallet · Dec 17, 2014 · 1a32a94 · 1a32a94
1 parent 6f37472
commit 1a32a94
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/src/js/model/output.js b/src/js/model/output.js
@@ -31,6 +31,18 @@ Output.prototype.markSpend = function(outId, height) {
     this.spendheight = height || 0;
 }
 
+/**
+ * Clear this output's spend
+ */
+Output.prototype.clearSpend = function() {
+    if (this.spendheight) {
+        throw Error("Output is already confirmed!");
+    }
+    this.spend = false;
+    this.spendpending = false;
+};
+
+
 /**
  * Receive id for the output ("hash:index")
  */

diff --git a/src/js/model/tx.js b/src/js/model/tx.js
@@ -272,9 +272,7 @@ Transaction.prototype.undo = function(tx, row) {
                 var walletAddress = wallet.getWalletAddress(output.address);
                 walletAddress.balance += output.value;
             }
-            delete output.spend;
-            delete output.spendheight;
-            delete output.spendpending;
+            output.clearSpend();
         }
     });
     tx.outs.forEach(function(anOut, i) {