mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-23 12:07:17 +00:00
73 lines
2.4 KiB
Go
73 lines
2.4 KiB
Go
// Package cli contains framework for building a command-line based Erigon node.
|
|
package cli
|
|
|
|
import (
|
|
"github.com/ledgerwatch/erigon/cmd/utils"
|
|
"github.com/ledgerwatch/erigon/internal/debug"
|
|
"github.com/ledgerwatch/erigon/internal/flags"
|
|
"github.com/ledgerwatch/erigon/node"
|
|
"github.com/ledgerwatch/erigon/params"
|
|
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
// MakeApp creates a cli application (based on `github.com/urlfave/cli` package).
|
|
// The application exits when `action` returns.
|
|
// Parameters:
|
|
// * action: the main function for the application. receives `*cli.Context` with parsed command-line flags. Returns no error, if some error could not be recovered from write to the log or panic.
|
|
// * cliFlags: the list of flags `cli.Flag` that the app should set and parse. By default, use `DefaultFlags()`. If you want to specify your own flag, use `append(DefaultFlags(), myFlag)` for this parameter.
|
|
func MakeApp(action func(*cli.Context), cliFlags []cli.Flag) *cli.App {
|
|
app := flags.NewApp("", "", "erigon experimental cli")
|
|
app.Action = action
|
|
app.Flags = append(cliFlags, debug.Flags...) // debug flags are required
|
|
app.Before = func(ctx *cli.Context) error {
|
|
return debug.Setup(ctx)
|
|
}
|
|
app.After = func(ctx *cli.Context) error {
|
|
debug.Exit()
|
|
return nil
|
|
}
|
|
app.Commands = []cli.Command{initCommand, snapshotCommand}
|
|
return app
|
|
}
|
|
|
|
func MigrateFlags(action func(ctx *cli.Context) error) func(*cli.Context) error {
|
|
return func(ctx *cli.Context) error {
|
|
for _, name := range ctx.FlagNames() {
|
|
if ctx.IsSet(name) {
|
|
ctx.GlobalSet(name, ctx.String(name))
|
|
}
|
|
}
|
|
return action(ctx)
|
|
}
|
|
}
|
|
|
|
func NewNodeConfig(ctx *cli.Context) *node.Config {
|
|
nodeConfig := node.DefaultConfig
|
|
// see simiar changes in `cmd/geth/config.go#defaultNodeConfig`
|
|
if commit := params.GitCommit; commit != "" {
|
|
nodeConfig.Version = params.VersionWithCommit(commit, "")
|
|
} else {
|
|
nodeConfig.Version = params.Version
|
|
}
|
|
nodeConfig.IPCPath = "" // force-disable IPC endpoint
|
|
nodeConfig.Name = "erigon"
|
|
if ctx.GlobalIsSet(utils.DataDirFlag.Name) {
|
|
nodeConfig.DataDir = ctx.GlobalString(utils.DataDirFlag.Name)
|
|
}
|
|
return &nodeConfig
|
|
}
|
|
|
|
func MakeConfigNodeDefault(ctx *cli.Context) *node.Node {
|
|
return makeConfigNode(NewNodeConfig(ctx))
|
|
}
|
|
|
|
func makeConfigNode(config *node.Config) *node.Node {
|
|
stack, err := node.New(config)
|
|
if err != nil {
|
|
utils.Fatalf("Failed to create Erigon node: %v", err)
|
|
}
|
|
|
|
return stack
|
|
}
|