mirror of
				https://github.com/0glabs/0g-chain.git
				synced 2025-11-04 00:27:41 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			171 lines
		
	
	
		
			4.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			171 lines
		
	
	
		
			4.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package main
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"os"
 | 
						|
	"path"
 | 
						|
 | 
						|
	"github.com/spf13/cobra"
 | 
						|
	"github.com/spf13/viper"
 | 
						|
 | 
						|
	"github.com/tendermint/tendermint/libs/cli"
 | 
						|
 | 
						|
	"github.com/cosmos/cosmos-sdk/client"
 | 
						|
	"github.com/cosmos/cosmos-sdk/client/flags"
 | 
						|
	"github.com/cosmos/cosmos-sdk/client/lcd"
 | 
						|
	"github.com/cosmos/cosmos-sdk/client/rpc"
 | 
						|
	"github.com/cosmos/cosmos-sdk/codec"
 | 
						|
	sdk "github.com/cosmos/cosmos-sdk/types"
 | 
						|
	"github.com/cosmos/cosmos-sdk/version"
 | 
						|
	"github.com/cosmos/cosmos-sdk/x/auth"
 | 
						|
	authcmd "github.com/cosmos/cosmos-sdk/x/auth/client/cli"
 | 
						|
	authrest "github.com/cosmos/cosmos-sdk/x/auth/client/rest"
 | 
						|
	"github.com/cosmos/cosmos-sdk/x/bank"
 | 
						|
	bankcmd "github.com/cosmos/cosmos-sdk/x/bank/client/cli"
 | 
						|
 | 
						|
	"github.com/kava-labs/kava/app"
 | 
						|
	"github.com/kava-labs/kava/migrate/rest_v0_3"
 | 
						|
)
 | 
						|
 | 
						|
