Files
clio/tools/snapshot/internal/parse_args/parse_args.go
2025-05-20 10:08:17 -04:00

96 lines
2.9 KiB
Go

package parse_args
import (
"fmt"
"os"
flag "github.com/spf13/pflag"
)
type Args struct {
ExportMode string
StartSeq uint32
EndSeq uint32
Path string
GrpcServer string
WsServer string
ServerMode bool
ShowRange bool
GetRange bool
// for authenticating DB
Host string
Username string
Password string
}
func Parse() (*Args, error) {
fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
fs.Usage = PrintUsage
exportMode := fs.StringP("export", "e", "", "Set export mode: 'full' (export full ledger data of start_seq) or 'delta' (only export ledger diff data)")
seq := fs.Uint32("start_seq", 0, "Starting sequence number")
endSeq := fs.Uint32("end_seq", 0, "Ending sequence number")
path := fs.StringP("path", "p", "", "Path to the data")
grpcServer := fs.StringP("grpc_server", "g", "0.0.0.0:50051", "rippled's gRPC server address")
wsServer := fs.StringP("ws_server", "w", "0.0.0.0:6006", "rippled's gRPC server address")
serverMode := fs.BoolP("server", "s", false, "Start server mode")
showRange := fs.BoolP("range", "r", false, "Show the range of the snapshot")
getRange := fs.Bool("getRange", false, "Get the available DB range of the host")
host := fs.String("host", "", "Database host (e.g., 127.0.0.1)")
username := fs.String("username", "", "Database username")
password := fs.String("password", "", "Database password")
fs.Parse(os.Args[1:])
if *serverMode && *exportMode != "" {
return nil, fmt.Errorf("invalid usage: --server and --export cannot be used at the same time")
}
if *serverMode {
if *grpcServer == "" || *wsServer == "" || *path == "" {
return nil, fmt.Errorf("invalid usage: --grpc_server and --ws_server and --path are required for server mode")
}
} else if *exportMode != "" {
if *exportMode == "full" || *exportMode == "delta" {
if *seq == 0 || *endSeq == 0 || *path == "" || *grpcServer == "" {
return nil, fmt.Errorf("invalid usage: --start_seq, --end_seq, --grpc_server and --path are required for export")
}
} else {
return nil, fmt.Errorf("invalid usage: Invalid export mode. Use 'full' or 'delta'")
}
} else if *showRange {
if *path == "" {
return nil, fmt.Errorf("invalid usage: --path is required for show range")
}
} else if *getRange {
if *host == "" || *username == "" || *password == "" {
return nil, fmt.Errorf("invalid usage: --host, --username, and --password are required for getRange")
}
} else {
return nil, fmt.Errorf("invalid usage: --export or --server or --range flag is required")
}
return &Args{
ExportMode: *exportMode,
StartSeq: *seq,
EndSeq: *endSeq,
Path: *path,
GrpcServer: *grpcServer,
WsServer: *wsServer,
ServerMode: *serverMode,
ShowRange: *showRange,
GetRange: *getRange,
Host: *host,
Username: *username,
Password: *password,
}, nil
}
func PrintUsage() {
fmt.Println("Usage: clio_snapshot [options]")
fmt.Println("Options:")
flag.PrintDefaults()
}