Add utility to drop wtxmgr namespace.
This is intended to be used to force a full rescan.
This commit is contained in:
parent
847c7d8df9
commit
f60b8207d2
1 changed files with 123 additions and 0 deletions
123
cmd/dropwtxmgr/main.go
Normal file
123
cmd/dropwtxmgr/main.go
Normal file
|
@ -0,0 +1,123 @@
|
|||
// Copyright (c) 2015 Conformal Systems LLC <info@conformal.com>
|
||||
//
|
||||
// Permission to use, copy, modify, and distribute this software for any
|
||||
// purpose with or without fee is hereby granted, provided that the above
|
||||
// copyright notice and this permission notice appear in all copies.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||||
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||||
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||||
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||||
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||||
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||||
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||||
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/btcsuite/btcutil"
|
||||
"github.com/btcsuite/btcwallet/walletdb"
|
||||
_ "github.com/btcsuite/btcwallet/walletdb/bdb"
|
||||
"github.com/btcsuite/go-flags"
|
||||
)
|
||||
|
||||
const defaultNet = "mainnet"
|
||||
|
||||
var datadir = btcutil.AppDataDir("btcwallet", false)
|
||||
|
||||
// Flags.
|
||||
var opts = struct {
|
||||
Force bool `short:"f" description:"Force removal without prompt"`
|
||||
DbPath string `long:"db" description:"Path to wallet database"`
|
||||
}{
|
||||
Force: false,
|
||||
DbPath: filepath.Join(datadir, defaultNet, "wallet.db"),
|
||||
}
|
||||
|
||||
func init() {
|
||||
_, err := flags.Parse(&opts)
|
||||
if err != nil {
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
// Namespace keys.
|
||||
var (
|
||||
wtxmgrNamespace = []byte("wtxmgr")
|
||||
)
|
||||
|
||||
func yes(s string) bool {
|
||||
switch s {
|
||||
case "y", "Y", "yes", "Yes":
|
||||
return true
|
||||
default:
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func no(s string) bool {
|
||||
switch s {
|
||||
case "n", "N", "no", "No":
|
||||
return true
|
||||
default:
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
os.Exit(mainInt())
|
||||
}
|
||||
|
||||
func mainInt() int {
|
||||
fmt.Println("Database path:", opts.DbPath)
|
||||
_, err := os.Stat(opts.DbPath)
|
||||
if os.IsNotExist(err) {
|
||||
fmt.Println("Database file does not exist")
|
||||
return 1
|
||||
}
|
||||
|
||||
for !opts.Force {
|
||||
fmt.Print("Drop all btcwallet transaction history? [y/N] ")
|
||||
|
||||
scanner := bufio.NewScanner(bufio.NewReader(os.Stdin))
|
||||
if !scanner.Scan() {
|
||||
// Exit on EOF.
|
||||
return 0
|
||||
}
|
||||
err := scanner.Err()
|
||||
if err != nil {
|
||||
fmt.Println()
|
||||
fmt.Println(err)
|
||||
return 1
|
||||
}
|
||||
resp := scanner.Text()
|
||||
if yes(resp) {
|
||||
break
|
||||
}
|
||||
if no(resp) || resp == "" {
|
||||
return 0
|
||||
}
|
||||
|
||||
fmt.Println("Enter yes or no.")
|
||||
}
|
||||
|
||||
db, err := walletdb.Open("bdb", opts.DbPath)
|
||||
if err != nil {
|
||||
fmt.Println("Failed to open database:", err)
|
||||
return 1
|
||||
}
|
||||
defer db.Close()
|
||||
fmt.Println("Dropping wtxmgr namespace")
|
||||
err = db.DeleteNamespace(wtxmgrNamespace)
|
||||
if err != nil && err != walletdb.ErrBucketNotFound {
|
||||
fmt.Println("Failed to drop namespace:", err)
|
||||
return 1
|
||||
}
|
||||
|
||||
return 0
|
||||
}
|
Loading…
Reference in a new issue