[lbry] claimtrie: minor refactoring of claimtrie CLI

This commit is contained in:
Roy Lee 2021-07-10 16:24:02 -07:00
parent 27c81de4e5
commit 3c85e6e56a
4 changed files with 13 additions and 19 deletions

View file

@ -6,22 +6,15 @@ import (
"strconv"
"github.com/btcsuite/btcd/claimtrie/block/blockrepo"
"github.com/btcsuite/btcd/claimtrie/config"
"github.com/btcsuite/btcd/claimtrie/merkletrie"
"github.com/btcsuite/btcd/claimtrie/merkletrie/merkletrierepo"
"github.com/btcsuite/btcd/claimtrie/param"
"github.com/btcsuite/btcd/claimtrie/temporal/temporalrepo"
"github.com/btcsuite/btcd/wire"
"github.com/spf13/cobra"
)
var localConfig *config.DBConfig
func init() {
param.SetNetwork(wire.MainNet, "mainnet")
localConfig = config.GenerateConfig(param.ClaimtrieDataFolder)
rootCmd.AddCommand(blockCmd)
blockCmd.AddCommand(blockLastCmd)
@ -106,9 +99,9 @@ var blockListCmd = &cobra.Command{
}
var blockNameCmd = &cobra.Command{
Use: "vertex <height> [<name>]",
Use: "vertex <height> <name>",
Short: "List the claim and child hashes at vertex name of block at height",
Args: cobra.RangeArgs(1, 2),
Args: cobra.RangeArgs(2, 2),
RunE: func(cmd *cobra.Command, args []string) error {
repo, err := blockrepo.NewPebble(localConfig.BlockRepoPebble.Path)

View file

@ -5,18 +5,13 @@ import (
"math"
"strconv"
"github.com/btcsuite/btcd/claimtrie/config"
"github.com/btcsuite/btcd/claimtrie/node"
"github.com/btcsuite/btcd/claimtrie/node/noderepo"
"github.com/btcsuite/btcd/claimtrie/param"
"github.com/btcsuite/btcd/wire"
"github.com/spf13/cobra"
)
func init() {
param.SetNetwork(wire.MainNet, "mainnet")
localConfig = config.GenerateConfig(param.ClaimtrieDataFolder)
rootCmd.AddCommand(nodeCmd)
nodeCmd.AddCommand(nodeDumpCmd)

View file

@ -1,9 +1,20 @@
package cmd
import (
"github.com/btcsuite/btcd/claimtrie/config"
"github.com/btcsuite/btcd/claimtrie/param"
"github.com/btcsuite/btcd/wire"
"github.com/spf13/cobra"
)
var localConfig *config.DBConfig
func init() {
param.SetNetwork(wire.MainNet, "mainnet")
localConfig = config.GenerateConfig(param.ClaimtrieDataFolder)
}
var rootCmd = &cobra.Command{
Use: "claimtrie",
Short: "ClaimTrie Command Line Interface",

View file

@ -5,17 +5,12 @@ import (
"log"
"strconv"
"github.com/btcsuite/btcd/claimtrie/config"
"github.com/btcsuite/btcd/claimtrie/param"
"github.com/btcsuite/btcd/claimtrie/temporal/temporalrepo"
"github.com/btcsuite/btcd/wire"
"github.com/spf13/cobra"
)
func init() {
param.SetNetwork(wire.MainNet, "mainnet")
localConfig = config.GenerateConfig(param.ClaimtrieDataFolder)
rootCmd.AddCommand(temporalCmd)
}