From fa26b1f0221d3a844d39a2c998efe318603c3f7e Mon Sep 17 00:00:00 2001
From: Oliver Gugger <gugger@gmail.com>
Date: Sat, 24 Oct 2020 16:26:47 +0200
Subject: [PATCH] cmd/dropwtxmgr: use DropTransactionHistory from wallet
 package

Now that the function to remove the wallet's transaction history is
exported as a utility function, we can use that in the dropwtxmgr
command.
---
 cmd/dropwtxmgr/main.go | 136 +----------------------------------------
 1 file changed, 2 insertions(+), 134 deletions(-)

diff --git a/cmd/dropwtxmgr/main.go b/cmd/dropwtxmgr/main.go
index 9447e6e..e1e3a1c 100644
--- a/cmd/dropwtxmgr/main.go
+++ b/cmd/dropwtxmgr/main.go
@@ -10,12 +10,10 @@ import (
 	"os"
 	"path/filepath"
 
-	"github.com/btcsuite/btcd/chaincfg/chainhash"
 	"github.com/btcsuite/btcutil"
-	"github.com/btcsuite/btcwallet/waddrmgr"
+	"github.com/btcsuite/btcwallet/wallet"
 	"github.com/btcsuite/btcwallet/walletdb"
 	_ "github.com/btcsuite/btcwallet/walletdb/bdb"
-	"github.com/btcsuite/btcwallet/wtxmgr"
 	"github.com/jessevdk/go-flags"
 )
 
@@ -40,15 +38,6 @@ func init() {
 	}
 }
 
-var (
-	// Namespace keys.
-	waddrmgrNamespace = []byte("waddrmgr")
-	wtxmgrNamespace   = []byte("wtxmgr")
-
-	// Bucket names.
-	bucketTxLabels = []byte("l")
-)
-
 func yes(s string) bool {
 	switch s {
 	case "y", "Y", "yes", "Yes":
@@ -113,62 +102,7 @@ func mainInt() int {
 
 	fmt.Println("Dropping btcwallet transaction history")
 
-	err = walletdb.Update(db, func(tx walletdb.ReadWriteTx) error {
-		// If we want to keep our tx labels, we read them out so we
-		// can re-add them after we have deleted our wtxmgr.
-		var labels map[chainhash.Hash]string
-		if !opts.DropLabels {
-			labels, err = fetchAllLabels(tx)
-			if err != nil {
-				return err
-			}
-		}
-
-		err := tx.DeleteTopLevelBucket(wtxmgrNamespace)
-		if err != nil && err != walletdb.ErrBucketNotFound {
-			return err
-		}
-		ns, err := tx.CreateTopLevelBucket(wtxmgrNamespace)
-		if err != nil {
-			return err
-		}
-		err = wtxmgr.Create(ns)
-		if err != nil {
-			return err
-		}
-
-		// If we want to re-add our labels, we do so now.
-		if !opts.DropLabels {
-			if err := putTxLabels(ns, labels); err != nil {
-				return err
-			}
-		}
-
-		ns = tx.ReadWriteBucket(waddrmgrNamespace)
-		birthdayBlock, err := waddrmgr.FetchBirthdayBlock(ns)
-		if err != nil {
-			fmt.Println("Wallet does not have a birthday block " +
-				"set, falling back to rescan from genesis")
-
-			startBlock, err := waddrmgr.FetchStartBlock(ns)
-			if err != nil {
-				return err
-			}
-			return waddrmgr.PutSyncedTo(ns, startBlock)
-		}
-
-		// We'll need to remove our birthday block first because it
-		// serves as a barrier when updating our state to detect reorgs
-		// due to the wallet not storing all block hashes of the chain.
-		if err := waddrmgr.DeleteBirthdayBlock(ns); err != nil {
-			return err
-		}
-
-		if err := waddrmgr.PutSyncedTo(ns, &birthdayBlock); err != nil {
-			return err
-		}
-		return waddrmgr.PutBirthdayBlock(ns, birthdayBlock)
-	})
+	err = wallet.DropTransactionHistory(db, !opts.DropLabels)
 	if err != nil {
 		fmt.Println("Failed to drop and re-create namespace:", err)
 		return 1
@@ -176,69 +110,3 @@ func mainInt() int {
 
 	return 0
 }
-
-// fetchAllLabels returns a map of hex-encoded txid to label.
-func fetchAllLabels(tx walletdb.ReadWriteTx) (map[chainhash.Hash]string,
-	error) {
-
-	// Get our top level bucket, if it does not exist we just exit.
-	txBucket := tx.ReadBucket(wtxmgrNamespace)
-	if txBucket == nil {
-		return nil, nil
-	}
-
-	// If we do not have a labels bucket, there are no labels so we exit.
-	labelsBucket := txBucket.NestedReadBucket(bucketTxLabels)
-	if labelsBucket == nil {
-		return nil, nil
-	}
-
-	labels := make(map[chainhash.Hash]string)
-	if err := labelsBucket.ForEach(func(k, v []byte) error {
-		txid, err := chainhash.NewHash(k)
-		if err != nil {
-			return err
-		}
-
-		label, err := wtxmgr.DeserializeLabel(v)
-		if err != nil {
-			return err
-		}
-
-		// Add an entry to our map of labels.
-		labels[*txid] = label
-
-		return nil
-	}); err != nil {
-		return nil, err
-	}
-
-	return labels, nil
-}
-
-// putTxLabels re-adds a nested labels bucket and entries to the bucket provided
-// if there are any labels present.
-func putTxLabels(ns walletdb.ReadWriteBucket,
-	labels map[chainhash.Hash]string) error {
-
-	// If there are no labels, exit early.
-	if len(labels) == 0 {
-		return nil
-	}
-
-	// First, we create a labels bucket which we will add all labels to.
-	labelBucket, err := ns.CreateBucketIfNotExists(bucketTxLabels)
-	if err != nil {
-		return err
-	}
-
-	// Next, we re-add every label to the bucket.
-	for txid, label := range labels {
-		err := wtxmgr.PutTxLabel(labelBucket, txid, label)
-		if err != nil {
-			return err
-		}
-	}
-
-	return nil
-}