[CLI] Add --rest optional flag to init command (#363)

This commit is contained in:
Louis Singer
2024-10-31 18:01:19 +01:00
committed by GitHub
parent 786a69da7d
commit 04e57f8179

View File

@@ -127,6 +127,12 @@ var (
Name: "force", Name: "force",
Usage: "force redemption without collaboration", Usage: "force redemption without collaboration",
} }
restFlag = &cli.BoolFlag{
Name: "rest",
Usage: "use REST client instead of gRPC",
Value: false,
DefaultText: "false",
}
) )
var ( var (
@@ -136,7 +142,7 @@ var (
Action: func(ctx *cli.Context) error { Action: func(ctx *cli.Context) error {
return initArkSdk(ctx) return initArkSdk(ctx)
}, },
Flags: []cli.Flag{networkFlag, passwordFlag, privateKeyFlag, urlFlag, explorerFlag}, Flags: []cli.Flag{networkFlag, passwordFlag, privateKeyFlag, urlFlag, explorerFlag, restFlag},
} }
configCommand = cli.Command{ configCommand = cli.Command{
Name: "config", Name: "config",
@@ -200,9 +206,14 @@ func initArkSdk(ctx *cli.Context) error {
return err return err
} }
clientType := arksdk.GrpcClient
if ctx.Bool(restFlag.Name) {
clientType = arksdk.RestClient
}
return arkSdkClient.Init( return arkSdkClient.Init(
ctx.Context, arksdk.InitArgs{ ctx.Context, arksdk.InitArgs{
ClientType: arksdk.GrpcClient, ClientType: clientType,
WalletType: arksdk.SingleKeyWallet, WalletType: arksdk.SingleKeyWallet,
AspUrl: ctx.String(urlFlag.Name), AspUrl: ctx.String(urlFlag.Name),
Seed: ctx.String(privateKeyFlag.Name), Seed: ctx.String(privateKeyFlag.Name),