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

AreaLayer · Oct 31, 2024 · 04e57f8 · 04e57f8
1 parent 786a69d
commit 04e57f8
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/client/main.go b/client/main.go
@@ -127,6 +127,12 @@ var (
 		Name:  "force",
 		Usage: "force redemption without collaboration",
 	}
+	restFlag = &cli.BoolFlag{
+		Name:        "rest",
+		Usage:       "use REST client instead of gRPC",
+		Value:       false,
+		DefaultText: "false",
+	}
 )
 
 var (
@@ -136,7 +142,7 @@ var (
 		Action: func(ctx *cli.Context) error {
 			return initArkSdk(ctx)
 		},
-		Flags: []cli.Flag{networkFlag, passwordFlag, privateKeyFlag, urlFlag, explorerFlag},
+		Flags: []cli.Flag{networkFlag, passwordFlag, privateKeyFlag, urlFlag, explorerFlag, restFlag},
 	}
 	configCommand = cli.Command{
 		Name:  "config",
@@ -200,9 +206,14 @@ func initArkSdk(ctx *cli.Context) error {
 		return err
 	}
 
+	clientType := arksdk.GrpcClient
+	if ctx.Bool(restFlag.Name) {
+		clientType = arksdk.RestClient
+	}
+
 	return arkSdkClient.Init(
 		ctx.Context, arksdk.InitArgs{
-			ClientType:  arksdk.GrpcClient,
+			ClientType:  clientType,
 			WalletType:  arksdk.SingleKeyWallet,
 			AspUrl:      ctx.String(urlFlag.Name),
 			Seed:        ctx.String(privateKeyFlag.Name),