mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 10:05:18 +00:00
116 lines
3.5 KiB
Go
116 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
dbm "github.com/tendermint/tm-db"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/libs/cli"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
tmtypes "github.com/tendermint/tendermint/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
|
"github.com/cosmos/cosmos-sdk/server"
|
|
"github.com/cosmos/cosmos-sdk/store"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
genutilcli "github.com/cosmos/cosmos-sdk/x/genutil/client/cli"
|
|
"github.com/cosmos/cosmos-sdk/x/staking"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
)
|
|
|
|
// kvd custom flags
|
|
const flagInvCheckPeriod = "inv-check-period"
|
|
|
|
var invCheckPeriod uint
|
|
|
|
func main() {
|
|
cdc := app.MakeCodec()
|
|
|
|
config := sdk.GetConfig()
|
|
app.SetBech32AddressPrefixes(config)
|
|
app.SetBip44CoinType(config)
|
|
config.Seal()
|
|
|
|
ctx := server.NewDefaultContext()
|
|
cobra.EnableCommandSorting = false
|
|
rootCmd := &cobra.Command{
|
|
Use: "kvd",
|
|
Short: "Kava Daemon (server)",
|
|
PersistentPreRunE: server.PersistentPreRunEFn(ctx),
|
|
}
|
|
|
|
rootCmd.AddCommand(
|
|
genutilcli.InitCmd(ctx, cdc, app.ModuleBasics, app.DefaultNodeHome),
|
|
genutilcli.CollectGenTxsCmd(ctx, cdc, auth.GenesisAccountIterator{}, app.DefaultNodeHome),
|
|
genutilcli.MigrateGenesisCmd(ctx, cdc),
|
|
genutilcli.GenTxCmd(
|
|
ctx,
|
|
cdc,
|
|
app.ModuleBasics,
|
|
staking.AppModuleBasic{},
|
|
auth.GenesisAccountIterator{},
|
|
app.DefaultNodeHome,
|
|
app.DefaultCLIHome),
|
|
genutilcli.ValidateGenesisCmd(ctx, cdc, app.ModuleBasics),
|
|
AddGenesisAccountCmd(ctx, cdc, app.DefaultNodeHome, app.DefaultCLIHome),
|
|
flags.NewCompletionCmd(rootCmd, true),
|
|
)
|
|
|
|
server.AddCommands(ctx, cdc, rootCmd, newApp, exportAppStateAndTMValidators)
|
|
|
|
// prepare and add flags
|
|
executor := cli.PrepareBaseCmd(rootCmd, "KA", app.DefaultNodeHome)
|
|
rootCmd.PersistentFlags().UintVar(&invCheckPeriod, flagInvCheckPeriod,
|
|
0, "Assert registered invariants every N blocks")
|
|
err := executor.Execute()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func newApp(logger log.Logger, db dbm.DB, traceStore io.Writer) abci.Application {
|
|
var cache sdk.MultiStorePersistentCache
|
|
|
|
if viper.GetBool(server.FlagInterBlockCache) {
|
|
cache = store.NewCommitKVStoreCacheManager()
|
|
}
|
|
|
|
skipUpgradeHeights := make(map[int64]bool)
|
|
for _, h := range viper.GetIntSlice(server.FlagUnsafeSkipUpgrades) {
|
|
skipUpgradeHeights[int64(h)] = true
|
|
}
|
|
|
|
return app.NewApp(
|
|
logger, db, traceStore, true, skipUpgradeHeights, invCheckPeriod,
|
|
baseapp.SetPruning(store.NewPruningOptionsFromString(viper.GetString("pruning"))),
|
|
baseapp.SetMinGasPrices(viper.GetString(server.FlagMinGasPrices)),
|
|
baseapp.SetHaltHeight(viper.GetUint64(server.FlagHaltHeight)),
|
|
baseapp.SetHaltTime(viper.GetUint64(server.FlagHaltTime)),
|
|
baseapp.SetInterBlockCache(cache),
|
|
)
|
|
}
|
|
|
|
func exportAppStateAndTMValidators(
|
|
logger log.Logger, db dbm.DB, traceStore io.Writer, height int64, forZeroHeight bool, jailWhiteList []string,
|
|
) (json.RawMessage, []tmtypes.GenesisValidator, error) {
|
|
|
|
if height != -1 {
|
|
tempApp := app.NewApp(logger, db, traceStore, false, map[int64]bool{}, uint(1))
|
|
err := tempApp.LoadHeight(height)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return tempApp.ExportAppStateAndValidators(forZeroHeight, jailWhiteList)
|
|
}
|
|
tempApp := app.NewApp(logger, db, traceStore, true, map[int64]bool{}, uint(1))
|
|
return tempApp.ExportAppStateAndValidators(forZeroHeight, jailWhiteList)
|
|
}
|