func main() {
 | 
						|
	// Configure cobra to sort commands
 | 
						|
	cobra.EnableCommandSorting = false
 | 
						|
 | 
						|
	// Instantiate the codec for the command line application
 | 
						|
	cdc := app.MakeCodec()
 | 
						|
 | 
						|
	// Set the global config
 | 
						|
	// config is not sealed as the cli supports two coin types for legacy reasons.
 | 
						|
	config := sdk.GetConfig()
 | 
						|
	app.SetBech32AddressPrefixes(config)
 | 
						|
	app.SetBip44CoinType(config)
 | 
						|
 | 
						|
	// TODO: setup keybase, viper object, etc. to be passed into
 | 
						|
	// the below functions and eliminate global vars, like we do
 | 
						|
	// with the cdc
 | 
						|
 | 
						|
	rootCmd := &cobra.Command{
 | 
						|
		Use:   "kvcli",
 | 
						|
		Short: "Command line interface for interacting with kvd",
 | 
						|
	}
 | 
						|
 | 
						|
	// Add --chain-id to persistent flags and mark it required
 | 
						|
	rootCmd.PersistentFlags().String(flags.FlagChainID, "", "Chain ID of tendermint node")
 | 
						|
	rootCmd.PersistentPreRunE = func(_ *cobra.Command, _ []string) error {
 | 
						|
		return initConfig(rootCmd)
 | 
						|
	}
 | 
						|
 | 
						|
	// Construct Root Command
 | 
						|
	rootCmd.AddCommand(
 | 
						|
		rpc.StatusCommand(),
 | 
						|
		client.ConfigCmd(app.DefaultCLIHome),
 | 
						|
		queryCmd(cdc),
 | 
						|
		txCmd(cdc),
 | 
						|
		flags.LineBreak,
 | 
						|
		lcd.ServeCommand(cdc, registerRoutes),
 | 
						|
		flags.LineBreak,
 | 
						|
		getModifiedKeysCmd(),
 | 
						|
		flags.LineBreak,
 | 
						|
		version.Cmd,
 | 
						|
		flags.NewCompletionCmd(rootCmd, true),
 | 
						|
	)
 | 
						|
 | 
						|
	// Add flags and prefix all env exposed with KA
 | 
						|
	executor := cli.PrepareMainCmd(rootCmd, "KA", app.DefaultCLIHome)
 | 
						|
 | 
						|
	err := executor.Execute()
 | 
						|
	if err != nil {
 | 
						|
		fmt.Printf("Failed executing CLI command: %s, exiting...\n", err)
 | 
						|
		os.Exit(1)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func queryCmd(cdc *codec.Codec) *cobra.Command {
 | 
						|
	queryCmd := &cobra.Command{
 | 
						|
		Use:     "query",
 | 
						|
		Aliases: []string{"q"},
 | 
						|
		Short:   "Querying subcommands",
 | 
						|
	}
 | 
						|
 | 
						|
	queryCmd.AddCommand(
 | 
						|
		authcmd.GetAccountCmd(cdc),
 | 
						|
		flags.LineBreak,
 | 
						|
		rpc.ValidatorCommand(cdc),
 | 
						|
		rpc.BlockCommand(),
 | 
						|
		authcmd.QueryTxsByEventsCmd(cdc),
 | 
						|
		authcmd.QueryTxCmd(cdc),
 | 
						|
		flags.LineBreak,
 | 
						|
	)
 | 
						|
 | 
						|
	// add modules' query commands
 | 
						|
	app.ModuleBasics.AddQueryCommands(queryCmd, cdc)
 | 
						|
 | 
						|
	return queryCmd
 | 
						|
}
 | 
						|
 | 
						|
func txCmd(cdc *codec.Codec) *cobra.Command {
 | 
						|
	txCmd := &cobra.Command{
 | 
						|
		Use:   "tx",
 | 
						|
		Short: "Transactions subcommands",
 | 
						|
	}
 | 
						|
 | 
						|
	txCmd.AddCommand(
 | 
						|
		bankcmd.SendTxCmd(cdc),
 | 
						|
		flags.LineBreak,
 | 
						|
		authcmd.GetSignCommand(cdc),
 | 
						|
		authcmd.GetMultiSignCommand(cdc),
 | 
						|
		flags.LineBreak,
 | 
						|
		authcmd.GetBroadcastCommand(cdc),
 | 
						|
		authcmd.GetEncodeCommand(cdc),
 | 
						|
		authcmd.GetDecodeCommand(cdc),
 | 
						|
		flags.LineBreak,
 | 
						|
	)
 | 
						|
 | 
						|
	// add modules' tx commands
 | 
						|
	app.ModuleBasics.AddTxCommands(txCmd, cdc)
 | 
						|
 | 
						|
	// remove auth and bank commands as they're mounted under the root tx command
 | 
						|
	var cmdsToRemove []*cobra.Command
 | 
						|
	for _, cmd := range txCmd.Commands() {
 | 
						|
		if cmd.Use == auth.ModuleName || cmd.Use == bank.ModuleName {
 | 
						|
			cmdsToRemove = append(cmdsToRemove, cmd)
 | 
						|
		}
 | 
						|
	}
 | 
						|
	txCmd.RemoveCommand(cmdsToRemove...)
 | 
						|
 | 
						|
	return txCmd
 | 
						|
}
 | 
						|
 | 
						|
// registerRoutes registers the routes from the different modules for the LCD.
 | 
						|
func registerRoutes(rs *lcd.RestServer) {
 | 
						|
	client.RegisterRoutes(rs.CliCtx, rs.Mux)
 | 
						|
	authrest.RegisterTxRoutes(rs.CliCtx, rs.Mux)
 | 
						|
	app.ModuleBasics.RegisterRESTRoutes(rs.CliCtx, rs.Mux)
 | 
						|
	// register legacy endpoints compatible with v0.3.x of kava
 | 
						|
	rest_v0_3.RegisterRoutes(rs.CliCtx, rs.Mux)
 | 
						|
}
 | 
						|
 | 
						|
// initConfig reads in and sets options from a config file (if one exists)
 | 
						|
func initConfig(cmd *cobra.Command) error {
 | 
						|
	home, err := cmd.PersistentFlags().GetString(cli.HomeFlag)
 | 
						|
	if err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	cfgFile := path.Join(home, "config", "config.toml")
 | 
						|
 | 
						|
	if _, err := os.Stat(cfgFile); err == nil {
 | 
						|
		viper.SetConfigFile(cfgFile)
 | 
						|
 | 
						|
		if err := viper.ReadInConfig(); err != nil {
 | 
						|
			return err
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if err := viper.BindPFlag(flags.FlagChainID, cmd.PersistentFlags().Lookup(flags.FlagChainID)); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	if err := viper.BindPFlag(cli.EncodingFlag, cmd.PersistentFlags().Lookup(cli.EncodingFlag)); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return viper.BindPFlag(cli.OutputFlag, cmd.PersistentFlags().Lookup(cli.OutputFlag))
 | 
						|
}
 |